bitmap.cc

   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 #include "bitmap.h"
  21
  22 Palette::Palette(UCHAR tBpp)
  23 {
  24   numColours = 0;
  25   setBpp(tBpp);
  26 }
  27
  28 void Palette::argb2yrba(ULONG argb, UCHAR& y, UCHAR& cr, UCHAR& cb, UCHAR& a)
  29 {
  30   a = ((argb & 0xFF000000) >> 24);
  31   int r = (argb & 0x00FF0000) >> 16;
  32   int g = (argb & 0x0000FF00) >> 8;
  33   int b = (argb & 0x000000FF);
  34   y  = (1052*r + 2065*g + 401*b + 4096*16 + 2048) / 4096;
  35   cr = (1799*r - 1508*g - 291*b + 4096*128 + 2048) / 4096;
  36   cb = (-608*r - 1191*g + 1799*b + 4096*128 + 2048) / 4096;
  37 }
  38
  39 ULONG Palette::yrba2argb(UCHAR y, UCHAR cr, UCHAR cb, UCHAR a)
  40 {
  41   int r, g, b;
  42   r = (4769*(y-16) + 6537*(cr-128) + 2048) / 4096;
  43   g = (4769*(y-16) - 3329*(cr-128) - 1604*(cb-128) + 2048) / 4096;
  44   b = (4769*(y-16) + 8263*(cb-128) + 2048) / 4096;
  45   if (r < 0) r = 0; if (r > 255) r = 255;
  46   if (g < 0) g = 0; if (g > 255) g = 255;
  47   if (b < 0) b = 0; if (b > 255) b = 255;
  48   return (a << 24) + (r << 16) + (g << 8) + b;
  49 }
  50
  51 void Palette::setBpp(UCHAR tBpp)
  52 {
  53   bpp = tBpp;
  54   if (bpp > MAX_DEPTH) bpp = MAX_DEPTH;
  55   maxColours = 1 << bpp;
  56   if (numColours > maxColours) numColours = maxColours;
  57   colour.resize(maxColours,0xFF000000);
  58   Y.resize(maxColours,16);
  59   Cr.resize(maxColours,128);
  60   Cb.resize(maxColours,128);
  61   A.resize(maxColours,255);
  62 }
  63
  64 void Palette::setColour(UCHAR index, ULONG tColour)
  65 {
  66   if (index >= maxColours) return;
  67   if (index >= numColours) numColours = index + 1;
  68   colour[index] = tColour;
  69   argb2yrba(tColour, Y[index], Cr[index], Cb[index], A[index]);
  70 }
  71
  72 void Palette::setYCrCbA(UCHAR index, UCHAR tY, UCHAR tCr, UCHAR tCb, UCHAR tA)
  73 {
  74   if (index >= maxColours) return;
  75   if (index >= numColours) numColours = index + 1;
  76   Y[index] = tY; Cr[index] = tCr; Cb[index] = tCb; A[index] = tA;
  77   colour[index] = yrba2argb(tY, tCr, tCb, tA);
  78 }
  79
  80 Bitmap::Bitmap(UINT tWidth, UINT tHeight, UCHAR tBpp)
  81 {
  82   setSize(tWidth, tHeight);
  83   palette.setBpp(tBpp);
  84 }
  85
  86 void Bitmap::setSize(UINT tWidth, UINT tHeight)
  87 {
  88   if (width == tWidth && height == tHeight) return;
  89   width = tWidth; height = tHeight;
  90   if (width == 0 || height == 0)
  91     bitmap.clear();
  92   else
  93     bitmap.assign(width * height, 0);
  94 }
  95
  96 bool Bitmap::setIndex(UINT x, UINT y, UCHAR index)
  97 {
  98   if (x <= width && y <= height)
  99   {
 100     bitmap[x + y*width] = index;
 101     return true;
 102   }
 103   else return false;
 104 }
 105
 106 UCHAR Bitmap::getIndex(UINT x, UINT y) const
 107 {
 108   if (x > width || y > height)
 109     return 0;
 110   else
 111     return bitmap[x + y*width];
 112 }
 113
 114 ULONG Bitmap::getColour(UINT x, UINT y) const
 115 {
 116   if (x > width || y > height)
 117     return 0;
 118   else
 119     return palette.getColour(bitmap[x + y*width]);
 120 }
 121
 122 void Bitmap::setAllIndices(UCHAR index)
 123 {
 124   bitmap.assign(width * height, index);
 125 }