Merge to XFA: Change FX_Alloc to FX_Try_Alloc in _JpegEncode
[pdfium.git] / core / src / fxcodec / codec / fx_codec_jpeg.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/fxcodec/fx_codec.h"
8 #include "../../../include/fxge/fx_dib.h"
9 #include "../../../src/fxcrt/fx_safe_types.h"
10 #include "codec_int.h"
11 extern "C" {
12     static void _JpegScanSOI(const FX_BYTE*& src_buf, FX_DWORD& src_size)
13     {
14         if (src_size == 0) {
15             return;
16         }
17         FX_DWORD offset = 0;
18         while (offset < src_size - 1) {
19             if (src_buf[offset] == 0xff && src_buf[offset + 1] == 0xd8) {
20                 src_buf += offset;
21                 src_size -= offset;
22                 return;
23             }
24             offset ++;
25         }
26     }
27 };
28 extern "C" {
29 #undef FAR
30 #include "../../fx_jpeglib.h"
31 }
32 extern "C" {
33     static void _src_do_nothing(struct jpeg_decompress_struct* cinfo) {}
34 };
35 extern "C" {
36     static void _error_fatal(j_common_ptr cinfo)
37     {
38         longjmp(*(jmp_buf*)cinfo->client_data, -1);
39     }
40 };
41 extern "C" {
42     static void _src_skip_data(struct jpeg_decompress_struct* cinfo, long num)
43     {
44         if (num > (long)cinfo->src->bytes_in_buffer) {
45             _error_fatal((j_common_ptr)cinfo);
46         }
47         cinfo->src->next_input_byte += num;
48         cinfo->src->bytes_in_buffer -= num;
49     }
50 };
51 extern "C" {
52     static boolean _src_fill_buffer(j_decompress_ptr cinfo)
53     {
54         return 0;
55     }
56 };
57 extern "C" {
58     static boolean _src_resync(j_decompress_ptr cinfo, int desired)
59     {
60         return 0;
61     }
62 };
63 extern "C" {
64     static void _error_do_nothing(j_common_ptr cinfo) {}
65 };
66 extern "C" {
67     static void _error_do_nothing1(j_common_ptr cinfo, int) {}
68 };
69 extern "C" {
70     static void _error_do_nothing2(j_common_ptr cinfo, char*) {}
71 };
72 #define JPEG_MARKER_EXIF                (JPEG_APP0 + 1)
73 #define JPEG_MARKER_ICC                 (JPEG_APP0 + 2)
74 #define JPEG_MARKER_AUTHORTIME  (JPEG_APP0 + 3)
75 #define JPEG_MARKER_MAXSIZE     0xFFFF
76 #define JPEG_OVERHEAD_LEN       14
77 static FX_BOOL _JpegIsIccMarker(jpeg_saved_marker_ptr marker)
78 {
79     if (marker->marker == JPEG_MARKER_ICC &&
80             marker->data_length >= JPEG_OVERHEAD_LEN &&
81             (FXSYS_memcmp32(marker->data, "\x49\x43\x43\x5f\x50\x52\x4f\x46\x49\x4c\x45\x00", 12) == 0)) {
82         return TRUE;
83     }
84     return FALSE;
85 }
86 static  FX_BOOL _JpegLoadIccProfile(j_decompress_ptr cinfo, FX_LPBYTE* icc_buf_ptr, FX_DWORD* icc_length)
87 {
88     if(icc_buf_ptr == NULL || icc_length == NULL) {
89         return FALSE;
90     }
91     *icc_buf_ptr = NULL;
92     *icc_length = 0;
93     FX_LPBYTE icc_data_ptr = NULL;
94     FX_DWORD icc_data_len = 0;
95     FX_BYTE count_icc_marker = 0;
96     FX_BYTE num_icc_marker = 0;
97     jpeg_saved_marker_ptr marker_list[256] = {NULL};
98     for (jpeg_saved_marker_ptr cur_marker = cinfo->marker_list;
99             cur_marker != NULL;
100             cur_marker = cur_marker->next) {
101         if(_JpegIsIccMarker(cur_marker)) {
102             if(count_icc_marker == 0) {
103                 num_icc_marker = cur_marker->data[13];
104             } else if(num_icc_marker != cur_marker->data[13]) {
105                 return FALSE;
106             }
107             int sn = cur_marker->data[12] - 1;
108             if(sn < 0 || sn >= num_icc_marker) {
109                 return FALSE;
110             }
111             if(marker_list[sn] == NULL) {
112                 marker_list[sn] = cur_marker;
113             } else {
114                 return FALSE;
115             }
116             count_icc_marker ++;
117             icc_data_len +=     (cur_marker->data_length - JPEG_OVERHEAD_LEN);
118         }
119     }
120     if(count_icc_marker != num_icc_marker) {
121         return FALSE;
122     }
123     if(num_icc_marker == 0) {
124         return TRUE;
125     }
126     icc_data_ptr = FX_Alloc(FX_BYTE, icc_data_len);
127     if(icc_buf_ptr == NULL)     {
128         return FALSE;
129     }
130     *icc_buf_ptr = icc_data_ptr;
131     *icc_length = icc_data_len;
132     for (int idx = 0; idx < num_icc_marker; idx++) {
133         icc_data_len = marker_list[idx]->data_length - JPEG_OVERHEAD_LEN;
134         FXSYS_memcpy32(icc_data_ptr, marker_list[idx]->data + JPEG_OVERHEAD_LEN, icc_data_len);
135         icc_data_ptr += icc_data_len;
136     }
137     return TRUE;
138 }
139 static  FX_BOOL _JpegEmbedIccProfile(j_compress_ptr cinfo, FX_LPCBYTE icc_buf_ptr, FX_DWORD icc_length)
140 {
141     if(icc_buf_ptr == NULL || icc_length == 0) {
142         return FALSE;
143     }
144     FX_DWORD icc_segment_size = (JPEG_MARKER_MAXSIZE - 2 - JPEG_OVERHEAD_LEN);
145     FX_DWORD icc_segment_num = (icc_length / icc_segment_size) + 1;
146     if (icc_segment_num > 255)  {
147         return FALSE;
148     }
149     FX_DWORD icc_data_length = JPEG_OVERHEAD_LEN + (icc_segment_num > 1 ? icc_segment_size : icc_length);
150     FX_LPBYTE icc_data = FX_Alloc(FX_BYTE, icc_data_length);
151     FXSYS_memcpy32(icc_data, "\x49\x43\x43\x5f\x50\x52\x4f\x46\x49\x4c\x45\x00", 12);
152     icc_data[13] = (FX_BYTE)icc_segment_num;
153     for (FX_BYTE i = 0; i < (icc_segment_num - 1); i++) {
154         icc_data[12] = i + 1;
155         FXSYS_memcpy32(icc_data + JPEG_OVERHEAD_LEN, icc_buf_ptr + i * icc_segment_size, icc_segment_size);
156         jpeg_write_marker(cinfo, JPEG_MARKER_ICC, icc_data, icc_data_length);
157     }
158     icc_data[12] = (FX_BYTE)icc_segment_num;
159     FX_DWORD icc_size = (icc_segment_num - 1) * icc_segment_size;
160     FXSYS_memcpy32(icc_data + JPEG_OVERHEAD_LEN, icc_buf_ptr + icc_size, icc_length - icc_size);
161     jpeg_write_marker(cinfo, JPEG_MARKER_ICC, icc_data, JPEG_OVERHEAD_LEN + icc_length - icc_size);
162     FX_Free(icc_data);
163     return TRUE;
164 }
165 extern "C" {
166     static void _dest_do_nothing(j_compress_ptr cinfo) {}
167 };
168 extern "C" {
169     static boolean _dest_empty(j_compress_ptr cinfo)
170     {
171         return FALSE;
172     }
173 };
174 #define JPEG_BLOCK_SIZE 1048576
175 static void _JpegEncode(const CFX_DIBSource* pSource, FX_LPBYTE& dest_buf, FX_STRSIZE& dest_size, int quality, FX_LPCBYTE icc_buf, FX_DWORD icc_length)
176 {
177     struct jpeg_error_mgr jerr;
178     jerr.error_exit = _error_do_nothing;
179     jerr.emit_message = _error_do_nothing1;
180     jerr.output_message = _error_do_nothing;
181     jerr.format_message = _error_do_nothing2;
182     jerr.reset_error_mgr = _error_do_nothing;
183
184     struct jpeg_compress_struct cinfo;
185     memset(&cinfo, 0, sizeof(cinfo));
186     cinfo.err = &jerr;
187     jpeg_create_compress(&cinfo);
188     int Bpp = pSource->GetBPP() / 8;
189     FX_DWORD nComponents = Bpp >= 3 ? (pSource->IsCmykImage() ? 4 : 3) : 1;
190     FX_DWORD pitch = pSource->GetPitch();
191     FX_DWORD width = pdfium::base::checked_cast<FX_DWORD>(pSource->GetWidth());
192     FX_DWORD height = pdfium::base::checked_cast<FX_DWORD>(pSource->GetHeight());
193     FX_SAFE_DWORD safe_buf_len = width;
194     safe_buf_len *= height;
195     safe_buf_len *= nComponents;
196     safe_buf_len += 1024;
197     if (icc_length) {
198         safe_buf_len += 255 * 18;
199         safe_buf_len += icc_length;
200     }
201     FX_DWORD dest_buf_length = 0;
202     if (!safe_buf_len.IsValid()) {
203         dest_buf = nullptr;
204     } else {
205         dest_buf_length = safe_buf_len.ValueOrDie();
206         dest_buf = FX_TryAlloc(FX_BYTE, dest_buf_length);
207         const int MIN_TRY_BUF_LEN = 1024;
208         while (!dest_buf && dest_buf_length > MIN_TRY_BUF_LEN) {
209             dest_buf_length >>= 1;
210             dest_buf = FX_TryAlloc(FX_BYTE, dest_buf_length);
211         }
212     }
213     if (!dest_buf) {
214         FX_OutOfMemoryTerminate(); 
215     }
216     struct jpeg_destination_mgr dest;
217     dest.init_destination = _dest_do_nothing;
218     dest.term_destination = _dest_do_nothing;
219     dest.empty_output_buffer = _dest_empty;
220     dest.next_output_byte = dest_buf;
221     dest.free_in_buffer = dest_buf_length;
222     cinfo.dest = &dest;
223     cinfo.image_width = width;
224     cinfo.image_height = height;
225     cinfo.input_components = nComponents;
226     if (nComponents == 1) {
227         cinfo.in_color_space = JCS_GRAYSCALE;
228     } else if (nComponents == 3) {
229         cinfo.in_color_space = JCS_RGB;
230     } else {
231         cinfo.in_color_space = JCS_CMYK;
232     }
233     FX_LPBYTE line_buf = NULL;
234     if (nComponents > 1) {
235         line_buf = FX_Alloc2D(FX_BYTE, width, nComponents);
236     }
237     jpeg_set_defaults(&cinfo);
238     if(quality != 75) {
239         jpeg_set_quality(&cinfo, quality, TRUE);
240     }
241     jpeg_start_compress(&cinfo, TRUE);
242     _JpegEmbedIccProfile(&cinfo, icc_buf, icc_length);
243     JSAMPROW row_pointer[1];
244     JDIMENSION row;
245     while (cinfo.next_scanline < cinfo.image_height) {
246         FX_LPCBYTE src_scan = pSource->GetScanline(cinfo.next_scanline);
247         if (nComponents > 1) {
248             FX_LPBYTE dest_scan = line_buf;
249             if (nComponents == 3) {
250                 for (int i = 0; i < width; i ++) {
251                     dest_scan[0] = src_scan[2];
252                     dest_scan[1] = src_scan[1];
253                     dest_scan[2] = src_scan[0];
254                     dest_scan += 3;
255                     src_scan += Bpp;
256                 }
257             } else {
258                 for (int i = 0; i < pitch; i ++) {
259                     *dest_scan++ = ~*src_scan++;
260                 }
261             }
262             row_pointer[0] = line_buf;
263         } else {
264             row_pointer[0] = (FX_LPBYTE)src_scan;
265         }
266         row = cinfo.next_scanline;
267         jpeg_write_scanlines(&cinfo, row_pointer, 1);
268         if (cinfo.next_scanline == row) {
269             dest_buf = FX_Realloc(FX_BYTE, dest_buf, dest_buf_length + JPEG_BLOCK_SIZE);
270             dest.next_output_byte = dest_buf + dest_buf_length - dest.free_in_buffer;
271             dest_buf_length += JPEG_BLOCK_SIZE;
272             dest.free_in_buffer += JPEG_BLOCK_SIZE;
273         }
274     }
275     jpeg_finish_compress(&cinfo);
276     jpeg_destroy_compress(&cinfo);
277     if (line_buf) {
278         FX_Free(line_buf);
279     }
280     dest_size = dest_buf_length - (FX_STRSIZE)dest.free_in_buffer;
281 }
282 static void _JpegLoadAttribute(struct jpeg_decompress_struct* pInfo, CFX_DIBAttribute* pAttribute)
283 {
284     if (pInfo == NULL || pAttribute == NULL) {
285         return;
286     }
287     if (pAttribute) {
288         pAttribute->m_nXDPI = pInfo->X_density;
289         pAttribute->m_nYDPI = pInfo->Y_density;
290         pAttribute->m_wDPIUnit = pInfo->density_unit;
291     }
292 }
293 static FX_BOOL _JpegLoadInfo(FX_LPCBYTE src_buf, FX_DWORD src_size, int& width, int& height,
294                              int& num_components, int& bits_per_components, FX_BOOL& color_transform,
295                              FX_LPBYTE* icc_buf_ptr, FX_DWORD* icc_length)
296 {
297     _JpegScanSOI(src_buf, src_size);
298     struct jpeg_decompress_struct cinfo;
299     struct jpeg_error_mgr jerr;
300     jerr.error_exit = _error_fatal;
301     jerr.emit_message = _error_do_nothing1;
302     jerr.output_message = _error_do_nothing;
303     jerr.format_message = _error_do_nothing2;
304     jerr.reset_error_mgr = _error_do_nothing;
305     jerr.trace_level = 0;
306     cinfo.err = &jerr;
307     jmp_buf mark;
308     cinfo.client_data = &mark;
309     if (setjmp(mark) == -1) {
310         return FALSE;
311     }
312     jpeg_create_decompress(&cinfo);
313     struct jpeg_source_mgr src;
314     src.init_source = _src_do_nothing;
315     src.term_source = _src_do_nothing;
316     src.skip_input_data = _src_skip_data;
317     src.fill_input_buffer = _src_fill_buffer;
318     src.resync_to_restart = _src_resync;
319     src.bytes_in_buffer = src_size;
320     src.next_input_byte = src_buf;
321     cinfo.src = &src;
322     if (setjmp(mark) == -1) {
323         jpeg_destroy_decompress(&cinfo);
324         return FALSE;
325     }
326     if(icc_buf_ptr && icc_length) {
327         jpeg_save_markers(&cinfo, JPEG_MARKER_ICC, JPEG_MARKER_MAXSIZE);
328     }
329     int ret = jpeg_read_header(&cinfo, TRUE);
330     if (ret != JPEG_HEADER_OK) {
331         jpeg_destroy_decompress(&cinfo);
332         return FALSE;
333     }
334     width = cinfo.image_width;
335     height = cinfo.image_height;
336     num_components = cinfo.num_components;
337     color_transform = cinfo.jpeg_color_space == JCS_YCbCr || cinfo.jpeg_color_space == JCS_YCCK;
338     bits_per_components = cinfo.data_precision;
339     if(icc_buf_ptr != NULL) {
340         *icc_buf_ptr = NULL;
341     }
342     if(icc_length != NULL) {
343         *icc_length = 0;
344     }
345     jpeg_destroy_decompress(&cinfo);
346     return TRUE;
347 }
348 class CCodec_JpegDecoder : public CCodec_ScanlineDecoder
349 {
350 public:
351     CCodec_JpegDecoder();
352     ~CCodec_JpegDecoder();
353     FX_BOOL                             Create(FX_LPCBYTE src_buf, FX_DWORD src_size, int width, int height, int nComps,
354                                FX_BOOL ColorTransform, IFX_JpegProvider* pJP);
355     virtual void                Destroy()
356     {
357         delete this;
358     }
359     virtual void                v_DownScale(int dest_width, int dest_height);
360     virtual FX_BOOL             v_Rewind();
361     virtual FX_LPBYTE   v_GetNextLine();
362     virtual FX_DWORD    GetSrcOffset();
363     jmp_buf             m_JmpBuf;
364     struct jpeg_decompress_struct cinfo;
365     struct jpeg_error_mgr jerr;
366     struct jpeg_source_mgr src;
367     FX_LPCBYTE  m_SrcBuf;
368     FX_DWORD    m_SrcSize;
369     FX_LPBYTE   m_pScanlineBuf;
370     FX_BOOL             InitDecode();
371     FX_BOOL             m_bInited, m_bStarted, m_bJpegTransform;
372 protected:
373     IFX_JpegProvider*   m_pExtProvider;
374     void*                               m_pExtContext;
375     FX_DWORD                    m_nDefaultScaleDenom;
376 };
377 CCodec_JpegDecoder::CCodec_JpegDecoder()
378 {
379     m_pScanlineBuf = NULL;
380     m_DownScale = 1;
381     m_bStarted = FALSE;
382     m_bInited = FALSE;
383     m_pExtProvider = NULL;
384     m_pExtContext = NULL;
385     FXSYS_memset32(&cinfo, 0, sizeof(cinfo));
386     FXSYS_memset32(&jerr, 0, sizeof(jerr));
387     FXSYS_memset32(&src, 0, sizeof(src));
388     m_nDefaultScaleDenom = 1;
389 }
390 CCodec_JpegDecoder::~CCodec_JpegDecoder()
391 {
392     if (m_pExtProvider) {
393         m_pExtProvider->DestroyDecoder(m_pExtContext);
394         return;
395     }
396     if (m_pScanlineBuf) {
397         FX_Free(m_pScanlineBuf);
398     }
399     if (m_bInited) {
400         jpeg_destroy_decompress(&cinfo);
401     }
402 }
403 FX_BOOL CCodec_JpegDecoder::InitDecode()
404 {
405     cinfo.err = &jerr;
406     cinfo.client_data = &m_JmpBuf;
407     if (setjmp(m_JmpBuf) == -1) {
408         return FALSE;
409     }
410     jpeg_create_decompress(&cinfo);
411     m_bInited = TRUE;
412     cinfo.src = &src;
413     src.bytes_in_buffer = m_SrcSize;
414     src.next_input_byte = m_SrcBuf;
415     if (setjmp(m_JmpBuf) == -1) {
416         jpeg_destroy_decompress(&cinfo);
417         m_bInited = FALSE;
418         return FALSE;
419     }
420     cinfo.image_width = m_OrigWidth;
421     cinfo.image_height = m_OrigHeight;
422     int ret = jpeg_read_header(&cinfo, TRUE);
423     if (ret != JPEG_HEADER_OK) {
424         return FALSE;
425     }
426     if (cinfo.saw_Adobe_marker) {
427         m_bJpegTransform = TRUE;
428     }
429     if (cinfo.num_components == 3 && !m_bJpegTransform) {
430         cinfo.out_color_space = cinfo.jpeg_color_space;
431     }
432     m_OrigWidth = cinfo.image_width;
433     m_OrigHeight = cinfo.image_height;
434     m_OutputWidth = m_OrigWidth;
435     m_OutputHeight = m_OrigHeight;
436     m_nDefaultScaleDenom = cinfo.scale_denom;
437     return TRUE;
438 }
439 FX_BOOL CCodec_JpegDecoder::Create(FX_LPCBYTE src_buf, FX_DWORD src_size, int width, int height,
440                                    int nComps, FX_BOOL ColorTransform, IFX_JpegProvider* pJP)
441 {
442     if (pJP) {
443         m_pExtProvider = pJP;
444         m_pExtContext = m_pExtProvider->CreateDecoder(src_buf, src_size, width, height, nComps, ColorTransform);
445         return m_pExtContext != NULL;
446     }
447     _JpegScanSOI(src_buf, src_size);
448     m_SrcBuf = src_buf;
449     m_SrcSize = src_size;
450     jerr.error_exit = _error_fatal;
451     jerr.emit_message = _error_do_nothing1;
452     jerr.output_message = _error_do_nothing;
453     jerr.format_message = _error_do_nothing2;
454     jerr.reset_error_mgr = _error_do_nothing;
455     src.init_source = _src_do_nothing;
456     src.term_source = _src_do_nothing;
457     src.skip_input_data = _src_skip_data;
458     src.fill_input_buffer = _src_fill_buffer;
459     src.resync_to_restart = _src_resync;
460     m_bJpegTransform = ColorTransform;
461     if(src_size > 1 && FXSYS_memcmp32(src_buf + src_size - 2, "\xFF\xD9", 2) != 0) {
462         ((FX_LPBYTE)src_buf)[src_size - 2] = 0xFF;
463         ((FX_LPBYTE)src_buf)[src_size - 1] = 0xD9;
464     }
465     m_OutputWidth = m_OrigWidth = width;
466     m_OutputHeight = m_OrigHeight = height;
467     if (!InitDecode()) {
468         return FALSE;
469     }
470     if (cinfo.num_components < nComps) {
471         return FALSE;
472     }
473     if ((int)cinfo.image_width < width) {
474         return FALSE;
475     }
476     m_Pitch = (cinfo.image_width * cinfo.num_components + 3) / 4 * 4;
477     m_pScanlineBuf = FX_Alloc(FX_BYTE, m_Pitch);
478     m_nComps = cinfo.num_components;
479     m_bpc = 8;
480     m_bColorTransformed = FALSE;
481     m_bStarted = FALSE;
482     return TRUE;
483 }
484 extern "C" {
485     FX_INT32 FX_GetDownsampleRatio(FX_INT32 originWidth, FX_INT32 originHeight, FX_INT32 downsampleWidth, FX_INT32 downsampleHeight)
486     {
487         int iratio_w = originWidth / downsampleWidth;
488         int iratio_h = originHeight / downsampleHeight;
489         int ratio = (iratio_w > iratio_h) ? iratio_h : iratio_w;
490         if (ratio >= 8) {
491             return 8;
492         } else if (ratio >= 4) {
493             return 4;
494         } else if (ratio >= 2) {
495             return 2;
496         }
497         return 1;
498     }
499 }
500 void CCodec_JpegDecoder::v_DownScale(int dest_width, int dest_height)
501 {
502     if (m_pExtProvider) {
503         m_pExtProvider->DownScale(m_pExtContext, dest_width, dest_height);
504         return;
505     }
506     int old_scale = m_DownScale;
507     m_DownScale = FX_GetDownsampleRatio(m_OrigWidth, m_OrigHeight, dest_width, dest_height);
508     m_OutputWidth = (m_OrigWidth + m_DownScale - 1) / m_DownScale;
509     m_OutputHeight = (m_OrigHeight + m_DownScale - 1) / m_DownScale;
510     m_Pitch = (m_OutputWidth * m_nComps + 3) / 4 * 4;
511     if (old_scale != m_DownScale) {
512         m_NextLine = -1;
513     }
514 }
515 FX_BOOL CCodec_JpegDecoder::v_Rewind()
516 {
517     if (m_pExtProvider) {
518         return m_pExtProvider->Rewind(m_pExtContext);
519     }
520     if (m_bStarted) {
521         jpeg_destroy_decompress(&cinfo);
522         if (!InitDecode()) {
523             return FALSE;
524         }
525     }
526     if (setjmp(m_JmpBuf) == -1) {
527         return FALSE;
528     }
529     cinfo.scale_denom = m_nDefaultScaleDenom * m_DownScale;
530     m_OutputWidth = (m_OrigWidth + m_DownScale - 1) / m_DownScale;
531     m_OutputHeight = (m_OrigHeight + m_DownScale - 1) / m_DownScale;
532     if (!jpeg_start_decompress(&cinfo)) {
533         jpeg_destroy_decompress(&cinfo);
534         return FALSE;
535     }
536     if ((int)cinfo.output_width > m_OrigWidth) {
537         FXSYS_assert(FALSE);
538         return FALSE;
539     }
540     m_bStarted = TRUE;
541     return TRUE;
542 }
543 FX_LPBYTE CCodec_JpegDecoder::v_GetNextLine()
544 {
545     if (m_pExtProvider) {
546         return m_pExtProvider->GetNextLine(m_pExtContext);
547     }
548     int nlines = jpeg_read_scanlines(&cinfo, &m_pScanlineBuf, 1);
549     if (nlines < 1) {
550         return NULL;
551     }
552     return m_pScanlineBuf;
553 }
554 FX_DWORD CCodec_JpegDecoder::GetSrcOffset()
555 {
556     if (m_pExtProvider) {
557         return m_pExtProvider->GetSrcOffset(m_pExtContext);
558     }
559     return (FX_DWORD)(m_SrcSize - src.bytes_in_buffer);
560 }
561 ICodec_ScanlineDecoder* CCodec_JpegModule::CreateDecoder(FX_LPCBYTE src_buf, FX_DWORD src_size,
562         int width, int height, int nComps, FX_BOOL ColorTransform)
563 {
564     if (src_buf == NULL || src_size == 0) {
565         return NULL;
566     }
567     CCodec_JpegDecoder* pDecoder = new CCodec_JpegDecoder;
568     if (!pDecoder->Create(src_buf, src_size, width, height, nComps, ColorTransform, m_pExtProvider)) {
569         delete pDecoder;
570         return NULL;
571     }
572     return pDecoder;
573 }
574 FX_BOOL CCodec_JpegModule::LoadInfo(FX_LPCBYTE src_buf, FX_DWORD src_size, int& width, int& height,
575                                     int& num_components, int& bits_per_components, FX_BOOL& color_transform,
576                                     FX_LPBYTE* icc_buf_ptr, FX_DWORD* icc_length)
577 {
578     if (m_pExtProvider) {
579         return m_pExtProvider->LoadInfo(src_buf, src_size, width, height,
580                                         num_components, bits_per_components, color_transform,
581                                         icc_buf_ptr, icc_length);
582     }
583     return _JpegLoadInfo(src_buf, src_size, width, height, num_components, bits_per_components, color_transform, icc_buf_ptr, icc_length);
584 }
585 FX_BOOL CCodec_JpegModule::Encode(const CFX_DIBSource* pSource, FX_LPBYTE& dest_buf, FX_STRSIZE& dest_size, int quality, FX_LPCBYTE icc_buf, FX_DWORD icc_length)
586 {
587     if (m_pExtProvider) {
588         return m_pExtProvider->Encode(pSource, dest_buf, dest_size, quality, icc_buf, icc_length);
589     }
590     if(pSource->GetBPP() < 8 || pSource->GetPalette() != NULL) {
591         ASSERT(pSource->GetBPP() >= 8 && pSource->GetPalette() == NULL);
592         return FALSE;
593     }
594     _JpegEncode(pSource, dest_buf, dest_size, quality, icc_buf, icc_length);
595     return TRUE;
596 }
597 struct FXJPEG_Context {
598     jmp_buf                     m_JumpMark;
599     jpeg_decompress_struct m_Info;
600     jpeg_error_mgr      m_ErrMgr;
601     jpeg_source_mgr     m_SrcMgr;
602     unsigned int        m_SkipSize;
603     void*               (*m_AllocFunc)(unsigned int);
604     void                (*m_FreeFunc)(void*);
605 };
606 extern "C" {
607     static void _error_fatal1(j_common_ptr cinfo)
608     {
609         longjmp(((FXJPEG_Context*)cinfo->client_data)->m_JumpMark, -1);
610     }
611 };
612 extern "C" {
613     static void _src_skip_data1(struct jpeg_decompress_struct* cinfo, long num)
614     {
615         if (cinfo->src->bytes_in_buffer < (size_t)num) {
616             ((FXJPEG_Context*)cinfo->client_data)->m_SkipSize = (unsigned int)(num - cinfo->src->bytes_in_buffer);
617             cinfo->src->bytes_in_buffer = 0;
618         } else {
619             cinfo->src->next_input_byte += num;
620             cinfo->src->bytes_in_buffer -= num;
621         }
622     }
623 };
624 static void* jpeg_alloc_func(unsigned int size)
625 {
626     return FX_Alloc(char, size);
627 }
628 static void jpeg_free_func(void* p)
629 {
630     FX_Free(p);
631 }
632 void* CCodec_JpegModule::Start()
633 {
634     if (m_pExtProvider) {
635         return m_pExtProvider->Start();
636     }
637     FXJPEG_Context* p = (FXJPEG_Context*)FX_Alloc(FX_BYTE, sizeof(FXJPEG_Context));
638     p->m_AllocFunc = jpeg_alloc_func;
639     p->m_FreeFunc = jpeg_free_func;
640     p->m_ErrMgr.error_exit = _error_fatal1;
641     p->m_ErrMgr.emit_message = _error_do_nothing1;
642     p->m_ErrMgr.output_message = _error_do_nothing;
643     p->m_ErrMgr.format_message = _error_do_nothing2;
644     p->m_ErrMgr.reset_error_mgr = _error_do_nothing;
645     p->m_SrcMgr.init_source = _src_do_nothing;
646     p->m_SrcMgr.term_source = _src_do_nothing;
647     p->m_SrcMgr.skip_input_data = _src_skip_data1;
648     p->m_SrcMgr.fill_input_buffer = _src_fill_buffer;
649     p->m_SrcMgr.resync_to_restart = _src_resync;
650     p->m_Info.client_data = p;
651     p->m_Info.err = &p->m_ErrMgr;
652     if (setjmp(p->m_JumpMark) == -1) {
653         return 0;
654     }
655     jpeg_create_decompress(&p->m_Info);
656     p->m_Info.src = &p->m_SrcMgr;
657     p->m_SkipSize = 0;
658     return p;
659 }
660 void CCodec_JpegModule::Finish(void* pContext)
661 {
662     if (m_pExtProvider) {
663         m_pExtProvider->Finish(pContext);
664         return;
665     }
666     FXJPEG_Context* p = (FXJPEG_Context*)pContext;
667     jpeg_destroy_decompress(&p->m_Info);
668     p->m_FreeFunc(p);
669 }
670 void CCodec_JpegModule::Input(void* pContext, const unsigned char* src_buf, FX_DWORD src_size)
671 {
672     if (m_pExtProvider) {
673         m_pExtProvider->Input(pContext, src_buf, src_size);
674         return;
675     }
676     FXJPEG_Context* p = (FXJPEG_Context*)pContext;
677     if (p->m_SkipSize) {
678         if (p->m_SkipSize > src_size) {
679             p->m_SrcMgr.bytes_in_buffer = 0;
680             p->m_SkipSize -= src_size;
681             return;
682         }
683         src_size -= p->m_SkipSize;
684         src_buf += p->m_SkipSize;
685         p->m_SkipSize = 0;
686     }
687     p->m_SrcMgr.next_input_byte = src_buf;
688     p->m_SrcMgr.bytes_in_buffer = src_size;
689 }
690 int CCodec_JpegModule::ReadHeader(void* pContext, int* width, int* height, int* nComps, CFX_DIBAttribute* pAttribute)
691 {
692     if (m_pExtProvider) {
693         return m_pExtProvider->ReadHeader(pContext, width, height, nComps, pAttribute);
694     }
695     FXJPEG_Context* p = (FXJPEG_Context*)pContext;
696     if (setjmp(p->m_JumpMark) == -1) {
697         return 1;
698     }
699     int ret = jpeg_read_header(&p->m_Info, true);
700     if (ret == JPEG_SUSPENDED) {
701         return 2;
702     }
703     if (ret != JPEG_HEADER_OK) {
704         return 1;
705     }
706     *width = p->m_Info.image_width;
707     *height = p->m_Info.image_height;
708     *nComps = p->m_Info.num_components;
709     _JpegLoadAttribute(&p->m_Info, pAttribute);
710     return 0;
711 }
712 FX_BOOL CCodec_JpegModule::StartScanline(void* pContext, int down_scale)
713 {
714     if (m_pExtProvider) {
715         return m_pExtProvider->StartScanline(pContext, down_scale);
716     }
717     FXJPEG_Context* p = (FXJPEG_Context*)pContext;
718     if (setjmp(p->m_JumpMark) == -1) {
719         return FALSE;
720     }
721     p->m_Info.scale_denom = down_scale;
722     return jpeg_start_decompress(&p->m_Info);
723 }
724 FX_BOOL CCodec_JpegModule::ReadScanline(void* pContext, unsigned char* dest_buf)
725 {
726     if (m_pExtProvider) {
727         return m_pExtProvider->ReadScanline(pContext, dest_buf);
728     }
729     FXJPEG_Context* p = (FXJPEG_Context*)pContext;
730     if (setjmp(p->m_JumpMark) == -1) {
731         return FALSE;
732     }
733     int nlines = jpeg_read_scanlines(&p->m_Info, &dest_buf, 1);
734     return nlines == 1;
735 }
736 FX_DWORD CCodec_JpegModule::GetAvailInput(void* pContext, FX_LPBYTE* avail_buf_ptr)
737 {
738     if (m_pExtProvider) {
739         return m_pExtProvider->GetAvailInput(pContext, avail_buf_ptr);
740     }
741     if(avail_buf_ptr != NULL) {
742         *avail_buf_ptr = NULL;
743         if(((FXJPEG_Context*)pContext)->m_SrcMgr.bytes_in_buffer > 0) {
744             *avail_buf_ptr = (FX_LPBYTE)((FXJPEG_Context*)pContext)->m_SrcMgr.next_input_byte;
745         }
746     }
747     return (FX_DWORD)((FXJPEG_Context*)pContext)->m_SrcMgr.bytes_in_buffer;
748 }