Mercurial > mplayer.hg
annotate libmpcodecs/ae_lame.c @ 28939:cbf56e4c9662
Check mpi type before returning an DR buffer in get_image, fixes jerkiness
with MPEG1/2 and -dr -slices
author | reimar |
---|---|
date | Mon, 16 Mar 2009 14:25:03 +0000 |
parents | 1f4b3aaefcd8 |
children | 0f1b5b68af32 |
rev | line source |
---|---|
15234 | 1 #include <stdio.h> |
2 #include <stdlib.h> | |
3 #include <inttypes.h> | |
15238 | 4 #include <unistd.h> |
15234 | 5 #include <string.h> |
15240 | 6 #include <sys/types.h> |
15234 | 7 #include "m_option.h" |
17012 | 8 #include "mp_msg.h" |
22601
ed8f90096c65
Add explicit location for headers from the libmpdemux/ directory.
diego
parents:
22600
diff
changeset
|
9 #include "libmpdemux/aviheader.h" |
ed8f90096c65
Add explicit location for headers from the libmpdemux/ directory.
diego
parents:
22600
diff
changeset
|
10 #include "libmpdemux/ms_hdr.h" |
22600
3c2b4a866c6a
Add explicit location for headers from the stream/ directory.
diego
parents:
21660
diff
changeset
|
11 #include "stream/stream.h" |
22601
ed8f90096c65
Add explicit location for headers from the libmpdemux/ directory.
diego
parents:
22600
diff
changeset
|
12 #include "libmpdemux/muxer.h" |
17012 | 13 #include "help_mp.h" |
15234 | 14 #include "ae_pcm.h" |
17012 | 15 #include "libaf/af_format.h" |
16 #include "libmpdemux/mp3_hdr.h" | |
15234 | 17 |
18 #undef CDECL | |
19 #include <lame/lame.h> | |
20 | |
21 lame_global_flags *lame; | |
22 static int lame_param_quality=0; // best | |
23 static int lame_param_algqual=5; // same as old default | |
24 static int lame_param_vbr=vbr_default; | |
25 static int lame_param_mode=-1; // unset | |
26 static int lame_param_padding=-1; // unset | |
27 static int lame_param_br=-1; // unset | |
28 static int lame_param_ratio=-1; // unset | |
29 static float lame_param_scale=-1; // unset | |
30 static int lame_param_lowpassfreq = 0; //auto | |
31 static int lame_param_highpassfreq = 0; //auto | |
32 static int lame_param_free_format = 0; //disabled | |
33 static int lame_param_br_min = 0; //not specified | |
34 static int lame_param_br_max = 0; //not specified | |
35 | |
27363
40057010b1fa
Change a bunch of codec-specific preprocessor directives from a HAVE_
diego
parents:
26754
diff
changeset
|
36 #ifdef CONFIG_MP3LAME_PRESET |
15234 | 37 int lame_param_fast=0; // unset |
38 static char* lame_param_preset=NULL; // unset | |
39 static int lame_presets_set( lame_t gfp, int fast, int cbr, const char* preset_name ); | |
40 #endif | |
41 | |
42 | |
43 m_option_t lameopts_conf[]={ | |
44 {"q", &lame_param_quality, CONF_TYPE_INT, CONF_RANGE, 0, 9, NULL}, | |
45 {"aq", &lame_param_algqual, CONF_TYPE_INT, CONF_RANGE, 0, 9, NULL}, | |
46 {"vbr", &lame_param_vbr, CONF_TYPE_INT, CONF_RANGE, 0, vbr_max_indicator, NULL}, | |
47 {"cbr", &lame_param_vbr, CONF_TYPE_FLAG, 0, 0, 0, NULL}, | |
48 {"abr", &lame_param_vbr, CONF_TYPE_FLAG, 0, 0, vbr_abr, NULL}, | |
49 {"mode", &lame_param_mode, CONF_TYPE_INT, CONF_RANGE, 0, MAX_INDICATOR, NULL}, | |
50 {"padding", &lame_param_padding, CONF_TYPE_INT, CONF_RANGE, 0, PAD_MAX_INDICATOR, NULL}, | |
51 {"br", &lame_param_br, CONF_TYPE_INT, CONF_RANGE, 0, 1024, NULL}, | |
52 {"ratio", &lame_param_ratio, CONF_TYPE_INT, CONF_RANGE, 0, 100, NULL}, | |
53 {"vol", &lame_param_scale, CONF_TYPE_FLOAT, CONF_RANGE, 0, 10, NULL}, | |
54 {"lowpassfreq",&lame_param_lowpassfreq, CONF_TYPE_INT, CONF_RANGE, -1, 48000,0}, | |
55 {"highpassfreq",&lame_param_highpassfreq, CONF_TYPE_INT, CONF_RANGE, -1, 48000,0}, | |
56 {"nofree", &lame_param_free_format, CONF_TYPE_FLAG, 0, 0, 0, NULL}, | |
57 {"free", &lame_param_free_format, CONF_TYPE_FLAG, 0, 0, 1, NULL}, | |
58 {"br_min", &lame_param_br_min, CONF_TYPE_INT, CONF_RANGE, 0, 1024, NULL}, | |
59 {"br_max", &lame_param_br_max, CONF_TYPE_INT, CONF_RANGE, 0, 1024, NULL}, | |
27363
40057010b1fa
Change a bunch of codec-specific preprocessor directives from a HAVE_
diego
parents:
26754
diff
changeset
|
60 #ifdef CONFIG_MP3LAME_PRESET |
15234 | 61 {"fast", &lame_param_fast, CONF_TYPE_FLAG, 0, 0, 1, NULL}, |
62 {"preset", &lame_param_preset, CONF_TYPE_STRING, 0, 0, 0, NULL}, | |
63 #else | |
64 {"fast", "MPlayer was built without -lameopts fast support (requires libmp3lame >=3.92).\n", CONF_TYPE_PRINT, CONF_NOCFG, 0, 0, NULL}, | |
65 {"preset", "MPlayer was built without -lameopts preset support (requires libmp3lame >=3.92).\n", CONF_TYPE_PRINT, CONF_NOCFG, 0, 0, NULL}, | |
66 #endif | |
67 {"help", MSGTR_MEncoderMP3LameHelp, CONF_TYPE_PRINT, CONF_NOCFG, 0, 0, NULL}, | |
68 {NULL, NULL, 0, 0, 0, 0, NULL} | |
69 }; | |
70 | |
71 | |
72 static int bind_lame(audio_encoder_t *encoder, muxer_stream_t *mux_a) | |
73 { | |
21621 | 74 mp_msg(MSGT_MENCODER, MSGL_INFO, MSGTR_MP3AudioSelected); |
15234 | 75 mux_a->h.dwSampleSize=0; // VBR |
76 mux_a->h.dwRate=encoder->params.sample_rate; | |
77 mux_a->h.dwScale=encoder->params.samples_per_frame; // samples/frame | |
78 if(sizeof(MPEGLAYER3WAVEFORMAT)!=30) mp_msg(MSGT_MENCODER,MSGL_WARN,MSGTR_MP3WaveFormatSizeNot30,sizeof(MPEGLAYER3WAVEFORMAT)); | |
79 mux_a->wf=malloc(sizeof(MPEGLAYER3WAVEFORMAT)); // should be 30 | |
80 mux_a->wf->wFormatTag=0x55; // MP3 | |
81 mux_a->wf->nChannels= (lame_param_mode<0) ? encoder->params.channels : ((lame_param_mode==3) ? 1 : 2); | |
82 mux_a->wf->nSamplesPerSec=mux_a->h.dwRate; | |
83 if(! lame_param_vbr) | |
84 mux_a->wf->nAvgBytesPerSec=lame_param_br * 125; | |
85 else | |
86 mux_a->wf->nAvgBytesPerSec=192000/8; // FIXME! | |
87 mux_a->wf->nBlockAlign=encoder->params.samples_per_frame; // required for l3codeca.acm + WMP 6.4 | |
88 mux_a->wf->wBitsPerSample=0; //16; | |
89 // from NaNdub: (requires for l3codeca.acm) | |
90 mux_a->wf->cbSize=12; | |
91 ((MPEGLAYER3WAVEFORMAT*)(mux_a->wf))->wID=1; | |
92 ((MPEGLAYER3WAVEFORMAT*)(mux_a->wf))->fdwFlags=2; | |
93 ((MPEGLAYER3WAVEFORMAT*)(mux_a->wf))->nBlockSize=encoder->params.samples_per_frame; // ??? | |
94 ((MPEGLAYER3WAVEFORMAT*)(mux_a->wf))->nFramesPerBlock=1; | |
95 ((MPEGLAYER3WAVEFORMAT*)(mux_a->wf))->nCodecDelay=0; | |
96 | |
19397
0074c9c77a0e
Set encoder's input format to "native endiannes", instead of little endian.
gpoirier
parents:
17932
diff
changeset
|
97 encoder->input_format = AF_FORMAT_S16_NE; |
15234 | 98 encoder->min_buffer_size = 4608; |
99 encoder->max_buffer_size = mux_a->h.dwRate * mux_a->wf->nChannels * 2; | |
100 | |
101 return 1; | |
102 } | |
103 | |
104 #define min(a, b) ((a) <= (b) ? (a) : (b)) | |
105 | |
106 static int get_frame_size(audio_encoder_t *encoder) | |
107 { | |
108 int sz; | |
109 if(encoder->stream->buffer_len < 4) | |
110 return 0; | |
111 sz = mp_decode_mp3_header(encoder->stream->buffer); | |
112 if(sz <= 0) | |
113 return 0; | |
114 return sz; | |
115 } | |
116 | |
117 static int encode_lame(audio_encoder_t *encoder, uint8_t *dest, void *src, int len, int max_size) | |
118 { | |
119 int n = 0; | |
120 if(encoder->params.channels == 1) | |
121 n = lame_encode_buffer(lame, (short *)src, (short *)src, len/2, dest, max_size); | |
122 else | |
123 n = lame_encode_buffer_interleaved(lame,(short *)src, len/4, dest, max_size); | |
124 | |
26754
63630c09e237
cosmetics: Remove pointless parentheses from return calls.
diego
parents:
25503
diff
changeset
|
125 return n < 0 ? 0 : n; |
15234 | 126 } |
127 | |
128 | |
129 static int close_lame(audio_encoder_t *encoder) | |
130 { | |
131 return 1; | |
132 } | |
133 | |
134 static void fixup(audio_encoder_t *encoder) | |
135 { | |
136 // fixup CBR mp3 audio header: | |
137 if(!lame_param_vbr) { | |
138 encoder->stream->h.dwSampleSize=1; | |
28347
1f4b3aaefcd8
Avoid a division by 0 when using -oac mp3lame but no audio data actually is encoded.
reimar
parents:
27363
diff
changeset
|
139 if (encoder->stream->h.dwLength) |
15234 | 140 ((MPEGLAYER3WAVEFORMAT*)(encoder->stream->wf))->nBlockSize= |
141 (encoder->stream->size+(encoder->stream->h.dwLength>>1))/encoder->stream->h.dwLength; | |
142 encoder->stream->h.dwLength=encoder->stream->size; | |
143 encoder->stream->h.dwRate=encoder->stream->wf->nAvgBytesPerSec; | |
144 encoder->stream->h.dwScale=1; | |
145 encoder->stream->wf->nBlockAlign=1; | |
21621 | 146 mp_msg(MSGT_MENCODER, MSGL_V, MSGTR_CBRAudioByterate, |
15234 | 147 encoder->stream->h.dwRate,((MPEGLAYER3WAVEFORMAT*)(encoder->stream->wf))->nBlockSize); |
148 } | |
149 } | |
150 | |
151 int mpae_init_lame(audio_encoder_t *encoder) | |
152 { | |
153 encoder->params.bitrate = lame_param_br * 125; | |
154 encoder->params.samples_per_frame = encoder->params.sample_rate < 32000 ? 576 : 1152; | |
155 encoder->decode_buffer_size = 2304; | |
156 | |
157 lame=lame_init(); | |
158 lame_set_bWriteVbrTag(lame,0); | |
159 lame_set_in_samplerate(lame,encoder->params.sample_rate); | |
160 //lame_set_in_samplerate(lame,sh_audio->samplerate); // if resampling done by lame | |
161 lame_set_num_channels(lame,encoder->params.channels); | |
162 lame_set_out_samplerate(lame,encoder->params.sample_rate); | |
163 lame_set_quality(lame,lame_param_algqual); // 0 = best q | |
164 if(lame_param_free_format) lame_set_free_format(lame,1); | |
165 if(lame_param_vbr){ // VBR: | |
166 lame_set_VBR(lame,lame_param_vbr); // vbr mode | |
167 lame_set_VBR_q(lame,lame_param_quality); // 0 = best vbr q 5=~128k | |
168 if(lame_param_br>0) lame_set_VBR_mean_bitrate_kbps(lame,lame_param_br); | |
169 if(lame_param_br_min>0) lame_set_VBR_min_bitrate_kbps(lame,lame_param_br_min); | |
170 if(lame_param_br_max>0) lame_set_VBR_max_bitrate_kbps(lame,lame_param_br_max); | |
171 } else { // CBR: | |
172 if(lame_param_br>0) lame_set_brate(lame,lame_param_br); | |
173 } | |
174 if(lame_param_mode>=0) lame_set_mode(lame,lame_param_mode); // j-st | |
175 if(lame_param_ratio>0) lame_set_compression_ratio(lame,lame_param_ratio); | |
176 if(lame_param_scale>0) { | |
21621 | 177 mp_msg(MSGT_MENCODER, MSGL_V, MSGTR_SettingAudioInputGain, lame_param_scale); |
15234 | 178 lame_set_scale(lame,lame_param_scale); |
179 } | |
180 if(lame_param_lowpassfreq>=-1) lame_set_lowpassfreq(lame,lame_param_lowpassfreq); | |
181 if(lame_param_highpassfreq>=-1) lame_set_highpassfreq(lame,lame_param_highpassfreq); | |
27363
40057010b1fa
Change a bunch of codec-specific preprocessor directives from a HAVE_
diego
parents:
26754
diff
changeset
|
182 #ifdef CONFIG_MP3LAME_PRESET |
15234 | 183 if(lame_param_preset != NULL) { |
21621 | 184 mp_msg(MSGT_MENCODER, MSGL_V, MSGTR_LamePresetEquals,lame_param_preset); |
15234 | 185 if(lame_presets_set(lame,lame_param_fast, (lame_param_vbr==0), lame_param_preset) < 0) |
186 return 0; | |
187 } | |
188 #endif | |
189 if(lame_init_params(lame) == -1) { | |
190 mp_msg(MSGT_MENCODER, MSGL_FATAL, MSGTR_LameCantInit); | |
191 return 0; | |
192 } | |
17932 | 193 if( mp_msg_test(MSGT_MENCODER,MSGL_V) ) { |
15234 | 194 lame_print_config(lame); |
195 lame_print_internals(lame); | |
196 } | |
197 | |
198 encoder->bind = bind_lame; | |
199 encoder->get_frame_size = get_frame_size; | |
200 encoder->encode = encode_lame; | |
201 encoder->fixup = fixup; | |
202 encoder->close = close_lame; | |
203 return 1; | |
204 } | |
205 | |
27363
40057010b1fa
Change a bunch of codec-specific preprocessor directives from a HAVE_
diego
parents:
26754
diff
changeset
|
206 #ifdef CONFIG_MP3LAME_PRESET |
15234 | 207 /* lame_presets_set |
208 taken out of presets_set in lame-3.93.1/frontend/parse.c and modified */ | |
209 static int lame_presets_set( lame_t gfp, int fast, int cbr, const char* preset_name ) | |
210 { | |
211 int mono = 0; | |
212 | |
213 if (strcmp(preset_name, "help") == 0) { | |
21621 | 214 mp_msg(MSGT_MENCODER, MSGL_FATAL, MSGTR_LameVersion, get_lame_version(), get_lame_url()); |
21622
5cd2a0041019
Simplify: get rid of useless lame_presets_longinfo_dm function.
reimar
parents:
21621
diff
changeset
|
215 mp_msg(MSGT_MENCODER, MSGL_FATAL, MSGTR_LamePresetsLongInfo); |
15234 | 216 return -1; |
217 } | |
218 | |
219 //aliases for compatibility with old presets | |
220 | |
221 if (strcmp(preset_name, "phone") == 0) { | |
222 preset_name = "16"; | |
223 mono = 1; | |
224 } | |
225 if ( (strcmp(preset_name, "phon+") == 0) || | |
226 (strcmp(preset_name, "lw") == 0) || | |
227 (strcmp(preset_name, "mw-eu") == 0) || | |
228 (strcmp(preset_name, "sw") == 0)) { | |
229 preset_name = "24"; | |
230 mono = 1; | |
231 } | |
232 if (strcmp(preset_name, "mw-us") == 0) { | |
233 preset_name = "40"; | |
234 mono = 1; | |
235 } | |
236 if (strcmp(preset_name, "voice") == 0) { | |
237 preset_name = "56"; | |
238 mono = 1; | |
239 } | |
240 if (strcmp(preset_name, "fm") == 0) { | |
241 preset_name = "112"; | |
242 } | |
243 if ( (strcmp(preset_name, "radio") == 0) || | |
244 (strcmp(preset_name, "tape") == 0)) { | |
245 preset_name = "112"; | |
246 } | |
247 if (strcmp(preset_name, "hifi") == 0) { | |
248 preset_name = "160"; | |
249 } | |
250 if (strcmp(preset_name, "cd") == 0) { | |
251 preset_name = "192"; | |
252 } | |
253 if (strcmp(preset_name, "studio") == 0) { | |
254 preset_name = "256"; | |
255 } | |
256 | |
27363
40057010b1fa
Change a bunch of codec-specific preprocessor directives from a HAVE_
diego
parents:
26754
diff
changeset
|
257 #ifdef CONFIG_MP3LAME_PRESET_MEDIUM |
15234 | 258 if (strcmp(preset_name, "medium") == 0) { |
259 if (fast > 0) | |
260 lame_set_preset(gfp, MEDIUM_FAST); | |
261 else | |
262 lame_set_preset(gfp, MEDIUM); | |
263 | |
264 return 0; | |
265 } | |
266 #endif | |
267 | |
268 if (strcmp(preset_name, "standard") == 0) { | |
269 if (fast > 0) | |
270 lame_set_preset(gfp, STANDARD_FAST); | |
271 else | |
272 lame_set_preset(gfp, STANDARD); | |
273 | |
274 return 0; | |
275 } | |
276 | |
277 else if (strcmp(preset_name, "extreme") == 0){ | |
278 if (fast > 0) | |
279 lame_set_preset(gfp, EXTREME_FAST); | |
280 else | |
281 lame_set_preset(gfp, EXTREME); | |
282 | |
283 return 0; | |
284 } | |
285 | |
286 else if (((strcmp(preset_name, "insane") == 0) || | |
287 (strcmp(preset_name, "320" ) == 0)) && (fast < 1)) { | |
288 | |
289 lame_set_preset(gfp, INSANE); | |
290 | |
291 return 0; | |
292 } | |
293 | |
294 // Generic ABR Preset | |
295 if (((atoi(preset_name)) > 0) && (fast < 1)) { | |
296 if ((atoi(preset_name)) >= 8 && (atoi(preset_name)) <= 320){ | |
297 lame_set_preset(gfp, atoi(preset_name)); | |
298 | |
299 if (cbr == 1 ) | |
300 lame_set_VBR(gfp, vbr_off); | |
301 | |
302 if (mono == 1 ) { | |
303 lame_set_mode(gfp, MONO); | |
304 } | |
305 | |
306 return 0; | |
307 | |
308 } | |
309 else { | |
21621 | 310 mp_msg(MSGT_MENCODER, MSGL_FATAL, MSGTR_LameVersion, get_lame_version(), get_lame_url()); |
311 mp_msg(MSGT_MENCODER, MSGL_FATAL, MSGTR_InvalidBitrateForLamePreset); | |
15234 | 312 return -1; |
313 } | |
314 } | |
315 | |
21621 | 316 mp_msg(MSGT_MENCODER, MSGL_FATAL, MSGTR_LameVersion, get_lame_version(), get_lame_url()); |
317 mp_msg(MSGT_MENCODER, MSGL_FATAL, MSGTR_InvalidLamePresetOptions); | |
15234 | 318 return -1; |
319 } | |
320 #endif |