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
|
#include <format>
#include <stdexcept>
#include <string>
#include <kms++/kms++.h>
#include <kms++util/kms++util.h>
#include <pixpat/pixpat.h>
namespace kms
{
static void fill_pixpat_buffer(pixpat_buffer& buf, IFramebuffer& fb)
{
const auto& info = get_pixel_format_info(fb.format());
buf.format = info.name.c_str();
buf.width = fb.width();
buf.height = fb.height();
buf.num_planes = fb.num_planes();
if (buf.num_planes > 4)
throw std::runtime_error("Too many planes");
for (unsigned i = 0; i < buf.num_planes; ++i) {
buf.planes[i] = fb.map(i);
buf.strides[i] = fb.stride(i);
}
}
static void fill_pattern_opts(pixpat_pattern_opts& popts, const TestPatternOptions& options)
{
switch (options.rec) {
case RecStandard::BT601: popts.rec = PIXPAT_REC_BT601; break;
case RecStandard::BT709: popts.rec = PIXPAT_REC_BT709; break;
case RecStandard::BT2020: popts.rec = PIXPAT_REC_BT2020; break;
}
popts.range = options.range == ColorRange::Full ? PIXPAT_RANGE_FULL
: PIXPAT_RANGE_LIMITED;
popts.num_threads = 0;
}
void draw_test_pattern(IFramebuffer& fb, const TestPatternOptions& options)
{
pixpat_buffer buf{};
fill_pixpat_buffer(buf, fb);
pixpat_pattern_opts popts{};
fill_pattern_opts(popts, options);
const char* pattern = options.pattern.empty() ? nullptr : options.pattern.c_str();
std::string params;
if (pattern) {
struct {
const char* alias;
const char* color;
} static const solid_aliases[] = {
{ "red", "ff0000" },
{ "green", "00ff00" },
{ "blue", "0000ff" },
{ "white", "ffffff" },
{ "black", "000000" },
};
for (const auto& a : solid_aliases) {
if (options.pattern == a.alias) {
pattern = "plain";
params = std::string("color=") + a.color;
popts.params = params.c_str();
break;
}
}
}
if (pixpat_draw_pattern(&buf, pattern, &popts) != 0)
throw std::runtime_error("pixpat_draw_pattern failed");
}
void draw_vbar_pattern(IFramebuffer& fb, unsigned x, unsigned width,
const TestPatternOptions& options)
{
pixpat_buffer buf{};
fill_pixpat_buffer(buf, fb);
pixpat_pattern_opts popts{};
fill_pattern_opts(popts, options);
std::string params = std::format("pos={},width={}", x, width);
popts.params = params.c_str();
if (pixpat_draw_pattern(&buf, "vbar", &popts) != 0)
throw std::runtime_error("pixpat_draw_pattern failed");
}
} // namespace kms
|
