1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-08-04 22:03:09 +02:00

libtheoraenc: Add encoding speed level option

Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
This commit is contained in:
Bernat Arlandis
2025-07-19 18:26:53 +02:00
committed by Michael Niedermayer
parent 157d3b007e
commit 3cd4fa9b89

View File

@ -37,6 +37,7 @@
#include "libavutil/pixdesc.h"
#include "libavutil/log.h"
#include "libavutil/base64.h"
#include "libavutil/opt.h"
#include "avcodec.h"
#include "codec_internal.h"
#include "encode.h"
@ -45,6 +46,7 @@
#include <theora/theoraenc.h>
typedef struct TheoraContext {
AVClass *av_class; /**< class for AVOptions */
th_enc_ctx *t_state;
uint8_t *stats;
int stats_size;
@ -52,8 +54,21 @@ typedef struct TheoraContext {
int uv_hshift;
int uv_vshift;
int keyframe_mask;
int speed_level;
} TheoraContext;
static const AVOption options[] = {
{ "speed_level", "Sets the encoding speed level", offsetof(TheoraContext, speed_level), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM },
{ NULL }
};
static const AVClass theora_class = {
.class_name = "libtheora",
.item_name = av_default_item_name,
.option = options,
.version = LIBAVUTIL_VERSION_INT,
};
/** Concatenate an ogg_packet into the extradata. */
static int concatenate_packet(unsigned int* offset,
AVCodecContext* avc_context,
@ -244,6 +259,15 @@ static av_cold int encode_init(AVCodecContext* avc_context)
return AVERROR_EXTERNAL;
}
// Set encoding speed level
if (h->speed_level != -1) {
int max_speed_level;
int speed_level = h->speed_level;
th_encode_ctl(h->t_state, TH_ENCCTL_GET_SPLEVEL_MAX, &max_speed_level, sizeof(max_speed_level));
speed_level = FFMIN(speed_level, max_speed_level);
th_encode_ctl(h->t_state, TH_ENCCTL_SET_SPLEVEL, &speed_level, sizeof(speed_level));
}
// need to enable 2 pass (via TH_ENCCTL_2PASS_) before encoding headers
if (avc_context->flags & AV_CODEC_FLAG_PASS1) {
if ((ret = get_stats(avc_context, 0)) < 0)
@ -389,6 +413,7 @@ const FFCodec ff_libtheora_encoder = {
.close = encode_close,
FF_CODEC_ENCODE_CB(encode_frame),
CODEC_PIXFMTS(AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P),
.p.priv_class = &theora_class,
.color_ranges = AVCOL_RANGE_MPEG,
.p.wrapper_name = "libtheora",
};