22 #include <wels/codec_api.h>
23 #include <wels/codec_ver.h>
37 #if !OPENH264_VER_AT_LEAST(1, 6)
38 #define SM_SIZELIMITED_SLICE SM_DYN_SLICE
41 #define TARGET_BITRATE_DEFAULT 2*1000*1000
52 #if FF_API_OPENH264_CABAC
61 #define OFFSET(x) offsetof(SVCContext, x)
62 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
63 #define DEPRECATED AV_OPT_FLAG_DEPRECATED
65 #if FF_API_OPENH264_SLICE_MODE
66 #if OPENH264_VER_AT_LEAST(1, 6)
67 {
"slice_mode",
"set slice mode, use slices/max_nal_size",
OFFSET(slice_mode),
AV_OPT_TYPE_INT, { .i64 = SM_FIXEDSLCNUM_SLICE }, SM_SINGLE_SLICE, SM_RESERVED,
VE|
DEPRECATED,
"slice_mode" },
69 {
"slice_mode",
"set slice mode, use slices/max_nal_size",
OFFSET(slice_mode),
AV_OPT_TYPE_INT, { .i64 = SM_AUTO_SLICE }, SM_SINGLE_SLICE, SM_RESERVED,
VE|
DEPRECATED,
"slice_mode" },
71 {
"fixed",
"a fixed number of slices", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_FIXEDSLCNUM_SLICE }, 0, 0,
VE,
"slice_mode" },
72 #if OPENH264_VER_AT_LEAST(1, 6)
76 {
"rowmb",
"one slice per row of macroblocks", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_ROWMB_SLICE }, 0, 0,
VE,
"slice_mode" },
77 {
"auto",
"automatic number of slices according to number of threads", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_AUTO_SLICE }, 0, 0,
VE,
"slice_mode" },
78 {
"dyn",
"Dynamic slicing", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_DYN_SLICE }, 0, 0,
VE,
"slice_mode" },
83 #define PROFILE(name, value) name, NULL, 0, AV_OPT_TYPE_CONST, { .i64 = value }, 0, 0, VE, "profile"
88 {
"max_nal_size",
"set maximum NAL size in bytes",
OFFSET(max_nal_size),
AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
VE },
89 {
"allow_skip_frames",
"allow skipping frames to hit the target bitrate",
OFFSET(skip_frames),
AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1,
VE },
90 #if FF_API_OPENH264_CABAC
100 {
"rc_mode",
"Select rate control mode",
OFFSET(
rc_mode),
AV_OPT_TYPE_INT, { .i64 = RC_QUALITY_MODE }, RC_OFF_MODE, RC_TIMESTAMP_MODE,
VE,
"rc_mode" },
101 {
"off",
"bit rate control off", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_OFF_MODE }, 0, 0,
VE,
"rc_mode" },
102 {
"quality",
"quality mode", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_QUALITY_MODE }, 0, 0,
VE,
"rc_mode" },
103 {
"bitrate",
"bitrate mode", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_BITRATE_MODE }, 0, 0,
VE,
"rc_mode" },
104 {
"buffer",
"using buffer status to adjust the video quality (no bitrate control)", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_BUFFERBASED_MODE }, 0, 0,
VE,
"rc_mode" },
105 #if OPENH264_VER_AT_LEAST(1, 4)
106 {
"timestamp",
"bit rate control based on timestamp", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_TIMESTAMP_MODE }, 0, 0,
VE,
"rc_mode" },
124 WelsDestroySVCEncoder(
s->encoder);
133 SEncParamExt param = { 0 };
136 WelsTraceCallback callback_function;
142 if (WelsCreateSVCEncoder(&
s->encoder)) {
148 log_level = WELS_LOG_DETAIL;
149 (*
s->encoder)->SetOption(
s->encoder, ENCODER_OPTION_TRACE_LEVEL, &log_level);
153 (*
s->encoder)->SetOption(
s->encoder, ENCODER_OPTION_TRACE_CALLBACK, &callback_function);
156 (*
s->encoder)->SetOption(
s->encoder, ENCODER_OPTION_TRACE_CALLBACK_CONTEXT, &avctx);
158 (*
s->encoder)->GetDefaultParams(
s->encoder, ¶m);
165 "Could not set framerate for libopenh264enc: integer overflow\n");
170 param.iPicWidth = avctx->
width;
171 param.iPicHeight = avctx->
height;
174 param.iRCMode =
s->rc_mode;
175 if (avctx->
qmax >= 0)
177 if (avctx->
qmin >= 0)
178 param.iMinQp =
av_clip(avctx->
qmin, 1, param.iMaxQp);
179 param.iTemporalLayerNum = 1;
180 param.iSpatialLayerNum = 1;
181 param.bEnableDenoise = 0;
182 param.bEnableBackgroundDetection = 1;
183 param.bEnableAdaptiveQuant = 1;
184 param.bEnableFrameSkip =
s->skip_frames;
185 param.bEnableLongTermReference = 0;
186 param.iLtrMarkPeriod = 30;
188 param.uiIntraPeriod = avctx->
gop_size;
189 #if OPENH264_VER_AT_LEAST(1, 4)
190 param.eSpsPpsIdStrategy = CONSTANT_ID;
192 param.bEnableSpsPpsIdAddition = 0;
194 param.bPrefixNalAddingCtrl = 0;
195 param.iLoopFilterDisableIdc = !
s->loopfilter;
196 param.iEntropyCodingModeFlag = 0;
210 "Unsupported avctx->profile: %d.\n", avctx->
profile);
216 #if OPENH264_VER_AT_LEAST(1, 8)
222 switch (
s->profile) {
223 #if OPENH264_VER_AT_LEAST(1, 8)
225 param.iEntropyCodingModeFlag = 1;
227 "select EProfileIdc PRO_HIGH in libopenh264.\n");
231 param.iEntropyCodingModeFlag = 1;
233 "select EProfileIdc PRO_MAIN in libopenh264.\n");
238 param.iEntropyCodingModeFlag = 0;
240 "select EProfileIdc PRO_BASELINE in libopenh264.\n");
243 param.iEntropyCodingModeFlag = 0;
245 "select EProfileIdc PRO_BASELINE in libopenh264.\n");
249 param.sSpatialLayers[0].iVideoWidth = param.iPicWidth;
250 param.sSpatialLayers[0].iVideoHeight = param.iPicHeight;
251 param.sSpatialLayers[0].fFrameRate = param.fMaxFrameRate;
252 param.sSpatialLayers[0].iSpatialBitrate = param.iTargetBitrate;
253 param.sSpatialLayers[0].iMaxSpatialBitrate = param.iMaxBitrate;
255 #if OPENH264_VER_AT_LEAST(1, 7)
260 { 1, 1 }, { 12, 11 }, { 10, 11 }, { 16, 11 },
261 { 40, 33 }, { 24, 11 }, { 20, 11 }, { 32, 11 },
262 { 80, 33 }, { 18, 11 }, { 15, 11 }, { 64, 33 },
265 static const ESampleAspectRatio asp_idc[] = {
267 ASP_1x1, ASP_12x11, ASP_10x11, ASP_16x11,
268 ASP_40x33, ASP_24x11, ASP_20x11, ASP_32x11,
269 ASP_80x33, ASP_18x11, ASP_15x11, ASP_64x33,
278 if (num == sar_idc[
i].num &&
279 den == sar_idc[
i].den)
283 param.sSpatialLayers[0].eAspectRatio = ASP_EXT_SAR;
284 param.sSpatialLayers[0].sAspectRatioExtWidth = num;
285 param.sSpatialLayers[0].sAspectRatioExtHeight = den;
287 param.sSpatialLayers[0].eAspectRatio = asp_idc[
i];
289 param.sSpatialLayers[0].bAspectRatioPresent =
true;
291 param.sSpatialLayers[0].bAspectRatioPresent =
false;
295 if ((avctx->
slices > 1) && (
s->max_nal_size)) {
297 "Invalid combination -slices %d and -max_nal_size %d.\n",
298 avctx->
slices,
s->max_nal_size);
303 s->slice_mode = SM_FIXEDSLCNUM_SLICE;
308 #if OPENH264_VER_AT_LEAST(1, 6)
309 param.sSpatialLayers[0].sSliceArgument.uiSliceMode =
s->slice_mode;
310 param.sSpatialLayers[0].sSliceArgument.uiSliceNum = avctx->
slices;
312 param.sSpatialLayers[0].sSliceCfg.uiSliceMode =
s->slice_mode;
313 param.sSpatialLayers[0].sSliceCfg.sSliceArgument.uiSliceNum = avctx->
slices;
315 if (avctx->
slices == 0 &&
s->slice_mode == SM_FIXEDSLCNUM_SLICE)
319 if (
s->max_nal_size) {
320 param.uiMaxNalSize =
s->max_nal_size;
321 #if OPENH264_VER_AT_LEAST(1, 6)
322 param.sSpatialLayers[0].sSliceArgument.uiSliceSizeConstraint =
s->max_nal_size;
324 param.sSpatialLayers[0].sSliceCfg.sSliceArgument.uiSliceSizeConstraint =
s->max_nal_size;
328 "specify a valid max_nal_size to use -slice_mode dyn\n");
333 if ((*
s->encoder)->InitializeExt(
s->encoder, ¶m) != cmResultSuccess) {
339 SFrameBSInfo fbi = { 0 };
341 (*
s->encoder)->EncodeParameterSets(
s->encoder, &fbi);
342 for (
i = 0;
i < fbi.sLayerInfo[0].iNalCount;
i++)
343 size += fbi.sLayerInfo[0].pNalLengthInByte[
i];
364 SFrameBSInfo fbi = { 0 };
367 SSourcePicture
sp = { 0 };
368 int size = 0, layer, first_layer = 0;
369 int layer_size[MAX_LAYER_NUM_OF_FRAME] = { 0 };
371 sp.iColorFormat = videoFormatI420;
372 for (
i = 0;
i < 3;
i++) {
380 (*
s->encoder)->ForceIntraFrame(
s->encoder,
true);
383 encoded = (*
s->encoder)->EncodeFrame(
s->encoder, &
sp, &fbi);
384 if (encoded != cmResultSuccess) {
388 if (fbi.eFrameType == videoFrameTypeSkip) {
399 first_layer = fbi.iLayerNum - 1;
401 for (layer = first_layer; layer < fbi.iLayerNum; layer++) {
402 for (
i = 0;
i < fbi.sLayerInfo[layer].iNalCount;
i++)
403 layer_size[layer] += fbi.sLayerInfo[layer].pNalLengthInByte[
i];
404 size += layer_size[layer];
406 av_log(avctx,
AV_LOG_DEBUG,
"%d slices\n", fbi.sLayerInfo[fbi.iLayerNum - 1].iNalCount);
412 for (layer = first_layer; layer < fbi.iLayerNum; layer++) {
413 memcpy(avpkt->
data +
size, fbi.sLayerInfo[layer].pBsBuf, layer_size[layer]);
414 size += layer_size[layer];
417 if (fbi.eFrameType == videoFrameTypeIDR)
432 .
name =
"libopenh264",
446 .priv_class = &
class,
447 .wrapper_name =
"libopenh264",