bitmap.h

   1 /*
   2     Copyright 2008 Mark Calderbank
   3
   4     This file is part of VOMP.
   5
   6     VOMP is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     VOMP is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with VOMP; if not, write to the Free Software Foundation, Inc.,
  18     51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19 */
  20 #ifndef BITMAP_H
  21 #define BITMAP_H
  22
  23 #include "defines.h"
  24 #include <vector>
  25
  26 class Palette
  27 {
  28   public:
  29     Palette(UCHAR tBpp = 8);
  30     UCHAR getBpp() const { return bpp; }
  31     void reset() { numColours = 0; }
  32     void setBpp(UCHAR tBpp);
  33     ULONG getColour(UCHAR index) const {
  34       return index < maxColours ? colour[index] : 0; }
  35     void setColour(UCHAR index, ULONG tColour);
  36     void setYCrCbA(UCHAR index, UCHAR tY, UCHAR tCr, UCHAR tCb, UCHAR tA);
  37     const std::vector<ULONG>& getColourVector() const { return colour; }
  38     const std::vector<UCHAR>& getYVector() const { return Y; }
  39     const std::vector<UCHAR>& getCrVector() const { return Cr; }
  40     const std::vector<UCHAR>& getCbVector() const { return Cb; }
  41     const std::vector<UCHAR>& getAVector() const { return A; }
  42     const UINT getNumColours()const{return numColours;};
  43   private:
  44     const static UINT MAX_DEPTH = 8;
  45     std::vector<ULONG> colour;
  46     std::vector<UCHAR> Y;
  47     std::vector<UCHAR> Cr;
  48     std::vector<UCHAR> Cb;
  49     std::vector<UCHAR> A;
  50     UCHAR bpp;
  51     UINT maxColours, numColours;
  52     void argb2yrba(ULONG argb, UCHAR& y, UCHAR& cr, UCHAR& cb, UCHAR& a);
  53     ULONG yrba2argb(UCHAR y, UCHAR cr, UCHAR cb, UCHAR a);
  54 };
  55
  56 class Bitmap
  57 {
  58   private:
  59     std::vector<UCHAR> bitmap;
  60     UINT width, height;
  61   public:
  62     Bitmap(UINT tWidth = 0, UINT tHeight = 0, UCHAR tBpp = 8);
  63     Palette palette;
  64     UINT getWidth() const { return width; }
  65     UINT getHeight() const { return height; }
  66     UCHAR getIndex(UINT x, UINT y) const;
  67     ULONG getColour(UINT x, UINT y) const;
  68     const std::vector<UCHAR> & rawData() const { return bitmap; }
  69     void setSize(UINT tWidth, UINT tHeight);
  70     bool setIndex(UINT x, UINT y, UCHAR index);
  71     void setAllIndices(UCHAR index);
  72 };
  73
  74
  75 #endif