msgCompressor.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2010 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/
/*********************************************************************
** $RCSfile: msgCompressor.h,v $ $Revision: $ $State: Exp $
*********************************************************************/


#ifndef msgCompressor_H_
#define msgCompressor_H_

#include <rtiMsConfig.h>
#include <vlutil/vlMachineTypes.h>
#include <vlutil/vlSmartPointers.h>
#include <vector>

#include "msgHeaderReader.h"

struct z_stream_s;
typedef struct z_stream_s* z_streamp;

class DT_DLL_RTIUTIL DtMsgCompressor
{
public:

   DtMsgCompressor(DtBoost::shared_ptr<DtMsgHeaderReader> hdrReader,
      size_t maxUncompressedMsgSize, MAKRti::DtU8 compressionLevel = 1);

   virtual ~DtMsgCompressor();

   virtual bool compressMsg(char* msg, size_t msgSize, char** compressedMsg,
      size_t& compressedSize);

   virtual bool isCompressed(const char* bufferPtr, size_t size) const;

   virtual bool uncompressMsg(char* msg, size_t msgSize, char** uncompressedMsg,
      size_t& uncompressedSize);

   virtual MAKRti::DtU8 compressionLevel() const { return myCompressionLevel; }
   virtual void setCompressionLevel(MAKRti::DtU8 level);

   virtual size_t maxUncompressedMsgSize() const { return myMaxUncompressedMsgSize; }

protected:

   std::vector<char> myCompressedBuffer;

   std::vector<char> myUncompressedBuffer;

   size_t myUncompressedDataSize;

   DtBoost::shared_ptr<DtMsgHeaderReader> myHeaderReader;

   MAKRti::DtU8 myCompressionLevel;

   size_t myMaxUncompressedMsgSize;

   z_streamp myCompressionStream;

   z_streamp myUncompressionStream;

};

#endif //! msgCompressor_H_