barrier.h

Go to the documentation of this file.

#ifndef ION_PORT_BARRIER_H_
#define ION_PORT_BARRIER_H_

#if defined(ION_PLATFORM_WINDOWS)
#  include <windows.h>
#else
#  include <pthread.h>
#endif

#include "base/integral_types.h"
#include "base/macros.h"
#include "ion/port/atomic.h"

namespace ion {
namespace port {

class Barrier {
 public:
  explicit Barrier(uint32 thread_count);
  ~Barrier();

  bool IsValid() const { return is_valid_; }

  void Wait();

 private:
#if defined(ION_PLATFORM_WINDOWS)
  void WaitInternal(int32 increment, int32 limit, HANDLE turnstile);

  const int32 thread_count_;
  std::atomic<int32> wait_count_;
  HANDLE turnstile1_;
  HANDLE turnstile2_;
#elif defined(ION_PLATFORM_LINUX) || defined(ION_PLATFORM_QNX)
  pthread_barrier_t barrier_;
  std::atomic<int32> waiting_count_;
#else
  pthread_cond_t condition_;
  pthread_cond_t exit_condition_;
  pthread_mutex_t mutex_;
  const int32 thread_count_;
  int32 wait_count_;
  int32 exit_count_;
#endif
  bool is_valid_;

  DISALLOW_COPY_AND_ASSIGN(Barrier);
};

}  // namespace port
}  // namespace ion

#endif  // ION_PORT_BARRIER_H_