1 /*****************************************************************************
2 * vorbis.c: vorbis decoder module making use of libvorbis.
3 *****************************************************************************
4 * Copyright (C) 1999-2001 VideoLAN
5 * $Id: vorbis.c,v 1.5 2002/11/14 22:38:47 massiot Exp $
7 * Authors: Gildas Bazin <gbazin@netcourrier.com>
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
27 #include <stdlib.h> /* malloc(), free() */
28 #include <string.h> /* memcpy(), memset() */
32 #include <vlc/decoder.h>
33 #include <input_ext-dec.h>
35 #include <vlc/input.h>
38 #include <vorbis/codec.h>
40 /*****************************************************************************
41 * dec_thread_t : vorbis decoder thread descriptor
42 *****************************************************************************/
43 typedef struct dec_thread_t
48 vlc_thread_t thread_id; /* id for thread functions */
53 vorbis_info vi; /* struct that stores all the static vorbis bitstream
55 vorbis_comment vc; /* struct that stores all the bitstream user
57 vorbis_dsp_state vd; /* central working state for the packet->PCM
59 vorbis_block vb; /* local working space for packet->PCM decode */
64 decoder_fifo_t *p_fifo; /* stores the PES stream data */
65 pes_packet_t *p_pes; /* current PES we are decoding */
70 aout_instance_t *p_aout;
71 aout_input_t *p_aout_input;
72 audio_sample_format_t output_format;
73 audio_date_t end_date;
77 static int pi_channels_maps[6] =
80 AOUT_CHAN_CENTER, AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
81 AOUT_CHAN_CENTER | AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
82 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT,
83 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
84 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
87 /*****************************************************************************
89 *****************************************************************************/
90 static int OpenDecoder ( vlc_object_t * );
91 static int RunDecoder ( decoder_fifo_t * );
92 static void CloseDecoder ( dec_thread_t * );
94 static void DecodePacket ( dec_thread_t * );
95 static int GetOggPacket ( dec_thread_t *, ogg_packet *, mtime_t * );
97 static void Interleave ( float *, const float **, int, int );
99 /*****************************************************************************
101 *****************************************************************************/
103 set_description( _("Vorbis decoder module") );
104 set_capability( "decoder", 100 );
105 set_callbacks( OpenDecoder, NULL );
108 /*****************************************************************************
109 * OpenDecoder: probe the decoder and return score
110 *****************************************************************************/
111 static int OpenDecoder( vlc_object_t *p_this )
113 decoder_fifo_t *p_fifo = (decoder_fifo_t*) p_this;
115 if( p_fifo->i_fourcc != VLC_FOURCC('v','o','r','b') )
120 p_fifo->pf_run = RunDecoder;
124 /*****************************************************************************
125 * RunDecoder: the vorbis decoder
126 *****************************************************************************/
127 static int RunDecoder( decoder_fifo_t * p_fifo )
130 ogg_packet oggpacket;
133 /* Allocate the memory needed to store the thread's structure */
134 if( (p_dec = (dec_thread_t *)malloc (sizeof(dec_thread_t)) )
137 msg_Err( p_fifo, "out of memory" );
141 /* Initialize the thread properties */
142 memset( p_dec, 0, sizeof(dec_thread_t) );
143 p_dec->p_fifo = p_fifo;
146 /* Take care of the initial Vorbis header */
147 vorbis_info_init( &p_dec->vi );
148 vorbis_comment_init( &p_dec->vc );
150 if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
153 oggpacket.b_o_s = 1; /* yes this actually is a b_o_s packet :) */
154 if( vorbis_synthesis_headerin( &p_dec->vi, &p_dec->vc, &oggpacket ) < 0 )
156 msg_Err( p_dec->p_fifo, "This bitstream does not contain Vorbis "
161 /* The next two packets in order are the comment and codebook headers.
162 We need to watch out that these packets are not missing as a
163 missing or corrupted header is fatal. */
164 if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
167 if( vorbis_synthesis_headerin( &p_dec->vi, &p_dec->vc, &oggpacket ) < 0 )
169 msg_Err( p_dec->p_fifo, "2nd Vorbis header is corrupted" );
173 if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
176 if( vorbis_synthesis_headerin( &p_dec->vi, &p_dec->vc, &oggpacket ) < 0 )
178 msg_Err( p_dec->p_fifo, "3rd Vorbis header is corrupted" );
182 /* Initialize the Vorbis packet->PCM decoder */
183 vorbis_synthesis_init( &p_dec->vd, &p_dec->vi );
184 vorbis_block_init( &p_dec->vd, &p_dec->vb );
186 p_dec->output_format.i_format = VLC_FOURCC('f','l','3','2');
187 p_dec->output_format.i_channels = pi_channels_maps[p_dec->vi.channels];
188 p_dec->output_format.i_rate = p_dec->vi.rate;
190 aout_DateInit( &p_dec->end_date, p_dec->vi.rate );
191 p_dec->p_aout = NULL;
192 p_dec->p_aout_input = aout_DecNew( p_dec->p_fifo,
194 &p_dec->output_format );
196 if( p_dec->p_aout_input == NULL )
198 msg_Err( p_dec->p_fifo, "failed to create aout fifo" );
202 /* vorbis decoder thread's main loop */
203 while( (!p_dec->p_fifo->b_die) && (!p_dec->p_fifo->b_error) )
205 DecodePacket( p_dec );
208 /* If b_error is set, the vorbis decoder thread enters the error loop */
209 if( p_dec->p_fifo->b_error )
211 DecoderError( p_dec->p_fifo );
214 /* End of the vorbis decoder thread */
215 CloseDecoder( p_dec );
220 DecoderError( p_fifo );
224 p_dec->p_fifo->b_error = 1;
226 /* End of the vorbis decoder thread */
227 CloseDecoder( p_dec );
234 /*****************************************************************************
235 * DecodePacket: decodes a Vorbis packet.
236 *****************************************************************************/
237 static void DecodePacket( dec_thread_t *p_dec )
239 aout_buffer_t *p_aout_buffer;
240 ogg_packet oggpacket;
245 if( GetOggPacket( p_dec, &oggpacket, &i_pts ) != VLC_SUCCESS )
247 /* This should mean an eos */
251 /* Date management */
252 if( i_pts > 0 && i_pts != aout_DateGet( &p_dec->end_date ) )
254 aout_DateSet( &p_dec->end_date, i_pts );
257 if( vorbis_synthesis( &p_dec->vb, &oggpacket ) == 0 )
258 vorbis_synthesis_blockin( &p_dec->vd, &p_dec->vb );
260 /* **pp_pcm is a multichannel float vector. In stereo, for
261 * example, pp_pcm[0] is left, and pp_pcm[1] is right. i_samples is
262 * the size of each channel. Convert the float values
263 * (-1.<=range<=1.) to whatever PCM format and write it out */
265 while( ( i_samples = vorbis_synthesis_pcmout( &p_dec->vd, &pp_pcm ) ) > 0 )
268 p_aout_buffer = aout_DecNewBuffer( p_dec->p_aout, p_dec->p_aout_input,
272 msg_Err( p_dec->p_fifo, "cannot get aout buffer" );
273 p_dec->p_fifo->b_error = 1;
277 /* Interleave the samples */
278 Interleave( (float *)p_aout_buffer->p_buffer, (const float **)pp_pcm,
279 p_dec->vi.channels, i_samples );
281 /* Tell libvorbis how many samples we actually consumed */
282 vorbis_synthesis_read( &p_dec->vd, i_samples );
284 /* Date management */
285 p_aout_buffer->start_date = aout_DateGet( &p_dec->end_date );
286 p_aout_buffer->end_date = aout_DateIncrement( &p_dec->end_date,
289 aout_DecPlay( p_dec->p_aout, p_dec->p_aout_input, p_aout_buffer );
294 /*****************************************************************************
295 * GetOggPacket: get the following vorbis packet from the stream and send back
296 * the result in an ogg packet (for easy decoding by libvorbis).
297 *****************************************************************************
298 * Returns VLC_EGENERIC in case of eof.
299 *****************************************************************************/
300 static int GetOggPacket( dec_thread_t *p_dec, ogg_packet *p_oggpacket,
303 if( p_dec->p_pes ) input_DeletePES( p_dec->p_fifo->p_packets_mgt,
306 input_ExtractPES( p_dec->p_fifo, &p_dec->p_pes );
307 if( !p_dec->p_pes ) return VLC_EGENERIC;
309 p_oggpacket->packet = p_dec->p_pes->p_first->p_payload_start;
310 p_oggpacket->bytes = p_dec->p_pes->i_pes_size;
311 p_oggpacket->granulepos = p_dec->p_pes->i_dts;
312 p_oggpacket->b_o_s = 0;
313 p_oggpacket->e_o_s = 0;
314 p_oggpacket->packetno = 0;
316 *p_pts = p_dec->p_pes->i_pts;
321 /*****************************************************************************
322 * Interleave: helper function to interleave channels
323 *****************************************************************************/
324 static void Interleave( float *p_out, const float **pp_in, int i_nb_channels,
329 for ( j = 0; j < i_samples; j++ )
331 for ( i = 0; i < i_nb_channels; i++ )
333 p_out[j * i_nb_channels + i] = pp_in[i][j];
338 /*****************************************************************************
339 * CloseDecoder: vorbis decoder destruction
340 *****************************************************************************/
341 static void CloseDecoder( dec_thread_t * p_dec )
343 if( p_dec->p_aout_input != NULL )
345 aout_DecDelete( p_dec->p_aout, p_dec->p_aout_input );
351 input_DeletePES( p_dec->p_fifo->p_packets_mgt, p_dec->p_pes );
352 vorbis_block_clear( &p_dec->vb );
353 vorbis_dsp_clear( &p_dec->vd );
354 vorbis_comment_clear( &p_dec->vc );
355 vorbis_info_clear( &p_dec->vi ); /* must be called last */