]> git.sesse.net Git - ffmpeg/blob - libavcodec/ratecontrol.h
avcodec/pthread_frame: remove usage of AVCodecContext accessors
[ffmpeg] / libavcodec / ratecontrol.h
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 */