Move core/fpdfapi/fpdf_page to core/fpdfapi/page

BUG=pdfium:603

Review-Url: https://codereview.chromium.org/2386423004

1 files changed, 411 insertions, 0 deletions

diff --git a/core/fpdfapi/page/cpdf_image.cpp b/core/fpdfapi/page/cpdf_image.cpp
new file mode 100644
index 0000000000..eb59dc750e
--- /dev/null
+++ b/core/fpdfapi/page/cpdf_image.cpp
@@ -0,0 +1,411 @@
+// 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/page/cpdf_image.h"
+
+#include <algorithm>
+#include <memory>
+#include <vector>
+
+#include "core/fpdfapi/cpdf_modulemgr.h"
+#include "core/fpdfapi/fpdf_parser/cpdf_array.h"
+#include "core/fpdfapi/fpdf_parser/cpdf_boolean.h"
+#include "core/fpdfapi/fpdf_parser/cpdf_document.h"
+#include "core/fpdfapi/fpdf_parser/cpdf_string.h"
+#include "core/fpdfapi/fpdf_render/cpdf_pagerendercache.h"
+#include "core/fpdfapi/fpdf_render/render_int.h"
+#include "core/fpdfapi/page/cpdf_page.h"
+#include "core/fpdfapi/page/pageint.h"
+#include "core/fxcodec/fx_codec.h"
+#include "core/fxge/fx_dib.h"
+
+CPDF_Image::CPDF_Image(CPDF_Document* pDoc)
+    : CPDF_Image(pDoc, nullptr, false) {}
+
+CPDF_Image::CPDF_Image(CPDF_Document* pDoc, CPDF_Stream* pStream, bool bInline)
+    : m_pDIBSource(nullptr),
+      m_pMask(nullptr),
+      m_MatteColor(0),
+      m_pStream(pStream),
+      m_bInline(bInline),
+      m_pInlineDict(nullptr),
+      m_Height(0),
+      m_Width(0),
+      m_bIsMask(false),
+      m_bInterpolate(false),
+      m_pDocument(pDoc),
+      m_pOC(nullptr) {
+  if (!pStream)
+    return;
+
+  CPDF_Dictionary* pDict = pStream->GetDict();
+  if (m_bInline)
+    m_pInlineDict = ToDictionary(pDict->Clone());
+
+  m_pOC = pDict->GetDictFor("OC");
+  m_bIsMask =
+      !pDict->KeyExist("ColorSpace") || pDict->GetIntegerFor("ImageMask");
+  m_bInterpolate = !!pDict->GetIntegerFor("Interpolate");
+  m_Height = pDict->GetIntegerFor("Height");
+  m_Width = pDict->GetIntegerFor("Width");
+}
+
+CPDF_Image::~CPDF_Image() {
+  if (m_bInline) {
+    if (m_pStream)
+      m_pStream->Release();
+    if (m_pInlineDict)
+      m_pInlineDict->Release();
+  }
+}
+
+CPDF_Image* CPDF_Image::Clone() {
+  if (m_pStream->GetObjNum())
+    return m_pDocument->GetPageData()->GetImage(m_pStream);
+
+  CPDF_Image* pImage =
+      new CPDF_Image(m_pDocument, ToStream(m_pStream->Clone()), m_bInline);
+  if (m_bInline)
+    pImage->SetInlineDict(ToDictionary(m_pInlineDict->CloneDirectObject()));
+
+  return pImage;
+}
+
+CPDF_Dictionary* CPDF_Image::InitJPEG(uint8_t* pData, uint32_t size) {
+  int32_t width;
+  int32_t height;
+  int32_t num_comps;
+  int32_t bits;
+  bool color_trans;
+  if (!CPDF_ModuleMgr::Get()->GetJpegModule()->LoadInfo(
+          pData, size, &width, &height, &num_comps, &bits, &color_trans)) {
+    return nullptr;
+  }
+
+  CPDF_Dictionary* pDict =
+      new CPDF_Dictionary(m_pDocument->GetByteStringPool());
+  pDict->SetNameFor("Type", "XObject");
+  pDict->SetNameFor("Subtype", "Image");
+  pDict->SetIntegerFor("Width", width);
+  pDict->SetIntegerFor("Height", height);
+  const FX_CHAR* csname = nullptr;
+  if (num_comps == 1) {
+    csname = "DeviceGray";
+  } else if (num_comps == 3) {
+    csname = "DeviceRGB";
+  } else if (num_comps == 4) {
+    csname = "DeviceCMYK";
+    CPDF_Array* pDecode = new CPDF_Array;
+    for (int n = 0; n < 4; n++) {
+      pDecode->AddInteger(1);
+      pDecode->AddInteger(0);
+    }
+    pDict->SetFor("Decode", pDecode);
+  }
+  pDict->SetNameFor("ColorSpace", csname);
+  pDict->SetIntegerFor("BitsPerComponent", bits);
+  pDict->SetNameFor("Filter", "DCTDecode");
+  if (!color_trans) {
+    CPDF_Dictionary* pParms =
+        new CPDF_Dictionary(m_pDocument->GetByteStringPool());
+    pDict->SetFor("DecodeParms", pParms);
+    pParms->SetIntegerFor("ColorTransform", 0);
+  }
+  m_bIsMask = FALSE;
+  m_Width = width;
+  m_Height = height;
+  if (!m_pStream)
+    m_pStream = new CPDF_Stream;
+  return pDict;
+}
+
+void CPDF_Image::SetJpegImage(IFX_FileRead* pFile) {
+  uint32_t size = (uint32_t)pFile->GetSize();
+  if (!size)
+    return;
+
+  uint32_t dwEstimateSize = std::min(size, 8192U);
+  std::vector<uint8_t> data(dwEstimateSize);
+  pFile->ReadBlock(data.data(), 0, dwEstimateSize);
+  CPDF_Dictionary* pDict = InitJPEG(data.data(), dwEstimateSize);
+  if (!pDict && size > dwEstimateSize) {
+    data.resize(size);
+    pFile->ReadBlock(data.data(), 0, size);
+    pDict = InitJPEG(data.data(), size);
+  }
+  if (!pDict)
+    return;
+
+  m_pStream->InitStreamFromFile(pFile, pDict);
+}
+
+void CPDF_Image::SetImage(const CFX_DIBitmap* pBitmap, int32_t iCompress) {
+  int32_t BitmapWidth = pBitmap->GetWidth();
+  int32_t BitmapHeight = pBitmap->GetHeight();
+  if (BitmapWidth < 1 || BitmapHeight < 1) {
+    return;
+  }
+  uint8_t* src_buf = pBitmap->GetBuffer();
+  int32_t src_pitch = pBitmap->GetPitch();
+  int32_t bpp = pBitmap->GetBPP();
+
+  CPDF_Dictionary* pDict =
+      new CPDF_Dictionary(m_pDocument->GetByteStringPool());
+  pDict->SetNameFor("Type", "XObject");
+  pDict->SetNameFor("Subtype", "Image");
+  pDict->SetIntegerFor("Width", BitmapWidth);
+  pDict->SetIntegerFor("Height", BitmapHeight);
+  uint8_t* dest_buf = nullptr;
+  FX_STRSIZE dest_pitch = 0, dest_size = 0, opType = -1;
+  if (bpp == 1) {
+    int32_t reset_a = 0, reset_r = 0, reset_g = 0, reset_b = 0;
+    int32_t set_a = 0, set_r = 0, set_g = 0, set_b = 0;
+    if (!pBitmap->IsAlphaMask()) {
+      ArgbDecode(pBitmap->GetPaletteArgb(0), reset_a, reset_r, reset_g,
+                 reset_b);
+      ArgbDecode(pBitmap->GetPaletteArgb(1), set_a, set_r, set_g, set_b);
+    }
+    if (set_a == 0 || reset_a == 0) {
+      pDict->SetFor("ImageMask", new CPDF_Boolean(TRUE));
+      if (reset_a == 0) {
+        CPDF_Array* pArray = new CPDF_Array;
+        pArray->AddInteger(1);
+        pArray->AddInteger(0);
+        pDict->SetFor("Decode", pArray);
+      }
+    } else {
+      CPDF_Array* pCS = new CPDF_Array;
+      pCS->AddName("Indexed");
+      pCS->AddName("DeviceRGB");
+      pCS->AddInteger(1);
+      CFX_ByteString ct;
+      FX_CHAR* pBuf = ct.GetBuffer(6);
+      pBuf[0] = (FX_CHAR)reset_r;
+      pBuf[1] = (FX_CHAR)reset_g;
+      pBuf[2] = (FX_CHAR)reset_b;
+      pBuf[3] = (FX_CHAR)set_r;
+      pBuf[4] = (FX_CHAR)set_g;
+      pBuf[5] = (FX_CHAR)set_b;
+      ct.ReleaseBuffer(6);
+      pCS->Add(new CPDF_String(ct, TRUE));
+      pDict->SetFor("ColorSpace", pCS);
+    }
+    pDict->SetIntegerFor("BitsPerComponent", 1);
+    dest_pitch = (BitmapWidth + 7) / 8;
+    if ((iCompress & 0x03) == PDF_IMAGE_NO_COMPRESS) {
+      opType = 1;
+    } else {
+      opType = 0;
+    }
+  } else if (bpp == 8) {
+    int32_t iPalette = pBitmap->GetPaletteSize();
+    if (iPalette > 0) {
+      CPDF_Array* pCS = new CPDF_Array;
+      pCS->AddName("Indexed");
+      pCS->AddName("DeviceRGB");
+      pCS->AddInteger(iPalette - 1);
+      uint8_t* pColorTable = FX_Alloc2D(uint8_t, iPalette, 3);
+      uint8_t* ptr = pColorTable;
+      for (int32_t i = 0; i < iPalette; i++) {
+        uint32_t argb = pBitmap->GetPaletteArgb(i);
+        ptr[0] = (uint8_t)(argb >> 16);
+        ptr[1] = (uint8_t)(argb >> 8);
+        ptr[2] = (uint8_t)argb;
+        ptr += 3;
+      }
+      CPDF_Stream* pCTS = new CPDF_Stream(
+          pColorTable, iPalette * 3,
+          new CPDF_Dictionary(m_pDocument->GetByteStringPool()));
+      pCS->AddReference(m_pDocument, m_pDocument->AddIndirectObject(pCTS));
+      pDict->SetReferenceFor("ColorSpace", m_pDocument,
+                             m_pDocument->AddIndirectObject(pCS));
+    } else {
+      pDict->SetNameFor("ColorSpace", "DeviceGray");
+    }
+    pDict->SetIntegerFor("BitsPerComponent", 8);
+    if ((iCompress & 0x03) == PDF_IMAGE_NO_COMPRESS) {
+      dest_pitch = BitmapWidth;
+      opType = 1;
+    } else {
+      opType = 0;
+    }
+  } else {
+    pDict->SetNameFor("ColorSpace", "DeviceRGB");
+    pDict->SetIntegerFor("BitsPerComponent", 8);
+    if ((iCompress & 0x03) == PDF_IMAGE_NO_COMPRESS) {
+      dest_pitch = BitmapWidth * 3;
+      opType = 2;
+    } else {
+      opType = 0;
+    }
+  }
+  const CFX_DIBitmap* pMaskBitmap = nullptr;
+  FX_BOOL bDeleteMask = FALSE;
+  if (pBitmap->HasAlpha()) {
+    pMaskBitmap = pBitmap->GetAlphaMask();
+    bDeleteMask = TRUE;
+  }
+  if (pMaskBitmap) {
+    int32_t maskWidth = pMaskBitmap->GetWidth();
+    int32_t maskHeight = pMaskBitmap->GetHeight();
+    uint8_t* mask_buf = nullptr;
+    FX_STRSIZE mask_size = 0;
+    CPDF_Dictionary* pMaskDict =
+        new CPDF_Dictionary(m_pDocument->GetByteStringPool());
+    pMaskDict->SetNameFor("Type", "XObject");
+    pMaskDict->SetNameFor("Subtype", "Image");
+    pMaskDict->SetIntegerFor("Width", maskWidth);
+    pMaskDict->SetIntegerFor("Height", maskHeight);
+    pMaskDict->SetNameFor("ColorSpace", "DeviceGray");
+    pMaskDict->SetIntegerFor("BitsPerComponent", 8);
+    if (pMaskBitmap->GetBPP() == 8 &&
+        (iCompress & PDF_IMAGE_MASK_LOSSY_COMPRESS) != 0) {
+    } else if (pMaskBitmap->GetFormat() == FXDIB_1bppMask) {
+    } else {
+      mask_buf = FX_Alloc2D(uint8_t, maskHeight, maskWidth);
+      mask_size = maskHeight * maskWidth;  // Safe since checked alloc returned.
+      for (int32_t a = 0; a < maskHeight; a++) {
+        FXSYS_memcpy(mask_buf + a * maskWidth, pMaskBitmap->GetScanline(a),
+                     maskWidth);
+      }
+    }
+    pMaskDict->SetIntegerFor("Length", mask_size);
+    pDict->SetReferenceFor("SMask", m_pDocument,
+                           m_pDocument->AddIndirectObject(new CPDF_Stream(
+                               mask_buf, mask_size, pMaskDict)));
+    if (bDeleteMask)
+      delete pMaskBitmap;
+  }
+  if (opType == 0) {
+    if (iCompress & PDF_IMAGE_LOSSLESS_COMPRESS) {
+    } else {
+      if (pBitmap->GetBPP() == 1) {
+      } else if (pBitmap->GetBPP() >= 8 && pBitmap->GetPalette()) {
+        CFX_DIBitmap* pNewBitmap = new CFX_DIBitmap();
+        pNewBitmap->Copy(pBitmap);
+        pNewBitmap->ConvertFormat(FXDIB_Rgb);
+        SetImage(pNewBitmap, iCompress);
+        if (pDict) {
+          pDict->Release();
+          pDict = nullptr;
+        }
+        FX_Free(dest_buf);
+        dest_buf = nullptr;
+        dest_size = 0;
+        delete pNewBitmap;
+        return;
+      }
+    }
+  } else if (opType == 1) {
+    dest_buf = FX_Alloc2D(uint8_t, dest_pitch, BitmapHeight);
+    dest_size = dest_pitch * BitmapHeight;  // Safe as checked alloc returned.
+
+    uint8_t* pDest = dest_buf;
+    for (int32_t i = 0; i < BitmapHeight; i++) {
+      FXSYS_memcpy(pDest, src_buf, dest_pitch);
+      pDest += dest_pitch;
+      src_buf += src_pitch;
+    }
+  } else if (opType == 2) {
+    dest_buf = FX_Alloc2D(uint8_t, dest_pitch, BitmapHeight);
+    dest_size = dest_pitch * BitmapHeight;  // Safe as checked alloc returned.
+
+    uint8_t* pDest = dest_buf;
+    int32_t src_offset = 0;
+    int32_t dest_offset = 0;
+    for (int32_t row = 0; row < BitmapHeight; row++) {
+      src_offset = row * src_pitch;
+      for (int32_t column = 0; column < BitmapWidth; column++) {
+        FX_FLOAT alpha = 1;
+        pDest[dest_offset] = (uint8_t)(src_buf[src_offset + 2] * alpha);
+        pDest[dest_offset + 1] = (uint8_t)(src_buf[src_offset + 1] * alpha);
+        pDest[dest_offset + 2] = (uint8_t)(src_buf[src_offset] * alpha);
+        dest_offset += 3;
+        src_offset += bpp == 24 ? 3 : 4;
+      }
+
+      pDest += dest_pitch;
+      dest_offset = 0;
+    }
+  }
+  if (!m_pStream)
+    m_pStream = new CPDF_Stream;
+
+  m_pStream->InitStream(dest_buf, dest_size, pDict);
+  m_bIsMask = pBitmap->IsAlphaMask();
+  m_Width = BitmapWidth;
+  m_Height = BitmapHeight;
+  FX_Free(dest_buf);
+}
+
+void CPDF_Image::ResetCache(CPDF_Page* pPage, const CFX_DIBitmap* pBitmap) {
+  pPage->GetRenderCache()->ResetBitmap(m_pStream, pBitmap);
+}
+
+CFX_DIBSource* CPDF_Image::LoadDIBSource(CFX_DIBSource** ppMask,
+                                         uint32_t* pMatteColor,
+                                         FX_BOOL bStdCS,
+                                         uint32_t GroupFamily,
+                                         FX_BOOL bLoadMask) const {
+  std::unique_ptr<CPDF_DIBSource> source(new CPDF_DIBSource);
+  if (source->Load(m_pDocument, m_pStream,
+                   reinterpret_cast<CPDF_DIBSource**>(ppMask), pMatteColor,
+                   nullptr, nullptr, bStdCS, GroupFamily, bLoadMask)) {
+    return source.release();
+  }
+  return nullptr;
+}
+
+CFX_DIBSource* CPDF_Image::DetachBitmap() {
+  CFX_DIBSource* pBitmap = m_pDIBSource;
+  m_pDIBSource = nullptr;
+  return pBitmap;
+}
+
+CFX_DIBSource* CPDF_Image::DetachMask() {
+  CFX_DIBSource* pBitmap = m_pMask;
+  m_pMask = nullptr;
+  return pBitmap;
+}
+
+FX_BOOL CPDF_Image::StartLoadDIBSource(CPDF_Dictionary* pFormResource,
+                                       CPDF_Dictionary* pPageResource,
+                                       FX_BOOL bStdCS,
+                                       uint32_t GroupFamily,
+                                       FX_BOOL bLoadMask) {
+  std::unique_ptr<CPDF_DIBSource> source(new CPDF_DIBSource);
+  int ret =
+      source->StartLoadDIBSource(m_pDocument, m_pStream, TRUE, pFormResource,
+                                 pPageResource, bStdCS, GroupFamily, bLoadMask);
+  if (ret == 2) {
+    m_pDIBSource = source.release();
+    return TRUE;
+  }
+  if (!ret) {
+    m_pDIBSource = nullptr;
+    return FALSE;
+  }
+  m_pMask = source->DetachMask();
+  m_MatteColor = source->GetMatteColor();
+  m_pDIBSource = source.release();
+  return FALSE;
+}
+
+FX_BOOL CPDF_Image::Continue(IFX_Pause* pPause) {
+  CPDF_DIBSource* pSource = static_cast<CPDF_DIBSource*>(m_pDIBSource);
+  int ret = pSource->ContinueLoadDIBSource(pPause);
+  if (ret == 2) {
+    return TRUE;
+  }
+  if (!ret) {
+    delete m_pDIBSource;
+    m_pDIBSource = nullptr;
+    return FALSE;
+  }
+  m_pMask = pSource->DetachMask();
+  m_MatteColor = pSource->GetMatteColor();
+  return FALSE;
+}