1 files changed, 119 insertions, 0 deletions
diff --git a/core/fpdfapi/fpdf_page/cpdf_meshstream.cpp b/core/fpdfapi/fpdf_page/cpdf_meshstream.cpp
new file mode 100644
index 0000000000..1d983d2b25
--- /dev/null
+++ b/core/fpdfapi/fpdf_page/cpdf_meshstream.cpp
@@ -0,0 +1,119 @@
+// Copyright 2016 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 "core/fpdfapi/fpdf_page/cpdf_meshstream.h"
+
+#include "core/fpdfapi/fpdf_page/include/cpdf_colorspace.h"
+#include "core/fpdfapi/fpdf_page/pageint.h"
+#include "core/fpdfapi/fpdf_parser/include/cpdf_array.h"
+
+FX_BOOL CPDF_MeshStream::Load(CPDF_Stream* pShadingStream,
+                              CPDF_Function** pFuncs,
+                              int nFuncs,
+                              CPDF_ColorSpace* pCS) {
+  m_Stream.LoadAllData(pShadingStream);
+  m_BitStream.Init(m_Stream.GetData(), m_Stream.GetSize());
+  m_pFuncs = pFuncs;
+  m_nFuncs = nFuncs;
+  m_pCS = pCS;
+  CPDF_Dictionary* pDict = pShadingStream->GetDict();
+  m_nCoordBits = pDict->GetIntegerBy("BitsPerCoordinate");
+  m_nCompBits = pDict->GetIntegerBy("BitsPerComponent");
+  m_nFlagBits = pDict->GetIntegerBy("BitsPerFlag");
+  if (!m_nCoordBits || !m_nCompBits)
+    return FALSE;
+
+  FX_DWORD nComps = pCS->CountComponents();
+  if (nComps > 8)
+    return FALSE;
+
+  m_nComps = nFuncs ? 1 : nComps;
+  if (((int)m_nComps < 0) || m_nComps > 8)
+    return FALSE;
+
+  m_CoordMax = m_nCoordBits == 32 ? -1 : (1 << m_nCoordBits) - 1;
+  m_CompMax = (1 << m_nCompBits) - 1;
+  CPDF_Array* pDecode = pDict->GetArrayBy("Decode");
+  if (!pDecode || pDecode->GetCount() != 4 + m_nComps * 2)
+    return FALSE;
+
+  m_xmin = pDecode->GetNumberAt(0);
+  m_xmax = pDecode->GetNumberAt(1);
+  m_ymin = pDecode->GetNumberAt(2);
+  m_ymax = pDecode->GetNumberAt(3);
+  for (FX_DWORD i = 0; i < m_nComps; i++) {
+    m_ColorMin[i] = pDecode->GetNumberAt(i * 2 + 4);
+    m_ColorMax[i] = pDecode->GetNumberAt(i * 2 + 5);
+  }
+  return TRUE;
+}
+
+FX_DWORD CPDF_MeshStream::GetFlag() {
+  return m_BitStream.GetBits(m_nFlagBits) & 0x03;
+}
+
+void CPDF_MeshStream::GetCoords(FX_FLOAT& x, FX_FLOAT& y) {
+  if (m_nCoordBits == 32) {
+    x = m_xmin + (FX_FLOAT)(m_BitStream.GetBits(m_nCoordBits) *
+                            (m_xmax - m_xmin) / (double)m_CoordMax);
+    y = m_ymin + (FX_FLOAT)(m_BitStream.GetBits(m_nCoordBits) *
+                            (m_ymax - m_ymin) / (double)m_CoordMax);
+  } else {
+    x = m_xmin +
+        m_BitStream.GetBits(m_nCoordBits) * (m_xmax - m_xmin) / m_CoordMax;
+    y = m_ymin +
+        m_BitStream.GetBits(m_nCoordBits) * (m_ymax - m_ymin) / m_CoordMax;
+  }
+}
+
+void CPDF_MeshStream::GetColor(FX_FLOAT& r, FX_FLOAT& g, FX_FLOAT& b) {
+  FX_DWORD i;
+  FX_FLOAT color_value[8];
+  for (i = 0; i < m_nComps; i++) {
+    color_value[i] = m_ColorMin[i] +
+                     m_BitStream.GetBits(m_nCompBits) *
+                         (m_ColorMax[i] - m_ColorMin[i]) / m_CompMax;
+  }
+  if (m_nFuncs) {
+    static const int kMaxResults = 8;
+    FX_FLOAT result[kMaxResults];
+    int nResults;
+    FXSYS_memset(result, 0, sizeof(result));
+    for (FX_DWORD i = 0; i < m_nFuncs; i++) {
+      if (m_pFuncs[i] && m_pFuncs[i]->CountOutputs() <= kMaxResults) {
+        m_pFuncs[i]->Call(color_value, 1, result, nResults);
+      }
+    }
+    m_pCS->GetRGB(result, r, g, b);
+  } else {
+    m_pCS->GetRGB(color_value, r, g, b);
+  }
+}
+
+FX_DWORD CPDF_MeshStream::GetVertex(CPDF_MeshVertex& vertex,
+                                    CFX_Matrix* pObject2Bitmap) {
+  FX_DWORD flag = GetFlag();
+  GetCoords(vertex.x, vertex.y);
+  pObject2Bitmap->Transform(vertex.x, vertex.y);
+  GetColor(vertex.r, vertex.g, vertex.b);
+  m_BitStream.ByteAlign();
+  return flag;
+}
+
+FX_BOOL CPDF_MeshStream::GetVertexRow(CPDF_MeshVertex* vertex,
+                                      int count,
+                                      CFX_Matrix* pObject2Bitmap) {
+  for (int i = 0; i < count; i++) {
+    if (m_BitStream.IsEOF())
+      return FALSE;
+
+    GetCoords(vertex[i].x, vertex[i].y);
+    pObject2Bitmap->Transform(vertex[i].x, vertex[i].y);
+    GetColor(vertex[i].r, vertex[i].g, vertex[i].b);
+    m_BitStream.ByteAlign();
+  }
+  return TRUE;
+}