Return to RangeCoderBitTree.h CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / elwix / tools / oldlzma / SRC / 7zip / Compress / RangeCoder

Annotation of elwix/tools/oldlzma/SRC/7zip/Compress/RangeCoder/RangeCoderBitTree.h, revision 1.1

1.1     ! misho       1: // Compress/RangeCoder/RangeCoderBitTree.h
        !             2: 
        !             3: #ifndef __COMPRESS_RANGECODER_BIT_TREE_H
        !             4: #define __COMPRESS_RANGECODER_BIT_TREE_H
        !             5: 
        !             6: #include "RangeCoderBit.h"
        !             7: #include "RangeCoderOpt.h"
        !             8: 
        !             9: namespace NCompress {
        !            10: namespace NRangeCoder {
        !            11: 
        !            12: template <int numMoveBits, int NumBitLevels>
        !            13: class CBitTreeEncoder
        !            14: {
        !            15:   CBitEncoder<numMoveBits> Models[1 << NumBitLevels];
        !            16: public:
        !            17:   void Init()
        !            18:   {
        !            19:     for(UInt32 i = 1; i < (1 << NumBitLevels); i++)
        !            20:       Models[i].Init();
        !            21:   }
        !            22:   void Encode(CEncoder *rangeEncoder, UInt32 symbol)
        !            23:   {
        !            24:     UInt32 modelIndex = 1;
        !            25:     for (int bitIndex = NumBitLevels; bitIndex != 0 ;)
        !            26:     {
        !            27:       bitIndex--;
        !            28:       UInt32 bit = (symbol >> bitIndex) & 1;
        !            29:       Models[modelIndex].Encode(rangeEncoder, bit);
        !            30:       modelIndex = (modelIndex << 1) | bit;
        !            31:     }
        !            32:   };
        !            33:   void ReverseEncode(CEncoder *rangeEncoder, UInt32 symbol)
        !            34:   {
        !            35:     UInt32 modelIndex = 1;
        !            36:     for (int i = 0; i < NumBitLevels; i++)
        !            37:     {
        !            38:       UInt32 bit = symbol & 1;
        !            39:       Models[modelIndex].Encode(rangeEncoder, bit);
        !            40:       modelIndex = (modelIndex << 1) | bit;
        !            41:       symbol >>= 1;
        !            42:     }
        !            43:   }
        !            44:   UInt32 GetPrice(UInt32 symbol) const
        !            45:   {
        !            46:     symbol |= (1 << NumBitLevels);
        !            47:     UInt32 price = 0;
        !            48:     while (symbol != 1)
        !            49:     {
        !            50:       price += Models[symbol >> 1].GetPrice(symbol & 1);
        !            51:       symbol >>= 1;
        !            52:     }
        !            53:     return price;
        !            54:   }
        !            55:   UInt32 ReverseGetPrice(UInt32 symbol) const
        !            56:   {
        !            57:     UInt32 price = 0;
        !            58:     UInt32 modelIndex = 1;
        !            59:     for (int i = NumBitLevels; i != 0; i--)
        !            60:     {
        !            61:       UInt32 bit = symbol & 1;
        !            62:       symbol >>= 1;
        !            63:       price += Models[modelIndex].GetPrice(bit);
        !            64:       modelIndex = (modelIndex << 1) | bit;
        !            65:     }
        !            66:     return price;
        !            67:   }
        !            68: };
        !            69: 
        !            70: template <int numMoveBits, int NumBitLevels>
        !            71: class CBitTreeDecoder
        !            72: {
        !            73:   CBitDecoder<numMoveBits> Models[1 << NumBitLevels];
        !            74: public:
        !            75:   void Init()
        !            76:   {
        !            77:     for(UInt32 i = 1; i < (1 << NumBitLevels); i++)
        !            78:       Models[i].Init();
        !            79:   }
        !            80:   UInt32 Decode(CDecoder *rangeDecoder)
        !            81:   {
        !            82:     UInt32 modelIndex = 1;
        !            83:     RC_INIT_VAR
        !            84:     for(int bitIndex = NumBitLevels; bitIndex != 0; bitIndex--)
        !            85:     {
        !            86:       // modelIndex = (modelIndex << 1) + Models[modelIndex].Decode(rangeDecoder);
        !            87:       RC_GETBIT(numMoveBits, Models[modelIndex].Prob, modelIndex)
        !            88:     }
        !            89:     RC_FLUSH_VAR
        !            90:     return modelIndex - (1 << NumBitLevels);
        !            91:   };
        !            92:   UInt32 ReverseDecode(CDecoder *rangeDecoder)
        !            93:   {
        !            94:     UInt32 modelIndex = 1;
        !            95:     UInt32 symbol = 0;
        !            96:     RC_INIT_VAR
        !            97:     for(int bitIndex = 0; bitIndex < NumBitLevels; bitIndex++)
        !            98:     {
        !            99:       // UInt32 bit = Models[modelIndex].Decode(rangeDecoder);
        !           100:       // modelIndex <<= 1;
        !           101:       // modelIndex += bit;
        !           102:       // symbol |= (bit << bitIndex);
        !           103:       RC_GETBIT2(numMoveBits, Models[modelIndex].Prob, modelIndex, ; , symbol |= (1 << bitIndex))
        !           104:     }
        !           105:     RC_FLUSH_VAR
        !           106:     return symbol;
        !           107:   }
        !           108: };
        !           109: 
        !           110: template <int numMoveBits>
        !           111: void ReverseBitTreeEncode(CBitEncoder<numMoveBits> *Models, 
        !           112:     CEncoder *rangeEncoder, int NumBitLevels, UInt32 symbol)
        !           113: {
        !           114:   UInt32 modelIndex = 1;
        !           115:   for (int i = 0; i < NumBitLevels; i++)
        !           116:   {
        !           117:     UInt32 bit = symbol & 1;
        !           118:     Models[modelIndex].Encode(rangeEncoder, bit);
        !           119:     modelIndex = (modelIndex << 1) | bit;
        !           120:     symbol >>= 1;
        !           121:   }
        !           122: }
        !           123: 
        !           124: template <int numMoveBits>
        !           125: UInt32 ReverseBitTreeGetPrice(CBitEncoder<numMoveBits> *Models, 
        !           126:     UInt32 NumBitLevels, UInt32 symbol)
        !           127: {
        !           128:   UInt32 price = 0;
        !           129:   UInt32 modelIndex = 1;
        !           130:   for (int i = NumBitLevels; i != 0; i--)
        !           131:   {
        !           132:     UInt32 bit = symbol & 1;
        !           133:     symbol >>= 1;
        !           134:     price += Models[modelIndex].GetPrice(bit);
        !           135:     modelIndex = (modelIndex << 1) | bit;
        !           136:   }
        !           137:   return price;
        !           138: }
        !           139: 
        !           140: template <int numMoveBits>
        !           141: UInt32 ReverseBitTreeDecode(CBitDecoder<numMoveBits> *Models, 
        !           142:     CDecoder *rangeDecoder, int NumBitLevels)
        !           143: {
        !           144:   UInt32 modelIndex = 1;
        !           145:   UInt32 symbol = 0;
        !           146:   RC_INIT_VAR
        !           147:   for(int bitIndex = 0; bitIndex < NumBitLevels; bitIndex++)
        !           148:   {
        !           149:     // UInt32 bit = Models[modelIndex].Decode(rangeDecoder);
        !           150:     // modelIndex <<= 1;
        !           151:     // modelIndex += bit;
        !           152:     // symbol |= (bit << bitIndex);
        !           153:     RC_GETBIT2(numMoveBits, Models[modelIndex].Prob, modelIndex, ; , symbol |= (1 << bitIndex))
        !           154:   }
        !           155:   RC_FLUSH_VAR
        !           156:   return symbol;
        !           157: }
        !           158: 
        !           159: }}
        !           160: 
        !           161: #endif