| // 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_edit/include/cpdf_pagecontentgenerator.h" |
| |
| #include "core/fpdfapi/fpdf_edit/include/cpdf_creator.h" |
| #include "core/fpdfapi/fpdf_page/include/cpdf_image.h" |
| #include "core/fpdfapi/fpdf_page/include/cpdf_imageobject.h" |
| #include "core/fpdfapi/fpdf_page/include/cpdf_page.h" |
| #include "core/fpdfapi/fpdf_page/include/cpdf_pageobject.h" |
| #include "core/fpdfapi/fpdf_page/pageint.h" |
| #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" |
| #include "core/fpdfapi/fpdf_parser/include/cpdf_dictionary.h" |
| #include "core/fpdfapi/fpdf_parser/include/cpdf_document.h" |
| #include "core/fpdfapi/fpdf_parser/include/cpdf_stream.h" |
| #include "core/fpdfapi/fpdf_parser/include/cpdf_stream_acc.h" |
| #include "core/fpdfapi/fpdf_parser/include/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->GetDirectObjectBy("Contents") : nullptr; |
| if (pContent) { |
| pPageDict->RemoveAt("Contents"); |
| } |
| CPDF_Stream* pStream = new CPDF_Stream(nullptr, 0, nullptr); |
| pStream->SetData(buf.GetBuffer(), buf.GetLength(), FALSE, FALSE); |
| m_pDocument->AddIndirectObject(pStream); |
| pPageDict->SetAtReference("Contents", m_pDocument, pStream->GetObjNum()); |
| } |
| CFX_ByteString CPDF_PageContentGenerator::RealizeResource( |
| CPDF_Object* pResourceObj, |
| const CFX_ByteString& bsType) { |
| if (!m_pPage->m_pResources) { |
| m_pPage->m_pResources = new CPDF_Dictionary; |
| int objnum = m_pDocument->AddIndirectObject(m_pPage->m_pResources); |
| m_pPage->m_pFormDict->SetAtReference("Resources", m_pDocument, objnum); |
| } |
| CPDF_Dictionary* pResList = m_pPage->m_pResources->GetDictBy(bsType); |
| if (!pResList) { |
| pResList = new CPDF_Dictionary; |
| m_pPage->m_pResources->SetAt(bsType, pResList); |
| } |
| m_pDocument->AddIndirectObject(pResourceObj); |
| CFX_ByteString name; |
| int idnum = 1; |
| while (1) { |
| name.Format("FX%c%d", bsType[0], idnum); |
| if (!pResList->KeyExist(name)) { |
| break; |
| } |
| idnum++; |
| } |
| pResList->SetAtReference(name, m_pDocument, pResourceObj->GetObjNum()); |
| 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 "; |
| if (!pImageObj->m_pImage->IsInline()) { |
| CPDF_Stream* pStream = pImageObj->m_pImage->GetStream(); |
| uint32_t dwSavedObjNum = pStream->GetObjNum(); |
| CFX_ByteString name = RealizeResource(pStream, "XObject"); |
| if (dwSavedObjNum == 0) { |
| if (pImageObj->m_pImage) |
| pImageObj->m_pImage->Release(); |
| pImageObj->m_pImage = 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_Stream* pStream = new CPDF_Stream(nullptr, 0, nullptr); |
| CPDF_Dictionary* pFormDict = new CPDF_Dictionary; |
| pFormDict->SetAtName("Type", "XObject"); |
| pFormDict->SetAtName("Subtype", "Form"); |
| CFX_FloatRect bbox = m_pPage->GetPageBBox(); |
| matrix.TransformRect(bbox); |
| pFormDict->SetAtRect("BBox", bbox); |
| 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->GetDirectObjectBy("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(nullptr, 0, nullptr); |
| pStream->SetData(buf.GetBuffer(), buf.GetLength(), FALSE, FALSE); |
| m_pDocument->AddIndirectObject(pStream); |
| m_pPage->m_pFormDict->SetAtReference("Contents", m_pDocument, |
| pStream->GetObjNum()); |
| } |