| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 PDFium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | |
| 6 | |
| 7 #include "core/src/fxcodec/jbig2/JBig2_HuffmanTable.h" | |
| 8 | |
| 9 #include <algorithm> | |
| 10 #include <vector> | |
| 11 | |
| 12 #include "core/include/fxcrt/fx_memory.h" | |
| 13 #include "core/src/fxcodec/jbig2/JBig2_BitStream.h" | |
| 14 #include "core/src/fxcodec/jbig2/JBig2_Define.h" | |
| 15 #include "core/src/fxcodec/jbig2/JBig2_HuffmanTable_Standard.h" | |
| 16 | |
| 17 CJBig2_HuffmanTable::CJBig2_HuffmanTable(const JBig2TableLine* pTable, | |
| 18 FX_DWORD nLines, | |
| 19 bool bHTOOB) | |
| 20 : m_bOK(true), HTOOB(bHTOOB), NTEMP(nLines) { | |
| 21 ParseFromStandardTable(pTable); | |
| 22 } | |
| 23 | |
| 24 CJBig2_HuffmanTable::CJBig2_HuffmanTable(CJBig2_BitStream* pStream) | |
| 25 : HTOOB(false), NTEMP(0) { | |
| 26 m_bOK = ParseFromCodedBuffer(pStream); | |
| 27 } | |
| 28 | |
| 29 CJBig2_HuffmanTable::~CJBig2_HuffmanTable() { | |
| 30 } | |
| 31 | |
| 32 void CJBig2_HuffmanTable::ParseFromStandardTable(const JBig2TableLine* pTable) { | |
| 33 PREFLEN.resize(NTEMP); | |
| 34 RANGELEN.resize(NTEMP); | |
| 35 RANGELOW.resize(NTEMP); | |
| 36 for (FX_DWORD i = 0; i < NTEMP; ++i) { | |
| 37 PREFLEN[i] = pTable[i].PREFLEN; | |
| 38 RANGELEN[i] = pTable[i].RANDELEN; | |
| 39 RANGELOW[i] = pTable[i].RANGELOW; | |
| 40 } | |
| 41 InitCodes(); | |
| 42 } | |
| 43 | |
| 44 bool CJBig2_HuffmanTable::ParseFromCodedBuffer(CJBig2_BitStream* pStream) { | |
| 45 unsigned char cTemp; | |
| 46 if (pStream->read1Byte(&cTemp) == -1) | |
| 47 return false; | |
| 48 | |
| 49 HTOOB = !!(cTemp & 0x01); | |
| 50 unsigned char HTPS = ((cTemp >> 1) & 0x07) + 1; | |
| 51 unsigned char HTRS = ((cTemp >> 4) & 0x07) + 1; | |
| 52 FX_DWORD HTLOW; | |
| 53 FX_DWORD HTHIGH; | |
| 54 if (pStream->readInteger(&HTLOW) == -1 || | |
| 55 pStream->readInteger(&HTHIGH) == -1) { | |
| 56 return false; | |
| 57 } | |
| 58 | |
| 59 const int low = static_cast<int>(HTLOW); | |
| 60 const int high = static_cast<int>(HTHIGH); | |
| 61 if (low > high) | |
| 62 return false; | |
| 63 | |
| 64 ExtendBuffers(false); | |
| 65 int cur_low = low; | |
| 66 do { | |
| 67 if ((pStream->readNBits(HTPS, &PREFLEN[NTEMP]) == -1) || | |
| 68 (pStream->readNBits(HTRS, &RANGELEN[NTEMP]) == -1)) { | |
| 69 return false; | |
| 70 } | |
| 71 RANGELOW[NTEMP] = cur_low; | |
| 72 cur_low += (1 << RANGELEN[NTEMP]); | |
| 73 ExtendBuffers(true); | |
| 74 } while (cur_low < high); | |
| 75 | |
| 76 if (pStream->readNBits(HTPS, &PREFLEN[NTEMP]) == -1) | |
| 77 return false; | |
| 78 | |
| 79 RANGELEN[NTEMP] = 32; | |
| 80 RANGELOW[NTEMP] = low - 1; | |
| 81 ExtendBuffers(true); | |
| 82 | |
| 83 if (pStream->readNBits(HTPS, &PREFLEN[NTEMP]) == -1) | |
| 84 return false; | |
| 85 | |
| 86 RANGELEN[NTEMP] = 32; | |
| 87 RANGELOW[NTEMP] = high; | |
| 88 ExtendBuffers(true); | |
| 89 | |
| 90 if (HTOOB) { | |
| 91 if (pStream->readNBits(HTPS, &PREFLEN[NTEMP]) == -1) | |
| 92 return false; | |
| 93 | |
| 94 ++NTEMP; | |
| 95 } | |
| 96 | |
| 97 InitCodes(); | |
| 98 return true; | |
| 99 } | |
| 100 | |
| 101 void CJBig2_HuffmanTable::InitCodes() { | |
| 102 int lenmax = 0; | |
| 103 for (FX_DWORD i = 0; i < NTEMP; ++i) | |
| 104 lenmax = std::max(PREFLEN[i], lenmax); | |
| 105 | |
| 106 CODES.resize(NTEMP); | |
| 107 std::vector<int> LENCOUNT(lenmax + 1); | |
| 108 std::vector<int> FIRSTCODE(lenmax + 1); | |
| 109 for (int len : PREFLEN) | |
| 110 ++LENCOUNT[len]; | |
| 111 | |
| 112 FIRSTCODE[0] = 0; | |
| 113 LENCOUNT[0] = 0; | |
| 114 for (int i = 1; i <= lenmax; ++i) { | |
| 115 FIRSTCODE[i] = (FIRSTCODE[i - 1] + LENCOUNT[i - 1]) << 1; | |
| 116 int CURCODE = FIRSTCODE[i]; | |
| 117 for (FX_DWORD j = 0; j < NTEMP; ++j) { | |
| 118 if (PREFLEN[j] == i) | |
| 119 CODES[j] = CURCODE++; | |
| 120 } | |
| 121 } | |
| 122 } | |
| 123 | |
| 124 void CJBig2_HuffmanTable::ExtendBuffers(bool increment) { | |
| 125 if (increment) | |
| 126 ++NTEMP; | |
| 127 | |
| 128 size_t size = PREFLEN.size(); | |
| 129 if (NTEMP < size) | |
| 130 return; | |
| 131 | |
| 132 size += 16; | |
| 133 ASSERT(NTEMP < size); | |
| 134 PREFLEN.resize(size); | |
| 135 RANGELEN.resize(size); | |
| 136 RANGELOW.resize(size); | |
| 137 } | |
| OLD | NEW |