cc0af10e516eff2ab3b61294d9fb5ffb044af438
[vlc.git] / src / input / input_dec.c
1 /*****************************************************************************
2  * input_dec.c: Functions for the management of decoders
3  *****************************************************************************
4  * Copyright (C) 1999-2001 VideoLAN
5  * $Id: input_dec.c,v 1.34 2002/05/14 21:23:44 massiot Exp $
6  *
7  * Authors: Christophe Massiot <massiot@via.ecp.fr>
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  * Preamble
26  *****************************************************************************/
27 #include <stdlib.h>
28 #include <string.h>                                    /* memcpy(), memset() */
29 #include <sys/types.h>                                              /* off_t */
30
31 #include <videolan/vlc.h>
32
33 #include "stream_control.h"
34 #include "input_ext-dec.h"
35 #include "input_ext-intf.h"
36 #include "input_ext-plugins.h"
37
38 static decoder_config_t * CreateDecoderConfig( input_thread_t * p_input,
39                                                es_descriptor_t * p_es );
40 static void DeleteDecoderConfig( decoder_config_t * p_config );
41
42 /*****************************************************************************
43  * input_RunDecoder: spawns a new decoder thread
44  *****************************************************************************/
45 vlc_thread_t input_RunDecoder( input_thread_t * p_input,
46                                es_descriptor_t * p_es )
47 {
48     vlc_thread_t thread_id;
49     char * psz_plugin = NULL;
50
51     if( p_es->i_type == MPEG1_AUDIO_ES || p_es->i_type == MPEG2_AUDIO_ES )
52     {
53         psz_plugin = config_GetPszVariable( "mpeg-adec" );
54     }
55     if( p_es->i_type == AC3_AUDIO_ES )
56     {
57         psz_plugin = config_GetPszVariable( "ac3-adec" );
58     }
59
60     /* Get a suitable module */
61     p_es->p_module = module_Need( MODULE_CAPABILITY_DECODER, psz_plugin,
62                                   (void *)&p_es->i_type );
63     if( psz_plugin ) free( psz_plugin );
64     if( p_es->p_module == NULL )
65     {
66         intf_ErrMsg( "input error: no suitable decoder module for type 0x%x",
67                       p_es->i_type );
68         return( 0 );
69     }
70
71     /* Create the decoder configuration structure */
72     p_es->p_config = CreateDecoderConfig( p_input, p_es );
73
74     if( p_es->p_config == NULL )
75     {
76         intf_ErrMsg( "input error: could not create decoder config" );
77         module_Unneed( p_es->p_module );
78         return( 0 );
79     }
80
81     /* Spawn the decoder thread */
82     if ( vlc_thread_create( &thread_id, "decoder",
83          (vlc_thread_func_t)p_es->p_module->
84              p_functions->dec.functions.dec.pf_run,
85          (void *)p_es->p_config) ) 
86     {
87         intf_ErrMsg( "input error: can't spawn decoder thread \"%s\"",
88                      p_es->p_module->psz_name );
89         free( p_es->p_config );
90         module_Unneed( p_es->p_module );
91         return( 0 );
92     }
93
94     p_input->stream.b_changed = 1;
95
96     return thread_id;
97 }
98
99
100 /*****************************************************************************
101  * input_EndDecoder: kills a decoder thread and waits until it's finished
102  *****************************************************************************/
103 void input_EndDecoder( input_thread_t * p_input, es_descriptor_t * p_es )
104 {
105     int i_dummy;
106
107     p_es->p_decoder_fifo->b_die = 1;
108
109     /* Make sure the thread leaves the NextDataPacket() function by
110      * sending it a few null packets. */
111     for( i_dummy = 0; i_dummy < PADDING_PACKET_NUMBER; i_dummy++ )
112     {
113         input_NullPacket( p_input, p_es );
114     }
115
116     if( p_es->p_pes != NULL )
117     {
118         input_DecodePES( p_es->p_decoder_fifo, p_es->p_pes );
119     }
120
121     /* Waiting for the thread to exit */
122     /* I thought that unlocking was better since thread join can be long
123      * but it actually creates late pictures and freezes --stef */
124 //    vlc_mutex_unlock( &p_input->stream.stream_lock );
125     vlc_thread_join( p_es->thread_id );
126 //    vlc_mutex_lock( &p_input->stream.stream_lock );
127
128     /* Delete decoder configuration */
129     DeleteDecoderConfig( p_es->p_config );
130
131     /* Unneed module */
132     module_Unneed( p_es->p_module );
133
134     /* Tell the input there is no more decoder */
135     p_es->p_decoder_fifo = NULL;
136
137     p_input->stream.b_changed = 1;
138 }
139
140 /*****************************************************************************
141  * input_DecodePES
142  *****************************************************************************
143  * Put a PES in the decoder's fifo.
144  *****************************************************************************/
145 void input_DecodePES( decoder_fifo_t * p_decoder_fifo, pes_packet_t * p_pes )
146 {
147     vlc_mutex_lock( &p_decoder_fifo->data_lock );
148
149     p_pes->p_next = NULL;
150     *p_decoder_fifo->pp_last = p_pes;
151     p_decoder_fifo->pp_last = &p_pes->p_next;
152     p_decoder_fifo->i_depth++;
153
154     /* Warn the decoder that it's got work to do. */
155     vlc_cond_signal( &p_decoder_fifo->data_wait );
156     vlc_mutex_unlock( &p_decoder_fifo->data_lock );
157 }
158
159 /*****************************************************************************
160  * input_EscapeDiscontinuity: send a NULL packet to the decoders
161  *****************************************************************************/
162 void input_EscapeDiscontinuity( input_thread_t * p_input )
163 {
164     int     i_es, i;
165
166     for( i_es = 0; i_es < p_input->stream.i_selected_es_number; i_es++ )
167     {
168         es_descriptor_t * p_es = p_input->stream.pp_selected_es[i_es];
169
170         if( p_es->p_decoder_fifo != NULL )
171         {
172             for( i = 0; i < PADDING_PACKET_NUMBER; i++ )
173             {
174                 input_NullPacket( p_input, p_es );
175             }
176         }
177     }
178 }
179
180 /*****************************************************************************
181  * input_EscapeAudioDiscontinuity: send a NULL packet to the audio decoders
182  *****************************************************************************/
183 void input_EscapeAudioDiscontinuity( input_thread_t * p_input )
184 {
185     int     i_es, i;
186
187     for( i_es = 0; i_es < p_input->stream.i_selected_es_number; i_es++ )
188     {
189         es_descriptor_t * p_es = p_input->stream.pp_selected_es[i_es];
190
191         if( p_es->p_decoder_fifo != NULL && p_es->b_audio )
192         {
193             for( i = 0; i < PADDING_PACKET_NUMBER; i++ )
194             {
195                 input_NullPacket( p_input, p_es );
196             }
197         }
198     }
199 }
200
201 /*****************************************************************************
202  * CreateDecoderConfig: create a decoder_config_t
203  *****************************************************************************/
204 static decoder_config_t * CreateDecoderConfig( input_thread_t * p_input,
205                                                es_descriptor_t * p_es )
206 {
207     decoder_config_t * p_config;
208
209     p_config = (decoder_config_t *)malloc( sizeof(decoder_config_t) );
210     if( p_config == NULL )
211     {
212         intf_ErrMsg( "Unable to allocate memory in CreateDecoderConfig" );
213         return NULL;
214     }
215
216     /* Decoder FIFO */
217     if( (p_config->p_decoder_fifo =
218             (decoder_fifo_t *)malloc( sizeof(decoder_fifo_t) )) == NULL )
219     {
220         intf_ErrMsg( "Out of memory" );
221         free( p_config );
222         return NULL;
223     }
224
225     /* Select a new ES */
226     p_input->stream.i_selected_es_number++;
227     p_input->stream.pp_selected_es = realloc(
228                                        p_input->stream.pp_selected_es,
229                                        p_input->stream.i_selected_es_number
230                                         * sizeof(es_descriptor_t *) );
231     if( p_input->stream.pp_selected_es == NULL )
232     {
233         intf_ErrMsg( "Unable to realloc memory" );
234         free( p_config->p_decoder_fifo );
235         free( p_config );
236         return NULL;
237     }
238     p_input->stream.pp_selected_es[p_input->stream.i_selected_es_number - 1]
239             = p_es;
240
241     /* Initialize the p_config structure */
242     vlc_mutex_init(&p_config->p_decoder_fifo->data_lock);
243     vlc_cond_init(&p_config->p_decoder_fifo->data_wait);
244     p_es->p_decoder_fifo = p_config->p_decoder_fifo;
245
246     p_config->i_id = p_es->i_id;
247     p_config->i_type = p_es->i_type;
248     p_config->p_demux_data = p_es->p_demux_data;
249     
250     p_config->p_stream_ctrl = &p_input->stream.control;
251
252     p_config->p_decoder_fifo->p_first = NULL;
253     p_config->p_decoder_fifo->pp_last = &p_config->p_decoder_fifo->p_first;
254     p_config->p_decoder_fifo->i_depth = 0;
255     p_config->p_decoder_fifo->b_die = p_config->p_decoder_fifo->b_error = 0;
256     p_config->p_decoder_fifo->p_packets_mgt = p_input->p_method_data;
257
258     return p_config;
259 }
260
261 /*****************************************************************************
262  * DeleteDecoderConfig: create a decoder_config_t
263  *****************************************************************************/
264 static void DeleteDecoderConfig( decoder_config_t * p_config )
265 {
266     intf_StatMsg( "input stats: killing decoder for 0x%x, type 0x%x, %d PES in FIFO",
267                   p_config->i_id, p_config->i_type,
268                   p_config->p_decoder_fifo->i_depth );
269     /* Free all packets still in the decoder fifo. */
270     input_DeletePES( p_config->p_decoder_fifo->p_packets_mgt,
271                      p_config->p_decoder_fifo->p_first );
272
273     /* Destroy the lock and cond */
274     vlc_cond_destroy( &p_config->p_decoder_fifo->data_wait );
275     vlc_mutex_destroy( &p_config->p_decoder_fifo->data_lock );
276
277     free( p_config->p_decoder_fifo );
278
279     free( p_config );
280 }
281