F:/My Projects/LSEngine/Modules/LSThreadLib/Src/CriticalSection/LSHCriticalSection.h

#ifndef __LSH_CRITICALSECTION_H__
#define __LSH_CRITICALSECTION_H__

#include "../LSHThreadLib.h"


// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// MACROS
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
//#define LSH_SINGLE_THREADED


namespace lsh {

        class CCriticalSection {
        public :
                // == Various constructors.
                LSE_INLINE LSE_CALL                                                     CCriticalSection();
                LSE_INLINE LSE_CALL                                                     ~CCriticalSection();


                // == Types.
                class CLocker {
                public :
                        // == Various constructors.
                        LSE_INLINE LSE_CALL                                             CLocker( CCriticalSection &_csLockMe ) {
                                m_pcsSection = &_csLockMe;
                                m_pcsSection->Lock();
                        }
                        LSE_INLINE LSE_CALL                                             ~CLocker() {
                                m_pcsSection->UnLock();
                        }

                private :
                        // == Members.
                        CCriticalSection *                                              m_pcsSection;
                };

                class CLockerS {
                public :
                        // == Various constructors.
                        LSE_INLINE LSE_CALL                                             CLockerS( CCriticalSection &_csLockMe ) {
#ifndef LSH_SINGLE_THREADED
                                m_pcsSection = &_csLockMe;
                                m_pcsSection->Lock();
#else   // #ifndef LSH_SINGLE_THREADED
                                // Remove warning C4100.
                                static_cast<CCriticalSection &>(_csLockMe);
#endif  // #ifndef LSH_SINGLE_THREADED
                        }
                        LSE_INLINE LSE_CALL                                             ~CLockerS() {
#ifndef LSH_SINGLE_THREADED
                                m_pcsSection->UnLock();
#endif  // #ifndef LSH_SINGLE_THREADED
                        }

                private :
#ifndef LSH_SINGLE_THREADED
                        // == Members.
                        // The section we lock and unlock.
                        CCriticalSection *                                              m_pcsSection;
#endif  // #ifndef LSH_SINGLE_THREADED
                };


                // == Functions.
                // Lock the critical section.
                LSE_INLINE LSVOID LSE_CALL                                      Lock();

                // Unlock the critical section.
                LSE_INLINE LSVOID LSE_CALL                                      UnLock();

        protected :
                // == Members.
#ifdef LSE_WIN32
                CRITICAL_SECTION                                                        m_csCrit;
#elif defined( LSE_MAC )
                pthread_mutex_t                                                         m_ptmCrit;
#endif  // #ifdef LSE_WIN32
        };


        // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
        // DEFINITIONS
        // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
        // == Various constructors.
        LSE_INLINE LSE_CALL CCriticalSection::CCriticalSection() {
#ifdef LSE_WIN32
                ::InitializeCriticalSection( &m_csCrit );
#elif defined( LSE_MAC )
                pthread_mutexattr_t tmaAtt;
                ::pthread_mutexattr_init( &tmaAtt );
                ::pthread_mutexattr_settype( &tmaAtt, PTHREAD_MUTEX_RECURSIVE );
                ::pthread_mutex_init( &m_ptmCrit, &tmaAtt );
                ::pthread_mutexattr_destroy( &tmaAtt );
#endif  // #ifdef LSE_WIN32
        }
        LSE_INLINE LSE_CALL CCriticalSection::~CCriticalSection() {
#ifdef LSE_WIN32
                ::DeleteCriticalSection( &m_csCrit );
#elif defined( LSE_MAC )
                ::pthread_mutex_destroy( &m_ptmCrit );
#endif  // #ifdef LSE_WIN32
        }

        // == Functions.
        // Lock the critical section.
        LSE_INLINE LSVOID LSE_CALL CCriticalSection::Lock() {
#ifdef LSE_WIN32
                ::EnterCriticalSection( &m_csCrit );
#elif defined( LSE_MAC )
                ::pthread_mutex_lock( &m_ptmCrit );
#endif  // #ifdef LSE_WIN32
        }

        // Unlock the critical section.
        LSE_INLINE LSVOID LSE_CALL CCriticalSection::UnLock() {
#ifdef LSE_WIN32
                ::LeaveCriticalSection( &m_csCrit );
#elif defined( LSE_MAC )
                ::pthread_mutex_unlock( &m_ptmCrit );
#endif  // #ifdef LSE_WIN32
        }

}       // namespace lsh

#endif  // __LSH_CRITICALSECTION_H__