Mercurial > libavcodec.hg
annotate xvid_rc.c @ 3988:3041b7b5476c libavcodec
Move CFLAGS handling to common.mak.
author | diego |
---|---|
date | Wed, 11 Oct 2006 08:30:13 +0000 |
parents | c8c591fe26f8 |
children | f7e99d20ef62 |
rev | line source |
---|---|
3064 | 1 /* |
2 * xvid Rate control wrapper for lavc video encoders | |
3 * | |
4 * Copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at> | |
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 | 9 * modify it under the terms of the GNU Lesser General Public |
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 | 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 | 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 * Lesser General Public License for more details. | |
17 * | |
18 * You should have received a copy of the GNU Lesser General Public | |
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 | 20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
21 */ | |
22 | |
23 #include <xvid.h> | |
24 #include <unistd.h> | |
25 #include "avcodec.h" | |
3233
18af2f7788c6
- Add new file internal.h for common internal-use-only functions.
corey
parents:
3204
diff
changeset
|
26 #include "internal.h" |
3064 | 27 //#include "dsputil.h" |
28 #include "mpegvideo.h" | |
29 | |
30 #undef NDEBUG | |
31 #include <assert.h> | |
32 | |
33 extern unsigned int xvid_debug; | |
34 | |
35 int ff_xvid_rate_control_init(MpegEncContext *s){ | |
36 char *tmp_name; | |
37 int fd, i; | |
38 xvid_plg_create_t xvid_plg_create; | |
39 xvid_plugin_2pass2_t xvid_2pass2; | |
40 | |
41 //xvid_debug=-1; | |
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 | 47 } |
48 | |
3204 | 49 for(i=0; i<s->rc_context.num_entries; i++){ |
3064 | 50 static const char *frame_types = " ipbs"; |
51 char tmp[256]; | |
52 RateControlEntry *rce; | |
53 | |
54 rce= &s->rc_context.entry[i]; | |
55 | |
56 snprintf(tmp, sizeof(tmp), "%c %d %d %d %d %d %d\n", | |
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 | 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 | 59 |
3204 | 60 //av_log(NULL, AV_LOG_ERROR, "%s\n", tmp); |
3064 | 61 write(fd, tmp, strlen(tmp)); |
62 } | |
63 | |
64 close(fd); | |
65 | |
66 memset(&xvid_2pass2, 0, sizeof(xvid_2pass2)); | |
67 xvid_2pass2.version= XVID_MAKE_VERSION(1,1,0); | |
68 xvid_2pass2.filename= tmp_name; | |
69 xvid_2pass2.bitrate= s->avctx->bit_rate; | |
70 xvid_2pass2.vbv_size= s->avctx->rc_buffer_size; | |
71 xvid_2pass2.vbv_maxrate= s->avctx->rc_max_rate; | |
72 xvid_2pass2.vbv_initial= s->avctx->rc_initial_buffer_occupancy; | |
73 | |
74 memset(&xvid_plg_create, 0, sizeof(xvid_plg_create)); | |
75 xvid_plg_create.version= XVID_MAKE_VERSION(1,1,0); | |
76 xvid_plg_create.fbase= s->avctx->time_base.den; | |
77 xvid_plg_create.fincr= s->avctx->time_base.num; | |
78 xvid_plg_create.param= &xvid_2pass2; | |
79 | |
3204 | 80 if(xvid_plugin_2pass2(NULL, XVID_PLG_CREATE, &xvid_plg_create, &s->rc_context.non_lavc_opaque)<0){ |
81 av_log(NULL, AV_LOG_ERROR, "xvid_plugin_2pass2 failed\n"); | |
82 return -1; | |
83 } | |
84 return 0; | |
3064 | 85 } |
86 | |
87 float ff_xvid_rate_estimate_qscale(MpegEncContext *s, int dry_run){ | |
88 xvid_plg_data_t xvid_plg_data; | |
89 | |
90 memset(&xvid_plg_data, 0, sizeof(xvid_plg_data)); | |
91 xvid_plg_data.version= XVID_MAKE_VERSION(1,1,0); | |
92 xvid_plg_data.width = s->width; | |
93 xvid_plg_data.height= s->height; | |
94 xvid_plg_data.mb_width = s->mb_width; | |
95 xvid_plg_data.mb_height= s->mb_height; | |
96 xvid_plg_data.fbase= s->avctx->time_base.den; | |
97 xvid_plg_data.fincr= s->avctx->time_base.num; | |
98 xvid_plg_data.min_quant[0]= s->avctx->qmin; | |
99 xvid_plg_data.min_quant[1]= s->avctx->qmin; | |
100 xvid_plg_data.min_quant[2]= s->avctx->qmin; //FIXME i/b factor & offset | |
101 xvid_plg_data.max_quant[0]= s->avctx->qmax; | |
102 xvid_plg_data.max_quant[1]= s->avctx->qmax; | |
103 xvid_plg_data.max_quant[2]= s->avctx->qmax; //FIXME i/b factor & offset | |
3204 | 104 xvid_plg_data.bquant_offset = 0; // 100 * s->avctx->b_quant_offset; |
105 xvid_plg_data.bquant_ratio = 100; // * s->avctx->b_quant_factor; | |
3064 | 106 |
107 #if 0 | |
108 xvid_plg_data.stats.hlength= X | |
109 #endif | |
110 | |
111 if(!s->rc_context.dry_run_qscale){ | |
112 if(s->picture_number){ | |
113 xvid_plg_data.length= | |
114 xvid_plg_data.stats.length= (s->frame_bits + 7)/8; | |
3204 | 115 xvid_plg_data.frame_num= s->rc_context.last_picture_number; |
3064 | 116 xvid_plg_data.quant= s->qscale; |
117 | |
118 xvid_plg_data.type= s->last_pict_type; | |
119 if(xvid_plugin_2pass2(s->rc_context.non_lavc_opaque, XVID_PLG_AFTER, &xvid_plg_data, NULL)){ | |
120 av_log(s->avctx, AV_LOG_ERROR, "xvid_plugin_2pass2(handle, XVID_PLG_AFTER, ...) FAILED\n"); | |
121 return -1; | |
122 } | |
123 } | |
3204 | 124 s->rc_context.last_picture_number= |
3064 | 125 xvid_plg_data.frame_num= s->picture_number; |
126 xvid_plg_data.quant= 0; | |
127 if(xvid_plugin_2pass2(s->rc_context.non_lavc_opaque, XVID_PLG_BEFORE, &xvid_plg_data, NULL)){ | |
128 av_log(s->avctx, AV_LOG_ERROR, "xvid_plugin_2pass2(handle, XVID_PLG_BEFORE, ...) FAILED\n"); | |
129 return -1; | |
130 } | |
131 s->rc_context.dry_run_qscale= xvid_plg_data.quant; | |
132 } | |
133 xvid_plg_data.quant= s->rc_context.dry_run_qscale; | |
134 if(!dry_run) | |
135 s->rc_context.dry_run_qscale= 0; | |
136 | |
3204 | 137 if(s->pict_type == B_TYPE) //FIXME this is not exactly identical to xvid |
138 return xvid_plg_data.quant * FF_QP2LAMBDA * s->avctx->b_quant_factor + s->avctx->b_quant_offset; | |
139 else | |
140 return xvid_plg_data.quant * FF_QP2LAMBDA; | |
3064 | 141 } |
142 | |
143 void ff_xvid_rate_control_uninit(MpegEncContext *s){ | |
144 xvid_plg_destroy_t xvid_plg_destroy; | |
145 | |
146 xvid_plugin_2pass2(s->rc_context.non_lavc_opaque, XVID_PLG_DESTROY, &xvid_plg_destroy, NULL); | |
147 } | |
148 |