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