| OLD | NEW |
| 1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| 6 | 6 |
| 7 #include "core/include/fpdfapi/fpdf_module.h" | 7 #include "core/include/fpdfapi/fpdf_module.h" |
| 8 #include "core/include/fpdfapi/fpdf_page.h" | 8 #include "core/include/fpdfapi/fpdf_page.h" |
| 9 #include "core/include/fpdfapi/fpdf_serial.h" | 9 #include "core/include/fpdfapi/fpdf_serial.h" |
| 10 #include "core/src/fpdfapi/fpdf_page/pageint.h" | 10 #include "core/src/fpdfapi/fpdf_page/pageint.h" |
| 11 | 11 |
| 12 CFX_ByteTextBuf& operator<<(CFX_ByteTextBuf& ar, CFX_Matrix& matrix) { | 12 CFX_ByteTextBuf& operator<<(CFX_ByteTextBuf& ar, CFX_Matrix& matrix) { |
| 13 ar << matrix.a << " " << matrix.b << " " << matrix.c << " " << matrix.d << " " | 13 ar << matrix.a << " " << matrix.b << " " << matrix.c << " " << matrix.d << " " |
| 14 << matrix.e << " " << matrix.f; | 14 << matrix.e << " " << matrix.f; |
| 15 return ar; | 15 return ar; |
| 16 } | 16 } |
| 17 CPDF_PageContentGenerate::CPDF_PageContentGenerate(CPDF_Page* pPage) | 17 CPDF_PageContentGenerator::CPDF_PageContentGenerator(CPDF_Page* pPage) |
| 18 : m_pPage(pPage) { | 18 : m_pPage(pPage) { |
| 19 m_pDocument = NULL; | 19 m_pDocument = NULL; |
| 20 if (m_pPage) { | 20 if (m_pPage) { |
| 21 m_pDocument = m_pPage->m_pDocument; | 21 m_pDocument = m_pPage->m_pDocument; |
| 22 } | 22 } |
| 23 FX_POSITION pos = pPage->GetFirstObjectPosition(); | 23 FX_POSITION pos = pPage->GetFirstObjectPosition(); |
| 24 while (pos) { | 24 while (pos) { |
| 25 InsertPageObject(pPage->GetNextObject(pos)); | 25 InsertPageObject(pPage->GetNextObject(pos)); |
| 26 } | 26 } |
| 27 } | 27 } |
| 28 CPDF_PageContentGenerate::~CPDF_PageContentGenerate() {} | 28 CPDF_PageContentGenerator::~CPDF_PageContentGenerator() {} |
| 29 FX_BOOL CPDF_PageContentGenerate::InsertPageObject( | 29 FX_BOOL CPDF_PageContentGenerator::InsertPageObject( |
| 30 CPDF_PageObject* pPageObject) { | 30 CPDF_PageObject* pPageObject) { |
| 31 if (!pPageObject) { | 31 if (!pPageObject) { |
| 32 return FALSE; | 32 return FALSE; |
| 33 } | 33 } |
| 34 return m_pageObjects.Add(pPageObject); | 34 return m_pageObjects.Add(pPageObject); |
| 35 } | 35 } |
| 36 void CPDF_PageContentGenerate::GenerateContent() { | 36 void CPDF_PageContentGenerator::GenerateContent() { |
| 37 CFX_ByteTextBuf buf; | 37 CFX_ByteTextBuf buf; |
| 38 CPDF_Dictionary* pPageDict = m_pPage->m_pFormDict; | 38 CPDF_Dictionary* pPageDict = m_pPage->m_pFormDict; |
| 39 for (int i = 0; i < m_pageObjects.GetSize(); ++i) { | 39 for (int i = 0; i < m_pageObjects.GetSize(); ++i) { |
| 40 CPDF_PageObject* pPageObj = m_pageObjects[i]; | 40 CPDF_PageObject* pPageObj = m_pageObjects[i]; |
| 41 if (!pPageObj || pPageObj->m_Type != PDFPAGE_IMAGE) { | 41 if (!pPageObj || pPageObj->m_Type != PDFPAGE_IMAGE) { |
| 42 continue; | 42 continue; |
| 43 } | 43 } |
| 44 ProcessImage(buf, (CPDF_ImageObject*)pPageObj); | 44 ProcessImage(buf, (CPDF_ImageObject*)pPageObj); |
| 45 } | 45 } |
| 46 CPDF_Object* pContent = | 46 CPDF_Object* pContent = |
| 47 pPageDict ? pPageDict->GetElementValue("Contents") : NULL; | 47 pPageDict ? pPageDict->GetElementValue("Contents") : NULL; |
| 48 if (pContent) { | 48 if (pContent) { |
| 49 pPageDict->RemoveAt("Contents"); | 49 pPageDict->RemoveAt("Contents"); |
| 50 } | 50 } |
| 51 CPDF_Stream* pStream = new CPDF_Stream(NULL, 0, NULL); | 51 CPDF_Stream* pStream = new CPDF_Stream(NULL, 0, NULL); |
| 52 pStream->SetData(buf.GetBuffer(), buf.GetLength(), FALSE, FALSE); | 52 pStream->SetData(buf.GetBuffer(), buf.GetLength(), FALSE, FALSE); |
| 53 m_pDocument->AddIndirectObject(pStream); | 53 m_pDocument->AddIndirectObject(pStream); |
| 54 pPageDict->SetAtReference("Contents", m_pDocument, pStream->GetObjNum()); | 54 pPageDict->SetAtReference("Contents", m_pDocument, pStream->GetObjNum()); |
| 55 } | 55 } |
| 56 CFX_ByteString CPDF_PageContentGenerate::RealizeResource( | 56 CFX_ByteString CPDF_PageContentGenerator::RealizeResource( |
| 57 CPDF_Object* pResourceObj, | 57 CPDF_Object* pResourceObj, |
| 58 const FX_CHAR* szType) { | 58 const FX_CHAR* szType) { |
| 59 if (!m_pPage->m_pResources) { | 59 if (!m_pPage->m_pResources) { |
| 60 m_pPage->m_pResources = new CPDF_Dictionary; | 60 m_pPage->m_pResources = new CPDF_Dictionary; |
| 61 int objnum = m_pDocument->AddIndirectObject(m_pPage->m_pResources); | 61 int objnum = m_pDocument->AddIndirectObject(m_pPage->m_pResources); |
| 62 m_pPage->m_pFormDict->SetAtReference("Resources", m_pDocument, objnum); | 62 m_pPage->m_pFormDict->SetAtReference("Resources", m_pDocument, objnum); |
| 63 } | 63 } |
| 64 CPDF_Dictionary* pResList = m_pPage->m_pResources->GetDict(szType); | 64 CPDF_Dictionary* pResList = m_pPage->m_pResources->GetDict(szType); |
| 65 if (!pResList) { | 65 if (!pResList) { |
| 66 pResList = new CPDF_Dictionary; | 66 pResList = new CPDF_Dictionary; |
| 67 m_pPage->m_pResources->SetAt(szType, pResList); | 67 m_pPage->m_pResources->SetAt(szType, pResList); |
| 68 } | 68 } |
| 69 m_pDocument->AddIndirectObject(pResourceObj); | 69 m_pDocument->AddIndirectObject(pResourceObj); |
| 70 CFX_ByteString name; | 70 CFX_ByteString name; |
| 71 int idnum = 1; | 71 int idnum = 1; |
| 72 while (1) { | 72 while (1) { |
| 73 name.Format("FX%c%d", szType[0], idnum); | 73 name.Format("FX%c%d", szType[0], idnum); |
| 74 if (!pResList->KeyExist(name)) { | 74 if (!pResList->KeyExist(name)) { |
| 75 break; | 75 break; |
| 76 } | 76 } |
| 77 idnum++; | 77 idnum++; |
| 78 } | 78 } |
| 79 pResList->AddReference(name, m_pDocument, pResourceObj->GetObjNum()); | 79 pResList->AddReference(name, m_pDocument, pResourceObj->GetObjNum()); |
| 80 return name; | 80 return name; |
| 81 } | 81 } |
| 82 void CPDF_PageContentGenerate::ProcessImage(CFX_ByteTextBuf& buf, | 82 void CPDF_PageContentGenerator::ProcessImage(CFX_ByteTextBuf& buf, |
| 83 CPDF_ImageObject* pImageObj) { | 83 CPDF_ImageObject* pImageObj) { |
| 84 if ((pImageObj->m_Matrix.a == 0 && pImageObj->m_Matrix.b == 0) || | 84 if ((pImageObj->m_Matrix.a == 0 && pImageObj->m_Matrix.b == 0) || |
| 85 (pImageObj->m_Matrix.c == 0 && pImageObj->m_Matrix.d == 0)) { | 85 (pImageObj->m_Matrix.c == 0 && pImageObj->m_Matrix.d == 0)) { |
| 86 return; | 86 return; |
| 87 } | 87 } |
| 88 buf << "q " << pImageObj->m_Matrix << " cm "; | 88 buf << "q " << pImageObj->m_Matrix << " cm "; |
| 89 if (!pImageObj->m_pImage->IsInline()) { | 89 if (!pImageObj->m_pImage->IsInline()) { |
| 90 CPDF_Stream* pStream = pImageObj->m_pImage->GetStream(); | 90 CPDF_Stream* pStream = pImageObj->m_pImage->GetStream(); |
| 91 FX_DWORD dwSavedObjNum = pStream->GetObjNum(); | 91 FX_DWORD dwSavedObjNum = pStream->GetObjNum(); |
| 92 CFX_ByteString name = RealizeResource(pStream, "XObject"); | 92 CFX_ByteString name = RealizeResource(pStream, "XObject"); |
| 93 if (dwSavedObjNum == 0) { | 93 if (dwSavedObjNum == 0) { |
| 94 if (pImageObj->m_pImage) | 94 if (pImageObj->m_pImage) |
| 95 pImageObj->m_pImage->Release(); | 95 pImageObj->m_pImage->Release(); |
| 96 pImageObj->m_pImage = m_pDocument->GetPageData()->GetImage(pStream); | 96 pImageObj->m_pImage = m_pDocument->GetPageData()->GetImage(pStream); |
| 97 } | 97 } |
| 98 buf << "/" << PDF_NameEncode(name) << " Do Q\n"; | 98 buf << "/" << PDF_NameEncode(name) << " Do Q\n"; |
| 99 } | 99 } |
| 100 } | 100 } |
| 101 void CPDF_PageContentGenerate::ProcessForm(CFX_ByteTextBuf& buf, | 101 void CPDF_PageContentGenerator::ProcessForm(CFX_ByteTextBuf& buf, |
| 102 const uint8_t* data, | 102 const uint8_t* data, |
| 103 FX_DWORD size, | 103 FX_DWORD size, |
| 104 CFX_Matrix& matrix) { | 104 CFX_Matrix& matrix) { |
| 105 if (!data || !size) { | 105 if (!data || !size) { |
| 106 return; | 106 return; |
| 107 } | 107 } |
| 108 CPDF_Stream* pStream = new CPDF_Stream(NULL, 0, NULL); | 108 CPDF_Stream* pStream = new CPDF_Stream(NULL, 0, NULL); |
| 109 CPDF_Dictionary* pFormDict = new CPDF_Dictionary; | 109 CPDF_Dictionary* pFormDict = new CPDF_Dictionary; |
| 110 pFormDict->SetAtName("Type", "XObject"); | 110 pFormDict->SetAtName("Type", "XObject"); |
| 111 pFormDict->SetAtName("Subtype", "Form"); | 111 pFormDict->SetAtName("Subtype", "Form"); |
| 112 CFX_FloatRect bbox = m_pPage->GetPageBBox(); | 112 CFX_FloatRect bbox = m_pPage->GetPageBBox(); |
| 113 matrix.TransformRect(bbox); | 113 matrix.TransformRect(bbox); |
| 114 pFormDict->SetAtRect("BBox", bbox); | 114 pFormDict->SetAtRect("BBox", bbox); |
| 115 pStream->InitStream((uint8_t*)data, size, pFormDict); | 115 pStream->InitStream((uint8_t*)data, size, pFormDict); |
| 116 buf << "q " << matrix << " cm "; | 116 buf << "q " << matrix << " cm "; |
| 117 CFX_ByteString name = RealizeResource(pStream, "XObject"); | 117 CFX_ByteString name = RealizeResource(pStream, "XObject"); |
| 118 buf << "/" << PDF_NameEncode(name) << " Do Q\n"; | 118 buf << "/" << PDF_NameEncode(name) << " Do Q\n"; |
| 119 } | 119 } |
| 120 void CPDF_PageContentGenerate::TransformContent(CFX_Matrix& matrix) { | 120 void CPDF_PageContentGenerator::TransformContent(CFX_Matrix& matrix) { |
| 121 CPDF_Dictionary* pDict = m_pPage->m_pFormDict; | 121 CPDF_Dictionary* pDict = m_pPage->m_pFormDict; |
| 122 CPDF_Object* pContent = pDict ? pDict->GetElementValue("Contents") : NULL; | 122 CPDF_Object* pContent = pDict ? pDict->GetElementValue("Contents") : NULL; |
| 123 if (!pContent) | 123 if (!pContent) |
| 124 return; | 124 return; |
| 125 | 125 |
| 126 CFX_ByteTextBuf buf; | 126 CFX_ByteTextBuf buf; |
| 127 if (CPDF_Array* pArray = pContent->AsArray()) { | 127 if (CPDF_Array* pArray = pContent->AsArray()) { |
| 128 int iCount = pArray->GetCount(); | 128 int iCount = pArray->GetCount(); |
| 129 CPDF_StreamAcc** pContentArray = FX_Alloc(CPDF_StreamAcc*, iCount); | 129 CPDF_StreamAcc** pContentArray = FX_Alloc(CPDF_StreamAcc*, iCount); |
| 130 int size = 0; | 130 int size = 0; |
| (...skipping 25 matching lines...) Expand all Loading... |
| 156 CPDF_StreamAcc contentStream; | 156 CPDF_StreamAcc contentStream; |
| 157 contentStream.LoadAllData(pStream); | 157 contentStream.LoadAllData(pStream); |
| 158 ProcessForm(buf, contentStream.GetData(), contentStream.GetSize(), matrix); | 158 ProcessForm(buf, contentStream.GetData(), contentStream.GetSize(), matrix); |
| 159 } | 159 } |
| 160 CPDF_Stream* pStream = new CPDF_Stream(NULL, 0, NULL); | 160 CPDF_Stream* pStream = new CPDF_Stream(NULL, 0, NULL); |
| 161 pStream->SetData(buf.GetBuffer(), buf.GetLength(), FALSE, FALSE); | 161 pStream->SetData(buf.GetBuffer(), buf.GetLength(), FALSE, FALSE); |
| 162 m_pDocument->AddIndirectObject(pStream); | 162 m_pDocument->AddIndirectObject(pStream); |
| 163 m_pPage->m_pFormDict->SetAtReference("Contents", m_pDocument, | 163 m_pPage->m_pFormDict->SetAtReference("Contents", m_pDocument, |
| 164 pStream->GetObjNum()); | 164 pStream->GetObjNum()); |
| 165 } | 165 } |
| OLD | NEW |