other-licenses/7zstub/src/7zip/Common/OutBuffer.h

branch
TOR_BUG_9701
changeset 14
925c144e1f1f
equal deleted inserted replaced
-1:000000000000 0:84e3c88e3122
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

mercurial