AutoMessage.h

#ifndef _AUTOMESSAGE_H_
#define DISABLE_ASSERTS
#define _AUTOMESSAGE_H_

#include "LTClient.h"
#include "iltmessage.h"
#include <Windows.h>
#define DISABLE_ASSERTS
#include <d3d9.h>

class ILTMessage_Write;
class ILTMessage_Read;

//extern CCommonLT* g_CommonLT;

static class CCommonLT* g_CommonLT;

class CAutoMessage
{
public:
    CAutoMessage() : m_pMsg(NULL)
    {
        Init();
    }
    ~CAutoMessage()
    {
        Term();
    }
    CAutoMessage(const ILTMessage_Read &cMsg) : m_pMsg(NULL)
    {
        Init();
        CLTMsgRef_Read cReadMsg(cMsg.Clone());
        WriteMessage(cReadMsg);
    }
    CAutoMessage(const char *pStr) : m_pMsg(NULL)
    {
        Init();
        WriteString(pStr);
    }
    CAutoMessage(const wchar_t *pStr) : m_pMsg(NULL)
    {
        Init();
        WriteWString(pStr);
    }
    template <class T>
    CAutoMessage(const T &tValue) : m_pMsg(NULL)
    {
        Init();
        WriteType(tValue);
    }

    void Reset() { Init(); }
    bool IsValid() { return m_pMsg != NULL ; }

    inline operator ILTMessage_Write*() { return m_pMsg; }
    inline operator const ILTMessage_Write*() const { return m_pMsg; }
    inline operator ILTMessage_Write&() { return *m_pMsg; }
    inline operator const ILTMessage_Write&() const { return *m_pMsg; }

    inline CLTMsgRef_Read Read() { return CLTMsgRef_Read(m_pMsg->Read()); }
    inline uint32 Size() const { return m_pMsg->Size(); }
    inline void WriteBits(uint32 nValue, uint32 nSize) { m_pMsg->WriteBits(nValue, nSize); }
    inline void WriteBits64(uint64 nValue, uint32 nSize) { m_pMsg->WriteBits64(nValue, nSize); }
    inline void WriteData(const void *pData, uint32 nSize) { m_pMsg->WriteData(pData, nSize); }
    inline void WriteMessage(ILTMessage_Read *pMsg) { m_pMsg->WriteMessage(pMsg); }
    inline void WriteMessageRaw(ILTMessage_Read *pMsg) { m_pMsg->WriteMessageRaw(pMsg); }
    inline void WriteString(const char *pString) { m_pMsg->WriteString(pString); }
    inline void WriteWString(const wchar_t *pString) { m_pMsg->WriteWString(pString); }
    inline void WriteCompLTVector(const LTVector &vVec) { m_pMsg->WriteCompLTVector(vVec); }
    inline void WriteCompPos(const LTVector &vPos) { m_pMsg->WriteCompPos(vPos); }
    inline void WriteCompLTRotation(const LTRotation &cRotation) { m_pMsg->WriteCompLTRotation(cRotation); }
    inline void WriteObject(HOBJECT hObj) { m_pMsg->WriteObject(hObj); }
    inline void WriteYRotation(const LTRotation &cRotation) { m_pMsg->WriteYRotation(cRotation); }
    inline void WriteDatabaseRecord( IDatabaseMgr *pDatabase, HRECORD hRecord ) { m_pMsg->WriteDatabaseRecord( pDatabase, hRecord ); }
    inline void Writebool(bool bValue) { WriteBits(bValue ? 1 : 0, 1); }
    inline void Writeuint8(uint8 nValue) { WriteBits(nValue, 8); }
    inline void Writeuint16(uint16 nValue) { WriteBits(nValue, 16); }
    inline void Writeuint32(uint32 nValue) { WriteBits(nValue, 32); }
    inline void Writeuint64(uint64 nValue) { WriteBits64(nValue, 64); }
    inline void Writeint8(int8 nValue) { WriteBits((uint32)nValue, 8); }
    inline void Writeint16(int16 nValue) { WriteBits((uint32)nValue, 16); }
    inline void Writeint32(int32 nValue) { WriteBits((uint32)nValue, 32); }
    inline void Writeint64(int32 nValue) { WriteBits64((uint64)nValue, 32); }
    inline void Writefloat(float fValue) { WriteBits(reinterpret_cast<const uint32&>(fValue), 32); }
    inline void Writedouble(double fValue) { WriteBits64(reinterpret_cast<const uint64&>(fValue), 64); }
    inline void WriteLTVector(const LTVector &vValue) { WriteType(vValue); }
    inline void WriteLTRotation(const LTRotation &cValue) { WriteType(cValue); }
    inline void WriteLTRigidTransform(const LTRigidTransform &tfValue) { WriteType( tfValue ); }
    inline void WriteLTTransform(const LTTransform &tfValue) { WriteType( tfValue ); }
    inline void WriteLTPolarCoord(const LTPolarCoord &polarCoord) { m_pMsg->WriteLTPolarCoord( polarCoord ); }
    inline void WriteCompLTPolarCoord(const LTPolarCoord &polarCoord) { m_pMsg->WriteCompLTPolarCoord( polarCoord ); }
    template <class T>
    inline void WriteType(const T &tValue) { m_pMsg->WriteType(tValue); }
private:
    inline void Init()
    {
        Term();
        m_pMsg = NULL;
        ILTMessage_Write *pMsg;
        LTRESULT nResult = g_CommonLT->CreateMessage(pMsg);
        if (nResult == LT_OK)
            m_pMsg = pMsg;
        ASSERT(nResult == LT_OK);
    }
    inline void Term()
    {
        m_pMsg = NULL;
    }
    CLTMsgRef_Write m_pMsg;
};

#endif