annotate xvid_rc.c @ 4166:eced83504436 libavcodec

mp3 header (de)compression bitstream filter this will make mp3 frames 4 bytes smaller, it will not give you binary identical mp3 files, but it will give you mp3 files which decode to binary identical output this will only work in containers providing at least packet size, sample_rate and number of channels bugreports about mp3 files for which this fails are welcome and this is experimental (dont expect compatibility and dont even expect to be able to decompress what you compressed, hell dont even expect this to work without editing the source a little)
author michael
date Fri, 10 Nov 2006 01:41:53 +0000
parents f7e99d20ef62
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
1 /*
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
2 * xvid Rate control wrapper for lavc video encoders
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
3 *
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
4 * Copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at>
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
5 *
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3233
diff changeset
6 * This file is part of FFmpeg.
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3233
diff changeset
7 *
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3233
diff changeset
8 * FFmpeg is free software; you can redistribute it and/or
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
9 * modify it under the terms of the GNU Lesser General Public
a5e0b58b4471 xvid ratecontrol support
michael
parents:
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: 3233
diff changeset
11 * version 2.1 of the License, or (at your option) any later version.
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
12 *
3947
c8c591fe26f8 Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 3233
diff changeset
13 * FFmpeg is distributed in the hope that it will be useful,
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
16 * Lesser General Public License for more details.
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
17 *
a5e0b58b4471 xvid ratecontrol support
michael
parents:
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: 3233
diff changeset
19 * License along with FFmpeg; if not, write to the Free Software
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
21 */
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
22
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
23 #include <xvid.h>
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
24 #include <unistd.h>
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
25 #include "avcodec.h"
4004
f7e99d20ef62 Rename internal.h --> xvid_internal.h, it clashes with libavutil/internal.h.
diego
parents: 3947
diff changeset
26 #include "xvid_internal.h"
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
27 //#include "dsputil.h"
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
28 #include "mpegvideo.h"
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
29
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
30 #undef NDEBUG
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
31 #include <assert.h>
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
32
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
33 extern unsigned int xvid_debug;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
34
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
35 int ff_xvid_rate_control_init(MpegEncContext *s){
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
36 char *tmp_name;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
37 int fd, i;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
38 xvid_plg_create_t xvid_plg_create;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
39 xvid_plugin_2pass2_t xvid_2pass2;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
40
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
41 //xvid_debug=-1;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
42
3233
18af2f7788c6 - Add new file internal.h for common internal-use-only functions.
corey
parents: 3204
diff changeset
43 fd=av_tempfile("xvidrc.", &tmp_name);
18af2f7788c6 - Add new file internal.h for common internal-use-only functions.
corey
parents: 3204
diff changeset
44 if (fd == -1) {
18af2f7788c6 - Add new file internal.h for common internal-use-only functions.
corey
parents: 3204
diff changeset
45 av_log(NULL, AV_LOG_ERROR, "Can't create temporary pass2 file.\n");
18af2f7788c6 - Add new file internal.h for common internal-use-only functions.
corey
parents: 3204
diff changeset
46 return -1;
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
47 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
48
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
49 for(i=0; i<s->rc_context.num_entries; i++){
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
50 static const char *frame_types = " ipbs";
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
51 char tmp[256];
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
52 RateControlEntry *rce;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
53
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
54 rce= &s->rc_context.entry[i];
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
55
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
56 snprintf(tmp, sizeof(tmp), "%c %d %d %d %d %d %d\n",
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
57 frame_types[rce->pict_type], (int)lrintf(rce->qscale / FF_QP2LAMBDA), rce->i_count, s->mb_num - rce->i_count - rce->skip_count,
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
58 rce->skip_count, (rce->i_tex_bits + rce->p_tex_bits + rce->misc_bits+7)/8, (rce->header_bits+rce->mv_bits+7)/8);
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
59
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
60 //av_log(NULL, AV_LOG_ERROR, "%s\n", tmp);
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
61 write(fd, tmp, strlen(tmp));
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
62 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
63
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
64 close(fd);
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
65
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
66 memset(&xvid_2pass2, 0, sizeof(xvid_2pass2));
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
67 xvid_2pass2.version= XVID_MAKE_VERSION(1,1,0);
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
68 xvid_2pass2.filename= tmp_name;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
69 xvid_2pass2.bitrate= s->avctx->bit_rate;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
70 xvid_2pass2.vbv_size= s->avctx->rc_buffer_size;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
71 xvid_2pass2.vbv_maxrate= s->avctx->rc_max_rate;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
72 xvid_2pass2.vbv_initial= s->avctx->rc_initial_buffer_occupancy;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
73
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
74 memset(&xvid_plg_create, 0, sizeof(xvid_plg_create));
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
75 xvid_plg_create.version= XVID_MAKE_VERSION(1,1,0);
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
76 xvid_plg_create.fbase= s->avctx->time_base.den;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
77 xvid_plg_create.fincr= s->avctx->time_base.num;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
78 xvid_plg_create.param= &xvid_2pass2;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
79
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
80 if(xvid_plugin_2pass2(NULL, XVID_PLG_CREATE, &xvid_plg_create, &s->rc_context.non_lavc_opaque)<0){
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
81 av_log(NULL, AV_LOG_ERROR, "xvid_plugin_2pass2 failed\n");
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
82 return -1;
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
83 }
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
84 return 0;
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
85 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
86
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
87 float ff_xvid_rate_estimate_qscale(MpegEncContext *s, int dry_run){
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
88 xvid_plg_data_t xvid_plg_data;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
89
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
90 memset(&xvid_plg_data, 0, sizeof(xvid_plg_data));
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
91 xvid_plg_data.version= XVID_MAKE_VERSION(1,1,0);
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
92 xvid_plg_data.width = s->width;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
93 xvid_plg_data.height= s->height;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
94 xvid_plg_data.mb_width = s->mb_width;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
95 xvid_plg_data.mb_height= s->mb_height;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
96 xvid_plg_data.fbase= s->avctx->time_base.den;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
97 xvid_plg_data.fincr= s->avctx->time_base.num;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
98 xvid_plg_data.min_quant[0]= s->avctx->qmin;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
99 xvid_plg_data.min_quant[1]= s->avctx->qmin;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
100 xvid_plg_data.min_quant[2]= s->avctx->qmin; //FIXME i/b factor & offset
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
101 xvid_plg_data.max_quant[0]= s->avctx->qmax;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
102 xvid_plg_data.max_quant[1]= s->avctx->qmax;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
103 xvid_plg_data.max_quant[2]= s->avctx->qmax; //FIXME i/b factor & offset
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
104 xvid_plg_data.bquant_offset = 0; // 100 * s->avctx->b_quant_offset;
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
105 xvid_plg_data.bquant_ratio = 100; // * s->avctx->b_quant_factor;
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
106
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
107 #if 0
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
108 xvid_plg_data.stats.hlength= X
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
109 #endif
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
110
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
111 if(!s->rc_context.dry_run_qscale){
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
112 if(s->picture_number){
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
113 xvid_plg_data.length=
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
114 xvid_plg_data.stats.length= (s->frame_bits + 7)/8;
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
115 xvid_plg_data.frame_num= s->rc_context.last_picture_number;
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
116 xvid_plg_data.quant= s->qscale;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
117
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
118 xvid_plg_data.type= s->last_pict_type;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
119 if(xvid_plugin_2pass2(s->rc_context.non_lavc_opaque, XVID_PLG_AFTER, &xvid_plg_data, NULL)){
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
120 av_log(s->avctx, AV_LOG_ERROR, "xvid_plugin_2pass2(handle, XVID_PLG_AFTER, ...) FAILED\n");
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
121 return -1;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
122 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
123 }
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
124 s->rc_context.last_picture_number=
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
125 xvid_plg_data.frame_num= s->picture_number;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
126 xvid_plg_data.quant= 0;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
127 if(xvid_plugin_2pass2(s->rc_context.non_lavc_opaque, XVID_PLG_BEFORE, &xvid_plg_data, NULL)){
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
128 av_log(s->avctx, AV_LOG_ERROR, "xvid_plugin_2pass2(handle, XVID_PLG_BEFORE, ...) FAILED\n");
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
129 return -1;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
130 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
131 s->rc_context.dry_run_qscale= xvid_plg_data.quant;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
132 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
133 xvid_plg_data.quant= s->rc_context.dry_run_qscale;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
134 if(!dry_run)
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
135 s->rc_context.dry_run_qscale= 0;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
136
3204
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
137 if(s->pict_type == B_TYPE) //FIXME this is not exactly identical to xvid
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
138 return xvid_plg_data.quant * FF_QP2LAMBDA * s->avctx->b_quant_factor + s->avctx->b_quant_offset;
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
139 else
5fc631919a98 various fixes for xvid_rc
michael
parents: 3202
diff changeset
140 return xvid_plg_data.quant * FF_QP2LAMBDA;
3064
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
141 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
142
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
143 void ff_xvid_rate_control_uninit(MpegEncContext *s){
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
144 xvid_plg_destroy_t xvid_plg_destroy;
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
145
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
146 xvid_plugin_2pass2(s->rc_context.non_lavc_opaque, XVID_PLG_DESTROY, &xvid_plg_destroy, NULL);
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
147 }
a5e0b58b4471 xvid ratecontrol support
michael
parents:
diff changeset
148