// AUTOGENERATED COPYRIGHT HEADER START // Copyright (C) 2020-2023 Michael Fabian 'Xaymar' Dirks // Copyright (C) 2022 lainon // AUTOGENERATED COPYRIGHT HEADER END #include "nvenc_shared.hpp" #include "encoders/encoder-ffmpeg.hpp" #include "ffmpeg/tools.hpp" extern "C" { #include "warning-disable.hpp" #include #include "warning-enable.hpp" } #define ST_I18N_PRESET "Encoder.FFmpeg.NVENC.Preset" #define ST_I18N_PRESET_(x) ST_I18N_PRESET "." D_VSTR(x) #define ST_KEY_PRESET "Preset" #define ST_I18N_TUNE "Encoder.FFmpeg.NVENC.Tune" #define ST_I18N_TUNE_(x) ST_I18N_TUNE "." D_VSTR(x) #define ST_KEY_TUNE "Tune" #define ST_I18N_RATECONTROL "Encoder.FFmpeg.NVENC.RateControl" #define ST_I18N_RATECONTROL_MODE ST_I18N_RATECONTROL ".Mode" #define ST_I18N_RATECONTROL_MODE_(x) ST_I18N_RATECONTROL_MODE "." D_VSTR(x) #define ST_KEY_RATECONTROL_MODE "RateControl.Mode" #define ST_I18N_RATECONTROL_TWOPASS ST_I18N_RATECONTROL ".TwoPass" #define ST_KEY_RATECONTROL_TWOPASS "RateControl.TwoPass" #define ST_I18N_RATECONTROL_MULTIPASS ST_I18N_RATECONTROL ".MultiPass" #define ST_KEY_RATECONTROL_MULTIPASS "RateControl.MultiPass" #define ST_I18N_RATECONTROL_LOOKAHEAD ST_I18N_RATECONTROL ".LookAhead" #define ST_KEY_RATECONTROL_LOOKAHEAD "RateControl.LookAhead" #define ST_I18N_RATECONTROL_ADAPTIVEI ST_I18N_RATECONTROL ".AdaptiveI" #define ST_KEY_RATECONTROL_ADAPTIVEI "RateControl.AdaptiveI" #define ST_I18N_RATECONTROL_ADAPTIVEB ST_I18N_RATECONTROL ".AdaptiveB" #define ST_KEY_RATECONTROL_ADAPTIVEB "RateControl.AdaptiveB" #define ST_I18N_RATECONTROL_LIMITS ST_I18N_RATECONTROL ".Limits" #define ST_I18N_RATECONTROL_LIMITS_BUFFERSIZE ST_I18N_RATECONTROL_LIMITS ".BufferSize" #define ST_KEY_RATECONTROL_LIMITS_BUFFERSIZE "RateControl.Limits.BufferSize" #define ST_I18N_RATECONTROL_LIMITS_QUALITY ST_I18N_RATECONTROL_LIMITS ".Quality" #define ST_KEY_RATECONTROL_LIMITS_QUALITY "RateControl.Limits.Quality" #define ST_I18N_RATECONTROL_LIMITS_BITRATE ST_I18N_RATECONTROL_LIMITS ".Bitrate" #define ST_I18N_RATECONTROL_LIMITS_BITRATE_TARGET ST_I18N_RATECONTROL_LIMITS_BITRATE ".Target" #define ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET "RateControl.Limits.Bitrate.Target" #define ST_I18N_RATECONTROL_LIMITS_BITRATE_MAXIMUM ST_I18N_RATECONTROL_LIMITS_BITRATE ".Maximum" #define ST_KEY_RATECONTROL_LIMITS_BITRATE_MAXIMUM "RateControl.Limits.Bitrate.Maximum" #define ST_I18N_RATECONTROL_QP ST_I18N_RATECONTROL ".QP" #define ST_I18N_RATECONTROL_QP_MINIMUM ST_I18N_RATECONTROL_QP ".Minimum" #define ST_KEY_RATECONTROL_QP_MINIMUM "RateControl.Quality.Minimum" #define ST_I18N_RATECONTROL_QP_MAXIMUM ST_I18N_RATECONTROL_QP ".Maximum" #define ST_KEY_RATECONTROL_QP_MAXIMUM "RateControl.Quality.Maximum" #define ST_I18N_RATECONTROL_QP_I ST_I18N_RATECONTROL_QP ".I" #define ST_KEY_RATECONTROL_QP_I "RateControl.QP.I" #define ST_I18N_RATECONTROL_QP_P ST_I18N_RATECONTROL_QP ".P" #define ST_KEY_RATECONTROL_QP_P "RateControl.QP.P" #define ST_I18N_RATECONTROL_QP_B ST_I18N_RATECONTROL_QP ".B" #define ST_KEY_RATECONTROL_QP_B "RateControl.QP.B" #define ST_I18N_AQ "Encoder.FFmpeg.NVENC.AQ" #define ST_I18N_AQ_SPATIAL ST_I18N_AQ ".Spatial" #define ST_KEY_AQ_SPATIAL "AQ.Spatial" #define ST_I18N_AQ_TEMPORAL ST_I18N_AQ ".Temporal" #define ST_KEY_AQ_TEMPORAL "AQ.Temporal" #define ST_I18N_AQ_STRENGTH ST_I18N_AQ ".Strength" #define ST_KEY_AQ_STRENGTH "AQ.Strength" #define ST_I18N_OTHER "Encoder.FFmpeg.NVENC.Other" #define ST_I18N_OTHER_BFRAMES ST_I18N_OTHER ".BFrames" #define ST_KEY_OTHER_BFRAMES "Other.BFrames" #define ST_I18N_OTHER_BFRAMEREFERENCEMODE ST_I18N_OTHER ".BFrameReferenceMode" #define ST_KEY_OTHER_BFRAMEREFERENCEMODE "Other.BFrameReferenceMode" #define ST_I18N_OTHER_ZEROLATENCY ST_I18N_OTHER ".ZeroLatency" #define ST_KEY_OTHER_ZEROLATENCY "Other.ZeroLatency" #define ST_I18N_OTHER_WEIGHTEDPREDICTION ST_I18N_OTHER ".WeightedPrediction" #define ST_KEY_OTHER_WEIGHTEDPREDICTION "Other.WeightedPrediction" #define ST_I18N_OTHER_NONREFERENCEPFRAMES ST_I18N_OTHER ".NonReferencePFrames" #define ST_KEY_OTHER_NONREFERENCEPFRAMES "Other.NonReferencePFrames" #define ST_I18N_OTHER_REFERENCEFRAMES ST_I18N_OTHER ".ReferenceFrames" #define ST_KEY_OTHER_REFERENCEFRAMES "Other.ReferenceFrames" #define ST_I18N_OTHER_LOWDELAYKEYFRAMESCALE ST_I18N_OTHER ".LowDelayKeyFrameScale" #define ST_KEY_OTHER_LOWDELAYKEYFRAMESCALE "Other.LowDelayKeyFrameScale" using namespace streamfx::encoder::ffmpeg::handler; inline bool is_cqp(std::string_view rc) { return std::string_view("constqp") == rc; } inline bool is_cbr(std::string_view rc) { return std::string_view("cbr") == rc; } inline bool is_vbr(std::string_view rc) { return std::string_view("vbr") == rc; } bool streamfx::encoder::ffmpeg::handler::nvenc::is_available() { #if defined(D_PLATFORM_WINDOWS) #if defined(D_PLATFORM_64BIT) std::filesystem::path lib_name = "nvEncodeAPI64.dll"; #else std::filesystem::path lib_name = "nvEncodeAPI.dll"; #endif #else std::filesystem::path lib_name = "libnvidia-encode.so.1"; #endif try { streamfx::util::library::load(lib_name); return true; } catch (...) { return false; } } void nvenc::override_update(ffmpeg_instance* instance, obs_data_t*) { AVCodecContext* context = const_cast(instance->get_avcodeccontext()); int64_t rclookahead = 0; int64_t surfaces = 0; int64_t async_depth = 0; av_opt_get_int(context, "rc-lookahead", AV_OPT_SEARCH_CHILDREN, &rclookahead); av_opt_get_int(context, "surfaces", AV_OPT_SEARCH_CHILDREN, &surfaces); av_opt_get_int(context, "async_depth", AV_OPT_SEARCH_CHILDREN, &async_depth); // Calculate and set the number of surfaces to allocate (if not user overridden). if (surfaces == 0) { surfaces = std::max(4ll, (context->max_b_frames + 1ll) * 4ll); if (rclookahead > 0) { surfaces = std::max(1ll, std::max(surfaces, rclookahead + (context->max_b_frames + 5ll))); } else if (context->max_b_frames > 0) { surfaces = std::max(4ll, (context->max_b_frames + 1ll) * 4ll); } else { surfaces = 4; } av_opt_set_int(context, "surfaces", surfaces, AV_OPT_SEARCH_CHILDREN); } // Set delay context->delay = std::min(std::max(static_cast(async_depth), 3), static_cast(surfaces - 1)); } void nvenc::get_defaults(obs_data_t* settings, const AVCodec*, AVCodecContext*) { obs_data_set_default_string(settings, ST_KEY_PRESET, "default"); obs_data_set_default_string(settings, ST_I18N_TUNE, "hq"); obs_data_set_default_string(settings, ST_KEY_RATECONTROL_MODE, "cbr"); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_TWOPASS, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_MULTIPASS, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_LOOKAHEAD, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_ADAPTIVEI, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_ADAPTIVEB, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET, 6000); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_LIMITS_BITRATE_MAXIMUM, 0); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_LIMITS_BUFFERSIZE, 0); obs_data_set_default_double(settings, ST_KEY_RATECONTROL_LIMITS_QUALITY, 0); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_QP_MINIMUM, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_QP_MAXIMUM, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_QP_I, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_QP_P, -1); obs_data_set_default_int(settings, ST_KEY_RATECONTROL_QP_B, -1); obs_data_set_default_int(settings, ST_KEY_AQ_SPATIAL, -1); obs_data_set_default_int(settings, ST_KEY_AQ_STRENGTH, -1); obs_data_set_default_int(settings, ST_KEY_AQ_TEMPORAL, -1); obs_data_set_default_int(settings, ST_KEY_OTHER_BFRAMES, -1); obs_data_set_default_int(settings, ST_KEY_OTHER_BFRAMEREFERENCEMODE, -1); obs_data_set_default_int(settings, ST_KEY_OTHER_ZEROLATENCY, -1); obs_data_set_default_int(settings, ST_KEY_OTHER_WEIGHTEDPREDICTION, -1); obs_data_set_default_int(settings, ST_KEY_OTHER_NONREFERENCEPFRAMES, -1); obs_data_set_default_int(settings, ST_KEY_OTHER_REFERENCEFRAMES, -1); obs_data_set_default_int(settings, ST_KEY_OTHER_LOWDELAYKEYFRAMESCALE, -1); // Replay Buffer obs_data_set_default_int(settings, "bitrate", 0); } static bool modified_ratecontrol(obs_properties_t* props, obs_property_t*, obs_data_t* settings) noexcept { // Decode the name into useful flags. const char* value = obs_data_get_string(settings, ST_KEY_RATECONTROL_MODE); bool have_bitrate = false; bool have_bitrate_range = false; bool have_quality = false; bool have_qp_limits = false; bool have_qp = false; if (value == std::string_view("cbr")) { have_bitrate = true; } else if (value == std::string_view("vbr")) { have_bitrate = true; have_bitrate_range = true; have_quality = true; have_qp_limits = true; have_qp = true; } else if (value == std::string_view("constqp")) { have_qp = true; } else { have_bitrate = true; have_bitrate_range = true; have_quality = true; have_qp_limits = true; have_qp = true; } obs_property_set_visible(obs_properties_get(props, ST_I18N_RATECONTROL_LIMITS), have_bitrate || have_quality); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_LIMITS_BUFFERSIZE), have_bitrate); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_LIMITS_QUALITY), have_quality); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET), have_bitrate); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_LIMITS_BITRATE_MAXIMUM), have_bitrate_range); obs_property_set_visible(obs_properties_get(props, ST_I18N_RATECONTROL_QP), have_qp || have_qp_limits); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_QP_MINIMUM), have_qp_limits); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_QP_MAXIMUM), have_qp_limits); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_QP_I), have_qp); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_QP_P), have_qp); obs_property_set_visible(obs_properties_get(props, ST_KEY_RATECONTROL_QP_B), have_qp); return true; } static bool modified_aq(obs_properties_t* props, obs_property_t*, obs_data_t* settings) noexcept { bool spatial_aq = streamfx::util::is_tristate_enabled(obs_data_get_int(settings, ST_KEY_AQ_SPATIAL)); obs_property_set_visible(obs_properties_get(props, ST_KEY_AQ_STRENGTH), spatial_aq); return true; } void nvenc::get_properties_pre(obs_properties_t* props, const AVCodec*, const AVCodecContext* context) { { auto p = obs_properties_add_list(props, ST_KEY_PRESET, D_TRANSLATE(ST_I18N_PRESET), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING); streamfx::ffmpeg::tools::avoption_list_add_entries(context->priv_data, "preset", [&p](const AVOption* opt) { char buffer[1024]; snprintf(buffer, sizeof(buffer), "%s.%s", ST_I18N_PRESET, opt->name); obs_property_list_add_string(p, D_TRANSLATE(buffer), opt->name); }); } if (streamfx::ffmpeg::tools::avoption_exists(context->priv_data, "tune")) { auto p = obs_properties_add_list(props, ST_KEY_TUNE, D_TRANSLATE(ST_I18N_TUNE), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING); streamfx::ffmpeg::tools::avoption_list_add_entries(context->priv_data, "tune", [&p](const AVOption* opt) { char buffer[1024]; snprintf(buffer, sizeof(buffer), "%s.%s", ST_I18N_TUNE, opt->name); obs_property_list_add_string(p, D_TRANSLATE(buffer), opt->name); }); } } void nvenc::get_properties_post(obs_properties_t* props, const AVCodec* codec, const AVCodecContext* context) { { // Rate Control obs_properties_t* grp = props; if (!streamfx::util::are_property_groups_broken()) { grp = obs_properties_create(); obs_properties_add_group(props, ST_I18N_RATECONTROL, D_TRANSLATE(ST_I18N_RATECONTROL), OBS_GROUP_NORMAL, grp); } { auto p = obs_properties_add_list(grp, ST_KEY_RATECONTROL_MODE, D_TRANSLATE(ST_I18N_RATECONTROL_MODE), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING); obs_property_set_modified_callback(p, modified_ratecontrol); obs_property_list_add_string(p, D_TRANSLATE(S_STATE_DEFAULT), ""); streamfx::ffmpeg::tools::avoption_list_add_entries(context->priv_data, "rc", [&p](const AVOption* opt) { // Ignore options that are "deprecated" but not flagged as such. if (opt->default_val.i64 & (1 << 23)) return; char buffer[1024]; snprintf(buffer, sizeof(buffer), "%s.%s", ST_I18N_RATECONTROL_MODE, opt->name); obs_property_list_add_string(p, D_TRANSLATE(buffer), opt->name); }); } if (streamfx::ffmpeg::tools::avoption_exists(context->priv_data, "multipass")) { auto p = obs_properties_add_list(grp, ST_KEY_RATECONTROL_MULTIPASS, D_TRANSLATE(ST_I18N_RATECONTROL_MULTIPASS), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING); obs_property_list_add_string(p, D_TRANSLATE(S_STATE_DEFAULT), ""); streamfx::ffmpeg::tools::avoption_list_add_entries(context->priv_data, "multipass", [&p](const AVOption* opt) { char buffer[1024]; snprintf(buffer, sizeof(buffer), "%s.%s", ST_I18N_RATECONTROL_MULTIPASS, opt->name); obs_property_list_add_string(p, D_TRANSLATE(buffer), opt->name); }); } else { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_RATECONTROL_TWOPASS, D_TRANSLATE(ST_I18N_RATECONTROL_TWOPASS)); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_RATECONTROL_LOOKAHEAD, D_TRANSLATE(ST_I18N_RATECONTROL_LOOKAHEAD), -1, 32, 1); obs_property_int_set_suffix(p, " frames"); //obs_property_set_modified_callback(p, modified_lookahead); } { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_RATECONTROL_ADAPTIVEI, D_TRANSLATE(ST_I18N_RATECONTROL_ADAPTIVEI)); } if (strcmp(codec->name, "h264_nvenc") == 0) { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_RATECONTROL_ADAPTIVEB, D_TRANSLATE(ST_I18N_RATECONTROL_ADAPTIVEB)); } } { obs_properties_t* grp = props; if (!streamfx::util::are_property_groups_broken()) { grp = obs_properties_create(); obs_properties_add_group(props, ST_I18N_RATECONTROL_LIMITS, D_TRANSLATE(ST_I18N_RATECONTROL_LIMITS), OBS_GROUP_NORMAL, grp); } { auto p = obs_properties_add_float_slider(grp, ST_KEY_RATECONTROL_LIMITS_QUALITY, D_TRANSLATE(ST_I18N_RATECONTROL_LIMITS_QUALITY), 0, 51, 0.01); } { auto p = obs_properties_add_int(grp, ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET, D_TRANSLATE(ST_I18N_RATECONTROL_LIMITS_BITRATE_TARGET), -1, std::numeric_limits::max(), 1); obs_property_int_set_suffix(p, " kbit/s"); } { auto p = obs_properties_add_int(grp, ST_KEY_RATECONTROL_LIMITS_BITRATE_MAXIMUM, D_TRANSLATE(ST_I18N_RATECONTROL_LIMITS_BITRATE_MAXIMUM), -1, std::numeric_limits::max(), 1); obs_property_int_set_suffix(p, " kbit/s"); } { auto p = obs_properties_add_int(grp, ST_KEY_RATECONTROL_LIMITS_BUFFERSIZE, D_TRANSLATE(ST_I18N_RATECONTROL_LIMITS_BUFFERSIZE), 0, std::numeric_limits::max(), 1); obs_property_int_set_suffix(p, " kbit"); } } { obs_properties_t* grp = props; if (!streamfx::util::are_property_groups_broken()) { grp = obs_properties_create(); obs_properties_add_group(props, ST_I18N_RATECONTROL_QP, D_TRANSLATE(ST_I18N_RATECONTROL_QP), OBS_GROUP_NORMAL, grp); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_RATECONTROL_QP_MINIMUM, D_TRANSLATE(ST_I18N_RATECONTROL_QP_MINIMUM), -1, 51, 1); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_RATECONTROL_QP_MAXIMUM, D_TRANSLATE(ST_I18N_RATECONTROL_QP_MAXIMUM), -1, 51, 1); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_RATECONTROL_QP_I, D_TRANSLATE(ST_I18N_RATECONTROL_QP_I), -1, 51, 1); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_RATECONTROL_QP_P, D_TRANSLATE(ST_I18N_RATECONTROL_QP_P), -1, 51, 1); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_RATECONTROL_QP_B, D_TRANSLATE(ST_I18N_RATECONTROL_QP_B), -1, 51, 1); } } { obs_properties_t* grp = props; if (!streamfx::util::are_property_groups_broken()) { grp = obs_properties_create(); obs_properties_add_group(props, ST_I18N_AQ, D_TRANSLATE(ST_I18N_AQ), OBS_GROUP_NORMAL, grp); } { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_AQ_SPATIAL, D_TRANSLATE(ST_I18N_AQ_SPATIAL)); obs_property_set_modified_callback(p, modified_aq); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_AQ_STRENGTH, D_TRANSLATE(ST_I18N_AQ_STRENGTH), -1, 15, 1); } { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_AQ_TEMPORAL, D_TRANSLATE(ST_I18N_AQ_TEMPORAL)); } } { obs_properties_t* grp = props; if (!streamfx::util::are_property_groups_broken()) { grp = obs_properties_create(); obs_properties_add_group(props, ST_I18N_OTHER, D_TRANSLATE(ST_I18N_OTHER), OBS_GROUP_NORMAL, grp); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_OTHER_BFRAMES, D_TRANSLATE(ST_I18N_OTHER_BFRAMES), -1, 4, 1); obs_property_int_set_suffix(p, " frames"); } { auto p = obs_properties_add_list(grp, ST_KEY_OTHER_BFRAMEREFERENCEMODE, D_TRANSLATE(ST_I18N_OTHER_BFRAMEREFERENCEMODE), OBS_COMBO_TYPE_LIST, OBS_COMBO_FORMAT_STRING); obs_property_list_add_string(p, D_TRANSLATE(S_STATE_DEFAULT), ""); streamfx::ffmpeg::tools::avoption_list_add_entries(context->priv_data, "b_ref_mode", [&p](const AVOption* opt) { char buffer[1024]; snprintf(buffer, sizeof(buffer), "%s.%s", ST_I18N_OTHER_BFRAMEREFERENCEMODE, opt->name); obs_property_list_add_string(p, D_TRANSLATE(buffer), opt->name); }); } { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_OTHER_ZEROLATENCY, D_TRANSLATE(ST_I18N_OTHER_ZEROLATENCY)); } { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_OTHER_WEIGHTEDPREDICTION, D_TRANSLATE(ST_I18N_OTHER_WEIGHTEDPREDICTION)); } { auto p = streamfx::util::obs_properties_add_tristate(grp, ST_KEY_OTHER_NONREFERENCEPFRAMES, D_TRANSLATE(ST_I18N_OTHER_NONREFERENCEPFRAMES)); } { auto p = obs_properties_add_int_slider(grp, ST_KEY_OTHER_REFERENCEFRAMES, D_TRANSLATE(ST_I18N_OTHER_REFERENCEFRAMES), -1, (strcmp(codec->name, "h264_nvenc") == 0) ? 16 : 4, 1); obs_property_int_set_suffix(p, " frames"); } if (streamfx::ffmpeg::tools::avoption_exists(context->priv_data, "ldkfs")) { auto p = obs_properties_add_int_slider(grp, ST_KEY_OTHER_LOWDELAYKEYFRAMESCALE, D_TRANSLATE(ST_I18N_OTHER_LOWDELAYKEYFRAMESCALE), -1, 255, 1); } } } void nvenc::get_runtime_properties(obs_properties_t* props, const AVCodec*, AVCodecContext*) { obs_property_set_enabled(obs_properties_get(props, ST_KEY_PRESET), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_TUNE), false); obs_property_set_enabled(obs_properties_get(props, ST_I18N_RATECONTROL), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_MODE), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_TWOPASS), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_MULTIPASS), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_LOOKAHEAD), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_ADAPTIVEI), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_ADAPTIVEB), false); obs_property_set_enabled(obs_properties_get(props, ST_I18N_RATECONTROL_LIMITS), true); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_LIMITS_BUFFERSIZE), true); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET), true); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_LIMITS_BITRATE_MAXIMUM), true); obs_property_set_enabled(obs_properties_get(props, ST_I18N_RATECONTROL_LIMITS_QUALITY), false); obs_property_set_enabled(obs_properties_get(props, ST_I18N_RATECONTROL_QP), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_QP_MINIMUM), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_QP_MAXIMUM), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_QP_I), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_QP_P), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_RATECONTROL_QP_B), false); obs_property_set_enabled(obs_properties_get(props, ST_I18N_AQ), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_AQ_SPATIAL), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_AQ_STRENGTH), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_AQ_TEMPORAL), false); obs_property_set_enabled(obs_properties_get(props, ST_I18N_OTHER), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_OTHER_BFRAMES), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_OTHER_BFRAMEREFERENCEMODE), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_OTHER_ZEROLATENCY), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_OTHER_WEIGHTEDPREDICTION), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_OTHER_NONREFERENCEPFRAMES), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_OTHER_REFERENCEFRAMES), false); obs_property_set_enabled(obs_properties_get(props, ST_KEY_OTHER_LOWDELAYKEYFRAMESCALE), false); } void nvenc::update(obs_data_t* settings, const AVCodec* codec, AVCodecContext* context) { if (const char* v = obs_data_get_string(settings, ST_KEY_PRESET); !context->internal && (v != nullptr) && (v[0] != '\0')) { av_opt_set(context->priv_data, "preset", v, AV_OPT_SEARCH_CHILDREN); } { // Rate Control const char* v = obs_data_get_string(settings, ST_KEY_RATECONTROL_MODE); if (!context->internal && (v != nullptr) && (v[0] != '\0')) { av_opt_set(context->priv_data, "rc", v, AV_OPT_SEARCH_CHILDREN); } // Decode the name into useful flags. bool have_bitrate = false; bool have_bitrate_range = false; bool have_quality = false; bool have_qp_limits = false; bool have_qp = false; if (v && is_cbr(v)) { have_bitrate = true; if (!context->internal) av_opt_set_int(context->priv_data, "cbr", 1, AV_OPT_SEARCH_CHILDREN); // Support for OBS Studio obs_data_set_string(settings, "rate_control", "CBR"); } else if (v && is_vbr(v)) { have_bitrate = true; have_bitrate_range = true; have_quality = true; have_qp_limits = true; have_qp = true; if (!context->internal) av_opt_set_int(context->priv_data, "cbr", 0, AV_OPT_SEARCH_CHILDREN); // Support for OBS Studio obs_data_set_string(settings, "rate_control", "VBR"); } else if (v && is_cqp(v)) { have_qp = true; if (!context->internal) av_opt_set_int(context->priv_data, "cbr", 0, AV_OPT_SEARCH_CHILDREN); // Support for OBS Studio obs_data_set_string(settings, "rate_control", "CQP"); } else { have_bitrate = true; have_bitrate_range = true; have_quality = true; have_qp_limits = true; have_qp = true; if (!context->internal) av_opt_set_int(context->priv_data, "cbr", 0, AV_OPT_SEARCH_CHILDREN); } if (!context->internal) { if (streamfx::ffmpeg::tools::avoption_exists(context->priv_data, "multipass")) { // Multi-Pass if (const char* v = obs_data_get_string(settings, ST_KEY_RATECONTROL_MULTIPASS); (v != nullptr) && (v[0] != '\0')) { av_opt_set(context->priv_data, "multipass", v, AV_OPT_SEARCH_CHILDREN); av_opt_set_int(context->priv_data, "2pass", 0, AV_OPT_SEARCH_CHILDREN); } } else { // Two-Pass if (int tp = static_cast(obs_data_get_int(settings, ST_KEY_RATECONTROL_TWOPASS)); tp > -1) { av_opt_set_int(context->priv_data, "2pass", tp ? 1 : 0, AV_OPT_SEARCH_CHILDREN); } } // Look Ahead # of Frames int la = static_cast(obs_data_get_int(settings, ST_KEY_RATECONTROL_LOOKAHEAD)); if (!streamfx::util::is_tristate_default(la)) { av_opt_set_int(context->priv_data, "rc-lookahead", la, AV_OPT_SEARCH_CHILDREN); } // Adaptive I-Frames if (int64_t adapt_i = obs_data_get_int(settings, ST_KEY_RATECONTROL_ADAPTIVEI); !streamfx::util::is_tristate_default(adapt_i) && (la != 0)) { // no-scenecut is inverted compared to our UI. av_opt_set_int(context->priv_data, "no-scenecut", 1 - adapt_i, AV_OPT_SEARCH_CHILDREN); } // Adaptive B-Frames if (std::string_view("h264_nvenc") == codec->name) { if (int64_t adapt_b = obs_data_get_int(settings, ST_KEY_RATECONTROL_ADAPTIVEB); !streamfx::util::is_tristate_default(adapt_b) && (la != 0)) { av_opt_set_int(context->priv_data, "b_adapt", adapt_b, AV_OPT_SEARCH_CHILDREN); } } } if (have_bitrate) { int64_t v = obs_data_get_int(settings, ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET); // Allow OBS to specify a maximum allowed bitrate. if (obs_data_get_int(settings, "bitrate") != obs_data_get_default_int(settings, "bitrate")) { // obs_data_has_user_value(X, Y) is also true if obs_data_set_Z(X, Y, obs_data_get_Z(X, Y)) v = std::clamp(v, -1, obs_data_get_int(settings, "bitrate")); } if (v > -1) { context->bit_rate = static_cast(v * 1000); } } else { context->bit_rate = 0; } if (have_bitrate_range) { if (int64_t max = obs_data_get_int(settings, ST_KEY_RATECONTROL_LIMITS_BITRATE_MAXIMUM); max > -1) { context->rc_max_rate = static_cast(max * 1000); } else { context->rc_max_rate = context->bit_rate; } context->rc_min_rate = context->bit_rate; } else { context->rc_min_rate = context->bit_rate; context->rc_max_rate = context->bit_rate; } { // Support for OBS Studio obs_data_set_int(settings, "bitrate", context->rc_max_rate); } // Buffer Size if (have_bitrate || have_bitrate_range) { if (int64_t v = obs_data_get_int(settings, ST_KEY_RATECONTROL_LIMITS_BUFFERSIZE); v > -1) context->rc_buffer_size = static_cast(v * 1000); } else { context->rc_buffer_size = 0; } if (!context->internal) { // Quality Limits if (have_qp_limits) { if (int qmin = static_cast(obs_data_get_int(settings, ST_KEY_RATECONTROL_QP_MINIMUM)); qmin > -1) context->qmin = qmin; if (int qmax = static_cast(obs_data_get_int(settings, ST_KEY_RATECONTROL_QP_MAXIMUM)); qmax > -1) context->qmax = qmax; } else { context->qmin = -1; context->qmax = -1; } // Quality Target if (have_quality) { if (double_t v = obs_data_get_double(settings, ST_KEY_RATECONTROL_LIMITS_QUALITY); v > 0) { av_opt_set_double(context->priv_data, "cq", v, AV_OPT_SEARCH_CHILDREN); } } else { av_opt_set_double(context->priv_data, "cq", 0, AV_OPT_SEARCH_CHILDREN); } // QP Settings if (have_qp) { if (int64_t qp = obs_data_get_int(settings, ST_KEY_RATECONTROL_QP_I); qp > -1) av_opt_set_int(context->priv_data, "init_qpI", static_cast(qp), AV_OPT_SEARCH_CHILDREN); if (int64_t qp = obs_data_get_int(settings, ST_KEY_RATECONTROL_QP_P); qp > -1) av_opt_set_int(context->priv_data, "init_qpP", static_cast(qp), AV_OPT_SEARCH_CHILDREN); if (int64_t qp = obs_data_get_int(settings, ST_KEY_RATECONTROL_QP_B); qp > -1) av_opt_set_int(context->priv_data, "init_qpB", static_cast(qp), AV_OPT_SEARCH_CHILDREN); } } } if (!context->internal) { // AQ int64_t saq = obs_data_get_int(settings, ST_KEY_AQ_SPATIAL); int64_t taq = obs_data_get_int(settings, ST_KEY_AQ_TEMPORAL); if (strcmp(codec->name, "h264_nvenc") == 0) { if (!streamfx::util::is_tristate_default(saq)) av_opt_set_int(context->priv_data, "spatial-aq", saq, AV_OPT_SEARCH_CHILDREN); if (!streamfx::util::is_tristate_default(taq)) av_opt_set_int(context->priv_data, "temporal-aq", taq, AV_OPT_SEARCH_CHILDREN); } else { if (!streamfx::util::is_tristate_default(saq)) av_opt_set_int(context->priv_data, "spatial_aq", saq, AV_OPT_SEARCH_CHILDREN); if (!streamfx::util::is_tristate_default(taq)) av_opt_set_int(context->priv_data, "temporal_aq", taq, AV_OPT_SEARCH_CHILDREN); } if (streamfx::util::is_tristate_enabled(saq)) if (int64_t aqs = obs_data_get_int(settings, ST_KEY_AQ_STRENGTH); aqs > -1) av_opt_set_int(context->priv_data, "aq-strength", static_cast(aqs), AV_OPT_SEARCH_CHILDREN); } if (!context->internal) { // Other if (int64_t bf = obs_data_get_int(settings, ST_KEY_OTHER_BFRAMES); bf > -1) av_opt_set_int(context, "bf", bf, AV_OPT_SEARCH_CHILDREN); if (int64_t zl = obs_data_get_int(settings, ST_KEY_OTHER_ZEROLATENCY); !streamfx::util::is_tristate_default(zl)) av_opt_set_int(context->priv_data, "zerolatency", zl, AV_OPT_SEARCH_CHILDREN); if (int64_t nrp = obs_data_get_int(settings, ST_KEY_OTHER_NONREFERENCEPFRAMES); !streamfx::util::is_tristate_default(nrp)) av_opt_set_int(context->priv_data, "nonref_p", nrp, AV_OPT_SEARCH_CHILDREN); if (int64_t v = obs_data_get_int(settings, ST_KEY_OTHER_REFERENCEFRAMES); v > -1) av_opt_set_int(context, "refs", v, AV_OPT_SEARCH_CHILDREN); int64_t wp = obs_data_get_int(settings, ST_KEY_OTHER_WEIGHTEDPREDICTION); if ((context->max_b_frames > 0) && streamfx::util::is_tristate_enabled(wp)) { DLOG_WARNING("[%s] Weighted Prediction disabled because of B-Frames being used.", codec->name); av_opt_set_int(context->priv_data, "weighted_pred", 0, AV_OPT_SEARCH_CHILDREN); } else if (!streamfx::util::is_tristate_default(wp)) { av_opt_set_int(context->priv_data, "weighted_pred", wp, AV_OPT_SEARCH_CHILDREN); } if (const char* v = obs_data_get_string(settings, ST_KEY_OTHER_BFRAMEREFERENCEMODE); (v != nullptr) && (v[0] != '\0')) { av_opt_set(context->priv_data, "b_ref_mode", v, AV_OPT_SEARCH_CHILDREN); } if (auto v = obs_data_get_int(settings, ST_KEY_OTHER_LOWDELAYKEYFRAMESCALE); v > -1) { av_opt_set_int(context->priv_data, "ldkfs", v, AV_OPT_SEARCH_CHILDREN); } } } void nvenc::log_options(obs_data_t*, const AVCodec* codec, AVCodecContext* context) { using namespace ::streamfx::ffmpeg; DLOG_INFO("[%s] NVIDIA NVENC:", codec->name); tools::print_av_option_string2(context, "preset", " Preset", [](int64_t v, std::string_view o) { return std::string(o); }); tools::print_av_option_string2(context, "rc", " Rate Control", [](int64_t v, std::string_view o) { return std::string(o); }); tools::print_av_option_bool(context, "2pass", " Two Pass"); tools::print_av_option_string2(context, "multipass", " Multi-Pass", [](int64_t v, std::string_view o) { return std::string(o); }); tools::print_av_option_int(context, "rc-lookahead", " Look-Ahead", "Frames"); tools::print_av_option_bool(context, "no-scenecut", " Adaptive I-Frames", true); if (strcmp(codec->name, "h264_nvenc") == 0) tools::print_av_option_bool(context, "b_adapt", " Adaptive B-Frames"); DLOG_INFO("[%s] Bitrate:", codec->name); tools::print_av_option_int(context, "b", " Target", "bits/sec"); tools::print_av_option_int(context, "minrate", " Minimum", "bits/sec"); tools::print_av_option_int(context, "maxrate", " Maximum", "bits/sec"); tools::print_av_option_int(context, "bufsize", " Buffer", "bits"); DLOG_INFO("[%s] Quality:", codec->name); tools::print_av_option_int(context, "cq", " Target", ""); tools::print_av_option_int(context, "qmin", " Minimum", ""); tools::print_av_option_int(context, "qmax", " Maximum", ""); DLOG_INFO("[%s] Quantization Parameters:", codec->name); tools::print_av_option_int(context, "init_qpI", " I-Frame", ""); tools::print_av_option_int(context, "init_qpP", " P-Frame", ""); tools::print_av_option_int(context, "init_qpB", " B-Frame", ""); tools::print_av_option_int(context, "qp_cb_offset", " CB Offset", ""); tools::print_av_option_int(context, "qp_cr_offset", " CR Offset", ""); tools::print_av_option_int(context, "bf", " B-Frames", "Frames"); tools::print_av_option_string2(context, "b_ref_mode", " Reference Mode", [](int64_t v, std::string_view o) { return std::string(o); }); DLOG_INFO("[%s] Adaptive Quantization:", codec->name); if (strcmp(codec->name, "h264_nvenc") == 0) { tools::print_av_option_bool(context, "spatial-aq", " Spatial AQ"); tools::print_av_option_int(context, "aq-strength", " Strength", ""); tools::print_av_option_bool(context, "temporal-aq", " Temporal AQ"); } else { tools::print_av_option_bool(context, "spatial_aq", " Spatial AQ"); tools::print_av_option_int(context, "aq-strength", " Strength", ""); tools::print_av_option_bool(context, "temporal_aq", " Temporal AQ"); } DLOG_INFO("[%s] Other:", codec->name); tools::print_av_option_bool(context, "zerolatency", " Zero Latency"); tools::print_av_option_bool(context, "weighted_pred", " Weighted Prediction"); tools::print_av_option_bool(context, "nonref_p", " Non-reference P-Frames"); tools::print_av_option_int(context, "refs", " Reference Frames", "Frames"); tools::print_av_option_bool(context, "strict_gop", " Strict GOP"); tools::print_av_option_bool(context, "aud", " Access Unit Delimiters"); tools::print_av_option_bool(context, "bluray-compat", " Bluray Compatibility"); tools::print_av_option_bool(context, "a53cc", " A53 Closed Captions"); tools::print_av_option_int(context, "dpb_size", " DPB Size", "Frames"); tools::print_av_option_int(context, "ldkfs", " DPB Size", "Frames"); tools::print_av_option_bool(context, "extra_sei", " Extra SEI Data"); tools::print_av_option_bool(context, "udu_sei", " User SEI Data"); tools::print_av_option_bool(context, "intra-refresh", " Intra-Refresh"); tools::print_av_option_bool(context, "single-slice-intra-refresh", " Single Slice Intra-Refresh"); tools::print_av_option_bool(context, "constrained-encoding", " Constrained Encoding"); } void streamfx::encoder::ffmpeg::handler::nvenc::migrate(obs_data_t* settings, uint64_t version, const AVCodec* codec, AVCodecContext* context) { // Only test for A.B.C in A.B.C.D version = version & STREAMFX_MASK_UPDATE; #define COPY_UNSET(TYPE, FROM, TO) \ if (obs_data_has_user_value(settings, FROM)) { \ obs_data_set_##TYPE(settings, TO, obs_data_get_##TYPE(settings, FROM)); \ obs_data_unset_user_value(settings, FROM); \ } if (version <= STREAMFX_MAKE_VERSION(0, 8, 0, 0)) { COPY_UNSET(int, "RateControl.Bitrate.Target", ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET); COPY_UNSET(int, "RateControl.Bitrate.Maximum", ST_KEY_RATECONTROL_LIMITS_BITRATE_TARGET); COPY_UNSET(int, "RateControl.BufferSize", ST_KEY_RATECONTROL_LIMITS_BUFFERSIZE); COPY_UNSET(int, "RateControl.Quality.Minimum", ST_KEY_RATECONTROL_QP_MINIMUM); COPY_UNSET(int, "RateControl.Quality.Maximum", ST_KEY_RATECONTROL_QP_MAXIMUM); COPY_UNSET(double, "RateControl.Quality.Target", ST_KEY_RATECONTROL_LIMITS_QUALITY); } if (version < STREAMFX_MAKE_VERSION(0, 11, 0, 0)) { obs_data_unset_user_value(settings, "Other.AccessUnitDelimiter"); obs_data_unset_user_value(settings, "Other.DecodedPictureBufferSize"); } if (version < STREAMFX_MAKE_VERSION(0, 11, 1, 0)) { // Preset if (auto v = obs_data_get_int(settings, ST_KEY_PRESET); v != -1) { std::map preset{ {0, "default"}, {1, "slow"}, {2, "medium"}, {3, "fast"}, {4, "hp"}, {5, "hq"}, {6, "bd"}, {7, "ll"}, {8, "llhq"}, {9, "llhp"}, {10, "lossless"}, {11, "losslesshp"}, }; if (auto k = preset.find(v); k != preset.end()) { obs_data_set_string(settings, ST_KEY_PRESET, k->second.data()); } } // Rate Control Mode if (auto v = obs_data_get_int(settings, ST_KEY_RATECONTROL_MODE); v != -1) { if (!obs_data_has_user_value(settings, ST_KEY_RATECONTROL_MODE)) v = 4; switch (v) { case 0: // CQP obs_data_set_string(settings, ST_KEY_RATECONTROL_MODE, "constqp"); break; case 2: // VBR_HQ obs_data_set_int(settings, ST_KEY_RATECONTROL_TWOPASS, 1); obs_data_set_string(settings, ST_KEY_RATECONTROL_MULTIPASS, "qres"); case 1: // VBR obs_data_set_string(settings, ST_KEY_RATECONTROL_MODE, "vbr"); break; case 5: // CBR_LD_HQ obs_data_set_int(settings, ST_KEY_OTHER_LOWDELAYKEYFRAMESCALE, 1); case 4: // CBR_HQ obs_data_set_int(settings, ST_KEY_RATECONTROL_TWOPASS, 1); obs_data_set_string(settings, ST_KEY_RATECONTROL_MULTIPASS, "qres"); case 3: // CBR obs_data_set_string(settings, ST_KEY_RATECONTROL_MODE, "cbr"); break; } } // Target Quality if (auto v = obs_data_get_double(settings, ST_KEY_RATECONTROL_LIMITS_QUALITY); v > 0) { obs_data_set_double(settings, ST_KEY_RATECONTROL_LIMITS_QUALITY, (v / 100.) * 51.); } // B-Frame Reference Modes if (auto v = obs_data_get_int(settings, ST_KEY_OTHER_BFRAMEREFERENCEMODE); v != -1) { std::map preset{ {0, "default"}, {1, "each"}, {2, "middle"}, }; if (auto k = preset.find(v); k != preset.end()) { obs_data_set_string(settings, ST_KEY_OTHER_BFRAMEREFERENCEMODE, k->second.data()); } } } #undef COPY_UNSET }