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