/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef MOZILLA_GFX_CRITICALSECTION_H_
#define MOZILLA_GFX_CRITICALSECTION_H_

#ifdef WIN32
#  include <windows.h>
#else
#  include <pthread.h>
#  include "mozilla/DebugOnly.h"
#endif

namespace mozilla {
namespace gfx {

#ifdef WIN32

class CriticalSection {
 public:
  CriticalSection() { ::InitializeCriticalSection(&mCriticalSection); }

  ~CriticalSection() { ::DeleteCriticalSection(&mCriticalSection); }

  void Enter() { ::EnterCriticalSection(&mCriticalSection); }

  void Leave() { ::LeaveCriticalSection(&mCriticalSection); }

 protected:
  CRITICAL_SECTION mCriticalSection;
};

#else
// posix

class PosixCondvar;
class CriticalSection {
 public:
  CriticalSection() {
    DebugOnly<int> err = pthread_mutex_init(&mMutex, nullptr);
    MOZ_ASSERT(!err);
  }

  ~CriticalSection() {
    DebugOnly<int> err = pthread_mutex_destroy(&mMutex);
    MOZ_ASSERT(!err);
  }

  void Enter() {
    DebugOnly<int> err = pthread_mutex_lock(&mMutex);
    MOZ_ASSERT(!err);
  }

  void Leave() {
    DebugOnly<int> err = pthread_mutex_unlock(&mMutex);
    MOZ_ASSERT(!err);
  }

 protected:
  pthread_mutex_t mMutex;
  friend class PosixCondVar;
};

#endif

/// RAII helper.
struct CriticalSectionAutoEnter final {
  explicit CriticalSectionAutoEnter(CriticalSection* aSection)
      : mSection(aSection) {
    mSection->Enter();
  }
  ~CriticalSectionAutoEnter() { mSection->Leave(); }

 protected:
  CriticalSection* mSection;
};

}  // namespace gfx
}  // namespace mozilla

#endif