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