1 /*****************************************************************************
3 *****************************************************************************
4 * Copyright (C) 1999-2008 the VideoLAN team
7 * Author: Laurent Aimar <fenrir _AT_ videolan _DOT_ org>
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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
31 #include <vlc_codec.h>
33 /****************************************************************************
34 * stream_Demux*: create a demuxer for an outpout stream (allow demuxer chain)
35 ****************************************************************************/
52 static int DStreamRead ( stream_t
*, void *p_read
, unsigned int i_read
);
53 static int DStreamPeek ( stream_t
*, const uint8_t **pp_peek
, unsigned int i_peek
);
54 static int DStreamControl( stream_t
*, int i_query
, va_list );
55 static void DStreamDelete ( stream_t
* );
56 static void* DStreamThread ( void * );
59 stream_t
*stream_DemuxNew( demux_t
*p_demux
, const char *psz_demux
, es_out_t
*out
)
61 vlc_object_t
*p_obj
= VLC_OBJECT(p_demux
);
62 /* We create a stream reader, and launch a thread */
66 s
= stream_CommonNew( p_obj
);
69 s
->p_input
= p_demux
->p_input
;
70 s
->psz_path
= strdup(""); /* N/A */
71 s
->pf_read
= DStreamRead
;
72 s
->pf_peek
= DStreamPeek
;
73 s
->pf_control
= DStreamControl
;
74 s
->pf_destroy
= DStreamDelete
;
76 s
->p_sys
= p_sys
= malloc( sizeof( *p_sys
) );
77 if( !s
->psz_path
|| !s
->p_sys
)
79 stream_CommonDelete( s
);
85 p_sys
->p_demux
= NULL
;
86 p_sys
->p_block
= NULL
;
87 p_sys
->psz_name
= strdup( psz_demux
);
90 if( ( p_sys
->p_fifo
= block_FifoNew() ) == NULL
)
92 stream_CommonDelete( s
);
93 free( p_sys
->psz_name
);
98 if( vlc_clone( &p_sys
->thread
, DStreamThread
, s
, VLC_THREAD_PRIORITY_INPUT
) )
100 stream_CommonDelete( s
);
101 free( p_sys
->psz_name
);
109 void stream_DemuxSend( stream_t
*s
, block_t
*p_block
)
111 stream_sys_t
*p_sys
= s
->p_sys
;
112 block_FifoPut( p_sys
->p_fifo
, p_block
);
115 static void DStreamDelete( stream_t
*s
)
117 stream_sys_t
*p_sys
= s
->p_sys
;
120 vlc_object_kill( s
);
121 #warning FIXME: not thread-safe:
123 vlc_object_kill( p_sys
->p_demux
);
124 p_empty
= block_New( s
, 1 ); p_empty
->i_buffer
= 0;
125 block_FifoPut( p_sys
->p_fifo
, p_empty
);
126 vlc_join( p_sys
->thread
, NULL
);
129 demux_Delete( p_sys
->p_demux
);
131 block_Release( p_sys
->p_block
);
133 block_FifoRelease( p_sys
->p_fifo
);
134 free( p_sys
->psz_name
);
136 stream_CommonDelete( s
);
140 static int DStreamRead( stream_t
*s
, void *p_read
, unsigned int i_read
)
142 stream_sys_t
*p_sys
= s
->p_sys
;
143 uint8_t *p_out
= p_read
;
146 //msg_Dbg( s, "DStreamRead: wanted %d bytes", i_read );
148 while( vlc_object_alive( s
) && !s
->b_error
&& i_read
)
150 block_t
*p_block
= p_sys
->p_block
;
155 p_block
= block_FifoGet( p_sys
->p_fifo
);
156 if( !p_block
) s
->b_error
= 1;
157 p_sys
->p_block
= p_block
;
160 if( p_block
&& i_read
)
162 i_copy
= __MIN( i_read
, p_block
->i_buffer
);
163 if( p_out
&& i_copy
) memcpy( p_out
, p_block
->p_buffer
, i_copy
);
167 p_block
->i_buffer
-= i_copy
;
168 p_block
->p_buffer
+= i_copy
;
170 if( !p_block
->i_buffer
)
172 block_Release( p_block
);
173 p_sys
->p_block
= NULL
;
178 p_sys
->i_pos
+= i_out
;
182 static int DStreamPeek( stream_t
*s
, const uint8_t **pp_peek
, unsigned int i_peek
)
184 stream_sys_t
*p_sys
= s
->p_sys
;
185 block_t
**pp_block
= &p_sys
->p_block
;
189 //msg_Dbg( s, "DStreamPeek: wanted %d bytes", i_peek );
191 while( vlc_object_alive( s
) && !s
->b_error
&& i_peek
)
197 *pp_block
= block_FifoGet( p_sys
->p_fifo
);
198 if( !*pp_block
) s
->b_error
= 1;
201 if( *pp_block
&& i_peek
)
203 i_copy
= __MIN( i_peek
, (*pp_block
)->i_buffer
);
207 if( i_peek
) pp_block
= &(*pp_block
)->p_next
;
213 p_sys
->p_block
= block_ChainGather( p_sys
->p_block
);
214 *pp_peek
= p_sys
->p_block
->p_buffer
;
220 static int DStreamControl( stream_t
*s
, int i_query
, va_list args
)
222 stream_sys_t
*p_sys
= s
->p_sys
;
228 case STREAM_GET_SIZE
:
229 p_i64
= va_arg( args
, uint64_t * );
233 case STREAM_CAN_SEEK
:
234 p_b
= (bool*) va_arg( args
, bool * );
238 case STREAM_CAN_FASTSEEK
:
239 p_b
= (bool*) va_arg( args
, bool * );
243 case STREAM_GET_POSITION
:
244 p_i64
= va_arg( args
, uint64_t * );
245 *p_i64
= p_sys
->i_pos
;
248 case STREAM_SET_POSITION
:
250 uint64_t i64
= va_arg( args
, uint64_t );
251 if( i64
< p_sys
->i_pos
)
254 uint64_t i_skip
= i64
- p_sys
->i_pos
;
257 int i_read
= DStreamRead( s
, NULL
, __MIN(i_skip
, INT_MAX
) );
265 case STREAM_CONTROL_ACCESS
:
266 case STREAM_GET_CONTENT_TYPE
:
267 case STREAM_SET_RECORD_STATE
:
271 msg_Err( s
, "invalid DStreamControl query=0x%x", i_query
);
276 static void* DStreamThread( void *obj
)
278 stream_t
*s
= (stream_t
*)obj
;
279 stream_sys_t
*p_sys
= s
->p_sys
;
281 int canc
= vlc_savecancel();
283 /* Create the demuxer */
284 if( !(p_demux
= demux_New( s
, s
->p_input
, "", p_sys
->psz_name
, "", s
, p_sys
->out
,
290 /* stream_Demux cannot apply DVB filters.
291 * Get all programs and let the E/S output sort them out. */
292 demux_Control( p_demux
, DEMUX_SET_GROUP
, -1, NULL
);
293 p_sys
->p_demux
= p_demux
;
296 while( vlc_object_alive( s
) && vlc_object_alive( p_demux
) )
298 if( demux_Demux( p_demux
) <= 0 ) break;
301 vlc_restorecancel( canc
);
302 vlc_object_kill( p_demux
);