-// Compress/RangeCoder/RangeCoderBitTree.h\r
-\r
-#ifndef __COMPRESS_RANGECODER_BIT_TREE_H\r
-#define __COMPRESS_RANGECODER_BIT_TREE_H\r
-\r
-#include "RangeCoderBit.h"\r
-#include "RangeCoderOpt.h"\r
-\r
-namespace NCompress {\r
-namespace NRangeCoder {\r
-\r
-template <int numMoveBits, int NumBitLevels>\r
-class CBitTreeEncoder\r
-{\r
- CBitEncoder<numMoveBits> Models[1 << NumBitLevels];\r
-public:\r
- void Init()\r
- {\r
- for(UInt32 i = 1; i < (1 << NumBitLevels); i++)\r
- Models[i].Init();\r
- }\r
- void Encode(CEncoder *rangeEncoder, UInt32 symbol)\r
- {\r
- UInt32 modelIndex = 1;\r
- for (int bitIndex = NumBitLevels; bitIndex != 0 ;)\r
- {\r
- bitIndex--;\r
- UInt32 bit = (symbol >> bitIndex) & 1;\r
- Models[modelIndex].Encode(rangeEncoder, bit);\r
- modelIndex = (modelIndex << 1) | bit;\r
- }\r
- };\r
- void ReverseEncode(CEncoder *rangeEncoder, UInt32 symbol)\r
- {\r
- UInt32 modelIndex = 1;\r
- for (int i = 0; i < NumBitLevels; i++)\r
- {\r
- UInt32 bit = symbol & 1;\r
- Models[modelIndex].Encode(rangeEncoder, bit);\r
- modelIndex = (modelIndex << 1) | bit;\r
- symbol >>= 1;\r
- }\r
- }\r
- UInt32 GetPrice(UInt32 symbol) const\r
- {\r
- symbol |= (1 << NumBitLevels);\r
- UInt32 price = 0;\r
- while (symbol != 1)\r
- {\r
- price += Models[symbol >> 1].GetPrice(symbol & 1);\r
- symbol >>= 1;\r
- }\r
- return price;\r
- }\r
- UInt32 ReverseGetPrice(UInt32 symbol) const\r
- {\r
- UInt32 price = 0;\r
- UInt32 modelIndex = 1;\r
- for (int i = NumBitLevels; i != 0; i--)\r
- {\r
- UInt32 bit = symbol & 1;\r
- symbol >>= 1;\r
- price += Models[modelIndex].GetPrice(bit);\r
- modelIndex = (modelIndex << 1) | bit;\r
- }\r
- return price;\r
- }\r
-};\r
-\r
-template <int numMoveBits, int NumBitLevels>\r
-class CBitTreeDecoder\r
-{\r
- CBitDecoder<numMoveBits> Models[1 << NumBitLevels];\r
-public:\r
- void Init()\r
- {\r
- for(UInt32 i = 1; i < (1 << NumBitLevels); i++)\r
- Models[i].Init();\r
- }\r
- UInt32 Decode(CDecoder *rangeDecoder)\r
- {\r
- UInt32 modelIndex = 1;\r
- RC_INIT_VAR\r
- for(int bitIndex = NumBitLevels; bitIndex != 0; bitIndex--)\r
- {\r
- // modelIndex = (modelIndex << 1) + Models[modelIndex].Decode(rangeDecoder);\r
- RC_GETBIT(numMoveBits, Models[modelIndex].Prob, modelIndex)\r
- }\r
- RC_FLUSH_VAR\r
- return modelIndex - (1 << NumBitLevels);\r
- };\r
- UInt32 ReverseDecode(CDecoder *rangeDecoder)\r
- {\r
- UInt32 modelIndex = 1;\r
- UInt32 symbol = 0;\r
- RC_INIT_VAR\r
- for(int bitIndex = 0; bitIndex < NumBitLevels; bitIndex++)\r
- {\r
- // UInt32 bit = Models[modelIndex].Decode(rangeDecoder);\r
- // modelIndex <<= 1;\r
- // modelIndex += bit;\r
- // symbol |= (bit << bitIndex);\r
- RC_GETBIT2(numMoveBits, Models[modelIndex].Prob, modelIndex, ; , symbol |= (1 << bitIndex))\r
- }\r
- RC_FLUSH_VAR\r
- return symbol;\r
- }\r
-};\r
-\r
-template <int numMoveBits>\r
-void ReverseBitTreeEncode(CBitEncoder<numMoveBits> *Models, \r
- CEncoder *rangeEncoder, int NumBitLevels, UInt32 symbol)\r
-{\r
- UInt32 modelIndex = 1;\r
- for (int i = 0; i < NumBitLevels; i++)\r
- {\r
- UInt32 bit = symbol & 1;\r
- Models[modelIndex].Encode(rangeEncoder, bit);\r
- modelIndex = (modelIndex << 1) | bit;\r
- symbol >>= 1;\r
- }\r
-}\r
-\r
-template <int numMoveBits>\r
-UInt32 ReverseBitTreeGetPrice(CBitEncoder<numMoveBits> *Models, \r
- UInt32 NumBitLevels, UInt32 symbol)\r
-{\r
- UInt32 price = 0;\r
- UInt32 modelIndex = 1;\r
- for (int i = NumBitLevels; i != 0; i--)\r
- {\r
- UInt32 bit = symbol & 1;\r
- symbol >>= 1;\r
- price += Models[modelIndex].GetPrice(bit);\r
- modelIndex = (modelIndex << 1) | bit;\r
- }\r
- return price;\r
-}\r
-\r
-template <int numMoveBits>\r
-UInt32 ReverseBitTreeDecode(CBitDecoder<numMoveBits> *Models, \r
- CDecoder *rangeDecoder, int NumBitLevels)\r
-{\r
- UInt32 modelIndex = 1;\r
- UInt32 symbol = 0;\r
- RC_INIT_VAR\r
- for(int bitIndex = 0; bitIndex < NumBitLevels; bitIndex++)\r
- {\r
- // UInt32 bit = Models[modelIndex].Decode(rangeDecoder);\r
- // modelIndex <<= 1;\r
- // modelIndex += bit;\r
- // symbol |= (bit << bitIndex);\r
- RC_GETBIT2(numMoveBits, Models[modelIndex].Prob, modelIndex, ; , symbol |= (1 << bitIndex))\r
- }\r
- RC_FLUSH_VAR\r
- return symbol;\r
-}\r
-\r
-}}\r
-\r
-#endif\r
+// Compress/RangeCoder/RangeCoderBitTree.h
+
+#ifndef __COMPRESS_RANGECODER_BIT_TREE_H
+#define __COMPRESS_RANGECODER_BIT_TREE_H
+
+#include "RangeCoderBit.h"
+#include "RangeCoderOpt.h"
+
+namespace NCompress {
+namespace NRangeCoder {
+
+template <int numMoveBits, int NumBitLevels>
+class CBitTreeEncoder
+{
+ CBitEncoder<numMoveBits> Models[1 << NumBitLevels];
+public:
+ void Init()
+ {
+ for(UInt32 i = 1; i < (1 << NumBitLevels); i++)
+ Models[i].Init();
+ }
+ void Encode(CEncoder *rangeEncoder, UInt32 symbol)
+ {
+ UInt32 modelIndex = 1;
+ for (int bitIndex = NumBitLevels; bitIndex != 0 ;)
+ {
+ bitIndex--;
+ UInt32 bit = (symbol >> bitIndex) & 1;
+ Models[modelIndex].Encode(rangeEncoder, bit);
+ modelIndex = (modelIndex << 1) | bit;
+ }
+ };
+ void ReverseEncode(CEncoder *rangeEncoder, UInt32 symbol)
+ {
+ UInt32 modelIndex = 1;
+ for (int i = 0; i < NumBitLevels; i++)
+ {
+ UInt32 bit = symbol & 1;
+ Models[modelIndex].Encode(rangeEncoder, bit);
+ modelIndex = (modelIndex << 1) | bit;
+ symbol >>= 1;
+ }
+ }
+ UInt32 GetPrice(UInt32 symbol) const
+ {
+ symbol |= (1 << NumBitLevels);
+ UInt32 price = 0;
+ while (symbol != 1)
+ {
+ price += Models[symbol >> 1].GetPrice(symbol & 1);
+ symbol >>= 1;
+ }
+ return price;
+ }
+ UInt32 ReverseGetPrice(UInt32 symbol) const
+ {
+ UInt32 price = 0;
+ UInt32 modelIndex = 1;
+ for (int i = NumBitLevels; i != 0; i--)
+ {
+ UInt32 bit = symbol & 1;
+ symbol >>= 1;
+ price += Models[modelIndex].GetPrice(bit);
+ modelIndex = (modelIndex << 1) | bit;
+ }
+ return price;
+ }
+};
+
+template <int numMoveBits, int NumBitLevels>
+class CBitTreeDecoder
+{
+ CBitDecoder<numMoveBits> Models[1 << NumBitLevels];
+public:
+ void Init()
+ {
+ for(UInt32 i = 1; i < (1 << NumBitLevels); i++)
+ Models[i].Init();
+ }
+ UInt32 Decode(CDecoder *rangeDecoder)
+ {
+ UInt32 modelIndex = 1;
+ RC_INIT_VAR
+ for(int bitIndex = NumBitLevels; bitIndex != 0; bitIndex--)
+ {
+ // modelIndex = (modelIndex << 1) + Models[modelIndex].Decode(rangeDecoder);
+ RC_GETBIT(numMoveBits, Models[modelIndex].Prob, modelIndex)
+ }
+ RC_FLUSH_VAR
+ return modelIndex - (1 << NumBitLevels);
+ };
+ UInt32 ReverseDecode(CDecoder *rangeDecoder)
+ {
+ UInt32 modelIndex = 1;
+ UInt32 symbol = 0;
+ RC_INIT_VAR
+ for(int bitIndex = 0; bitIndex < NumBitLevels; bitIndex++)
+ {
+ // UInt32 bit = Models[modelIndex].Decode(rangeDecoder);
+ // modelIndex <<= 1;
+ // modelIndex += bit;
+ // symbol |= (bit << bitIndex);
+ RC_GETBIT2(numMoveBits, Models[modelIndex].Prob, modelIndex, ; , symbol |= (1 << bitIndex))
+ }
+ RC_FLUSH_VAR
+ return symbol;
+ }
+};
+
+template <int numMoveBits>
+void ReverseBitTreeEncode(CBitEncoder<numMoveBits> *Models,
+ CEncoder *rangeEncoder, int NumBitLevels, UInt32 symbol)
+{
+ UInt32 modelIndex = 1;
+ for (int i = 0; i < NumBitLevels; i++)
+ {
+ UInt32 bit = symbol & 1;
+ Models[modelIndex].Encode(rangeEncoder, bit);
+ modelIndex = (modelIndex << 1) | bit;
+ symbol >>= 1;
+ }
+}
+
+template <int numMoveBits>
+UInt32 ReverseBitTreeGetPrice(CBitEncoder<numMoveBits> *Models,
+ UInt32 NumBitLevels, UInt32 symbol)
+{
+ UInt32 price = 0;
+ UInt32 modelIndex = 1;
+ for (int i = NumBitLevels; i != 0; i--)
+ {
+ UInt32 bit = symbol & 1;
+ symbol >>= 1;
+ price += Models[modelIndex].GetPrice(bit);
+ modelIndex = (modelIndex << 1) | bit;
+ }
+ return price;
+}
+
+template <int numMoveBits>
+UInt32 ReverseBitTreeDecode(CBitDecoder<numMoveBits> *Models,
+ CDecoder *rangeDecoder, int NumBitLevels)
+{
+ UInt32 modelIndex = 1;
+ UInt32 symbol = 0;
+ RC_INIT_VAR
+ for(int bitIndex = 0; bitIndex < NumBitLevels; bitIndex++)
+ {
+ // UInt32 bit = Models[modelIndex].Decode(rangeDecoder);
+ // modelIndex <<= 1;
+ // modelIndex += bit;
+ // symbol |= (bit << bitIndex);
+ RC_GETBIT2(numMoveBits, Models[modelIndex].Prob, modelIndex, ; , symbol |= (1 << bitIndex))
+ }
+ RC_FLUSH_VAR
+ return symbol;
+}
+
+}}
+
+#endif