annotate ratecontrol.c @ 12197:fbf4d5b1b664 libavcodec

Remove FF_MM_SSE2/3 flags for CPUs where this is generally not faster than regular MMX code. Examples of this are the Core1 CPU. Instead, set a new flag, FF_MM_SSE2/3SLOW, which can be checked for particular SSE2/3 functions that have been checked specifically on such CPUs and are actually faster than their MMX counterparts. In addition, use this flag to enable particular VP8 and LPC SSE2 functions that are faster than their MMX counterparts. Based on a patch by Loren Merritt <lorenm AT u washington edu>.
author rbultje
date Mon, 19 Jul 2010 22:38:23 +0000
parents e522dbf60abd
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
1 /*
428
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
2 * Rate control for video encoders
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
3 *
1739
07a484280a82 copyright year update of the files i touched and remembered, things look annoyingly unmaintained otherwise
michael
parents: 1708
diff changeset
4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
428
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
5 *
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3789
diff changeset
6 * This file is part of FFmpeg.
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3789
diff changeset
7 *
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3789
diff changeset
8 * FFmpeg is free software; you can redistribute it and/or
428
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
9 * modify it under the terms of the GNU Lesser General Public
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
10 * License as published by the Free Software Foundation; either
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3789
diff changeset
11 * version 2.1 of the License, or (at your option) any later version.
428
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
12 *
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3789
diff changeset
13 * FFmpeg is distributed in the hope that it will be useful,
428
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
16 * Lesser General Public License for more details.
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
17 *
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
18 * You should have received a copy of the GNU Lesser General Public
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3789
diff changeset
19 * License along with FFmpeg; if not, write to the Free Software
3036
0b546eab515d Update licensing information: The FSF changed postal address.
diego
parents: 2981
diff changeset
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
428
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
21 */
1106
1e39f273ecd6 per file doxy
michaelni
parents: 1057
diff changeset
22
1e39f273ecd6 per file doxy
michaelni
parents: 1057
diff changeset
23 /**
11644
7dd2a45249a9 Remove explicit filename from Doxygen @file commands.
diego
parents: 11614
diff changeset
24 * @file
1106
1e39f273ecd6 per file doxy
michaelni
parents: 1057
diff changeset
25 * Rate control for video encoders.
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
26 */
1106
1e39f273ecd6 per file doxy
michaelni
parents: 1057
diff changeset
27
11417
421c9441c11e Move ff_sqrt() to libavutil/intmath.h
mru
parents: 9476
diff changeset
28 #include "libavutil/intmath.h"
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
29 #include "avcodec.h"
428
dd4f4c3d7171 consistent include usage
glantau
parents: 396
diff changeset
30 #include "dsputil.h"
3789
730ad999e379 Move the ratecontrol related code from mpegvideo.h to a separate header file.
takis
parents: 3786
diff changeset
31 #include "ratecontrol.h"
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
32 #include "mpegvideo.h"
11832
e522dbf60abd Move eval.c and eval.h from libavcodec to libavutil, and make the eval
stefano
parents: 11801
diff changeset
33 #include "libavutil/eval.h"
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
34
4795
522e52c630bd typos/grammar
diego
parents: 4594
diff changeset
35 #undef NDEBUG // Always check asserts, the speed effect is far too small to disable them.
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
36 #include <assert.h>
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
37
627
79c43f519d02 undefined M_PI / M_E fix
michaelni
parents: 620
diff changeset
38 #ifndef M_E
79c43f519d02 undefined M_PI / M_E fix
michaelni
parents: 620
diff changeset
39 #define M_E 2.718281828
79c43f519d02 undefined M_PI / M_E fix
michaelni
parents: 620
diff changeset
40 #endif
79c43f519d02 undefined M_PI / M_E fix
michaelni
parents: 620
diff changeset
41
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
42 static int init_pass2(MpegEncContext *s);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
43 static double get_qscale(MpegEncContext *s, RateControlEntry *rce, double rate_factor, int frame_num);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
44
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
45 void ff_write_pass1_stats(MpegEncContext *s){
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
46 snprintf(s->avctx->stats_out, 256, "in:%d out:%d type:%d q:%d itex:%d ptex:%d mv:%d misc:%d fcode:%d bcode:%d mc-var:%d var:%d icount:%d skipcount:%d hbits:%d;\n",
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
47 s->current_picture_ptr->display_picture_number, s->current_picture_ptr->coded_picture_number, s->pict_type,
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
48 s->current_picture.quality, s->i_tex_bits, s->p_tex_bits, s->mv_bits, s->misc_bits,
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
49 s->f_code, s->b_code, s->current_picture.mc_mb_var_sum, s->current_picture.mb_var_sum, s->i_count, s->skip_count, s->header_bits);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
50 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
51
4084
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
52 static inline double qp2bits(RateControlEntry *rce, double qp){
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
53 if(qp<=0.0){
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
54 av_log(NULL, AV_LOG_ERROR, "qp<=0.0\n");
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
55 }
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
56 return rce->qscale * (double)(rce->i_tex_bits + rce->p_tex_bits+1)/ qp;
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
57 }
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
58
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
59 static inline double bits2qp(RateControlEntry *rce, double bits){
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
60 if(bits<0.9){
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
61 av_log(NULL, AV_LOG_ERROR, "bits<0.9\n");
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
62 }
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
63 return rce->qscale * (double)(rce->i_tex_bits + rce->p_tex_bits+1)/ bits;
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
64 }
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
65
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
66 int ff_rate_control_init(MpegEncContext *s)
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
67 {
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
68 RateControlContext *rcc= &s->rc_context;
11801
026edf66e3a9 Make ff_parse_expr() and ff_parse_and_eval_expr() return an int
stefano
parents: 11749
diff changeset
69 int i, res;
7129
322023e630a6 mark read-only data as const
stefang
parents: 6908
diff changeset
70 static const char * const const_names[]={
4084
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
71 "PI",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
72 "E",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
73 "iTex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
74 "pTex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
75 "tex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
76 "mv",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
77 "fCode",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
78 "iCount",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
79 "mcVar",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
80 "var",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
81 "isI",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
82 "isP",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
83 "isB",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
84 "avgQP",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
85 "qComp",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
86 /* "lastIQP",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
87 "lastPQP",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
88 "lastBQP",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
89 "nextNonBQP",*/
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
90 "avgIITex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
91 "avgPITex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
92 "avgPPTex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
93 "avgBPTex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
94 "avgTex",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
95 NULL
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
96 };
7978
dc1a7a6ec58d Add (additional) const to many global tables.
reimar
parents: 7587
diff changeset
97 static double (* const func1[])(void *, double)={
4084
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
98 (void *)bits2qp,
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
99 (void *)qp2bits,
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
100 NULL
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
101 };
7129
322023e630a6 mark read-only data as const
stefang
parents: 6908
diff changeset
102 static const char * const func1_names[]={
4084
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
103 "bits2qp",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
104 "qp2bits",
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
105 NULL
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
106 };
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
107 emms_c();
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
108
11832
e522dbf60abd Move eval.c and eval.h from libavcodec to libavutil, and make the eval
stefano
parents: 11801
diff changeset
109 res = av_parse_expr(&rcc->rc_eq_eval, s->avctx->rc_eq ? s->avctx->rc_eq : "tex^qComp", const_names, func1_names, func1, NULL, NULL, 0, s->avctx);
11801
026edf66e3a9 Make ff_parse_expr() and ff_parse_and_eval_expr() return an int
stefano
parents: 11749
diff changeset
110 if (res < 0) {
11749
c6368258b694 Change eval API to take parent log context and log level offset.
michael
parents: 11739
diff changeset
111 av_log(s->avctx, AV_LOG_ERROR, "Error parsing rc_eq \"%s\"\n", s->avctx->rc_eq);
11801
026edf66e3a9 Make ff_parse_expr() and ff_parse_and_eval_expr() return an int
stefano
parents: 11749
diff changeset
112 return res;
4084
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
113 }
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
114
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
115 for(i=0; i<5; i++){
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
116 rcc->pred[i].coeff= FF_QP2LAMBDA * 7.0;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
117 rcc->pred[i].count= 1.0;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
118
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
119 rcc->pred[i].decay= 0.4;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
120 rcc->i_cplx_sum [i]=
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
121 rcc->p_cplx_sum [i]=
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
122 rcc->mv_bits_sum[i]=
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
123 rcc->qscale_sum [i]=
6525
52925923273f cosmetics: typo fixes
diego
parents: 6481
diff changeset
124 rcc->frame_count[i]= 1; // 1 is better because of 1/0 and such
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
125 rcc->last_qscale_for[i]=FF_QP2LAMBDA * 5;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
126 }
1683
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
127 rcc->buffer_index= s->avctx->rc_initial_buffer_occupancy;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
128
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
129 if(s->flags&CODEC_FLAG_PASS2){
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
130 int i;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
131 char *p;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
132
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
133 /* find number of pics */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
134 p= s->avctx->stats_in;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
135 for(i=-1; p; i++){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
136 p= strchr(p+1, ';');
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
137 }
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
138 i+= s->max_b_frames;
2422
18b8b2dcc037 various security fixes and precautionary checks
michael
parents: 1806
diff changeset
139 if(i<=0 || i>=INT_MAX / sizeof(RateControlEntry))
18b8b2dcc037 various security fixes and precautionary checks
michael
parents: 1806
diff changeset
140 return -1;
5958
ed05a3d964fa stupid code (casting of void*) found by checktree.sh
michael
parents: 5672
diff changeset
141 rcc->entry = av_mallocz(i*sizeof(RateControlEntry));
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
142 rcc->num_entries= i;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
143
2628
511e3afc43e1 Ministry of English Composition, reporting for duty (and the word is "skipped", not "skiped"; "skiped" would rhyme with "hyped")
melanson
parents: 2494
diff changeset
144 /* init all to skipped p frames (with b frames we might have a not encoded frame at the end FIXME) */
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
145 for(i=0; i<rcc->num_entries; i++){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
146 RateControlEntry *rce= &rcc->entry[i];
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
147 rce->pict_type= rce->new_pict_type=FF_P_TYPE;
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
148 rce->qscale= rce->new_qscale=FF_QP2LAMBDA * 2;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
149 rce->misc_bits= s->mb_num + 10;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
150 rce->mb_var_sum= s->mb_num*100;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
151 }
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
152
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
153 /* read stats */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
154 p= s->avctx->stats_in;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
155 for(i=0; i<rcc->num_entries - s->max_b_frames; i++){
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
156 RateControlEntry *rce;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
157 int picture_number;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
158 int e;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
159 char *next;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
160
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
161 next= strchr(p, ';');
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
162 if(next){
5127
4dbe6578f811 misc spelling fixes
diego
parents: 5087
diff changeset
163 (*next)=0; //sscanf in unbelievably slow on looong strings //FIXME copy / do not write
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
164 next++;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
165 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
166 e= sscanf(p, " in:%d ", &picture_number);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
167
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
168 assert(picture_number >= 0);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
169 assert(picture_number < rcc->num_entries);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
170 rce= &rcc->entry[picture_number];
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
171
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
172 e+=sscanf(p, " in:%*d out:%*d type:%d q:%f itex:%d ptex:%d mv:%d misc:%d fcode:%d bcode:%d mc-var:%d var:%d icount:%d skipcount:%d hbits:%d",
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
173 &rce->pict_type, &rce->qscale, &rce->i_tex_bits, &rce->p_tex_bits, &rce->mv_bits, &rce->misc_bits,
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
174 &rce->f_code, &rce->b_code, &rce->mc_mb_var_sum, &rce->mb_var_sum, &rce->i_count, &rce->skip_count, &rce->header_bits);
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
175 if(e!=14){
1598
932d306bf1dc av_log() patch by (Michel Bardiaux <mbardiaux at peaktime dot be>)
michael
parents: 1505
diff changeset
176 av_log(s->avctx, AV_LOG_ERROR, "statistics are damaged at line %d, parser out=%d\n", i, e);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
177 return -1;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
178 }
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
179
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
180 p= next;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
181 }
3203
a78cc2e4b90e fix frametypes with xvid_rc
michael
parents: 3200
diff changeset
182
a78cc2e4b90e fix frametypes with xvid_rc
michael
parents: 3200
diff changeset
183 if(init_pass2(s) < 0) return -1;
a78cc2e4b90e fix frametypes with xvid_rc
michael
parents: 3200
diff changeset
184
3200
646f6344472d make ff_rate_control_init() bail out if rc_strategy==1 and lavc wasn't
corey
parents: 3065
diff changeset
185 //FIXME maybe move to end
646f6344472d make ff_rate_control_init() bail out if rc_strategy==1 and lavc wasn't
corey
parents: 3065
diff changeset
186 if((s->flags&CODEC_FLAG_PASS2) && s->avctx->rc_strategy == FF_RC_STRATEGY_XVID) {
8590
7a463923ecd1 Change semantic of CONFIG_*, HAVE_* and ARCH_*.
aurel
parents: 8227
diff changeset
187 #if CONFIG_LIBXVID
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
188 return ff_xvid_rate_control_init(s);
3200
646f6344472d make ff_rate_control_init() bail out if rc_strategy==1 and lavc wasn't
corey
parents: 3065
diff changeset
189 #else
6908
25dcd46b8160 cosmetics: XviD is now called Xvid.
diego
parents: 6526
diff changeset
190 av_log(s->avctx, AV_LOG_ERROR, "Xvid ratecontrol requires libavcodec compiled with Xvid support.\n");
3200
646f6344472d make ff_rate_control_init() bail out if rc_strategy==1 and lavc wasn't
corey
parents: 3065
diff changeset
191 return -1;
3065
b2c352a66a73 10l forgot #ifdef CONFIG_XVID
michael
parents: 3064
diff changeset
192 #endif
3200
646f6344472d make ff_rate_control_init() bail out if rc_strategy==1 and lavc wasn't
corey
parents: 3065
diff changeset
193 }
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
194 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
195
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
196 if(!(s->flags&CODEC_FLAG_PASS2)){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
197
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
198 rcc->short_term_qsum=0.001;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
199 rcc->short_term_qcount=0.001;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
200
707
c0a914cc90c0 renaming variable
michaelni
parents: 693
diff changeset
201 rcc->pass1_rc_eq_output_sum= 0.001;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
202 rcc->pass1_wanted_bits=0.001;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
203
5087
6c7f4ece59ed check qblur > 1 (prevent assert failure / segfault)
michael
parents: 4795
diff changeset
204 if(s->avctx->qblur > 1.0){
6c7f4ece59ed check qblur > 1 (prevent assert failure / segfault)
michael
parents: 4795
diff changeset
205 av_log(s->avctx, AV_LOG_ERROR, "qblur too large\n");
6c7f4ece59ed check qblur > 1 (prevent assert failure / segfault)
michael
parents: 4795
diff changeset
206 return -1;
6c7f4ece59ed check qblur > 1 (prevent assert failure / segfault)
michael
parents: 4795
diff changeset
207 }
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
208 /* init stuff with the user specified complexity */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
209 if(s->avctx->rc_initial_cplx){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
210 for(i=0; i<60*30; i++){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
211 double bits= s->avctx->rc_initial_cplx * (i/10000.0 + 1.0)*s->mb_num;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
212 RateControlEntry rce;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
213
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
214 if (i%((s->gop_size+3)/4)==0) rce.pict_type= FF_I_TYPE;
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
215 else if(i%(s->max_b_frames+1)) rce.pict_type= FF_B_TYPE;
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
216 else rce.pict_type= FF_P_TYPE;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
217
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
218 rce.new_pict_type= rce.pict_type;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
219 rce.mc_mb_var_sum= bits*s->mb_num/100000;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
220 rce.mb_var_sum = s->mb_num;
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
221 rce.qscale = FF_QP2LAMBDA * 2;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
222 rce.f_code = 2;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
223 rce.b_code = 1;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
224 rce.misc_bits= 1;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
225
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
226 if(s->pict_type== FF_I_TYPE){
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
227 rce.i_count = s->mb_num;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
228 rce.i_tex_bits= bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
229 rce.p_tex_bits= 0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
230 rce.mv_bits= 0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
231 }else{
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
232 rce.i_count = 0; //FIXME we do know this approx
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
233 rce.i_tex_bits= 0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
234 rce.p_tex_bits= bits*0.9;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
235 rce.mv_bits= bits*0.1;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
236 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
237 rcc->i_cplx_sum [rce.pict_type] += rce.i_tex_bits*rce.qscale;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
238 rcc->p_cplx_sum [rce.pict_type] += rce.p_tex_bits*rce.qscale;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
239 rcc->mv_bits_sum[rce.pict_type] += rce.mv_bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
240 rcc->frame_count[rce.pict_type] ++;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
241
9476
2b2bac59038e Remove 2 useless assignments from ff_rate_control_init() found by CSA.
michael
parents: 9470
diff changeset
242 get_qscale(s, &rce, rcc->pass1_wanted_bits/rcc->pass1_rc_eq_output_sum, i);
5408
20bea6a9950c cosmetics: misc typo fixes
diego
parents: 5203
diff changeset
243 rcc->pass1_wanted_bits+= s->bit_rate/(1/av_q2d(s->avctx->time_base)); //FIXME misbehaves a little for variable fps
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
244 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
245 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
246
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
247 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
248
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
249 return 0;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
250 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
251
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
252 void ff_rate_control_uninit(MpegEncContext *s)
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
253 {
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
254 RateControlContext *rcc= &s->rc_context;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
255 emms_c();
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
256
11832
e522dbf60abd Move eval.c and eval.h from libavcodec to libavutil, and make the eval
stefano
parents: 11801
diff changeset
257 av_free_expr(rcc->rc_eq_eval);
396
fce0a2520551 removed useless header includes - use av memory functions
glantau
parents: 388
diff changeset
258 av_freep(&rcc->entry);
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
259
8590
7a463923ecd1 Change semantic of CONFIG_*, HAVE_* and ARCH_*.
aurel
parents: 8227
diff changeset
260 #if CONFIG_LIBXVID
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
261 if((s->flags&CODEC_FLAG_PASS2) && s->avctx->rc_strategy == FF_RC_STRATEGY_XVID)
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
262 ff_xvid_rate_control_uninit(s);
3065
b2c352a66a73 10l forgot #ifdef CONFIG_XVID
michael
parents: 3064
diff changeset
263 #endif
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
264 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
265
1684
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
266 int ff_vbv_update(MpegEncContext *s, int frame_size){
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
267 RateControlContext *rcc= &s->rc_context;
2637
ef44d24680d1 switch to native time bases
michael
parents: 2628
diff changeset
268 const double fps= 1/av_q2d(s->avctx->time_base);
1697
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
269 const int buffer_size= s->avctx->rc_buffer_size;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
270 const double min_rate= s->avctx->rc_min_rate/fps;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
271 const double max_rate= s->avctx->rc_max_rate/fps;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
272
1697
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
273 //printf("%d %f %d %f %f\n", buffer_size, rcc->buffer_index, frame_size, min_rate, max_rate);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
274 if(buffer_size){
1683
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
275 int left;
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
276
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
277 rcc->buffer_index-= frame_size;
1683
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
278 if(rcc->buffer_index < 0){
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
279 av_log(s->avctx, AV_LOG_ERROR, "rc buffer underflow\n");
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
280 rcc->buffer_index= 0;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
281 }
1683
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
282
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
283 left= buffer_size - rcc->buffer_index - 1;
4594
a96d905dcbaa Add av_ prefix to clip functions
reimar
parents: 4588
diff changeset
284 rcc->buffer_index += av_clip(left, min_rate, max_rate);
1683
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
285
1697
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
286 if(rcc->buffer_index > buffer_size){
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
287 int stuffing= ceil((rcc->buffer_index - buffer_size)/8);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
288
1684
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
289 if(stuffing < 4 && s->codec_id == CODEC_ID_MPEG4)
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
290 stuffing=4;
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
291 rcc->buffer_index -= 8*stuffing;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
292
1684
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
293 if(s->avctx->debug & FF_DEBUG_RC)
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
294 av_log(s->avctx, AV_LOG_DEBUG, "stuffing %d bytes\n", stuffing);
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
295
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
296 return stuffing;
1683
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
297 }
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
298 }
1684
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
299 return 0;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
300 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
301
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
302 /**
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
303 * modifies the bitrate curve from pass1 for one frame
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
304 */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
305 static double get_qscale(MpegEncContext *s, RateControlEntry *rce, double rate_factor, int frame_num){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
306 RateControlContext *rcc= &s->rc_context;
1687
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
307 AVCodecContext *a= s->avctx;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
308 double q, bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
309 const int pict_type= rce->new_pict_type;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
310 const double mb_num= s->mb_num;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
311 int i;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
312
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
313 double const_values[]={
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
314 M_PI,
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
315 M_E,
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
316 rce->i_tex_bits*rce->qscale,
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
317 rce->p_tex_bits*rce->qscale,
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
318 (rce->i_tex_bits + rce->p_tex_bits)*(double)rce->qscale,
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
319 rce->mv_bits/mb_num,
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
320 rce->pict_type == FF_B_TYPE ? (rce->f_code + rce->b_code)*0.5 : rce->f_code,
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
321 rce->i_count/mb_num,
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
322 rce->mc_mb_var_sum/mb_num,
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
323 rce->mb_var_sum/mb_num,
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
324 rce->pict_type == FF_I_TYPE,
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
325 rce->pict_type == FF_P_TYPE,
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
326 rce->pict_type == FF_B_TYPE,
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
327 rcc->qscale_sum[pict_type] / (double)rcc->frame_count[pict_type],
1687
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
328 a->qcompress,
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
329 /* rcc->last_qscale_for[FF_I_TYPE],
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
330 rcc->last_qscale_for[FF_P_TYPE],
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
331 rcc->last_qscale_for[FF_B_TYPE],
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
332 rcc->next_non_b_qscale,*/
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
333 rcc->i_cplx_sum[FF_I_TYPE] / (double)rcc->frame_count[FF_I_TYPE],
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
334 rcc->i_cplx_sum[FF_P_TYPE] / (double)rcc->frame_count[FF_P_TYPE],
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
335 rcc->p_cplx_sum[FF_P_TYPE] / (double)rcc->frame_count[FF_P_TYPE],
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
336 rcc->p_cplx_sum[FF_B_TYPE] / (double)rcc->frame_count[FF_B_TYPE],
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
337 (rcc->i_cplx_sum[pict_type] + rcc->p_cplx_sum[pict_type]) / (double)rcc->frame_count[pict_type],
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
338 0
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
339 };
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
340
11832
e522dbf60abd Move eval.c and eval.h from libavcodec to libavutil, and make the eval
stefano
parents: 11801
diff changeset
341 bits = av_eval_expr(rcc->rc_eq_eval, const_values, rce);
3766
acf9ca729bd2 Handle possible failure of ff_eval.
takis
parents: 3681
diff changeset
342 if (isnan(bits)) {
4084
90e848a24c70 update ratecontrol to new ff_eval API
ods15
parents: 4001
diff changeset
343 av_log(s->avctx, AV_LOG_ERROR, "Error evaluating rc_eq \"%s\"\n", s->avctx->rc_eq);
3766
acf9ca729bd2 Handle possible failure of ff_eval.
takis
parents: 3681
diff changeset
344 return -1;
acf9ca729bd2 Handle possible failure of ff_eval.
takis
parents: 3681
diff changeset
345 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
346
707
c0a914cc90c0 renaming variable
michaelni
parents: 693
diff changeset
347 rcc->pass1_rc_eq_output_sum+= bits;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
348 bits*=rate_factor;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
349 if(bits<0.0) bits=0.0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
350 bits+= 1.0; //avoid 1/0 issues
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
351
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
352 /* user override */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
353 for(i=0; i<s->avctx->rc_override_count; i++){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
354 RcOverride *rco= s->avctx->rc_override;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
355 if(rco[i].start_frame > frame_num) continue;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
356 if(rco[i].end_frame < frame_num) continue;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
357
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
358 if(rco[i].qscale)
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
359 bits= qp2bits(rce, rco[i].qscale); //FIXME move at end to really force it?
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
360 else
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
361 bits*= rco[i].quality_factor;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
362 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
363
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
364 q= bits2qp(rce, bits);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
365
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
366 /* I/B difference */
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
367 if (pict_type==FF_I_TYPE && s->avctx->i_quant_factor<0.0)
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
368 q= -q*s->avctx->i_quant_factor + s->avctx->i_quant_offset;
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
369 else if(pict_type==FF_B_TYPE && s->avctx->b_quant_factor<0.0)
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
370 q= -q*s->avctx->b_quant_factor + s->avctx->b_quant_offset;
5421
0e593c7d0ebc prevent assert failure with negative quant_offset
michael
parents: 5408
diff changeset
371 if(q<1) q=1;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
372
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
373 return q;
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
374 }
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
375
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
376 static double get_diff_limited_q(MpegEncContext *s, RateControlEntry *rce, double q){
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
377 RateControlContext *rcc= &s->rc_context;
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
378 AVCodecContext *a= s->avctx;
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
379 const int pict_type= rce->new_pict_type;
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
380 const double last_p_q = rcc->last_qscale_for[FF_P_TYPE];
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
381 const double last_non_b_q= rcc->last_qscale_for[rcc->last_non_b_pict_type];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
382
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
383 if (pict_type==FF_I_TYPE && (a->i_quant_factor>0.0 || rcc->last_non_b_pict_type==FF_P_TYPE))
4001
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3947
diff changeset
384 q= last_p_q *FFABS(a->i_quant_factor) + a->i_quant_offset;
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
385 else if(pict_type==FF_B_TYPE && a->b_quant_factor>0.0)
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
386 q= last_non_b_q* a->b_quant_factor + a->b_quant_offset;
5421
0e593c7d0ebc prevent assert failure with negative quant_offset
michael
parents: 5408
diff changeset
387 if(q<1) q=1;
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
388
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
389 /* last qscale / qdiff stuff */
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
390 if(rcc->last_non_b_pict_type==pict_type || pict_type!=FF_I_TYPE){
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
391 double last_q= rcc->last_qscale_for[pict_type];
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
392 const int maxdiff= FF_QP2LAMBDA * a->max_qdiff;
930
6bcb214d6a17 more debug output
michaelni
parents: 915
diff changeset
393
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
394 if (q > last_q + maxdiff) q= last_q + maxdiff;
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
395 else if(q < last_q - maxdiff) q= last_q - maxdiff;
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
396 }
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
397
5127
4dbe6578f811 misc spelling fixes
diego
parents: 5087
diff changeset
398 rcc->last_qscale_for[pict_type]= q; //Note we cannot do that after blurring
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
399
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
400 if(pict_type!=FF_B_TYPE)
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
401 rcc->last_non_b_pict_type= pict_type;
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
402
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
403 return q;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
404 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
405
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
406 /**
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
407 * gets the qmin & qmax for pict_type
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
408 */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
409 static void get_qminmax(int *qmin_ret, int *qmax_ret, MpegEncContext *s, int pict_type){
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
410 int qmin= s->avctx->lmin;
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
411 int qmax= s->avctx->lmax;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
412
1365
d58fa7cc6008 fix qmin==qmax==1 && msmpeg4 && intra bug
michaelni
parents: 1264
diff changeset
413 assert(qmin <= qmax);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
414
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
415 if(pict_type==FF_B_TYPE){
4001
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3947
diff changeset
416 qmin= (int)(qmin*FFABS(s->avctx->b_quant_factor)+s->avctx->b_quant_offset + 0.5);
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3947
diff changeset
417 qmax= (int)(qmax*FFABS(s->avctx->b_quant_factor)+s->avctx->b_quant_offset + 0.5);
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
418 }else if(pict_type==FF_I_TYPE){
4001
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3947
diff changeset
419 qmin= (int)(qmin*FFABS(s->avctx->i_quant_factor)+s->avctx->i_quant_offset + 0.5);
34fdffe98bd0 Rename ABS macro to FFABS.
diego
parents: 3947
diff changeset
420 qmax= (int)(qmax*FFABS(s->avctx->i_quant_factor)+s->avctx->i_quant_offset + 0.5);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
421 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
422
4594
a96d905dcbaa Add av_ prefix to clip functions
reimar
parents: 4588
diff changeset
423 qmin= av_clip(qmin, 1, FF_LAMBDA_MAX);
a96d905dcbaa Add av_ prefix to clip functions
reimar
parents: 4588
diff changeset
424 qmax= av_clip(qmax, 1, FF_LAMBDA_MAX);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
425
1365
d58fa7cc6008 fix qmin==qmax==1 && msmpeg4 && intra bug
michaelni
parents: 1264
diff changeset
426 if(qmax<qmin) qmax= qmin;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
427
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
428 *qmin_ret= qmin;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
429 *qmax_ret= qmax;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
430 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
431
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
432 static double modify_qscale(MpegEncContext *s, RateControlEntry *rce, double q, int frame_num){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
433 RateControlContext *rcc= &s->rc_context;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
434 int qmin, qmax;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
435 const int pict_type= rce->new_pict_type;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
436 const double buffer_size= s->avctx->rc_buffer_size;
2637
ef44d24680d1 switch to native time bases
michael
parents: 2628
diff changeset
437 const double fps= 1/av_q2d(s->avctx->time_base);
1683
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
438 const double min_rate= s->avctx->rc_min_rate / fps;
043b6d933fc6 CBR improvements
michael
parents: 1598
diff changeset
439 const double max_rate= s->avctx->rc_max_rate / fps;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
440
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
441 get_qminmax(&qmin, &qmax, s, pict_type);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
442
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
443 /* modulation */
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
444 if(s->avctx->rc_qmod_freq && frame_num%s->avctx->rc_qmod_freq==0 && pict_type==FF_P_TYPE)
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
445 q*= s->avctx->rc_qmod_amp;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
446
678
michaelni
parents: 627
diff changeset
447 //printf("q:%f\n", q);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
448 /* buffer overflow/underflow protection */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
449 if(buffer_size){
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
450 double expected_size= rcc->buffer_index;
1697
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
451 double q_limit;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
452
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
453 if(min_rate){
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
454 double d= 2*(buffer_size - expected_size)/buffer_size;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
455 if(d>1.0) d=1.0;
678
michaelni
parents: 627
diff changeset
456 else if(d<0.0001) d=0.0001;
michaelni
parents: 627
diff changeset
457 q*= pow(d, 1.0/s->avctx->rc_buffer_aggressivity);
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
458
8227
596677256482 Implement the fields rc_max_available_vbv_use and
stefano
parents: 7978
diff changeset
459 q_limit= bits2qp(rce, FFMAX((min_rate - buffer_size + rcc->buffer_index) * s->avctx->rc_min_vbv_overflow_use, 1));
1697
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
460 if(q > q_limit){
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
461 if(s->avctx->debug&FF_DEBUG_RC){
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
462 av_log(s->avctx, AV_LOG_DEBUG, "limiting QP %f -> %f\n", q, q_limit);
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
463 }
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
464 q= q_limit;
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
465 }
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
466 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
467
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
468 if(max_rate){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
469 double d= 2*expected_size/buffer_size;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
470 if(d>1.0) d=1.0;
678
michaelni
parents: 627
diff changeset
471 else if(d<0.0001) d=0.0001;
michaelni
parents: 627
diff changeset
472 q/= pow(d, 1.0/s->avctx->rc_buffer_aggressivity);
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
473
8227
596677256482 Implement the fields rc_max_available_vbv_use and
stefano
parents: 7978
diff changeset
474 q_limit= bits2qp(rce, FFMAX(rcc->buffer_index * s->avctx->rc_max_available_vbv_use, 1));
1697
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
475 if(q < q_limit){
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
476 if(s->avctx->debug&FF_DEBUG_RC){
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
477 av_log(s->avctx, AV_LOG_DEBUG, "limiting QP %f -> %f\n", q, q_limit);
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
478 }
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
479 q= q_limit;
f3287b5d8d9f vbv_delay
michael
parents: 1687
diff changeset
480 }
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
481 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
482 }
678
michaelni
parents: 627
diff changeset
483 //printf("q:%f max:%f min:%f size:%f index:%d bits:%f agr:%f\n", q,max_rate, min_rate, buffer_size, rcc->buffer_index, bits, s->avctx->rc_buffer_aggressivity);
615
459a715f0661 fixing qmin==qmax && qsquish==1
michaelni
parents: 612
diff changeset
484 if(s->avctx->rc_qsquish==0.0 || qmin==qmax){
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
485 if (q<qmin) q=qmin;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
486 else if(q>qmax) q=qmax;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
487 }else{
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
488 double min2= log(qmin);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
489 double max2= log(qmax);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
490
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
491 q= log(q);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
492 q= (q - min2)/(max2-min2) - 0.5;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
493 q*= -4.0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
494 q= 1.0/(1.0 + exp(q));
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
495 q= q*(max2-min2) + min2;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
496
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
497 q= exp(q);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
498 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
499
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
500 return q;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
501 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
502
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
503 //----------------------------------
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
504 // 1 Pass Code
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
505
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
506 static double predict_size(Predictor *p, double q, double var)
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
507 {
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
508 return p->coeff*var / (q*p->count);
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
509 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
510
949
693a0797398f warnings patch by (bubu <bubu at bubu dot net>)
michaelni
parents: 932
diff changeset
511 /*
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
512 static double predict_qp(Predictor *p, double size, double var)
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
513 {
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
514 //printf("coeff:%f, count:%f, var:%f, size:%f//\n", p->coeff, p->count, var, size);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
515 return p->coeff*var / (size*p->count);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
516 }
949
693a0797398f warnings patch by (bubu <bubu at bubu dot net>)
michaelni
parents: 932
diff changeset
517 */
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
518
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
519 static void update_predictor(Predictor *p, double q, double var, double size)
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
520 {
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
521 double new_coeff= size*q / (var + 1);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
522 if(var<10) return;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
523
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
524 p->count*= p->decay;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
525 p->coeff*= p->decay;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
526 p->count++;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
527 p->coeff+= new_coeff;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
528 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
529
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
530 static void adaptive_quantization(MpegEncContext *s, double q){
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
531 int i;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
532 const float lumi_masking= s->avctx->lumi_masking / (128.0*128.0);
693
b6a7ff92df57 darkness masking (lumi masking does only bright stuff now)
michaelni
parents: 690
diff changeset
533 const float dark_masking= s->avctx->dark_masking / (128.0*128.0);
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
534 const float temp_cplx_masking= s->avctx->temporal_cplx_masking;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
535 const float spatial_cplx_masking = s->avctx->spatial_cplx_masking;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
536 const float p_masking = s->avctx->p_masking;
2493
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
537 const float border_masking = s->avctx->border_masking;
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
538 float bits_sum= 0.0;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
539 float cplx_sum= 0.0;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
540 float cplx_tab[s->mb_num];
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
541 float bits_tab[s->mb_num];
2494
36d70fbb31c5 mb_lmin/max to limit the per mb quality for the ratecontrol independant from the frame limits
michael
parents: 2493
diff changeset
542 const int qmin= s->avctx->mb_lmin;
36d70fbb31c5 mb_lmin/max to limit the per mb quality for the ratecontrol independant from the frame limits
michael
parents: 2493
diff changeset
543 const int qmax= s->avctx->mb_lmax;
903
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
544 Picture * const pic= &s->current_picture;
2493
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
545 const int mb_width = s->mb_width;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
546 const int mb_height = s->mb_height;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
547
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
548 for(i=0; i<s->mb_num; i++){
1180
ea015319b9e8 10l (adaptve quant fix)
michaelni
parents: 1141
diff changeset
549 const int mb_xy= s->mb_index2xy[i];
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
550 float temp_cplx= sqrt(pic->mc_mb_var[mb_xy]); //FIXME merge in pow()
1180
ea015319b9e8 10l (adaptve quant fix)
michaelni
parents: 1141
diff changeset
551 float spat_cplx= sqrt(pic->mb_var[mb_xy]);
ea015319b9e8 10l (adaptve quant fix)
michaelni
parents: 1141
diff changeset
552 const int lumi= pic->mb_mean[mb_xy];
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
553 float bits, cplx, factor;
2493
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
554 int mb_x = mb_xy % s->mb_stride;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
555 int mb_y = mb_xy / s->mb_stride;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
556 int mb_distance;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
557 float mb_factor = 0.0;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
558 #if 0
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
559 if(spat_cplx < q/3) spat_cplx= q/3; //FIXME finetune
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
560 if(temp_cplx < q/3) temp_cplx= q/3; //FIXME finetune
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
561 #endif
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
562 if(spat_cplx < 4) spat_cplx= 4; //FIXME finetune
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
563 if(temp_cplx < 4) temp_cplx= 4; //FIXME finetune
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
564
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
565 if((s->mb_type[mb_xy]&CANDIDATE_MB_TYPE_INTRA)){//FIXME hq mode
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
566 cplx= spat_cplx;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
567 factor= 1.0 + p_masking;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
568 }else{
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
569 cplx= temp_cplx;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
570 factor= pow(temp_cplx, - temp_cplx_masking);
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
571 }
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
572 factor*=pow(spat_cplx, - spatial_cplx_masking);
693
b6a7ff92df57 darkness masking (lumi masking does only bright stuff now)
michaelni
parents: 690
diff changeset
573
b6a7ff92df57 darkness masking (lumi masking does only bright stuff now)
michaelni
parents: 690
diff changeset
574 if(lumi>127)
b6a7ff92df57 darkness masking (lumi masking does only bright stuff now)
michaelni
parents: 690
diff changeset
575 factor*= (1.0 - (lumi-128)*(lumi-128)*lumi_masking);
b6a7ff92df57 darkness masking (lumi masking does only bright stuff now)
michaelni
parents: 690
diff changeset
576 else
b6a7ff92df57 darkness masking (lumi masking does only bright stuff now)
michaelni
parents: 690
diff changeset
577 factor*= (1.0 - (lumi-128)*(lumi-128)*dark_masking);
2493
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
578
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
579 if(mb_x < mb_width/5){
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
580 mb_distance = mb_width/5 - mb_x;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
581 mb_factor = (float)mb_distance / (float)(mb_width/5);
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
582 }else if(mb_x > 4*mb_width/5){
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
583 mb_distance = mb_x - 4*mb_width/5;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
584 mb_factor = (float)mb_distance / (float)(mb_width/5);
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
585 }
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
586 if(mb_y < mb_height/5){
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
587 mb_distance = mb_height/5 - mb_y;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
588 mb_factor = FFMAX(mb_factor, (float)mb_distance / (float)(mb_height/5));
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
589 }else if(mb_y > 4*mb_height/5){
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
590 mb_distance = mb_y - 4*mb_height/5;
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
591 mb_factor = FFMAX(mb_factor, (float)mb_distance / (float)(mb_height/5));
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
592 }
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
593
258120c61eea Border processing adaptive quant patch by (Christophe Massiot |cmassiot freebox fr)
michael
parents: 2423
diff changeset
594 factor*= 1.0 - border_masking*mb_factor;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
595
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
596 if(factor<0.00001) factor= 0.00001;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
597
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
598 bits= cplx*factor;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
599 cplx_sum+= cplx;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
600 bits_sum+= bits;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
601 cplx_tab[i]= cplx;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
602 bits_tab[i]= bits;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
603 }
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
604
5127
4dbe6578f811 misc spelling fixes
diego
parents: 5087
diff changeset
605 /* handle qmin/qmax clipping */
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
606 if(s->flags&CODEC_FLAG_NORMALIZE_AQP){
1806
2721e1859e19 normalize adaptive quantizatiuon fix (based upon a patch by (Jindrich Makovicka <makovick at kmlinux dot fjfi dot cvut dot cz>))
michael
parents: 1739
diff changeset
607 float factor= bits_sum/cplx_sum;
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
608 for(i=0; i<s->mb_num; i++){
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
609 float newq= q*cplx_tab[i]/bits_tab[i];
1806
2721e1859e19 normalize adaptive quantizatiuon fix (based upon a patch by (Jindrich Makovicka <makovick at kmlinux dot fjfi dot cvut dot cz>))
michael
parents: 1739
diff changeset
610 newq*= factor;
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
611
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
612 if (newq > qmax){
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
613 bits_sum -= bits_tab[i];
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
614 cplx_sum -= cplx_tab[i]*q/qmax;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
615 }
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
616 else if(newq < qmin){
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
617 bits_sum -= bits_tab[i];
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
618 cplx_sum -= cplx_tab[i]*q/qmin;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
619 }
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
620 }
1806
2721e1859e19 normalize adaptive quantizatiuon fix (based upon a patch by (Jindrich Makovicka <makovick at kmlinux dot fjfi dot cvut dot cz>))
michael
parents: 1739
diff changeset
621 if(bits_sum < 0.001) bits_sum= 0.001;
2721e1859e19 normalize adaptive quantizatiuon fix (based upon a patch by (Jindrich Makovicka <makovick at kmlinux dot fjfi dot cvut dot cz>))
michael
parents: 1739
diff changeset
622 if(cplx_sum < 0.001) cplx_sum= 0.001;
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
623 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
624
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
625 for(i=0; i<s->mb_num; i++){
1180
ea015319b9e8 10l (adaptve quant fix)
michaelni
parents: 1141
diff changeset
626 const int mb_xy= s->mb_index2xy[i];
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
627 float newq= q*cplx_tab[i]/bits_tab[i];
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
628 int intq;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
629
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
630 if(s->flags&CODEC_FLAG_NORMALIZE_AQP){
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
631 newq*= bits_sum/cplx_sum;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
632 }
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
633
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
634 intq= (int)(newq + 0.5);
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
635
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
636 if (intq > qmax) intq= qmax;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
637 else if(intq < qmin) intq= qmin;
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
638 //if(i%s->mb_width==0) printf("\n");
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
639 //printf("%2d%3d ", intq, ff_sqrt(s->mc_mb_var[i]));
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
640 s->lambda_table[mb_xy]= intq;
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
641 }
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
642 }
2981
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
643
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
644 void ff_get_2pass_fcode(MpegEncContext *s){
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
645 RateControlContext *rcc= &s->rc_context;
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
646 int picture_number= s->picture_number;
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
647 RateControlEntry *rce;
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
648
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
649 rce= &rcc->entry[picture_number];
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
650 s->f_code= rce->f_code;
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
651 s->b_code= rce->b_code;
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
652 }
33d4fb0df0d3 better f_code guessing in 2nd pass
michael
parents: 2974
diff changeset
653
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
654 //FIXME rd or at least approx for dquant
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
655
2974
e7189a8c22ed use correct qp & lambda for ME on the second pass
michael
parents: 2967
diff changeset
656 float ff_rate_estimate_qscale(MpegEncContext *s, int dry_run)
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
657 {
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
658 float q;
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
659 int qmin, qmax;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
660 float br_compensation;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
661 double diff;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
662 double short_term_q;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
663 double fps;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
664 int picture_number= s->picture_number;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
665 int64_t wanted_bits;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
666 RateControlContext *rcc= &s->rc_context;
1687
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
667 AVCodecContext *a= s->avctx;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
668 RateControlEntry local_rce, *rce;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
669 double bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
670 double rate_factor;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
671 int var;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
672 const int pict_type= s->pict_type;
903
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
673 Picture * const pic= &s->current_picture;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
674 emms_c();
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
675
8590
7a463923ecd1 Change semantic of CONFIG_*, HAVE_* and ARCH_*.
aurel
parents: 8227
diff changeset
676 #if CONFIG_LIBXVID
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
677 if((s->flags&CODEC_FLAG_PASS2) && s->avctx->rc_strategy == FF_RC_STRATEGY_XVID)
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
678 return ff_xvid_rate_estimate_qscale(s, dry_run);
3065
b2c352a66a73 10l forgot #ifdef CONFIG_XVID
michael
parents: 3064
diff changeset
679 #endif
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents: 3036
diff changeset
680
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
681 get_qminmax(&qmin, &qmax, s, pict_type);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
682
2637
ef44d24680d1 switch to native time bases
michael
parents: 2628
diff changeset
683 fps= 1/av_q2d(s->avctx->time_base);
678
michaelni
parents: 627
diff changeset
684 //printf("input_pic_num:%d pic_num:%d frame_rate:%d\n", s->input_picture_number, s->picture_number, s->frame_rate);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
685 /* update predictors */
2974
e7189a8c22ed use correct qp & lambda for ME on the second pass
michael
parents: 2967
diff changeset
686 if(picture_number>2 && !dry_run){
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
687 const int last_var= s->last_pict_type == FF_I_TYPE ? rcc->last_mb_var_sum : rcc->last_mc_mb_var_sum;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
688 update_predictor(&rcc->pred[s->last_pict_type], rcc->last_qscale, sqrt(last_var), s->frame_bits);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
689 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
690
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
691 if(s->flags&CODEC_FLAG_PASS2){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
692 assert(picture_number>=0);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
693 assert(picture_number<rcc->num_entries);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
694 rce= &rcc->entry[picture_number];
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
695 wanted_bits= rce->expected_bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
696 }else{
5672
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
697 Picture *dts_pic;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
698 rce= &local_rce;
5672
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
699
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
700 //FIXME add a dts field to AVFrame and ensure its set and use it here instead of reordering
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
701 //but the reordering is simpler for now until h.264 b pyramid must be handeld
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
702 if(s->pict_type == FF_B_TYPE || s->low_delay)
5672
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
703 dts_pic= s->current_picture_ptr;
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
704 else
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
705 dts_pic= s->last_picture_ptr;
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
706
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
707 //if(dts_pic)
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
708 // av_log(NULL, AV_LOG_ERROR, "%Ld %Ld %Ld %d\n", s->current_picture_ptr->pts, s->user_specified_pts, dts_pic->pts, picture_number);
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
709
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
710 if(!dts_pic || dts_pic->pts == AV_NOPTS_VALUE)
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
711 wanted_bits= (uint64_t)(s->bit_rate*(double)picture_number/fps);
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
712 else
ce58a38cd391 fix timestamps used for ratecontrol
michael
parents: 5421
diff changeset
713 wanted_bits= (uint64_t)(s->bit_rate*(double)dts_pic->pts/fps);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
714 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
715
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
716 diff= s->total_bits - wanted_bits;
1687
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
717 br_compensation= (a->bit_rate_tolerance - diff)/a->bit_rate_tolerance;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
718 if(br_compensation<=0.0) br_compensation=0.001;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
719
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
720 var= pict_type == FF_I_TYPE ? pic->mb_var_sum : pic->mc_mb_var_sum;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
721
1455
c4539ef4d8cb removed warnings
bellard
parents: 1365
diff changeset
722 short_term_q = 0; /* avoid warning */
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
723 if(s->flags&CODEC_FLAG_PASS2){
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
724 if(pict_type!=FF_I_TYPE)
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
725 assert(pict_type == rce->new_pict_type);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
726
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
727 q= rce->new_qscale / br_compensation;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
728 //printf("%f %f %f last:%d var:%d type:%d//\n", q, rce->new_qscale, br_compensation, s->frame_bits, var, pict_type);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
729 }else{
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
730 rce->pict_type=
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
731 rce->new_pict_type= pict_type;
903
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
732 rce->mc_mb_var_sum= pic->mc_mb_var_sum;
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
733 rce->mb_var_sum = pic-> mb_var_sum;
1505
010f76d07a27 use lagrange multipler instead of qp for ratecontrol, this may break some things, tell me ASAP if u notice anything broken
michaelni
parents: 1455
diff changeset
734 rce->qscale = FF_QP2LAMBDA * 2;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
735 rce->f_code = s->f_code;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
736 rce->b_code = s->b_code;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
737 rce->misc_bits= 1;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
738
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
739 bits= predict_size(&rcc->pred[pict_type], rce->qscale, sqrt(var));
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
740 if(pict_type== FF_I_TYPE){
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
741 rce->i_count = s->mb_num;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
742 rce->i_tex_bits= bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
743 rce->p_tex_bits= 0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
744 rce->mv_bits= 0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
745 }else{
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
746 rce->i_count = 0; //FIXME we do know this approx
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
747 rce->i_tex_bits= 0;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
748 rce->p_tex_bits= bits*0.9;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
749
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
750 rce->mv_bits= bits*0.1;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
751 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
752 rcc->i_cplx_sum [pict_type] += rce->i_tex_bits*rce->qscale;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
753 rcc->p_cplx_sum [pict_type] += rce->p_tex_bits*rce->qscale;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
754 rcc->mv_bits_sum[pict_type] += rce->mv_bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
755 rcc->frame_count[pict_type] ++;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
756
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
757 bits= rce->i_tex_bits + rce->p_tex_bits;
707
c0a914cc90c0 renaming variable
michaelni
parents: 693
diff changeset
758 rate_factor= rcc->pass1_wanted_bits/rcc->pass1_rc_eq_output_sum * br_compensation;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
759
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
760 q= get_qscale(s, rce, rate_factor, picture_number);
3766
acf9ca729bd2 Handle possible failure of ff_eval.
takis
parents: 3681
diff changeset
761 if (q < 0)
acf9ca729bd2 Handle possible failure of ff_eval.
takis
parents: 3681
diff changeset
762 return -1;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
763
615
459a715f0661 fixing qmin==qmax && qsquish==1
michaelni
parents: 612
diff changeset
764 assert(q>0.0);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
765 //printf("%f ", q);
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
766 q= get_diff_limited_q(s, rce, q);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
767 //printf("%f ", q);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
768 assert(q>0.0);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
769
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
770 if(pict_type==FF_P_TYPE || s->intra_only){ //FIXME type dependent blur like in 2-pass
1687
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
771 rcc->short_term_qsum*=a->qblur;
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
772 rcc->short_term_qcount*=a->qblur;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
773
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
774 rcc->short_term_qsum+= q;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
775 rcc->short_term_qcount++;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
776 //printf("%f ", q);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
777 q= short_term_q= rcc->short_term_qsum/rcc->short_term_qcount;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
778 //printf("%f ", q);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
779 }
678
michaelni
parents: 627
diff changeset
780 assert(q>0.0);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
781
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
782 q= modify_qscale(s, rce, q, picture_number);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
783
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
784 rcc->pass1_wanted_bits+= s->bit_rate/fps;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
785
615
459a715f0661 fixing qmin==qmax && qsquish==1
michaelni
parents: 612
diff changeset
786 assert(q>0.0);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
787 }
930
6bcb214d6a17 more debug output
michaelni
parents: 915
diff changeset
788
6bcb214d6a17 more debug output
michaelni
parents: 915
diff changeset
789 if(s->avctx->debug&FF_DEBUG_RC){
1598
932d306bf1dc av_log() patch by (Michel Bardiaux <mbardiaux at peaktime dot be>)
michael
parents: 1505
diff changeset
790 av_log(s->avctx, AV_LOG_DEBUG, "%c qp:%d<%2.1f<%d %d want:%d total:%d comp:%f st_q:%2.2f size:%d var:%d/%d br:%d fps:%d\n",
1264
2fa34e615c76 cleanup
michaelni
parents: 1180
diff changeset
791 av_get_pict_type_char(pict_type), qmin, q, qmax, picture_number, (int)wanted_bits/1000, (int)s->total_bits/1000,
930
6bcb214d6a17 more debug output
michaelni
parents: 915
diff changeset
792 br_compensation, short_term_q, s->frame_bits, pic->mb_var_sum, pic->mc_mb_var_sum, s->bit_rate/1000, (int)fps
6bcb214d6a17 more debug output
michaelni
parents: 915
diff changeset
793 );
6bcb214d6a17 more debug output
michaelni
parents: 915
diff changeset
794 }
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
795
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
796 if (q<qmin) q=qmin;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
797 else if(q>qmax) q=qmax;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
798
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
799 if(s->adaptive_quant)
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
800 adaptive_quantization(s, q);
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
801 else
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
802 q= (int)(q + 0.5);
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
803
2974
e7189a8c22ed use correct qp & lambda for ME on the second pass
michael
parents: 2967
diff changeset
804 if(!dry_run){
e7189a8c22ed use correct qp & lambda for ME on the second pass
michael
parents: 2967
diff changeset
805 rcc->last_qscale= q;
e7189a8c22ed use correct qp & lambda for ME on the second pass
michael
parents: 2967
diff changeset
806 rcc->last_mc_mb_var_sum= pic->mc_mb_var_sum;
e7189a8c22ed use correct qp & lambda for ME on the second pass
michael
parents: 2967
diff changeset
807 rcc->last_mb_var_sum= pic->mb_var_sum;
e7189a8c22ed use correct qp & lambda for ME on the second pass
michael
parents: 2967
diff changeset
808 }
903
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
809 #if 0
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
810 {
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
811 static int mvsum=0, texsum=0;
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
812 mvsum += s->mv_bits;
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
813 texsum += s->i_tex_bits + s->p_tex_bits;
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
814 printf("%d %d//\n\n", mvsum, texsum);
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
815 }
22ee74da2cd3 cleanup
michaelni
parents: 847
diff changeset
816 #endif
690
a1c69cb685b3 adaptive quantization (lumi/temporal & spatial complexity masking)
michaelni
parents: 679
diff changeset
817 return q;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
818 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
819
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
820 //----------------------------------------------
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
821 // 2-Pass code
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
822
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
823 static int init_pass2(MpegEncContext *s)
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
824 {
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
825 RateControlContext *rcc= &s->rc_context;
1687
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
826 AVCodecContext *a= s->avctx;
3681
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
827 int i, toobig;
2637
ef44d24680d1 switch to native time bases
michael
parents: 2628
diff changeset
828 double fps= 1/av_q2d(s->avctx->time_base);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
829 double complexity[5]={0,0,0,0,0}; // aproximate bits at quant=1
4588
fc155ff94878 cosmetics: Fix another common typo, dependAnt --> dependEnt.
diego
parents: 4084
diff changeset
830 uint64_t const_bits[5]={0,0,0,0,0}; // quantizer independent bits
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
831 uint64_t all_const_bits;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
832 uint64_t all_available_bits= (uint64_t)(s->bit_rate*(double)rcc->num_entries/fps);
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
833 double rate_factor=0;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
834 double step;
949
693a0797398f warnings patch by (bubu <bubu at bubu dot net>)
michaelni
parents: 932
diff changeset
835 //int last_i_frame=-10000000;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
836 const int filter_size= (int)(a->qblur*4) | 1;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
837 double expected_bits;
6526
737b74bec1e5 Fix typo in variable name: blured --> blurred.
diego
parents: 6525
diff changeset
838 double *qscale, *blurred_qscale, qscale_sum;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
839
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
840 /* find complexity & const_bits & decide the pict_types */
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
841 for(i=0; i<rcc->num_entries; i++){
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
842 RateControlEntry *rce= &rcc->entry[i];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
843
915
75ee49a4a516 fixing 2pass assert failure
michaelni
parents: 903
diff changeset
844 rce->new_pict_type= rce->pict_type;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
845 rcc->i_cplx_sum [rce->pict_type] += rce->i_tex_bits*rce->qscale;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
846 rcc->p_cplx_sum [rce->pict_type] += rce->p_tex_bits*rce->qscale;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
847 rcc->mv_bits_sum[rce->pict_type] += rce->mv_bits;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
848 rcc->frame_count[rce->pict_type] ++;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
849
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
850 complexity[rce->new_pict_type]+= (rce->i_tex_bits+ rce->p_tex_bits)*(double)rce->qscale;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
851 const_bits[rce->new_pict_type]+= rce->mv_bits + rce->misc_bits;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
852 }
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6373
diff changeset
853 all_const_bits= const_bits[FF_I_TYPE] + const_bits[FF_P_TYPE] + const_bits[FF_B_TYPE];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
854
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
855 if(all_available_bits < all_const_bits){
3674
be8084e35485 error message spelling typo
corey
parents: 3319
diff changeset
856 av_log(s->avctx, AV_LOG_ERROR, "requested bitrate is too low\n");
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
857 return -1;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
858 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
859
1031
19de1445beb2 use av_malloc() functions - added av_strdup and av_realloc()
bellard
parents: 977
diff changeset
860 qscale= av_malloc(sizeof(double)*rcc->num_entries);
6526
737b74bec1e5 Fix typo in variable name: blured --> blurred.
diego
parents: 6525
diff changeset
861 blurred_qscale= av_malloc(sizeof(double)*rcc->num_entries);
3681
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
862 toobig = 0;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
863
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
864 for(step=256*256; step>0.0000001; step*=0.5){
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
865 expected_bits=0;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
866 rate_factor+= step;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
867
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
868 rcc->buffer_index= s->avctx->rc_buffer_size/2;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
869
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
870 /* find qscale */
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
871 for(i=0; i<rcc->num_entries; i++){
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
872 qscale[i]= get_qscale(s, &rcc->entry[i], rate_factor, i);
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
873 }
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
874 assert(filter_size%2==1);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
875
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
876 /* fixed I/B QP relative to P mode */
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
877 for(i=rcc->num_entries-1; i>=0; i--){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
878 RateControlEntry *rce= &rcc->entry[i];
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
879
679
3d2da9b44cd8 better min/max rate handling
michaelni
parents: 678
diff changeset
880 qscale[i]= get_diff_limited_q(s, rce, qscale[i]);
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
881 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
882
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
883 /* smooth curve */
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
884 for(i=0; i<rcc->num_entries; i++){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
885 RateControlEntry *rce= &rcc->entry[i];
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
886 const int pict_type= rce->new_pict_type;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
887 int j;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
888 double q=0.0, sum=0.0;
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
889
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
890 for(j=0; j<filter_size; j++){
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
891 int index= i+j-filter_size/2;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
892 double d= index-i;
1687
cdc3d4106fb6 cleanup
michael
parents: 1684
diff changeset
893 double coeff= a->qblur==0 ? 1.0 : exp(-d*d/(a->qblur * a->qblur));
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
894
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
895 if(index < 0 || index >= rcc->num_entries) continue;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
896 if(pict_type != rcc->entry[index].new_pict_type) continue;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
897 q+= qscale[index] * coeff;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
898 sum+= coeff;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
899 }
6526
737b74bec1e5 Fix typo in variable name: blured --> blurred.
diego
parents: 6525
diff changeset
900 blurred_qscale[i]= q/sum;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
901 }
2967
ef2149182f1c COSMETICS: Remove all trailing whitespace.
diego
parents: 2637
diff changeset
902
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
903 /* find expected bits */
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
904 for(i=0; i<rcc->num_entries; i++){
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
905 RateControlEntry *rce= &rcc->entry[i];
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
906 double bits;
6526
737b74bec1e5 Fix typo in variable name: blured --> blurred.
diego
parents: 6525
diff changeset
907 rce->new_qscale= modify_qscale(s, rce, blurred_qscale[i], i);
616
0fe52ab8042c forgot the const bits in 2pass curve matching (patch (with rounding removed) by Rmi Guyomarch <rguyom at pobox dot com>)
michaelni
parents: 615
diff changeset
908 bits= qp2bits(rce, rce->new_qscale) + rce->mv_bits + rce->misc_bits;
6526
737b74bec1e5 Fix typo in variable name: blured --> blurred.
diego
parents: 6525
diff changeset
909 //printf("%d %f\n", rce->new_bits, blurred_qscale[i]);
1684
19e781619e3f stuffing to stay above min_bitrate
michael
parents: 1683
diff changeset
910 bits += 8*ff_vbv_update(s, bits);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
911
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
912 rce->expected_bits= expected_bits;
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
913 expected_bits += bits;
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
914 }
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
915
3681
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
916 /*
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
917 av_log(s->avctx, AV_LOG_INFO,
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
918 "expected_bits: %f all_available_bits: %d rate_factor: %f\n",
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
919 expected_bits, (int)all_available_bits, rate_factor);
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
920 */
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
921 if(expected_bits > all_available_bits) {
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
922 rate_factor-= step;
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
923 ++toobig;
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
924 }
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
925 }
1031
19de1445beb2 use av_malloc() functions - added av_strdup and av_realloc()
bellard
parents: 977
diff changeset
926 av_free(qscale);
6526
737b74bec1e5 Fix typo in variable name: blured --> blurred.
diego
parents: 6525
diff changeset
927 av_free(blurred_qscale);
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
928
3681
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
929 /* check bitrate calculations and print info */
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
930 qscale_sum = 0.0;
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
931 for(i=0; i<rcc->num_entries; i++){
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
932 /* av_log(s->avctx, AV_LOG_DEBUG, "[lavc rc] entry[%d].new_qscale = %.3f qp = %.3f\n",
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
933 i, rcc->entry[i].new_qscale, rcc->entry[i].new_qscale / FF_QP2LAMBDA); */
4594
a96d905dcbaa Add av_ prefix to clip functions
reimar
parents: 4588
diff changeset
934 qscale_sum += av_clip(rcc->entry[i].new_qscale / FF_QP2LAMBDA, s->avctx->qmin, s->avctx->qmax);
3681
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
935 }
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
936 assert(toobig <= 40);
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
937 av_log(s->avctx, AV_LOG_DEBUG,
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
938 "[lavc rc] requested bitrate: %d bps expected bitrate: %d bps\n",
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
939 s->bit_rate,
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
940 (int)(expected_bits / ((double)all_available_bits/s->bit_rate)));
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
941 av_log(s->avctx, AV_LOG_DEBUG,
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
942 "[lavc rc] estimated target average qp: %.3f\n",
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
943 (float)qscale_sum / rcc->num_entries);
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
944 if (toobig == 0) {
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
945 av_log(s->avctx, AV_LOG_INFO,
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
946 "[lavc rc] Using all of requested bitrate is not "
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
947 "necessary for this video with these parameters.\n");
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
948 } else if (toobig == 40) {
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
949 av_log(s->avctx, AV_LOG_ERROR,
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
950 "[lavc rc] Error: bitrate too low for this video "
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
951 "with these parameters.\n");
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
952 return -1;
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
953 } else if (fabs(expected_bits/all_available_bits - 1.0) > 0.01) {
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
954 av_log(s->avctx, AV_LOG_ERROR,
4f547f0da680 second pass encoding changes:
corey
parents: 3676
diff changeset
955 "[lavc rc] Error: 2pass curve failed to converge\n");
612
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
956 return -1;
c0005de2be59 new ratecontrol code
michaelni
parents: 463
diff changeset
957 }
329
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
958
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
959 return 0;
5cc47d0ba53e fixed ratecontrol & b-frames
michaelni
parents:
diff changeset
960 }