| // 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/page/cpdf_docpagedata.h" |
| #include "core/fpdfapi/page/cpdf_image.h" |
| #include "core/fpdfapi/page/cpdf_imageobject.h" |
| #include "core/fpdfapi/page/cpdf_page.h" |
| #include "core/fpdfapi/page/cpdf_path.h" |
| #include "core/fpdfapi/page/cpdf_pathobject.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_reference.h" |
| #include "core/fpdfapi/parser/cpdf_stream.h" |
| #include "core/fpdfapi/parser/fpdf_parser_decode.h" |
| |
| namespace { |
| |
| CFX_ByteTextBuf& operator<<(CFX_ByteTextBuf& ar, const CFX_Matrix& matrix) { |
| ar << matrix.a << " " << matrix.b << " " << matrix.c << " " << matrix.d << " " |
| << matrix.e << " " << matrix.f; |
| return ar; |
| } |
| |
| } // namespace |
| |
| CPDF_PageContentGenerator::CPDF_PageContentGenerator(CPDF_Page* pPage) |
| : m_pPage(pPage), m_pDocument(m_pPage->m_pDocument) { |
| for (const auto& pObj : *pPage->GetPageObjectList()) { |
| if (pObj) |
| m_pageObjects.push_back(pObj.get()); |
| } |
| } |
| |
| CPDF_PageContentGenerator::~CPDF_PageContentGenerator() {} |
| |
| void CPDF_PageContentGenerator::GenerateContent() { |
| CFX_ByteTextBuf buf; |
| for (CPDF_PageObject* pPageObj : m_pageObjects) { |
| if (CPDF_ImageObject* pImageObject = pPageObj->AsImage()) |
| ProcessImage(&buf, pImageObject); |
| else if (CPDF_PathObject* pPathObj = pPageObj->AsPath()) |
| ProcessPath(&buf, pPathObj); |
| } |
| CPDF_Dictionary* pPageDict = m_pPage->m_pFormDict; |
| CPDF_Object* pContent = |
| pPageDict ? pPageDict->GetDirectObjectFor("Contents") : nullptr; |
| if (pContent) |
| pPageDict->RemoveFor("Contents"); |
| |
| CPDF_Stream* pStream = m_pDocument->NewIndirect<CPDF_Stream>(); |
| pStream->SetData(buf.GetBuffer(), buf.GetLength()); |
| pPageDict->SetNewFor<CPDF_Reference>("Contents", m_pDocument, |
| pStream->GetObjNum()); |
| } |
| |
| CFX_ByteString CPDF_PageContentGenerator::RealizeResource( |
| uint32_t dwResourceObjNum, |
| const CFX_ByteString& bsType) { |
| ASSERT(dwResourceObjNum); |
| if (!m_pPage->m_pResources) { |
| m_pPage->m_pResources = m_pDocument->NewIndirect<CPDF_Dictionary>(); |
| m_pPage->m_pFormDict->SetNewFor<CPDF_Reference>( |
| "Resources", m_pDocument, m_pPage->m_pResources->GetObjNum()); |
| } |
| CPDF_Dictionary* pResList = m_pPage->m_pResources->GetDictFor(bsType); |
| if (!pResList) |
| pResList = m_pPage->m_pResources->SetNewFor<CPDF_Dictionary>(bsType); |
| |
| CFX_ByteString name; |
| int idnum = 1; |
| while (1) { |
| name.Format("FX%c%d", bsType[0], idnum); |
| if (!pResList->KeyExist(name)) { |
| break; |
| } |
| idnum++; |
| } |
| pResList->SetNewFor<CPDF_Reference>(name, m_pDocument, dwResourceObjNum); |
| return name; |
| } |
| |
| void CPDF_PageContentGenerator::ProcessImage(CFX_ByteTextBuf* buf, |
| CPDF_ImageObject* pImageObj) { |
| if ((pImageObj->matrix().a == 0 && pImageObj->matrix().b == 0) || |
| (pImageObj->matrix().c == 0 && pImageObj->matrix().d == 0)) { |
| return; |
| } |
| *buf << "q " << pImageObj->matrix() << " cm "; |
| |
| CPDF_Image* pImage = pImageObj->GetImage(); |
| if (pImage->IsInline()) |
| return; |
| |
| CPDF_Stream* pStream = pImage->GetStream(); |
| if (!pStream) |
| return; |
| |
| bool bWasInline = pStream->IsInline(); |
| if (bWasInline) |
| pImage->ConvertStreamToIndirectObject(); |
| |
| uint32_t dwObjNum = pStream->GetObjNum(); |
| CFX_ByteString name = RealizeResource(dwObjNum, "XObject"); |
| if (bWasInline) |
| pImageObj->SetUnownedImage(m_pDocument->GetPageData()->GetImage(dwObjNum)); |
| |
| *buf << "/" << PDF_NameEncode(name) << " Do Q\n"; |
| } |
| |
| // Processing path with operators from Tables 4.9 and 4.10 of PDF spec 1.7: |
| // "re" appends a rectangle (here, used only if the whole path is a rectangle) |
| // "m" moves current point to the given coordinates |
| // "l" creates a line from current point to the new point |
| // "c" adds a Bezier curve from current to last point, using the two other |
| // points as the Bezier control points |
| // Note: "l", "c" change the current point |
| // "h" closes the subpath (appends a line from current to starting point) |
| // Path painting operators: "S", "n", "B", "f", "B*", "f*", depending on |
| // the filling mode and whether we want stroking the path or not. |
| void CPDF_PageContentGenerator::ProcessPath(CFX_ByteTextBuf* buf, |
| CPDF_PathObject* pPathObj) { |
| const FX_PATHPOINT* pPoints = pPathObj->m_Path.GetPoints(); |
| if (pPathObj->m_Path.IsRect()) { |
| *buf << pPoints[0].m_PointX << " " << pPoints[0].m_PointY << " " |
| << (pPoints[2].m_PointX - pPoints[0].m_PointX) << " " |
| << (pPoints[2].m_PointY - pPoints[0].m_PointY) << " re"; |
| } else { |
| int numPoints = pPathObj->m_Path.GetPointCount(); |
| for (int i = 0; i < numPoints; i++) { |
| if (i > 0) |
| *buf << " "; |
| *buf << pPoints[i].m_PointX << " " << pPoints[i].m_PointY; |
| int pointFlag = pPoints[i].m_Flag; |
| if (pointFlag == FXPT_MOVETO) { |
| *buf << " m"; |
| } else if (pointFlag & FXPT_LINETO) { |
| *buf << " l"; |
| } else if (pointFlag & FXPT_BEZIERTO) { |
| if (i + 2 >= numPoints || pPoints[i].m_Flag != FXPT_BEZIERTO || |
| pPoints[i + 1].m_Flag != FXPT_BEZIERTO || |
| (pPoints[i + 2].m_Flag & FXPT_BEZIERTO) == 0) { |
| // If format is not supported, close the path and paint |
| *buf << " h"; |
| break; |
| } |
| *buf << " " << pPoints[i + 1].m_PointX << " " << pPoints[i + 1].m_PointY |
| << " " << pPoints[i + 2].m_PointX << " " << pPoints[i + 2].m_PointY |
| << " c"; |
| if (pPoints[i + 2].m_Flag & FXPT_CLOSEFIGURE) |
| *buf << " h"; |
| i += 2; |
| } |
| if (pointFlag & FXPT_CLOSEFIGURE) |
| *buf << " h"; |
| } |
| } |
| if (pPathObj->m_FillType == 0) |
| *buf << (pPathObj->m_bStroke ? " S" : " n"); |
| else if (pPathObj->m_FillType == FXFILL_WINDING) |
| *buf << (pPathObj->m_bStroke ? " B" : " f"); |
| else if (pPathObj->m_FillType == FXFILL_ALTERNATE) |
| *buf << (pPathObj->m_bStroke ? " B*" : " f*"); |
| *buf << "\n"; |
| } |