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 "core/include/fxcodec/fx_codec.h" 8 #include "core/include/fxge/fx_dib.h" 9 #include "codec_int.h" 10 #include "core/src/fxcodec/lbmp/fx_bmp.h" 11 struct FXBMP_Context { 12 bmp_decompress_struct_p bmp_ptr; 13 void* parent_ptr; 14 void* child_ptr; 15 16 void* (*m_AllocFunc)(unsigned int); 17 void (*m_FreeFunc)(void*); 18 }; 19 extern "C" { 20 static void* _bmp_alloc_func(unsigned int size) { 21 return FX_Alloc(char, size); 22 } 23 static void _bmp_free_func(void* p) { 24 if (p != NULL) { 25 FX_Free(p); 26 } 27 } 28 }; 29 static void _bmp_error_data(bmp_decompress_struct_p bmp_ptr, 30 const FX_CHAR* err_msg) { 31 FXSYS_strncpy((char*)bmp_ptr->err_ptr, err_msg, BMP_MAX_ERROR_SIZE - 1); 32 longjmp(bmp_ptr->jmpbuf, 1); 33 } 34 static void _bmp_read_scanline(bmp_decompress_struct_p bmp_ptr, 35 int32_t row_num, 36 uint8_t* row_buf) { 37 FXBMP_Context* p = (FXBMP_Context*)bmp_ptr->context_ptr; 38 CCodec_BmpModule* pModule = (CCodec_BmpModule*)p->parent_ptr; 39 pModule->ReadScanlineCallback(p->child_ptr, row_num, row_buf); 40 } 41 static FX_BOOL _bmp_get_data_position(bmp_decompress_struct_p bmp_ptr, 42 FX_DWORD rcd_pos) { 43 FXBMP_Context* p = (FXBMP_Context*)bmp_ptr->context_ptr; 44 CCodec_BmpModule* pModule = (CCodec_BmpModule*)p->parent_ptr; 45 return pModule->InputImagePositionBufCallback(p->child_ptr, rcd_pos); 46 } 47 void* CCodec_BmpModule::Start(void* pModule) { 48 FXBMP_Context* p = (FXBMP_Context*)FX_Alloc(uint8_t, sizeof(FXBMP_Context)); 49 if (p == NULL) { 50 return NULL; 51 } 52 FXSYS_memset(p, 0, sizeof(FXBMP_Context)); 53 if (p == NULL) { 54 return NULL; 55 } 56 p->m_AllocFunc = _bmp_alloc_func; 57 p->m_FreeFunc = _bmp_free_func; 58 p->bmp_ptr = NULL; 59 p->parent_ptr = (void*)this; 60 p->child_ptr = pModule; 61 p->bmp_ptr = _bmp_create_decompress(); 62 if (p->bmp_ptr == NULL) { 63 FX_Free(p); 64 return NULL; 65 } 66 p->bmp_ptr->context_ptr = (void*)p; 67 p->bmp_ptr->err_ptr = m_szLastError; 68 p->bmp_ptr->_bmp_error_fn = _bmp_error_data; 69 p->bmp_ptr->_bmp_get_row_fn = _bmp_read_scanline; 70 p->bmp_ptr->_bmp_get_data_position_fn = _bmp_get_data_position; 71 return p; 72 } 73 void CCodec_BmpModule::Finish(void* pContext) { 74 FXBMP_Context* p = (FXBMP_Context*)pContext; 75 if (p != NULL) { 76 _bmp_destroy_decompress(&p->bmp_ptr); 77 p->m_FreeFunc(p); 78 } 79 } 80 int32_t CCodec_BmpModule::ReadHeader(void* pContext, 81 int32_t* width, 82 int32_t* height, 83 FX_BOOL* tb_flag, 84 int32_t* components, 85 int32_t* pal_num, 86 FX_DWORD** pal_pp, 87 CFX_DIBAttribute* pAttribute) { 88 FXBMP_Context* p = (FXBMP_Context*)pContext; 89 if (setjmp(p->bmp_ptr->jmpbuf)) { 90 return 0; 91 } 92 int32_t ret = _bmp_read_header(p->bmp_ptr); 93 if (ret != 1) { 94 return ret; 95 } 96 *width = p->bmp_ptr->width; 97 *height = p->bmp_ptr->height; 98 *tb_flag = p->bmp_ptr->imgTB_flag; 99 *components = p->bmp_ptr->components; 100 *pal_num = p->bmp_ptr->pal_num; 101 *pal_pp = p->bmp_ptr->pal_ptr; 102 if (pAttribute) { 103 pAttribute->m_wDPIUnit = FXCODEC_RESUNIT_METER; 104 pAttribute->m_nXDPI = p->bmp_ptr->dpi_x; 105 pAttribute->m_nYDPI = p->bmp_ptr->dpi_y; 106 pAttribute->m_nBmpCompressType = p->bmp_ptr->compress_flag; 107 } 108 return 1; 109 } 110 int32_t CCodec_BmpModule::LoadImage(void* pContext) { 111 FXBMP_Context* p = (FXBMP_Context*)pContext; 112 if (setjmp(p->bmp_ptr->jmpbuf)) { 113 return 0; 114 } 115 return _bmp_decode_image(p->bmp_ptr); 116 } 117 FX_DWORD CCodec_BmpModule::GetAvailInput(void* pContext, 118 uint8_t** avial_buf_ptr) { 119 FXBMP_Context* p = (FXBMP_Context*)pContext; 120 return _bmp_get_avail_input(p->bmp_ptr, avial_buf_ptr); 121 } 122 void CCodec_BmpModule::Input(void* pContext, 123 const uint8_t* src_buf, 124 FX_DWORD src_size) { 125 FXBMP_Context* p = (FXBMP_Context*)pContext; 126 _bmp_input_buffer(p->bmp_ptr, (uint8_t*)src_buf, src_size); 127 } 128