22 #include <wels/codec_api.h>
23 #include <wels/codec_ver.h>
38 #if !OPENH264_VER_AT_LEAST(1, 6)
39 #define SM_SIZELIMITED_SLICE SM_DYN_SLICE
42 #define TARGET_BITRATE_DEFAULT 2*1000*1000
53 #if FF_API_OPENH264_CABAC
62 #define OFFSET(x) offsetof(SVCContext, x)
63 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
64 #define DEPRECATED AV_OPT_FLAG_DEPRECATED
66 #if FF_API_OPENH264_SLICE_MODE
67 #if OPENH264_VER_AT_LEAST(1, 6)
68 {
"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" },
70 {
"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" },
72 {
"fixed",
"a fixed number of slices", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_FIXEDSLCNUM_SLICE }, 0, 0,
VE,
"slice_mode" },
73 #if OPENH264_VER_AT_LEAST(1, 6)
77 {
"rowmb",
"one slice per row of macroblocks", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_ROWMB_SLICE }, 0, 0,
VE,
"slice_mode" },
78 {
"auto",
"automatic number of slices according to number of threads", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_AUTO_SLICE }, 0, 0,
VE,
"slice_mode" },
79 {
"dyn",
"Dynamic slicing", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_DYN_SLICE }, 0, 0,
VE,
"slice_mode" },
84 #define PROFILE(name, value) name, NULL, 0, AV_OPT_TYPE_CONST, { .i64 = value }, 0, 0, VE, "profile"
89 {
"max_nal_size",
"set maximum NAL size in bytes",
OFFSET(max_nal_size),
AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
VE },
90 {
"allow_skip_frames",
"allow skipping frames to hit the target bitrate",
OFFSET(skip_frames),
AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1,
VE },
91 #if FF_API_OPENH264_CABAC
101 {
"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" },
102 {
"off",
"bit rate control off", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_OFF_MODE }, 0, 0,
VE,
"rc_mode" },
103 {
"quality",
"quality mode", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_QUALITY_MODE }, 0, 0,
VE,
"rc_mode" },
104 {
"bitrate",
"bitrate mode", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_BITRATE_MODE }, 0, 0,
VE,
"rc_mode" },
105 {
"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" },
106 #if OPENH264_VER_AT_LEAST(1, 4)
107 {
"timestamp",
"bit rate control based on timestamp", 0,
AV_OPT_TYPE_CONST, { .i64 = RC_TIMESTAMP_MODE }, 0, 0,
VE,
"rc_mode" },
125 WelsDestroySVCEncoder(
s->encoder);
134 SEncParamExt param = { 0 };
137 WelsTraceCallback callback_function;
143 if (WelsCreateSVCEncoder(&
s->encoder)) {
149 log_level = WELS_LOG_DETAIL;
150 (*
s->encoder)->SetOption(
s->encoder, ENCODER_OPTION_TRACE_LEVEL, &log_level);
154 (*
s->encoder)->SetOption(
s->encoder, ENCODER_OPTION_TRACE_CALLBACK, &callback_function);
157 (*
s->encoder)->SetOption(
s->encoder, ENCODER_OPTION_TRACE_CALLBACK_CONTEXT, &avctx);
159 (*
s->encoder)->GetDefaultParams(
s->encoder, ¶m);
166 "Could not set framerate for libopenh264enc: integer overflow\n");
171 param.iPicWidth = avctx->
width;
172 param.iPicHeight = avctx->
height;
175 param.iRCMode =
s->rc_mode;
176 if (avctx->
qmax >= 0)
178 if (avctx->
qmin >= 0)
179 param.iMinQp =
av_clip(avctx->
qmin, 1, param.iMaxQp);
180 param.iTemporalLayerNum = 1;
181 param.iSpatialLayerNum = 1;
182 param.bEnableDenoise = 0;
183 param.bEnableBackgroundDetection = 1;
184 param.bEnableAdaptiveQuant = 1;
185 param.bEnableFrameSkip =
s->skip_frames;
186 param.bEnableLongTermReference = 0;
187 param.iLtrMarkPeriod = 30;
189 param.uiIntraPeriod = avctx->
gop_size;
190 #if OPENH264_VER_AT_LEAST(1, 4)
191 param.eSpsPpsIdStrategy = CONSTANT_ID;
193 param.bEnableSpsPpsIdAddition = 0;
195 param.bPrefixNalAddingCtrl = 0;
196 param.iLoopFilterDisableIdc = !
s->loopfilter;
197 param.iEntropyCodingModeFlag =
s->coder >= 0 ?
s->coder : 1;
211 "Unsupported avctx->profile: %d.\n", avctx->
profile);
217 #if OPENH264_VER_AT_LEAST(1, 8)
223 switch (
s->profile) {
226 "select EProfileIdc PRO_HIGH in libopenh264.\n",
227 param.iEntropyCodingModeFlag ?
"CABAC" :
"CAVLC");
231 "select EProfileIdc PRO_MAIN in libopenh264.\n",
232 param.iEntropyCodingModeFlag ?
"CABAC" :
"CAVLC");
237 param.iEntropyCodingModeFlag = 0;
239 "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;
254 param.sSpatialLayers[0].uiProfileIdc =
s->profile;
256 #if OPENH264_VER_AT_LEAST(1, 7)
261 { 1, 1 }, { 12, 11 }, { 10, 11 }, { 16, 11 },
262 { 40, 33 }, { 24, 11 }, { 20, 11 }, { 32, 11 },
263 { 80, 33 }, { 18, 11 }, { 15, 11 }, { 64, 33 },
266 static const ESampleAspectRatio asp_idc[] = {
268 ASP_1x1, ASP_12x11, ASP_10x11, ASP_16x11,
269 ASP_40x33, ASP_24x11, ASP_20x11, ASP_32x11,
270 ASP_80x33, ASP_18x11, ASP_15x11, ASP_64x33,
279 if (num == sar_idc[
i].num &&
280 den == sar_idc[
i].den)
284 param.sSpatialLayers[0].eAspectRatio = ASP_EXT_SAR;
285 param.sSpatialLayers[0].sAspectRatioExtWidth = num;
286 param.sSpatialLayers[0].sAspectRatioExtHeight = den;
288 param.sSpatialLayers[0].eAspectRatio = asp_idc[
i];
290 param.sSpatialLayers[0].bAspectRatioPresent =
true;
292 param.sSpatialLayers[0].bAspectRatioPresent =
false;
296 if ((avctx->
slices > 1) && (
s->max_nal_size)) {
298 "Invalid combination -slices %d and -max_nal_size %d.\n",
299 avctx->
slices,
s->max_nal_size);
304 s->slice_mode = SM_FIXEDSLCNUM_SLICE;
309 #if OPENH264_VER_AT_LEAST(1, 6)
310 param.sSpatialLayers[0].sSliceArgument.uiSliceMode =
s->slice_mode;
311 param.sSpatialLayers[0].sSliceArgument.uiSliceNum = avctx->
slices;
313 param.sSpatialLayers[0].sSliceCfg.uiSliceMode =
s->slice_mode;
314 param.sSpatialLayers[0].sSliceCfg.sSliceArgument.uiSliceNum = avctx->
slices;
316 if (avctx->
slices == 0 &&
s->slice_mode == SM_FIXEDSLCNUM_SLICE)
320 if (
s->max_nal_size) {
321 param.uiMaxNalSize =
s->max_nal_size;
322 #if OPENH264_VER_AT_LEAST(1, 6)
323 param.sSpatialLayers[0].sSliceArgument.uiSliceSizeConstraint =
s->max_nal_size;
325 param.sSpatialLayers[0].sSliceCfg.sSliceArgument.uiSliceSizeConstraint =
s->max_nal_size;
329 "specify a valid max_nal_size to use -slice_mode dyn\n");
334 #if OPENH264_VER_AT_LEAST(1, 6)
335 param.sSpatialLayers[0].uiVideoFormat = VF_UNDEF;
337 param.sSpatialLayers[0].bVideoSignalTypePresent =
true;
344 param.sSpatialLayers[0].bVideoSignalTypePresent =
true;
345 param.sSpatialLayers[0].bColorDescriptionPresent =
true;
349 param.sSpatialLayers[0].uiColorMatrix = avctx->
colorspace;
353 param.sSpatialLayers[0].uiTransferCharacteristics = avctx->
color_trc;
356 if ((*
s->encoder)->InitializeExt(
s->encoder, ¶m) != cmResultSuccess) {
362 SFrameBSInfo fbi = { 0 };
364 (*
s->encoder)->EncodeParameterSets(
s->encoder, &fbi);
365 for (
i = 0;
i < fbi.sLayerInfo[0].iNalCount;
i++)
366 size += fbi.sLayerInfo[0].pNalLengthInByte[
i];
387 SFrameBSInfo fbi = { 0 };
390 SSourcePicture
sp = { 0 };
391 int size = 0, layer, first_layer = 0;
392 int layer_size[MAX_LAYER_NUM_OF_FRAME] = { 0 };
394 sp.iColorFormat = videoFormatI420;
395 for (
i = 0;
i < 3;
i++) {
403 (*
s->encoder)->ForceIntraFrame(
s->encoder,
true);
406 encoded = (*
s->encoder)->EncodeFrame(
s->encoder, &
sp, &fbi);
407 if (encoded != cmResultSuccess) {
411 if (fbi.eFrameType == videoFrameTypeSkip) {
422 first_layer = fbi.iLayerNum - 1;
424 for (layer = first_layer; layer < fbi.iLayerNum; layer++) {
425 for (
i = 0;
i < fbi.sLayerInfo[layer].iNalCount;
i++)
426 layer_size[layer] += fbi.sLayerInfo[layer].pNalLengthInByte[
i];
427 size += layer_size[layer];
429 av_log(avctx,
AV_LOG_DEBUG,
"%d slices\n", fbi.sLayerInfo[fbi.iLayerNum - 1].iNalCount);
435 for (layer = first_layer; layer < fbi.iLayerNum; layer++) {
436 memcpy(avpkt->
data +
size, fbi.sLayerInfo[layer].pBsBuf, layer_size[layer]);
437 size += layer_size[layer];
440 if (fbi.eFrameType == videoFrameTypeIDR)
455 .
p.
name =
"libopenh264",
469 .p.priv_class = &
class,
470 .p.wrapper_name =
"libopenh264",