22 #include <wels/codec_api.h>
23 #include <wels/codec_ver.h>
36 #if !OPENH264_VER_AT_LEAST(1, 6)
37 #define SM_SIZELIMITED_SLICE SM_DYN_SLICE
52 #define OFFSET(x) offsetof(SVCContext, x)
53 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
55 #if OPENH264_VER_AT_LEAST(1, 6)
56 {
"slice_mode",
"set slice mode",
OFFSET(slice_mode),
AV_OPT_TYPE_INT, { .i64 = SM_FIXEDSLCNUM_SLICE }, SM_SINGLE_SLICE, SM_RESERVED,
VE,
"slice_mode" },
58 {
"slice_mode",
"set slice mode",
OFFSET(slice_mode),
AV_OPT_TYPE_INT, { .i64 = SM_AUTO_SLICE }, SM_SINGLE_SLICE, SM_RESERVED,
VE,
"slice_mode" },
60 {
"fixed",
"a fixed number of slices", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_FIXEDSLCNUM_SLICE }, 0, 0,
VE,
"slice_mode" },
61 #if OPENH264_VER_AT_LEAST(1, 6)
65 {
"rowmb",
"one slice per row of macroblocks", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_ROWMB_SLICE }, 0, 0,
VE,
"slice_mode" },
66 {
"auto",
"automatic number of slices according to number of threads", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_AUTO_SLICE }, 0, 0,
VE,
"slice_mode" },
67 {
"dyn",
"Dynamic slicing", 0,
AV_OPT_TYPE_CONST, { .i64 = SM_DYN_SLICE }, 0, 0,
VE,
"slice_mode" },
71 {
"max_nal_size",
"set maximum NAL size in bytes",
OFFSET(max_nal_size),
AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
VE },
72 {
"allow_skip_frames",
"allow skipping frames to hit the target bitrate",
OFFSET(skip_frames),
AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1,
VE },
86 WelsDestroySVCEncoder(s->
encoder);
95 SEncParamExt param = { 0 };
98 WelsTraceCallback callback_function;
104 if (WelsCreateSVCEncoder(&s->
encoder)) {
110 log_level = WELS_LOG_DETAIL;
111 (*s->
encoder)->SetOption(s->
encoder, ENCODER_OPTION_TRACE_LEVEL, &log_level);
115 (*s->
encoder)->SetOption(s->
encoder, ENCODER_OPTION_TRACE_CALLBACK, &callback_function);
118 (*s->
encoder)->SetOption(s->
encoder, ENCODER_OPTION_TRACE_CALLBACK_CONTEXT, &avctx);
122 #if FF_API_CODER_TYPE
130 param.iPicWidth = avctx->
width;
131 param.iPicHeight = avctx->
height;
132 param.iTargetBitrate = avctx->
bit_rate;
134 param.iRCMode = RC_QUALITY_MODE;
135 param.iTemporalLayerNum = 1;
136 param.iSpatialLayerNum = 1;
137 param.bEnableDenoise = 0;
138 param.bEnableBackgroundDetection = 1;
139 param.bEnableAdaptiveQuant = 1;
141 param.bEnableLongTermReference = 0;
142 param.iLtrMarkPeriod = 30;
143 param.uiIntraPeriod = avctx->
gop_size;
144 #if OPENH264_VER_AT_LEAST(1, 4)
145 param.eSpsPpsIdStrategy = CONSTANT_ID;
147 param.bEnableSpsPpsIdAddition = 0;
149 param.bPrefixNalAddingCtrl = 0;
151 param.iEntropyCodingModeFlag = 0;
154 param.iEntropyCodingModeFlag = 1;
156 param.iEntropyCodingModeFlag = 1;
158 param.sSpatialLayers[0].iVideoWidth = param.iPicWidth;
159 param.sSpatialLayers[0].iVideoHeight = param.iPicHeight;
160 param.sSpatialLayers[0].fFrameRate = param.fMaxFrameRate;
161 param.sSpatialLayers[0].iSpatialBitrate = param.iTargetBitrate;
162 param.sSpatialLayers[0].iMaxSpatialBitrate = param.iMaxBitrate;
166 "Invalid combination -slices %d and -max_nal_size %d.\n",
177 #if OPENH264_VER_AT_LEAST(1, 6)
178 param.sSpatialLayers[0].sSliceArgument.uiSliceMode = s->
slice_mode;
179 param.sSpatialLayers[0].sSliceArgument.uiSliceNum = avctx->
slices;
181 param.sSpatialLayers[0].sSliceCfg.uiSliceMode = s->
slice_mode;
182 param.sSpatialLayers[0].sSliceCfg.sSliceArgument.uiSliceNum = avctx->
slices;
188 #if OPENH264_VER_AT_LEAST(1, 6)
189 param.sSpatialLayers[0].sSliceArgument.uiSliceSizeConstraint = s->
max_nal_size;
191 param.sSpatialLayers[0].sSliceCfg.sSliceArgument.uiSliceSizeConstraint = s->
max_nal_size;
195 "specify a valid max_nal_size to use -slice_mode dyn\n");
200 if ((*s->
encoder)->InitializeExt(s->
encoder, ¶m) != cmResultSuccess) {
206 SFrameBSInfo fbi = { 0 };
209 for (i = 0; i < fbi.sLayerInfo[0].iNalCount; i++)
210 size += fbi.sLayerInfo[0].pNalLengthInByte[i];
215 memcpy(avctx->
extradata, fbi.sLayerInfo[0].pBsBuf, size);
231 SFrameBSInfo fbi = { 0 };
234 SSourcePicture
sp = { 0 };
235 int size = 0, layer, first_layer = 0;
236 int layer_size[MAX_LAYER_NUM_OF_FRAME] = { 0 };
238 sp.iColorFormat = videoFormatI420;
239 for (i = 0; i < 3; i++) {
241 sp.pData[i] = frame->
data[i];
243 sp.iPicWidth = avctx->
width;
244 sp.iPicHeight = avctx->
height;
247 if (encoded != cmResultSuccess) {
251 if (fbi.eFrameType == videoFrameTypeSkip) {
262 first_layer = fbi.iLayerNum - 1;
264 for (layer = first_layer; layer < fbi.iLayerNum; layer++) {
265 for (i = 0; i < fbi.sLayerInfo[layer].iNalCount; i++)
266 layer_size[layer] += fbi.sLayerInfo[layer].pNalLengthInByte[i];
267 size += layer_size[layer];
269 av_log(avctx,
AV_LOG_DEBUG,
"%d slices\n", fbi.sLayerInfo[fbi.iLayerNum - 1].iNalCount);
276 for (layer = first_layer; layer < fbi.iLayerNum; layer++) {
277 memcpy(avpkt->
data + size, fbi.sLayerInfo[layer].pBsBuf, layer_size[layer]);
278 size += layer_size[layer];
281 if (fbi.eFrameType == videoFrameTypeIDR)
288 .
name =
"libopenh264",
300 .priv_class = &
class,
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
void ff_libopenh264_trace_callback(void *ctx, int level, const char *msg)
This structure describes decoded (raw) audio or video data.
#define AV_LOG_WARNING
Something somehow does not look correct.
int64_t bit_rate
the average bitrate
#define LIBAVUTIL_VERSION_INT
static av_cold int init(AVCodecContext *avctx)
int max_bitrate
Maximum bitrate of the stream, in bits per second.
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
#define AV_CODEC_CAP_AUTO_THREADS
Codec supports avctx->thread_count == 0 (auto).
Macro definitions for various function/variable attributes.
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
static double av_q2d(AVRational a)
Convert an AVRational to a double.
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
static const AVOption options[]
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
int flags
AV_CODEC_FLAG_*.
const char * name
Name of the codec implementation.
static av_cold int svc_encode_close(AVCodecContext *avctx)
int flags
A combination of AV_PKT_FLAG values.
static av_cold int svc_encode_init(AVCodecContext *avctx)
common internal API header
int width
picture width / height.
attribute_deprecated int coder_type
int thread_count
thread count is used to decide how many independent tasks should be passed to execute() ...
This structure describes the bitrate properties of an encoded bitstream.
Libavcodec external API header.
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
main external API structure.
Describe the class of an AVClass context structure.
static int svc_encode_frame(AVCodecContext *avctx, AVPacket *avpkt, const AVFrame *frame, int *got_packet)
#define SM_SIZELIMITED_SLICE
int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size, int64_t min_size)
Check AVPacket size and/or allocate data.
static enum AVPixelFormat pix_fmts[]
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
#define AV_CODEC_FLAG_GLOBAL_HEADER
Place global headers in extradata instead of every keyframe.
int gop_size
the number of pictures in a group of pictures, or 0 for intra_only
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
#define FF_DISABLE_DEPRECATION_WARNINGS
common internal api header.
common internal and external API header
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
AVCPBProperties * ff_add_cpb_side_data(AVCodecContext *avctx)
Add a CPB properties side data to an encoding context.
#define AV_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding...
int slices
Number of slices.
#define FF_ENABLE_DEPRECATION_WARNINGS
int avg_bitrate
Average bitrate of the stream, in bits per second.
int ff_libopenh264_check_version(void *logctx)
AVCodec ff_libopenh264_encoder
AVPixelFormat
Pixel format.
This structure stores compressed data.
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
int64_t rc_max_rate
maximum bitrate