| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475 |
- // InBuffer.h
- #ifndef __INBUFFER_H
- #define __INBUFFER_H
- #include "../IStream.h"
- #include "../../Common/MyCom.h"
- #include "../../Common/MyException.h"
- #ifndef _NO_EXCEPTIONS
- struct CInBufferException: public CSystemException
- {
- CInBufferException(HRESULT errorCode): CSystemException(errorCode) {}
- };
- #endif
- class CInBuffer
- {
- Byte *_buffer;
- Byte *_bufferLimit;
- Byte *_bufferBase;
- CMyComPtr<ISequentialInStream> _stream;
- UInt64 _processedSize;
- UInt32 _bufferSize;
- bool _wasFinished;
- bool ReadBlock();
- Byte ReadBlock2();
- public:
- #ifdef _NO_EXCEPTIONS
- HRESULT ErrorCode;
- #endif
- CInBuffer();
- ~CInBuffer() { Free(); }
- bool Create(UInt32 bufferSize);
- void Free();
-
- void SetStream(ISequentialInStream *stream);
- void Init();
- void ReleaseStream() { _stream.Release(); }
- bool ReadByte(Byte &b)
- {
- if(_buffer >= _bufferLimit)
- if(!ReadBlock())
- return false;
- b = *_buffer++;
- return true;
- }
- Byte ReadByte()
- {
- if(_buffer >= _bufferLimit)
- return ReadBlock2();
- return *_buffer++;
- }
- void ReadBytes(void *data, UInt32 size, UInt32 &processedSize)
- {
- for(processedSize = 0; processedSize < size; processedSize++)
- if (!ReadByte(((Byte *)data)[processedSize]))
- return;
- }
- bool ReadBytes(void *data, UInt32 size)
- {
- UInt32 processedSize;
- ReadBytes(data, size, processedSize);
- return (processedSize == size);
- }
- UInt64 GetProcessedSize() const { return _processedSize + (_buffer - _bufferBase); }
- bool WasFinished() const { return _wasFinished; }
- };
- #endif
|