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