diff options
| author | 3gg <3gg@shellblade.net> | 2025-12-27 12:03:39 -0800 |
|---|---|---|
| committer | 3gg <3gg@shellblade.net> | 2025-12-27 12:03:39 -0800 |
| commit | 5a079a2d114f96d4847d1ee305d5b7c16eeec50e (patch) | |
| tree | 8926ab44f168acf787d8e19608857b3af0f82758 /contrib/SDL-3.2.8/src/video/SDL_bmp.c | |
Initial commit
Diffstat (limited to 'contrib/SDL-3.2.8/src/video/SDL_bmp.c')
| -rw-r--r-- | contrib/SDL-3.2.8/src/video/SDL_bmp.c | 879 |
1 files changed, 879 insertions, 0 deletions
diff --git a/contrib/SDL-3.2.8/src/video/SDL_bmp.c b/contrib/SDL-3.2.8/src/video/SDL_bmp.c new file mode 100644 index 0000000..688086f --- /dev/null +++ b/contrib/SDL-3.2.8/src/video/SDL_bmp.c | |||
| @@ -0,0 +1,879 @@ | |||
| 1 | /* | ||
| 2 | Simple DirectMedia Layer | ||
| 3 | Copyright (C) 1997-2025 Sam Lantinga <slouken@libsdl.org> | ||
| 4 | |||
| 5 | This software is provided 'as-is', without any express or implied | ||
| 6 | warranty. In no event will the authors be held liable for any damages | ||
| 7 | arising from the use of this software. | ||
| 8 | |||
| 9 | Permission is granted to anyone to use this software for any purpose, | ||
| 10 | including commercial applications, and to alter it and redistribute it | ||
| 11 | freely, subject to the following restrictions: | ||
| 12 | |||
| 13 | 1. The origin of this software must not be misrepresented; you must not | ||
| 14 | claim that you wrote the original software. If you use this software | ||
| 15 | in a product, an acknowledgment in the product documentation would be | ||
| 16 | appreciated but is not required. | ||
| 17 | 2. Altered source versions must be plainly marked as such, and must not be | ||
| 18 | misrepresented as being the original software. | ||
| 19 | 3. This notice may not be removed or altered from any source distribution. | ||
| 20 | */ | ||
| 21 | #include "SDL_internal.h" | ||
| 22 | |||
| 23 | /* | ||
| 24 | Code to load and save surfaces in Windows BMP format. | ||
| 25 | |||
| 26 | Why support BMP format? Well, it's a native format for Windows, and | ||
| 27 | most image processing programs can read and write it. It would be nice | ||
| 28 | to be able to have at least one image format that we can natively load | ||
| 29 | and save, and since PNG is so complex that it would bloat the library, | ||
| 30 | BMP is a good alternative. | ||
| 31 | |||
| 32 | This code currently supports Win32 DIBs in uncompressed 8 and 24 bpp. | ||
| 33 | */ | ||
| 34 | |||
| 35 | #include "SDL_pixels_c.h" | ||
| 36 | #include "SDL_surface_c.h" | ||
| 37 | |||
| 38 | #define SAVE_32BIT_BMP | ||
| 39 | |||
| 40 | // Compression encodings for BMP files | ||
| 41 | #ifndef BI_RGB | ||
| 42 | #define BI_RGB 0 | ||
| 43 | #define BI_RLE8 1 | ||
| 44 | #define BI_RLE4 2 | ||
| 45 | #define BI_BITFIELDS 3 | ||
| 46 | #endif | ||
| 47 | |||
| 48 | // Logical color space values for BMP files | ||
| 49 | // https://learn.microsoft.com/en-us/openspecs/windows_protocols/ms-wmf/eb4bbd50-b3ce-4917-895c-be31f214797f | ||
| 50 | #ifndef LCS_WINDOWS_COLOR_SPACE | ||
| 51 | // 0x57696E20 == "Win " | ||
| 52 | #define LCS_WINDOWS_COLOR_SPACE 0x57696E20 | ||
| 53 | #endif | ||
| 54 | |||
| 55 | #ifndef LCS_sRGB | ||
| 56 | // 0x73524742 == "sRGB" | ||
| 57 | #define LCS_sRGB 0x73524742 | ||
| 58 | #endif | ||
| 59 | |||
| 60 | // Logical/physical color relationship | ||
| 61 | // https://learn.microsoft.com/en-us/openspecs/windows_protocols/ms-wmf/9fec0834-607d-427d-abd5-ab240fb0db38 | ||
| 62 | #ifndef LCS_GM_GRAPHICS | ||
| 63 | #define LCS_GM_GRAPHICS 0x00000002 | ||
| 64 | #endif | ||
| 65 | |||
| 66 | static bool readRlePixels(SDL_Surface *surface, SDL_IOStream *src, int isRle8) | ||
| 67 | { | ||
| 68 | /* | ||
| 69 | | Sets the surface pixels from src. A bmp image is upside down. | ||
| 70 | */ | ||
| 71 | int pitch = surface->pitch; | ||
| 72 | int height = surface->h; | ||
| 73 | Uint8 *start = (Uint8 *)surface->pixels; | ||
| 74 | Uint8 *end = start + (height * pitch); | ||
| 75 | Uint8 *bits = end - pitch, *spot; | ||
| 76 | int ofs = 0; | ||
| 77 | Uint8 ch; | ||
| 78 | Uint8 needsPad; | ||
| 79 | const int pixels_per_byte = (isRle8 ? 1 : 2); | ||
| 80 | |||
| 81 | #define COPY_PIXEL(x) \ | ||
| 82 | spot = &bits[ofs++]; \ | ||
| 83 | if (spot >= start && spot < end) \ | ||
| 84 | *spot = (x) | ||
| 85 | |||
| 86 | for (;;) { | ||
| 87 | if (!SDL_ReadU8(src, &ch)) { | ||
| 88 | return false; | ||
| 89 | } | ||
| 90 | /* | ||
| 91 | | encoded mode starts with a run length, and then a byte | ||
| 92 | | with two colour indexes to alternate between for the run | ||
| 93 | */ | ||
| 94 | if (ch) { | ||
| 95 | Uint8 pixel; | ||
| 96 | if (!SDL_ReadU8(src, &pixel)) { | ||
| 97 | return false; | ||
| 98 | } | ||
| 99 | ch /= pixels_per_byte; | ||
| 100 | do { | ||
| 101 | COPY_PIXEL(pixel); | ||
| 102 | } while (--ch); | ||
| 103 | } else { | ||
| 104 | /* | ||
| 105 | | A leading zero is an escape; it may signal the end of the bitmap, | ||
| 106 | | a cursor move, or some absolute data. | ||
| 107 | | zero tag may be absolute mode or an escape | ||
| 108 | */ | ||
| 109 | if (!SDL_ReadU8(src, &ch)) { | ||
| 110 | return false; | ||
| 111 | } | ||
| 112 | switch (ch) { | ||
| 113 | case 0: // end of line | ||
| 114 | ofs = 0; | ||
| 115 | bits -= pitch; // go to previous | ||
| 116 | break; | ||
| 117 | case 1: // end of bitmap | ||
| 118 | return true; // success! | ||
| 119 | case 2: // delta | ||
| 120 | if (!SDL_ReadU8(src, &ch)) { | ||
| 121 | return false; | ||
| 122 | } | ||
| 123 | ofs += ch / pixels_per_byte; | ||
| 124 | |||
| 125 | if (!SDL_ReadU8(src, &ch)) { | ||
| 126 | return false; | ||
| 127 | } | ||
| 128 | bits -= ((ch / pixels_per_byte) * pitch); | ||
| 129 | break; | ||
| 130 | default: // no compression | ||
| 131 | ch /= pixels_per_byte; | ||
| 132 | needsPad = (ch & 1); | ||
| 133 | do { | ||
| 134 | Uint8 pixel; | ||
| 135 | if (!SDL_ReadU8(src, &pixel)) { | ||
| 136 | return false; | ||
| 137 | } | ||
| 138 | COPY_PIXEL(pixel); | ||
| 139 | } while (--ch); | ||
| 140 | |||
| 141 | // pad at even boundary | ||
| 142 | if (needsPad && !SDL_ReadU8(src, &ch)) { | ||
| 143 | return false; | ||
| 144 | } | ||
| 145 | break; | ||
| 146 | } | ||
| 147 | } | ||
| 148 | } | ||
| 149 | } | ||
| 150 | |||
| 151 | static void CorrectAlphaChannel(SDL_Surface *surface) | ||
| 152 | { | ||
| 153 | // Check to see if there is any alpha channel data | ||
| 154 | bool hasAlpha = false; | ||
| 155 | #if SDL_BYTEORDER == SDL_BIG_ENDIAN | ||
| 156 | int alphaChannelOffset = 0; | ||
| 157 | #else | ||
| 158 | int alphaChannelOffset = 3; | ||
| 159 | #endif | ||
| 160 | Uint8 *alpha = ((Uint8 *)surface->pixels) + alphaChannelOffset; | ||
| 161 | Uint8 *end = alpha + surface->h * surface->pitch; | ||
| 162 | |||
| 163 | while (alpha < end) { | ||
| 164 | if (*alpha != 0) { | ||
| 165 | hasAlpha = true; | ||
| 166 | break; | ||
| 167 | } | ||
| 168 | alpha += 4; | ||
| 169 | } | ||
| 170 | |||
| 171 | if (!hasAlpha) { | ||
| 172 | alpha = ((Uint8 *)surface->pixels) + alphaChannelOffset; | ||
| 173 | while (alpha < end) { | ||
| 174 | *alpha = SDL_ALPHA_OPAQUE; | ||
| 175 | alpha += 4; | ||
| 176 | } | ||
| 177 | } | ||
| 178 | } | ||
| 179 | |||
| 180 | SDL_Surface *SDL_LoadBMP_IO(SDL_IOStream *src, bool closeio) | ||
| 181 | { | ||
| 182 | bool was_error = true; | ||
| 183 | Sint64 fp_offset = 0; | ||
| 184 | int i, pad; | ||
| 185 | SDL_Surface *surface; | ||
| 186 | Uint32 Rmask = 0; | ||
| 187 | Uint32 Gmask = 0; | ||
| 188 | Uint32 Bmask = 0; | ||
| 189 | Uint32 Amask = 0; | ||
| 190 | Uint8 *bits; | ||
| 191 | Uint8 *top, *end; | ||
| 192 | bool topDown; | ||
| 193 | bool haveRGBMasks = false; | ||
| 194 | bool haveAlphaMask = false; | ||
| 195 | bool correctAlpha = false; | ||
| 196 | |||
| 197 | // The Win32 BMP file header (14 bytes) | ||
| 198 | char magic[2]; | ||
| 199 | // Uint32 bfSize; | ||
| 200 | // Uint16 bfReserved1; | ||
| 201 | // Uint16 bfReserved2; | ||
| 202 | Uint32 bfOffBits; | ||
| 203 | |||
| 204 | // The Win32 BITMAPINFOHEADER struct (40 bytes) | ||
| 205 | Uint32 biSize; | ||
| 206 | Sint32 biWidth = 0; | ||
| 207 | Sint32 biHeight = 0; | ||
| 208 | // Uint16 biPlanes; | ||
| 209 | Uint16 biBitCount = 0; | ||
| 210 | Uint32 biCompression = 0; | ||
| 211 | // Uint32 biSizeImage; | ||
| 212 | // Sint32 biXPelsPerMeter; | ||
| 213 | // Sint32 biYPelsPerMeter; | ||
| 214 | Uint32 biClrUsed = 0; | ||
| 215 | // Uint32 biClrImportant; | ||
| 216 | |||
| 217 | // Make sure we are passed a valid data source | ||
| 218 | surface = NULL; | ||
| 219 | if (!src) { | ||
| 220 | SDL_InvalidParamError("src"); | ||
| 221 | goto done; | ||
| 222 | } | ||
| 223 | |||
| 224 | // Read in the BMP file header | ||
| 225 | fp_offset = SDL_TellIO(src); | ||
| 226 | if (fp_offset < 0) { | ||
| 227 | goto done; | ||
| 228 | } | ||
| 229 | SDL_ClearError(); | ||
| 230 | if (SDL_ReadIO(src, magic, 2) != 2) { | ||
| 231 | goto done; | ||
| 232 | } | ||
| 233 | if (SDL_strncmp(magic, "BM", 2) != 0) { | ||
| 234 | SDL_SetError("File is not a Windows BMP file"); | ||
| 235 | goto done; | ||
| 236 | } | ||
| 237 | if (!SDL_ReadU32LE(src, NULL /* bfSize */) || | ||
| 238 | !SDL_ReadU16LE(src, NULL /* bfReserved1 */) || | ||
| 239 | !SDL_ReadU16LE(src, NULL /* bfReserved2 */) || | ||
| 240 | !SDL_ReadU32LE(src, &bfOffBits)) { | ||
| 241 | goto done; | ||
| 242 | } | ||
| 243 | |||
| 244 | // Read the Win32 BITMAPINFOHEADER | ||
| 245 | if (!SDL_ReadU32LE(src, &biSize)) { | ||
| 246 | goto done; | ||
| 247 | } | ||
| 248 | if (biSize == 12) { // really old BITMAPCOREHEADER | ||
| 249 | Uint16 biWidth16, biHeight16; | ||
| 250 | if (!SDL_ReadU16LE(src, &biWidth16) || | ||
| 251 | !SDL_ReadU16LE(src, &biHeight16) || | ||
| 252 | !SDL_ReadU16LE(src, NULL /* biPlanes */) || | ||
| 253 | !SDL_ReadU16LE(src, &biBitCount)) { | ||
| 254 | goto done; | ||
| 255 | } | ||
| 256 | biWidth = biWidth16; | ||
| 257 | biHeight = biHeight16; | ||
| 258 | biCompression = BI_RGB; | ||
| 259 | // biSizeImage = 0; | ||
| 260 | // biXPelsPerMeter = 0; | ||
| 261 | // biYPelsPerMeter = 0; | ||
| 262 | biClrUsed = 0; | ||
| 263 | // biClrImportant = 0; | ||
| 264 | } else if (biSize >= 40) { // some version of BITMAPINFOHEADER | ||
| 265 | Uint32 headerSize; | ||
| 266 | if (!SDL_ReadS32LE(src, &biWidth) || | ||
| 267 | !SDL_ReadS32LE(src, &biHeight) || | ||
| 268 | !SDL_ReadU16LE(src, NULL /* biPlanes */) || | ||
| 269 | !SDL_ReadU16LE(src, &biBitCount) || | ||
| 270 | !SDL_ReadU32LE(src, &biCompression) || | ||
| 271 | !SDL_ReadU32LE(src, NULL /* biSizeImage */) || | ||
| 272 | !SDL_ReadU32LE(src, NULL /* biXPelsPerMeter */) || | ||
| 273 | !SDL_ReadU32LE(src, NULL /* biYPelsPerMeter */) || | ||
| 274 | !SDL_ReadU32LE(src, &biClrUsed) || | ||
| 275 | !SDL_ReadU32LE(src, NULL /* biClrImportant */)) { | ||
| 276 | goto done; | ||
| 277 | } | ||
| 278 | |||
| 279 | // 64 == BITMAPCOREHEADER2, an incompatible OS/2 2.x extension. Skip this stuff for now. | ||
| 280 | if (biSize != 64) { | ||
| 281 | /* This is complicated. If compression is BI_BITFIELDS, then | ||
| 282 | we have 3 DWORDS that specify the RGB masks. This is either | ||
| 283 | stored here in an BITMAPV2INFOHEADER (which only differs in | ||
| 284 | that it adds these RGB masks) and biSize >= 52, or we've got | ||
| 285 | these masks stored in the exact same place, but strictly | ||
| 286 | speaking, this is the bmiColors field in BITMAPINFO immediately | ||
| 287 | following the legacy v1 info header, just past biSize. */ | ||
| 288 | if (biCompression == BI_BITFIELDS) { | ||
| 289 | haveRGBMasks = true; | ||
| 290 | if (!SDL_ReadU32LE(src, &Rmask) || | ||
| 291 | !SDL_ReadU32LE(src, &Gmask) || | ||
| 292 | !SDL_ReadU32LE(src, &Bmask)) { | ||
| 293 | goto done; | ||
| 294 | } | ||
| 295 | |||
| 296 | // ...v3 adds an alpha mask. | ||
| 297 | if (biSize >= 56) { // BITMAPV3INFOHEADER; adds alpha mask | ||
| 298 | haveAlphaMask = true; | ||
| 299 | if (!SDL_ReadU32LE(src, &Amask)) { | ||
| 300 | goto done; | ||
| 301 | } | ||
| 302 | } | ||
| 303 | } else { | ||
| 304 | // the mask fields are ignored for v2+ headers if not BI_BITFIELD. | ||
| 305 | if (biSize >= 52) { // BITMAPV2INFOHEADER; adds RGB masks | ||
| 306 | if (!SDL_ReadU32LE(src, NULL /* Rmask */) || | ||
| 307 | !SDL_ReadU32LE(src, NULL /* Gmask */) || | ||
| 308 | !SDL_ReadU32LE(src, NULL /* Bmask */)) { | ||
| 309 | goto done; | ||
| 310 | } | ||
| 311 | } | ||
| 312 | if (biSize >= 56) { // BITMAPV3INFOHEADER; adds alpha mask | ||
| 313 | if (!SDL_ReadU32LE(src, NULL /* Amask */)) { | ||
| 314 | goto done; | ||
| 315 | } | ||
| 316 | } | ||
| 317 | } | ||
| 318 | |||
| 319 | /* Insert other fields here; Wikipedia and MSDN say we're up to | ||
| 320 | v5 of this header, but we ignore those for now (they add gamma, | ||
| 321 | color spaces, etc). Ignoring the weird OS/2 2.x format, we | ||
| 322 | currently parse up to v3 correctly (hopefully!). */ | ||
| 323 | } | ||
| 324 | |||
| 325 | // skip any header bytes we didn't handle... | ||
| 326 | headerSize = (Uint32)(SDL_TellIO(src) - (fp_offset + 14)); | ||
| 327 | if (biSize > headerSize) { | ||
| 328 | if (SDL_SeekIO(src, (biSize - headerSize), SDL_IO_SEEK_CUR) < 0) { | ||
| 329 | goto done; | ||
| 330 | } | ||
| 331 | } | ||
| 332 | } | ||
| 333 | if (biWidth <= 0 || biHeight == 0) { | ||
| 334 | SDL_SetError("BMP file with bad dimensions (%" SDL_PRIs32 "x%" SDL_PRIs32 ")", biWidth, biHeight); | ||
| 335 | goto done; | ||
| 336 | } | ||
| 337 | if (biHeight < 0) { | ||
| 338 | topDown = true; | ||
| 339 | biHeight = -biHeight; | ||
| 340 | } else { | ||
| 341 | topDown = false; | ||
| 342 | } | ||
| 343 | |||
| 344 | // Check for read error | ||
| 345 | if (SDL_strcmp(SDL_GetError(), "") != 0) { | ||
| 346 | goto done; | ||
| 347 | } | ||
| 348 | |||
| 349 | // Reject invalid bit depths | ||
| 350 | switch (biBitCount) { | ||
| 351 | case 0: | ||
| 352 | case 3: | ||
| 353 | case 5: | ||
| 354 | case 6: | ||
| 355 | case 7: | ||
| 356 | SDL_SetError("%u bpp BMP images are not supported", biBitCount); | ||
| 357 | goto done; | ||
| 358 | default: | ||
| 359 | break; | ||
| 360 | } | ||
| 361 | |||
| 362 | // RLE4 and RLE8 BMP compression is supported | ||
| 363 | switch (biCompression) { | ||
| 364 | case BI_RGB: | ||
| 365 | // If there are no masks, use the defaults | ||
| 366 | SDL_assert(!haveRGBMasks); | ||
| 367 | SDL_assert(!haveAlphaMask); | ||
| 368 | // Default values for the BMP format | ||
| 369 | switch (biBitCount) { | ||
| 370 | case 15: | ||
| 371 | case 16: | ||
| 372 | // SDL_PIXELFORMAT_XRGB1555 or SDL_PIXELFORMAT_ARGB1555 if Amask | ||
| 373 | Rmask = 0x7C00; | ||
| 374 | Gmask = 0x03E0; | ||
| 375 | Bmask = 0x001F; | ||
| 376 | break; | ||
| 377 | case 24: | ||
| 378 | #if SDL_BYTEORDER == SDL_BIG_ENDIAN | ||
| 379 | // SDL_PIXELFORMAT_RGB24 | ||
| 380 | Rmask = 0x000000FF; | ||
| 381 | Gmask = 0x0000FF00; | ||
| 382 | Bmask = 0x00FF0000; | ||
| 383 | #else | ||
| 384 | // SDL_PIXELFORMAT_BGR24 | ||
| 385 | Rmask = 0x00FF0000; | ||
| 386 | Gmask = 0x0000FF00; | ||
| 387 | Bmask = 0x000000FF; | ||
| 388 | #endif | ||
| 389 | break; | ||
| 390 | case 32: | ||
| 391 | // We don't know if this has alpha channel or not | ||
| 392 | correctAlpha = true; | ||
| 393 | // SDL_PIXELFORMAT_RGBA8888 | ||
| 394 | Amask = 0xFF000000; | ||
| 395 | Rmask = 0x00FF0000; | ||
| 396 | Gmask = 0x0000FF00; | ||
| 397 | Bmask = 0x000000FF; | ||
| 398 | break; | ||
| 399 | default: | ||
| 400 | break; | ||
| 401 | } | ||
| 402 | break; | ||
| 403 | |||
| 404 | case BI_BITFIELDS: | ||
| 405 | break; // we handled this in the info header. | ||
| 406 | |||
| 407 | default: | ||
| 408 | break; | ||
| 409 | } | ||
| 410 | |||
| 411 | // Create a compatible surface, note that the colors are RGB ordered | ||
| 412 | { | ||
| 413 | SDL_PixelFormat format; | ||
| 414 | |||
| 415 | // Get the pixel format | ||
| 416 | format = SDL_GetPixelFormatForMasks(biBitCount, Rmask, Gmask, Bmask, Amask); | ||
| 417 | surface = SDL_CreateSurface(biWidth, biHeight, format); | ||
| 418 | |||
| 419 | if (!surface) { | ||
| 420 | goto done; | ||
| 421 | } | ||
| 422 | } | ||
| 423 | |||
| 424 | // Load the palette, if any | ||
| 425 | if (SDL_ISPIXELFORMAT_INDEXED(surface->format)) { | ||
| 426 | SDL_Palette *palette = SDL_CreateSurfacePalette(surface); | ||
| 427 | if (!palette) { | ||
| 428 | goto done; | ||
| 429 | } | ||
| 430 | |||
| 431 | if (SDL_SeekIO(src, fp_offset + 14 + biSize, SDL_IO_SEEK_SET) < 0) { | ||
| 432 | SDL_SetError("Error seeking in datastream"); | ||
| 433 | goto done; | ||
| 434 | } | ||
| 435 | |||
| 436 | if (biBitCount >= 32) { // we shift biClrUsed by this value later. | ||
| 437 | SDL_SetError("Unsupported or incorrect biBitCount field"); | ||
| 438 | goto done; | ||
| 439 | } | ||
| 440 | |||
| 441 | if (biClrUsed == 0) { | ||
| 442 | biClrUsed = 1 << biBitCount; | ||
| 443 | } | ||
| 444 | |||
| 445 | if (biClrUsed > (Uint32)palette->ncolors) { | ||
| 446 | biClrUsed = 1 << biBitCount; // try forcing it? | ||
| 447 | if (biClrUsed > (Uint32)palette->ncolors) { | ||
| 448 | SDL_SetError("Unsupported or incorrect biClrUsed field"); | ||
| 449 | goto done; | ||
| 450 | } | ||
| 451 | } | ||
| 452 | palette->ncolors = biClrUsed; | ||
| 453 | |||
| 454 | if (biSize == 12) { | ||
| 455 | for (i = 0; i < palette->ncolors; ++i) { | ||
| 456 | if (!SDL_ReadU8(src, &palette->colors[i].b) || | ||
| 457 | !SDL_ReadU8(src, &palette->colors[i].g) || | ||
| 458 | !SDL_ReadU8(src, &palette->colors[i].r)) { | ||
| 459 | goto done; | ||
| 460 | } | ||
| 461 | palette->colors[i].a = SDL_ALPHA_OPAQUE; | ||
| 462 | } | ||
| 463 | } else { | ||
| 464 | for (i = 0; i < palette->ncolors; ++i) { | ||
| 465 | if (!SDL_ReadU8(src, &palette->colors[i].b) || | ||
| 466 | !SDL_ReadU8(src, &palette->colors[i].g) || | ||
| 467 | !SDL_ReadU8(src, &palette->colors[i].r) || | ||
| 468 | !SDL_ReadU8(src, &palette->colors[i].a)) { | ||
| 469 | goto done; | ||
| 470 | } | ||
| 471 | |||
| 472 | /* According to Microsoft documentation, the fourth element | ||
| 473 | is reserved and must be zero, so we shouldn't treat it as | ||
| 474 | alpha. | ||
| 475 | */ | ||
| 476 | palette->colors[i].a = SDL_ALPHA_OPAQUE; | ||
| 477 | } | ||
| 478 | } | ||
| 479 | } | ||
| 480 | |||
| 481 | // Read the surface pixels. Note that the bmp image is upside down | ||
| 482 | if (SDL_SeekIO(src, fp_offset + bfOffBits, SDL_IO_SEEK_SET) < 0) { | ||
| 483 | SDL_SetError("Error seeking in datastream"); | ||
| 484 | goto done; | ||
| 485 | } | ||
| 486 | if ((biCompression == BI_RLE4) || (biCompression == BI_RLE8)) { | ||
| 487 | if (!readRlePixels(surface, src, biCompression == BI_RLE8)) { | ||
| 488 | SDL_SetError("Error reading from datastream"); | ||
| 489 | goto done; | ||
| 490 | } | ||
| 491 | |||
| 492 | // Success! | ||
| 493 | was_error = false; | ||
| 494 | goto done; | ||
| 495 | } | ||
| 496 | top = (Uint8 *)surface->pixels; | ||
| 497 | end = (Uint8 *)surface->pixels + (surface->h * surface->pitch); | ||
| 498 | pad = ((surface->pitch % 4) ? (4 - (surface->pitch % 4)) : 0); | ||
| 499 | if (topDown) { | ||
| 500 | bits = top; | ||
| 501 | } else { | ||
| 502 | bits = end - surface->pitch; | ||
| 503 | } | ||
| 504 | while (bits >= top && bits < end) { | ||
| 505 | if (SDL_ReadIO(src, bits, surface->pitch) != (size_t)surface->pitch) { | ||
| 506 | goto done; | ||
| 507 | } | ||
| 508 | if (biBitCount == 8 && surface->palette && biClrUsed < (1u << biBitCount)) { | ||
| 509 | for (i = 0; i < surface->w; ++i) { | ||
| 510 | if (bits[i] >= biClrUsed) { | ||
| 511 | SDL_SetError("A BMP image contains a pixel with a color out of the palette"); | ||
| 512 | goto done; | ||
| 513 | } | ||
| 514 | } | ||
| 515 | } | ||
| 516 | #if SDL_BYTEORDER == SDL_BIG_ENDIAN | ||
| 517 | /* Byte-swap the pixels if needed. Note that the 24bpp | ||
| 518 | case has already been taken care of above. */ | ||
| 519 | switch (biBitCount) { | ||
| 520 | case 15: | ||
| 521 | case 16: | ||
| 522 | { | ||
| 523 | Uint16 *pix = (Uint16 *)bits; | ||
| 524 | for (i = 0; i < surface->w; i++) { | ||
| 525 | pix[i] = SDL_Swap16(pix[i]); | ||
| 526 | } | ||
| 527 | break; | ||
| 528 | } | ||
| 529 | |||
| 530 | case 32: | ||
| 531 | { | ||
| 532 | Uint32 *pix = (Uint32 *)bits; | ||
| 533 | for (i = 0; i < surface->w; i++) { | ||
| 534 | pix[i] = SDL_Swap32(pix[i]); | ||
| 535 | } | ||
| 536 | break; | ||
| 537 | } | ||
| 538 | } | ||
| 539 | #endif | ||
| 540 | |||
| 541 | // Skip padding bytes, ugh | ||
| 542 | if (pad) { | ||
| 543 | Uint8 padbyte; | ||
| 544 | for (i = 0; i < pad; ++i) { | ||
| 545 | if (!SDL_ReadU8(src, &padbyte)) { | ||
| 546 | goto done; | ||
| 547 | } | ||
| 548 | } | ||
| 549 | } | ||
| 550 | if (topDown) { | ||
| 551 | bits += surface->pitch; | ||
| 552 | } else { | ||
| 553 | bits -= surface->pitch; | ||
| 554 | } | ||
| 555 | } | ||
| 556 | if (correctAlpha) { | ||
| 557 | CorrectAlphaChannel(surface); | ||
| 558 | } | ||
| 559 | |||
| 560 | was_error = false; | ||
| 561 | |||
| 562 | done: | ||
| 563 | if (was_error) { | ||
| 564 | if (src) { | ||
| 565 | SDL_SeekIO(src, fp_offset, SDL_IO_SEEK_SET); | ||
| 566 | } | ||
| 567 | SDL_DestroySurface(surface); | ||
| 568 | surface = NULL; | ||
| 569 | } | ||
| 570 | if (closeio && src) { | ||
| 571 | SDL_CloseIO(src); | ||
| 572 | } | ||
| 573 | return surface; | ||
| 574 | } | ||
| 575 | |||
| 576 | SDL_Surface *SDL_LoadBMP(const char *file) | ||
| 577 | { | ||
| 578 | SDL_IOStream *stream = SDL_IOFromFile(file, "rb"); | ||
| 579 | if (!stream) { | ||
| 580 | return NULL; | ||
| 581 | } | ||
| 582 | return SDL_LoadBMP_IO(stream, true); | ||
| 583 | } | ||
| 584 | |||
| 585 | bool SDL_SaveBMP_IO(SDL_Surface *surface, SDL_IOStream *dst, bool closeio) | ||
| 586 | { | ||
| 587 | bool was_error = true; | ||
| 588 | Sint64 fp_offset, new_offset; | ||
| 589 | int i, pad; | ||
| 590 | SDL_Surface *intermediate_surface = NULL; | ||
| 591 | Uint8 *bits; | ||
| 592 | bool save32bit = false; | ||
| 593 | bool saveLegacyBMP = false; | ||
| 594 | |||
| 595 | // The Win32 BMP file header (14 bytes) | ||
| 596 | char magic[2] = { 'B', 'M' }; | ||
| 597 | Uint32 bfSize; | ||
| 598 | Uint16 bfReserved1; | ||
| 599 | Uint16 bfReserved2; | ||
| 600 | Uint32 bfOffBits; | ||
| 601 | |||
| 602 | // The Win32 BITMAPINFOHEADER struct (40 bytes) | ||
| 603 | Uint32 biSize; | ||
| 604 | Sint32 biWidth; | ||
| 605 | Sint32 biHeight; | ||
| 606 | Uint16 biPlanes; | ||
| 607 | Uint16 biBitCount; | ||
| 608 | Uint32 biCompression; | ||
| 609 | Uint32 biSizeImage; | ||
| 610 | Sint32 biXPelsPerMeter; | ||
| 611 | Sint32 biYPelsPerMeter; | ||
| 612 | Uint32 biClrUsed; | ||
| 613 | Uint32 biClrImportant; | ||
| 614 | |||
| 615 | // The additional header members from the Win32 BITMAPV4HEADER struct (108 bytes in total) | ||
| 616 | Uint32 bV4RedMask = 0; | ||
| 617 | Uint32 bV4GreenMask = 0; | ||
| 618 | Uint32 bV4BlueMask = 0; | ||
| 619 | Uint32 bV4AlphaMask = 0; | ||
| 620 | Uint32 bV4CSType = 0; | ||
| 621 | Sint32 bV4Endpoints[3 * 3] = { 0 }; | ||
| 622 | Uint32 bV4GammaRed = 0; | ||
| 623 | Uint32 bV4GammaGreen = 0; | ||
| 624 | Uint32 bV4GammaBlue = 0; | ||
| 625 | |||
| 626 | // The additional header members from the Win32 BITMAPV5HEADER struct (124 bytes in total) | ||
| 627 | Uint32 bV5Intent = 0; | ||
| 628 | Uint32 bV5ProfileData = 0; | ||
| 629 | Uint32 bV5ProfileSize = 0; | ||
| 630 | Uint32 bV5Reserved = 0; | ||
| 631 | |||
| 632 | // Make sure we have somewhere to save | ||
| 633 | if (!SDL_SurfaceValid(surface)) { | ||
| 634 | SDL_InvalidParamError("surface"); | ||
| 635 | goto done; | ||
| 636 | } | ||
| 637 | if (!dst) { | ||
| 638 | SDL_InvalidParamError("dst"); | ||
| 639 | goto done; | ||
| 640 | } | ||
| 641 | |||
| 642 | #ifdef SAVE_32BIT_BMP | ||
| 643 | // We can save alpha information in a 32-bit BMP | ||
| 644 | if (SDL_BITSPERPIXEL(surface->format) >= 8 && | ||
| 645 | (SDL_ISPIXELFORMAT_ALPHA(surface->format) || | ||
| 646 | surface->map.info.flags & SDL_COPY_COLORKEY)) { | ||
| 647 | save32bit = true; | ||
| 648 | } | ||
| 649 | #endif // SAVE_32BIT_BMP | ||
| 650 | |||
| 651 | if (surface->palette && !save32bit) { | ||
| 652 | if (SDL_BITSPERPIXEL(surface->format) == 8) { | ||
| 653 | intermediate_surface = surface; | ||
| 654 | } else { | ||
| 655 | SDL_SetError("%u bpp BMP files not supported", | ||
| 656 | SDL_BITSPERPIXEL(surface->format)); | ||
| 657 | goto done; | ||
| 658 | } | ||
| 659 | } else if ((surface->format == SDL_PIXELFORMAT_BGR24 && !save32bit) || | ||
| 660 | (surface->format == SDL_PIXELFORMAT_BGRA32 && save32bit)) { | ||
| 661 | intermediate_surface = surface; | ||
| 662 | } else { | ||
| 663 | SDL_PixelFormat pixel_format; | ||
| 664 | |||
| 665 | /* If the surface has a colorkey or alpha channel we'll save a | ||
| 666 | 32-bit BMP with alpha channel, otherwise save a 24-bit BMP. */ | ||
| 667 | if (save32bit) { | ||
| 668 | pixel_format = SDL_PIXELFORMAT_BGRA32; | ||
| 669 | } else { | ||
| 670 | pixel_format = SDL_PIXELFORMAT_BGR24; | ||
| 671 | } | ||
| 672 | intermediate_surface = SDL_ConvertSurface(surface, pixel_format); | ||
| 673 | if (!intermediate_surface) { | ||
| 674 | SDL_SetError("Couldn't convert image to %d bpp", | ||
| 675 | (int)SDL_BITSPERPIXEL(pixel_format)); | ||
| 676 | goto done; | ||
| 677 | } | ||
| 678 | } | ||
| 679 | |||
| 680 | if (save32bit) { | ||
| 681 | saveLegacyBMP = SDL_GetHintBoolean(SDL_HINT_BMP_SAVE_LEGACY_FORMAT, false); | ||
| 682 | } | ||
| 683 | |||
| 684 | if (SDL_LockSurface(intermediate_surface)) { | ||
| 685 | const size_t bw = intermediate_surface->w * intermediate_surface->fmt->bytes_per_pixel; | ||
| 686 | |||
| 687 | // Set the BMP file header values | ||
| 688 | bfSize = 0; // We'll write this when we're done | ||
| 689 | bfReserved1 = 0; | ||
| 690 | bfReserved2 = 0; | ||
| 691 | bfOffBits = 0; // We'll write this when we're done | ||
| 692 | |||
| 693 | // Write the BMP file header values | ||
| 694 | fp_offset = SDL_TellIO(dst); | ||
| 695 | if (fp_offset < 0) { | ||
| 696 | goto done; | ||
| 697 | } | ||
| 698 | if (SDL_WriteIO(dst, magic, 2) != 2 || | ||
| 699 | !SDL_WriteU32LE(dst, bfSize) || | ||
| 700 | !SDL_WriteU16LE(dst, bfReserved1) || | ||
| 701 | !SDL_WriteU16LE(dst, bfReserved2) || | ||
| 702 | !SDL_WriteU32LE(dst, bfOffBits)) { | ||
| 703 | goto done; | ||
| 704 | } | ||
| 705 | |||
| 706 | // Set the BMP info values | ||
| 707 | biSize = 40; | ||
| 708 | biWidth = intermediate_surface->w; | ||
| 709 | biHeight = intermediate_surface->h; | ||
| 710 | biPlanes = 1; | ||
| 711 | biBitCount = intermediate_surface->fmt->bits_per_pixel; | ||
| 712 | biCompression = BI_RGB; | ||
| 713 | biSizeImage = intermediate_surface->h * intermediate_surface->pitch; | ||
| 714 | biXPelsPerMeter = 0; | ||
| 715 | biYPelsPerMeter = 0; | ||
| 716 | if (intermediate_surface->palette) { | ||
| 717 | biClrUsed = intermediate_surface->palette->ncolors; | ||
| 718 | } else { | ||
| 719 | biClrUsed = 0; | ||
| 720 | } | ||
| 721 | biClrImportant = 0; | ||
| 722 | |||
| 723 | // Set the BMP info values | ||
| 724 | if (save32bit && !saveLegacyBMP) { | ||
| 725 | biSize = 124; | ||
| 726 | // Version 4 values | ||
| 727 | biCompression = BI_BITFIELDS; | ||
| 728 | // The BMP format is always little endian, these masks stay the same | ||
| 729 | bV4RedMask = 0x00ff0000; | ||
| 730 | bV4GreenMask = 0x0000ff00; | ||
| 731 | bV4BlueMask = 0x000000ff; | ||
| 732 | bV4AlphaMask = 0xff000000; | ||
| 733 | bV4CSType = LCS_sRGB; | ||
| 734 | bV4GammaRed = 0; | ||
| 735 | bV4GammaGreen = 0; | ||
| 736 | bV4GammaBlue = 0; | ||
| 737 | // Version 5 values | ||
| 738 | bV5Intent = LCS_GM_GRAPHICS; | ||
| 739 | bV5ProfileData = 0; | ||
| 740 | bV5ProfileSize = 0; | ||
| 741 | bV5Reserved = 0; | ||
| 742 | } | ||
| 743 | |||
| 744 | // Write the BMP info values | ||
| 745 | if (!SDL_WriteU32LE(dst, biSize) || | ||
| 746 | !SDL_WriteS32LE(dst, biWidth) || | ||
| 747 | !SDL_WriteS32LE(dst, biHeight) || | ||
| 748 | !SDL_WriteU16LE(dst, biPlanes) || | ||
| 749 | !SDL_WriteU16LE(dst, biBitCount) || | ||
| 750 | !SDL_WriteU32LE(dst, biCompression) || | ||
| 751 | !SDL_WriteU32LE(dst, biSizeImage) || | ||
| 752 | !SDL_WriteU32LE(dst, biXPelsPerMeter) || | ||
| 753 | !SDL_WriteU32LE(dst, biYPelsPerMeter) || | ||
| 754 | !SDL_WriteU32LE(dst, biClrUsed) || | ||
| 755 | !SDL_WriteU32LE(dst, biClrImportant)) { | ||
| 756 | goto done; | ||
| 757 | } | ||
| 758 | |||
| 759 | // Write the BMP info values | ||
| 760 | if (save32bit && !saveLegacyBMP) { | ||
| 761 | // Version 4 values | ||
| 762 | if (!SDL_WriteU32LE(dst, bV4RedMask) || | ||
| 763 | !SDL_WriteU32LE(dst, bV4GreenMask) || | ||
| 764 | !SDL_WriteU32LE(dst, bV4BlueMask) || | ||
| 765 | !SDL_WriteU32LE(dst, bV4AlphaMask) || | ||
| 766 | !SDL_WriteU32LE(dst, bV4CSType)) { | ||
| 767 | goto done; | ||
| 768 | } | ||
| 769 | for (i = 0; i < 3 * 3; i++) { | ||
| 770 | if (!SDL_WriteU32LE(dst, bV4Endpoints[i])) { | ||
| 771 | goto done; | ||
| 772 | } | ||
| 773 | } | ||
| 774 | if (!SDL_WriteU32LE(dst, bV4GammaRed) || | ||
| 775 | !SDL_WriteU32LE(dst, bV4GammaGreen) || | ||
| 776 | !SDL_WriteU32LE(dst, bV4GammaBlue)) { | ||
| 777 | goto done; | ||
| 778 | } | ||
| 779 | // Version 5 values | ||
| 780 | if (!SDL_WriteU32LE(dst, bV5Intent) || | ||
| 781 | !SDL_WriteU32LE(dst, bV5ProfileData) || | ||
| 782 | !SDL_WriteU32LE(dst, bV5ProfileSize) || | ||
| 783 | !SDL_WriteU32LE(dst, bV5Reserved)) { | ||
| 784 | goto done; | ||
| 785 | } | ||
| 786 | } | ||
| 787 | |||
| 788 | // Write the palette (in BGR color order) | ||
| 789 | if (intermediate_surface->palette) { | ||
| 790 | SDL_Color *colors; | ||
| 791 | int ncolors; | ||
| 792 | |||
| 793 | colors = intermediate_surface->palette->colors; | ||
| 794 | ncolors = intermediate_surface->palette->ncolors; | ||
| 795 | for (i = 0; i < ncolors; ++i) { | ||
| 796 | if (!SDL_WriteU8(dst, colors[i].b) || | ||
| 797 | !SDL_WriteU8(dst, colors[i].g) || | ||
| 798 | !SDL_WriteU8(dst, colors[i].r) || | ||
| 799 | !SDL_WriteU8(dst, colors[i].a)) { | ||
| 800 | goto done; | ||
| 801 | } | ||
| 802 | } | ||
| 803 | } | ||
| 804 | |||
| 805 | // Write the bitmap offset | ||
| 806 | bfOffBits = (Uint32)(SDL_TellIO(dst) - fp_offset); | ||
| 807 | if (SDL_SeekIO(dst, fp_offset + 10, SDL_IO_SEEK_SET) < 0) { | ||
| 808 | goto done; | ||
| 809 | } | ||
| 810 | if (!SDL_WriteU32LE(dst, bfOffBits)) { | ||
| 811 | goto done; | ||
| 812 | } | ||
| 813 | if (SDL_SeekIO(dst, fp_offset + bfOffBits, SDL_IO_SEEK_SET) < 0) { | ||
| 814 | goto done; | ||
| 815 | } | ||
| 816 | |||
| 817 | // Write the bitmap image upside down | ||
| 818 | bits = (Uint8 *)intermediate_surface->pixels + (intermediate_surface->h * intermediate_surface->pitch); | ||
| 819 | pad = ((bw % 4) ? (4 - (bw % 4)) : 0); | ||
| 820 | while (bits > (Uint8 *)intermediate_surface->pixels) { | ||
| 821 | bits -= intermediate_surface->pitch; | ||
| 822 | if (SDL_WriteIO(dst, bits, bw) != bw) { | ||
| 823 | goto done; | ||
| 824 | } | ||
| 825 | if (pad) { | ||
| 826 | const Uint8 padbyte = 0; | ||
| 827 | for (i = 0; i < pad; ++i) { | ||
| 828 | if (!SDL_WriteU8(dst, padbyte)) { | ||
| 829 | goto done; | ||
| 830 | } | ||
| 831 | } | ||
| 832 | } | ||
| 833 | } | ||
| 834 | |||
| 835 | // Write the BMP file size | ||
| 836 | new_offset = SDL_TellIO(dst); | ||
| 837 | if (new_offset < 0) { | ||
| 838 | goto done; | ||
| 839 | } | ||
| 840 | bfSize = (Uint32)(new_offset - fp_offset); | ||
| 841 | if (SDL_SeekIO(dst, fp_offset + 2, SDL_IO_SEEK_SET) < 0) { | ||
| 842 | goto done; | ||
| 843 | } | ||
| 844 | if (!SDL_WriteU32LE(dst, bfSize)) { | ||
| 845 | goto done; | ||
| 846 | } | ||
| 847 | if (SDL_SeekIO(dst, fp_offset + bfSize, SDL_IO_SEEK_SET) < 0) { | ||
| 848 | goto done; | ||
| 849 | } | ||
| 850 | |||
| 851 | // Close it up.. | ||
| 852 | SDL_UnlockSurface(intermediate_surface); | ||
| 853 | |||
| 854 | was_error = false; | ||
| 855 | } | ||
| 856 | |||
| 857 | done: | ||
| 858 | if (intermediate_surface && intermediate_surface != surface) { | ||
| 859 | SDL_DestroySurface(intermediate_surface); | ||
| 860 | } | ||
| 861 | if (closeio && dst) { | ||
| 862 | if (!SDL_CloseIO(dst)) { | ||
| 863 | was_error = true; | ||
| 864 | } | ||
| 865 | } | ||
| 866 | if (was_error) { | ||
| 867 | return false; | ||
| 868 | } | ||
| 869 | return true; | ||
| 870 | } | ||
| 871 | |||
| 872 | bool SDL_SaveBMP(SDL_Surface *surface, const char *file) | ||
| 873 | { | ||
| 874 | SDL_IOStream *stream = SDL_IOFromFile(file, "wb"); | ||
| 875 | if (!stream) { | ||
| 876 | return false; | ||
| 877 | } | ||
| 878 | return SDL_SaveBMP_IO(surface, stream, true); | ||
| 879 | } | ||
