mlBarrier.h

Go to the documentation of this file.

/*************************************************************************************
**
** Copyright 2009, MeVis Medical Solutions AG
**
** The user may use this file in accordance with the license agreement provided with
** the Software or, alternatively, in accordance with the terms contained in a
** written agreement between the user and MeVis Medical Solutions AG.
**
** For further information use the contact form at https://www.mevislab.de/contact
**
**************************************************************************************/
 
#ifndef ML_BARRIER_H
#define ML_BARRIER_H

 
#include "mlUtilsSystem.h"
 
#include "mlMutex.h"
#include "mlWaitCondition.h"
 
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