|
1 // OutBuffer.h |
|
2 |
|
3 #ifndef __OUTBUFFER_H |
|
4 #define __OUTBUFFER_H |
|
5 |
|
6 #include "../IStream.h" |
|
7 #include "../../Common/MyCom.h" |
|
8 |
|
9 #ifndef _NO_EXCEPTIONS |
|
10 struct COutBufferException |
|
11 { |
|
12 HRESULT ErrorCode; |
|
13 COutBufferException(HRESULT errorCode): ErrorCode(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 void FlushWithCheck(); |
|
32 public: |
|
33 #ifdef _NO_EXCEPTIONS |
|
34 HRESULT ErrorCode; |
|
35 #endif |
|
36 |
|
37 COutBuffer(): _buffer(0), _pos(0), _stream(0), _buffer2(0) {} |
|
38 ~COutBuffer() { Free(); } |
|
39 |
|
40 bool Create(UInt32 bufferSize); |
|
41 void Free(); |
|
42 |
|
43 void SetMemStream(Byte *buffer) { _buffer2 = buffer; } |
|
44 void SetStream(ISequentialOutStream *stream); |
|
45 void Init(); |
|
46 HRESULT Flush(); |
|
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 |