]> git.sesse.net Git - vlc/blob - modules/demux/mpeg/mpgv.c
* modules/demux/*: removed useless probing messages.
[vlc] / modules / demux / mpeg / mpgv.c
1 /*****************************************************************************
2  * mpgv.c : MPEG-I/II Video demuxer
3  *****************************************************************************
4  * Copyright (C) 2001-2004 VideoLAN
5  * $Id$
6  *
7  * Authors: Laurent Aimar <fenrir@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>                                      /* malloc(), free() */
28
29 #include <vlc/vlc.h>
30 #include <vlc/input.h>
31 #include "vlc_codec.h"
32
33 /*****************************************************************************
34  * Module descriptor
35  *****************************************************************************/
36 static int  Open ( vlc_object_t * );
37 static void Close( vlc_object_t * );
38
39 vlc_module_begin();
40     set_category( CAT_INPUT );
41     set_subcategory( SUBCAT_INPUT_DEMUX );
42     set_description( _("MPEG-I/II video demuxer" ) );
43     set_capability( "demux2", 100 );
44     set_callbacks( Open, Close );
45     add_shortcut( "mpgv" );
46 vlc_module_end();
47
48 /*****************************************************************************
49  * Local prototypes
50  *****************************************************************************/
51 struct demux_sys_t
52 {
53     vlc_bool_t  b_start;
54
55     es_out_id_t *p_es;
56
57     decoder_t *p_packetizer;
58 };
59
60 static int Demux( demux_t * );
61 static int Control( demux_t *, int, va_list );
62
63 #define MPGV_PACKET_SIZE 4096
64
65 /*****************************************************************************
66  * Open: initializes demux structures
67  *****************************************************************************/
68 static int Open( vlc_object_t * p_this )
69 {
70     demux_t     *p_demux = (demux_t*)p_this;
71     demux_sys_t *p_sys;
72     vlc_bool_t   b_forced = VLC_FALSE;
73
74     uint8_t     *p_peek;
75
76     es_format_t  fmt;
77
78     if( stream_Peek( p_demux->s, &p_peek, 4 ) < 4 )
79     {
80         msg_Err( p_demux, "cannot peek" );
81         return VLC_EGENERIC;
82     }
83
84     if( !strncmp( p_demux->psz_demux, "mpgv", 4 ) )
85     {
86         b_forced = VLC_TRUE;
87     }
88
89     if( p_peek[0] != 0x00 || p_peek[1] != 0x00 || p_peek[2] != 0x01 )
90     {
91         if( !b_forced ) return VLC_EGENERIC;
92
93         msg_Err( p_demux, "this doesn't look like an MPEG ES stream, continuing" );
94     }
95
96     if( p_peek[3] > 0xb9 )
97     {
98         if( !b_forced ) return VLC_EGENERIC;
99         msg_Err( p_demux, "this seems to be a system stream (PS plug-in ?), but continuing" );
100     }
101
102     p_demux->pf_demux  = Demux;
103     p_demux->pf_control= Control;
104     p_demux->p_sys     = p_sys = malloc( sizeof( demux_sys_t ) );
105     p_sys->b_start     = VLC_TRUE;
106     p_sys->p_es        = NULL;
107
108     /*
109      * Load the mpegvideo packetizer
110      */
111     p_sys->p_packetizer = vlc_object_create( p_demux, VLC_OBJECT_PACKETIZER );
112     p_sys->p_packetizer->pf_decode_audio = NULL;
113     p_sys->p_packetizer->pf_decode_video = NULL;
114     p_sys->p_packetizer->pf_decode_sub = NULL;
115     p_sys->p_packetizer->pf_packetize = NULL;
116     es_format_Init( &p_sys->p_packetizer->fmt_in, VIDEO_ES,
117                     VLC_FOURCC( 'm', 'p', 'g', 'v' ) );
118     es_format_Init( &p_sys->p_packetizer->fmt_out, UNKNOWN_ES, 0 );
119     p_sys->p_packetizer->p_module =
120         module_Need( p_sys->p_packetizer, "packetizer", NULL, 0 );
121
122     if( p_sys->p_packetizer->p_module == NULL)
123     {
124         vlc_object_destroy( p_sys->p_packetizer );
125         msg_Err( p_demux, "cannot find mpgv packetizer" );
126         free( p_sys );
127         return VLC_EGENERIC;
128     }
129
130     /*
131      * create the output
132      */
133     es_format_Init( &fmt, VIDEO_ES, VLC_FOURCC( 'm', 'p', 'g', 'v' ) );
134     p_sys->p_es = es_out_Add( p_demux->out, &fmt );
135
136     return VLC_SUCCESS;
137 }
138
139 /*****************************************************************************
140  * Close: frees unused data
141  *****************************************************************************/
142 static void Close( vlc_object_t * p_this )
143 {
144     demux_t     *p_demux = (demux_t*)p_this;
145     demux_sys_t *p_sys = p_demux->p_sys;
146
147     module_Unneed( p_sys->p_packetizer, p_sys->p_packetizer->p_module );
148     vlc_object_destroy( p_sys->p_packetizer );
149
150     free( p_sys );
151 }
152
153 /*****************************************************************************
154  * Demux: reads and demuxes data packets
155  *****************************************************************************
156  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
157  *****************************************************************************/
158 static int Demux( demux_t *p_demux )
159 {
160     demux_sys_t  *p_sys = p_demux->p_sys;
161     block_t *p_block_in, *p_block_out;
162
163     if( ( p_block_in = stream_Block( p_demux->s, MPGV_PACKET_SIZE ) ) == NULL )
164     {
165         return 0;
166     }
167
168     if( p_sys->b_start )
169     {
170         p_block_in->i_pts =
171         p_block_in->i_dts = 1;
172     }
173     else
174     {
175         p_block_in->i_pts =
176         p_block_in->i_dts = 0;
177     }
178
179     while( (p_block_out = p_sys->p_packetizer->pf_packetize( p_sys->p_packetizer, &p_block_in )) )
180     {
181         p_sys->b_start = VLC_FALSE;
182
183         while( p_block_out )
184         {
185             block_t *p_next = p_block_out->p_next;
186
187             es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_block_out->i_dts );
188
189             p_block_out->p_next = NULL;
190             es_out_Send( p_demux->out, p_sys->p_es, p_block_out );
191
192             p_block_out = p_next;
193         }
194     }
195     return 1;
196 }
197
198 /*****************************************************************************
199  * Control:
200  *****************************************************************************/
201 static int Control( demux_t *p_demux, int i_query, va_list args )
202 {
203     /* demux_sys_t *p_sys  = p_demux->p_sys; */
204     /* FIXME calculate the bitrate */
205     if( i_query == DEMUX_SET_TIME )
206         return VLC_EGENERIC;
207     else
208         return demux2_vaControlHelper( p_demux->s,
209                                        0, -1,
210                                        0, 1, i_query, args );
211 }
212