summaryrefslogtreecommitdiff
path: root/tests/ttmtest/reconf
diff options
context:
space:
mode:
Diffstat (limited to 'tests/ttmtest/reconf')
0 files changed, 0 insertions, 0 deletions
generated by cgit v1.2.3 (git 2.25.1) at 2025-07-31 00:08:33 +0000
>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 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 
#include <cstdio>
#include <poll.h>
#include <unistd.h>
#include <algorithm>
#include <regex>
#include <fstream>
#include <map>
#include <system_error>
#include <fmt/format.h>

#include <kms++/kms++.h>
#include <kms++util/kms++util.h>
#include <kms++util/videodevice.h>

const uint32_t NUM_SRC_BUFS = 2;
const uint32_t NUM_DST_BUFS = 2;

using namespace std;
using namespace kms;

static const char* usage_str =
	"Usage: wbm2m [OPTIONS]\n\n"
	"Options:\n"
	"  -f, --format=4CC          Output format\n"
	"  -c, --crop=CROP           CROP is <x>,<y>-<w>x<h>\n"
	"  -h, --help                Print this help\n";

const int bar_speed = 4;
const int bar_width = 10;

static unsigned get_bar_pos(DumbFramebuffer* fb, unsigned frame_num)
{
	return (frame_num * bar_speed) % (fb->width() - bar_width + 1);
}

static void read_frame(DumbFramebuffer* fb, unsigned frame_num)
{
	static map<DumbFramebuffer*, int> s_bar_pos_map;

	int old_pos = -1;
	if (s_bar_pos_map.find(fb) != s_bar_pos_map.end())
		old_pos = s_bar_pos_map[fb];

	int pos = get_bar_pos(fb, frame_num);
	draw_color_bar(*fb, old_pos, pos, bar_width);
	draw_text(*fb, fb->width() / 2, 0, to_string(frame_num), RGB(255, 255, 255));
	s_bar_pos_map[fb] = pos;
}

static void parse_crop(const string& crop_str, uint32_t& c_left, uint32_t& c_top,
		       uint32_t& c_width, uint32_t& c_height)
{
	const regex crop_re("(\\d+),(\\d+)-(\\d+)x(\\d+)"); // 400,400-400x400

	smatch sm;
	if (!regex_match(crop_str, sm, crop_re))
		EXIT("Failed to parse crop option '%s'", crop_str.c_str());

	c_left = stoul(sm[1]);
	c_top = stoul(sm[2]);
	c_width = stoul(sm[3]);
	c_height = stoul(sm[4]);
}

int main(int argc, char** argv)
{
	// XXX get from args
	const uint32_t src_width = 800;
	const uint32_t src_height = 480;
	const auto src_fmt = PixelFormat::XRGB8888;
	const uint32_t num_src_frames = 10;

	const uint32_t dst_width = 800;
	const uint32_t dst_height = 480;
	uint32_t c_top, c_left, c_width, c_height;

	auto dst_fmt = PixelFormat::XRGB8888;
	bool use_selection = false;

	OptionSet optionset = {
		Option("f|format=", [&](string s) {
			dst_fmt = FourCCToPixelFormat(s);
		}),
		Option("c|crop=", [&](string s) {
			parse_crop(s, c_left, c_top, c_width, c_height);
			use_selection = true;
		}),
		Option("h|help", [&]() {
			puts(usage_str);
			exit(-1);
		}),
	};

	optionset.parse(argc, argv);

	if (optionset.params().size() > 0) {
		puts(usage_str);
		exit(-1);
	}

	printf("%ux%u-%s -> %ux%u-%s\n", src_width, src_height, PixelFormatToFourCC(src_fmt).c_str(),
	       dst_width, dst_height, PixelFormatToFourCC(dst_fmt).c_str());

	const string filename = fmt::format("wb-out-{}x{}-{}.raw", dst_width, dst_height,
					    PixelFormatToFourCC(dst_fmt));

	printf("writing to %s\n", filename.c_str());

	VideoDevice vid("/dev/video10");

	Card card;

	uint32_t src_frame_num = 0;
	uint32_t dst_frame_num = 0;

	VideoStreamer* out = vid.get_output_streamer();
	VideoStreamer* in = vid.get_capture_streamer();

	out->set_format(src_fmt, src_width, src_height);
	in->set_format(dst_fmt, dst_width, dst_height);

	if (use_selection) {
		out->set_selection(c_left, c_top, c_width, c_height);
		printf("crop -> %u,%u-%ux%u\n", c_left, c_top, c_width, c_height);
	}

	out->set_queue_size(NUM_SRC_BUFS);
	in->set_queue_size(NUM_DST_BUFS);

	for (unsigned i = 0; i < min(NUM_SRC_BUFS, num_src_frames); ++i) {
		auto fb = new DumbFramebuffer(card, src_width, src_height, src_fmt);

		read_frame(fb, src_frame_num++);

		out->queue(fb);
	}

	for (unsigned i = 0; i < min(NUM_DST_BUFS, num_src_frames); ++i) {
		auto fb = new DumbFramebuffer(card, dst_width, dst_height, dst_fmt);
		in->queue(fb);
	}

	vector<pollfd> fds(3);

	fds[0].fd = 0;
	fds[0].events = POLLIN;
	fds[1].fd = vid.fd();
	fds[1].events = POLLIN;