]> git.sesse.net Git - vlc/blob - src/ac3_decoder/ac3_decoder_thread.c
* Fixed an alignment issue with the bit stream and the bit stream
[vlc] / src / ac3_decoder / ac3_decoder_thread.c
1 /*****************************************************************************
2  * ac3_decoder_thread.c: ac3 decoder thread
3  *****************************************************************************
4  * Copyright (C) 1999, 2000 VideoLAN
5  * $Id: ac3_decoder_thread.c,v 1.29 2001/04/25 10:22:32 massiot Exp $
6  *
7  * Authors: Michel Lespinasse <walken@zoy.org>
8  *
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.
13  * 
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.
18  *
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  *****************************************************************************/
23
24 /*
25  * TODO :
26  *
27  * - vérifier l'état de la fifo de sortie avant d'y stocker les samples
28  *   décodés ;
29  * - vlc_cond_signal() / vlc_cond_wait()
30  *
31  */
32
33 /*****************************************************************************
34  * Preamble
35  *****************************************************************************/
36 #include "defs.h"
37
38 #include <unistd.h>                                              /* getpid() */
39
40 #include <stdio.h>                                           /* "intf_msg.h" */
41 #include <stdlib.h>                                      /* malloc(), free() */
42
43 #include "config.h"
44 #include "common.h"
45 #include "threads.h"
46 #include "mtime.h"
47
48 #include "intf_msg.h"                        /* intf_DbgMsg(), intf_ErrMsg() */
49
50 #include "stream_control.h"
51 #include "input_ext-dec.h"
52
53 #include "audio_output.h"
54
55 #include "ac3_decoder.h"
56 #include "ac3_decoder_thread.h"
57
58 #define AC3DEC_FRAME_SIZE (2*1536) 
59
60 /*****************************************************************************
61  * Local prototypes
62  *****************************************************************************/
63 static int      InitThread              (ac3dec_thread_t * p_adec);
64 static void     RunThread               (ac3dec_thread_t * p_adec);
65 static void     ErrorThread             (ac3dec_thread_t * p_adec);
66 static void     EndThread               (ac3dec_thread_t * p_adec);
67 static void     BitstreamCallback       ( bit_stream_t *p_bit_stream,
68                                               boolean_t b_new_pes );
69
70
71 /*****************************************************************************
72  * ac3dec_CreateThread: creates an ac3 decoder thread
73  *****************************************************************************/
74 vlc_thread_t ac3dec_CreateThread( adec_config_t * p_config )
75 {
76     ac3dec_thread_t *   p_ac3dec_t;
77
78     intf_DbgMsg( "ac3dec debug: creating ac3 decoder thread" );
79
80     /* Allocate the memory needed to store the thread's structure */
81     if((p_ac3dec_t = (ac3dec_thread_t *)malloc(sizeof(ac3dec_thread_t)))==NULL)
82     {
83         intf_ErrMsg ( "ac3dec error: not enough memory "
84                       "for ac3dec_CreateThread() to create the new thread");
85         return 0;
86     }
87     
88     /*
89      * Initialize the thread properties
90      */
91     p_ac3dec_t->p_config = p_config;
92     p_ac3dec_t->p_fifo = p_config->decoder_config.p_decoder_fifo;
93
94     /* Initialize the ac3 decoder structures */
95     ac3_init (&p_ac3dec_t->ac3_decoder);
96
97     /*
98      * Initialize the output properties
99      */
100     p_ac3dec_t->p_aout = p_config->p_aout;
101     p_ac3dec_t->p_aout_fifo = NULL;
102
103     /* Spawn the ac3 decoder thread */
104     if (vlc_thread_create(&p_ac3dec_t->thread_id, "ac3 decoder", 
105                 (vlc_thread_func_t)RunThread, (void *)p_ac3dec_t))
106     {
107         intf_ErrMsg( "ac3dec error: can't spawn ac3 decoder thread" );
108         free (p_ac3dec_t);
109         return 0;
110     }
111
112     intf_DbgMsg ("ac3dec debug: ac3 decoder thread (%p) created", p_ac3dec_t);
113     return p_ac3dec_t->thread_id;
114 }
115
116 /* Following functions are local */
117
118 /*****************************************************************************
119  * InitThread : initialize an ac3 decoder thread
120  *****************************************************************************/
121 static int InitThread (ac3dec_thread_t * p_ac3dec_t)
122 {
123     aout_fifo_t         aout_fifo;
124
125     intf_DbgMsg("ac3dec debug: initializing ac3 decoder thread %p",p_ac3dec_t);
126
127     p_ac3dec_t->p_config->decoder_config.pf_init_bit_stream(
128             &p_ac3dec_t->ac3_decoder.bit_stream,
129             p_ac3dec_t->p_config->decoder_config.p_decoder_fifo,
130             BitstreamCallback, (void *) p_ac3dec_t );
131
132     
133     aout_fifo.i_type = AOUT_ADEC_STEREO_FIFO;
134     aout_fifo.i_channels = 2;
135     aout_fifo.b_stereo = 1;
136
137     aout_fifo.l_frame_size = AC3DEC_FRAME_SIZE;
138
139     /* Creating the audio output fifo */
140     if ((p_ac3dec_t->p_aout_fifo = aout_CreateFifo(p_ac3dec_t->p_aout, &aout_fifo)) == NULL)
141     {
142         return -1;
143     }
144
145     intf_DbgMsg("ac3dec debug: ac3 decoder thread %p initialized", p_ac3dec_t);
146     return 0;
147 }
148
149 /*****************************************************************************
150  * RunThread : ac3 decoder thread
151  *****************************************************************************/
152 static void RunThread (ac3dec_thread_t * p_ac3dec_t)
153 {
154     int sync;
155
156     intf_DbgMsg ("ac3dec debug: running ac3 decoder thread (%p) (pid == %i)", p_ac3dec_t, getpid());
157
158     /* Initializing the ac3 decoder thread */
159     if (InitThread (p_ac3dec_t)) /* XXX?? */
160     {
161         p_ac3dec_t->p_fifo->b_error = 1;
162     }
163
164     sync = 0;
165     p_ac3dec_t->sync_ptr = 0;
166
167     /* ac3 decoder thread's main loop */
168     /* FIXME : do we have enough room to store the decoded frames ?? */
169     while ((!p_ac3dec_t->p_fifo->b_die) && (!p_ac3dec_t->p_fifo->b_error))
170     {
171         s16 * buffer;
172         ac3_sync_info_t sync_info;
173                         
174         if (!sync) {
175             do {
176                 GetBits(&p_ac3dec_t->ac3_decoder.bit_stream,8);
177             } while ((!p_ac3dec_t->sync_ptr) && (!p_ac3dec_t->p_fifo->b_die)
178                     && (!p_ac3dec_t->p_fifo->b_error));
179
180             /* we are in sync now */
181             sync = 1;
182         }
183
184         if (DECODER_FIFO_START(*p_ac3dec_t->p_fifo)->i_pts)
185         {
186             p_ac3dec_t->p_aout_fifo->date[p_ac3dec_t->p_aout_fifo->l_end_frame] =
187                 DECODER_FIFO_START(*p_ac3dec_t->p_fifo)->i_pts;
188             DECODER_FIFO_START(*p_ac3dec_t->p_fifo)->i_pts = 0;
189         } else {
190             p_ac3dec_t->p_aout_fifo->date[p_ac3dec_t->p_aout_fifo->l_end_frame] =
191                 LAST_MDATE;
192         }
193     
194         if (ac3_sync_frame (&p_ac3dec_t->ac3_decoder, &sync_info))
195         {
196             sync = 0;
197             goto bad_frame;
198         }
199
200         p_ac3dec_t->p_aout_fifo->l_rate = sync_info.sample_rate;
201
202         buffer = ((s16 *)p_ac3dec_t->p_aout_fifo->buffer) + 
203             (p_ac3dec_t->p_aout_fifo->l_end_frame * AC3DEC_FRAME_SIZE);
204
205         if (ac3_decode_frame (&p_ac3dec_t->ac3_decoder, buffer))
206         {
207             sync = 0;
208             goto bad_frame;
209         }
210         
211         vlc_mutex_lock (&p_ac3dec_t->p_aout_fifo->data_lock);
212         p_ac3dec_t->p_aout_fifo->l_end_frame = 
213             (p_ac3dec_t->p_aout_fifo->l_end_frame + 1) & AOUT_FIFO_SIZE;
214         vlc_cond_signal (&p_ac3dec_t->p_aout_fifo->data_wait);
215         vlc_mutex_unlock (&p_ac3dec_t->p_aout_fifo->data_lock);
216
217         bad_frame:
218             RealignBits(&p_ac3dec_t->ac3_decoder.bit_stream);
219     }
220
221     /* If b_error is set, the ac3 decoder thread enters the error loop */
222     if (p_ac3dec_t->p_fifo->b_error)
223     {
224         ErrorThread (p_ac3dec_t);
225     }
226
227     /* End of the ac3 decoder thread */
228     EndThread (p_ac3dec_t);
229 }
230
231 /*****************************************************************************
232  * ErrorThread : ac3 decoder's RunThread() error loop
233  *****************************************************************************/
234 static void ErrorThread (ac3dec_thread_t * p_ac3dec_t)
235 {
236     /* We take the lock, because we are going to read/write the start/end
237      * indexes of the decoder fifo */
238     vlc_mutex_lock (&p_ac3dec_t->p_fifo->data_lock);
239
240     /* Wait until a `die' order is sent */
241     while (!p_ac3dec_t->p_fifo->b_die)
242     {
243         /* Trash all received PES packets */
244         while (!DECODER_FIFO_ISEMPTY(*p_ac3dec_t->p_fifo))
245         {
246             p_ac3dec_t->p_fifo->pf_delete_pes(p_ac3dec_t->p_fifo->p_packets_mgt,
247                     DECODER_FIFO_START(*p_ac3dec_t->p_fifo));
248             DECODER_FIFO_INCSTART (*p_ac3dec_t->p_fifo);
249         }
250
251         /* Waiting for the input thread to put new PES packets in the fifo */
252         vlc_cond_wait (&p_ac3dec_t->p_fifo->data_wait,
253                        &p_ac3dec_t->p_fifo->data_lock);
254     }
255
256     /* We can release the lock before leaving */
257     vlc_mutex_unlock (&p_ac3dec_t->p_fifo->data_lock);
258 }
259
260 /*****************************************************************************
261  * EndThread : ac3 decoder thread destruction
262  *****************************************************************************/
263 static void EndThread (ac3dec_thread_t * p_ac3dec_t)
264 {
265     intf_DbgMsg ("ac3dec debug: destroying ac3 decoder thread %p", p_ac3dec_t);
266
267     /* If the audio output fifo was created, we destroy it */
268     if (p_ac3dec_t->p_aout_fifo != NULL)
269     {
270         aout_DestroyFifo (p_ac3dec_t->p_aout_fifo);
271
272         /* Make sure the output thread leaves the NextFrame() function */
273         vlc_mutex_lock (&(p_ac3dec_t->p_aout_fifo->data_lock));
274         vlc_cond_signal (&(p_ac3dec_t->p_aout_fifo->data_wait));
275         vlc_mutex_unlock (&(p_ac3dec_t->p_aout_fifo->data_lock));
276         
277     }
278
279     /* Destroy descriptor */
280     free( p_ac3dec_t->p_config );
281     free( p_ac3dec_t );
282
283     intf_DbgMsg ("ac3dec debug: ac3 decoder thread %p destroyed", p_ac3dec_t);
284 }
285
286 /*****************************************************************************
287 * BitstreamCallback: Import parameters from the new data/PES packet
288 *****************************************************************************
289 * This function is called by input's NextDataPacket.
290 *****************************************************************************/
291 static void BitstreamCallback ( bit_stream_t * p_bit_stream,
292                                         boolean_t b_new_pes)
293 {
294
295     ac3dec_thread_t *p_ac3dec_t=(ac3dec_thread_t *)p_bit_stream->p_callback_arg;
296
297     if( b_new_pes )
298     {
299         int ptr;
300         
301         ptr = *(p_bit_stream->p_byte + 1);
302         ptr <<= 8;
303         ptr |= *(p_bit_stream->p_byte + 2);
304         p_ac3dec_t->sync_ptr = ptr;
305         p_bit_stream->p_byte += 3;                                                            
306     }
307 }