blob: d8c77477ec94603a95845dc9265942387647acd1 [file] [log] [blame] [edit]
// 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 "public/fpdf_ppo.h"
#include <algorithm>
#include <map>
#include <memory>
#include <utility>
#include <vector>
#include "constants/page_object.h"
#include "core/fpdfapi/page/cpdf_page.h"
#include "core/fpdfapi/page/cpdf_pageobject.h"
#include "core/fpdfapi/parser/cpdf_array.h"
#include "core/fpdfapi/parser/cpdf_dictionary.h"
#include "core/fpdfapi/parser/cpdf_document.h"
#include "core/fpdfapi/parser/cpdf_name.h"
#include "core/fpdfapi/parser/cpdf_number.h"
#include "core/fpdfapi/parser/cpdf_object.h"
#include "core/fpdfapi/parser/cpdf_reference.h"
#include "core/fpdfapi/parser/cpdf_stream.h"
#include "core/fpdfapi/parser/cpdf_stream_acc.h"
#include "core/fpdfapi/parser/cpdf_string.h"
#include "core/fxcrt/retain_ptr.h"
#include "core/fxcrt/unowned_ptr.h"
#include "fpdfsdk/cpdfsdk_helpers.h"
#include "public/cpp/fpdf_scopers.h"
#include "third_party/base/ptr_util.h"
namespace {
// Struct that stores sub page origin and scale information. When importing
// more than one pages onto the same page, most likely the pages will need to be
// scaled down, and scale is in range of (0, 1) exclusive.
struct NupPageSettings {
CFX_PointF subPageStartPoint;
float scale;
};
// Calculates the N-up parameters. When importing multiple pages into one page.
// The space of output page is evenly divided along the X axis and Y axis based
// on the input |numPagesOnXAxis| and |numPagesOnYAxis|.
class NupState {
public:
NupState(const CFX_SizeF& pagesize,
unsigned int numPagesOnXAxis,
unsigned int numPagesOnYAxis);
// Calculate sub page origin and scale with the source page of |pagesize| and
// new page of |m_subPageSize|.
NupPageSettings CalculateNewPagePosition(const CFX_SizeF& pagesize);
private:
// Helper function to get the subX, subY pair based on |m_subPageIndex|.
// The space of output page is evenly divided into slots along x and y axis.
// subX and subY are 0-based indices that indicate which allocation slot to
// use.
std::pair<size_t, size_t> ConvertPageOrder() const;
// Given the |subX| and |subY| subpage position within a page, and a source
// page with dimensions of |pagesize|, calculate the sub page's origin and
// scale.
NupPageSettings CalculatePageEdit(size_t subX,
size_t subY,
const CFX_SizeF& pagesize) const;
const CFX_SizeF m_destPageSize;
const size_t m_numPagesOnXAxis;
const size_t m_numPagesOnYAxis;
const size_t m_numPagesPerSheet;
CFX_SizeF m_subPageSize;
// A 0-based index, in range of [0, m_numPagesPerSheet - 1).
size_t m_subPageIndex = 0;
};
NupState::NupState(const CFX_SizeF& pagesize,
unsigned int numPagesOnXAxis,
unsigned int numPagesOnYAxis)
: m_destPageSize(pagesize),
m_numPagesOnXAxis(numPagesOnXAxis),
m_numPagesOnYAxis(numPagesOnYAxis),
m_numPagesPerSheet(numPagesOnXAxis * numPagesOnYAxis) {
ASSERT(m_numPagesOnXAxis > 0);
ASSERT(m_numPagesOnYAxis > 0);
ASSERT(m_destPageSize.width > 0);
ASSERT(m_destPageSize.height > 0);
m_subPageSize.width = m_destPageSize.width / m_numPagesOnXAxis;
m_subPageSize.height = m_destPageSize.height / m_numPagesOnYAxis;
}
std::pair<size_t, size_t> NupState::ConvertPageOrder() const {
size_t subX = m_subPageIndex % m_numPagesOnXAxis;
size_t subY = m_subPageIndex / m_numPagesOnXAxis;
// Y Axis, pages start from the top of the output page.
subY = m_numPagesOnYAxis - subY - 1;
return {subX, subY};
}
NupPageSettings NupState::CalculatePageEdit(size_t subX,
size_t subY,
const CFX_SizeF& pagesize) const {
NupPageSettings settings;
settings.subPageStartPoint.x = subX * m_subPageSize.width;
settings.subPageStartPoint.y = subY * m_subPageSize.height;
const float xScale = m_subPageSize.width / pagesize.width;
const float yScale = m_subPageSize.height / pagesize.height;
settings.scale = std::min(xScale, yScale);
float subWidth = pagesize.width * settings.scale;
float subHeight = pagesize.height * settings.scale;
if (xScale > yScale)
settings.subPageStartPoint.x += (m_subPageSize.width - subWidth) / 2;
else
settings.subPageStartPoint.y += (m_subPageSize.height - subHeight) / 2;
return settings;
}
NupPageSettings NupState::CalculateNewPagePosition(const CFX_SizeF& pagesize) {
if (m_subPageIndex >= m_numPagesPerSheet)
m_subPageIndex = 0;
size_t subX;
size_t subY;
std::tie(subX, subY) = ConvertPageOrder();
++m_subPageIndex;
return CalculatePageEdit(subX, subY, pagesize);
}
const CPDF_Object* PageDictGetInheritableTag(const CPDF_Dictionary* pDict,
const ByteString& bsSrcTag) {
if (!pDict || bsSrcTag.IsEmpty())
return nullptr;
if (!pDict->KeyExist(pdfium::page_object::kParent) ||
!pDict->KeyExist(pdfium::page_object::kType)) {
return nullptr;
}
const CPDF_Object* pType =
pDict->GetObjectFor(pdfium::page_object::kType)->GetDirect();
if (!ToName(pType))
return nullptr;
if (pType->GetString().Compare("Page"))
return nullptr;
const CPDF_Dictionary* pp = ToDictionary(
pDict->GetObjectFor(pdfium::page_object::kParent)->GetDirect());
if (!pp)
return nullptr;
if (pDict->KeyExist(bsSrcTag))
return pDict->GetObjectFor(bsSrcTag);
while (pp) {
if (pp->KeyExist(bsSrcTag))
return pp->GetObjectFor(bsSrcTag);
if (!pp->KeyExist(pdfium::page_object::kParent))
break;
pp = ToDictionary(
pp->GetObjectFor(pdfium::page_object::kParent)->GetDirect());
}
return nullptr;
}
CFX_FloatRect GetMediaBox(const CPDF_Dictionary* pPageDict) {
const CPDF_Object* pMediaBox =
PageDictGetInheritableTag(pPageDict, pdfium::page_object::kMediaBox);
const CPDF_Array* pArray = ToArray(pMediaBox->GetDirect());
if (!pArray)
return CFX_FloatRect();
return pArray->GetRect();
}
CFX_FloatRect GetCropBox(const CPDF_Dictionary* pPageDict) {
if (pPageDict->KeyExist(pdfium::page_object::kCropBox))
return pPageDict->GetRectFor(pdfium::page_object::kCropBox);
return GetMediaBox(pPageDict);
}
const CPDF_Object* GetPageOrganizerPageContent(
const CPDF_Dictionary* pPageDict) {
return pPageDict
? pPageDict->GetDirectObjectFor(pdfium::page_object::kContents)
: nullptr;
}
bool CopyInheritable(CPDF_Dictionary* pDestPageDict,
const CPDF_Dictionary* pSrcPageDict,
const ByteString& key) {
if (pDestPageDict->KeyExist(key))
return true;
const CPDF_Object* pInheritable =
PageDictGetInheritableTag(pSrcPageDict, key);
if (!pInheritable)
return false;
pDestPageDict->SetFor(key, pInheritable->Clone());
return true;
}
bool ParsePageRangeString(const ByteString& bsPageRange,
uint32_t nCount,
std::vector<uint32_t>* pageArray) {
ByteString bsStrippedPageRange = bsPageRange;
bsStrippedPageRange.Remove(' ');
size_t nLength = bsStrippedPageRange.GetLength();
if (nLength == 0)
return true;
static const ByteString cbCompareString("0123456789-,");
for (size_t i = 0; i < nLength; ++i) {
if (!cbCompareString.Contains(bsStrippedPageRange[i]))
return false;
}
ByteString cbMidRange;
size_t nStringFrom = 0;
size_t nStringTo = 0;
while (nStringTo < nLength) {
nStringTo = bsStrippedPageRange.Find(',', nStringFrom).value_or(nLength);
cbMidRange = bsStrippedPageRange.Mid(nStringFrom, nStringTo - nStringFrom);
Optional<size_t> nDashPosition = cbMidRange.Find('-');
if (nDashPosition) {
size_t nMid = nDashPosition.value();
uint32_t nStartPageNum = pdfium::base::checked_cast<uint32_t>(
atoi(cbMidRange.Left(nMid).c_str()));
if (nStartPageNum == 0)
return false;
++nMid;
size_t nEnd = cbMidRange.GetLength() - nMid;
if (nEnd == 0)
return false;
uint32_t nEndPageNum = pdfium::base::checked_cast<uint32_t>(
atoi(cbMidRange.Mid(nMid, nEnd).c_str()));
if (nStartPageNum < 0 || nStartPageNum > nEndPageNum ||
nEndPageNum > nCount) {
return false;
}
for (uint32_t i = nStartPageNum; i <= nEndPageNum; ++i) {
pageArray->push_back(i);
}
} else {
uint32_t nPageNum =
pdfium::base::checked_cast<uint32_t>(atoi(cbMidRange.c_str()));
if (nPageNum <= 0 || nPageNum > nCount)
return false;
pageArray->push_back(nPageNum);
}
nStringFrom = nStringTo + 1;
}
return true;
}
std::vector<uint32_t> GetPageNumbers(const CPDF_Document& doc,
const ByteString& bsPageRange) {
std::vector<uint32_t> page_numbers;
uint32_t nCount = doc.GetPageCount();
if (bsPageRange.IsEmpty()) {
for (uint32_t i = 1; i <= nCount; ++i)
page_numbers.push_back(i);
} else {
if (!ParsePageRangeString(bsPageRange, nCount, &page_numbers))
page_numbers.clear();
}
return page_numbers;
}
class CPDF_PageOrganizer {
protected:
// Map source page object number to XObject object number.
using ObjectNumberMap = std::map<uint32_t, uint32_t>;
CPDF_PageOrganizer(CPDF_Document* pDestPDFDoc, CPDF_Document* pSrcPDFDoc);
~CPDF_PageOrganizer();
// Must be called after construction before doing anything else.
bool PDFDocInit();
bool UpdateReference(CPDF_Object* pObj, ObjectNumberMap* pObjNumberMap);
CPDF_Document* dest() { return m_pDestPDFDoc.Get(); }
const CPDF_Document* dest() const { return m_pDestPDFDoc.Get(); }
CPDF_Document* src() { return m_pSrcPDFDoc.Get(); }
const CPDF_Document* src() const { return m_pSrcPDFDoc.Get(); }
private:
uint32_t GetNewObjId(ObjectNumberMap* pObjNumberMap, CPDF_Reference* pRef);
UnownedPtr<CPDF_Document> const m_pDestPDFDoc;
UnownedPtr<CPDF_Document> const m_pSrcPDFDoc;
};
CPDF_PageOrganizer::CPDF_PageOrganizer(CPDF_Document* pDestPDFDoc,
CPDF_Document* pSrcPDFDoc)
: m_pDestPDFDoc(pDestPDFDoc), m_pSrcPDFDoc(pSrcPDFDoc) {}
CPDF_PageOrganizer::~CPDF_PageOrganizer() {}
bool CPDF_PageOrganizer::PDFDocInit() {
ASSERT(m_pDestPDFDoc);
ASSERT(m_pSrcPDFDoc);
CPDF_Dictionary* pNewRoot = dest()->GetRoot();
if (!pNewRoot)
return false;
CPDF_Dictionary* pDocInfoDict = dest()->GetInfo();
if (!pDocInfoDict)
return false;
pDocInfoDict->SetNewFor<CPDF_String>("Producer", "PDFium", false);
ByteString cbRootType = pNewRoot->GetStringFor("Type", ByteString());
if (cbRootType.IsEmpty())
pNewRoot->SetNewFor<CPDF_Name>("Type", "Catalog");
CPDF_Object* pElement = pNewRoot->GetObjectFor("Pages");
CPDF_Dictionary* pNewPages =
pElement ? ToDictionary(pElement->GetDirect()) : nullptr;
if (!pNewPages) {
pNewPages = dest()->NewIndirect<CPDF_Dictionary>();
pNewRoot->SetFor("Pages", pNewPages->MakeReference(dest()));
}
ByteString cbPageType = pNewPages->GetStringFor("Type", ByteString());
if (cbPageType.IsEmpty())
pNewPages->SetNewFor<CPDF_Name>("Type", "Pages");
if (!pNewPages->GetArrayFor("Kids")) {
pNewPages->SetNewFor<CPDF_Number>("Count", 0);
pNewPages->SetFor("Kids",
dest()->NewIndirect<CPDF_Array>()->MakeReference(dest()));
}
return true;
}
bool CPDF_PageOrganizer::UpdateReference(CPDF_Object* pObj,
ObjectNumberMap* pObjNumberMap) {
switch (pObj->GetType()) {
case CPDF_Object::kReference: {
CPDF_Reference* pReference = pObj->AsReference();
uint32_t newobjnum = GetNewObjId(pObjNumberMap, pReference);
if (newobjnum == 0)
return false;
pReference->SetRef(dest(), newobjnum);
break;
}
case CPDF_Object::kDictionary: {
CPDF_Dictionary* pDict = pObj->AsDictionary();
std::vector<ByteString> bad_keys;
{
CPDF_DictionaryLocker locker(pDict);
for (auto it = locker.begin(); it != locker.end(); ++it) {
const ByteString& key = it->first;
if (key == "Parent" || key == "Prev" || key == "First")
continue;
CPDF_Object* pNextObj = it->second.get();
if (!pNextObj)
return false;
if (!UpdateReference(pNextObj, pObjNumberMap))
bad_keys.push_back(key);
}
}
for (const auto& key : bad_keys)
pDict->RemoveFor(key);
break;
}
case CPDF_Object::kArray: {
CPDF_Array* pArray = pObj->AsArray();
for (size_t i = 0; i < pArray->size(); ++i) {
CPDF_Object* pNextObj = pArray->GetObjectAt(i);
if (!pNextObj)
return false;
if (!UpdateReference(pNextObj, pObjNumberMap))
return false;
}
break;
}
case CPDF_Object::kStream: {
CPDF_Stream* pStream = pObj->AsStream();
CPDF_Dictionary* pDict = pStream->GetDict();
if (!pDict)
return false;
if (!UpdateReference(pDict, pObjNumberMap))
return false;
break;
}
default:
break;
}
return true;
}
uint32_t CPDF_PageOrganizer::GetNewObjId(ObjectNumberMap* pObjNumberMap,
CPDF_Reference* pRef) {
if (!pRef)
return 0;
uint32_t dwObjnum = pRef->GetRefObjNum();
uint32_t dwNewObjNum = 0;
const auto it = pObjNumberMap->find(dwObjnum);
if (it != pObjNumberMap->end())
dwNewObjNum = it->second;
if (dwNewObjNum)
return dwNewObjNum;
CPDF_Object* pDirect = pRef->GetDirect();
if (!pDirect)
return 0;
std::unique_ptr<CPDF_Object> pClone = pDirect->Clone();
if (CPDF_Dictionary* pDictClone = pClone->AsDictionary()) {
if (pDictClone->KeyExist("Type")) {
ByteString strType = pDictClone->GetStringFor("Type");
if (!FXSYS_stricmp(strType.c_str(), "Pages"))
return 4;
if (!FXSYS_stricmp(strType.c_str(), "Page"))
return 0;
}
}
CPDF_Object* pUnownedClone = dest()->AddIndirectObject(std::move(pClone));
dwNewObjNum = pUnownedClone->GetObjNum();
(*pObjNumberMap)[dwObjnum] = dwNewObjNum;
if (!UpdateReference(pUnownedClone, pObjNumberMap))
return 0;
return dwNewObjNum;
}
// Copies pages from a source document into a destination document.
// This class is intended to be used once via ExportPage() and then destroyed.
class CPDF_PageExporter final : public CPDF_PageOrganizer {
public:
CPDF_PageExporter(CPDF_Document* pDestPDFDoc, CPDF_Document* pSrcPDFDoc);
~CPDF_PageExporter();
// For the pages from the source document with |pageNums| as their page
// numbers, insert them into the destination document at page |nIndex|.
// |pageNums| is 1-based.
// |nIndex| is 0-based.
bool ExportPage(const std::vector<uint32_t>& pageNums, int nIndex);
};
CPDF_PageExporter::CPDF_PageExporter(CPDF_Document* pDestPDFDoc,
CPDF_Document* pSrcPDFDoc)
: CPDF_PageOrganizer(pDestPDFDoc, pSrcPDFDoc) {}
CPDF_PageExporter::~CPDF_PageExporter() = default;
bool CPDF_PageExporter::ExportPage(const std::vector<uint32_t>& pageNums,
int nIndex) {
if (!PDFDocInit())
return false;
int curpage = nIndex;
auto pObjNumberMap = pdfium::MakeUnique<ObjectNumberMap>();
for (size_t i = 0; i < pageNums.size(); ++i) {
CPDF_Dictionary* pDestPageDict = dest()->CreateNewPage(curpage);
auto* pSrcPageDict = src()->GetPageDictionary(pageNums[i] - 1);
if (!pSrcPageDict || !pDestPageDict)
return false;
// Clone the page dictionary
CPDF_DictionaryLocker locker(pSrcPageDict);
for (const auto& it : locker) {
const ByteString& cbSrcKeyStr = it.first;
if (cbSrcKeyStr == pdfium::page_object::kType ||
cbSrcKeyStr == pdfium::page_object::kParent) {
continue;
}
CPDF_Object* pObj = it.second.get();
pDestPageDict->SetFor(cbSrcKeyStr, pObj->Clone());
}
// inheritable item
// Even though some entries are required by the PDF spec, there exist
// PDFs that omit them. Set some defaults in this case.
// 1 MediaBox - required
if (!CopyInheritable(pDestPageDict, pSrcPageDict,
pdfium::page_object::kMediaBox)) {
// Search for "CropBox" in the source page dictionary.
// If it does not exist, use the default letter size.
const CPDF_Object* pInheritable = PageDictGetInheritableTag(
pSrcPageDict, pdfium::page_object::kCropBox);
if (pInheritable) {
pDestPageDict->SetFor(pdfium::page_object::kMediaBox,
pInheritable->Clone());
} else {
// Make the default size letter size (8.5"x11")
static const CFX_FloatRect kDefaultLetterRect(0, 0, 612, 792);
pDestPageDict->SetRectFor(pdfium::page_object::kMediaBox,
kDefaultLetterRect);
}
}
// 2 Resources - required
if (!CopyInheritable(pDestPageDict, pSrcPageDict,
pdfium::page_object::kResources)) {
// Use a default empty resources if it does not exist.
pDestPageDict->SetNewFor<CPDF_Dictionary>(
pdfium::page_object::kResources);
}
// 3 CropBox - optional
CopyInheritable(pDestPageDict, pSrcPageDict, pdfium::page_object::kCropBox);
// 4 Rotate - optional
CopyInheritable(pDestPageDict, pSrcPageDict, pdfium::page_object::kRotate);
// Update the reference
uint32_t dwOldPageObj = pSrcPageDict->GetObjNum();
uint32_t dwNewPageObj = pDestPageDict->GetObjNum();
(*pObjNumberMap)[dwOldPageObj] = dwNewPageObj;
UpdateReference(pDestPageDict, pObjNumberMap.get());
++curpage;
}
return true;
}
// Copies pages from a source document into a destination document. Creates 1
// page in the destination document for every N source pages. This class is
// intended to be used once via ExportNPagesToOne() and then destroyed.
class CPDF_NPageToOneExporter final : public CPDF_PageOrganizer {
public:
CPDF_NPageToOneExporter(CPDF_Document* pDestPDFDoc,
CPDF_Document* pSrcPDFDoc);
~CPDF_NPageToOneExporter();
// For the pages from the source document with |pageNums| as their page
// numbers, insert them into the destination document, starting at page 0.
// |pageNums| is 1-based.
// |destPageSize| is the destination document page dimensions, measured in
// PDF "user space" units.
// |numPagesOnXAxis| and |numPagesOnXAxis| together defines how many source
// pages fit on one destination page.
bool ExportNPagesToOne(const std::vector<uint32_t>& pageNums,
const CFX_SizeF& destPageSize,
unsigned int numPagesOnXAxis,
unsigned int numPagesOnYAxis);
private:
// Map page object number to XObject object name.
using PageXObjectMap = std::map<uint32_t, ByteString>;
// Map XObject's object name to it's object number.
using XObjectNameNumberMap = std::map<ByteString, uint32_t>;
// Creates a xobject from the source page dictionary, and appends the
// bsContent string with the xobject reference surrounded by the
// transformation matrix.
void AddSubPage(const CPDF_Dictionary* pSrcPageDict,
const NupPageSettings& settings,
ObjectNumberMap* pObjNumberMap,
PageXObjectMap* pPageXObjectMap,
XObjectNameNumberMap* pXObjNameNumberMap,
ByteString* bsContent);
uint32_t MakeXObject(const CPDF_Dictionary* pSrcPageDict,
ObjectNumberMap* pObjNumberMap);
void FinishPage(CPDF_Dictionary* pDestPageDict,
const ByteString& bsContent,
const XObjectNameNumberMap& xObjNameNumberMap);
uint32_t m_xobjectNum = 0;
XObjectNameNumberMap m_xobjs;
};
CPDF_NPageToOneExporter::CPDF_NPageToOneExporter(CPDF_Document* pDestPDFDoc,
CPDF_Document* pSrcPDFDoc)
: CPDF_PageOrganizer(pDestPDFDoc, pSrcPDFDoc) {}
CPDF_NPageToOneExporter::~CPDF_NPageToOneExporter() = default;
bool CPDF_NPageToOneExporter::ExportNPagesToOne(
const std::vector<uint32_t>& pageNums,
const CFX_SizeF& destPageSize,
unsigned int numPagesOnXAxis,
unsigned int numPagesOnYAxis) {
if (!PDFDocInit())
return false;
FX_SAFE_SIZE_T safe_numPagesPerSheet = numPagesOnXAxis;
safe_numPagesPerSheet *= numPagesOnYAxis;
if (!safe_numPagesPerSheet.IsValid())
return false;
size_t numPagesPerSheet = safe_numPagesPerSheet.ValueOrDie();
// Mapping of source page object number and XObject object number.
// Used to update refernece.
ObjectNumberMap objectNumberMap;
// Mapping of source page object number and XObject name of the entire doc.
// If there are two pages that are identical and have the same object number,
// we can reuse one created XObject.
PageXObjectMap pageXObjectMap;
NupState nupState(destPageSize, numPagesOnXAxis, numPagesOnYAxis);
size_t curpage = 0;
const CFX_FloatRect destPageRect(0, 0, destPageSize.width,
destPageSize.height);
for (size_t outerPage = 0; outerPage < pageNums.size();
outerPage += numPagesPerSheet) {
// Create a new page
CPDF_Dictionary* pDestPageDict = dest()->CreateNewPage(curpage);
if (!pDestPageDict)
return false;
pDestPageDict->SetRectFor(pdfium::page_object::kMediaBox, destPageRect);
ByteString bsContent;
size_t innerPageMax =
std::min(outerPage + numPagesPerSheet, pageNums.size());
// Mapping of XObject name and XObject object number of one page.
XObjectNameNumberMap xObjNameNumberMap;
for (size_t innerPage = outerPage; innerPage < innerPageMax; ++innerPage) {
auto* pSrcPageDict = src()->GetPageDictionary(pageNums[innerPage] - 1);
if (!pSrcPageDict)
return false;
auto srcPage = pdfium::MakeRetain<CPDF_Page>(src(), pSrcPageDict, true);
NupPageSettings settings =
nupState.CalculateNewPagePosition(srcPage->GetPageSize());
AddSubPage(pSrcPageDict, settings, &objectNumberMap, &pageXObjectMap,
&xObjNameNumberMap, &bsContent);
}
// Finish up the current page.
FinishPage(pDestPageDict, bsContent, xObjNameNumberMap);
++curpage;
}
return true;
}
void CPDF_NPageToOneExporter::AddSubPage(
const CPDF_Dictionary* pSrcPageDict,
const NupPageSettings& settings,
ObjectNumberMap* pObjNumberMap,
PageXObjectMap* pPageXObjectMap,
XObjectNameNumberMap* pXObjNameNumberMap,
ByteString* bsContent) {
uint32_t dwPageObjnum = pSrcPageDict->GetObjNum();
ByteString bsXObjectName;
const auto it = pPageXObjectMap->find(dwPageObjnum);
if (it != pPageXObjectMap->end()) {
bsXObjectName = it->second;
} else {
++m_xobjectNum;
// TODO(Xlou): A better name schema to avoid possible object name collision.
bsXObjectName = ByteString::Format("X%d", m_xobjectNum);
m_xobjs[bsXObjectName] = MakeXObject(pSrcPageDict, pObjNumberMap);
(*pPageXObjectMap)[dwPageObjnum] = bsXObjectName;
}
(*pXObjNameNumberMap)[bsXObjectName] = m_xobjs[bsXObjectName];
CFX_Matrix matrix;
matrix.Scale(settings.scale, settings.scale);
matrix.Translate(settings.subPageStartPoint.x, settings.subPageStartPoint.y);
std::ostringstream contentStream;
contentStream << "q\n"
<< matrix.a << " " << matrix.b << " " << matrix.c << " "
<< matrix.d << " " << matrix.e << " " << matrix.f << " cm\n"
<< "/" << bsXObjectName << " Do Q\n";
*bsContent += ByteString(contentStream);
}
uint32_t CPDF_NPageToOneExporter::MakeXObject(
const CPDF_Dictionary* pSrcPageDict,
ObjectNumberMap* pObjNumberMap) {
ASSERT(pSrcPageDict);
const CPDF_Object* pSrcContentObj = GetPageOrganizerPageContent(pSrcPageDict);
CPDF_Stream* pNewXObject = dest()->NewIndirect<CPDF_Stream>(
nullptr, 0, dest()->New<CPDF_Dictionary>());
CPDF_Dictionary* pNewXObjectDict = pNewXObject->GetDict();
const ByteString bsResourceString = "Resources";
if (!CopyInheritable(pNewXObjectDict, pSrcPageDict, bsResourceString)) {
// Use a default empty resources if it does not exist.
pNewXObjectDict->SetNewFor<CPDF_Dictionary>(bsResourceString);
}
uint32_t dwSrcPageObj = pSrcPageDict->GetObjNum();
uint32_t dwNewXobjectObj = pNewXObjectDict->GetObjNum();
(*pObjNumberMap)[dwSrcPageObj] = dwNewXobjectObj;
UpdateReference(pNewXObjectDict, pObjNumberMap);
pNewXObjectDict->SetNewFor<CPDF_Name>("Type", "XObject");
pNewXObjectDict->SetNewFor<CPDF_Name>("Subtype", "Form");
pNewXObjectDict->SetNewFor<CPDF_Number>("FormType", 1);
pNewXObjectDict->SetRectFor("BBox", GetCropBox(pSrcPageDict));
// TODO(xlou): add matrix field to pNewXObjectDict.
if (const CPDF_Array* pSrcContentArray = ToArray(pSrcContentObj)) {
ByteString bsSrcContentStream;
for (size_t i = 0; i < pSrcContentArray->size(); ++i) {
const CPDF_Stream* pStream = pSrcContentArray->GetStreamAt(i);
auto pAcc = pdfium::MakeRetain<CPDF_StreamAcc>(pStream);
pAcc->LoadAllDataFiltered();
bsSrcContentStream += ByteString(pAcc->GetData(), pAcc->GetSize());
bsSrcContentStream += "\n";
}
pNewXObject->SetDataAndRemoveFilter(bsSrcContentStream.AsRawSpan());
} else {
ByteString bsStream;
{
const CPDF_Stream* pStream = pSrcContentObj->AsStream();
auto pAcc = pdfium::MakeRetain<CPDF_StreamAcc>(pStream);
pAcc->LoadAllDataFiltered();
bsStream = ByteString(pAcc->GetData(), pAcc->GetSize());
}
pNewXObject->SetDataAndRemoveFilter(bsStream.AsRawSpan());
}
return pNewXObject->GetObjNum();
}
void CPDF_NPageToOneExporter::FinishPage(
CPDF_Dictionary* pDestPageDict,
const ByteString& bsContent,
const XObjectNameNumberMap& xObjNameNumberMap) {
ASSERT(pDestPageDict);
CPDF_Dictionary* pRes =
pDestPageDict->GetDictFor(pdfium::page_object::kResources);
if (!pRes) {
pRes = pDestPageDict->SetNewFor<CPDF_Dictionary>(
pdfium::page_object::kResources);
}
CPDF_Dictionary* pPageXObject = pRes->GetDictFor("XObject");
if (!pPageXObject)
pPageXObject = pRes->SetNewFor<CPDF_Dictionary>("XObject");
for (auto& it : xObjNameNumberMap)
pPageXObject->SetNewFor<CPDF_Reference>(it.first, dest(), it.second);
auto pDict = dest()->New<CPDF_Dictionary>();
CPDF_Stream* pStream =
dest()->NewIndirect<CPDF_Stream>(nullptr, 0, std::move(pDict));
pStream->SetData(bsContent.AsRawSpan());
pDestPageDict->SetFor(pdfium::page_object::kContents,
pStream->MakeReference(dest()));
}
} // namespace
FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV FPDF_ImportPages(FPDF_DOCUMENT dest_doc,
FPDF_DOCUMENT src_doc,
FPDF_BYTESTRING pagerange,
int index) {
CPDF_Document* pDestDoc = CPDFDocumentFromFPDFDocument(dest_doc);
if (!dest_doc)
return false;
CPDF_Document* pSrcDoc = CPDFDocumentFromFPDFDocument(src_doc);
if (!pSrcDoc)
return false;
std::vector<uint32_t> page_numbers = GetPageNumbers(*pSrcDoc, pagerange);
if (page_numbers.empty())
return false;
CPDF_PageExporter exporter(pDestDoc, pSrcDoc);
return exporter.ExportPage(page_numbers, index);
}
FPDF_EXPORT FPDF_DOCUMENT FPDF_CALLCONV
FPDF_ImportNPagesToOne(FPDF_DOCUMENT src_doc,
float output_width,
float output_height,
unsigned int num_pages_on_x_axis,
unsigned int num_pages_on_y_axis) {
CPDF_Document* pSrcDoc = CPDFDocumentFromFPDFDocument(src_doc);
if (!pSrcDoc)
return nullptr;
if (output_width <= 0 || output_height <= 0 || num_pages_on_x_axis <= 0 ||
num_pages_on_y_axis <= 0) {
return nullptr;
}
ScopedFPDFDocument output_doc(FPDF_CreateNewDocument());
if (!output_doc)
return nullptr;
CPDF_Document* pDestDoc = CPDFDocumentFromFPDFDocument(output_doc.get());
ASSERT(pDestDoc);
std::vector<uint32_t> page_numbers = GetPageNumbers(*pSrcDoc, ByteString());
if (page_numbers.empty())
return nullptr;
if (num_pages_on_x_axis == 1 && num_pages_on_y_axis == 1) {
CPDF_PageExporter exporter(pDestDoc, pSrcDoc);
if (!exporter.ExportPage(page_numbers, 0))
return nullptr;
return output_doc.release();
}
CPDF_NPageToOneExporter exporter(pDestDoc, pSrcDoc);
if (!exporter.ExportNPagesToOne(page_numbers,
CFX_SizeF(output_width, output_height),
num_pages_on_x_axis, num_pages_on_y_axis)) {
return nullptr;
}
return output_doc.release();
}
FPDF_EXPORT FPDF_BOOL FPDF_CALLCONV
FPDF_CopyViewerPreferences(FPDF_DOCUMENT dest_doc, FPDF_DOCUMENT src_doc) {
CPDF_Document* pDstDoc = CPDFDocumentFromFPDFDocument(dest_doc);
if (!pDstDoc)
return false;
CPDF_Document* pSrcDoc = CPDFDocumentFromFPDFDocument(src_doc);
if (!pSrcDoc)
return false;
const CPDF_Dictionary* pSrcDict = pSrcDoc->GetRoot();
pSrcDict = pSrcDict->GetDictFor("ViewerPreferences");
if (!pSrcDict)
return false;
CPDF_Dictionary* pDstDict = pDstDoc->GetRoot();
if (!pDstDict)
return false;
pDstDict->SetFor("ViewerPreferences", pSrcDict->CloneDirectObject());
return true;
}