git.cworth.org Git - apitrace/blob - bmp.hpp

   1 /**************************************************************************
   2  *
   3  * Copyright 2008-2010 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  *
  24  **************************************************************************/
  25
  26 #ifndef _BMP_HPP_
  27 #define _BMP_HPP_
  28
  29
  30 #include <fstream>
  31
  32
  33 namespace BMP {
  34
  35 #pragma pack(push,2)
  36 struct FileHeader {
  37    uint16_t bfType;
  38    uint32_t bfSize;
  39    uint16_t bfReserved1;
  40    uint16_t bfReserved2;
  41    uint32_t bfOffBits;
  42 };
  43 #pragma pack(pop)
  44
  45 struct InfoHeader {
  46    uint32_t biSize;
  47    int32_t biWidth;
  48    int32_t biHeight;
  49    uint16_t biPlanes;
  50    uint16_t biBitCount;
  51    uint32_t biCompression;
  52    uint32_t biSizeImage;
  53    int32_t biXPelsPerMeter;
  54    int32_t biYPelsPerMeter;
  55    uint32_t biClrUsed;
  56    uint32_t biClrImportant;
  57 };
  58
  59 struct Pixel {
  60    uint8_t rgbBlue;
  61    uint8_t rgbGreen;
  62    uint8_t rgbRed;
  63    uint8_t rgbAlpha;
  64 };
  65
  66
  67 static inline void
  68 write(const char *filename,
  69       const unsigned char *rgba,
  70       unsigned width, unsigned height,
  71       unsigned stride,
  72       bool flip = false)
  73 {
  74    struct FileHeader bmfh;
  75    struct InfoHeader bmih;
  76    unsigned x, y;
  77
  78    bmfh.bfType = 0x4d42;
  79    bmfh.bfSize = 14 + 40 + height*width*4;
  80    bmfh.bfReserved1 = 0;
  81    bmfh.bfReserved2 = 0;
  82    bmfh.bfOffBits = 14 + 40;
  83
  84    bmih.biSize = 40;
  85    bmih.biWidth = width;
  86    bmih.biHeight = height;
  87    bmih.biPlanes = 1;
  88    bmih.biBitCount = 32;
  89    bmih.biCompression = 0;
  90    bmih.biSizeImage = height*width*4;
  91    bmih.biXPelsPerMeter = 0;
  92    bmih.biYPelsPerMeter = 0;
  93    bmih.biClrUsed = 0;
  94    bmih.biClrImportant = 0;
  95
  96    std::ofstream stream(filename, std::ofstream::binary);
  97
  98    stream.write((const char *)&bmfh, 14);
  99    stream.write((const char *)&bmih, 40);
 100
 101    if (flip) {
 102        y = height;
 103        while (y--) {
 104           const unsigned char *ptr = rgba + y * stride;
 105           for (x = 0; x < width; ++x) {
 106              struct Pixel pixel;
 107              pixel.rgbRed   = ptr[x*4 + 0];
 108              pixel.rgbGreen = ptr[x*4 + 1];
 109              pixel.rgbBlue  = ptr[x*4 + 2];
 110              pixel.rgbAlpha = ptr[x*4 + 3];
 111              stream.write((const char *)&pixel, 4);
 112           }
 113        }
 114    } else {
 115        for (y = 0; y < height; ++y) {
 116           const unsigned char *ptr = rgba + y * stride;
 117           for (x = 0; x < width; ++x) {
 118              struct Pixel pixel;
 119              pixel.rgbRed   = ptr[x*4 + 0];
 120              pixel.rgbGreen = ptr[x*4 + 1];
 121              pixel.rgbBlue  = ptr[x*4 + 2];
 122              pixel.rgbAlpha = ptr[x*4 + 3];
 123              stream.write((const char *)&pixel, 4);
 124           }
 125        }
 126    }
 127
 128    stream.close();
 129 }
 130
 131
 132 } /* namespace BMP */
 133
 134
 135 #endif /* _BMP_HPP_ */