aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/main.cpp
diff options
context:
space:
mode:
authordec05eba <dec05eba@protonmail.com>2026-08-01 21:17:53 +0200
committerdec05eba <dec05eba@protonmail.com>2026-08-01 21:17:53 +0200
commitd02955183135dbc72072492cdfcb35f2387e2c00 (patch)
treee12defe454c7a7e3661bd1cabd9e11ef5d68d44e /src/main.cpp
parente8fca67df16025598f6ca7bcc31cfdcfb79feb7d (diff)
Move video/audio codec selection from main.cpp into codec_select module
Codec selection returns gsr_error values instead of exiting, and takes gsr_recorder_settings instead of the argument parser.
Diffstat (limited to 'src/main.cpp')
-rw-r--r--src/main.cpp516
1 files changed, 8 insertions, 508 deletions
diff --git a/src/main.cpp b/src/main.cpp
index b618a3b..70ef9e8 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -2,6 +2,8 @@ extern "C" {
#include "../include/ffmpeg_utils.h"
#include "../include/recorder/audio_codec.h"
#include "../include/recorder/video_codec.h"
+#include "../include/recorder/codec_select.h"
+#include "../include/recorder/error.h"
#include "../include/capture/nvfbc.h"
#include "../include/capture/xcomposite.h"
#include "../include/capture/ximage.h"
@@ -811,72 +813,6 @@ fail:
return err;
}
-static gsr_video_encoder* create_video_encoder(gsr_egl *egl, const args_parser &arg_parser) {
- const gsr_color_depth color_depth = video_codec_to_bit_depth(arg_parser.settings.video_codec);
- gsr_video_encoder *video_encoder = nullptr;
-
- if(arg_parser.settings.video_encoder == GSR_VIDEO_ENCODER_HW_CPU) {
- gsr_video_encoder_software_params params;
- params.egl = egl;
- params.color_depth = color_depth;
- video_encoder = gsr_video_encoder_software_create(&params);
- return video_encoder;
- }
-
- if(video_codec_is_vulkan(arg_parser.settings.video_codec)) {
- gsr_video_encoder_vulkan_params params;
- params.egl = egl;
- params.color_depth = color_depth;
- video_encoder = gsr_video_encoder_vulkan_create(&params);
- return video_encoder;
- }
-
- switch(egl->gpu_info.vendor) {
- case GSR_GPU_VENDOR_AMD:
- case GSR_GPU_VENDOR_INTEL:
- case GSR_GPU_VENDOR_BROADCOM: {
- gsr_video_encoder_vaapi_params params;
- params.egl = egl;
- params.color_depth = color_depth;
- video_encoder = gsr_video_encoder_vaapi_create(&params);
- break;
- }
- case GSR_GPU_VENDOR_NVIDIA: {
- gsr_video_encoder_nvenc_params params;
- params.egl = egl;
- params.color_depth = color_depth;
- video_encoder = gsr_video_encoder_nvenc_create(&params);
- break;
- }
- }
-
- return video_encoder;
-}
-
-static bool get_supported_video_codecs(gsr_egl *egl, gsr_video_codec video_codec, bool use_software_video_encoder, bool cleanup, gsr_supported_video_codecs *video_codecs) {
- memset(video_codecs, 0, sizeof(*video_codecs));
-
- if(use_software_video_encoder) {
- video_codecs->h264.supported = avcodec_find_encoder_by_name("libx264");
- video_codecs->h264.max_resolution = {4096, 2304};
- return true;
- }
-
- if(video_codec_is_vulkan(video_codec))
- return gsr_get_supported_video_codecs_vulkan(video_codecs, egl->card_path, &egl->vulkan_device_index, cleanup);
-
- switch(egl->gpu_info.vendor) {
- case GSR_GPU_VENDOR_AMD:
- case GSR_GPU_VENDOR_INTEL:
- case GSR_GPU_VENDOR_BROADCOM:
- return gsr_get_supported_video_codecs_vaapi(video_codecs, egl->card_path, cleanup);
- case GSR_GPU_VENDOR_NVIDIA:
- return gsr_get_supported_video_codecs_nvenc(video_codecs, cleanup);
- }
-
- return false;
-}
-
static void xwayland_check_callback(const gsr_monitor *monitor, void *userdata) {
bool *xwayland_found = (bool*)userdata;
if(monitor->name_len >= 8 && strncmp(monitor->name, "XWAYLAND", 8) == 0)
@@ -949,80 +885,6 @@ static void list_gpu_info(gsr_egl *egl) {
printf("card_path|%s\n", egl->card_path);
}
-static const AVCodec* get_ffmpeg_video_codec(gsr_video_codec video_codec, gsr_gpu_vendor vendor) {
- switch(video_codec) {
- case GSR_VIDEO_CODEC_H264:
- return avcodec_find_encoder_by_name(vendor == GSR_GPU_VENDOR_NVIDIA ? "h264_nvenc" : "h264_vaapi");
- case GSR_VIDEO_CODEC_HEVC:
- case GSR_VIDEO_CODEC_HEVC_HDR:
- case GSR_VIDEO_CODEC_HEVC_10BIT:
- return avcodec_find_encoder_by_name(vendor == GSR_GPU_VENDOR_NVIDIA ? "hevc_nvenc" : "hevc_vaapi");
- case GSR_VIDEO_CODEC_AV1:
- case GSR_VIDEO_CODEC_AV1_HDR:
- case GSR_VIDEO_CODEC_AV1_10BIT:
- return avcodec_find_encoder_by_name(vendor == GSR_GPU_VENDOR_NVIDIA ? "av1_nvenc" : "av1_vaapi");
- case GSR_VIDEO_CODEC_VP8:
- return avcodec_find_encoder_by_name(vendor == GSR_GPU_VENDOR_NVIDIA ? "vp8_nvenc" : "vp8_vaapi");
- case GSR_VIDEO_CODEC_VP9:
- return avcodec_find_encoder_by_name(vendor == GSR_GPU_VENDOR_NVIDIA ? "vp9_nvenc" : "vp9_vaapi");
- case GSR_VIDEO_CODEC_H264_VULKAN:
- return avcodec_find_encoder_by_name("h264_vulkan");
- case GSR_VIDEO_CODEC_HEVC_VULKAN:
- case GSR_VIDEO_CODEC_HEVC_HDR_VULKAN:
- case GSR_VIDEO_CODEC_HEVC_10BIT_VULKAN:
- return avcodec_find_encoder_by_name("hevc_vulkan");
- case GSR_VIDEO_CODEC_AV1_VULKAN:
- case GSR_VIDEO_CODEC_AV1_HDR_VULKAN:
- case GSR_VIDEO_CODEC_AV1_10BIT_VULKAN:
- return avcodec_find_encoder_by_name("av1_vulkan");
- }
- return nullptr;
-}
-
-static void set_supported_video_codecs_ffmpeg(gsr_supported_video_codecs *supported_video_codecs, gsr_supported_video_codecs *supported_video_codecs_vulkan, gsr_gpu_vendor vendor) {
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_H264, vendor)) {
- supported_video_codecs->h264.supported = false;
- }
-
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_HEVC, vendor)) {
- supported_video_codecs->hevc.supported = false;
- supported_video_codecs->hevc_hdr.supported = false;
- supported_video_codecs->hevc_10bit.supported = false;
- }
-
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_AV1, vendor)) {
- supported_video_codecs->av1.supported = false;
- supported_video_codecs->av1_hdr.supported = false;
- supported_video_codecs->av1_10bit.supported = false;
- }
-
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_VP8, vendor)) {
- supported_video_codecs->vp8.supported = false;
- }
-
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_VP9, vendor)) {
- supported_video_codecs->vp9.supported = false;
- }
-
- if(supported_video_codecs_vulkan) {
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_H264_VULKAN, vendor)) {
- supported_video_codecs_vulkan->h264.supported = false;
- }
-
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_HEVC_VULKAN, vendor)) {
- supported_video_codecs_vulkan->hevc.supported = false;
- supported_video_codecs_vulkan->hevc_hdr.supported = false;
- supported_video_codecs_vulkan->hevc_10bit.supported = false;
- }
-
- if(!get_ffmpeg_video_codec(GSR_VIDEO_CODEC_AV1_VULKAN, vendor)) {
- supported_video_codecs_vulkan->av1.supported = false;
- supported_video_codecs_vulkan->av1_hdr.supported = false;
- supported_video_codecs_vulkan->av1_10bit.supported = false;
- }
- }
-}
-
static void list_supported_video_codecs(gsr_egl *egl, bool wayland) {
// Dont clean it up on purpose to increase shutdown speed
gsr_supported_video_codecs supported_video_codecs;
@@ -2364,372 +2226,7 @@ static void validate_merged_audio_inputs_app_audio(const std::vector<MergedAudio
}
}
-static gsr_audio_codec select_audio_codec_with_fallback(gsr_audio_codec audio_codec, const std::string &file_extension, bool uses_amix) {
- switch(audio_codec) {
- case GSR_AUDIO_CODEC_AAC: {
- if(file_extension == "webm") {
- //audio_codec_to_use = "opus";
- audio_codec = GSR_AUDIO_CODEC_OPUS;
- gsr_log(GSR_LOG_LEVEL_WARNING, ".webm files only support opus audio codec, changing audio codec from aac to opus");
- }
- break;
- }
- case GSR_AUDIO_CODEC_OPUS: {
- if(file_extension != "mp4" && file_extension != "mkv" && file_extension != "webm" && file_extension != "ts" && file_extension != "whip") {
- //audio_codec_to_use = "aac";
- audio_codec = GSR_AUDIO_CODEC_AAC;
- gsr_log(GSR_LOG_LEVEL_WARNING, "opus audio codec is only supported by .mp4, .mkv, .webm and .ts files, falling back to aac instead");
- }
- break;
- }
- case GSR_AUDIO_CODEC_FLAC: {
- // TODO: Also check mpegts?
- if(file_extension == "webm") {
- //audio_codec_to_use = "opus";
- audio_codec = GSR_AUDIO_CODEC_OPUS;
- gsr_log(GSR_LOG_LEVEL_WARNING, ".webm files only support opus audio codec, changing audio codec from flac to opus");
- } else if(file_extension != "mp4" && file_extension != "mkv") {
- //audio_codec_to_use = "aac";
- audio_codec = GSR_AUDIO_CODEC_AAC;
- gsr_log(GSR_LOG_LEVEL_WARNING, "flac audio codec is only supported by .mp4 and .mkv files, falling back to aac instead");
- } else if(uses_amix) {
- // TODO: remove this? is it true anymore?
- //audio_codec_to_use = "opus";
- audio_codec = GSR_AUDIO_CODEC_OPUS;
- gsr_log(GSR_LOG_LEVEL_WARNING, "flac audio codec is not supported when mixing audio sources, falling back to opus instead");
- }
- break;
- }
- }
- return audio_codec;
-}
-
-static bool video_codec_only_supports_low_power_mode(const gsr_supported_video_codecs &supported_video_codecs, gsr_video_codec video_codec) {
- switch(video_codec) {
- case GSR_VIDEO_CODEC_H264: return supported_video_codecs.h264.low_power;
- case GSR_VIDEO_CODEC_HEVC: return supported_video_codecs.hevc.low_power;
- case GSR_VIDEO_CODEC_HEVC_HDR: return supported_video_codecs.hevc_hdr.low_power;
- case GSR_VIDEO_CODEC_HEVC_10BIT: return supported_video_codecs.hevc_10bit.low_power;
- case GSR_VIDEO_CODEC_AV1: return supported_video_codecs.av1.low_power;
- case GSR_VIDEO_CODEC_AV1_HDR: return supported_video_codecs.av1_hdr.low_power;
- case GSR_VIDEO_CODEC_AV1_10BIT: return supported_video_codecs.av1_10bit.low_power;
- case GSR_VIDEO_CODEC_VP8: return supported_video_codecs.vp8.low_power;
- case GSR_VIDEO_CODEC_VP9: return supported_video_codecs.vp9.low_power;
- case GSR_VIDEO_CODEC_H264_VULKAN: return supported_video_codecs.h264.low_power;
- case GSR_VIDEO_CODEC_HEVC_VULKAN: return supported_video_codecs.hevc.low_power;
- case GSR_VIDEO_CODEC_HEVC_HDR_VULKAN: return supported_video_codecs.hevc_hdr.low_power;
- case GSR_VIDEO_CODEC_HEVC_10BIT_VULKAN: return supported_video_codecs.hevc_10bit.low_power;
- case GSR_VIDEO_CODEC_AV1_VULKAN: return supported_video_codecs.av1.low_power;
- case GSR_VIDEO_CODEC_AV1_HDR_VULKAN: return supported_video_codecs.av1_hdr.low_power;
- case GSR_VIDEO_CODEC_AV1_10BIT_VULKAN: return supported_video_codecs.av1_10bit.low_power;
- }
- return false;
-}
-
-static const AVCodec* get_av_codec_if_supported(gsr_video_codec video_codec, gsr_egl *egl, bool use_software_video_encoder, const gsr_supported_video_codecs *supported_video_codecs) {
- switch(video_codec) {
- case GSR_VIDEO_CODEC_H264:
- case GSR_VIDEO_CODEC_H264_VULKAN: {
- if(use_software_video_encoder)
- return avcodec_find_encoder_by_name("libx264");
- else if(supported_video_codecs->h264.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_HEVC:
- case GSR_VIDEO_CODEC_HEVC_VULKAN: {
- if(supported_video_codecs->hevc.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_HEVC_HDR:
- case GSR_VIDEO_CODEC_HEVC_HDR_VULKAN: {
- if(supported_video_codecs->hevc_hdr.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_HEVC_10BIT:
- case GSR_VIDEO_CODEC_HEVC_10BIT_VULKAN: {
- if(supported_video_codecs->hevc_10bit.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_AV1:
- case GSR_VIDEO_CODEC_AV1_VULKAN: {
- if(supported_video_codecs->av1.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_AV1_HDR:
- case GSR_VIDEO_CODEC_AV1_HDR_VULKAN: {
- if(supported_video_codecs->av1_hdr.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_AV1_10BIT:
- case GSR_VIDEO_CODEC_AV1_10BIT_VULKAN: {
- if(supported_video_codecs->av1_10bit.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_VP8: {
- if(supported_video_codecs->vp8.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- case GSR_VIDEO_CODEC_VP9: {
- if(supported_video_codecs->vp9.supported)
- return get_ffmpeg_video_codec(video_codec, egl->gpu_info.vendor);
- break;
- }
- }
- return nullptr;
-}
-
-static vec2i codec_get_max_resolution(gsr_video_codec video_codec, bool use_software_video_encoder, const gsr_supported_video_codecs *supported_video_codecs) {
- switch(video_codec) {
- case GSR_VIDEO_CODEC_H264:
- case GSR_VIDEO_CODEC_H264_VULKAN: {
- if(use_software_video_encoder)
- return {4096, 2304};
- else if(supported_video_codecs->h264.supported)
- return supported_video_codecs->h264.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_HEVC:
- case GSR_VIDEO_CODEC_HEVC_VULKAN: {
- if(supported_video_codecs->hevc.supported)
- return supported_video_codecs->hevc.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_HEVC_HDR:
- case GSR_VIDEO_CODEC_HEVC_HDR_VULKAN: {
- if(supported_video_codecs->hevc_hdr.supported)
- return supported_video_codecs->hevc_hdr.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_HEVC_10BIT:
- case GSR_VIDEO_CODEC_HEVC_10BIT_VULKAN: {
- if(supported_video_codecs->hevc_10bit.supported)
- return supported_video_codecs->hevc_10bit.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_AV1:
- case GSR_VIDEO_CODEC_AV1_VULKAN: {
- if(supported_video_codecs->av1.supported)
- return supported_video_codecs->av1.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_AV1_HDR:
- case GSR_VIDEO_CODEC_AV1_HDR_VULKAN: {
- if(supported_video_codecs->av1_hdr.supported)
- return supported_video_codecs->av1_hdr.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_AV1_10BIT:
- case GSR_VIDEO_CODEC_AV1_10BIT_VULKAN: {
- if(supported_video_codecs->av1_10bit.supported)
- return supported_video_codecs->av1_10bit.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_VP8: {
- if(supported_video_codecs->vp8.supported)
- return supported_video_codecs->vp8.max_resolution;
- break;
- }
- case GSR_VIDEO_CODEC_VP9: {
- if(supported_video_codecs->vp9.supported)
- return supported_video_codecs->vp9.max_resolution;
- break;
- }
- }
- return {0, 0};
-}
-
-static bool codec_supports_resolution(vec2i codec_max_resolution, vec2i capture_resolution) {
- if(codec_max_resolution.x == 0 || codec_max_resolution.y == 0)
- return true;
- return codec_max_resolution.x >= capture_resolution.x && codec_max_resolution.y >= capture_resolution.y;
-}
-
-static void print_codec_error(gsr_video_codec video_codec) {
- if(video_codec == (gsr_video_codec)GSR_VIDEO_CODEC_AUTO)
- video_codec = GSR_VIDEO_CODEC_H264;
-
- const char *video_codec_name = video_codec_to_string(video_codec);
- gsr_log(GSR_LOG_LEVEL_ERROR, "your gpu does not support '%s' video codec. If you are sure that your gpu does support '%s' video encoding and you are using an AMD/Intel GPU,\n then make sure you have installed the GPU specific vaapi packages (intel-media-driver, libva-intel-driver, libva-mesa-driver and linux-firmware).\n It's also possible that your distro has disabled hardware accelerated video encoding for '%s' video codec.\n This may be the case on corporate distros such as Manjaro, Fedora or OpenSUSE.\n You can test this by running 'vainfo | grep VAEntrypointEncSlice' to see if it matches any H264/HEVC/AV1/VP8/VP9 profile.\n On such distros, you need to manually install mesa from source to enable H264/HEVC hardware acceleration, or use a more user friendly distro. Alternatively record with AV1 if supported by your GPU.\n You can alternatively use the flatpak version of GPU Screen Recorder (https://flathub.org/apps/com.dec05eba.gpu_screen_recorder) which bypasses system issues with patented H264/HEVC codecs.\n If your GPU doesn't support hardware accelerated video encoding then you can use '-fallback-cpu-encoding yes' option to encode with your cpu instead.", video_codec_name, video_codec_name, video_codec_name);
-}
-
-static void force_cpu_encoding(args_parser *args_parser) {
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_H264;
- args_parser->settings.video_encoder = GSR_VIDEO_ENCODER_HW_CPU;
- if(args_parser->settings.bitrate_mode == GSR_BITRATE_MODE_VBR) {
- gsr_log(GSR_LOG_LEVEL_WARNING, "bitrate mode has been forcefully set to qp because software encoding option doesn't support vbr option");
- args_parser->settings.bitrate_mode = GSR_BITRATE_MODE_QP;
- }
-}
-
-static const AVCodec* pick_video_codec(gsr_egl *egl, args_parser *args_parser, bool use_fallback_codec, bool *low_power, gsr_supported_video_codecs *supported_video_codecs) {
- // TODO: software encoder for hevc, av1, vp8 and vp9
- *low_power = false;
- const AVCodec *video_codec_f = get_av_codec_if_supported(args_parser->settings.video_codec, egl, args_parser->settings.video_encoder == GSR_VIDEO_ENCODER_HW_CPU, supported_video_codecs);
-
- if(!video_codec_f && use_fallback_codec && args_parser->settings.video_encoder != GSR_VIDEO_ENCODER_HW_CPU) {
- switch(args_parser->settings.video_codec) {
- case GSR_VIDEO_CODEC_H264: {
- gsr_log(GSR_LOG_LEVEL_ERROR, "selected video codec h264 is not supported by your hardware");
- if(args_parser->settings.fallback_cpu_encoding) {
- gsr_log(GSR_LOG_LEVEL_WARNING, "gpu encoding is not available on your system, trying cpu encoding instead because -fallback-cpu-encoding is enabled. Install the proper vaapi drivers on your system (if supported) if you experience performance issues");
- force_cpu_encoding(args_parser);
- }
- break;
- }
- case GSR_VIDEO_CODEC_HEVC:
- case GSR_VIDEO_CODEC_HEVC_HDR:
- case GSR_VIDEO_CODEC_HEVC_10BIT: {
- gsr_log(GSR_LOG_LEVEL_WARNING, "selected video codec hevc is not supported by your hardware, trying h264 instead");
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_H264;
- return pick_video_codec(egl, args_parser, true, low_power, supported_video_codecs);
- }
- case GSR_VIDEO_CODEC_AV1:
- case GSR_VIDEO_CODEC_AV1_HDR:
- case GSR_VIDEO_CODEC_AV1_10BIT: {
- gsr_log(GSR_LOG_LEVEL_WARNING, "selected video codec av1 is not supported by your hardware, trying h264 instead");
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_H264;
- return pick_video_codec(egl, args_parser, true, low_power, supported_video_codecs);
- }
- case GSR_VIDEO_CODEC_VP8:
- case GSR_VIDEO_CODEC_VP9:
- // TODO: Cant fallback to other codec because webm only supports vp8/vp9
- break;
- case GSR_VIDEO_CODEC_H264_VULKAN: {
- gsr_log(GSR_LOG_LEVEL_WARNING, "selected video codec h264_vulkan is not supported by your hardware, trying h264 instead");
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_H264;
- // Need to do a query again because this time it's without vulkan
- if(!get_supported_video_codecs(egl, args_parser->settings.video_codec, false, true, supported_video_codecs)) {
- gsr_log(GSR_LOG_LEVEL_ERROR, "failed to query for supported video codecs");
- print_codec_error(args_parser->settings.video_codec);
- _exit(11);
- }
- return pick_video_codec(egl, args_parser, true, low_power, supported_video_codecs);
- }
- case GSR_VIDEO_CODEC_HEVC_VULKAN:
- case GSR_VIDEO_CODEC_HEVC_HDR_VULKAN:
- case GSR_VIDEO_CODEC_HEVC_10BIT_VULKAN: {
- gsr_log(GSR_LOG_LEVEL_WARNING, "selected video codec hevc_vulkan is not supported by your hardware, trying hevc instead");
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_HEVC;
- // Need to do a query again because this time it's without vulkan
- if(!get_supported_video_codecs(egl, args_parser->settings.video_codec, false, true, supported_video_codecs)) {
- gsr_log(GSR_LOG_LEVEL_ERROR, "failed to query for supported video codecs");
- print_codec_error(args_parser->settings.video_codec);
- _exit(11);
- }
- return pick_video_codec(egl, args_parser, true, low_power, supported_video_codecs);
- }
- case GSR_VIDEO_CODEC_AV1_VULKAN:
- case GSR_VIDEO_CODEC_AV1_HDR_VULKAN:
- case GSR_VIDEO_CODEC_AV1_10BIT_VULKAN: {
- gsr_log(GSR_LOG_LEVEL_WARNING, "selected video codec av1_vulkan is not supported by your hardware, trying av1 instead");
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_AV1;
- // Need to do a query again because this time it's without vulkan
- if(!get_supported_video_codecs(egl, args_parser->settings.video_codec, false, true, supported_video_codecs)) {
- gsr_log(GSR_LOG_LEVEL_ERROR, "failed to query for supported video codecs");
- print_codec_error(args_parser->settings.video_codec);
- _exit(11);
- }
- return pick_video_codec(egl, args_parser, true, low_power, supported_video_codecs);
- }
- }
-
- video_codec_f = get_av_codec_if_supported(args_parser->settings.video_codec, egl, args_parser->settings.video_encoder == GSR_VIDEO_ENCODER_HW_CPU, supported_video_codecs);
- }
-
- if(!video_codec_f) {
- print_codec_error(args_parser->settings.video_codec);
- _exit(54);
- }
-
- *low_power = video_codec_only_supports_low_power_mode(*supported_video_codecs, args_parser->settings.video_codec);
-
- return video_codec_f;
-}
-
/* Returns -1 if none is available */
-static gsr_video_codec select_appropriate_video_codec_automatically(vec2i video_size, const gsr_supported_video_codecs *supported_video_codecs) {
- if(supported_video_codecs->h264.supported && codec_supports_resolution(supported_video_codecs->h264.max_resolution, video_size)) {
- gsr_log(GSR_LOG_LEVEL_INFO, "using h264 encoder because a codec was not specified");
- return GSR_VIDEO_CODEC_H264;
- } else if(supported_video_codecs->hevc.supported && codec_supports_resolution(supported_video_codecs->hevc.max_resolution, video_size)) {
- gsr_log(GSR_LOG_LEVEL_INFO, "using hevc encoder because a codec was not specified and h264 supported max resolution (%dx%d) is less than the capture resolution (%dx%d)", supported_video_codecs->h264.max_resolution.x, supported_video_codecs->h264.max_resolution.y,
- video_size.x, video_size.y);
- return GSR_VIDEO_CODEC_HEVC;
- } else if(supported_video_codecs->av1.supported && codec_supports_resolution(supported_video_codecs->av1.max_resolution, video_size)) {
- gsr_log(GSR_LOG_LEVEL_INFO, "using av1 encoder because a codec was not specified and hevc supported max resolution (%dx%d) is less than the capture resolution (%dx%d)", supported_video_codecs->hevc.max_resolution.x, supported_video_codecs->hevc.max_resolution.y,
- video_size.x, video_size.y);
- return GSR_VIDEO_CODEC_AV1;
- } else {
- return (gsr_video_codec)-1;
- }
-}
-
-static const AVCodec* select_video_codec_with_fallback(vec2i video_size, args_parser *args_parser, const char *file_extension, gsr_egl *egl, bool *low_power) {
- gsr_supported_video_codecs supported_video_codecs_non_vulkan;
- get_supported_video_codecs(egl, args_parser->settings.video_codec, args_parser->settings.video_encoder == GSR_VIDEO_ENCODER_HW_CPU, true, &supported_video_codecs_non_vulkan);
-
- gsr_supported_video_codecs supported_video_codecs_vulkan = supported_video_codecs_non_vulkan;
- set_supported_video_codecs_ffmpeg(&supported_video_codecs_non_vulkan, &supported_video_codecs_vulkan, egl->gpu_info.vendor);
-
- gsr_supported_video_codecs *supported_video_codecs = video_codec_is_vulkan(args_parser->settings.video_codec)
- ? &supported_video_codecs_vulkan
- : &supported_video_codecs_non_vulkan;
-
- const bool video_codec_auto = args_parser->settings.video_codec == (gsr_video_codec)GSR_VIDEO_CODEC_AUTO;
- if(video_codec_auto) {
- if(strcmp(file_extension, "webm") == 0) {
- gsr_log(GSR_LOG_LEVEL_INFO, "using vp8 encoder because a codec was not specified and the file extension is .webm");
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_VP8;
- } else if(args_parser->settings.video_encoder == GSR_VIDEO_ENCODER_HW_CPU) {
- gsr_log(GSR_LOG_LEVEL_INFO, "using h264 encoder because a codec was not specified");
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_H264;
- } else if(args_parser->settings.video_encoder != GSR_VIDEO_ENCODER_HW_CPU) {
- args_parser->settings.video_codec = select_appropriate_video_codec_automatically(video_size, &supported_video_codecs_non_vulkan);
- if(args_parser->settings.video_codec == (gsr_video_codec)-1) {
- if(args_parser->settings.fallback_cpu_encoding) {
- gsr_log(GSR_LOG_LEVEL_WARNING, "gpu encoding is not available on your system or your gpu doesn't support recording at the resolution you are trying to record, trying cpu encoding instead because -fallback-cpu-encoding is enabled. Install the proper vaapi drivers on your system (if supported) if you experience performance issues");
- force_cpu_encoding(args_parser);
- } else {
- gsr_log(GSR_LOG_LEVEL_ERROR, "no video encoder was specified and neither h264, hevc nor av1 are supported on your system or you are trying to capture at a resolution higher than your system supports for each codec.");
- fprintf(stderr, " Ensure that you have installed the proper vaapi driver. If your gpu doesn't support video encoding then you can run gpu-screen-recorder with \"-fallback-cpu-encoding yes\" option to use cpu encoding.\n");
- _exit(52);
- }
- }
- }
- }
-
- if(LIBAVFORMAT_VERSION_INT < AV_VERSION_INT(60, 10, 100) && strcmp(file_extension, "flv") == 0) {
- if(args_parser->settings.video_codec != GSR_VIDEO_CODEC_H264) {
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_H264;
- gsr_log(GSR_LOG_LEVEL_WARNING, "hevc/av1 is not compatible with flv in your outdated version of ffmpeg, falling back to h264 instead.");
- }
- } else if(strcmp(file_extension, "m3u8") == 0) {
- if(video_codec_is_av1(args_parser->settings.video_codec)) {
- args_parser->settings.video_codec = GSR_VIDEO_CODEC_HEVC;
- gsr_log(GSR_LOG_LEVEL_WARNING, "av1 is not compatible with hls (m3u8), falling back to hevc instead.");
- }
- }
-
- const AVCodec *codec = pick_video_codec(egl, args_parser, true, low_power, supported_video_codecs);
-
- const vec2i codec_max_resolution = codec_get_max_resolution(args_parser->settings.video_codec, args_parser->settings.video_encoder == GSR_VIDEO_ENCODER_HW_CPU, supported_video_codecs);
- if(!codec_supports_resolution(codec_max_resolution, video_size)) {
- const char *video_codec_name = video_codec_to_string(args_parser->settings.video_codec);
- gsr_log(GSR_LOG_LEVEL_ERROR, "The max resolution for video codec %s is %dx%d while you are trying to capture at resolution %dx%d. Change capture resolution or video codec and try again", video_codec_name, codec_max_resolution.x, codec_max_resolution.y, video_size.x, video_size.y);
- _exit(53);
- }
-
- return codec;
-}
-
static std::vector<AudioDeviceData> create_device_audio_inputs(const std::vector<AudioInput> &audio_inputs, AVCodecContext *audio_codec_context, int num_channels, double num_audio_frames_shift, std::vector<AVFilterContext*> &src_filter_ctx, bool use_amix) {
std::vector<AudioDeviceData> audio_track_audio_devices;
for(size_t i = 0; i < audio_inputs.size(); ++i) {
@@ -3224,7 +2721,7 @@ int main(int argc, char **argv) {
const double target_fps = 1.0 / (double)arg_parser.settings.fps;
const bool uses_amix = merged_audio_inputs_should_use_amix(requested_audio_inputs);
- arg_parser.settings.audio_codec = select_audio_codec_with_fallback(arg_parser.settings.audio_codec, file_extension, uses_amix);
+ arg_parser.settings.audio_codec = select_audio_codec_with_fallback(arg_parser.settings.audio_codec, file_extension.c_str(), uses_amix);
vec2i video_size = {0, 0};
std::vector<VideoSource> video_sources = create_video_sources(arg_parser, &egl, false, capture_sources, video_size);
@@ -3247,7 +2744,10 @@ int main(int argc, char **argv) {
}
bool low_power = false;
- const AVCodec *video_codec_f = select_video_codec_with_fallback(video_size, &arg_parser, file_extension.c_str(), &egl, &low_power);
+ const AVCodec *video_codec_f = nullptr;
+ const int select_video_codec_result = select_video_codec_with_fallback(video_size, &arg_parser.settings, file_extension.c_str(), &egl, &low_power, &video_codec_f);
+ if(select_video_codec_result != GSR_ERROR_OK)
+ _exit(gsr_error_to_exit_code(select_video_codec_result));
const enum AVPixelFormat video_pix_fmt = get_pixel_format(arg_parser.settings.video_codec, egl.gpu_info.vendor, arg_parser.settings.video_encoder == GSR_VIDEO_ENCODER_HW_CPU);
AVCodecContext *video_codec_context = create_video_codec_context(video_pix_fmt, video_codec_f, &egl, &arg_parser.settings, video_size.x, video_size.y);
@@ -3275,7 +2775,7 @@ int main(int argc, char **argv) {
_exit(1);
}
- gsr_video_encoder *video_encoder = create_video_encoder(&egl, arg_parser);
+ gsr_video_encoder *video_encoder = create_video_encoder(&egl, &arg_parser.settings);
if(!video_encoder) {
gsr_log(GSR_LOG_LEVEL_ERROR, "failed to create video encoder");
_exit(1);