1 // OutBuffer.h

     2 

     3 #ifndef __OUTBUFFER_H

     4 #define __OUTBUFFER_H

     5 

     6 #include "../IStream.h"

     7 #include "../../Common/MyCom.h"

     8 #include "../../Common/MyException.h"

     9 

    10 #ifndef _NO_EXCEPTIONS

    11 struct COutBufferException: public CSystemException 

    12 { 

    13   COutBufferException(HRESULT errorCode): CSystemException(errorCode) {} 

    14 };

    15 #endif

    16 

    17 class COutBuffer

    18 {

    19 protected:

    20   Byte *_buffer;

    21   UInt32 _pos;

    22   UInt32 _limitPos;

    23   UInt32 _streamPos;

    24   UInt32 _bufferSize;

    25   CMyComPtr<ISequentialOutStream> _stream;

    26   UInt64 _processedSize;

    27   Byte  *_buffer2;

    28   bool _overDict;

    29 

    30   HRESULT FlushPart();

    31 public:

    32   #ifdef _NO_EXCEPTIONS

    33   HRESULT ErrorCode;

    34   #endif

    35 

    36   COutBuffer(): _buffer(0), _pos(0), _stream(0), _buffer2(0) {}

    37   ~COutBuffer() { Free(); }

    38   

    39   bool Create(UInt32 bufferSize);

    40   void Free();

    41 

    42   void SetMemStream(Byte *buffer) { _buffer2 = buffer; }

    43   void SetStream(ISequentialOutStream *stream);

    44   void Init();

    45   HRESULT Flush();

    46   void FlushWithCheck();

    47   void ReleaseStream() {  _stream.Release(); }

    48 

    49   void WriteByte(Byte b)

    50   {

    51     _buffer[_pos++] = b;

    52     if(_pos == _limitPos)

    53       FlushWithCheck();

    54   }

    55   void WriteBytes(const void *data, size_t size)

    56   {

    57     for (size_t i = 0; i < size; i++)

    58       WriteByte(((const Byte *)data)[i]);

    59   }

    60 

    61   UInt64 GetProcessedSize() const;

    62 };

    63 

    64 #endif