diff options
Diffstat (limited to 'core/fpdfapi/edit/cpdf_pagecontentgenerator.cpp')
| -rw-r--r-- | core/fpdfapi/edit/cpdf_pagecontentgenerator.cpp | 178 |
1 files changed, 178 insertions, 0 deletions
diff --git a/core/fpdfapi/edit/cpdf_pagecontentgenerator.cpp b/core/fpdfapi/edit/cpdf_pagecontentgenerator.cpp new file mode 100644 index 0000000000..e5d9683da6 --- /dev/null +++ b/core/fpdfapi/edit/cpdf_pagecontentgenerator.cpp @@ -0,0 +1,178 @@ +// 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/edit/cpdf_pagecontentgenerator.h" + +#include "core/fpdfapi/edit/cpdf_creator.h" +#include "core/fpdfapi/fpdf_page/cpdf_image.h" +#include "core/fpdfapi/fpdf_page/cpdf_imageobject.h" +#include "core/fpdfapi/fpdf_page/cpdf_page.h" +#include "core/fpdfapi/fpdf_page/cpdf_pageobject.h" +#include "core/fpdfapi/fpdf_page/pageint.h" +#include "core/fpdfapi/fpdf_parser/cpdf_array.h" +#include "core/fpdfapi/fpdf_parser/cpdf_dictionary.h" +#include "core/fpdfapi/fpdf_parser/cpdf_document.h" +#include "core/fpdfapi/fpdf_parser/cpdf_stream.h" +#include "core/fpdfapi/fpdf_parser/cpdf_stream_acc.h" +#include "core/fpdfapi/fpdf_parser/fpdf_parser_decode.h" + +CFX_ByteTextBuf& operator<<(CFX_ByteTextBuf& ar, CFX_Matrix& matrix) { + ar << matrix.a << " " << matrix.b << " " << matrix.c << " " << matrix.d << " " + << matrix.e << " " << matrix.f; + return ar; +} + +CPDF_PageContentGenerator::CPDF_PageContentGenerator(CPDF_Page* pPage) + : m_pPage(pPage), m_pDocument(m_pPage->m_pDocument) { + for (const auto& pObj : *pPage->GetPageObjectList()) + InsertPageObject(pObj.get()); +} + +CPDF_PageContentGenerator::~CPDF_PageContentGenerator() {} + +FX_BOOL CPDF_PageContentGenerator::InsertPageObject( + CPDF_PageObject* pPageObject) { + return pPageObject && m_pageObjects.Add(pPageObject); +} + +void CPDF_PageContentGenerator::GenerateContent() { + CFX_ByteTextBuf buf; + CPDF_Dictionary* pPageDict = m_pPage->m_pFormDict; + for (int i = 0; i < m_pageObjects.GetSize(); ++i) { + CPDF_PageObject* pPageObj = m_pageObjects[i]; + if (!pPageObj || !pPageObj->IsImage()) { + continue; + } + ProcessImage(buf, pPageObj->AsImage()); + } + CPDF_Object* pContent = + pPageDict ? pPageDict->GetDirectObjectFor("Contents") : nullptr; + if (pContent) + pPageDict->RemoveFor("Contents"); + + CPDF_Stream* pStream = new CPDF_Stream; + pStream->SetData(buf.GetBuffer(), buf.GetLength()); + pPageDict->SetReferenceFor("Contents", m_pDocument, + m_pDocument->AddIndirectObject(pStream)); +} + +CFX_ByteString CPDF_PageContentGenerator::RealizeResource( + CPDF_Object* pResourceObj, + const CFX_ByteString& bsType) { + if (!m_pPage->m_pResources) { + m_pPage->m_pResources = + new CPDF_Dictionary(m_pDocument->GetByteStringPool()); + m_pPage->m_pFormDict->SetReferenceFor( + "Resources", m_pDocument, + m_pDocument->AddIndirectObject(m_pPage->m_pResources)); + } + CPDF_Dictionary* pResList = m_pPage->m_pResources->GetDictFor(bsType); + if (!pResList) { + pResList = new CPDF_Dictionary(m_pDocument->GetByteStringPool()); + m_pPage->m_pResources->SetFor(bsType, pResList); + } + CFX_ByteString name; + int idnum = 1; + while (1) { + name.Format("FX%c%d", bsType[0], idnum); + if (!pResList->KeyExist(name)) { + break; + } + idnum++; + } + pResList->SetReferenceFor(name, m_pDocument, + m_pDocument->AddIndirectObject(pResourceObj)); + return name; +} + +void CPDF_PageContentGenerator::ProcessImage(CFX_ByteTextBuf& buf, + CPDF_ImageObject* pImageObj) { + if ((pImageObj->m_Matrix.a == 0 && pImageObj->m_Matrix.b == 0) || + (pImageObj->m_Matrix.c == 0 && pImageObj->m_Matrix.d == 0)) { + return; + } + buf << "q " << pImageObj->m_Matrix << " cm "; + CPDF_Image* pImage = pImageObj->GetImage(); + if (!pImage->IsInline()) { + CPDF_Stream* pStream = pImage->GetStream(); + uint32_t dwSavedObjNum = pStream->GetObjNum(); + CFX_ByteString name = RealizeResource(pStream, "XObject"); + if (dwSavedObjNum == 0) { + pImageObj->SetUnownedImage(m_pDocument->GetPageData()->GetImage(pStream)); + } + buf << "/" << PDF_NameEncode(name) << " Do Q\n"; + } +} + +void CPDF_PageContentGenerator::ProcessForm(CFX_ByteTextBuf& buf, + const uint8_t* data, + uint32_t size, + CFX_Matrix& matrix) { + if (!data || !size) + return; + + CPDF_Dictionary* pFormDict = + new CPDF_Dictionary(m_pDocument->GetByteStringPool()); + pFormDict->SetNameFor("Type", "XObject"); + pFormDict->SetNameFor("Subtype", "Form"); + + CFX_FloatRect bbox = m_pPage->GetPageBBox(); + matrix.TransformRect(bbox); + pFormDict->SetRectFor("BBox", bbox); + + CPDF_Stream* pStream = new CPDF_Stream; + pStream->InitStream(data, size, pFormDict); + buf << "q " << matrix << " cm "; + + CFX_ByteString name = RealizeResource(pStream, "XObject"); + buf << "/" << PDF_NameEncode(name) << " Do Q\n"; +} + +void CPDF_PageContentGenerator::TransformContent(CFX_Matrix& matrix) { + CPDF_Dictionary* pDict = m_pPage->m_pFormDict; + CPDF_Object* pContent = + pDict ? pDict->GetDirectObjectFor("Contents") : nullptr; + if (!pContent) + return; + + CFX_ByteTextBuf buf; + if (CPDF_Array* pArray = pContent->AsArray()) { + size_t iCount = pArray->GetCount(); + CPDF_StreamAcc** pContentArray = FX_Alloc(CPDF_StreamAcc*, iCount); + size_t size = 0; + for (size_t i = 0; i < iCount; ++i) { + pContent = pArray->GetObjectAt(i); + CPDF_Stream* pStream = ToStream(pContent); + if (!pStream) + continue; + + CPDF_StreamAcc* pStreamAcc = new CPDF_StreamAcc(); + pStreamAcc->LoadAllData(pStream); + pContentArray[i] = pStreamAcc; + size += pContentArray[i]->GetSize() + 1; + } + int pos = 0; + uint8_t* pBuf = FX_Alloc(uint8_t, size); + for (size_t i = 0; i < iCount; ++i) { + FXSYS_memcpy(pBuf + pos, pContentArray[i]->GetData(), + pContentArray[i]->GetSize()); + pos += pContentArray[i]->GetSize() + 1; + pBuf[pos - 1] = ' '; + delete pContentArray[i]; + } + ProcessForm(buf, pBuf, size, matrix); + FX_Free(pBuf); + FX_Free(pContentArray); + } else if (CPDF_Stream* pStream = pContent->AsStream()) { + CPDF_StreamAcc contentStream; + contentStream.LoadAllData(pStream); + ProcessForm(buf, contentStream.GetData(), contentStream.GetSize(), matrix); + } + CPDF_Stream* pStream = new CPDF_Stream; + pStream->SetData(buf.GetBuffer(), buf.GetLength()); + m_pPage->m_pFormDict->SetReferenceFor( + "Contents", m_pDocument, m_pDocument->AddIndirectObject(pStream)); +} |