Revert "FX Bool considered harmful, part 3"
[pdfium.git] / core / src / fpdfapi / fpdf_edit / fpdf_edit_image.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_module.h"
8 #include "../../../include/fpdfapi/fpdf_page.h"
9 #include "../../../include/fxcodec/fx_codec.h"
10 #include "../../../include/fpdfapi/fpdf_render.h"
11 #include "../fpdf_page/pageint.h"
12 #include "../fpdf_render/render_int.h"
13 CPDF_Dictionary* CPDF_Image::InitJPEG(uint8_t* pData, FX_DWORD size)
14 {
15     int32_t width;
16     int32_t height;
17     int32_t num_comps;
18     int32_t bits;
19     FX_BOOL color_trans;
20     if (!CPDF_ModuleMgr::Get()->GetJpegModule()->
21             LoadInfo(pData, size, width, height, num_comps, bits, color_trans)) {
22         return NULL;
23     }
24     CPDF_Dictionary* pDict = new CPDF_Dictionary;
25     pDict->SetAtName("Type", "XObject");
26     pDict->SetAtName("Subtype", "Image");
27     pDict->SetAtInteger("Width", width);
28     pDict->SetAtInteger("Height", height);
29     const FX_CHAR* csname = NULL;
30     if (num_comps == 1) {
31         csname = "DeviceGray";
32     } else if (num_comps == 3) {
33         csname = "DeviceRGB";
34     } else if (num_comps == 4) {
35         csname = "DeviceCMYK";
36         CPDF_Array* pDecode = CPDF_Array::Create();
37         for (int n = 0; n < 4; n ++) {
38             pDecode->AddInteger(1);
39             pDecode->AddInteger(0);
40         }
41         pDict->SetAt(FX_BSTRC("Decode"), pDecode);
42     }
43     pDict->SetAtName("ColorSpace", csname);
44     pDict->SetAtInteger("BitsPerComponent", bits);
45     pDict->SetAtName("Filter", "DCTDecode");
46     if (!color_trans) {
47         CPDF_Dictionary* pParms = new CPDF_Dictionary;
48         pDict->SetAt("DecodeParms", pParms);
49         pParms->SetAtInteger("ColorTransform", 0);
50     }
51     m_bIsMask = FALSE;
52     m_Width = width;
53     m_Height = height;
54     if (m_pStream == NULL) {
55         m_pStream = new CPDF_Stream(NULL, 0, NULL);
56     }
57     return pDict;
58 }
59 void CPDF_Image::SetJpegImage(uint8_t* pData, FX_DWORD size)
60 {
61     CPDF_Dictionary *pDict = InitJPEG(pData, size);
62     if (!pDict) {
63         return;
64     }
65     m_pStream->InitStream(pData, size, pDict);
66 }
67 void CPDF_Image::SetJpegImage(IFX_FileRead *pFile)
68 {
69     FX_DWORD size = (FX_DWORD)pFile->GetSize();
70     if (!size) {
71         return;
72     }
73     FX_DWORD dwEstimateSize = size;
74     if (dwEstimateSize > 8192) {
75         dwEstimateSize = 8192;
76     }
77     uint8_t* pData = FX_Alloc(uint8_t, dwEstimateSize);
78     pFile->ReadBlock(pData, 0, dwEstimateSize);
79     CPDF_Dictionary *pDict = InitJPEG(pData, dwEstimateSize);
80     FX_Free(pData);
81     if (!pDict && size > dwEstimateSize) {
82         pData = FX_Alloc(uint8_t, size);
83         pFile->ReadBlock(pData, 0, size);
84         pDict = InitJPEG(pData, size);
85         FX_Free(pData);
86     }
87     if (!pDict) {
88         return;
89     }
90     m_pStream->InitStream(pFile, pDict);
91 }
92 void _DCTEncodeBitmap(CPDF_Dictionary *pBitmapDict, const CFX_DIBitmap* pBitmap, int quality, uint8_t* &buf, FX_STRSIZE &size)
93 {
94 }
95 void _JBIG2EncodeBitmap(CPDF_Dictionary *pBitmapDict, const CFX_DIBitmap *pBitmap, CPDF_Document *pDoc, uint8_t* &buf, FX_STRSIZE &size, FX_BOOL bLossLess)
96 {
97 }
98 void CPDF_Image::SetImage(const CFX_DIBitmap* pBitmap, int32_t iCompress, IFX_FileWrite *pFileWrite, IFX_FileRead *pFileRead, const CFX_DIBitmap* pMask, const CPDF_ImageSetParam* pParam)
99 {
100     int32_t BitmapWidth = pBitmap->GetWidth();
101     int32_t BitmapHeight = pBitmap->GetHeight();
102     if (BitmapWidth < 1 || BitmapHeight < 1) {
103         return;
104     }
105     uint8_t* src_buf = pBitmap->GetBuffer();
106     int32_t src_pitch = pBitmap->GetPitch();
107     int32_t bpp = pBitmap->GetBPP();
108     FX_BOOL bUseMatte = pParam && pParam->pMatteColor && (pBitmap->GetFormat() == FXDIB_Argb);
109     CPDF_Dictionary* pDict = new CPDF_Dictionary;
110     pDict->SetAtName(FX_BSTRC("Type"), FX_BSTRC("XObject"));
111     pDict->SetAtName(FX_BSTRC("Subtype"), FX_BSTRC("Image"));
112     pDict->SetAtInteger(FX_BSTRC("Width"), BitmapWidth);
113     pDict->SetAtInteger(FX_BSTRC("Height"), BitmapHeight);
114     uint8_t* dest_buf = NULL;
115     FX_STRSIZE dest_pitch = 0, dest_size = 0, opType = -1;
116     if (bpp == 1) {
117         int32_t reset_a = 0, reset_r = 0, reset_g = 0, reset_b = 0;
118         int32_t set_a = 0, set_r = 0, set_g = 0, set_b = 0;
119         if (!pBitmap->IsAlphaMask()) {
120             ArgbDecode(pBitmap->GetPaletteArgb(0), reset_a, reset_r, reset_g, reset_b);
121             ArgbDecode(pBitmap->GetPaletteArgb(1), set_a, set_r, set_g, set_b);
122         }
123         if (set_a == 0 || reset_a == 0) {
124             pDict->SetAt(FX_BSTRC("ImageMask"), new CPDF_Boolean(TRUE));
125             if (reset_a == 0) {
126                 CPDF_Array* pArray = new CPDF_Array;
127                 pArray->AddInteger(1);
128                 pArray->AddInteger(0);
129                 pDict->SetAt(FX_BSTRC("Decode"), pArray);
130             }
131         } else {
132             CPDF_Array* pCS = new CPDF_Array;
133             pCS->AddName(FX_BSTRC("Indexed"));
134             pCS->AddName(FX_BSTRC("DeviceRGB"));
135             pCS->AddInteger(1);
136             CFX_ByteString ct;
137             FX_CHAR* pBuf = ct.GetBuffer(6);
138             pBuf[0] = (FX_CHAR)reset_r;
139             pBuf[1] = (FX_CHAR)reset_g;
140             pBuf[2] = (FX_CHAR)reset_b;
141             pBuf[3] = (FX_CHAR)set_r;
142             pBuf[4] = (FX_CHAR)set_g;
143             pBuf[5] = (FX_CHAR)set_b;
144             ct.ReleaseBuffer(6);
145             pCS->Add(CPDF_String::Create(ct, TRUE));
146             pDict->SetAt(FX_BSTRC("ColorSpace"), pCS);
147         }
148         pDict->SetAtInteger(FX_BSTRC("BitsPerComponent"), 1);
149         dest_pitch = (BitmapWidth + 7) / 8;
150         if ((iCompress & 0x03) == PDF_IMAGE_NO_COMPRESS) {
151             opType = 1;
152         } else {
153             opType = 0;
154         }
155     } else if (bpp == 8) {
156         int32_t iPalette = pBitmap->GetPaletteSize();
157         if (iPalette > 0) {
158             CPDF_Array* pCS = new CPDF_Array;
159             m_pDocument->AddIndirectObject(pCS);
160             pCS->AddName(FX_BSTRC("Indexed"));
161             pCS->AddName(FX_BSTRC("DeviceRGB"));
162             pCS->AddInteger(iPalette - 1);
163             uint8_t* pColorTable = FX_Alloc2D(uint8_t, iPalette, 3);
164             uint8_t* ptr = pColorTable;
165             for (int32_t i = 0; i < iPalette; i ++) {
166                 FX_DWORD argb = pBitmap->GetPaletteArgb(i);
167                 ptr[0] = (uint8_t)(argb >> 16);
168                 ptr[1] = (uint8_t)(argb >> 8);
169                 ptr[2] = (uint8_t)argb;
170                 ptr += 3;
171             }
172             CPDF_Stream *pCTS = CPDF_Stream::Create(pColorTable, iPalette * 3, CPDF_Dictionary::Create());
173             m_pDocument->AddIndirectObject(pCTS);
174             pCS->AddReference(m_pDocument, pCTS);
175             pDict->SetAtReference(FX_BSTRC("ColorSpace"), m_pDocument, pCS);
176         } else {
177             pDict->SetAtName(FX_BSTRC("ColorSpace"), FX_BSTRC("DeviceGray"));
178         }
179         pDict->SetAtInteger(FX_BSTRC("BitsPerComponent"), 8);
180         if ((iCompress & 0x03) == PDF_IMAGE_NO_COMPRESS) {
181             dest_pitch = BitmapWidth;
182             opType = 1;
183         } else {
184             opType = 0;
185         }
186     } else {
187         pDict->SetAtName(FX_BSTRC("ColorSpace"), FX_BSTRC("DeviceRGB"));
188         pDict->SetAtInteger(FX_BSTRC("BitsPerComponent"), 8);
189         if ((iCompress & 0x03) == PDF_IMAGE_NO_COMPRESS) {
190             dest_pitch = BitmapWidth * 3;
191             opType = 2;
192         } else {
193             opType = 0;
194         }
195     }
196     const CFX_DIBitmap* pMaskBitmap = NULL;
197     FX_BOOL bDeleteMask = FALSE;
198     if (pBitmap->HasAlpha()) {
199         pMaskBitmap = pBitmap->GetAlphaMask();
200         bDeleteMask = TRUE;
201     }
202     if (!pMaskBitmap && pMask) {
203         FXDIB_Format maskFormat = pMask->GetFormat();
204         if (maskFormat == FXDIB_1bppMask || maskFormat == FXDIB_8bppMask) {
205             pMaskBitmap = pMask;
206         }
207     }
208     if (pMaskBitmap) {
209         int32_t maskWidth = pMaskBitmap->GetWidth();
210         int32_t maskHeight = pMaskBitmap->GetHeight();
211         uint8_t* mask_buf = NULL;
212         FX_STRSIZE mask_size;
213         CPDF_Dictionary* pMaskDict = new CPDF_Dictionary;
214         pMaskDict->SetAtName(FX_BSTRC("Type"), FX_BSTRC("XObject"));
215         pMaskDict->SetAtName(FX_BSTRC("Subtype"), FX_BSTRC("Image"));
216         pMaskDict->SetAtInteger(FX_BSTRC("Width"), maskWidth);
217         pMaskDict->SetAtInteger(FX_BSTRC("Height"), maskHeight);
218         pMaskDict->SetAtName(FX_BSTRC("ColorSpace"), FX_BSTRC("DeviceGray"));
219         pMaskDict->SetAtInteger(FX_BSTRC("BitsPerComponent"), 8);
220         if (pMaskBitmap->GetBPP() == 8 && (iCompress & PDF_IMAGE_MASK_LOSSY_COMPRESS) != 0) {
221             _DCTEncodeBitmap(pMaskDict, pMaskBitmap, pParam ? pParam->nQuality : 75, mask_buf, mask_size);
222         } else if (pMaskBitmap->GetFormat() == FXDIB_1bppMask) {
223             _JBIG2EncodeBitmap(pMaskDict, pMaskBitmap, m_pDocument, mask_buf, mask_size, TRUE);
224         } else {
225             mask_buf = FX_Alloc2D(uint8_t, maskHeight, maskWidth);
226             mask_size = maskHeight * maskWidth;  // Safe since checked alloc returned.
227             for (int32_t a = 0; a < maskHeight; a ++) {
228                 FXSYS_memcpy(mask_buf + a * maskWidth, pMaskBitmap->GetScanline(a), maskWidth);
229             }
230         }
231         pMaskDict->SetAtInteger(FX_BSTRC("Length"), mask_size);
232         if (bUseMatte) {
233             int a, r, g, b;
234             ArgbDecode(*(pParam->pMatteColor), a, r, g, b);
235             CPDF_Array* pMatte = new CPDF_Array;
236             pMatte->AddInteger(r);
237             pMatte->AddInteger(g);
238             pMatte->AddInteger(b);
239             pMaskDict->SetAt(FX_BSTRC("Matte"), pMatte);
240         }
241         CPDF_Stream* pMaskStream = new CPDF_Stream(mask_buf, mask_size, pMaskDict);
242         m_pDocument->AddIndirectObject(pMaskStream);
243         pDict->SetAtReference(FX_BSTRC("SMask"), m_pDocument, pMaskStream);
244         if (bDeleteMask) {
245             delete pMaskBitmap;
246         }
247     }
248     FX_BOOL bStream = pFileWrite != NULL && pFileRead != NULL;
249     if (opType == 0) {
250         if (iCompress & PDF_IMAGE_LOSSLESS_COMPRESS) {
251             if (pBitmap->GetBPP() == 1) {
252                 _JBIG2EncodeBitmap(pDict, pBitmap, m_pDocument, dest_buf, dest_size, TRUE);
253             }
254         } else {
255             if (pBitmap->GetBPP() == 1) {
256                 _JBIG2EncodeBitmap(pDict, pBitmap, m_pDocument, dest_buf, dest_size, FALSE);
257             } else if (pBitmap->GetBPP() >= 8 && pBitmap->GetPalette() != NULL) {
258                 CFX_DIBitmap *pNewBitmap = new CFX_DIBitmap();
259                 pNewBitmap->Copy(pBitmap);
260                 pNewBitmap->ConvertFormat(FXDIB_Rgb);
261                 SetImage(pNewBitmap, iCompress, pFileWrite, pFileRead);
262                 if (pDict) {
263                     pDict->Release();
264                     pDict = NULL;
265                 }
266                 if (dest_buf) {
267                     FX_Free(dest_buf);
268                     dest_buf = NULL;
269                 }
270                 dest_size = 0;
271                 delete pNewBitmap;
272                 return;
273             } else {
274                 if (bUseMatte) {
275                     CFX_DIBitmap *pNewBitmap = new CFX_DIBitmap();
276                     pNewBitmap->Create(BitmapWidth, BitmapHeight, FXDIB_Argb);
277                     uint8_t* dst_buf = pNewBitmap->GetBuffer();
278                     int32_t src_offset = 0;
279                     for (int32_t row = 0; row < BitmapHeight; row ++) {
280                         src_offset = row * src_pitch;
281                         for (int32_t column = 0; column < BitmapWidth; column ++) {
282                             FX_FLOAT alpha = src_buf[src_offset + 3] / 255.0f;
283                             dst_buf[src_offset] = (uint8_t)(src_buf[src_offset] * alpha);
284                             dst_buf[src_offset + 1] = (uint8_t)(src_buf[src_offset + 1] * alpha);
285                             dst_buf[src_offset + 2] = (uint8_t)(src_buf[src_offset + 2] * alpha);
286                             dst_buf[src_offset + 3] = (uint8_t)(src_buf[src_offset + 3]);
287                             src_offset += 4;
288                         }
289                     }
290                     _DCTEncodeBitmap(pDict, pNewBitmap, pParam ? pParam->nQuality : 75, dest_buf, dest_size);
291                     delete pNewBitmap;
292                 } else {
293                     _DCTEncodeBitmap(pDict, pBitmap, pParam ? pParam->nQuality : 75, dest_buf, dest_size);
294                 }
295             }
296         }
297         if (bStream) {
298             pFileWrite->WriteBlock(dest_buf, dest_size);
299             FX_Free(dest_buf);
300             dest_buf = NULL;
301         }
302     } else if (opType == 1) {
303         if (!bStream) {
304             dest_buf = FX_Alloc2D(uint8_t, dest_pitch, BitmapHeight);
305             dest_size = dest_pitch * BitmapHeight;  // Safe since checked alloc returned.
306         }
307         uint8_t* pDest = dest_buf;
308         for (int32_t i = 0; i < BitmapHeight; i ++) {
309             if (!bStream) {
310                 FXSYS_memcpy(pDest, src_buf, dest_pitch);
311                 pDest += dest_pitch;
312             } else {
313                 pFileWrite->WriteBlock(src_buf, dest_pitch);
314             }
315             src_buf += src_pitch;
316         }
317     } else if (opType == 2) {
318         if (!bStream) {
319             dest_buf = FX_Alloc2D(uint8_t, dest_pitch, BitmapHeight);
320             dest_size = dest_pitch * BitmapHeight;  // Safe since checked alloc returned.
321         } else {
322             dest_buf = FX_Alloc(uint8_t, dest_pitch);
323         }
324         uint8_t* pDest = dest_buf;
325         int32_t src_offset = 0;
326         int32_t dest_offset = 0;
327         for (int32_t row = 0; row < BitmapHeight; row ++) {
328             src_offset = row * src_pitch;
329             for (int32_t column = 0; column < BitmapWidth; column ++) {
330                 FX_FLOAT alpha = bUseMatte ? src_buf[src_offset + 3] / 255.0f : 1;
331                 pDest[dest_offset] = (uint8_t)(src_buf[src_offset + 2] * alpha);
332                 pDest[dest_offset + 1] = (uint8_t)(src_buf[src_offset + 1] * alpha);
333                 pDest[dest_offset + 2] = (uint8_t)(src_buf[src_offset] * alpha);
334                 dest_offset += 3;
335                 src_offset += bpp == 24 ? 3 : 4;
336             }
337             if (bStream) {
338                 pFileWrite->WriteBlock(pDest, dest_pitch);
339                 pDest = dest_buf;
340             } else {
341                 pDest += dest_pitch;
342             }
343             dest_offset = 0;
344         }
345         if (bStream) {
346             FX_Free(dest_buf);
347             dest_buf = NULL;
348         }
349     }
350     if (m_pStream == NULL) {
351         m_pStream = new CPDF_Stream(NULL, 0, NULL);
352     }
353     if (!bStream) {
354         m_pStream->InitStream(dest_buf, dest_size, pDict);
355     } else {
356         pFileWrite->Flush();
357         m_pStream->InitStream(pFileRead, pDict);
358     }
359     m_bIsMask = pBitmap->IsAlphaMask();
360     m_Width = BitmapWidth;
361     m_Height = BitmapHeight;
362     if (dest_buf) {
363         FX_Free(dest_buf);
364     }
365 }
366 void CPDF_Image::ResetCache(CPDF_Page* pPage, const CFX_DIBitmap* pBitmap)
367 {
368     pPage->GetRenderCache()->ResetBitmap(m_pStream, pBitmap);
369 }