barrier.hpp

Go to the documentation of this file.

 
#ifndef PARALLEL_CELLULAR_AUTOMATA_BARRIER_HPP
#define PARALLEL_CELLULAR_AUTOMATA_BARRIER_HPP
#include <condition_variable>
#include <functional>
#include <mutex>
#include <thread>
#include <vector>
 
namespace ca
{
class Barrier
{
  private:
    std::mutex m;
    std::condition_variable cond;
    const unsigned n_threads;
    unsigned count;
 
    // to make the barrier reusable first we count down and then up.
    enum Direction : unsigned char
    {
        DECREASING,
        INCREASING
    };
 
    Direction direction;
 
  public:
    explicit Barrier(unsigned n);
    void wait();
    void wait(std::function<void()> fun);
 
    void busy_wait();
 
    void busy_wait(std::function<void()> fun);
};
} // namespace ca
#endif