git.cworth.org Git - apitrace/blob - image/image.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 /*
  27  * Image I/O.
  28  */
  29
  30 #ifndef _IMAGE_HPP_
  31 #define _IMAGE_HPP_
  32
  33
  34 #include <fstream>
  35
  36
  37 namespace image {
  38
  39
  40 class Image {
  41 public:
  42     unsigned width;
  43     unsigned height;
  44     unsigned channels;
  45
  46     // Flipped vertically or not
  47     bool flipped;
  48
  49     // Pixels in RGBA format
  50     unsigned char *pixels;
  51
  52     inline Image(unsigned w, unsigned h, unsigned c = 4, bool f = false) :
  53         width(w),
  54         height(h),
  55         channels(c),
  56         flipped(f),
  57         pixels(new unsigned char[h*w*c])
  58     {}
  59
  60     inline ~Image() {
  61         delete [] pixels;
  62     }
  63
  64     inline unsigned char *start(void) {
  65         return flipped ? pixels + (height - 1)*width*channels : pixels;
  66     }
  67
  68     inline const unsigned char *start(void) const {
  69         return flipped ? pixels + (height - 1)*width*channels : pixels;
  70     }
  71
  72     inline unsigned char *end(void) {
  73         return flipped ? pixels - width*channels : pixels + height*width*channels;
  74     }
  75
  76     inline const unsigned char *end(void) const {
  77         return flipped ? pixels - width*channels : pixels + height*width*channels;
  78     }
  79
  80     inline signed stride(void) const {
  81         return flipped ? -(signed)(width*channels) : width*channels;
  82     }
  83
  84     bool writeBMP(const char *filename) const;
  85
  86     void writePNM(std::ostream &os, const char *comment = NULL) const;
  87
  88     inline bool writePNM(const char *filename, const char *comment = NULL) const {
  89         std::ofstream os(filename, std::ofstream::binary);
  90         if (!os) {
  91             return false;
  92         }
  93         writePNM(os, comment);
  94         return true;
  95     }
  96
  97     bool
  98     writePNG(std::ostream &os) const;
  99
 100     inline bool
 101     writePNG(const char *filename) const {
 102         std::ofstream os(filename, std::ofstream::binary);
 103         if (!os) {
 104             return false;
 105         }
 106         return writePNG(os);
 107     }
 108
 109     double compare(Image &ref);
 110 };
 111
 112
 113 Image *
 114 readPNG(const char *filename);
 115
 116 const char *
 117 readPNMHeader(const char *buffer, size_t size, unsigned *channels, unsigned *width, unsigned *height);
 118
 119
 120 } /* namespace image */
 121
 122
 123 #endif /* _IMAGE_HPP_ */