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