mirror of
				https://github.com/OpenMW/openmw.git
				synced 2025-10-27 03:26:38 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			51 lines
		
	
	
	
		
			1.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			51 lines
		
	
	
	
		
			1.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #ifndef OPENMW_BARRIER_H
 | |
| #define OPENMW_BARRIER_H
 | |
| 
 | |
| #include <condition_variable>
 | |
| #include <functional>
 | |
| #include <mutex>
 | |
| 
 | |
| namespace Misc
 | |
| {
 | |
|     /// @brief Synchronize several threads
 | |
|     class Barrier
 | |
|     {
 | |
|         public:
 | |
|             using BarrierCallback = std::function<void(void)>;
 | |
|             /// @param count number of threads to wait on
 | |
|             /// @param func callable to be executed once after all threads have met
 | |
|             Barrier(int count, BarrierCallback&& func) : mThreadCount(count), mRendezvousCount(0), mGeneration(0)
 | |
|                                                        , mFunc(std::forward<BarrierCallback>(func))
 | |
|             {}
 | |
| 
 | |
|             /// @brief stop execution of threads until count distinct threads reach this point
 | |
|             void wait()
 | |
|             {
 | |
|                 std::unique_lock lock(mMutex);
 | |
| 
 | |
|                 ++mRendezvousCount;
 | |
|                 const int currentGeneration = mGeneration;
 | |
|                 if (mRendezvousCount == mThreadCount)
 | |
|                 {
 | |
|                     ++mGeneration;
 | |
|                     mRendezvousCount = 0;
 | |
|                     mFunc();
 | |
|                     mRendezvous.notify_all();
 | |
|                 }
 | |
|                 else
 | |
|                 {
 | |
|                     mRendezvous.wait(lock, [&]() { return mGeneration != currentGeneration; });
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|         private:
 | |
|             int mThreadCount;
 | |
|             int mRendezvousCount;
 | |
|             int mGeneration;
 | |
|             mutable std::mutex mMutex;
 | |
|             std::condition_variable mRendezvous;
 | |
|             BarrierCallback mFunc;
 | |
|     };
 | |
| }
 | |
| 
 | |
| #endif
 |