Barrier.h

Go to the documentation of this file.

 
#pragma once
 
#include <condition_variable>
#include <cstdint>
#include <mutex>
 
namespace ogdf {
 
 
class Barrier {
    std::condition_variable m_allThreadsReachedSync;
    std::mutex m_numThreadsReachedSyncLock;
 
    uint32_t m_threadCount; 
    uint32_t m_numThreadsReachedSync; 
    uint32_t m_syncNumber; 
 
public:
    explicit Barrier(uint32_t numThreads) : m_threadCount(numThreads) {
        m_numThreadsReachedSync = 0;
        m_syncNumber = 0;
    }
 
 
    void threadSync() {
        std::unique_lock<std::mutex> lk(m_numThreadsReachedSyncLock);
 
        uint32_t syncNr = m_syncNumber;
        m_numThreadsReachedSync++;
        if (m_numThreadsReachedSync == m_threadCount) {
            m_syncNumber++;
            m_allThreadsReachedSync.notify_all();
            m_numThreadsReachedSync = 0;
 
        } else {
            m_allThreadsReachedSync.wait(lk, [syncNr, this] { return syncNr != m_syncNumber; });
        }
    }
};
 
}