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