| OLD | NEW |
| (Empty) |
| 1 package SevenZip.Compression.RangeCoder; | |
| 2 import java.io.IOException; | |
| 3 | |
| 4 public class BitTreeEncoder | |
| 5 { | |
| 6 short[] Models; | |
| 7 int NumBitLevels; | |
| 8 | |
| 9 public BitTreeEncoder(int numBitLevels) | |
| 10 { | |
| 11 NumBitLevels = numBitLevels; | |
| 12 Models = new short[1 << numBitLevels]; | |
| 13 } | |
| 14 | |
| 15 public void Init() | |
| 16 { | |
| 17 Decoder.InitBitModels(Models); | |
| 18 } | |
| 19 | |
| 20 public void Encode(Encoder rangeEncoder, int symbol) throws IOException | |
| 21 { | |
| 22 int m = 1; | |
| 23 for (int bitIndex = NumBitLevels; bitIndex != 0; ) | |
| 24 { | |
| 25 bitIndex--; | |
| 26 int bit = (symbol >>> bitIndex) & 1; | |
| 27 rangeEncoder.Encode(Models, m, bit); | |
| 28 m = (m << 1) | bit; | |
| 29 } | |
| 30 } | |
| 31 | |
| 32 public void ReverseEncode(Encoder rangeEncoder, int symbol) throws IOExc
eption | |
| 33 { | |
| 34 int m = 1; | |
| 35 for (int i = 0; i < NumBitLevels; i++) | |
| 36 { | |
| 37 int bit = symbol & 1; | |
| 38 rangeEncoder.Encode(Models, m, bit); | |
| 39 m = (m << 1) | bit; | |
| 40 symbol >>= 1; | |
| 41 } | |
| 42 } | |
| 43 | |
| 44 public int GetPrice(int symbol) | |
| 45 { | |
| 46 int price = 0; | |
| 47 int m = 1; | |
| 48 for (int bitIndex = NumBitLevels; bitIndex != 0; ) | |
| 49 { | |
| 50 bitIndex--; | |
| 51 int bit = (symbol >>> bitIndex) & 1; | |
| 52 price += Encoder.GetPrice(Models[m], bit); | |
| 53 m = (m << 1) + bit; | |
| 54 } | |
| 55 return price; | |
| 56 } | |
| 57 | |
| 58 public int ReverseGetPrice(int symbol) | |
| 59 { | |
| 60 int price = 0; | |
| 61 int m = 1; | |
| 62 for (int i = NumBitLevels; i != 0; i--) | |
| 63 { | |
| 64 int bit = symbol & 1; | |
| 65 symbol >>>= 1; | |
| 66 price += Encoder.GetPrice(Models[m], bit); | |
| 67 m = (m << 1) | bit; | |
| 68 } | |
| 69 return price; | |
| 70 } | |
| 71 | |
| 72 public static int ReverseGetPrice(short[] Models, int startIndex, | |
| 73 int NumBitLevels, int symbol) | |
| 74 { | |
| 75 int price = 0; | |
| 76 int m = 1; | |
| 77 for (int i = NumBitLevels; i != 0; i--) | |
| 78 { | |
| 79 int bit = symbol & 1; | |
| 80 symbol >>>= 1; | |
| 81 price += Encoder.GetPrice(Models[startIndex + m], bit); | |
| 82 m = (m << 1) | bit; | |
| 83 } | |
| 84 return price; | |
| 85 } | |
| 86 | |
| 87 public static void ReverseEncode(short[] Models, int startIndex, | |
| 88 Encoder rangeEncoder, int NumBitLevels, int symbol) thro
ws IOException | |
| 89 { | |
| 90 int m = 1; | |
| 91 for (int i = 0; i < NumBitLevels; i++) | |
| 92 { | |
| 93 int bit = symbol & 1; | |
| 94 rangeEncoder.Encode(Models, startIndex + m, bit); | |
| 95 m = (m << 1) | bit; | |
| 96 symbol >>= 1; | |
| 97 } | |
| 98 } | |
| 99 } | |
| OLD | NEW |