drop svn:externals in the tree and add it locally.
[coreboot.git] / payloads / bayou / util / pbuilder / lzma / C / 7zip / Compress / RangeCoder / RangeCoderBitTree.h
diff --git a/payloads/bayou/util/pbuilder/lzma/C/7zip/Compress/RangeCoder/RangeCoderBitTree.h b/payloads/bayou/util/pbuilder/lzma/C/7zip/Compress/RangeCoder/RangeCoderBitTree.h
new file mode 100644 (file)
index 0000000..1fa023f
--- /dev/null
@@ -0,0 +1,161 @@
+// 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