blob: 039e507ade4764283a976ecb22125d587855f7f8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
#ifndef __STREAM_WINDOWIN_H
#define __STREAM_WINDOWIN_H
#include "IInOutStreams.h"
namespace NStream {
namespace NWindow {
class CIn
{
BYTE *m_BufferBase; // pointer to buffer with data
ISequentialInStream* m_Stream;
UINT32 m_PosLimit; // offset (from m_Buffer) of first byte when new block reading must be done
bool m_StreamEndWasReached; // if (true) then m_StreamPos shows real end of stream
const BYTE *m_PointerToLastSafePosition;
protected:
BYTE *m_Buffer; // Pointer to virtual Buffer begin
UINT32 m_BlockSize; // Size of Allocated memory block
UINT32 m_Pos; // offset (from m_Buffer) of curent byte
UINT32 m_KeepSizeBefore; // how many BYTEs must be kept in buffer before m_Pos
UINT32 m_KeepSizeAfter; // how many BYTEs must be kept buffer after m_Pos
UINT32 m_KeepSizeReserv; // how many BYTEs must be kept as reserv
UINT32 m_StreamPos; // offset (from m_Buffer) of first not read byte from Stream
virtual void BeforeMoveBlock() {};
virtual void AfterMoveBlock() {};
void MoveBlock();
virtual HRESULT ReadBlock();
void Free();
public:
CIn();
void Create(UINT32 aKeepSizeBefore, UINT32 aKeepSizeAfter,
UINT32 aKeepSizeReserv = (1<<17));
virtual ~CIn();
HRESULT Init(ISequentialInStream *aStream);
BYTE *GetBuffer() const { return m_Buffer; }
const BYTE *GetPointerToCurrentPos() const { return m_Buffer + m_Pos; }
HRESULT MovePos()
{
m_Pos++;
if (m_Pos > m_PosLimit)
{
const BYTE *aPointerToPostion = m_Buffer + m_Pos;
if(aPointerToPostion > m_PointerToLastSafePosition)
MoveBlock();
return ReadBlock();
}
else
return S_OK;
}
// BYTE GetCurrentByte()const;
BYTE GetIndexByte(UINT32 anIndex)const
{ return m_Buffer[m_Pos + anIndex]; }
// UINT32 GetCurPos()const { return m_Pos;};
// BYTE *GetBufferBeg()const { return m_Buffer;};
// aIndex + aLimit have not to exceed m_KeepSizeAfter;
UINT32 GetMatchLen(UINT32 aIndex, UINT32 aBack, UINT32 aLimit) const
{
if(m_StreamEndWasReached)
if ((m_Pos + aIndex) + aLimit > m_StreamPos)
aLimit = m_StreamPos - (m_Pos + aIndex);
aBack++;
BYTE *pby = m_Buffer + m_Pos + aIndex;
UINT32 i;
for(i = 0; i < aLimit && pby[i] == pby[i - aBack]; i++);
return i;
}
UINT32 GetNumAvailableBytes() const { return m_StreamPos - m_Pos; }
void ReduceOffsets(UINT32 aSubValue)
{
m_Buffer += aSubValue;
m_PosLimit -= aSubValue;
m_Pos -= aSubValue;
m_StreamPos -= aSubValue;
}
};
}}
#endif
|