1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
#include <drm_fourcc.h>
#include <stdexcept>
#include "kms++.h"
#include "color.h"
namespace kms
{
static RGB read_rgb(DumbFramebuffer& fb, int x, int y)
{
uint32_t *pc = (uint32_t *)(fb.map(0) + fb.stride(0) * y);
uint32_t c = pc[x];
return RGB((c >> 16) & 0xff, (c >> 8) & 0xff, c & 0xff);
}
static YUV read_rgb_as_yuv(DumbFramebuffer& fb, int x, int y)
{
RGB rgb = read_rgb(fb, x, y);
return YUV(rgb);
}
static void fb_rgb_to_packed_yuv(DumbFramebuffer& dst_fb, DumbFramebuffer& src_fb)
{
unsigned w = src_fb.width();
unsigned h = src_fb.height();
uint8_t *dst = dst_fb.map(0);
for (unsigned y = 0; y < h; ++y) {
for (unsigned x = 0; x < w; x += 2) {
YUV yuv1 = read_rgb_as_yuv(src_fb, x + 0, y);
YUV yuv2 = read_rgb_as_yuv(src_fb, x + 1, y);
switch (dst_fb.format()) {
case PixelFormat::UYVY:
dst[x * 2 + 0] = (yuv1.u + yuv2.u) / 2;
dst[x * 2 + 1] = yuv1.y;
dst[x * 2 + 2] = (yuv1.v + yuv2.v) / 2;
dst[x * 2 + 3] = yuv2.y;
break;
case PixelFormat::YUYV:
dst[x * 2 + 0] = yuv1.y;
dst[x * 2 + 1] = (yuv1.u + yuv2.u) / 2;
dst[x * 2 + 2] = yuv2.y;
dst[x * 2 + 3] = (yuv1.v + yuv2.v) / 2;
break;
default:
throw std::invalid_argument("fo");
}
}
dst += dst_fb.stride(0);
}
}
static void fb_rgb_to_semiplanar_yuv(DumbFramebuffer& dst_fb, DumbFramebuffer& src_fb)
{
unsigned w = src_fb.width();
unsigned h = src_fb.height();
uint8_t *dst_y = dst_fb.map(0);
uint8_t *dst_uv = dst_fb.map(1);
for (unsigned y = 0; y < h; ++y) {
for (unsigned x = 0; x < w; ++x) {
YUV yuv = read_rgb_as_yuv(src_fb, x, y);
dst_y[x] = yuv.y;
}
dst_y += dst_fb.stride(0);
}
for (unsigned y = 0; y < h; y += 2) {
for (unsigned x = 0; x < w; x += 2) {
YUV yuv00 = read_rgb_as_yuv(src_fb, x + 0, y + 0);
YUV yuv01 = read_rgb_as_yuv(src_fb, x + 1, y + 0);
YUV yuv10 = read_rgb_as_yuv(src_fb, x + 0, y + 1);
YUV yuv11 = read_rgb_as_yuv(src_fb, x + 1, y + 1);
unsigned u = (yuv00.u + yuv01.u + yuv10.u + yuv11.u) / 4;
unsigned v = (yuv00.v + yuv01.v + yuv10.v + yuv11.v) / 4;
dst_uv[x + 0] = u;
dst_uv[x + 1] = v;
}
dst_uv += dst_fb.stride(1);
}
}
static void fb_rgb_to_rgb565(DumbFramebuffer& dst_fb, DumbFramebuffer& src_fb)
{
unsigned w = src_fb.width();
unsigned h = src_fb.height();
uint8_t *dst = dst_fb.map(0);
for (unsigned y = 0; y < h; ++y) {
for (unsigned x = 0; x < w; ++x) {
RGB rgb = read_rgb(src_fb, x, y);
unsigned r = rgb.r * 32 / 256;
unsigned g = rgb.g * 64 / 256;
unsigned b = rgb.b * 32 / 256;
((uint16_t *)dst)[x] = (r << 11) | (g << 5) | (b << 0);
}
dst += dst_fb.stride(0);
}
}
void color_convert(DumbFramebuffer& dst, DumbFramebuffer &src)
{
switch (dst.format()) {
case PixelFormat::NV12:
case PixelFormat::NV21:
fb_rgb_to_semiplanar_yuv(dst, src);
break;
case PixelFormat::YUYV:
case PixelFormat::UYVY:
fb_rgb_to_packed_yuv(dst, src);
break;
case PixelFormat::RGB565:
fb_rgb_to_rgb565(dst, src);
break;
default:
throw std::invalid_argument("fo");
}
}
}
|
