544 lines
12 KiB
C
544 lines
12 KiB
C
#include <rtthread.h>
|
|
#include <rtgui/dc_hw.h>
|
|
#include <rtgui/image.h>
|
|
#include <rtgui/rtgui_system.h>
|
|
#include <rtgui/image_bmp.h>
|
|
#include <rtgui/blit.h>
|
|
|
|
/* Compression encodings for BMP files */
|
|
#ifndef BI_RGB
|
|
#define BI_RGB 0
|
|
#define BI_RLE8 1
|
|
#define BI_RLE4 2
|
|
#define BI_BITFIELDS 3
|
|
#endif
|
|
|
|
#define hw_driver (rtgui_graphic_driver_get_default())
|
|
|
|
struct rtgui_image_bmp
|
|
{
|
|
rt_bool_t is_loaded;
|
|
|
|
rt_uint32_t Rmask;
|
|
rt_uint32_t Gmask;
|
|
rt_uint32_t Bmask;
|
|
|
|
rt_size_t pixel_offset;
|
|
|
|
rt_uint8_t byte_per_pixel;
|
|
rt_uint8_t pad;
|
|
rt_uint8_t ExpandBMP;
|
|
|
|
rt_uint8_t *pixels;
|
|
rt_uint32_t pitch;
|
|
|
|
struct rtgui_filerw* filerw;
|
|
};
|
|
|
|
struct rtgui_image_bmp_header
|
|
{
|
|
/* The Win32 BMP file header (14 bytes) */
|
|
char magic[2];
|
|
rt_uint32_t bfSize;
|
|
rt_uint16_t bfReserved1;
|
|
rt_uint16_t bfReserved2;
|
|
rt_uint32_t bfOffBits;
|
|
|
|
/* The Win32 BITMAPINFOHEADER struct (40 bytes) */
|
|
rt_uint32_t biSize;
|
|
rt_int32_t biWidth;
|
|
rt_int32_t biHeight;
|
|
rt_uint16_t biPlanes;
|
|
rt_uint16_t biBitCount;
|
|
rt_uint32_t biCompression;
|
|
rt_uint32_t biSizeImage;
|
|
rt_int32_t biXPelsPerMeter;
|
|
rt_int32_t biYPelsPerMeter;
|
|
rt_uint32_t biClrUsed;
|
|
rt_uint32_t biClrImportant;
|
|
};
|
|
|
|
static rt_bool_t rtgui_image_bmp_check(struct rtgui_filerw* file);
|
|
static rt_bool_t rtgui_image_bmp_load(struct rtgui_image* image, struct rtgui_filerw* file, rt_bool_t load);
|
|
static void rtgui_image_bmp_unload(struct rtgui_image* image);
|
|
static void rtgui_image_bmp_blit(struct rtgui_image* image, struct rtgui_dc* dc, struct rtgui_rect* rect);
|
|
|
|
struct rtgui_image_engine rtgui_image_bmp_engine =
|
|
{
|
|
"bmp",
|
|
{ RT_NULL },
|
|
rtgui_image_bmp_check,
|
|
rtgui_image_bmp_load,
|
|
rtgui_image_bmp_unload,
|
|
rtgui_image_bmp_blit
|
|
};
|
|
|
|
rt_inline rt_uint32_t rtgui_filerw_read32(struct rtgui_filerw* src)
|
|
{
|
|
rt_uint32_t value;
|
|
rtgui_filerw_read(src, &value, (sizeof value), 1);
|
|
return value;
|
|
}
|
|
|
|
rt_inline rt_uint16_t rtgui_filerw_read16(struct rtgui_filerw* src)
|
|
{
|
|
rt_uint16_t value;
|
|
rtgui_filerw_read(src, &value, (sizeof value), 1);
|
|
return value;
|
|
|
|
}
|
|
|
|
static rt_bool_t rtgui_image_bmp_check(struct rtgui_filerw* file)
|
|
{
|
|
char magic[2];
|
|
rt_bool_t is_bmp;
|
|
rt_off_t start;
|
|
|
|
if ( !file ) return 0;
|
|
|
|
start = rtgui_filerw_tell(file);
|
|
|
|
/* move to the beginning of file */
|
|
rtgui_filerw_seek(file, 0, RTGUI_FILE_SEEK_SET);
|
|
|
|
is_bmp = RT_FALSE;
|
|
if ( rtgui_filerw_read(file, magic, 1, sizeof(magic)) == sizeof(magic) )
|
|
{
|
|
if (magic[0] == 'B' &&
|
|
magic[1] == 'M')
|
|
{
|
|
is_bmp = RT_TRUE;
|
|
}
|
|
}
|
|
|
|
rtgui_filerw_seek(file, start, RTGUI_FILE_SEEK_SET);
|
|
|
|
return(is_bmp);
|
|
}
|
|
|
|
static struct rtgui_image_palette* rtgui_image_bmp_load_palette(struct rtgui_image_bmp_header* header, struct rtgui_filerw* src)
|
|
{
|
|
/* Load the palette, if any */
|
|
rt_size_t i;
|
|
struct rtgui_image_palette* palette;
|
|
|
|
if (header->biClrUsed == 0)
|
|
header->biClrUsed = 1 << header->biBitCount;
|
|
|
|
palette = rtgui_image_palette_create(header->biClrUsed);
|
|
if (palette == RT_NULL) return RT_NULL;
|
|
|
|
if ( header->biSize == 12 )
|
|
{
|
|
rt_uint8_t r, g, b;
|
|
for ( i = 0; i < (int)header->biClrUsed; ++i )
|
|
{
|
|
rtgui_filerw_read(src, &b, 1, 1);
|
|
rtgui_filerw_read(src, &g, 1, 1);
|
|
rtgui_filerw_read(src, &r, 1, 1);
|
|
palette->colors[i] = RTGUI_RGB(r, g, b);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rt_uint8_t r, g, b, a;
|
|
for ( i = 0; i < (int)header->biClrUsed; ++i )
|
|
{
|
|
rtgui_filerw_read(src, &b, 1, 1);
|
|
rtgui_filerw_read(src, &g, 1, 1);
|
|
rtgui_filerw_read(src, &r, 1, 1);
|
|
rtgui_filerw_read(src, &a, 1, 1);
|
|
palette->colors[i] = RTGUI_ARGB(a, r, g, b);
|
|
}
|
|
}
|
|
|
|
return palette;
|
|
}
|
|
|
|
static rt_bool_t rtgui_image_bmp_load(struct rtgui_image* image, struct rtgui_filerw* src, rt_bool_t load)
|
|
{
|
|
rt_uint32_t Rmask;
|
|
rt_uint32_t Gmask;
|
|
rt_uint32_t Bmask;
|
|
int ExpandBMP, bmpPitch;
|
|
struct rtgui_image_bmp* bmp;
|
|
struct rtgui_image_bmp_header* header;
|
|
struct rtgui_image_palette* palette;
|
|
|
|
bmp = RT_NULL;
|
|
header = RT_NULL;
|
|
palette = RT_NULL;
|
|
header = (struct rtgui_image_bmp_header*) rtgui_malloc(sizeof(struct rtgui_image_bmp_header));
|
|
if (header == RT_NULL) return RT_FALSE;
|
|
|
|
if ( rtgui_filerw_read(src, header->magic, 1, sizeof(header->magic)) ==
|
|
sizeof(sizeof(header->magic)) )
|
|
{
|
|
if (header->magic[0] != 'B' ||
|
|
header->magic[1] != 'M')
|
|
{
|
|
goto __exit;
|
|
}
|
|
}
|
|
|
|
header->bfSize = rtgui_filerw_read32(src);
|
|
header->bfReserved1 = rtgui_filerw_read16(src);
|
|
header->bfReserved2 = rtgui_filerw_read16(src);
|
|
header->bfOffBits = rtgui_filerw_read32(src);
|
|
|
|
/* Read the Win32 BITMAPINFOHEADER */
|
|
header->biSize = rtgui_filerw_read32(src);
|
|
if ( header->biSize == 12 )
|
|
{
|
|
header->biWidth = (rt_uint32_t)rtgui_filerw_read16(src);
|
|
header->biHeight = (rt_uint32_t)rtgui_filerw_read16(src);
|
|
header->biPlanes = rtgui_filerw_read16(src);
|
|
header->biBitCount = rtgui_filerw_read16(src);
|
|
header->biCompression = BI_RGB;
|
|
header->biSizeImage = 0;
|
|
header->biXPelsPerMeter = 0;
|
|
header->biYPelsPerMeter = 0;
|
|
header->biClrUsed = 0;
|
|
|
|
header->biClrImportant = 0;
|
|
|
|
}
|
|
else
|
|
{
|
|
header->biWidth = rtgui_filerw_read32(src);
|
|
header->biHeight = rtgui_filerw_read32(src);
|
|
header->biPlanes = rtgui_filerw_read16(src);
|
|
header->biBitCount = rtgui_filerw_read16(src);
|
|
header->biCompression = rtgui_filerw_read32(src);
|
|
header->biSizeImage = rtgui_filerw_read32(src);
|
|
header->biXPelsPerMeter = rtgui_filerw_read32(src);
|
|
header->biYPelsPerMeter = rtgui_filerw_read32(src);
|
|
header->biClrUsed = rtgui_filerw_read32(src);
|
|
header->biClrImportant = rtgui_filerw_read32(src);
|
|
|
|
}
|
|
|
|
/* allocate palette and expand 1 and 4 bit bitmaps to 8 bits per pixel */
|
|
switch (header->biBitCount)
|
|
{
|
|
case 1:
|
|
ExpandBMP = header->biBitCount;
|
|
palette = rtgui_image_bmp_load_palette(header, src);
|
|
header->biBitCount = 8;
|
|
break;
|
|
|
|
case 4:
|
|
ExpandBMP = header->biBitCount;
|
|
palette = rtgui_image_bmp_load_palette(header, src);
|
|
header->biBitCount = 8;
|
|
break;
|
|
|
|
case 8:
|
|
palette = rtgui_image_bmp_load_palette(header, src);
|
|
ExpandBMP = 0;
|
|
break;
|
|
|
|
default:
|
|
ExpandBMP = 0;
|
|
break;
|
|
}
|
|
|
|
/* We don't support any BMP compression right now */
|
|
Rmask = Gmask = Bmask = 0;
|
|
switch (header->biCompression)
|
|
{
|
|
case BI_RGB:
|
|
/* If there are no masks, use the defaults */
|
|
if ( header->bfOffBits == (14 + header->biSize) )
|
|
{
|
|
/* Default values for the BMP format */
|
|
switch (header->biBitCount)
|
|
{
|
|
case 15:
|
|
case 16:
|
|
Rmask = 0x7C00;
|
|
Gmask = 0x03E0;
|
|
Bmask = 0x001F;
|
|
break;
|
|
|
|
case 24:
|
|
case 32:
|
|
Rmask = 0x00FF0000;
|
|
Gmask = 0x0000FF00;
|
|
Bmask = 0x000000FF;
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
/* Fall through -- read the RGB masks */
|
|
|
|
case BI_BITFIELDS:
|
|
switch (header->biBitCount)
|
|
{
|
|
case 15:
|
|
case 16:
|
|
case 32:
|
|
Rmask = rtgui_filerw_read32(src);
|
|
Gmask = rtgui_filerw_read32(src);
|
|
Bmask = rtgui_filerw_read32(src);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
rt_kprintf("Compressed BMP files not supported\n");
|
|
goto __exit;
|
|
}
|
|
|
|
bmp = (struct rtgui_image_bmp*) rtgui_malloc(sizeof(struct rtgui_image_bmp));
|
|
if (bmp == RT_NULL)
|
|
goto __exit;
|
|
|
|
/* set image information */
|
|
image->w = header->biWidth;
|
|
image->h = header->biHeight;
|
|
image->engine = &rtgui_image_bmp_engine;
|
|
image->data = bmp;
|
|
bmp->filerw = src;
|
|
bmp->byte_per_pixel = header->biBitCount/8;
|
|
bmp->pitch = image->w * bmp->byte_per_pixel;
|
|
bmp->pixel_offset = header->bfOffBits;
|
|
bmp->Rmask = Rmask; bmp->Gmask = Gmask; bmp->Bmask = Bmask;
|
|
bmp->ExpandBMP = ExpandBMP;
|
|
if (palette != RT_NULL) image->palette = palette;
|
|
|
|
/* get padding */
|
|
switch (ExpandBMP)
|
|
{
|
|
case 1:
|
|
bmpPitch = (header->biWidth + 7) >> 3;
|
|
bmp->pad = (((bmpPitch)%4) ? (4-((bmpPitch)%4)) : 0);
|
|
break;
|
|
|
|
case 4:
|
|
bmpPitch = (header->biWidth + 1) >> 1;
|
|
bmp->pad = (((bmpPitch)%4) ? (4-((bmpPitch)%4)) : 0);
|
|
break;
|
|
|
|
default:
|
|
bmp->pad = ((bmp->pitch%4) ? (4-(bmp->pitch%4)) : 0);
|
|
break;
|
|
}
|
|
|
|
if (load == RT_TRUE)
|
|
{
|
|
rt_uint8_t *bits;
|
|
rt_uint32_t i;
|
|
|
|
/* load all pixels */
|
|
bmp->pixels = rtgui_malloc(image->h * bmp->pitch);
|
|
if (bmp->pixels == RT_NULL)
|
|
goto __exit;
|
|
|
|
/* Read the pixels. Note that the bmp image is upside down */
|
|
if ( rtgui_filerw_seek(src, bmp->pixel_offset, RTGUI_FILE_SEEK_SET) < 0)
|
|
goto __exit;
|
|
|
|
bits = bmp->pixels + image->h * bmp->pitch;
|
|
while ( bits > bmp->pixels )
|
|
{
|
|
bits -= bmp->pitch;
|
|
switch (ExpandBMP)
|
|
{
|
|
case 1:
|
|
case 4:
|
|
{
|
|
rt_uint8_t pixel = 0;
|
|
int shift = (8 - ExpandBMP);
|
|
for ( i=0; i < image->w; ++i )
|
|
{
|
|
if ( i % (8/ExpandBMP) == 0 )
|
|
{
|
|
if ( !rtgui_filerw_read(src, &pixel, 1, 1) )
|
|
goto __exit;
|
|
|
|
}
|
|
*(bits+i) = (pixel>>shift);
|
|
pixel <<= ExpandBMP;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
if ( rtgui_filerw_read(src, bits, 1, bmp->pitch) != bmp->pitch ) goto __exit;
|
|
break;
|
|
}
|
|
|
|
/* Skip padding bytes */
|
|
if ( bmp->pad )
|
|
{
|
|
rt_uint8_t padbyte;
|
|
for ( i=0; i < bmp->pad; ++i )
|
|
{
|
|
rtgui_filerw_read(src, &padbyte, 1, 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
rtgui_filerw_close(bmp->filerw);
|
|
bmp->filerw = RT_NULL;
|
|
bmp->pixel_offset = 0;
|
|
}
|
|
else
|
|
{
|
|
bmp->pixels = RT_NULL;
|
|
}
|
|
|
|
return RT_TRUE;
|
|
|
|
__exit:
|
|
rtgui_free(header);
|
|
if (palette != RT_NULL)
|
|
{
|
|
rtgui_free(palette);
|
|
image->palette = RT_NULL;
|
|
}
|
|
|
|
if (bmp != RT_NULL)
|
|
rtgui_free(bmp->pixels);
|
|
rtgui_free(bmp);
|
|
return RT_FALSE;
|
|
}
|
|
|
|
static void rtgui_image_bmp_unload(struct rtgui_image* image)
|
|
{
|
|
struct rtgui_image_bmp* bmp;
|
|
|
|
if (image != RT_NULL)
|
|
{
|
|
bmp = (struct rtgui_image_bmp*) image->data;
|
|
|
|
if (bmp->pixels != RT_NULL) rtgui_free(bmp->pixels);
|
|
if (bmp->filerw != RT_NULL)
|
|
{
|
|
rtgui_filerw_close(bmp->filerw);
|
|
bmp->filerw = RT_NULL;
|
|
}
|
|
|
|
/* release data */
|
|
rtgui_free(bmp);
|
|
}
|
|
}
|
|
|
|
static void rtgui_image_bmp_blit(struct rtgui_image* image, struct rtgui_dc* dc, struct rtgui_rect* dst_rect)
|
|
{
|
|
rt_uint16_t y, w, h;
|
|
struct rtgui_image_bmp* bmp;
|
|
|
|
RT_ASSERT(image != RT_NULL || dc != RT_NULL || dst_rect != RT_NULL);
|
|
|
|
/* this dc is not visible */
|
|
if (rtgui_dc_get_visible(dc) != RT_TRUE) return;
|
|
|
|
bmp = (struct rtgui_image_bmp*) image->data;
|
|
RT_ASSERT(bmp != RT_NULL);
|
|
|
|
/* the minimum rect */
|
|
if (image->w < rtgui_rect_width(*dst_rect)) w = image->w;
|
|
else w = rtgui_rect_width(*dst_rect);
|
|
if (image->h < rtgui_rect_height(*dst_rect)) h = image->h;
|
|
else h = rtgui_rect_height(*dst_rect);
|
|
|
|
if (bmp->pixels != RT_NULL)
|
|
{
|
|
rt_uint8_t* ptr;
|
|
|
|
/* get pixel pointer */
|
|
ptr = bmp->pixels;
|
|
if ((dc->type == RTGUI_DC_HW) || (dc->type == RTGUI_DC_CLIENT))
|
|
{
|
|
if (bmp->byte_per_pixel == hw_driver->byte_per_pixel)
|
|
{
|
|
for (y = 0; y < h; y ++)
|
|
{
|
|
dc->engine->blit_line(dc,
|
|
dst_rect->x1, dst_rect->x1 + w,
|
|
dst_rect->y1 + y,
|
|
ptr);
|
|
ptr += bmp->pitch;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rt_size_t pitch;
|
|
rt_uint8_t *line_ptr;
|
|
|
|
if (image->palette == RT_NULL)
|
|
{
|
|
rtgui_blit_line_func blit_line;
|
|
line_ptr = (rt_uint8_t*) rtgui_malloc(hw_driver->byte_per_pixel * w);
|
|
blit_line = rtgui_blit_line_get(hw_driver->byte_per_pixel , bmp->byte_per_pixel);
|
|
pitch = w * bmp->byte_per_pixel;
|
|
if (line_ptr != RT_NULL)
|
|
{
|
|
for (y = 0; y < h; y ++)
|
|
{
|
|
blit_line(line_ptr, ptr, pitch);
|
|
dc->engine->blit_line(dc,
|
|
dst_rect->x1, dst_rect->x1 + w,
|
|
dst_rect->y1 + y,
|
|
line_ptr);
|
|
ptr += bmp->pitch;
|
|
}
|
|
}
|
|
rtgui_free(line_ptr);
|
|
}
|
|
else
|
|
{
|
|
int x, p;
|
|
rtgui_color_t color;
|
|
|
|
/* use palette */
|
|
for (y = 0; y < h; y ++)
|
|
{
|
|
ptr = bmp->pixels + (y * bmp->pitch);
|
|
for (x = 0; x < w; x ++)
|
|
{
|
|
p = *ptr;
|
|
color = image->palette->colors[*ptr]; ptr ++;
|
|
rtgui_dc_draw_color_point(dc, dst_rect->x1 + x, dst_rect->y1 + y, color);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rt_uint8_t* ptr;
|
|
ptr = rtgui_malloc(bmp->pitch);
|
|
if (ptr == RT_NULL) return; /* no memory */
|
|
|
|
/* seek to the begin of pixel data */
|
|
rtgui_filerw_seek(bmp->filerw, bmp->pixel_offset, RTGUI_FILE_SEEK_SET);
|
|
|
|
for (y = 0; y < h; y ++)
|
|
{
|
|
/* read pixel data */
|
|
if (rtgui_filerw_read(bmp->filerw, ptr, 1, bmp->pitch) != bmp->pitch)
|
|
break; /* read data failed */
|
|
|
|
dc->engine->blit_line(dc, dst_rect->x1, dst_rect->x1 + w, dst_rect->y1 + y, ptr);
|
|
}
|
|
|
|
rtgui_free(ptr);
|
|
}
|
|
}
|
|
|
|
void rtgui_image_bmp_init()
|
|
{
|
|
/* register bmp on image system */
|
|
rtgui_image_register_engine(&rtgui_image_bmp_engine);
|
|
}
|