Fix include paths for fx_system.h
[pdfium.git] / core / src / fxcodec / jbig2 / JBig2_ArithIntDecoder.cpp
index bb4e0bb..49be941 100644 (file)
-// Copyright 2014 PDFium Authors. All rights reserved.\r
-// Use of this source code is governed by a BSD-style license that can be\r
-// found in the LICENSE file.\r
\r
-// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com\r
-\r
-#include "JBig2_ArithIntDecoder.h"\r
-CJBig2_ArithIntDecoder::CJBig2_ArithIntDecoder()\r
-{\r
-    IAx = (JBig2ArithCtx*)m_pModule->JBig2_Malloc2(sizeof(JBig2ArithCtx), 512);\r
-    JBIG2_memset(IAx, 0, sizeof(JBig2ArithCtx) * 512);\r
-}\r
-CJBig2_ArithIntDecoder::~CJBig2_ArithIntDecoder()\r
-{\r
-    m_pModule->JBig2_Free(IAx);\r
-}\r
-int CJBig2_ArithIntDecoder::decode(CJBig2_ArithDecoder *pArithDecoder, int *nResult)\r
-{\r
-    int PREV, V;\r
-    int S, D;\r
-    int nNeedBits, nTemp, i;\r
-    PREV = 1;\r
-    S = pArithDecoder->DECODE(IAx + PREV);\r
-    PREV = (PREV << 1) | S;\r
-    D = pArithDecoder->DECODE(IAx + PREV);\r
-    PREV = (PREV << 1) | D;\r
-    if(D) {\r
-        D = pArithDecoder->DECODE(IAx + PREV);\r
-        PREV = (PREV << 1) | D;\r
-        if(D) {\r
-            D = pArithDecoder->DECODE(IAx + PREV);\r
-            PREV = (PREV << 1) | D;\r
-            if(D) {\r
-                D = pArithDecoder->DECODE(IAx + PREV);\r
-                PREV = (PREV << 1) | D;\r
-                if(D) {\r
-                    D = pArithDecoder->DECODE(IAx + PREV);\r
-                    PREV = (PREV << 1) | D;\r
-                    if(D) {\r
-                        nNeedBits = 32;\r
-                        V = 4436;\r
-                    } else {\r
-                        nNeedBits = 12;\r
-                        V = 340;\r
-                    }\r
-                } else {\r
-                    nNeedBits = 8;\r
-                    V = 84;\r
-                }\r
-            } else {\r
-                nNeedBits = 6;\r
-                V = 20;\r
-            }\r
-        } else {\r
-            nNeedBits = 4;\r
-            V = 4;\r
-        }\r
-    } else {\r
-        nNeedBits = 2;\r
-        V = 0;\r
-    }\r
-    nTemp = 0;\r
-    for(i = 0; i < nNeedBits; i++) {\r
-        D = pArithDecoder->DECODE(IAx + PREV);\r
-        if(PREV < 256) {\r
-            PREV = (PREV << 1) | D;\r
-        } else {\r
-            PREV = (((PREV << 1) | D) & 511) | 256;\r
-        }\r
-        nTemp = (nTemp << 1) | D;\r
-    }\r
-    V += nTemp;\r
-    if(S == 1 && V > 0) {\r
-        V = -V;\r
-    }\r
-    *nResult = V;\r
-    if(S == 1 && V == 0) {\r
-        return JBIG2_OOB;\r
-    }\r
-    return 0;\r
-}\r
-CJBig2_ArithIaidDecoder::CJBig2_ArithIaidDecoder(unsigned char SBSYMCODELENA)\r
-{\r
-    SBSYMCODELEN = SBSYMCODELENA;\r
-    IAID = (JBig2ArithCtx*)m_pModule->JBig2_Malloc2(sizeof(JBig2ArithCtx), (1 << SBSYMCODELEN));\r
-    JBIG2_memset(IAID, 0, sizeof(JBig2ArithCtx) * (int)(1 << SBSYMCODELEN));\r
-}\r
-CJBig2_ArithIaidDecoder::~CJBig2_ArithIaidDecoder()\r
-{\r
-    m_pModule->JBig2_Free(IAID);\r
-}\r
-int CJBig2_ArithIaidDecoder::decode(CJBig2_ArithDecoder *pArithDecoder, int *nResult)\r
-{\r
-    int PREV;\r
-    int D;\r
-    int i;\r
-    PREV = 1;\r
-    for(i = 0; i < SBSYMCODELEN; i++) {\r
-        D = pArithDecoder->DECODE(IAID + PREV);\r
-        PREV = (PREV << 1) | D;\r
-    }\r
-    PREV = PREV - (1 << SBSYMCODELEN);\r
-    *nResult = PREV;\r
-    return 0;\r
-}\r
+// Copyright 2014 PDFium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
+
+#include "JBig2_ArithIntDecoder.h"
+
+#include "../../../include/fxcrt/fx_basic.h"
+
+namespace {
+
+int ShiftOr(int val, int bitwise_or_val) {
+  return (val << 1) | bitwise_or_val;
+}
+
+const struct ArithIntDecodeData {
+  int nNeedBits;
+  int nValue;
+} g_ArithIntDecodeData[] = {
+    {2, 0},
+    {4, 4},
+    {6, 20},
+    {8, 84},
+    {12, 340},
+    {32, 4436},
+};
+
+size_t RecursiveDecode(CJBig2_ArithDecoder* decoder,
+                       std::vector<JBig2ArithCtx>* context,
+                       int* prev,
+                       size_t depth) {
+  static const size_t kDepthEnd = FX_ArraySize(g_ArithIntDecodeData) - 1;
+  if (depth == kDepthEnd)
+    return kDepthEnd;
+
+  JBig2ArithCtx* pCX = &(*context)[*prev];
+  int D = decoder->DECODE(pCX);
+  *prev = ShiftOr(*prev, D);
+  if (!D)
+    return depth;
+  return RecursiveDecode(decoder, context, prev, depth + 1);
+}
+
+}  // namespace
+
+CJBig2_ArithIntDecoder::CJBig2_ArithIntDecoder() {
+  m_IAx.resize(512);
+}
+
+CJBig2_ArithIntDecoder::~CJBig2_ArithIntDecoder() {
+}
+
+bool CJBig2_ArithIntDecoder::decode(CJBig2_ArithDecoder* pArithDecoder,
+                                    int* nResult) {
+  int PREV = 1;
+  const int S = pArithDecoder->DECODE(&m_IAx[PREV]);
+  PREV = ShiftOr(PREV, S);
+
+  const size_t nDecodeDataIndex =
+      RecursiveDecode(pArithDecoder, &m_IAx, &PREV, 0);
+
+  int nTemp = 0;
+  for (int i = 0; i < g_ArithIntDecodeData[nDecodeDataIndex].nNeedBits; ++i) {
+    int D = pArithDecoder->DECODE(&m_IAx[PREV]);
+    PREV = ShiftOr(PREV, D);
+    if (PREV >= 256)
+      PREV = (PREV & 511) | 256;
+    nTemp = ShiftOr(nTemp, D);
+  }
+  int nValue = g_ArithIntDecodeData[nDecodeDataIndex].nValue;
+  nValue += nTemp;
+  if (S == 1 && nValue > 0)
+    nValue = -nValue;
+
+  *nResult = nValue;
+  return S != 1 || nValue != 0;
+}
+
+CJBig2_ArithIaidDecoder::CJBig2_ArithIaidDecoder(unsigned char SBSYMCODELENA)
+    : SBSYMCODELEN(SBSYMCODELENA) {
+  m_IAID.resize(1 << SBSYMCODELEN);
+}
+
+CJBig2_ArithIaidDecoder::~CJBig2_ArithIaidDecoder() {
+}
+
+void CJBig2_ArithIaidDecoder::decode(CJBig2_ArithDecoder* pArithDecoder,
+                                     FX_DWORD* nResult) {
+  int PREV = 1;
+  for (unsigned char i = 0; i < SBSYMCODELEN; ++i) {
+    JBig2ArithCtx* pCX = &m_IAID[PREV];
+    int D = pArithDecoder->DECODE(pCX);
+    PREV = ShiftOr(PREV, D);
+  }
+  *nResult = PREV - (1 << SBSYMCODELEN);
+}