1 /* $Id: enclame.c,v 1.9 2005/03/05 14:27:05 titer Exp $
3 This file is part of the HandBrake source code.
4 Homepage: <http://handbrake.fr/>.
5 It may be used under the terms of the GNU General Public License. */
11 int enclameInit( hb_work_object_t *, hb_job_t * );
12 int enclameWork( hb_work_object_t *, hb_buffer_t **, hb_buffer_t ** );
13 void enclameClose( hb_work_object_t * );
15 hb_work_object_t hb_enclame =
18 "MP3 encoder (libmp3lame)",
24 struct hb_work_private_s
29 lame_global_flags * lame;
32 unsigned long input_samples;
33 unsigned long output_bytes;
40 int enclameInit( hb_work_object_t * w, hb_job_t * job )
42 hb_work_private_t * pv = calloc( 1, sizeof( hb_work_private_t ) );
43 hb_audio_t * audio = w->audio;
48 hb_log( "enclame: opening libmp3lame" );
50 pv->lame = lame_init();
51 lame_set_brate( pv->lame, audio->config.out.bitrate );
52 lame_set_in_samplerate( pv->lame, audio->config.out.samplerate );
53 lame_set_out_samplerate( pv->lame, audio->config.out.samplerate );
54 lame_init_params( pv->lame );
56 pv->input_samples = 1152 * 2;
57 pv->output_bytes = LAME_MAXMP3BUFFER;
58 pv->buf = malloc( pv->input_samples * sizeof( float ) );
60 pv->list = hb_list_init();
66 /***********************************************************************
68 ***********************************************************************
70 **********************************************************************/
71 void enclameClose( hb_work_object_t * w )
73 hb_work_private_t * pv = w->private_data;
75 lame_close( pv->lame );
76 hb_list_empty( &pv->list );
79 w->private_data = NULL;
82 /***********************************************************************
84 ***********************************************************************
86 **********************************************************************/
87 static hb_buffer_t * Encode( hb_work_object_t * w )
89 hb_work_private_t * pv = w->private_data;
90 hb_audio_t * audio = w->audio;
92 int16_t samples_s16[1152 * 2];
96 if( hb_list_bytes( pv->list ) < pv->input_samples * sizeof( float ) )
101 hb_list_getbytes( pv->list, pv->buf, pv->input_samples * sizeof( float ),
104 for( i = 0; i < 1152 * 2; i++ )
106 samples_s16[i] = ((float*) pv->buf)[i];
109 buf = hb_buffer_init( pv->output_bytes );
110 buf->start = pts + 90000 * pos / 2 / sizeof( float ) / audio->config.out.samplerate;
111 buf->stop = buf->start + 90000 * 1152 / audio->config.out.samplerate;
112 buf->size = lame_encode_buffer_interleaved( pv->lame, samples_s16,
113 1152, buf->data, LAME_MAXMP3BUFFER );
114 buf->frametype = HB_FRAME_AUDIO;
118 /* Encoding was successful but we got no data. Try to encode
120 hb_buffer_close( &buf );
123 else if( buf->size < 0 )
125 hb_log( "enclame: lame_encode_buffer failed" );
126 hb_buffer_close( &buf );
133 /***********************************************************************
135 ***********************************************************************
137 **********************************************************************/
138 int enclameWork( hb_work_object_t * w, hb_buffer_t ** buf_in,
139 hb_buffer_t ** buf_out )
141 hb_work_private_t * pv = w->private_data;
142 hb_buffer_t * in = *buf_in;
145 if ( (*buf_in)->size <= 0 )
147 /* EOF on input - send it downstream & say we're done */
157 hb_fifo_push( w->fifo_in, in);
160 buf = hb_buffer_init( pv->output_bytes );
161 buf->size = lame_encode_flush( pv->lame, buf->data, LAME_MAXMP3BUFFER );
164 hb_buffer_close( &buf );
171 hb_list_add( pv->list, *buf_in );
174 *buf_out = buf = Encode( w );
178 buf->next = Encode( w );