[FFmpeg-devel] [PATCH 19/41] avcodec/mpegvideoenc: Move tmp bframes to MPVMainEncContext
Andreas Rheinhardt
andreas.rheinhardt at outlook.com
Sun Jan 30 08:27:27 EET 2022
Also move b_frame_strategy and b_sensitivity; they are all
only used by an encoder's main thread.
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt at outlook.com>
---
libavcodec/mpegvideo.h | 5 ----
libavcodec/mpegvideo_enc.c | 55 +++++++++++++++++++-------------------
libavcodec/mpegvideoenc.h | 13 ++++++---
3 files changed, 36 insertions(+), 37 deletions(-)
diff --git a/libavcodec/mpegvideo.h b/libavcodec/mpegvideo.h
index 1bad288ea5..e83061fa92 100644
--- a/libavcodec/mpegvideo.h
+++ b/libavcodec/mpegvideo.h
@@ -530,11 +530,6 @@ typedef struct MPVContext {
int error_rate;
- /* temporary frames used by b_frame_strategy = 2 */
- AVFrame *tmp_frames[MAX_B_FRAMES + 2];
- int b_frame_strategy;
- int b_sensitivity;
-
/* frame skip options for encoding */
int frame_skip_threshold;
int frame_skip_factor;
diff --git a/libavcodec/mpegvideo_enc.c b/libavcodec/mpegvideo_enc.c
index a42cc0fa51..bbff65ef41 100644
--- a/libavcodec/mpegvideo_enc.c
+++ b/libavcodec/mpegvideo_enc.c
@@ -610,10 +610,10 @@ av_cold int ff_mpv_encode_init(AVCodecContext *avctx)
return AVERROR(EINVAL);
}
- if (s->b_frame_strategy && (avctx->flags & AV_CODEC_FLAG_PASS2)) {
+ if (m->b_frame_strategy && (avctx->flags & AV_CODEC_FLAG_PASS2)) {
av_log(avctx, AV_LOG_INFO,
"notice: b_frame_strategy only affects the first pass\n");
- s->b_frame_strategy = 0;
+ m->b_frame_strategy = 0;
}
i = av_gcd(avctx->time_base.den, avctx->time_base.num);
@@ -902,17 +902,17 @@ av_cold int ff_mpv_encode_init(AVCodecContext *avctx)
if ((ret = ff_rate_control_init(m)) < 0)
return ret;
- if (s->b_frame_strategy == 2) {
+ if (m->b_frame_strategy == 2) {
for (i = 0; i < s->max_b_frames + 2; i++) {
- s->tmp_frames[i] = av_frame_alloc();
- if (!s->tmp_frames[i])
+ m->tmp_frames[i] = av_frame_alloc();
+ if (!m->tmp_frames[i])
return AVERROR(ENOMEM);
- s->tmp_frames[i]->format = AV_PIX_FMT_YUV420P;
- s->tmp_frames[i]->width = s->width >> s->brd_scale;
- s->tmp_frames[i]->height = s->height >> s->brd_scale;
+ m->tmp_frames[i]->format = AV_PIX_FMT_YUV420P;
+ m->tmp_frames[i]->width = s->width >> s->brd_scale;
+ m->tmp_frames[i]->height = s->height >> s->brd_scale;
- ret = av_frame_get_buffer(s->tmp_frames[i], 0);
+ ret = av_frame_get_buffer(m->tmp_frames[i], 0);
if (ret < 0)
return ret;
}
@@ -933,14 +933,13 @@ av_cold int ff_mpv_encode_end(AVCodecContext *avctx)
{
MPVMainEncContext *const m = avctx->priv_data;
MPVMainContext *const s = &m->common;
- int i;
ff_rate_control_uninit(m);
ff_mpv_common_end(&m->common);
- for (i = 0; i < FF_ARRAY_ELEMS(s->tmp_frames); i++)
- av_frame_free(&s->tmp_frames[i]);
+ for (int i = 0; i < FF_ARRAY_ELEMS(m->tmp_frames); i++)
+ av_frame_free(&m->tmp_frames[i]);
ff_mpv_picture_free(avctx, &s->new_picture);
@@ -1265,18 +1264,18 @@ static int estimate_best_b_count(MPVMainEncContext *m)
data[2] += INPLACE_OFFSET;
}
- s->mpvencdsp.shrink[scale](s->tmp_frames[i]->data[0],
- s->tmp_frames[i]->linesize[0],
+ s->mpvencdsp.shrink[scale](m->tmp_frames[i]->data[0],
+ m->tmp_frames[i]->linesize[0],
data[0],
pre_input.f->linesize[0],
width, height);
- s->mpvencdsp.shrink[scale](s->tmp_frames[i]->data[1],
- s->tmp_frames[i]->linesize[1],
+ s->mpvencdsp.shrink[scale](m->tmp_frames[i]->data[1],
+ m->tmp_frames[i]->linesize[1],
data[1],
pre_input.f->linesize[1],
width >> 1, height >> 1);
- s->mpvencdsp.shrink[scale](s->tmp_frames[i]->data[2],
- s->tmp_frames[i]->linesize[2],
+ s->mpvencdsp.shrink[scale](m->tmp_frames[i]->data[2],
+ m->tmp_frames[i]->linesize[2],
data[2],
pre_input.f->linesize[2],
width >> 1, height >> 1);
@@ -1313,10 +1312,10 @@ static int estimate_best_b_count(MPVMainEncContext *m)
goto fail;
- s->tmp_frames[0]->pict_type = AV_PICTURE_TYPE_I;
- s->tmp_frames[0]->quality = 1 * FF_QP2LAMBDA;
+ m->tmp_frames[0]->pict_type = AV_PICTURE_TYPE_I;
+ m->tmp_frames[0]->quality = 1 * FF_QP2LAMBDA;
- out_size = encode_frame(c, s->tmp_frames[0], pkt);
+ out_size = encode_frame(c, m->tmp_frames[0], pkt);
if (out_size < 0) {
ret = out_size;
goto fail;
@@ -1327,11 +1326,11 @@ static int estimate_best_b_count(MPVMainEncContext *m)
for (i = 0; i < s->max_b_frames + 1; i++) {
int is_p = i % (j + 1) == j || i == s->max_b_frames;
- s->tmp_frames[i + 1]->pict_type = is_p ?
+ m->tmp_frames[i + 1]->pict_type = is_p ?
AV_PICTURE_TYPE_P : AV_PICTURE_TYPE_B;
- s->tmp_frames[i + 1]->quality = is_p ? p_lambda : b_lambda;
+ m->tmp_frames[i + 1]->quality = is_p ? p_lambda : b_lambda;
- out_size = encode_frame(c, s->tmp_frames[i + 1], pkt);
+ out_size = encode_frame(c, m->tmp_frames[i + 1], pkt);
if (out_size < 0) {
ret = out_size;
goto fail;
@@ -1418,11 +1417,11 @@ static int select_input_picture(MPVMainEncContext *m)
}
}
- if (s->b_frame_strategy == 0) {
+ if (m->b_frame_strategy == 0) {
b_frames = s->max_b_frames;
while (b_frames && !s->input_picture[b_frames])
b_frames--;
- } else if (s->b_frame_strategy == 1) {
+ } else if (m->b_frame_strategy == 1) {
for (i = 1; i < s->max_b_frames + 1; i++) {
if (s->input_picture[i] &&
s->input_picture[i]->b_frame_score == 0) {
@@ -1436,7 +1435,7 @@ static int select_input_picture(MPVMainEncContext *m)
for (i = 0; i < s->max_b_frames + 1; i++) {
if (!s->input_picture[i] ||
s->input_picture[i]->b_frame_score - 1 >
- s->mb_num / s->b_sensitivity)
+ s->mb_num / m->b_sensitivity)
break;
}
@@ -1446,7 +1445,7 @@ static int select_input_picture(MPVMainEncContext *m)
for (i = 0; i < b_frames + 1; i++) {
s->input_picture[i]->b_frame_score = 0;
}
- } else if (s->b_frame_strategy == 2) {
+ } else if (m->b_frame_strategy == 2) {
b_frames = estimate_best_b_count(m);
if (b_frames < 0)
return b_frames;
diff --git a/libavcodec/mpegvideoenc.h b/libavcodec/mpegvideoenc.h
index 8369254bff..c1013316aa 100644
--- a/libavcodec/mpegvideoenc.h
+++ b/libavcodec/mpegvideoenc.h
@@ -35,6 +35,11 @@
typedef MPVContext MPVEncContext;
typedef struct MPVMainEncContext {
MPVMainContext common;
+
+ /* temporary frames used by b_frame_strategy == 2 */
+ AVFrame *tmp_frames[MAX_B_FRAMES + 2];
+ int b_frame_strategy;
+ int b_sensitivity;
} MPVMainEncContext;
/* mpegvideo_enc common options */
@@ -115,8 +120,8 @@ FF_MPV_OPT_CMP_FUNC, \
{"intra_penalty", "Penalty for intra blocks in block decision", FF_MPV_OFFSET(intra_penalty), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, INT_MAX/2, FF_MPV_OPT_FLAGS }, \
#define FF_MPV_COMMON_BFRAME_OPTS \
-{"b_strategy", "Strategy to choose between I/P/B-frames", FF_MPV_OFFSET(b_frame_strategy), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, 2, FF_MPV_OPT_FLAGS }, \
-{"b_sensitivity", "Adjust sensitivity of b_frame_strategy 1", FF_MPV_OFFSET(b_sensitivity), AV_OPT_TYPE_INT, {.i64 = 40 }, 1, INT_MAX, FF_MPV_OPT_FLAGS }, \
+{"b_strategy", "Strategy to choose between I/P/B-frames", FF_MPV_MAIN_OFFSET(b_frame_strategy), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, 2, FF_MPV_OPT_FLAGS }, \
+{"b_sensitivity", "Adjust sensitivity of b_frame_strategy 1", FF_MPV_MAIN_OFFSET(b_sensitivity), AV_OPT_TYPE_INT, {.i64 = 40 }, 1, INT_MAX, FF_MPV_OPT_FLAGS }, \
{"brd_scale", "Downscale frames for dynamic B-frame decision", FF_MPV_OFFSET(brd_scale), AV_OPT_TYPE_INT, {.i64 = 0 }, 0, 3, FF_MPV_OPT_FLAGS },
#if FF_API_MPEGVIDEO_OPTS
@@ -127,8 +132,8 @@ FF_MPV_OPT_CMP_FUNC, \
#define FF_MPV_DEPRECATED_MATRIX_OPT \
{ "force_duplicated_matrix", "Deprecated, does nothing", FF_MPV_OFFSET(dummy), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, FF_MPV_OPT_FLAGS | AV_OPT_FLAG_DEPRECATED },
#define FF_MPV_DEPRECATED_BFRAME_OPTS \
- { "b_strategy", "Deprecated, does nothing", FF_MPV_OFFSET(b_frame_strategy), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 2, FF_MPV_OPT_FLAGS | AV_OPT_FLAG_DEPRECATED }, \
- { "b_sensitivity", "Deprecated, does nothing", FF_MPV_OFFSET(b_sensitivity), AV_OPT_TYPE_INT, { .i64 = 40 }, 1, INT_MAX, FF_MPV_OPT_FLAGS | AV_OPT_FLAG_DEPRECATED }, \
+ { "b_strategy", "Deprecated, does nothing", FF_MPV_MAIN_OFFSET(b_frame_strategy), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 2, FF_MPV_OPT_FLAGS | AV_OPT_FLAG_DEPRECATED }, \
+ { "b_sensitivity", "Deprecated, does nothing", FF_MPV_MAIN_OFFSET(b_sensitivity), AV_OPT_TYPE_INT, { .i64 = 40 }, 1, INT_MAX, FF_MPV_OPT_FLAGS | AV_OPT_FLAG_DEPRECATED }, \
{ "brd_scale", "Deprecated, does nothing", FF_MPV_OFFSET(brd_scale), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 3, FF_MPV_OPT_FLAGS | AV_OPT_FLAG_DEPRECATED },
#endif
--
2.32.0
More information about the ffmpeg-devel
mailing list