• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 #include <kms++util/color.h>
2 
3 namespace kms
4 {
RGB()5 RGB::RGB()
6 {
7 	r = g = b = 0;
8 	a = 255;
9 }
10 
RGB(uint8_t r,uint8_t g,uint8_t b)11 RGB::RGB(uint8_t r, uint8_t g, uint8_t b)
12 	:RGB(255, r, g, b)
13 {
14 }
15 
RGB(uint8_t a,uint8_t r,uint8_t g,uint8_t b)16 RGB::RGB(uint8_t a, uint8_t r, uint8_t g, uint8_t b)
17 {
18 	this->r = r;
19 	this->g = g;
20 	this->b = b;
21 	this->a = a;
22 }
23 
RGB(uint32_t argb)24 RGB::RGB(uint32_t argb)
25 {
26 	this->b = (argb >> 0) & 0xff;
27 	this->g = (argb >> 8) & 0xff;
28 	this->r = (argb >> 16) & 0xff;
29 	this->a = (argb >> 24) & 0xff;
30 }
31 
rgb888() const32 uint32_t RGB::rgb888() const
33 {
34 	return (r << 16) | (g << 8) | (b << 0);
35 }
36 
bgr888() const37 uint32_t RGB::bgr888() const
38 {
39 	return (b << 16) | (g << 8) | (r << 0);
40 }
41 
argb8888() const42 uint32_t RGB::argb8888() const
43 {
44 	return (a << 24) | (r << 16) | (g << 8) | (b << 0);
45 }
46 
abgr8888() const47 uint32_t RGB::abgr8888() const
48 {
49 	return (a << 24) | (b << 16) | (g << 8) | (r << 0);
50 }
51 
rgb565() const52 uint16_t RGB::rgb565() const
53 {
54 	return ((r >> 3) << 11) | ((g >> 2) << 5) | ((b >> 3) << 0);
55 }
56 
bgr565() const57 uint16_t RGB::bgr565() const
58 {
59 	return ((b >> 3) << 11) | ((g >> 2) << 5) | ((r >> 3) << 0);
60 }
61 
yuv(YUVType type) const62 YUV RGB::yuv(YUVType type) const
63 {
64 	return YUV(*this, type);
65 }
66 
67 #define CF_ONE (256)
68 #define CF(a, b, c) { ((int) ((a) * CF_ONE)), ((int) ((b) * CF_ONE)), ((int) ((c) * CF_ONE)) }
69 #define CLAMP(a) ((a) > (CF_ONE-1) ? (CF_ONE-1) : (a) < 0 ? 0 : (a))
70 
71 const int YUVcoef[static_cast<unsigned>(YUVType::MAX)][3][3] = {
72 	[static_cast<unsigned>(YUVType::BT601_Lim)] = {
73 		CF( 0.257,  0.504,  0.098),
74 		CF(-0.148, -0.291,  0.439),
75 		CF( 0.439, -0.368, -0.071) },
76 	[static_cast<unsigned>(YUVType::BT601_Full)] = {
77 		CF( 0.299,  0.587,  0.114),
78 		CF(-0.169, -0.331,  0.500),
79 		CF( 0.500, -0.419, -0.081) },
80 	[static_cast<unsigned>(YUVType::BT709_Lim)] = {
81 		CF( 0.1826,  0.6142,  0.0620),
82 		CF(-0.1006, -0.3386,  0.4392),
83 		CF( 0.4392, -0.3989, -0.0403) },
84 	[static_cast<unsigned>(YUVType::BT709_Full)] = {
85 		CF( 0.2126,  0.7152,  0.0722),
86 		CF(-0.1146, -0.3854,  0.5000),
87 		CF( 0.5000, -0.4542, -0.0468) },
88 };
89 
90 const int YUVoffset[static_cast<unsigned>(YUVType::MAX)][3] = {
91 	[static_cast<unsigned>(YUVType::BT601_Lim)]  = CF(0.0625,  0.5,  0.5),
92 	[static_cast<unsigned>(YUVType::BT601_Full)] = CF(     0,  0.5,  0.5),
93 	[static_cast<unsigned>(YUVType::BT709_Lim)]  = CF(0.0625,  0.5,  0.5),
94 	[static_cast<unsigned>(YUVType::BT709_Full)] = CF(     0,  0.5,  0.5),
95 };
96 
YUV()97 YUV::YUV()
98 {
99 	y = u = v = a = 0;
100 }
101 
YUV(uint8_t y,uint8_t u,uint8_t v)102 YUV::YUV(uint8_t y, uint8_t u, uint8_t v)
103 {
104 	this->y = y;
105 	this->u = u;
106 	this->v = v;
107 	this->a = 0;
108 }
109 
110 static inline
MAKE_YUV_Y(uint8_t r,uint8_t g,uint8_t b,YUVType type)111 uint8_t MAKE_YUV_Y(uint8_t r, uint8_t g, uint8_t b, YUVType type)
112 {
113 	unsigned tidx = static_cast<unsigned>(type);
114 
115 	return CLAMP(((YUVcoef[tidx][0][0] * r + YUVcoef[tidx][0][1] * g +
116 		      YUVcoef[tidx][0][2] * b + CF_ONE/2) / CF_ONE) +
117 		     YUVoffset[tidx][0]);
118 }
119 
120 static inline
MAKE_YUV_U(uint8_t r,uint8_t g,uint8_t b,YUVType type)121 uint8_t MAKE_YUV_U(uint8_t r, uint8_t g, uint8_t b, YUVType type)
122 {
123 	unsigned tidx = static_cast<unsigned>(type);
124 
125 	return CLAMP(((YUVcoef[tidx][1][0] * r + YUVcoef[tidx][1][1] * g +
126 		       YUVcoef[tidx][1][2] * b + CF_ONE/2) / CF_ONE) +
127 		     YUVoffset[tidx][1]);
128 }
129 
130 static inline
MAKE_YUV_V(uint8_t r,uint8_t g,uint8_t b,YUVType type)131 uint8_t MAKE_YUV_V(uint8_t r, uint8_t g, uint8_t b, YUVType type)
132 {
133 	unsigned tidx = static_cast<unsigned>(type);
134 
135 	return CLAMP(((YUVcoef[tidx][2][0] * r + YUVcoef[tidx][2][1] * g +
136 		       YUVcoef[tidx][2][2] * b + CF_ONE/2) / CF_ONE) +
137 		     YUVoffset[tidx][2]);
138 }
139 
YUV(const RGB & rgb,YUVType type)140 YUV::YUV(const RGB& rgb, YUVType type)
141 {
142 	this->y = MAKE_YUV_Y(rgb.r, rgb.g, rgb.b, type);
143 	this->u = MAKE_YUV_U(rgb.r, rgb.g, rgb.b, type);
144 	this->v = MAKE_YUV_V(rgb.r, rgb.g, rgb.b, type);
145 	this->a = rgb.a;
146 }
147 }
148