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