Modifies CLI help message for audio track selection, to make it clearer how to specif...
[HandBrake.git] / libhb / encvorbis.c
bloba90b285d2537b74abb679de5cd77dc88ba130de4
1 /* $Id: encvorbis.c,v 1.6 2005/03/05 15:08:32 titer Exp $
3 This file is part of the HandBrake source code.
4 Homepage: <http://handbrake.m0k.org/>.
5 It may be used under the terms of the GNU General Public License. */
7 #include "hb.h"
9 #include "vorbis/vorbisenc.h"
11 #define OGGVORBIS_FRAME_SIZE 1024
13 int encvorbisInit( hb_work_object_t *, hb_job_t * );
14 int encvorbisWork( hb_work_object_t *, hb_buffer_t **, hb_buffer_t ** );
15 void encvorbisClose( hb_work_object_t * );
17 hb_work_object_t hb_encvorbis =
19 WORK_ENCVORBIS,
20 "Vorbis encoder (libvorbis)",
21 encvorbisInit,
22 encvorbisWork,
23 encvorbisClose
26 struct hb_work_private_s
28 hb_job_t * job;
30 vorbis_info vi;
31 vorbis_comment vc;
32 vorbis_dsp_state vd;
33 vorbis_block vb;
35 unsigned long input_samples;
36 uint8_t * buf;
37 uint64_t pts;
39 hb_list_t * list;
40 int out_discrete_channels;
41 int channel_map[6];
42 int64_t prev_blocksize;
45 int encvorbisInit( hb_work_object_t * w, hb_job_t * job )
47 int i;
48 ogg_packet header[3];
49 struct ovectl_ratemanage2_arg ctl_rate_arg;
51 hb_work_private_t * pv = calloc( 1, sizeof( hb_work_private_t ) );
52 w->private_data = pv;
53 pv->out_discrete_channels = HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT(w->amixdown);
55 pv->job = job;
57 hb_log( "encvorbis: opening libvorbis" );
59 /* 28kbps/channel seems to be the minimum for 6ch vorbis. */
60 int min_bitrate = 28 * pv->out_discrete_channels;
61 if (pv->out_discrete_channels > 2 && job->abitrate < min_bitrate)
63 hb_log( "encvorbis: Selected bitrate (%d kbps) too low for %d channel audio.", job->abitrate, pv->out_discrete_channels);
64 hb_log( "encvorbis: Resetting bitrate to %d kbps", min_bitrate);
65 job->abitrate = min_bitrate;
68 /* init */
69 vorbis_info_init( &pv->vi );
70 if( vorbis_encode_setup_managed( &pv->vi, pv->out_discrete_channels,
71 job->arate, -1, 1000 * job->abitrate, -1 ) )
73 hb_error( "encvorbis: vorbis_encode_setup_managed failed.\n" );
74 *job->die = 1;
75 return 0;
78 if( vorbis_encode_ctl( &pv->vi, OV_ECTL_RATEMANAGE2_GET, &ctl_rate_arg) )
80 hb_log( "encvorbis: vorbis_encode_ctl( ratemanage2_get ) failed" );
83 ctl_rate_arg.bitrate_average_kbps = 1000 * job->abitrate;
84 ctl_rate_arg.management_active = 1;
86 if( vorbis_encode_ctl( &pv->vi, OV_ECTL_RATEMANAGE2_SET, &ctl_rate_arg ) ||
87 vorbis_encode_setup_init( &pv->vi ) )
89 hb_error( "encvorbis: vorbis_encode_ctl( ratemanage2_set ) OR vorbis_encode_setup_init failed.\n" );
90 *job->die = 1;
91 return 0;
94 /* add a comment */
95 vorbis_comment_init( &pv->vc );
96 vorbis_comment_add_tag( &pv->vc, "Encoder", "HandBrake");
97 vorbis_comment_add_tag( &pv->vc, "LANGUAGE", w->config->vorbis.language);
99 /* set up the analysis state and auxiliary encoding storage */
100 vorbis_analysis_init( &pv->vd, &pv->vi);
101 vorbis_block_init( &pv->vd, &pv->vb);
103 /* get the 3 headers */
104 vorbis_analysis_headerout( &pv->vd, &pv->vc,
105 &header[0], &header[1], &header[2] );
106 for( i = 0; i < 3; i++ )
108 memcpy( w->config->vorbis.headers[i], &header[i],
109 sizeof( ogg_packet ) );
110 memcpy( w->config->vorbis.headers[i] + sizeof( ogg_packet ),
111 header[i].packet, header[i].bytes );
114 pv->input_samples = pv->out_discrete_channels * OGGVORBIS_FRAME_SIZE;
115 pv->buf = malloc( pv->input_samples * sizeof( float ) );
117 pv->list = hb_list_init();
119 switch (pv->out_discrete_channels) {
120 case 1:
121 pv->channel_map[0] = 0;
122 break;
123 case 6:
124 pv->channel_map[0] = 0;
125 pv->channel_map[1] = 2;
126 pv->channel_map[2] = 1;
127 pv->channel_map[3] = 4;
128 pv->channel_map[4] = 5;
129 pv->channel_map[5] = 3;
130 break;
131 default:
132 hb_log("encvorbis.c: Unable to correctly proccess %d channels, assuming stereo.", pv->out_discrete_channels);
133 case 2:
134 // Assume stereo
135 pv->channel_map[0] = 0;
136 pv->channel_map[1] = 1;
137 break;
140 return 0;
143 /***********************************************************************
144 * Close
145 ***********************************************************************
147 **********************************************************************/
148 void encvorbisClose( hb_work_object_t * w )
150 hb_work_private_t * pv = w->private_data;
152 vorbis_block_clear( &pv->vb );
153 vorbis_dsp_clear( &pv->vd );
154 vorbis_comment_clear( &pv->vc );
155 vorbis_info_clear( &pv->vi );
157 if (pv->list)
158 hb_list_empty( &pv->list );
160 free( pv->buf );
161 free( pv );
162 w->private_data = NULL;
165 /***********************************************************************
166 * Flush
167 ***********************************************************************
169 **********************************************************************/
170 static hb_buffer_t * Flush( hb_work_object_t * w )
172 hb_work_private_t * pv = w->private_data;
173 hb_buffer_t * buf;
174 int64_t blocksize = 0;
176 if( vorbis_analysis_blockout( &pv->vd, &pv->vb ) == 1 )
178 ogg_packet op;
180 vorbis_analysis( &pv->vb, NULL );
181 vorbis_bitrate_addblock( &pv->vb );
183 if( vorbis_bitrate_flushpacket( &pv->vd, &op ) )
185 buf = hb_buffer_init( sizeof( ogg_packet ) + op.bytes );
186 memcpy( buf->data, &op, sizeof( ogg_packet ) );
187 memcpy( buf->data + sizeof( ogg_packet ), op.packet,
188 op.bytes );
189 blocksize = vorbis_packet_blocksize(&pv->vi, &op);
190 buf->frametype = HB_FRAME_AUDIO;
191 buf->start = (int64_t)(vorbis_granule_time(&pv->vd, op.granulepos) * 90000);
192 buf->stop = (int64_t)(vorbis_granule_time(&pv->vd, (pv->prev_blocksize + blocksize)/4 + op.granulepos) * 90000);
193 /* The stop time isn't accurate for the first ~3 packets, as the actual blocksize depends on the previous _and_ current packets. */
194 pv->prev_blocksize = blocksize;
195 return buf;
199 return NULL;
202 /***********************************************************************
203 * Encode
204 ***********************************************************************
206 **********************************************************************/
207 static hb_buffer_t * Encode( hb_work_object_t * w )
209 hb_work_private_t * pv = w->private_data;
210 hb_buffer_t * buf;
211 float ** buffer;
212 int i, j;
214 /* Try to extract more data */
215 if( ( buf = Flush( w ) ) )
217 return buf;
220 if( hb_list_bytes( pv->list ) < pv->input_samples * sizeof( float ) )
222 return NULL;
225 /* Process more samples */
226 hb_list_getbytes( pv->list, pv->buf, pv->input_samples * sizeof( float ),
227 &pv->pts, NULL );
228 buffer = vorbis_analysis_buffer( &pv->vd, OGGVORBIS_FRAME_SIZE );
229 for( i = 0; i < OGGVORBIS_FRAME_SIZE; i++ )
231 for( j = 0; j < pv->out_discrete_channels; j++)
233 buffer[j][i] = ((float *) pv->buf)[(pv->out_discrete_channels * i + pv->channel_map[j])] / 32768.f;
236 vorbis_analysis_wrote( &pv->vd, OGGVORBIS_FRAME_SIZE );
238 /* Try to extract again */
239 return Flush( w );
242 /***********************************************************************
243 * Work
244 ***********************************************************************
246 **********************************************************************/
247 int encvorbisWork( hb_work_object_t * w, hb_buffer_t ** buf_in,
248 hb_buffer_t ** buf_out )
250 hb_work_private_t * pv = w->private_data;
251 hb_buffer_t * buf;
253 hb_list_add( pv->list, *buf_in );
254 *buf_in = NULL;
256 *buf_out = buf = Encode( w );
258 while( buf )
260 buf->next = Encode( w );
261 buf = buf->next;
264 return HB_WORK_OK;