#include <errno.h>
#include <math.h>
#include <stdio.h>
#include "common.h"
#include "filemem.h"
#include "output.h"
#include "bmp.h"
INTERNAL int zint_bmp_pixel_plot(struct zint_symbol *symbol, const unsigned char *pixelbuf) {
int i, row, column;
int bits_per_pixel;
int colour_count;
int resolution;
size_t row_size, data_offset, file_size;
struct filemem fm;
struct filemem *const fmp = &fm;
bitmap_file_header_t file_header;
bitmap_info_header_t info_header;
color_ref_t bg;
color_ref_t fg;
color_ref_t palette[8];
int ultra_fg_index = 9;
unsigned char map[128];
unsigned char *rowbuf;
(void) zint_out_colour_get_rgb(symbol->fgcolour, &fg.red, &fg.green, &fg.blue, NULL );
fg.reserved = 0x00;
(void) zint_out_colour_get_rgb(symbol->bgcolour, &bg.red, &bg.green, &bg.blue, NULL );
bg.reserved = 0x00;
if (symbol->symbology == BARCODE_ULTRA) {
static const unsigned char ultra_chars[8] = { 'C', 'B', 'M', 'R', 'Y', 'G', 'K', 'W' };
for (i = 0; i < 8; i++) {
zint_out_colour_char_to_rgb(ultra_chars[i], &palette[i].red, &palette[i].green, &palette[i].blue);
palette[i].reserved = 0x00;
if (memcmp(&palette[i], &fg, sizeof(fg)) == 0) {
ultra_fg_index = i + 1;
}
map[ultra_chars[i]] = (unsigned char) (i + 1);
}
bits_per_pixel = 4;
colour_count = ultra_fg_index == 9 ? 10 : 9;
map['0'] = 0;
map['1'] = (unsigned char) ultra_fg_index;
} else {
bits_per_pixel = 1;
colour_count = 2;
map['0'] = 0;
map['1'] = 0x80;
}
row_size = 4 * (((size_t) symbol->bitmap_width * bits_per_pixel + 31) / 32);
data_offset = sizeof(bitmap_file_header_t) + sizeof(bitmap_info_header_t);
data_offset += sizeof(color_ref_t) * colour_count;
file_size = data_offset + row_size * symbol->bitmap_height;
if (file_size != (uint32_t) file_size) {
return z_errtxt(ZINT_ERROR_MEMORY, symbol, 606, "Output size too large for file size field of BMP header");
}
if (!(rowbuf = (unsigned char *) malloc(row_size))) {
return z_errtxt(ZINT_ERROR_MEMORY, symbol, 602, "Insufficient memory for BMP row buffer");
}
zint_out_le_u16(file_header.header_field, 0x4d42);
zint_out_le_u32(file_header.file_size, file_size);
zint_out_le_u32(file_header.reserved, 0);
zint_out_le_u32(file_header.data_offset, data_offset);
zint_out_le_u32(info_header.header_size, sizeof(bitmap_info_header_t));
zint_out_le_i32(info_header.width, symbol->bitmap_width);
zint_out_le_i32(info_header.height, symbol->bitmap_height);
zint_out_le_u16(info_header.colour_planes, 1);
zint_out_le_u16(info_header.bits_per_pixel, bits_per_pixel);
zint_out_le_u32(info_header.compression_method, 0);
zint_out_le_u32(info_header.image_size, 0);
resolution = symbol->dpmm ? (int) roundf(z_stripf(symbol->dpmm * 1000.0f)) : 0;
zint_out_le_i32(info_header.horiz_res, resolution);
zint_out_le_i32(info_header.vert_res, resolution);
zint_out_le_u32(info_header.colours, colour_count);
zint_out_le_u32(info_header.important_colours, colour_count);
if (!zint_fm_open(fmp, symbol, "wb")) {
ZEXT z_errtxtf(0, symbol, 601, "Could not open BMP output file (%1$d: %2$s)", fmp->err, strerror(fmp->err));
free(rowbuf);
return ZINT_ERROR_FILE_ACCESS;
}
zint_fm_write(&file_header, sizeof(bitmap_file_header_t), 1, fmp);
zint_fm_write(&info_header, sizeof(bitmap_info_header_t), 1, fmp);
zint_fm_write(&bg, sizeof(color_ref_t), 1, fmp);
if (bits_per_pixel == 4) {
for (i = 0; i < 8; i++) {
zint_fm_write(&palette[i], sizeof(color_ref_t), 1, fmp);
}
if (ultra_fg_index == 9) {
zint_fm_write(&fg, sizeof(color_ref_t), 1, fmp);
}
} else {
zint_fm_write(&fg, sizeof(color_ref_t), 1, fmp);
}
if (bits_per_pixel == 4) {
for (row = 0; row < symbol->bitmap_height; row++) {
const unsigned char *pb = pixelbuf + ((size_t) symbol->bitmap_width * (symbol->bitmap_height - row - 1));
memset(rowbuf, 0, row_size);
for (column = 0; column < symbol->bitmap_width; column++) {
rowbuf[column >> 1] |= map[pb[column]] << (!(column & 1) << 2);
}
zint_fm_write(rowbuf, 1, row_size, fmp);
}
} else {
for (row = 0; row < symbol->bitmap_height; row++) {
const unsigned char *pb = pixelbuf + ((size_t) symbol->bitmap_width * (symbol->bitmap_height - row - 1));
memset(rowbuf, 0, row_size);
for (column = 0; column < symbol->bitmap_width; column++) {
rowbuf[column >> 3] |= map[pb[column]] >> (column & 7);
}
zint_fm_write(rowbuf, 1, row_size, fmp);
}
}
free(rowbuf);
if (zint_fm_error(fmp)) {
ZEXT z_errtxtf(0, symbol, 603, "Incomplete write of BMP output (%1$d: %2$s)", fmp->err, strerror(fmp->err));
(void) zint_fm_close(fmp, symbol);
return ZINT_ERROR_FILE_WRITE;
}
if (!zint_fm_close(fmp, symbol)) {
return ZEXT z_errtxtf(ZINT_ERROR_FILE_WRITE, symbol, 605, "Failure on closing BMP output file (%1$d: %2$s)",
fmp->err, strerror(fmp->err));
}
return 0;
}