#include "stdafx.h"
#ifdef _WIN32
#include <Windows.h>
#endif

CriticalSection::CriticalSection()
{
#ifdef _WIN32
    m_cs = new CRITICAL_SECTION;
    ::InitializeCriticalSection((CRITICAL_SECTION *)m_cs);
#else
    m_cs = new pthread_mutex_t;

    pthread_mutexattr_t mAttr;
    pthread_mutexattr_settype(&mAttr, PTHREAD_MUTEX_RECURSIVE_NP);
    pthread_mutex_init((pthread_mutex_t *)m_cs, &mAttr);
    pthread_mutexattr_destroy(&mAttr);
#endif
}

CriticalSection::~CriticalSection(void)
{
#ifdef _WIN32
    ::DeleteCriticalSection((CRITICAL_SECTION *)m_cs);
    delete (CRITICAL_SECTION *)m_cs;
#else
    pthread_mutex_destroy((pthread_mutex_t *)m_cs);
    delete (pthread_mutex_t *)m_cs;
#endif
}

/*
Enters a critical section of code.
Prevents other threads from accessing the section between the enter and leave sections simultaneously.
Note: It is good practice to try and keep the critical section code as little as possible, so that
other threads are not locked waiting for it.
*/

void CriticalSection::enter(void)
{
#ifdef _WIN32
    ::EnterCriticalSection((CRITICAL_SECTION *)m_cs);
#else
    pthread_mutex_lock((pthread_mutex_t *)m_cs);
#endif
}

/*
Leaves the critical section.
Allows threads access to the critical code section again.
Warning: Note that an exception occurring with a critical section may not result in the expected leave being
called.  To ensure that your critical section is exception safe, ensure that you wrap the critical
section in a try catch, and the catch calls the leave method.
*/

void CriticalSection::leave(void)
{
#ifdef _WIN32
    ::LeaveCriticalSection((CRITICAL_SECTION *)m_cs);
#else
    pthread_mutex_unlock((pthread_mutex_t *)m_cs);
#endif
}