1 /* 2 * Copyright 2011 Luc Verhaegen <libv (at) codethink.co.uk> 3 * 4 * Permission is hereby granted, free of charge, to any person obtaining a 5 * copy of this software and associated documentation files (the "Software"), 6 * to deal in the Software without restriction, including without limitation 7 * the rights to use, copy, modify, merge, publish, distribute, sub license, 8 * and/or sell copies of the Software, and to permit persons to whom the 9 * Software is furnished to do so, subject to the following conditions: 10 * 11 * The above copyright notice and this permission notice (including the 12 * next paragraph) shall be included in all copies or substantial portions 13 * of the Software. 14 * 15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL 18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER 21 * DEALINGS IN THE SOFTWARE. 22 * 23 */ 24 /* 25 * Quick 'n Dirty bitmap dumper. 26 */ 27 #include <stdio.h> 28 #include <unistd.h> 29 #include <sys/types.h> 30 #include <sys/stat.h> 31 #include <fcntl.h> 32 #include <string.h> 33 #include <errno.h> 34 35 #include "write_bmp.h" 36 37 #define FILENAME_SIZE 1024 38 39 struct bmp_header { 40 unsigned short magic; 41 unsigned int size; 42 unsigned int unused; 43 unsigned int start; 44 } __attribute__((__packed__)); 45 46 struct dib_header { 47 unsigned int size; 48 unsigned int width; 49 unsigned int height; 50 unsigned short planes; 51 unsigned short bpp; 52 unsigned int compression; 53 unsigned int data_size; 54 unsigned int h_res; 55 unsigned int v_res; 56 unsigned int colours; 57 unsigned int important_colours; 58 unsigned int red_mask; 59 unsigned int green_mask; 60 unsigned int blue_mask; 61 unsigned int alpha_mask; 62 unsigned int colour_space; 63 unsigned int unused[12]; 64 } __attribute__((__packed__)); 65 66 static void 67 bmp_header_write(int fd, int width, int height, int bgra, int noflip, int alpha) 68 { 69 struct bmp_header bmp_header = { 70 .magic = 0x4d42, 71 .size = (width * height * 4) + 72 sizeof(struct bmp_header) + sizeof(struct dib_header), 73 .start = sizeof(struct bmp_header) + sizeof(struct dib_header), 74 }; 75 struct dib_header dib_header = { 76 .size = sizeof(struct dib_header), 77 .width = width, 78 .height = noflip ? -height : height, 79 .planes = 1, 80 .bpp = 32, 81 .compression = 3, 82 .data_size = 4 * width * height, 83 .h_res = 0xB13, 84 .v_res = 0xB13, 85 .colours = 0, 86 .important_colours = 0, 87 .red_mask = 0x000000FF, 88 .green_mask = 0x0000FF00, 89 .blue_mask = 0x00FF0000, 90 .alpha_mask = alpha ? 0xFF000000 : 0x00000000, 91 .colour_space = 0x57696E20, 92 }; 93 94 if (bgra) { 95 dib_header.red_mask = 0x00FF0000; 96 dib_header.blue_mask = 0x000000FF; 97 } 98 99 write(fd, &bmp_header, sizeof(struct bmp_header)); 100 write(fd, &dib_header, sizeof(struct dib_header)); 101 } 102 103 void 104 bmp_dump32(char *buffer, unsigned width, unsigned height, bool bgra, const char *filename) 105 { 106 int fd; 107 108 fd = open(filename, O_WRONLY| O_TRUNC | O_CREAT, 0666); 109 if (fd == -1) { 110 printf("Failed to open %s: %s\n", filename, strerror(errno)); 111 return; 112 } 113 114 bmp_header_write(fd, width, height, bgra, false, true); 115 116 write(fd, buffer, width * height * 4); 117 } 118 119 void 120 bmp_dump32_noflip(char *buffer, unsigned width, unsigned height, bool bgra, const char *filename) 121 { 122 int fd; 123 124 fd = open(filename, O_WRONLY| O_TRUNC | O_CREAT, 0666); 125 if (fd == -1) { 126 printf("Failed to open %s: %s\n", filename, strerror(errno)); 127 return; 128 } 129 130 bmp_header_write(fd, width, height, bgra, true, true); 131 132 write(fd, buffer, width * height * 4); 133 } 134 135 void 136 bmp_dump32_ex(char *buffer, unsigned width, unsigned height, bool flip, bool bgra, bool alpha, const char *filename) 137 { 138 int fd; 139 140 fd = open(filename, O_WRONLY| O_TRUNC | O_CREAT, 0666); 141 if (fd == -1) { 142 printf("Failed to open %s: %s\n", filename, strerror(errno)); 143 return; 144 } 145 146 bmp_header_write(fd, width, height, bgra, flip, alpha); 147 148 write(fd, buffer, width * height * 4); 149 } 150