3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
4 * Copyright (c) 2002-2004 Michael Niedermayer
6 * This file is part of FFmpeg.
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 #ifndef AVCODEC_RATECONTROL_H
24 #define AVCODEC_RATECONTROL_H
33 #include "libavutil/eval.h"
35 typedef struct Predictor
{
41 typedef struct RateControlEntry
{
49 uint64_t expected_bits
;
52 int64_t mc_mb_var_sum
;
61 * rate control context.
63 typedef struct RateControlContext
{
64 int num_entries
; ///< number of RateControlEntries
65 RateControlEntry
*entry
;
66 double buffer_index
; ///< amount of bits in the video/audio buffer
68 double short_term_qsum
; ///< sum of recent qscales
69 double short_term_qcount
; ///< count of recent qscales
70 double pass1_rc_eq_output_sum
;///< sum of the output of the rc equation, this is used for normalization
71 double pass1_wanted_bits
; ///< bits which should have been output by the pass1 code (including complexity init)
73 double last_qscale_for
[5]; ///< last qscale for a specific pict type, used for max_diff & ipb factor stuff
74 int64_t last_mc_mb_var_sum
;
75 int64_t last_mb_var_sum
;
76 uint64_t i_cplx_sum
[5];
77 uint64_t p_cplx_sum
[5];
78 uint64_t mv_bits_sum
[5];
79 uint64_t qscale_sum
[5];
81 int last_non_b_pict_type
;
86 struct MpegEncContext
;
89 int ff_rate_control_init(struct MpegEncContext
*s
);
90 float ff_rate_estimate_qscale(struct MpegEncContext
*s
, int dry_run
);
91 void ff_write_pass1_stats(struct MpegEncContext
*s
);
92 void ff_rate_control_uninit(struct MpegEncContext
*s
);
93 int ff_vbv_update(struct MpegEncContext
*s
, int frame_size
);
94 void ff_get_2pass_fcode(struct MpegEncContext
*s
);
96 #endif /* AVCODEC_RATECONTROL_H */