blob: 2a7aaec644c426235ff05d2144a49a081aaaa185
1 | /* |
2 | * Ratecontrol |
3 | * Copyright (c) 2000, 2001, 2002 Fabrice Bellard |
4 | * Copyright (c) 2002-2004 Michael Niedermayer |
5 | * |
6 | * This file is part of FFmpeg. |
7 | * |
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. |
12 | * |
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. |
17 | * |
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 |
21 | */ |
22 | |
23 | #ifndef AVCODEC_RATECONTROL_H |
24 | #define AVCODEC_RATECONTROL_H |
25 | |
26 | /** |
27 | * @file |
28 | * ratecontrol header. |
29 | */ |
30 | |
31 | #include <stdio.h> |
32 | #include <stdint.h> |
33 | #include "libavutil/eval.h" |
34 | |
35 | typedef struct Predictor{ |
36 | double coeff; |
37 | double count; |
38 | double decay; |
39 | } Predictor; |
40 | |
41 | typedef struct RateControlEntry{ |
42 | int pict_type; |
43 | float qscale; |
44 | int mv_bits; |
45 | int i_tex_bits; |
46 | int p_tex_bits; |
47 | int misc_bits; |
48 | int header_bits; |
49 | uint64_t expected_bits; |
50 | int new_pict_type; |
51 | float new_qscale; |
52 | int64_t mc_mb_var_sum; |
53 | int64_t mb_var_sum; |
54 | int i_count; |
55 | int skip_count; |
56 | int f_code; |
57 | int b_code; |
58 | }RateControlEntry; |
59 | |
60 | /** |
61 | * rate control context. |
62 | */ |
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 |
67 | Predictor pred[5]; |
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) |
72 | double last_qscale; |
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]; |
80 | int frame_count[5]; |
81 | int last_non_b_pict_type; |
82 | |
83 | void *non_lavc_opaque; ///< context for non lavc rc code (for example xvid) |
84 | float dry_run_qscale; ///< for xvid rc |
85 | int last_picture_number; ///< for xvid rc |
86 | AVExpr * rc_eq_eval; |
87 | }RateControlContext; |
88 | |
89 | struct MpegEncContext; |
90 | |
91 | /* rate control */ |
92 | int ff_rate_control_init(struct MpegEncContext *s); |
93 | float ff_rate_estimate_qscale(struct MpegEncContext *s, int dry_run); |
94 | void ff_write_pass1_stats(struct MpegEncContext *s); |
95 | void ff_rate_control_uninit(struct MpegEncContext *s); |
96 | int ff_vbv_update(struct MpegEncContext *s, int frame_size); |
97 | void ff_get_2pass_fcode(struct MpegEncContext *s); |
98 | |
99 | #endif /* AVCODEC_RATECONTROL_H */ |
100 |