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