Decoder.java 9.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329
  1. package SevenZip.Compression.LZMA;
  2. import SevenZip.Compression.RangeCoder.BitTreeDecoder;
  3. import SevenZip.Compression.LZMA.Base;
  4. import SevenZip.Compression.LZ.OutWindow;
  5. import java.io.IOException;
  6. public class Decoder
  7. {
  8. class LenDecoder
  9. {
  10. short[] m_Choice = new short[2];
  11. BitTreeDecoder[] m_LowCoder = new BitTreeDecoder[Base.kNumPosStatesMax];
  12. BitTreeDecoder[] m_MidCoder = new BitTreeDecoder[Base.kNumPosStatesMax];
  13. BitTreeDecoder m_HighCoder = new BitTreeDecoder(Base.kNumHighLenBits);
  14. int m_NumPosStates = 0;
  15. public void Create(int numPosStates)
  16. {
  17. for (; m_NumPosStates < numPosStates; m_NumPosStates++)
  18. {
  19. m_LowCoder[m_NumPosStates] = new BitTreeDecoder(Base.kNumLowLenBits);
  20. m_MidCoder[m_NumPosStates] = new BitTreeDecoder(Base.kNumMidLenBits);
  21. }
  22. }
  23. public void Init()
  24. {
  25. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_Choice);
  26. for (int posState = 0; posState < m_NumPosStates; posState++)
  27. {
  28. m_LowCoder[posState].Init();
  29. m_MidCoder[posState].Init();
  30. }
  31. m_HighCoder.Init();
  32. }
  33. public int Decode(SevenZip.Compression.RangeCoder.Decoder rangeDecoder, int posState) throws IOException
  34. {
  35. if (rangeDecoder.DecodeBit(m_Choice, 0) == 0)
  36. return m_LowCoder[posState].Decode(rangeDecoder);
  37. int symbol = Base.kNumLowLenSymbols;
  38. if (rangeDecoder.DecodeBit(m_Choice, 1) == 0)
  39. symbol += m_MidCoder[posState].Decode(rangeDecoder);
  40. else
  41. symbol += Base.kNumMidLenSymbols + m_HighCoder.Decode(rangeDecoder);
  42. return symbol;
  43. }
  44. }
  45. class LiteralDecoder
  46. {
  47. class Decoder2
  48. {
  49. short[] m_Decoders = new short[0x300];
  50. public void Init()
  51. {
  52. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_Decoders);
  53. }
  54. public byte DecodeNormal(SevenZip.Compression.RangeCoder.Decoder rangeDecoder) throws IOException
  55. {
  56. int symbol = 1;
  57. do
  58. symbol = (symbol << 1) | rangeDecoder.DecodeBit(m_Decoders, symbol);
  59. while (symbol < 0x100);
  60. return (byte)symbol;
  61. }
  62. public byte DecodeWithMatchByte(SevenZip.Compression.RangeCoder.Decoder rangeDecoder, byte matchByte) throws IOException
  63. {
  64. int symbol = 1;
  65. do
  66. {
  67. int matchBit = (matchByte >> 7) & 1;
  68. matchByte <<= 1;
  69. int bit = rangeDecoder.DecodeBit(m_Decoders, ((1 + matchBit) << 8) + symbol);
  70. symbol = (symbol << 1) | bit;
  71. if (matchBit != bit)
  72. {
  73. while (symbol < 0x100)
  74. symbol = (symbol << 1) | rangeDecoder.DecodeBit(m_Decoders, symbol);
  75. break;
  76. }
  77. }
  78. while (symbol < 0x100);
  79. return (byte)symbol;
  80. }
  81. }
  82. Decoder2[] m_Coders;
  83. int m_NumPrevBits;
  84. int m_NumPosBits;
  85. int m_PosMask;
  86. public void Create(int numPosBits, int numPrevBits)
  87. {
  88. if (m_Coders != null && m_NumPrevBits == numPrevBits && m_NumPosBits == numPosBits)
  89. return;
  90. m_NumPosBits = numPosBits;
  91. m_PosMask = (1 << numPosBits) - 1;
  92. m_NumPrevBits = numPrevBits;
  93. int numStates = 1 << (m_NumPrevBits + m_NumPosBits);
  94. m_Coders = new Decoder2[numStates];
  95. for (int i = 0; i < numStates; i++)
  96. m_Coders[i] = new Decoder2();
  97. }
  98. public void Init()
  99. {
  100. int numStates = 1 << (m_NumPrevBits + m_NumPosBits);
  101. for (int i = 0; i < numStates; i++)
  102. m_Coders[i].Init();
  103. }
  104. Decoder2 GetDecoder(int pos, byte prevByte)
  105. {
  106. return m_Coders[((pos & m_PosMask) << m_NumPrevBits) + ((prevByte & 0xFF) >>> (8 - m_NumPrevBits))];
  107. }
  108. }
  109. OutWindow m_OutWindow = new OutWindow();
  110. SevenZip.Compression.RangeCoder.Decoder m_RangeDecoder = new SevenZip.Compression.RangeCoder.Decoder();
  111. short[] m_IsMatchDecoders = new short[Base.kNumStates << Base.kNumPosStatesBitsMax];
  112. short[] m_IsRepDecoders = new short[Base.kNumStates];
  113. short[] m_IsRepG0Decoders = new short[Base.kNumStates];
  114. short[] m_IsRepG1Decoders = new short[Base.kNumStates];
  115. short[] m_IsRepG2Decoders = new short[Base.kNumStates];
  116. short[] m_IsRep0LongDecoders = new short[Base.kNumStates << Base.kNumPosStatesBitsMax];
  117. BitTreeDecoder[] m_PosSlotDecoder = new BitTreeDecoder[Base.kNumLenToPosStates];
  118. short[] m_PosDecoders = new short[Base.kNumFullDistances - Base.kEndPosModelIndex];
  119. BitTreeDecoder m_PosAlignDecoder = new BitTreeDecoder(Base.kNumAlignBits);
  120. LenDecoder m_LenDecoder = new LenDecoder();
  121. LenDecoder m_RepLenDecoder = new LenDecoder();
  122. LiteralDecoder m_LiteralDecoder = new LiteralDecoder();
  123. int m_DictionarySize = -1;
  124. int m_DictionarySizeCheck = -1;
  125. int m_PosStateMask;
  126. public Decoder()
  127. {
  128. for (int i = 0; i < Base.kNumLenToPosStates; i++)
  129. m_PosSlotDecoder[i] = new BitTreeDecoder(Base.kNumPosSlotBits);
  130. }
  131. boolean SetDictionarySize(int dictionarySize)
  132. {
  133. if (dictionarySize < 0)
  134. return false;
  135. if (m_DictionarySize != dictionarySize)
  136. {
  137. m_DictionarySize = dictionarySize;
  138. m_DictionarySizeCheck = Math.max(m_DictionarySize, 1);
  139. m_OutWindow.Create(Math.max(m_DictionarySizeCheck, (1 << 12)));
  140. }
  141. return true;
  142. }
  143. boolean SetLcLpPb(int lc, int lp, int pb)
  144. {
  145. if (lc > Base.kNumLitContextBitsMax || lp > 4 || pb > Base.kNumPosStatesBitsMax)
  146. return false;
  147. m_LiteralDecoder.Create(lp, lc);
  148. int numPosStates = 1 << pb;
  149. m_LenDecoder.Create(numPosStates);
  150. m_RepLenDecoder.Create(numPosStates);
  151. m_PosStateMask = numPosStates - 1;
  152. return true;
  153. }
  154. void Init() throws IOException
  155. {
  156. m_OutWindow.Init(false);
  157. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_IsMatchDecoders);
  158. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_IsRep0LongDecoders);
  159. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_IsRepDecoders);
  160. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_IsRepG0Decoders);
  161. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_IsRepG1Decoders);
  162. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_IsRepG2Decoders);
  163. SevenZip.Compression.RangeCoder.Decoder.InitBitModels(m_PosDecoders);
  164. m_LiteralDecoder.Init();
  165. int i;
  166. for (i = 0; i < Base.kNumLenToPosStates; i++)
  167. m_PosSlotDecoder[i].Init();
  168. m_LenDecoder.Init();
  169. m_RepLenDecoder.Init();
  170. m_PosAlignDecoder.Init();
  171. m_RangeDecoder.Init();
  172. }
  173. public boolean Code(java.io.InputStream inStream, java.io.OutputStream outStream,
  174. long outSize) throws IOException
  175. {
  176. m_RangeDecoder.SetStream(inStream);
  177. m_OutWindow.SetStream(outStream);
  178. Init();
  179. int state = Base.StateInit();
  180. int rep0 = 0, rep1 = 0, rep2 = 0, rep3 = 0;
  181. long nowPos64 = 0;
  182. byte prevByte = 0;
  183. while (outSize < 0 || nowPos64 < outSize)
  184. {
  185. int posState = (int)nowPos64 & m_PosStateMask;
  186. if (m_RangeDecoder.DecodeBit(m_IsMatchDecoders, (state << Base.kNumPosStatesBitsMax) + posState) == 0)
  187. {
  188. LiteralDecoder.Decoder2 decoder2 = m_LiteralDecoder.GetDecoder((int)nowPos64, prevByte);
  189. if (!Base.StateIsCharState(state))
  190. prevByte = decoder2.DecodeWithMatchByte(m_RangeDecoder, m_OutWindow.GetByte(rep0));
  191. else
  192. prevByte = decoder2.DecodeNormal(m_RangeDecoder);
  193. m_OutWindow.PutByte(prevByte);
  194. state = Base.StateUpdateChar(state);
  195. nowPos64++;
  196. }
  197. else
  198. {
  199. int len;
  200. if (m_RangeDecoder.DecodeBit(m_IsRepDecoders, state) == 1)
  201. {
  202. len = 0;
  203. if (m_RangeDecoder.DecodeBit(m_IsRepG0Decoders, state) == 0)
  204. {
  205. if (m_RangeDecoder.DecodeBit(m_IsRep0LongDecoders, (state << Base.kNumPosStatesBitsMax) + posState) == 0)
  206. {
  207. state = Base.StateUpdateShortRep(state);
  208. len = 1;
  209. }
  210. }
  211. else
  212. {
  213. int distance;
  214. if (m_RangeDecoder.DecodeBit(m_IsRepG1Decoders, state) == 0)
  215. distance = rep1;
  216. else
  217. {
  218. if (m_RangeDecoder.DecodeBit(m_IsRepG2Decoders, state) == 0)
  219. distance = rep2;
  220. else
  221. {
  222. distance = rep3;
  223. rep3 = rep2;
  224. }
  225. rep2 = rep1;
  226. }
  227. rep1 = rep0;
  228. rep0 = distance;
  229. }
  230. if (len == 0)
  231. {
  232. len = m_RepLenDecoder.Decode(m_RangeDecoder, posState) + Base.kMatchMinLen;
  233. state = Base.StateUpdateRep(state);
  234. }
  235. }
  236. else
  237. {
  238. rep3 = rep2;
  239. rep2 = rep1;
  240. rep1 = rep0;
  241. len = Base.kMatchMinLen + m_LenDecoder.Decode(m_RangeDecoder, posState);
  242. state = Base.StateUpdateMatch(state);
  243. int posSlot = m_PosSlotDecoder[Base.GetLenToPosState(len)].Decode(m_RangeDecoder);
  244. if (posSlot >= Base.kStartPosModelIndex)
  245. {
  246. int numDirectBits = (posSlot >> 1) - 1;
  247. rep0 = ((2 | (posSlot & 1)) << numDirectBits);
  248. if (posSlot < Base.kEndPosModelIndex)
  249. rep0 += BitTreeDecoder.ReverseDecode(m_PosDecoders,
  250. rep0 - posSlot - 1, m_RangeDecoder, numDirectBits);
  251. else
  252. {
  253. rep0 += (m_RangeDecoder.DecodeDirectBits(
  254. numDirectBits - Base.kNumAlignBits) << Base.kNumAlignBits);
  255. rep0 += m_PosAlignDecoder.ReverseDecode(m_RangeDecoder);
  256. if (rep0 < 0)
  257. {
  258. if (rep0 == -1)
  259. break;
  260. return false;
  261. }
  262. }
  263. }
  264. else
  265. rep0 = posSlot;
  266. }
  267. if (rep0 >= nowPos64 || rep0 >= m_DictionarySizeCheck)
  268. {
  269. // m_OutWindow.Flush();
  270. return false;
  271. }
  272. m_OutWindow.CopyBlock(rep0, len);
  273. nowPos64 += len;
  274. prevByte = m_OutWindow.GetByte(0);
  275. }
  276. }
  277. m_OutWindow.Flush();
  278. m_OutWindow.ReleaseStream();
  279. m_RangeDecoder.ReleaseStream();
  280. return true;
  281. }
  282. public boolean SetDecoderProperties(byte[] properties)
  283. {
  284. if (properties.length < 5)
  285. return false;
  286. int val = properties[0] & 0xFF;
  287. int lc = val % 9;
  288. int remainder = val / 9;
  289. int lp = remainder % 5;
  290. int pb = remainder / 5;
  291. int dictionarySize = 0;
  292. for (int i = 0; i < 4; i++)
  293. dictionarySize += ((int)(properties[1 + i]) & 0xFF) << (i * 8);
  294. if (!SetLcLpPb(lc, lp, pb))
  295. return false;
  296. return SetDictionarySize(dictionarySize);
  297. }
  298. }