273 lines
7.5 KiB
C++
273 lines
7.5 KiB
C++
static bool str_to_bool(const char *str)
|
|
{
|
|
if (!str)
|
|
return false;
|
|
if (*str == '1')
|
|
return true;
|
|
if (*str == '0')
|
|
return false;
|
|
if (astrcmpi(str, "true") == 0)
|
|
return true;
|
|
if (astrcmpi(str, "false") == 0)
|
|
return false;
|
|
return false;
|
|
}
|
|
|
|
static void amf_apply_opt(amf_base *enc, obs_option *opt)
|
|
{
|
|
bool avc = enc->codec == amf_codec_type::AVC;
|
|
bool hevc = enc->codec == amf_codec_type::HEVC;
|
|
|
|
if (strcmp(opt->name, "g") == 0 || strcmp(opt->name, "keyint") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
if (enc->codec == amf_codec_type::AVC)
|
|
set_avc_opt(IDR_PERIOD, val);
|
|
else
|
|
set_hevc_opt(NUM_GOPS_PER_IDR, val);
|
|
|
|
} else if (strcmp(opt->name, "usage") == 0) {
|
|
|
|
if (strcmp(opt->value, "transcoding") == 0) {
|
|
set_enum_opt(USAGE, TRANSCONDING);
|
|
} else if (strcmp(opt->value, "ultralowlatency") == 0) {
|
|
set_enum_opt(USAGE, ULTRA_LOW_LATENCY);
|
|
} else if (strcmp(opt->value, "lowlatency") == 0) {
|
|
set_enum_opt(USAGE, LOW_LATENCY);
|
|
} else if (strcmp(opt->value, "webcam") == 0) {
|
|
set_enum_opt(USAGE, WEBCAM);
|
|
} else {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
}
|
|
|
|
} else if (strcmp(opt->name, "profile") == 0) {
|
|
|
|
if (strcmp(opt->value, "main") == 0) {
|
|
set_enum_opt(PROFILE, MAIN);
|
|
} else if (enc->codec != amf_codec_type::AVC) {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
return;
|
|
}
|
|
|
|
if (strcmp(opt->value, "high") == 0) {
|
|
set_opt(PROFILE, AMF_VIDEO_ENCODER_PROFILE_HIGH);
|
|
} else if (strcmp(opt->value, "constrained_baseline") == 0) {
|
|
set_opt(PROFILE,
|
|
AMF_VIDEO_ENCODER_PROFILE_CONSTRAINED_BASELINE);
|
|
} else if (strcmp(opt->value, "constrained_high") == 0) {
|
|
set_opt(PROFILE,
|
|
AMF_VIDEO_ENCODER_PROFILE_CONSTRAINED_HIGH);
|
|
} else {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
}
|
|
|
|
} else if (strcmp(opt->name, "level") == 0) {
|
|
|
|
std::string val = opt->value;
|
|
size_t pos = val.find('.');
|
|
if (pos != std::string::npos)
|
|
val.erase(pos, 1);
|
|
|
|
int level = std::stoi(val);
|
|
set_opt(PROFILE_LEVEL, level);
|
|
|
|
} else if (strcmp(opt->name, "quality") == 0) {
|
|
|
|
if (strcmp(opt->value, "speed") == 0) {
|
|
set_enum_opt(QUALITY_PRESET, SPEED);
|
|
} else if (strcmp(opt->value, "balanced") == 0) {
|
|
set_enum_opt(QUALITY_PRESET, BALANCED);
|
|
} else if (strcmp(opt->value, "quality") == 0) {
|
|
set_enum_opt(QUALITY_PRESET, QUALITY);
|
|
} else {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
}
|
|
|
|
} else if (strcmp(opt->name, "rc") == 0) {
|
|
|
|
if (strcmp(opt->value, "cqp") == 0) {
|
|
set_enum_opt(RATE_CONTROL_METHOD, CONSTANT_QP);
|
|
} else if (strcmp(opt->value, "cbr") == 0) {
|
|
set_enum_opt(RATE_CONTROL_METHOD, CBR);
|
|
} else if (strcmp(opt->value, "vbr_peak") == 0) {
|
|
set_enum_opt(RATE_CONTROL_METHOD, PEAK_CONSTRAINED_VBR);
|
|
} else if (strcmp(opt->value, "vbr_latency") == 0) {
|
|
set_enum_opt(RATE_CONTROL_METHOD,
|
|
LATENCY_CONSTRAINED_VBR);
|
|
} else {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
}
|
|
|
|
} else if (strcmp(opt->name, "enforce_hrd") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_opt(ENFORCE_HRD, val);
|
|
|
|
} else if (strcmp(opt->name, "filler_data") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_opt(FILLER_DATA_ENABLE, val);
|
|
|
|
} else if (strcmp(opt->name, "vbaq") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_opt(ENABLE_VBAQ, val);
|
|
|
|
} else if (strcmp(opt->name, "qp_i") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_opt(QP_I, val);
|
|
|
|
} else if (strcmp(opt->name, "qp_p") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_opt(QP_P, val);
|
|
|
|
} else if (strcmp(opt->name, "me_half_pel") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_opt(MOTION_HALF_PIXEL, val);
|
|
|
|
} else if (strcmp(opt->name, "me_quarter_pel") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_opt(MOTION_QUARTERPIXEL, val);
|
|
|
|
} else if (strcmp(opt->name, "aud") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_opt(INSERT_AUD, val);
|
|
|
|
} else if (strcmp(opt->name, "max_au_size") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_opt(MAX_AU_SIZE, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "preanalysis") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_avc_property(enc, PREENCODE_ENABLE, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "qp_b") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_avc_property(enc, QP_B, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "frame_skipping") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_avc_property(enc, RATE_CONTROL_SKIP_FRAME_ENABLE, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "header_spacing") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_avc_property(enc, HEADER_INSERTION_SPACING, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "bf_delta_qp") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_avc_property(enc, B_PIC_DELTA_QP, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "bf_ref") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_avc_property(enc, B_REFERENCE_ENABLE, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "bf_ref_delta_qp") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_avc_property(enc, REF_B_PIC_DELTA_QP, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "intra_refresh_mb") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_avc_property(enc, INTRA_REFRESH_NUM_MBS_PER_SLOT, val);
|
|
|
|
} else if (avc && strcmp(opt->name, "coder") == 0) {
|
|
|
|
if (strcmp(opt->value, "auto") == 0) {
|
|
set_avc_opt(CABAC_ENABLE, AMF_VIDEO_ENCODER_UNDEFINED);
|
|
} else if (strcmp(opt->value, "cavlc") == 0) {
|
|
set_avc_opt(CABAC_ENABLE, AMF_VIDEO_ENCODER_CALV);
|
|
} else if (strcmp(opt->value, "cabac") == 0) {
|
|
set_avc_opt(CABAC_ENABLE, AMF_VIDEO_ENCODER_CABAC);
|
|
} else {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
}
|
|
|
|
} else if (hevc && strcmp(opt->name, "profile_tier") == 0) {
|
|
|
|
if (strcmp(opt->value, "main") == 0) {
|
|
set_hevc_enum(TIER, MAIN);
|
|
} else if (strcmp(opt->value, "high") == 0) {
|
|
set_hevc_enum(TIER, HIGH);
|
|
} else {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
}
|
|
|
|
} else if (hevc && strcmp(opt->name, "header_insertion_mode") == 0) {
|
|
|
|
if (strcmp(opt->value, "none") == 0) {
|
|
set_hevc_enum(HEADER_INSERTION_MODE, NONE);
|
|
} else if (strcmp(opt->value, "gop") == 0) {
|
|
set_hevc_enum(HEADER_INSERTION_MODE, GOP_ALIGNED);
|
|
} else if (strcmp(opt->value, "idr") == 0) {
|
|
set_hevc_enum(HEADER_INSERTION_MODE, IDR_ALIGNED);
|
|
} else {
|
|
warn("Invalid value for %s: %s", opt->name, opt->value);
|
|
}
|
|
|
|
} else if (hevc && strcmp(opt->name, "skip_frame") == 0) {
|
|
|
|
bool val = str_to_bool(opt->value);
|
|
set_hevc_property(enc, RATE_CONTROL_SKIP_FRAME_ENABLE, val);
|
|
|
|
} else if (hevc && strcmp(opt->name, "gops_per_idr") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_hevc_property(enc, NUM_GOPS_PER_IDR, val);
|
|
|
|
} else if (hevc && strcmp(opt->name, "min_qp_i") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_hevc_property(enc, MIN_QP_I, val);
|
|
|
|
} else if (hevc && strcmp(opt->name, "max_qp_i") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_hevc_property(enc, MAX_QP_I, val);
|
|
|
|
} else if (hevc && strcmp(opt->name, "min_qp_p") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_hevc_property(enc, MIN_QP_P, val);
|
|
|
|
} else if (hevc && strcmp(opt->name, "max_qp_p") == 0) {
|
|
|
|
int val = atoi(opt->value);
|
|
set_hevc_property(enc, MAX_QP_P, val);
|
|
} else {
|
|
wchar_t wname[256];
|
|
int val;
|
|
bool is_bool = false;
|
|
|
|
if (astrcmpi(opt->value, "true") == 0) {
|
|
is_bool = true;
|
|
val = 1;
|
|
} else if (astrcmpi(opt->value, "false") == 0) {
|
|
is_bool = true;
|
|
val = 0;
|
|
} else {
|
|
val = atoi(opt->value);
|
|
}
|
|
|
|
os_utf8_to_wcs(opt->name, 0, wname, _countof(wname));
|
|
if (is_bool) {
|
|
bool bool_val = (bool)val;
|
|
set_amf_property(enc, wname, bool_val);
|
|
} else {
|
|
set_amf_property(enc, wname, val);
|
|
}
|
|
}
|
|
}
|