1 /*****************************************************************************
2 * ac3_decoder_thread.c: ac3 decoder thread
3 *****************************************************************************
4 * Copyright (C) 1999, 2000 VideoLAN
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * General Public License for more details.
18 * You should have received a copy of the GNU General Public
19 * License along with this program; if not, write to the
20 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
21 * Boston, MA 02111-1307, USA.
22 *****************************************************************************/
27 * - vérifier l'état de la fifo de sortie avant d'y stocker les samples
29 * - vlc_cond_signal() / vlc_cond_wait()
33 /*****************************************************************************
35 *****************************************************************************/
38 #include <unistd.h> /* getpid() */
40 #include <stdio.h> /* "intf_msg.h" */
41 #include <stdlib.h> /* malloc(), free() */
42 #include <sys/types.h> /* on BSD, uio.h needs types.h */
43 #include <sys/uio.h> /* "input.h" */
50 #include "debug.h" /* "input_netlist.h" */
52 #include "intf_msg.h" /* intf_DbgMsg(), intf_ErrMsg() */
54 #include "input.h" /* pes_packet_t */
55 #include "input_netlist.h" /* input_NetlistFreePES() */
56 #include "decoder_fifo.h" /* DECODER_FIFO_(ISEMPTY|START|INCSTART)() */
58 #include "audio_output.h"
60 #include "ac3_decoder.h"
61 #include "ac3_decoder_thread.h"
63 #define AC3DEC_FRAME_SIZE (2*1536)
64 typedef s16 ac3dec_frame_t[ AC3DEC_FRAME_SIZE ];
66 /*****************************************************************************
68 *****************************************************************************/
69 static int InitThread ( ac3dec_thread_t * p_adec );
70 static void RunThread ( ac3dec_thread_t * p_adec );
71 static void ErrorThread ( ac3dec_thread_t * p_adec );
72 static void EndThread ( ac3dec_thread_t * p_adec );
74 /*****************************************************************************
75 * ac3dec_CreateThread: creates an ac3 decoder thread
76 *****************************************************************************/
77 ac3dec_thread_t * ac3dec_CreateThread( input_thread_t * p_input )
79 ac3dec_thread_t * p_ac3dec;
81 intf_DbgMsg("ac3dec debug: creating ac3 decoder thread\n");
83 /* Allocate the memory needed to store the thread's structure */
84 if ( (p_ac3dec = (ac3dec_thread_t *)malloc( sizeof(ac3dec_thread_t) )) == NULL )
86 intf_ErrMsg("ac3dec error: not enough memory for ac3dec_CreateThread() to create the new thread\n");
91 * Initialize the thread properties
94 p_ac3dec->b_error = 0;
97 * Initialize the input properties
99 /* Initialize the decoder fifo's data lock and conditional variable and set
100 * its buffer as empty */
101 vlc_mutex_init( &p_ac3dec->fifo.data_lock );
102 vlc_cond_init( &p_ac3dec->fifo.data_wait );
103 p_ac3dec->fifo.i_start = 0;
104 p_ac3dec->fifo.i_end = 0;
106 /* Initialize the ac3 decoder structures */
107 ac3_init (&p_ac3dec->ac3_decoder);
109 /* Initialize the bit stream structure */
110 p_ac3dec->p_input = p_input;
113 * Initialize the output properties
115 p_ac3dec->p_aout = p_input->p_aout;
116 p_ac3dec->p_aout_fifo = NULL;
118 /* Spawn the ac3 decoder thread */
119 if ( vlc_thread_create(&p_ac3dec->thread_id, "ac3 decoder", (vlc_thread_func_t)RunThread, (void *)p_ac3dec) )
121 intf_ErrMsg( "ac3dec error: can't spawn ac3 decoder thread\n" );
126 intf_DbgMsg( "ac3dec debug: ac3 decoder thread (%p) created\n", p_ac3dec );
130 /*****************************************************************************
131 * ac3dec_DestroyThread: destroys an ac3 decoder thread
132 *****************************************************************************/
133 void ac3dec_DestroyThread( ac3dec_thread_t * p_ac3dec )
135 intf_DbgMsg( "ac3dec debug: requesting termination of ac3 decoder thread %p\n", p_ac3dec );
137 /* Ask thread to kill itself */
140 /* Make sure the decoder thread leaves the GetByte() function */
141 vlc_mutex_lock( &(p_ac3dec->fifo.data_lock) );
142 vlc_cond_signal( &(p_ac3dec->fifo.data_wait) );
143 vlc_mutex_unlock( &(p_ac3dec->fifo.data_lock) );
145 /* Waiting for the decoder thread to exit */
146 /* Remove this as soon as the "status" flag is implemented */
147 vlc_thread_join( p_ac3dec->thread_id );
150 /* Following functions are local */
152 /*****************************************************************************
153 * InitThread : initialize an ac3 decoder thread
154 *****************************************************************************/
155 static int InitThread( ac3dec_thread_t * p_ac3dec )
157 aout_fifo_t aout_fifo;
158 ac3_byte_stream_t * byte_stream;
160 intf_DbgMsg( "ac3dec debug: initializing ac3 decoder thread %p\n", p_ac3dec );
162 /* Our first job is to initialize the bit stream structure with the
163 * beginning of the input stream */
164 vlc_mutex_lock( &p_ac3dec->fifo.data_lock );
165 while ( DECODER_FIFO_ISEMPTY(p_ac3dec->fifo) )
167 if ( p_ac3dec->b_die )
169 vlc_mutex_unlock( &p_ac3dec->fifo.data_lock );
172 vlc_cond_wait( &p_ac3dec->fifo.data_wait, &p_ac3dec->fifo.data_lock );
174 p_ac3dec->p_ts = DECODER_FIFO_START( p_ac3dec->fifo )->p_first_ts;
175 byte_stream = ac3_byte_stream (&p_ac3dec->ac3_decoder);
176 byte_stream->p_byte =
177 p_ac3dec->p_ts->buffer + p_ac3dec->p_ts->i_payload_start;
179 p_ac3dec->p_ts->buffer + p_ac3dec->p_ts->i_payload_end;
180 byte_stream->info = p_ac3dec;
181 vlc_mutex_unlock( &p_ac3dec->fifo.data_lock );
183 aout_fifo.i_type = AOUT_ADEC_STEREO_FIFO;
184 aout_fifo.i_channels = 2;
185 aout_fifo.b_stereo = 1;
187 aout_fifo.l_frame_size = AC3DEC_FRAME_SIZE;
189 /* Creating the audio output fifo */
190 if ( (p_ac3dec->p_aout_fifo = aout_CreateFifo(p_ac3dec->p_aout, &aout_fifo)) == NULL )
195 intf_DbgMsg( "ac3dec debug: ac3 decoder thread %p initialized\n", p_ac3dec );
199 /*****************************************************************************
200 * RunThread : ac3 decoder thread
201 *****************************************************************************/
202 static void RunThread( ac3dec_thread_t * p_ac3dec )
206 intf_DbgMsg( "ac3dec debug: running ac3 decoder thread (%p) (pid == %i)\n", p_ac3dec, getpid() );
208 msleep( INPUT_PTS_DELAY );
210 /* Initializing the ac3 decoder thread */
211 if ( InitThread(p_ac3dec) ) /* XXX?? */
213 p_ac3dec->b_error = 1;
217 p_ac3dec->sync_ptr = 0;
219 /* ac3 decoder thread's main loop */
220 /* FIXME : do we have enough room to store the decoded frames ?? */
221 while ( (!p_ac3dec->b_die) && (!p_ac3dec->b_error) )
224 ac3_sync_info_t sync_info;
226 if (!sync) { /* have to find a synchro point */
228 ac3_byte_stream_t * p_byte_stream;
232 p_byte_stream = ac3_byte_stream (&p_ac3dec->ac3_decoder);
234 /* first read till next ac3 magic header */
236 ac3_byte_stream_next (p_byte_stream);
237 } while ((!p_ac3dec->sync_ptr) &&
238 (!p_ac3dec->b_die) &&
239 (!p_ac3dec->b_error));
240 /* skip the specified number of bytes */
242 ptr = p_ac3dec->sync_ptr;
243 while (--ptr && (!p_ac3dec->b_die) && (!p_ac3dec->b_error)) {
244 if (p_byte_stream->p_byte >= p_byte_stream->p_end) {
245 ac3_byte_stream_next (p_byte_stream);
247 p_byte_stream->p_byte++;
250 /* we are in sync now */
253 p_ac3dec->sync_ptr = 0;
256 if ( DECODER_FIFO_START(p_ac3dec->fifo)->b_has_pts )
258 p_ac3dec->p_aout_fifo->date[p_ac3dec->p_aout_fifo->l_end_frame] = DECODER_FIFO_START(p_ac3dec->fifo)->i_pts;
259 DECODER_FIFO_START(p_ac3dec->fifo)->b_has_pts = 0;
263 p_ac3dec->p_aout_fifo->date[p_ac3dec->p_aout_fifo->l_end_frame] = LAST_MDATE;
266 if (ac3_sync_frame (&p_ac3dec->ac3_decoder, &sync_info)) {
271 p_ac3dec->p_aout_fifo->l_rate = sync_info.sample_rate;
273 buffer = ((ac3dec_frame_t *)p_ac3dec->p_aout_fifo->buffer)[ p_ac3dec->p_aout_fifo->l_end_frame ];
275 if (ac3_decode_frame (&p_ac3dec->ac3_decoder, buffer)) {
280 vlc_mutex_lock( &p_ac3dec->p_aout_fifo->data_lock );
281 p_ac3dec->p_aout_fifo->l_end_frame = (p_ac3dec->p_aout_fifo->l_end_frame + 1) & AOUT_FIFO_SIZE;
282 vlc_cond_signal( &p_ac3dec->p_aout_fifo->data_wait );
283 vlc_mutex_unlock( &p_ac3dec->p_aout_fifo->data_lock );
288 /* If b_error is set, the ac3 decoder thread enters the error loop */
289 if ( p_ac3dec->b_error )
291 ErrorThread( p_ac3dec );
294 /* End of the ac3 decoder thread */
295 EndThread( p_ac3dec );
298 /*****************************************************************************
299 * ErrorThread : ac3 decoder's RunThread() error loop
300 *****************************************************************************/
301 static void ErrorThread( ac3dec_thread_t * p_ac3dec )
303 /* We take the lock, because we are going to read/write the start/end
304 * indexes of the decoder fifo */
305 vlc_mutex_lock( &p_ac3dec->fifo.data_lock );
307 /* Wait until a `die' order is sent */
308 while( !p_ac3dec->b_die )
310 /* Trash all received PES packets */
311 while( !DECODER_FIFO_ISEMPTY(p_ac3dec->fifo) )
313 input_NetlistFreePES( p_ac3dec->p_input, DECODER_FIFO_START(p_ac3dec->fifo) );
314 DECODER_FIFO_INCSTART( p_ac3dec->fifo );
317 /* Waiting for the input thread to put new PES packets in the fifo */
318 vlc_cond_wait( &p_ac3dec->fifo.data_wait, &p_ac3dec->fifo.data_lock );
321 /* We can release the lock before leaving */
322 vlc_mutex_unlock( &p_ac3dec->fifo.data_lock );
325 /*****************************************************************************
326 * EndThread : ac3 decoder thread destruction
327 *****************************************************************************/
328 static void EndThread( ac3dec_thread_t * p_ac3dec )
330 intf_DbgMsg( "ac3dec debug: destroying ac3 decoder thread %p\n", p_ac3dec );
332 /* If the audio output fifo was created, we destroy it */
333 if ( p_ac3dec->p_aout_fifo != NULL )
335 aout_DestroyFifo( p_ac3dec->p_aout_fifo );
337 /* Make sure the output thread leaves the NextFrame() function */
338 vlc_mutex_lock( &(p_ac3dec->p_aout_fifo->data_lock) );
339 vlc_cond_signal( &(p_ac3dec->p_aout_fifo->data_wait) );
340 vlc_mutex_unlock( &(p_ac3dec->p_aout_fifo->data_lock) );
343 /* Destroy descriptor */
346 intf_DbgMsg( "ac3dec debug: ac3 decoder thread %p destroyed\n", p_ac3dec );
349 void ac3_byte_stream_next (ac3_byte_stream_t * p_byte_stream)
351 ac3dec_thread_t * p_ac3dec = p_byte_stream->info;
353 /* We are looking for the next TS packet that contains real data,
354 * and not just a PES header */
356 /* We were reading the last TS packet of this PES packet... It's
357 * time to jump to the next PES packet */
358 if (p_ac3dec->p_ts->p_next_ts == NULL) {
361 /* We are going to read/write the start and end indexes of the
362 * decoder fifo and to use the fifo's conditional variable,
363 * that's why we need to take the lock before */
364 vlc_mutex_lock (&p_ac3dec->fifo.data_lock);
366 /* Is the input thread dying ? */
367 if (p_ac3dec->p_input->b_die) {
368 vlc_mutex_unlock (&(p_ac3dec->fifo.data_lock));
372 /* We should increase the start index of the decoder fifo, but
373 * if we do this now, the input thread could overwrite the
374 * pointer to the current PES packet, and we weren't able to
375 * give it back to the netlist. That's why we free the PES
377 input_NetlistFreePES (p_ac3dec->p_input, DECODER_FIFO_START(p_ac3dec->fifo) );
379 DECODER_FIFO_INCSTART (p_ac3dec->fifo);
381 while (DECODER_FIFO_ISEMPTY(p_ac3dec->fifo)) {
382 vlc_cond_wait (&p_ac3dec->fifo.data_wait, &p_ac3dec->fifo.data_lock );
384 if (p_ac3dec->p_input->b_die) {
385 vlc_mutex_unlock (&(p_ac3dec->fifo.data_lock));
390 /* The next byte could be found in the next PES packet */
391 p_ac3dec->p_ts = DECODER_FIFO_START (p_ac3dec->fifo)->p_first_ts;
393 /* parse ac3 magic header */
394 ptr = p_ac3dec->p_ts->buffer [p_ac3dec->p_ts->i_payload_start+2];
396 ptr |= p_ac3dec->p_ts->buffer [p_ac3dec->p_ts->i_payload_start+3];
397 p_ac3dec->sync_ptr = ptr;
398 p_ac3dec->p_ts->i_payload_start += 4;
400 /* We can release the fifo's data lock */
401 vlc_mutex_unlock (&p_ac3dec->fifo.data_lock);
404 /* Perhaps the next TS packet of the current PES packet contains
405 * real data (ie its payload's size is greater than 0) */
407 p_ac3dec->p_ts = p_ac3dec->p_ts->p_next_ts;
409 } while (p_ac3dec->p_ts->i_payload_start == p_ac3dec->p_ts->i_payload_end);
410 p_byte_stream->p_byte =
411 p_ac3dec->p_ts->buffer + p_ac3dec->p_ts->i_payload_start;
412 p_byte_stream->p_end =
413 p_ac3dec->p_ts->buffer + p_ac3dec->p_ts->i_payload_end;