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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#ifndef __LENCODER_H
#define __LENCODER_H
#include "BitTreeCoder.h"
namespace NLength {
const int kNumPosStatesBitsMax = 4;
const int kNumPosStatesMax = (1 << kNumPosStatesBitsMax);
const int kNumPosStatesBitsEncodingMax = 4;
const int kNumPosStatesEncodingMax = (1 << kNumPosStatesBitsEncodingMax);
const int kNumMoveBits = 5;
const int kNumLenBits = 3;
const int kNumLowSymbols = 1 << kNumLenBits;
const int kNumMidBits = 3;
const int kNumMidSymbols = 1 << kNumMidBits;
const int kNumHighBits = 8;
const int kNumSymbolsTotal = kNumLowSymbols + kNumMidSymbols + (1 << kNumHighBits);
class CEncoder
{
CMyBitEncoder<kNumMoveBits> m_Choice;
CBitTreeEncoder<kNumMoveBits, kNumLenBits> m_LowCoder[kNumPosStatesEncodingMax];
CMyBitEncoder<kNumMoveBits> m_Choice2;
CBitTreeEncoder<kNumMoveBits, kNumMidBits> m_MidCoder[kNumPosStatesEncodingMax];
CBitTreeEncoder<kNumMoveBits, kNumHighBits> m_HighCoder;
protected:
UINT32 m_NumPosStates;
public:
void Create(UINT32 aNumPosStates)
{ m_NumPosStates = aNumPosStates; }
void Init();
void Encode(CMyRangeEncoder *aRangeEncoder, UINT32 aSymbol, UINT32 aPosState);
UINT32 GetPrice(UINT32 aSymbol, UINT32 aPosState) const;
};
const int kNumSpecSymbols = kNumLowSymbols + kNumMidSymbols;
class CPriceTableEncoder: public CEncoder
{
UINT32 m_Prices[kNumSymbolsTotal][kNumPosStatesEncodingMax];
UINT32 m_TableSize;
UINT32 m_Counters[kNumPosStatesEncodingMax];
public:
void SetTableSize(UINT32 aTableSize)
{ m_TableSize = aTableSize; }
UINT32 GetPrice(UINT32 aSymbol, UINT32 aPosState) const
{ return m_Prices[aSymbol][aPosState]; }
void UpdateTable(UINT32 aPosState)
{
for (UINT32 aLen = 0; aLen < m_TableSize; aLen++)
m_Prices[aLen][aPosState] = CEncoder::GetPrice(aLen , aPosState);
m_Counters[aPosState] = m_TableSize;
}
void UpdateTables()
{
for (UINT32 aPosState = 0; aPosState < m_NumPosStates; aPosState++)
UpdateTable(aPosState);
}
void Encode(CMyRangeEncoder *aRangeEncoder, UINT32 aSymbol, UINT32 aPosState)
{
CEncoder::Encode(aRangeEncoder, aSymbol, aPosState);
if (--m_Counters[aPosState] == 0)
UpdateTable(aPosState);
}
};
class CDecoder
{
CMyBitDecoder<kNumMoveBits> m_Choice;
CBitTreeDecoder<kNumMoveBits, kNumLenBits> m_LowCoder[kNumPosStatesMax];
CMyBitDecoder<kNumMoveBits> m_Choice2;
CBitTreeDecoder<kNumMoveBits, kNumMidBits> m_MidCoder[kNumPosStatesMax];
CBitTreeDecoder<kNumMoveBits, kNumHighBits> m_HighCoder;
UINT32 m_NumPosStates;
public:
void Create(UINT32 aNumPosStates)
{ m_NumPosStates = aNumPosStates; }
void Init()
{
m_Choice.Init();
for (UINT32 aPosState = 0; aPosState < m_NumPosStates; aPosState++)
{
m_LowCoder[aPosState].Init();
m_MidCoder[aPosState].Init();
}
m_Choice2.Init();
m_HighCoder.Init();
}
UINT32 Decode(CMyRangeDecoder *aRangeDecoder, UINT32 aPosState)
{
if(m_Choice.Decode(aRangeDecoder) == 0)
return m_LowCoder[aPosState].Decode(aRangeDecoder);
else
{
UINT32 aSymbol = kNumLowSymbols;
if(m_Choice2.Decode(aRangeDecoder) == 0)
aSymbol += m_MidCoder[aPosState].Decode(aRangeDecoder);
else
{
aSymbol += kNumMidSymbols;
aSymbol += m_HighCoder.Decode(aRangeDecoder);
}
return aSymbol;
}
}
};
}
#endif
|