2 #include <chrono>
3 #include <cstring>
4 #include <cassert>
6 #include <xf86drm.h>
7 #include <xf86drmMode.h>
8 #include <drm_fourcc.h>
9 #include <drm.h>
10 #include <drm_mode.h>
12 #include "kms++.h"
13 #include "test.h"
15 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
17 namespace kms
18 {
19 static void draw_rgb_pixel(DumbFramebuffer& buf, unsigned x, unsigned y, RGB color)
20 {
21 switch (buf.format()) {
22 case PixelFormat::XRGB8888:
23 case PixelFormat::ARGB8888:
24 {
25 uint32_t *p = (uint32_t*)(buf.map(0) + buf.stride(0) * y + x * 4);
26 *p = color.rgb888();
27 break;
28 }
29 case PixelFormat::XBGR8888:
30 case PixelFormat::ABGR8888:
31 {
32 uint32_t *p = (uint32_t*)(buf.map(0) + buf.stride(0) * y + x * 4);
33 *p = color.bgr888();
34 break;
35 }
36 case PixelFormat::RGB565:
37 {
38 uint16_t *p = (uint16_t*)(buf.map(0) + buf.stride(0) * y + x * 2);
39 *p = color.rgb565();
40 break;
41 }
42 default:
43 throw std::invalid_argument("invalid pixelformat");
44 }
45 }
47 static void draw_yuv422_macropixel(DumbFramebuffer& buf, unsigned x, unsigned y, YUV yuv1, YUV yuv2)
48 {
49 ASSERT((x & 1) == 0);
51 uint8_t *p = (uint8_t*)(buf.map(0) + buf.stride(0) * y + x * 2);
53 uint8_t y0 = yuv1.y;
54 uint8_t y1 = yuv2.y;
55 uint8_t u = (yuv1.u + yuv2.u) / 2;
56 uint8_t v = (yuv1.v + yuv2.v) / 2;
58 switch (buf.format()) {
59 case PixelFormat::UYVY:
60 p[0] = u;
61 p[1] = y0;
62 p[2] = v;
63 p[3] = y1;
64 break;
66 case PixelFormat::YUYV:
67 p[0] = y0;
68 p[1] = u;
69 p[2] = y1;
70 p[3] = v;
71 break;
73 case PixelFormat::YVYU:
74 p[0] = y0;
75 p[1] = v;
76 p[2] = y1;
77 p[3] = u;
78 break;
80 case PixelFormat::VYUY:
81 p[0] = v;
82 p[1] = y0;
83 p[2] = u;
84 p[3] = y1;
85 break;
87 default:
88 throw std::invalid_argument("invalid pixelformat");
89 }
90 }
92 static void draw_yuv420_macropixel(DumbFramebuffer& buf, unsigned x, unsigned y,
93 YUV yuv1, YUV yuv2, YUV yuv3, YUV yuv4)
94 {
95 ASSERT((x & 1) == 0);
96 ASSERT((y & 1) == 0);
98 uint8_t *py1 = (uint8_t*)(buf.map(0) + buf.stride(0) * (y + 0) + x);
99 uint8_t *py2 = (uint8_t*)(buf.map(0) + buf.stride(0) * (y + 1) + x);
101 uint8_t *puv = (uint8_t*)(buf.map(1) + buf.stride(1) * (y / 2) + x);
103 uint8_t y0 = yuv1.y;
104 uint8_t y1 = yuv2.y;
105 uint8_t y2 = yuv3.y;
106 uint8_t y3 = yuv4.y;
107 uint8_t u = (yuv1.u + yuv2.u + yuv3.u + yuv4.u) / 4;
108 uint8_t v = (yuv1.v + yuv2.v + yuv3.v + yuv4.v) / 4;
110 switch (buf.format()) {
111 case PixelFormat::NV12:
112 py1[0] = y0;
113 py1[1] = y1;
114 py2[0] = y2;
115 py2[1] = y3;
116 puv[0] = u;
117 puv[1] = v;
118 break;
120 case PixelFormat::NV21:
121 py1[0] = y0;
122 py1[1] = y1;
123 py2[0] = y2;
124 py2[1] = y3;
125 puv[0] = v;
126 puv[1] = u;
127 break;
129 default:
130 throw std::invalid_argument("invalid pixelformat");
131 }
132 }
134 static RGB get_test_pattern_pixel(DumbFramebuffer& fb, unsigned x, unsigned y)
135 {
136 const unsigned w = fb.width();
137 const unsigned h = fb.height();
139 const unsigned mw = 20;
141 const unsigned xm1 = mw;
142 const unsigned xm2 = w - mw - 1;
143 const unsigned ym1 = mw;
144 const unsigned ym2 = h - mw - 1;
146 // white margin lines
147 if (x == xm1 || x == xm2 || y == ym1 || y == ym2)
148 return RGB(255, 255, 255);
149 // white box outlines to corners
150 else if ((x == 0 || x == w - 1) && (y < ym1 || y > ym2))
151 return RGB(255, 255, 255);
152 // white box outlines to corners
153 else if ((y == 0 || y == h - 1) && (x < xm1 || x > xm2))
154 return RGB(255, 255, 255);
155 // blue bar on the left
156 else if (x < xm1 && (y > ym1 && y < ym2))
157 return RGB(0, 0, 255);
158 // blue bar on the top
159 else if (y < ym1 && (x > xm1 && x < xm2))
160 return RGB(0, 0, 255);
161 // red bar on the right
162 else if (x > xm2 && (y > ym1 && y < ym2))
163 return RGB(255, 0, 0);
164 // red bar on the bottom
165 else if (y > ym2 && (x > xm1 && x < xm2))
166 return RGB(255, 0, 0);
167 // inside the margins
168 else if (x > xm1 && x < xm2 && y > ym1 && y < ym2) {
169 // diagonal line
170 if (x == y || w - x == h - y)
171 return RGB(255, 255, 255);
172 // diagonal line
173 else if (w - x == y || x == h - y)
174 return RGB(255, 255, 255);
175 else {
176 int t = (x - xm1 - 1) * 8 / (xm2 - xm1 - 1);
177 unsigned r = 0, g = 0, b = 0;
179 unsigned c = (y - ym1 - 1) % 256;
181 switch (t) {
182 case 0:
183 r = c;
184 break;
185 case 1:
186 g = c;
187 break;
188 case 2:
189 b = c;
190 break;
191 case 3:
192 g = b = c;
193 break;
194 case 4:
195 r = b = c;
196 break;
197 case 5:
198 r = g = c;
199 break;
200 case 6:
201 r = g = b = c;
202 break;
203 case 7:
204 break;
205 }
207 return RGB(r, g, b);
208 }
209 } else {
210 // black corners
211 return RGB(0, 0, 0);
212 }
213 }
215 static void draw_test_pattern_impl(DumbFramebuffer& fb)
216 {
217 unsigned x, y;
218 unsigned w = fb.width();
219 unsigned h = fb.height();
221 switch (fb.format()) {
222 case PixelFormat::XRGB8888:
223 case PixelFormat::XBGR8888:
224 case PixelFormat::ARGB8888:
225 case PixelFormat::ABGR8888:
226 case PixelFormat::RGB565:
227 for (y = 0; y < h; y++) {
228 for (x = 0; x < w; x++) {
229 RGB pixel = get_test_pattern_pixel(fb, x, y);
230 draw_rgb_pixel(fb, x, y, pixel);
231 }
232 }
233 break;
235 case PixelFormat::UYVY:
236 case PixelFormat::YUYV:
237 case PixelFormat::YVYU:
238 case PixelFormat::VYUY:
239 for (y = 0; y < h; y++) {
240 for (x = 0; x < w; x += 2) {
241 RGB pixel1 = get_test_pattern_pixel(fb, x, y);
242 RGB pixel2 = get_test_pattern_pixel(fb, x + 1, y);
243 draw_yuv422_macropixel(fb, x, y, pixel1.yuv(), pixel2.yuv());
244 }
245 }
246 break;
248 case PixelFormat::NV12:
249 case PixelFormat::NV21:
250 for (y = 0; y < h; y += 2) {
251 for (x = 0; x < w; x += 2) {
252 RGB pixel00 = get_test_pattern_pixel(fb, x, y);
253 RGB pixel10 = get_test_pattern_pixel(fb, x + 1, y);
254 RGB pixel01 = get_test_pattern_pixel(fb, x, y + 1);
255 RGB pixel11 = get_test_pattern_pixel(fb, x + 1, y + 1);
256 draw_yuv420_macropixel(fb, x, y,
257 pixel00.yuv(), pixel10.yuv(),
258 pixel01.yuv(), pixel11.yuv());
259 }
260 }
261 break;
262 default:
263 throw std::invalid_argument("unknown pixelformat");
264 }
265 }
267 void draw_test_pattern(DumbFramebuffer& fb)
268 {
269 using namespace std::chrono;
271 auto t1 = high_resolution_clock::now();
273 draw_test_pattern_impl(fb);
275 auto t2 = high_resolution_clock::now();
276 auto time_span = duration_cast<microseconds>(t2 - t1);
278 printf("draw took %u us\n", (unsigned)time_span.count());
279 }
280 }