Mercurial > mplayer.hg

comparison libmpcodecs/ae_twolame.c @ 15359:af245d15f502

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added twolame mp2 audio encoder
author nicodvb
date Sat, 07 May 2005 14:50:14 +0000
parents
children 4ff00aa141ef
comparison
equal deleted inserted replaced
15358:87121d373127 15359:af245d15f502
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <inttypes.h>
4 #include <unistd.h>
5 #include <string.h>
6 #include <sys/types.h>
7 #include "m_option.h"
8 #include "../mp_msg.h"
9 #include "aviheader.h"
10 #include "../libaf/af_format.h"
11 #include "ms_hdr.h"
12 #include "muxer.h"
13 #include "ae_twolame.h"
14 #include "../libmpdemux/mp3_hdr.h"
15
16
17 static int
18 param_bitrate = 192,
19 param_psy = 3,
20 param_maxvbr = 0,
21 param_errprot = 0,
22 param_debug = 0;
23
24 static float param_vbr = 0;
25 static char *param_mode = "stereo";
26
27 m_option_t twolameopts_conf[] = {
28 {"br", &param_bitrate, CONF_TYPE_INT, 0, 0, 0, NULL},
29 {"mode", &param_mode, CONF_TYPE_STRING, 0, 0, 0, NULL},
30 {"psy", &param_psy, CONF_TYPE_INT, CONF_RANGE, 0, 3, NULL},
31 {"vbr", &param_vbr, CONF_TYPE_FLOAT, CONF_RANGE, -50, 50, NULL},
32 {"maxvbr", &param_maxvbr, CONF_TYPE_INT, 0, 0, 0, NULL},
33 {"errprot", &param_errprot, CONF_TYPE_INT, CONF_RANGE, 0, 1, NULL},
34 {"debug", &param_debug, CONF_TYPE_INT, CONF_RANGE, 0, 100000000, NULL},
35 {NULL, NULL, 0, 0, 0, 0, NULL}
36 };
37
38
39 static int bind_twolame(audio_encoder_t *encoder, muxer_stream_t *mux_a)
40 {
41 mpae_twolame_ctx *ctx = (mpae_twolame_ctx *) encoder->priv;
42
43 mux_a->wf = malloc(sizeof(WAVEFORMATEX)+256);
44 mux_a->wf->wFormatTag = 0x50;
45 mux_a->wf->nChannels = encoder->params.channels;
46 mux_a->wf->nSamplesPerSec = encoder->params.sample_rate;
47 mux_a->wf->nAvgBytesPerSec = encoder->params.bitrate / 8;
48
49 if(ctx->vbr || ((mux_a->wf->nAvgBytesPerSec * encoder->params.samples_per_frame) % mux_a->wf->nSamplesPerSec))
50 {
51 mux_a->h.dwScale = encoder->params.samples_per_frame;
52 mux_a->h.dwRate = encoder->params.sample_rate;
53 mux_a->h.dwSampleSize = 0; // Blocksize not constant
54 }
55 else
56 {
57 mux_a->h.dwScale = (mux_a->wf->nAvgBytesPerSec * encoder->params.samples_per_frame)/ mux_a->wf->nSamplesPerSec; /* for cbr */
58 mux_a->h.dwRate = mux_a->wf->nAvgBytesPerSec;
59 mux_a->h.dwSampleSize = mux_a->h.dwScale;
60 }
61 mux_a->wf->nBlockAlign = mux_a->h.dwScale;
62 mux_a->h.dwSuggestedBufferSize = (encoder->params.audio_preload*mux_a->wf->nAvgBytesPerSec)/1000;
63 mux_a->h.dwSuggestedBufferSize -= mux_a->h.dwSuggestedBufferSize % mux_a->wf->nBlockAlign;
64
65 mux_a->wf->cbSize = 0; //12;
66 mux_a->wf->wBitsPerSample = 0; /* does not apply */
67 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->wID = 1;
68 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->fdwFlags = 2;
69 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->nBlockSize = mux_a->wf->nBlockAlign;
70 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->nFramesPerBlock = 1;
71 ((MPEGLAYER3WAVEFORMAT *) (mux_a->wf))->nCodecDelay = 0;
72
73 // Fix allocation
74 mux_a->wf = realloc(mux_a->wf, sizeof(WAVEFORMATEX)+mux_a->wf->cbSize);
75
76 encoder->input_format = AF_FORMAT_S16_NE;
77 encoder->min_buffer_size = mux_a->h.dwSuggestedBufferSize;
78 encoder->max_buffer_size = mux_a->h.dwSuggestedBufferSize*2;
79
80 return 1;
81 }
82
83 static int encode_twolame(audio_encoder_t *encoder, uint8_t *dest, void *src, int len, int max_size)
84 {
85 mpae_twolame_ctx *ctx = (mpae_twolame_ctx *)encoder->priv;
86 int ret_size = 0, r2;
87
88 len /= (2*encoder->params.channels);
89 ret_size = twolame_encode_buffer_interleaved(ctx->twolame_ctx, src, len, dest, max_size);
90 r2 = mp_decode_mp3_header(dest);
91 mp_msg(MSGT_MENCODER, MSGL_V, "\nSIZE: %d, max: %d, r2: %d\n", ret_size, max_size, r2);
92 if(r2 > 0)
93 ret_size = r2;
94 return ret_size;
95 }
96
97 int close_twolame(audio_encoder_t *encoder)
98 {
99 free(encoder->priv);
100 return 1;
101 }
102
103 static int get_frame_size(audio_encoder_t *encoder)
104 {
105 int sz;
106 if(encoder->stream->buffer_len < 4)
107 return 0;
108 sz = mp_decode_mp3_header(encoder->stream->buffer);
109 if(sz <= 0)
110 return 0;
111 return sz;
112 }
113
114
115 int mpae_init_twolame(audio_encoder_t *encoder)
116 {
117 int mode;
118 mpae_twolame_ctx *ctx = NULL;
119
120 if(encoder->params.channels == 1)
121 {
122 mp_msg(MSGT_MENCODER, MSGL_INFO, "ae_twolame, 1 audio channel, forcing mono mode\n");
123 mode = TWOLAME_MONO;
124 }
125 else if(encoder->params.channels == 2)
126 {
127 if(! strcasecmp(param_mode, "dual"))
128 mode = TWOLAME_DUAL_CHANNEL;
129 else if(! strcasecmp(param_mode, "jstereo"))
130 mode = TWOLAME_JOINT_STEREO;
131 else if(! strcasecmp(param_mode, "stereo"))
132 mode = TWOLAME_STEREO;
133 else
134 {
135 mp_msg(MSGT_MENCODER, MSGL_ERR, "ae_twolame, unknown mode %s, exiting\n", param_mode);
136 }
137 }
138 else
139 mp_msg(MSGT_MENCODER, MSGL_ERR, "ae_twolame, Twolame can't encode > 2 channels, exiting\n");
140
141 ctx = (mpae_twolame_ctx *) calloc(1, sizeof(mpae_twolame_ctx));
142 if(ctx == NULL)
143 {
144 mp_msg(MSGT_MENCODER, MSGL_ERR, "ae_twolame, couldn't alloc a %d bytes context, exiting\n", sizeof(mpae_twolame_ctx));
145 return 0;
146 }
147
148 ctx->twolame_ctx = twolame_init();
149 if(ctx->twolame_ctx == NULL)
150 {
151 mp_msg(MSGT_MENCODER, MSGL_ERR, "ae_twolame, couldn't initial parameters from libtwolame, exiting\n");
152 free(ctx);
153 return 0;
154 }
155 ctx->vbr = 0;
156
157 if(twolame_set_num_channels(ctx->twolame_ctx, encoder->params.channels) != 0)
158 return 0;
159 if(twolame_set_mode(ctx->twolame_ctx, mode) != 0)
160 return 0;
161
162 if(twolame_set_in_samplerate(ctx->twolame_ctx, encoder->params.sample_rate) != 0)
163 return 0;
164
165 if(twolame_set_out_samplerate(ctx->twolame_ctx, encoder->params.sample_rate) != 0)
166 return 0;
167
168 if(encoder->params.sample_rate < 32000)
169 twolame_set_version(ctx->twolame_ctx, TWOLAME_MPEG2);
170 else
171 twolame_set_version(ctx->twolame_ctx, TWOLAME_MPEG1);
172
173 if(twolame_set_psymodel(ctx->twolame_ctx, param_psy) != 0)
174 return 0;
175
176 if(twolame_set_bitrate(ctx->twolame_ctx, param_bitrate) != 0)
177 return 0;
178
179 if(param_errprot)
180 if(twolame_set_error_protection(ctx->twolame_ctx, TRUE) != 0)
181 return 0;
182
183 if(param_vbr != 0)
184 {
185 if(twolame_set_VBR(ctx->twolame_ctx, TRUE) != 0)
186 return 0;
187 if(twolame_set_VBR_q(ctx->twolame_ctx, param_vbr) != 0)
188 return 0;
189 if(twolame_set_padding(ctx->twolame_ctx, FALSE) != 0)
190 return 0;
191 if(param_maxvbr)
192 {
193 if(twolame_set_VBR_max_bitrate_kbps(ctx->twolame_ctx, param_maxvbr) != 0)
194 return 0;
195 }
196 ctx->vbr = 1;
197 }
198
199 if(twolame_set_verbosity(ctx->twolame_ctx, param_debug) != 0)
200 return 0;
201
202 if(twolame_init_params(ctx->twolame_ctx) != 0)
203 return 0;
204
205 encoder->params.bitrate = param_bitrate * 1000;
206 encoder->params.samples_per_frame = 1152;
207 encoder->priv = ctx;
208 encoder->decode_buffer_size = 1152 * 2 * encoder->params.channels;
209
210 encoder->bind = bind_twolame;
211 encoder->get_frame_size = get_frame_size;
212 encoder->encode = encode_twolame;
213 encoder->close = close_twolame;
214
215 return 1;
216 }
217