150 lines
4.0 KiB
C
150 lines
4.0 KiB
C
/*
|
|
* Copyright 2011 Luc Verhaegen <libv@codethink.co.uk>
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
* copy of this software and associated documentation files (the "Software"),
|
|
* to deal in the Software without restriction, including without limitation
|
|
* the rights to use, copy, modify, merge, publish, distribute, sub license,
|
|
* and/or sell copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice (including the
|
|
* next paragraph) shall be included in all copies or substantial portions
|
|
* of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
|
|
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
* DEALINGS IN THE SOFTWARE.
|
|
*
|
|
*/
|
|
/*
|
|
* Quick 'n Dirty bitmap dumper.
|
|
*/
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
|
|
#include "write_bmp.h"
|
|
|
|
#define FILENAME_SIZE 1024
|
|
|
|
struct bmp_header {
|
|
unsigned short magic;
|
|
unsigned int size;
|
|
unsigned int unused;
|
|
unsigned int start;
|
|
} __attribute__((__packed__));
|
|
|
|
struct dib_header {
|
|
unsigned int size;
|
|
unsigned int width;
|
|
unsigned int height;
|
|
unsigned short planes;
|
|
unsigned short bpp;
|
|
unsigned int compression;
|
|
unsigned int data_size;
|
|
unsigned int h_res;
|
|
unsigned int v_res;
|
|
unsigned int colours;
|
|
unsigned int important_colours;
|
|
unsigned int red_mask;
|
|
unsigned int green_mask;
|
|
unsigned int blue_mask;
|
|
unsigned int alpha_mask;
|
|
unsigned int colour_space;
|
|
unsigned int unused[12];
|
|
} __attribute__((__packed__));
|
|
|
|
static void
|
|
bmp_header_write(int fd, int width, int height, int bgra, int noflip, int alpha)
|
|
{
|
|
struct bmp_header bmp_header = {
|
|
.magic = 0x4d42,
|
|
.size = (width * height * 4) +
|
|
sizeof(struct bmp_header) + sizeof(struct dib_header),
|
|
.start = sizeof(struct bmp_header) + sizeof(struct dib_header),
|
|
};
|
|
struct dib_header dib_header = {
|
|
.size = sizeof(struct dib_header),
|
|
.width = width,
|
|
.height = noflip ? -height : height,
|
|
.planes = 1,
|
|
.bpp = 32,
|
|
.compression = 3,
|
|
.data_size = 4 * width * height,
|
|
.h_res = 0xB13,
|
|
.v_res = 0xB13,
|
|
.colours = 0,
|
|
.important_colours = 0,
|
|
.red_mask = 0x000000FF,
|
|
.green_mask = 0x0000FF00,
|
|
.blue_mask = 0x00FF0000,
|
|
.alpha_mask = alpha ? 0xFF000000 : 0x00000000,
|
|
.colour_space = 0x57696E20,
|
|
};
|
|
|
|
if (bgra) {
|
|
dib_header.red_mask = 0x00FF0000;
|
|
dib_header.blue_mask = 0x000000FF;
|
|
}
|
|
|
|
write(fd, &bmp_header, sizeof(struct bmp_header));
|
|
write(fd, &dib_header, sizeof(struct dib_header));
|
|
}
|
|
|
|
void
|
|
bmp_dump32(char *buffer, unsigned width, unsigned height, bool bgra, const char *filename)
|
|
{
|
|
int fd;
|
|
|
|
fd = open(filename, O_WRONLY| O_TRUNC | O_CREAT, 0666);
|
|
if (fd == -1) {
|
|
printf("Failed to open %s: %s\n", filename, strerror(errno));
|
|
return;
|
|
}
|
|
|
|
bmp_header_write(fd, width, height, bgra, false, true);
|
|
|
|
write(fd, buffer, width * height * 4);
|
|
}
|
|
|
|
void
|
|
bmp_dump32_noflip(char *buffer, unsigned width, unsigned height, bool bgra, const char *filename)
|
|
{
|
|
int fd;
|
|
|
|
fd = open(filename, O_WRONLY| O_TRUNC | O_CREAT, 0666);
|
|
if (fd == -1) {
|
|
printf("Failed to open %s: %s\n", filename, strerror(errno));
|
|
return;
|
|
}
|
|
|
|
bmp_header_write(fd, width, height, bgra, true, true);
|
|
|
|
write(fd, buffer, width * height * 4);
|
|
}
|
|
|
|
void
|
|
bmp_dump32_ex(char *buffer, unsigned width, unsigned height, bool flip, bool bgra, bool alpha, const char *filename)
|
|
{
|
|
int fd;
|
|
|
|
fd = open(filename, O_WRONLY| O_TRUNC | O_CREAT, 0666);
|
|
if (fd == -1) {
|
|
printf("Failed to open %s: %s\n", filename, strerror(errno));
|
|
return;
|
|
}
|
|
|
|
bmp_header_write(fd, width, height, bgra, flip, alpha);
|
|
|
|
write(fd, buffer, width * height * 4);
|
|
}
|