MeVis/Foundation/Sources/MLUtilities/mlBarrier.h

Go to the documentation of this file.

// **InsertLicense** code
//----------------------------------------------------------------------------------

//----------------------------------------------------------------------------------
#ifndef __mlBarrier_H
#define __mlBarrier_H

#ifndef __mlUtilsSystem_H
#include "mlUtilsSystem.h"
#endif

#ifndef __mlMutex_H
#include "mlMutex.h"
#endif
#ifndef __mlWaitCondition_H
#include "mlWaitCondition.h"
#endif

ML_UTILS_START_NAMESPACE

//----------------------------------------------------------------------------------
//----------------------------------------------------------------------------------
class ML_UTILS_EXPORT Barrier {

public:
  //----------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------
  Barrier(unsigned int count = 0);

  //----------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------
  void setNumberOfThreads(unsigned int count);

  //----------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------
  bool wait();

private:
  Barrier(const Barrier&);
  Barrier& operator =(const Barrier&);

  unsigned int _count;
  unsigned int _generation;
  unsigned int _numberOfThreads;

  Mutex _mutex;
  WaitCondition _finishedNotifier;
};

//----------------------------------------------------------------------------------
//----------------------------------------------------------------------------------
class ML_UTILS_EXPORT ScopedBarrierWaiter {

public:
  //----------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------
  ScopedBarrierWaiter(Barrier& barrier, unsigned int numberOfExpectedWaits = 1);
  ~ScopedBarrierWaiter();

  //----------------------------------------------------------------------------------
  //----------------------------------------------------------------------------------
  void wait();

private:
  ScopedBarrierWaiter(const ScopedBarrierWaiter&);
  ScopedBarrierWaiter& operator =(const ScopedBarrierWaiter&);

  Barrier& _barrier;
  unsigned int _numberOfExpectedWaits;
};

ML_UTILS_END_NAMESPACE

#endif // __mlBarrier_H