#pragma once

#include "CriticalSection.h"
#include "LogClass.h"

class CTraceModule
{
    TraceLevel m_Type;

public:
    CTraceModule () {  m_Type = TrLvError; }
    virtual ~CTraceModule () {}

    inline  void       SetTraceLevel ( TraceLevel Type ) { m_Type = Type; }
    inline  TraceLevel GetTraceLevel ( void ) const     { return m_Type; }
    virtual void       Write         ( const char * Message, bool EndOfLine ) = 0;
};

class CTraceFileLog : public CTraceModule
{
    enum { MB = 1024 * 1024 };

    CriticalSection m_CriticalSection;
    CLog            m_hLogFile;
    bool            m_FlushFile;

public:
    CTraceFileLog (const char * FileName, bool FlushFile = true);
    CTraceFileLog(const char * FileName, bool FlushFile, LOG_OPEN_MODE eMode, size_t dwMaxFileSize = 5);
    virtual ~CTraceFileLog ();

    void Write ( const char * Message, bool EndOfLine );
    void SetFlushFile ( bool bFlushFile );
};

class CDebugTraceLog : public CTraceModule
{
public:
    void Write ( const char * Message, bool EndOfLine );
};

CTraceModule * AddTraceModule    ( CTraceModule * TraceModule ); // Must be created with new
CTraceModule * RemoveTraceModule ( CTraceModule * TraceModule ); // Is not automaticly deleted