/*++

Copyright (c) 1995  Microsoft Corporation

Module Name:

    cs.h

Abstract:

    Declaration of Critical Section class

Author:

    Ronit Hartmann (ronith) ??-???-??

--*/

#ifndef __CS_H
#define __CS_H

//------------------------
//
//  class CCriticalSection
//
class CCriticalSection
{
    friend class CS;
    friend class ConditionalCS;

public:
    CCriticalSection(){InitializeCriticalSection(&_cs);}
    ~CCriticalSection(){DeleteCriticalSection(&_cs);}
    void Enter() {EnterCriticalSection(&_cs);}
    void Leave() {LeaveCriticalSection(&_cs);}

private:
    void Lock()     { Enter(); }
    void Unlock()   { Leave(); }

private:
    CRITICAL_SECTION _cs;
};


//------------------------
//
//  class CS
//
class CS {
    CCriticalSection* m_lock;

public:
    CS(CCriticalSection& lock);
   ~CS();
};

//- implementation -------
//
//  class CS
//
inline CS::CS(CCriticalSection& lock) :
    m_lock(&lock)
{
    m_lock->Lock();
}

inline CS::~CS()
{
    m_lock->Unlock();
}


//-----------------------------
//
//  ConditionalCS class
//
class ConditionalCS
{
public:
    ConditionalCS( CCriticalSection & cs) : m_cs(&cs),
                                         m_fLocked(FALSE)
    {}
    void Lock();
    ~ConditionalCS();
    
private:
    CCriticalSection *      m_cs;
    BOOL                    m_fLocked;


};

inline ConditionalCS::~ConditionalCS()
{
    if ( m_fLocked)
    {
        m_cs->Unlock();
    }
}

inline void ConditionalCS::Lock()
{
     m_fLocked = TRUE;
     m_cs->Lock();
}


#endif // __CS_H