git.cworth.org Git - apitrace/blob - image/image_bmp.cpp

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