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/fxcodec/codec/ccodec_bmpmodule.h"
8
9 #include "core/fxcodec/codec/codec_int.h"
10 #include "core/fxcodec/fx_codec.h"
11 #include "core/fxcodec/lbmp/fx_bmp.h"
12 #include "core/fxge/fx_dib.h"
13
14 struct FXBMP_Context {
15 bmp_decompress_struct_p bmp_ptr;
16 void* parent_ptr;
17
18 void* (*m_AllocFunc)(unsigned int);
19 void (*m_FreeFunc)(void*);
20 };
21 extern "C" {
bmp_alloc_func(unsigned int size)22 static void* bmp_alloc_func(unsigned int size) {
23 return FX_Alloc(char, size);
24 }
bmp_free_func(void * p)25 static void bmp_free_func(void* p) {
26 FX_Free(p);
27 }
28 };
bmp_error_data(bmp_decompress_struct_p bmp_ptr,const FX_CHAR * err_msg)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 }
bmp_read_scanline(bmp_decompress_struct_p bmp_ptr,int32_t row_num,uint8_t * row_buf)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->GetDelegate()->BmpReadScanline(row_num, row_buf);
40 }
bmp_get_data_position(bmp_decompress_struct_p bmp_ptr,uint32_t rcd_pos)41 static bool bmp_get_data_position(bmp_decompress_struct_p bmp_ptr,
42 uint32_t rcd_pos) {
43 FXBMP_Context* p = (FXBMP_Context*)bmp_ptr->context_ptr;
44 CCodec_BmpModule* pModule = (CCodec_BmpModule*)p->parent_ptr;
45 return pModule->GetDelegate()->BmpInputImagePositionBuf(rcd_pos);
46 }
47
CCodec_BmpModule()48 CCodec_BmpModule::CCodec_BmpModule() {
49 memset(m_szLastError, 0, sizeof(m_szLastError));
50 }
51
~CCodec_BmpModule()52 CCodec_BmpModule::~CCodec_BmpModule() {}
53
Start()54 FXBMP_Context* CCodec_BmpModule::Start() {
55 FXBMP_Context* p = FX_Alloc(FXBMP_Context, 1);
56 if (!p)
57 return nullptr;
58
59 FXSYS_memset(p, 0, sizeof(FXBMP_Context));
60 if (!p)
61 return nullptr;
62
63 p->m_AllocFunc = bmp_alloc_func;
64 p->m_FreeFunc = bmp_free_func;
65 p->bmp_ptr = nullptr;
66 p->parent_ptr = (void*)this;
67 p->bmp_ptr = bmp_create_decompress();
68 if (!p->bmp_ptr) {
69 FX_Free(p);
70 return nullptr;
71 }
72 p->bmp_ptr->context_ptr = (void*)p;
73 p->bmp_ptr->err_ptr = m_szLastError;
74 p->bmp_ptr->bmp_error_fn = bmp_error_data;
75 p->bmp_ptr->bmp_get_row_fn = bmp_read_scanline;
76 p->bmp_ptr->bmp_get_data_position_fn = bmp_get_data_position;
77 return p;
78 }
79
Finish(FXBMP_Context * ctx)80 void CCodec_BmpModule::Finish(FXBMP_Context* ctx) {
81 if (ctx) {
82 bmp_destroy_decompress(&ctx->bmp_ptr);
83 ctx->m_FreeFunc(ctx);
84 }
85 }
ReadHeader(FXBMP_Context * ctx,int32_t * width,int32_t * height,bool * tb_flag,int32_t * components,int32_t * pal_num,uint32_t ** pal_pp,CFX_DIBAttribute * pAttribute)86 int32_t CCodec_BmpModule::ReadHeader(FXBMP_Context* ctx,
87 int32_t* width,
88 int32_t* height,
89 bool* tb_flag,
90 int32_t* components,
91 int32_t* pal_num,
92 uint32_t** pal_pp,
93 CFX_DIBAttribute* pAttribute) {
94 if (setjmp(ctx->bmp_ptr->jmpbuf)) {
95 return 0;
96 }
97 int32_t ret = bmp_read_header(ctx->bmp_ptr);
98 if (ret != 1) {
99 return ret;
100 }
101 *width = ctx->bmp_ptr->width;
102 *height = ctx->bmp_ptr->height;
103 *tb_flag = ctx->bmp_ptr->imgTB_flag;
104 *components = ctx->bmp_ptr->components;
105 *pal_num = ctx->bmp_ptr->pal_num;
106 *pal_pp = ctx->bmp_ptr->pal_ptr;
107 if (pAttribute) {
108 pAttribute->m_wDPIUnit = FXCODEC_RESUNIT_METER;
109 pAttribute->m_nXDPI = ctx->bmp_ptr->dpi_x;
110 pAttribute->m_nYDPI = ctx->bmp_ptr->dpi_y;
111 pAttribute->m_nBmpCompressType = ctx->bmp_ptr->compress_flag;
112 }
113 return 1;
114 }
115
LoadImage(FXBMP_Context * ctx)116 int32_t CCodec_BmpModule::LoadImage(FXBMP_Context* ctx) {
117 if (setjmp(ctx->bmp_ptr->jmpbuf))
118 return 0;
119 return bmp_decode_image(ctx->bmp_ptr);
120 }
121
GetAvailInput(FXBMP_Context * ctx,uint8_t ** avail_buf_ptr)122 uint32_t CCodec_BmpModule::GetAvailInput(FXBMP_Context* ctx,
123 uint8_t** avail_buf_ptr) {
124 return bmp_get_avail_input(ctx->bmp_ptr, avail_buf_ptr);
125 }
126
Input(FXBMP_Context * ctx,const uint8_t * src_buf,uint32_t src_size)127 void CCodec_BmpModule::Input(FXBMP_Context* ctx,
128 const uint8_t* src_buf,
129 uint32_t src_size) {
130 bmp_input_buffer(ctx->bmp_ptr, (uint8_t*)src_buf, src_size);
131 }
132