/***************************************************************************** * vc1.c : VC1 Video demuxer ***************************************************************************** * Copyright (C) 2002-2004 VLC authors and VideoLAN * $Id$ * * Authors: Laurent Aimar * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU Lesser General Public License as published by * the Free Software Foundation; either version 2.1 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this program; if not, write to the Free Software Foundation, * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA. *****************************************************************************/ /***************************************************************************** * Preamble *****************************************************************************/ #ifdef HAVE_CONFIG_H # include "config.h" #endif #include #include #include #include /***************************************************************************** * Module descriptor *****************************************************************************/ static int Open ( vlc_object_t * ); static void Close( vlc_object_t * ); #define FPS_TEXT N_("Frames per Second") #define FPS_LONGTEXT N_("Desired frame rate for the VC-1 stream.") vlc_module_begin () set_shortname( "VC-1") set_category( CAT_INPUT ) set_subcategory( SUBCAT_INPUT_DEMUX ) set_description( N_("VC1 video demuxer" ) ) set_capability( "demux", 0 ) add_float( "vc1-fps", 25.0, FPS_TEXT, FPS_LONGTEXT, true ) set_callbacks( Open, Close ) add_shortcut( "vc1" ) vlc_module_end () /***************************************************************************** * Local prototypes *****************************************************************************/ struct demux_sys_t { vlc_tick_t i_dts; es_out_id_t *p_es; float f_fps; decoder_t *p_packetizer; }; static int Demux( demux_t * ); static int Control( demux_t *, int, va_list ); #define VC1_PACKET_SIZE 4096 /***************************************************************************** * Open: initializes demux structures *****************************************************************************/ static int Open( vlc_object_t * p_this ) { demux_t *p_demux = (demux_t*)p_this; demux_sys_t *p_sys; const uint8_t *p_peek; es_format_t fmt; if( vlc_stream_Peek( p_demux->s, &p_peek, 4 ) < 4 ) return VLC_EGENERIC; if( p_peek[0] != 0x00 || p_peek[1] != 0x00 || p_peek[2] != 0x01 || p_peek[3] != 0x0f ) /* Sequence header */ { if( !p_demux->obj.force ) { msg_Warn( p_demux, "vc-1 module discarded (no startcode)" ); return VLC_EGENERIC; } msg_Err( p_demux, "this doesn't look like a VC-1 ES stream, " "continuing anyway" ); } p_sys = malloc( sizeof( demux_sys_t ) ); if( unlikely(p_sys == NULL) ) return VLC_ENOMEM; p_demux->pf_demux = Demux; p_demux->pf_control= Control; p_demux->p_sys = p_sys; p_sys->p_es = NULL; p_sys->i_dts = 0; p_sys->f_fps = var_CreateGetFloat( p_demux, "vc1-fps" ); if( p_sys->f_fps < 0.001f ) p_sys->f_fps = 0.0f; /* Load the packetizer */ es_format_Init( &fmt, VIDEO_ES, VLC_CODEC_VC1 ); p_sys->p_packetizer = demux_PacketizerNew( p_demux, &fmt, "VC-1" ); if( !p_sys->p_packetizer ) { free( p_sys ); return VLC_EGENERIC; } return VLC_SUCCESS; } /***************************************************************************** * Close: frees unused data *****************************************************************************/ static void Close( vlc_object_t * p_this ) { demux_t *p_demux = (demux_t*)p_this; demux_sys_t *p_sys = p_demux->p_sys; demux_PacketizerDestroy( p_sys->p_packetizer ); free( p_sys ); } /***************************************************************************** * Demux: reads and demuxes data packets ***************************************************************************** * Returns -1 in case of error, 0 in case of EOF, 1 otherwise *****************************************************************************/ static int Demux( demux_t *p_demux) { demux_sys_t *p_sys = p_demux->p_sys; block_t *p_block_in, *p_block_out; bool b_eof = false; p_block_in = vlc_stream_Block( p_demux->s, VC1_PACKET_SIZE ); if( p_block_in == NULL ) { b_eof = true; } else { /* */ p_block_in->i_dts = VLC_TICK_0; p_block_in->i_pts = VLC_TICK_0; } while( (p_block_out = p_sys->p_packetizer->pf_packetize( p_sys->p_packetizer, p_block_in ? &p_block_in : NULL )) ) { while( p_block_out ) { block_t *p_next = p_block_out->p_next; p_block_out->p_next = NULL; if( p_sys->p_es == NULL ) { p_sys->p_packetizer->fmt_out.b_packetized = true; p_sys->p_es = es_out_Add( p_demux->out, &p_sys->p_packetizer->fmt_out); } es_out_SetPCR( p_demux->out, VLC_TICK_0 + p_sys->i_dts ); p_block_out->i_dts = VLC_TICK_0 + p_sys->i_dts; p_block_out->i_pts = VLC_TICK_0 + p_sys->i_dts; es_out_Send( p_demux->out, p_sys->p_es, p_block_out ); p_block_out = p_next; if( p_sys->p_packetizer->fmt_out.video.i_frame_rate > 0 && p_sys->p_packetizer->fmt_out.video.i_frame_rate_base > 0 ) p_sys->i_dts += CLOCK_FREQ * p_sys->p_packetizer->fmt_out.video.i_frame_rate_base / p_sys->p_packetizer->fmt_out.video.i_frame_rate; else if( p_sys->f_fps > 0.001f ) p_sys->i_dts += (int64_t)((float) CLOCK_FREQ / p_sys->f_fps); else p_sys->i_dts += CLOCK_FREQ / 25; } } return b_eof ? VLC_DEMUXER_EOF : VLC_DEMUXER_SUCCESS; } /***************************************************************************** * Control: *****************************************************************************/ static int Control( demux_t *p_demux, int i_query, va_list args ) { /* demux_sys_t *p_sys = p_demux->p_sys; */ /* FIXME calculate the bitrate */ if( i_query == DEMUX_SET_TIME ) return VLC_EGENERIC; else return demux_vaControlHelper( p_demux->s, 0, -1, 0, 1, i_query, args ); }