]> git.vomp.tv Git - vompclient-marten.git/blob - bitmap.cc
Motion Compensation, all implemented, all buggy
[vompclient-marten.git] / 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 }