mirror of
https://github.com/hyprwm/wlroots-hyprland.git
synced 2024-11-22 12:55:58 +01:00
examples/dmabuf-capture: use getopt
This makes it a little bit less annoying to provide the right arguments. All options have reasonable defaults.
This commit is contained in:
parent
90ede7f838
commit
61d6408fdb
1 changed files with 72 additions and 25 deletions
|
@ -205,10 +205,11 @@ static void remove_output(struct wayland_output *out) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static struct wayland_output *find_output(struct capture_context *ctx,
|
static struct wayland_output *find_output(struct capture_context *ctx,
|
||||||
struct wl_output *out, uint32_t id) {
|
struct wl_output *out, int id) {
|
||||||
struct wayland_output *output, *tmp;
|
struct wayland_output *output, *tmp;
|
||||||
wl_list_for_each_safe(output, tmp, &ctx->output_list, link) {
|
wl_list_for_each_safe(output, tmp, &ctx->output_list, link) {
|
||||||
if ((output->output == out) || (output->id == id)) {
|
if (output->output == out || (id >= 0 && output->id == (uint32_t)id)
|
||||||
|
|| id == -1) {
|
||||||
return output;
|
return output;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -766,16 +767,77 @@ static int init(struct capture_context *ctx) {
|
||||||
|
|
||||||
static void uninit(struct capture_context *ctx);
|
static void uninit(struct capture_context *ctx);
|
||||||
|
|
||||||
|
static const char usage[] = "usage: dmabuf-capture [options...] <destination file path>\n"
|
||||||
|
" -o <output ID>\n"
|
||||||
|
" -t <hardware device type>\n"
|
||||||
|
" -d <device path>\n"
|
||||||
|
" -e <encoder>\n"
|
||||||
|
" -f <pixel format>\n"
|
||||||
|
" -r <bitrate in Mbps>\n"
|
||||||
|
"\n"
|
||||||
|
"Example:\n"
|
||||||
|
" dmabuf-capture -o 32 -t vaapi -d /dev/dri/renderD129 \\\n"
|
||||||
|
" -e libx264 -f nv12 -r 12 recording.mkv\n";
|
||||||
|
|
||||||
int main(int argc, char *argv[]) {
|
int main(int argc, char *argv[]) {
|
||||||
int err;
|
struct capture_context ctx = {
|
||||||
struct capture_context ctx = { 0 };
|
.hardware_device = "/dev/dri/renderD128",
|
||||||
ctx.class = &((AVClass) {
|
.encoder_name = "libx264",
|
||||||
|
.out_bitrate = 12,
|
||||||
|
};
|
||||||
|
int output_id = -1;
|
||||||
|
const char *hw_device_type = "vaapi";
|
||||||
|
const char *software_format = "nv12";
|
||||||
|
int opt;
|
||||||
|
while ((opt = getopt(argc, argv, "ho:t:d:e:f:r:")) != -1) {
|
||||||
|
char *end;
|
||||||
|
switch (opt) {
|
||||||
|
case 'o':
|
||||||
|
output_id = strtol(optarg, &end, 10);
|
||||||
|
if (optarg[0] == '\0' || end[0] != '\0') {
|
||||||
|
fprintf(stderr, "Output ID is not an integer\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 't':
|
||||||
|
hw_device_type = optarg;
|
||||||
|
break;
|
||||||
|
case 'd':
|
||||||
|
ctx.hardware_device = optarg;
|
||||||
|
break;
|
||||||
|
case 'e':
|
||||||
|
ctx.encoder_name = optarg;
|
||||||
|
break;
|
||||||
|
case 'f':
|
||||||
|
software_format = optarg;
|
||||||
|
break;
|
||||||
|
case 'r':
|
||||||
|
ctx.out_bitrate = strtof(optarg, &end);
|
||||||
|
if (optarg[0] == '\0' || end[0] != '\0') {
|
||||||
|
fprintf(stderr, "Bitrate is not a floating-pointer number\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
fprintf(stderr, "%s", usage);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (optind >= argc) {
|
||||||
|
fprintf(stderr, "Missing destination file argument\n");
|
||||||
|
fprintf(stderr, "%s", usage);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
ctx.out_filename = argv[optind];
|
||||||
|
|
||||||
|
ctx.class = &((AVClass){
|
||||||
.class_name = "dmabuf-capture",
|
.class_name = "dmabuf-capture",
|
||||||
.item_name = av_default_item_name,
|
.item_name = av_default_item_name,
|
||||||
.version = LIBAVUTIL_VERSION_INT,
|
.version = LIBAVUTIL_VERSION_INT,
|
||||||
});
|
});
|
||||||
|
|
||||||
err = init(&ctx);
|
int err = init(&ctx);
|
||||||
if (err) {
|
if (err) {
|
||||||
goto end;
|
goto end;
|
||||||
}
|
}
|
||||||
|
@ -786,31 +848,16 @@ int main(int argc, char *argv[]) {
|
||||||
o->make, o->model, o->id);
|
o->make, o->model, o->id);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (argc != 8) {
|
o = find_output(&ctx, NULL, output_id);
|
||||||
printf("Invalid number of arguments! Usage and example:\n"
|
|
||||||
"./dmabuf-capture <source id> <hardware device type> <device> "
|
|
||||||
"<encoder name> <pixel format> <bitrate in Mbps> <file path>\n"
|
|
||||||
"./dmabuf-capture 0 vaapi /dev/dri/renderD129 libx264 nv12 12 "
|
|
||||||
"dmabuf_recording_01.mkv\n");
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
const int o_id = strtol(argv[1], NULL, 10);
|
|
||||||
o = find_output(&ctx, NULL, o_id);
|
|
||||||
if (!o) {
|
if (!o) {
|
||||||
printf("Unable to find output with ID %i!\n", o_id);
|
printf("Unable to find output with ID %d\n", output_id);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
ctx.target_output = o->output;
|
ctx.target_output = o->output;
|
||||||
ctx.with_cursor = true;
|
ctx.with_cursor = true;
|
||||||
ctx.hw_device_type = av_hwdevice_find_type_by_name(argv[2]);
|
ctx.hw_device_type = av_hwdevice_find_type_by_name(hw_device_type);
|
||||||
ctx.hardware_device = argv[3];
|
ctx.software_format = av_get_pix_fmt(software_format);
|
||||||
|
|
||||||
ctx.encoder_name = argv[4];
|
|
||||||
ctx.software_format = av_get_pix_fmt(argv[5]);
|
|
||||||
ctx.out_bitrate = strtof(argv[6], NULL);
|
|
||||||
ctx.out_filename = argv[7];
|
|
||||||
|
|
||||||
av_dict_set(&ctx.encoder_opts, "preset", "veryfast", 0);
|
av_dict_set(&ctx.encoder_opts, "preset", "veryfast", 0);
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue