* avi.c : AVI file Stream input module for vlc
*****************************************************************************
* Copyright (C) 2001 VideoLAN
- * $Id: avi.c,v 1.27 2003/01/20 13:01:53 fenrir Exp $
+ * $Id: avi.c,v 1.62 2003/09/13 17:42:15 fenrir Exp $
* Authors: Laurent Aimar <fenrir@via.ecp.fr>
*
* This program is free software; you can redistribute it and/or modify
* Preamble
*****************************************************************************/
#include <stdlib.h> /* malloc(), free() */
-#include <string.h> /* strdup() */
-#include <errno.h>
-#include <sys/types.h>
#include <vlc/vlc.h>
#include <vlc/input.h>
-
-#include "video.h"
+#include "codecs.h"
#include "libavi.h"
-
-#define __AVI_SUBTITLE__ 1
-
-#ifdef __AVI_SUBTITLE__
-# include "../util/sub.h"
-#endif
#include "avi.h"
-/*****************************************************************************
- * Local prototypes
- *****************************************************************************/
-static int AVIInit ( vlc_object_t * );
-static void __AVIEnd ( vlc_object_t * );
-static int AVISeek ( input_thread_t *, mtime_t, int );
-static int AVIDemux_Seekable ( input_thread_t * );
-static int AVIDemux_UnSeekable( input_thread_t *p_input );
-
-#define AVIEnd(a) __AVIEnd(VLC_OBJECT(a))
-
/*****************************************************************************
* Module descriptor
*****************************************************************************/
+static int Open ( vlc_object_t * );
+static void Close ( vlc_object_t * );
+
vlc_module_begin();
- add_category_hint( "demuxer", NULL );
+ add_category_hint( N_("avi-demuxer"), NULL, VLC_TRUE );
add_bool( "avi-interleaved", 0, NULL,
- "force interleaved method",
- "force interleaved method" );
+ N_("force interleaved method"),
+ N_("force interleaved method"), VLC_TRUE );
add_bool( "avi-index", 0, NULL,
- "force index creation",
- "force index creation" );
+ N_("force index creation"),
+ N_("force index creation"), VLC_TRUE );
- set_description( "avi demuxer" );
+ set_description( N_("AVI demuxer") );
set_capability( "demux", 212 );
- set_callbacks( AVIInit, __AVIEnd );
+ set_callbacks( Open, Close );
vlc_module_end();
/*****************************************************************************
- * Some useful functions to manipulate memory
+ * Local prototypes
*****************************************************************************/
+static int Control ( input_thread_t *, int, va_list );
+static int Seek ( input_thread_t *, mtime_t, int );
+static int Demux_Seekable ( input_thread_t * );
+static int Demux_UnSeekable( input_thread_t *p_input );
-static uint16_t GetWLE( uint8_t *p_buff )
-{
- return (uint16_t)p_buff[0] | ( ((uint16_t)p_buff[1]) << 8 );
-}
-
-static uint32_t GetDWLE( uint8_t *p_buff )
-{
- return (uint32_t)p_buff[0] | ( ((uint32_t)p_buff[1]) << 8 ) |
- ( ((uint32_t)p_buff[2]) << 16 ) | ( ((uint32_t)p_buff[3]) << 24 );
-}
-
-static uint32_t GetDWBE( uint8_t *p_buff )
-{
- return (uint32_t)p_buff[3] | ( ((uint32_t)p_buff[2]) << 8 ) |
- ( ((uint32_t)p_buff[1]) << 16 ) | ( ((uint32_t)p_buff[0]) << 24 );
-}
-static vlc_fourcc_t GetFOURCC( byte_t *p_buff )
-{
- return VLC_FOURCC( p_buff[0], p_buff[1], p_buff[2], p_buff[3] );
-}
+#define FREE( p ) if( p ) { free( p ); (p) = NULL; }
+#define __ABS( x ) ( (x) < 0 ? (-(x)) : (x) )
static inline off_t __EVEN( off_t i )
{
return (i & 1) ? i + 1 : i;
}
-#define __ABS( x ) ( (x) < 0 ? (-(x)) : (x) )
+static mtime_t AVI_PTSToChunk( avi_stream_t *, mtime_t i_pts );
+static mtime_t AVI_PTSToByte ( avi_stream_t *, mtime_t i_pts );
+static mtime_t AVI_GetDPTS ( avi_stream_t *, int64_t i_count );
+static mtime_t AVI_GetPTS ( avi_stream_t * );
-/* read data in a pes */
-static int input_ReadInPES( input_thread_t *p_input,
- pes_packet_t **pp_pes,
- size_t i_size )
-{
- pes_packet_t *p_pes;
- data_packet_t *p_data;
+static int AVI_StreamChunkFind( input_thread_t *, unsigned int i_stream );
+static int AVI_StreamChunkSet ( input_thread_t *,
+ unsigned int i_stream, unsigned int i_ck );
+static int AVI_StreamBytesSet ( input_thread_t *,
+ unsigned int i_stream, off_t i_byte );
- if( !(p_pes = input_NewPES( p_input->p_method_data ) ) )
- {
- pp_pes = NULL;
- return -1;
- }
+vlc_fourcc_t AVI_FourccGetCodec( unsigned int i_cat, vlc_fourcc_t );
+static int AVI_GetKeyFlag ( vlc_fourcc_t , uint8_t * );
- *pp_pes = p_pes;
+static vlc_bool_t AVI_Interleaved( input_thread_t *p_input );
- if( !i_size )
- {
- p_pes->p_first =
- p_pes->p_last =
- input_NewPacket( p_input->p_method_data, 0 );
- p_pes->i_nb_data = 1;
- p_pes->i_pes_size = 0;
- return 0;
- }
+static int AVI_PacketGetHeader( input_thread_t *, avi_packet_t *p_pk );
+static int AVI_PacketNext ( input_thread_t * );
+static int AVI_PacketRead ( input_thread_t *, avi_packet_t *, pes_packet_t **);
+static int AVI_PacketSearch ( input_thread_t * );
- p_pes->i_nb_data = 0;
- p_pes->i_pes_size = 0;
+static void AVI_IndexLoad ( input_thread_t * );
+static void AVI_IndexCreate ( input_thread_t * );
+static void AVI_IndexAddEntry( demux_sys_t *, int, AVIIndexEntry_t * );
- while( p_pes->i_pes_size < i_size )
- {
- int i_read;
+static mtime_t AVI_MovieGetLength( input_thread_t * );
- i_read = input_SplitBuffer(p_input,
- &p_data,
- __MIN( i_size -
- p_pes->i_pes_size, 2048 ) );
- if( i_read <= 0 )
- {
- return p_pes->i_pes_size;
- }
+/*****************************************************************************
+ * Stream management
+ *****************************************************************************/
+static vlc_bool_t AVI_StreamStart ( input_thread_t *, int );
+static void AVI_StreamStop ( input_thread_t *, int );
+static int AVI_StreamSeek ( input_thread_t *, int, mtime_t );
+static int AVI_StreamStopFinishedStreams( input_thread_t *);
- if( !p_pes->p_first )
- {
- p_pes->p_first = p_data;
- }
- else
- {
- p_pes->p_last->p_next = p_data;
- }
- p_pes->p_last = p_data;
- p_pes->i_nb_data++;
- p_pes->i_pes_size += i_read;
- }
+/*****************************************************************************
+ * Open: check file and initializes AVI structures
+ *****************************************************************************/
+static int Open( vlc_object_t * p_this )
+{
+ input_thread_t * p_input = (input_thread_t *)p_this;
+ avi_chunk_t ck_riff;
+ avi_chunk_list_t *p_riff = (avi_chunk_list_t*)&ck_riff;
+ avi_chunk_list_t *p_hdrl, *p_movi;
+#if 0
+ avi_chunk_list_t *p_INFO;
+ avi_chunk_strz_t *p_name;
+#endif
+ avi_chunk_avih_t *p_avih;
+ demux_sys_t *p_avi;
+ es_descriptor_t *p_es = NULL; /* avoid warning */
+ unsigned int i;
+ vlc_bool_t b_stream_audio, b_stream_video;
+ uint8_t *p_peek;
- return p_pes->i_pes_size;
-}
-/* Test if it seems that it's a key frame */
-static int AVI_GetKeyFlag( vlc_fourcc_t i_fourcc, uint8_t *p_byte )
-{
- switch( i_fourcc )
+ /* Is it an avi file ? */
+ if( input_Peek( p_input, &p_peek, 12 ) < 12 )
{
- case FOURCC_DIV1:
- /* we have:
- * startcode: 0x00000100 32bits
- * framenumber ? 5bits
- * piture type 0(I),1(P) 2bits
- */
- if( GetDWBE( p_byte ) != 0x00000100 )
- {
- /* it's not an msmpegv1 stream, strange...*/
- return AVIIF_KEYFRAME;
- }
- else
- {
- return p_byte[4] & 0x06 ? 0 : AVIIF_KEYFRAME;
- }
- case FOURCC_DIV2:
- case FOURCC_DIV3: // wmv1 also
- /* we have
- * picture type 0(I),1(P) 2bits
- */
- return p_byte[0] & 0xC0 ? 0 : AVIIF_KEYFRAME;
- case FOURCC_mp4v:
- /* we should find first occurence of 0x000001b6 (32bits)
- * startcode: 0x000001b6 32bits
- * piture type 0(I),1(P) 2bits
- */
- if( GetDWBE( p_byte ) != 0x000001b6 )
- {
- /* not true , need to find the first VOP header */
- return AVIIF_KEYFRAME;
- }
- else
- {
- return p_byte[4] & 0xC0 ? 0 : AVIIF_KEYFRAME;
- }
- default:
- /* I can't do it, so say yes */
- return AVIIF_KEYFRAME;
+ msg_Err( p_input, "cannot peek()" );
+ return VLC_EGENERIC;
}
-}
-
-vlc_fourcc_t AVI_FourccGetCodec( unsigned int i_cat, vlc_fourcc_t i_codec )
-{
- switch( i_cat )
+ if( strncmp( &p_peek[0], "RIFF", 4 ) || strncmp( &p_peek[8], "AVI ", 4 ) )
{
- case AUDIO_ES:
- switch( i_codec )
- {
- case WAVE_FORMAT_PCM:
- return VLC_FOURCC( 'a', 'r', 'a', 'w' );
- case WAVE_FORMAT_MPEG:
- case WAVE_FORMAT_MPEGLAYER3:
- return VLC_FOURCC( 'm', 'p', 'g', 'a' );
- case WAVE_FORMAT_A52:
- return VLC_FOURCC( 'a', '5', '2', ' ' );
- case WAVE_FORMAT_WMA1:
- return VLC_FOURCC( 'w', 'm', 'a', '1' );
- case WAVE_FORMAT_WMA2:
- return VLC_FOURCC( 'w', 'm', 'a', '2' );
- default:
- return VLC_FOURCC( 'm', 's',
- ( i_codec >> 8 )&0xff, i_codec&0xff );
- }
- case VIDEO_ES:
- // XXX DIV1 <- msmpeg4v1, DIV2 <- msmpeg4v2, DIV3 <- msmpeg4v3, mp4v for mpeg4
- switch( i_codec )
- {
- case FOURCC_DIV1:
- case FOURCC_div1:
- case FOURCC_MPG4:
- case FOURCC_mpg4:
- return FOURCC_DIV1;
- case FOURCC_DIV2:
- case FOURCC_div2:
- case FOURCC_MP42:
- case FOURCC_mp42:
- case FOURCC_MPG3:
- case FOURCC_mpg3:
- return FOURCC_DIV2;
- case FOURCC_div3:
- case FOURCC_MP43:
- case FOURCC_mp43:
- case FOURCC_DIV3:
- case FOURCC_DIV4:
- case FOURCC_div4:
- case FOURCC_DIV5:
- case FOURCC_div5:
- case FOURCC_DIV6:
- case FOURCC_div6:
- case FOURCC_AP41:
- case FOURCC_3IV1:
- return FOURCC_DIV3;
- case FOURCC_DIVX:
- case FOURCC_divx:
- case FOURCC_MP4S:
- case FOURCC_mp4s:
- case FOURCC_M4S2:
- case FOURCC_m4s2:
- case FOURCC_xvid:
- case FOURCC_XVID:
- case FOURCC_XviD:
- case FOURCC_DX50:
- case FOURCC_mp4v:
- case FOURCC_4:
- return FOURCC_mp4v;
- }
- default:
- return VLC_FOURCC( 'u', 'n', 'd', 'f' );
+ msg_Warn( p_input, "avi module discarded (invalid header)" );
+ return VLC_EGENERIC;
}
-}
-static void AVI_ParseStreamHeader( vlc_fourcc_t i_id,
- int *pi_number, int *pi_type )
-{
-#define SET_PTR( p, v ) if( p ) *(p) = (v);
- int c1, c2;
+ /* Initialize input structures. */
+ p_avi = p_input->p_demux_data = malloc( sizeof(demux_sys_t) );
+ memset( p_avi, 0, sizeof( demux_sys_t ) );
+ p_avi->i_time = 0;
+ p_avi->i_pcr = 0;
+ p_avi->i_movi_lastchunk_pos = 0;
+ p_avi->b_odml = VLC_FALSE;
+ p_avi->b_interleaved = VLC_FALSE;
- c1 = ((uint8_t *)&i_id)[0];
- c2 = ((uint8_t *)&i_id)[1];
+ stream_Control( p_input->s, STREAM_CAN_FASTSEEK, &p_avi->b_seekable );
- if( c1 < '0' || c1 > '9' || c2 < '0' || c2 > '9' )
+ p_input->pf_demux_control = Control;
+ p_input->pf_demux = Demux_Seekable;
+ /* For unseekable stream, automaticaly use Demux_UnSeekable */
+ if( !p_avi->b_seekable || config_GetInt( p_input, "avi-interleaved" ) )
{
- SET_PTR( pi_number, 100 ); /* > max stream number */
- SET_PTR( pi_type, UNKNOWN_ES );
+ p_input->pf_demux = Demux_UnSeekable;
}
- else
+
+ if( AVI_ChunkReadRoot( p_input->s, &p_avi->ck_root ) )
{
- SET_PTR( pi_number, (c1 - '0') * 10 + (c2 - '0' ) );
- switch( VLC_TWOCC( ((uint8_t *)&i_id)[2], ((uint8_t *)&i_id)[3] ) )
+ msg_Err( p_input, "avi module discarded (invalid file)" );
+ return VLC_EGENERIC;
+ }
+
+ if( AVI_ChunkCount( &p_avi->ck_root, AVIFOURCC_RIFF ) > 1 )
+ {
+ unsigned int i_count =
+ AVI_ChunkCount( &p_avi->ck_root, AVIFOURCC_RIFF );
+
+ msg_Warn( p_input, "multiple riff -> OpenDML ?" );
+ for( i = 1; i < i_count; i++ )
{
- case AVITWOCC_wb:
- SET_PTR( pi_type, AUDIO_ES );
- break;
- case AVITWOCC_dc:
- case AVITWOCC_db:
- SET_PTR( pi_type, VIDEO_ES );
- break;
- default:
- SET_PTR( pi_type, UNKNOWN_ES );
+ avi_chunk_list_t *p_avix;
+
+ p_avix = AVI_ChunkFind( &p_avi->ck_root, AVIFOURCC_RIFF, i );
+ if( p_avix->i_type == AVIFOURCC_AVIX )
+ {
+ msg_Warn( p_input, "detected OpenDML file" );
+ p_avi->b_odml = VLC_TRUE;
break;
+ }
}
}
-#undef SET_PTR
-}
-static int AVI_PacketGetHeader( input_thread_t *p_input, avi_packet_t *p_pk )
-{
- uint8_t *p_peek;
+ p_riff = AVI_ChunkFind( &p_avi->ck_root, AVIFOURCC_RIFF, 0 );
+ p_hdrl = AVI_ChunkFind( p_riff, AVIFOURCC_hdrl, 0 );
+ p_movi = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0 );
+#if 0
+ p_INFO = AVI_ChunkFind( p_riff, AVIFOURCC_INFO, 0 );
+ p_name = AVI_ChunkFind( p_INFO, AVIFOURCC_INAM, 0 );
+#endif
- if( input_Peek( p_input, &p_peek, 16 ) < 16 )
- {
- return VLC_EGENERIC;
- }
- p_pk->i_fourcc = GetFOURCC( p_peek );
- p_pk->i_size = GetDWLE( p_peek + 4 );
- p_pk->i_pos = AVI_TellAbsolute( p_input );
- if( p_pk->i_fourcc == AVIFOURCC_LIST )
- {
- p_pk->i_type = GetFOURCC( p_peek + 8 );
- }
- else
+ if( !p_hdrl || !p_movi )
{
- p_pk->i_type = 0;
+ msg_Err( p_input, "avi module discarded (invalid file)" );
+ goto error;
}
- memcpy( p_pk->i_peek, p_peek + 8, 8 );
-
- AVI_ParseStreamHeader( p_pk->i_fourcc, &p_pk->i_stream, &p_pk->i_cat );
- return VLC_SUCCESS;
-}
-
-static int AVI_PacketNext( input_thread_t *p_input )
-{
- avi_packet_t avi_ck;
-
- if( AVI_PacketGetHeader( p_input, &avi_ck ) )
+ if( !( p_avih = AVI_ChunkFind( p_hdrl, AVIFOURCC_avih, 0 ) ) )
{
- return VLC_EGENERIC;
+ msg_Err( p_input, "cannot find avih chunk" );
+ goto error;
}
- if( avi_ck.i_fourcc == AVIFOURCC_LIST && avi_ck.i_type == AVIFOURCC_rec )
+ p_avi->i_streams = AVI_ChunkCount( p_hdrl, AVIFOURCC_strl );
+ if( p_avih->i_streams != p_avi->i_streams )
{
- return AVI_SkipBytes( p_input, 12 );
+ msg_Warn( p_input,
+ "found %d stream but %d are declared",
+ p_avi->i_streams,
+ p_avih->i_streams );
}
- else
+ if( p_avi->i_streams == 0 )
{
- return AVI_SkipBytes( p_input, __EVEN( avi_ck.i_size ) + 8 );
+ msg_Err( p_input, "no stream defined!" );
+ goto error;
}
-}
-static int AVI_PacketRead( input_thread_t *p_input,
- avi_packet_t *p_pk,
- pes_packet_t **pp_pes )
-{
- size_t i_size;
- vlc_bool_t b_pad;
- i_size = __EVEN( p_pk->i_size + 8 );
- b_pad = ( i_size != p_pk->i_size + 8 );
-
- if( input_ReadInPES( p_input, pp_pes, i_size ) != (ssize_t)i_size )
+ /* create one program */
+ vlc_mutex_lock( &p_input->stream.stream_lock );
+ if( input_InitStream( p_input, 0 ) == -1)
{
- return VLC_EGENERIC;
+ vlc_mutex_unlock( &p_input->stream.stream_lock );
+ msg_Err( p_input, "cannot init stream" );
+ goto error;
}
- (*pp_pes)->p_first->p_payload_start += 8;
- (*pp_pes)->i_pes_size -= 8;
-
- if( b_pad )
+ if( input_AddProgram( p_input, 0, 0) == NULL )
{
- (*pp_pes)->p_last->p_payload_end--;
- (*pp_pes)->i_pes_size--;
+ vlc_mutex_unlock( &p_input->stream.stream_lock );
+ msg_Err( p_input, "cannot add program" );
+ goto error;
}
+ p_input->stream.p_selected_program = p_input->stream.pp_programs[0];
+ p_input->stream.i_mux_rate = 0; /* Fixed later */
+ vlc_mutex_unlock( &p_input->stream.stream_lock );
- return VLC_SUCCESS;
-}
+ /* print informations on streams */
+ msg_Dbg( p_input, "AVIH: %d stream, flags %s%s%s%s ",
+ p_avi->i_streams,
+ p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
+ p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
+ p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
+ p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
+ {
+ input_info_category_t *p_cat = input_InfoCategory( p_input, _("Avi") );
+ input_AddInfo( p_cat, _("Number of Streams"), "%d", p_avi->i_streams );
+ input_AddInfo( p_cat, _("Flags"), "%s%s%s%s",
+ p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
+ p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
+ p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
+ p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
+ }
-static int AVI_PacketSearch( input_thread_t *p_input )
-{
- demux_sys_t *p_avi = p_input->p_demux_data;
+ /* now read info on each stream and create ES */
+ p_avi->pp_info = calloc( p_avi->i_streams,
+ sizeof( avi_stream_t* ) );
+ memset( p_avi->pp_info,
+ 0,
+ sizeof( avi_stream_t* ) * p_avi->i_streams );
- avi_packet_t avi_pk;
- for( ;; )
+ for( i = 0 ; i < p_avi->i_streams; i++ )
{
- if( AVI_SkipBytes( p_input, 1 ) )
+ avi_chunk_list_t *p_avi_strl;
+ avi_chunk_strh_t *p_avi_strh;
+ avi_chunk_strf_auds_t *p_avi_strf_auds;
+ avi_chunk_strf_vids_t *p_avi_strf_vids;
+ int i_init_size;
+ void *p_init_data;
+#define p_info p_avi->pp_info[i]
+ p_info = malloc( sizeof(avi_stream_t ) );
+ memset( p_info, 0, sizeof( avi_stream_t ) );
+
+ p_info->p_index = NULL;
+ p_info->i_idxnb = 0;
+ p_info->i_idxmax = 0;
+
+ p_avi_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i );
+ p_avi_strh = AVI_ChunkFind( p_avi_strl, AVIFOURCC_strh, 0 );
+ p_avi_strf_auds = (void*)
+ p_avi_strf_vids = (void*)
+ AVI_ChunkFind( p_avi_strl, AVIFOURCC_strf, 0 );
+
+ if( !p_avi_strl || !p_avi_strh ||
+ ( !p_avi_strf_auds && !p_avi_strf_vids ) )
{
- return VLC_EGENERIC;
+ msg_Warn( p_input, "stream[%d] incomplete", i );
+ continue;
}
- AVI_PacketGetHeader( p_input, &avi_pk );
- if( avi_pk.i_stream < p_avi->i_streams &&
- ( avi_pk.i_cat == AUDIO_ES || avi_pk.i_cat == VIDEO_ES ) )
+
+ /* *** Init p_info *** */
+ p_info->i_rate = p_avi_strh->i_rate;
+ p_info->i_scale = p_avi_strh->i_scale;
+ p_info->i_samplesize = p_avi_strh->i_samplesize;
+ msg_Dbg( p_input, "stream[%d] rate:%d scale:%d samplesize:%d",
+ i,
+ p_info->i_rate, p_info->i_scale, p_info->i_samplesize );
+ switch( p_avi_strh->i_type )
{
- return VLC_SUCCESS;
+ case( AVIFOURCC_auds ):
+ p_info->i_cat = AUDIO_ES;
+ p_info->i_fourcc =
+ AVI_FourccGetCodec( AUDIO_ES,
+ p_avi_strf_auds->p_wf->wFormatTag );
+ p_info->i_codec = p_info->i_fourcc;
+ i_init_size = p_avi_strf_auds->i_chunk_size;
+ p_init_data = p_avi_strf_auds->p_wf;
+ msg_Dbg( p_input, "stream[%d] audio(0x%x) %d channels %dHz %dbits",
+ i,
+ p_avi_strf_auds->p_wf->wFormatTag,
+ p_avi_strf_auds->p_wf->nChannels,
+ p_avi_strf_auds->p_wf->nSamplesPerSec,
+ p_avi_strf_auds->p_wf->wBitsPerSample );
+ {
+ char psz_cat[ sizeof("Stream") + 10 ];
+ input_info_category_t *p_cat;
+ sprintf( psz_cat, _("Stream %d"), i );
+ p_cat = input_InfoCategory( p_input, psz_cat );
+ input_AddInfo( p_cat, _("Type"), "Audio(0x%x)",
+ p_avi_strf_auds->p_wf->wFormatTag );
+ input_AddInfo( p_cat, _("Codec"), "%4.4s",
+ (const char*)&(p_info->i_codec) );
+ input_AddInfo( p_cat, _("FOURCC"), "0x%x",
+ p_info->i_codec );
+ input_AddInfo( p_cat, _("Channels"), "%d",
+ p_avi_strf_auds->p_wf->nChannels );
+ input_AddInfo( p_cat, _("Sample Rate"), "%d",
+ p_avi_strf_auds->p_wf->nSamplesPerSec );
+ if( p_avi_strf_auds->p_wf->wBitsPerSample > 0
+ && p_info->i_scale != 0 )
+ {
+ input_AddInfo( p_cat, _("Bits Per Sample"), "%d",
+ p_avi_strf_auds->p_wf->wBitsPerSample );
+ input_AddInfo( p_cat, _("Audio Bitrate"), "%d",
+ p_info->i_samplesize * p_info->i_rate
+ / p_info->i_scale );
+ }
+ }
+ break;
+
+ case( AVIFOURCC_vids ):
+ p_info->i_cat = VIDEO_ES;
+ /* XXX quick hack for playing ffmpeg video, I don't know
+ who is doing something wrong */
+ p_info->i_samplesize = 0;
+ p_info->i_fourcc = p_avi_strf_vids->p_bih->biCompression;
+ p_info->i_codec =
+ AVI_FourccGetCodec( VIDEO_ES, p_info->i_fourcc );
+ i_init_size = p_avi_strf_vids->i_chunk_size;
+ p_init_data = p_avi_strf_vids->p_bih;
+ msg_Dbg( p_input, "stream[%d] video(%4.4s) %dx%d %dbpp %ffps",
+ i,
+ (char*)&p_avi_strf_vids->p_bih->biCompression,
+ p_avi_strf_vids->p_bih->biWidth,
+ p_avi_strf_vids->p_bih->biHeight,
+ p_avi_strf_vids->p_bih->biBitCount,
+ (float)p_info->i_rate /
+ (float)p_info->i_scale );
+ {
+ char psz_cat[ sizeof("Stream") + 10 ];
+ input_info_category_t *p_cat;
+ sprintf( psz_cat, "Stream %d", i );
+ p_cat = input_InfoCategory( p_input, psz_cat );
+ input_AddInfo( p_cat, _("Type"), _("Video") );
+ input_AddInfo( p_cat, _("Codec"), "%4.4s",
+ (const char*)&(p_info->i_codec) );
+ input_AddInfo( p_cat, _("FOURCC"), "0x%x",
+ p_info->i_codec );
+ input_AddInfo( p_cat, _("Resolution"), "%dx%d",
+ p_avi_strf_vids->p_bih->biWidth,
+ p_avi_strf_vids->p_bih->biHeight );
+ input_AddInfo( p_cat, _("Frame Rate"), "%f",
+ (float)p_info->i_rate /
+ (float)p_info->i_scale );
+ input_AddInfo( p_cat, _("Bits Per Pixel"), "%d",
+ p_avi_strf_vids->p_bih->biBitCount );
+ }
+ break;
+ default:
+ msg_Warn( p_input, "stream[%d] unknown type", i );
+ p_info->i_cat = UNKNOWN_ES;
+ i_init_size = 0;
+ p_init_data = NULL;
+ {
+ char psz_cat[32]; /* We'll clip i just in case */
+ input_info_category_t *p_cat;
+ sprintf( psz_cat, "Stream %d", __MIN( i, 100000 ) );
+ p_cat = input_InfoCategory( p_input, psz_cat );
+ input_AddInfo( p_cat, _("Type"), _("Unknown") );
+ }
+ break;
}
- switch( avi_pk.i_fourcc )
+ p_info->b_activated = VLC_FALSE;
+ /* add one ES */
+ vlc_mutex_lock( &p_input->stream.stream_lock );
+ p_info->p_es =
+ p_es = input_AddES( p_input, p_input->stream.p_selected_program,
+ 1+i, p_info->i_cat, NULL, 0 );
+ vlc_mutex_unlock( &p_input->stream.stream_lock );
+ p_es->i_stream_id =i; /* XXX: i don't use it */
+ p_es->i_fourcc = p_info->i_fourcc;
+ if( p_es->i_cat == AUDIO_ES )
{
- case AVIFOURCC_JUNK:
- case AVIFOURCC_LIST:
- case AVIFOURCC_idx1:
- return VLC_SUCCESS;
+ if( i_init_size < (int)sizeof( WAVEFORMATEX ) )
+ {
+ i_init_size = sizeof( WAVEFORMATEX );
+ }
+ p_es->p_waveformatex = malloc( i_init_size );
+ memcpy( p_es->p_waveformatex, p_init_data, i_init_size );
+ }
+ else if( p_es->i_cat == VIDEO_ES )
+ {
+ p_es->p_bitmapinfoheader = malloc( i_init_size );
+ memcpy( p_es->p_bitmapinfoheader, p_init_data, i_init_size );
}
+#undef p_info
}
-}
-
-static void __AVI_AddEntryIndex( avi_stream_t *p_info,
- AVIIndexEntry_t *p_index)
-{
- if( p_info->p_index == NULL )
+ if( config_GetInt( p_input, "avi-index" ) )
{
- p_info->i_idxmax = 16384;
- p_info->i_idxnb = 0;
- if( !( p_info->p_index = calloc( p_info->i_idxmax,
- sizeof( AVIIndexEntry_t ) ) ) )
+ if( p_avi->b_seekable )
{
- return;
+ AVI_IndexCreate( p_input );
}
- }
- if( p_info->i_idxnb >= p_info->i_idxmax )
- {
- p_info->i_idxmax += 16384;
- if( !( p_info->p_index = realloc( (void*)p_info->p_index,
- p_info->i_idxmax *
- sizeof( AVIIndexEntry_t ) ) ) )
+ else
{
- return;
+ msg_Warn( p_input, "cannot create index (unseekable stream)" );
+ AVI_IndexLoad( p_input );
}
}
- /* calculate cumulate length */
- if( p_info->i_idxnb > 0 )
- {
- p_index->i_lengthtotal =
- p_info->p_index[p_info->i_idxnb - 1].i_length +
- p_info->p_index[p_info->i_idxnb - 1].i_lengthtotal;
- }
else
{
- p_index->i_lengthtotal = 0;
+ AVI_IndexLoad( p_input );
}
- p_info->p_index[p_info->i_idxnb] = *p_index;
- p_info->i_idxnb++;
-
-}
-
-static void AVI_IndexAddEntry( demux_sys_t *p_avi,
- int i_stream,
- AVIIndexEntry_t *p_index)
-{
- __AVI_AddEntryIndex( p_avi->pp_info[i_stream],
- p_index );
- if( p_avi->i_movi_lastchunk_pos < p_index->i_pos )
+ /* *** movie length in sec *** */
+ p_avi->i_length = AVI_MovieGetLength( p_input );
+ if( p_avi->i_length < (mtime_t)p_avih->i_totalframes *
+ (mtime_t)p_avih->i_microsecperframe /
+ (mtime_t)1000000 )
{
- p_avi->i_movi_lastchunk_pos = p_index->i_pos;
+ msg_Warn( p_input, "broken or missing index, 'seek' will be axproximative or will have strange behavour" );
}
-}
+ /* fix some BeOS MediaKit generated file */
+ for( i = 0 ; i < p_avi->i_streams; i++ )
+ {
+ avi_chunk_list_t *p_avi_strl;
+ avi_chunk_strh_t *p_avi_strh;
+ avi_chunk_strf_auds_t *p_avi_strf_auds;
+#define p_stream p_avi->pp_info[i]
-static void AVI_IndexLoad( input_thread_t *p_input )
-{
- demux_sys_t *p_avi = p_input->p_demux_data;
+ if( p_stream->i_cat != AUDIO_ES )
+ {
+ continue;
+ }
+ if( p_stream->i_idxnb < 1 ||
+ p_stream->i_scale != 1 ||
+ p_stream->i_samplesize != 0 )
+ {
+ continue;
+ }
+ p_avi_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i );
+ p_avi_strh = AVI_ChunkFind( p_avi_strl, AVIFOURCC_strh, 0 );
+ p_avi_strf_auds = AVI_ChunkFind( p_avi_strl, AVIFOURCC_strf, 0 );
- avi_chunk_list_t *p_riff;
- avi_chunk_list_t *p_movi;
- avi_chunk_idx1_t *p_idx1;
+ if( p_avi_strf_auds->p_wf->wFormatTag != WAVE_FORMAT_PCM &&
+ (unsigned int)p_stream->i_rate == p_avi_strf_auds->p_wf->nSamplesPerSec )
+ {
+ int64_t i_track_length =
+ p_stream->p_index[p_stream->i_idxnb-1].i_length +
+ p_stream->p_index[p_stream->i_idxnb-1].i_lengthtotal;
+ mtime_t i_length = (mtime_t)p_avih->i_totalframes *
+ (mtime_t)p_avih->i_microsecperframe;
- unsigned int i_stream;
- unsigned int i_index;
- off_t i_offset;
+ if( i_length == 0 )
+ {
+ msg_Warn( p_input, "track[%d] cannot be fixed (BeOS MediaKit generated)", i );
+ continue;
+ }
+ p_stream->i_samplesize = 1;
+ p_stream->i_rate = i_track_length * (int64_t)1000000/ i_length;
+ msg_Warn( p_input, "track[%d] fixed with rate=%d scale=%d (BeOS MediaKit generated)", i, p_stream->i_rate, p_stream->i_scale );
+ }
+#undef p_stream
+ }
- p_riff = (avi_chunk_list_t*)AVI_ChunkFind( &p_avi->ck_root,
- AVIFOURCC_RIFF, 0);
+ if( p_avi->i_length )
+ {
+ p_input->stream.i_mux_rate =
+ stream_Size( p_input->s ) / 50 / p_avi->i_length;
+
+ p_avi->b_interleaved = AVI_Interleaved( p_input );
+ msg_Dbg( p_input, "interleaved=%s",
+ p_avi->b_interleaved ? "yes" : "no" );
+ }
- p_idx1 = (avi_chunk_idx1_t*)AVI_ChunkFind( p_riff, AVIFOURCC_idx1, 0);
- p_movi = (avi_chunk_list_t*)AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
+ b_stream_audio = VLC_FALSE;
+ b_stream_video = VLC_FALSE;
- if( !p_idx1 )
+ for( i = 0; i < p_avi->i_streams; i++ )
{
- msg_Warn( p_input, "cannot find idx1 chunk, no index defined" );
- return;
+#define tk p_avi->pp_info[i]
+ if( tk->p_es->i_cat == VIDEO_ES && !b_stream_video )
+ {
+ b_stream_video = AVI_StreamStart( p_input, i );
+ }
+ else if(tk->p_es->i_cat == AUDIO_ES && !b_stream_audio )
+ {
+ b_stream_audio = AVI_StreamStart( p_input, i );
+ }
+#undef tk
}
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
+
+ if( !b_stream_video )
{
- p_avi->pp_info[i_stream]->i_idxnb = 0;
- p_avi->pp_info[i_stream]->i_idxmax = 0;
- p_avi->pp_info[i_stream]->p_index = NULL;
+ msg_Warn( p_input, "no video stream found" );
}
- /* *** calculate offset *** */
- if( p_idx1->i_entry_count > 0 &&
- p_idx1->entry[0].i_pos < p_movi->i_chunk_pos )
+ if( !b_stream_audio )
{
- i_offset = p_movi->i_chunk_pos + 8;
+ msg_Warn( p_input, "no audio stream found!" );
}
- else
+
+ vlc_mutex_lock( &p_input->stream.stream_lock );
+ p_input->stream.p_selected_program->b_is_ok = 1;
+ vlc_mutex_unlock( &p_input->stream.stream_lock );
+
+ if( p_avi->b_seekable )
{
- i_offset = 0;
+ /* we have read all chunk so go back to movi */
+ stream_Seek( p_input->s, p_movi->i_chunk_pos );
}
+ /* Skip movi header */
+ stream_Read( p_input->s, NULL, 12 );
- for( i_index = 0; i_index < p_idx1->i_entry_count; i_index++ )
- {
- unsigned int i_cat;
-
- AVI_ParseStreamHeader( p_idx1->entry[i_index].i_fourcc,
- &i_stream,
- &i_cat );
- if( i_stream < p_avi->i_streams &&
- i_cat == p_avi->pp_info[i_stream]->i_cat )
- {
- AVIIndexEntry_t index;
- index.i_id = p_idx1->entry[i_index].i_fourcc;
- index.i_flags =
- p_idx1->entry[i_index].i_flags&(~AVIIF_FIXKEYFRAME);
- index.i_pos = p_idx1->entry[i_index].i_pos + i_offset;
- index.i_length = p_idx1->entry[i_index].i_length;
- AVI_IndexAddEntry( p_avi, i_stream, &index );
- }
- }
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
- {
- msg_Dbg( p_input,
- "stream[%d] creating %d index entries",
- i_stream,
- p_avi->pp_info[i_stream]->i_idxnb );
- }
+ p_avi->i_movi_begin = p_movi->i_chunk_pos;
+ return VLC_SUCCESS;
+error:
+ AVI_ChunkFreeRoot( p_input->s, &p_avi->ck_root );
+ free( p_avi );
+ return VLC_EGENERIC;
}
-static void AVI_IndexCreate( input_thread_t *p_input )
+/*****************************************************************************
+ * Close: frees unused data
+ *****************************************************************************/
+static void Close ( vlc_object_t * p_this )
{
- demux_sys_t *p_avi = p_input->p_demux_data;
-
- avi_chunk_list_t *p_riff;
- avi_chunk_list_t *p_movi;
-
- unsigned int i_stream;
- off_t i_movi_end;
-
- p_riff = (avi_chunk_list_t*)AVI_ChunkFind( &p_avi->ck_root,
- AVIFOURCC_RIFF, 0);
- p_movi = (avi_chunk_list_t*)AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
-
- if( !p_movi )
- {
- msg_Err( p_input, "cannot find p_movi" );
- return;
- }
-
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
- {
- p_avi->pp_info[i_stream]->i_idxnb = 0;
- p_avi->pp_info[i_stream]->i_idxmax = 0;
- p_avi->pp_info[i_stream]->p_index = NULL;
- }
- i_movi_end = __MIN( (off_t)(p_movi->i_chunk_pos + p_movi->i_chunk_size),
- p_input->stream.p_selected_area->i_size );
+ input_thread_t * p_input = (input_thread_t *)p_this;
+ unsigned int i;
+ demux_sys_t *p_avi = p_input->p_demux_data ;
- AVI_SeekAbsolute( p_input, p_movi->i_chunk_pos + 12);
- msg_Warn( p_input, "creating index from LIST-movi, will take time !" );
- for( ;; )
+ for( i = 0; i < p_avi->i_streams; i++ )
{
- avi_packet_t pk;
-
- if( AVI_PacketGetHeader( p_input, &pk ) )
- {
- break;
- }
- if( pk.i_stream < p_avi->i_streams &&
- pk.i_cat == p_avi->pp_info[pk.i_stream]->i_cat )
- {
- AVIIndexEntry_t index;
- index.i_id = pk.i_fourcc;
- index.i_flags =
- AVI_GetKeyFlag(p_avi->pp_info[pk.i_stream]->i_codec, pk.i_peek);
- index.i_pos = pk.i_pos;
- index.i_length = pk.i_size;
- AVI_IndexAddEntry( p_avi, pk.i_stream, &index );
- }
- else
- {
- switch( pk.i_fourcc )
- {
- case AVIFOURCC_idx1:
- goto print_stat;
- case AVIFOURCC_rec:
- case AVIFOURCC_JUNK:
- break;
- default:
- msg_Warn( p_input, "need resync, probably broken avi" );
- if( AVI_PacketSearch( p_input ) )
- {
- msg_Warn( p_input, "lost sync, abord index creation" );
- goto print_stat;
- }
- }
- }
- if( pk.i_pos + pk.i_size >= i_movi_end ||
- AVI_PacketNext( p_input ) )
+ if( p_avi->pp_info[i] )
{
- break;
+ FREE( p_avi->pp_info[i]->p_index );
+ free( p_avi->pp_info[i] );
}
}
+ FREE( p_avi->pp_info );
+ AVI_ChunkFreeRoot( p_input->s, &p_avi->ck_root );
-print_stat:
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
- {
- msg_Dbg( p_input,
- "stream[%d] creating %d index entries",
- i_stream,
- p_avi->pp_info[i_stream]->i_idxnb );
- }
+ free( p_avi );
}
-
/*****************************************************************************
- * Stream management
+ * Demux_Seekable: reads and demuxes data packets for stream seekable
+ *****************************************************************************
+ * AVIDemux: reads and demuxes data packets
+ *****************************************************************************
+ * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
*****************************************************************************/
-static vlc_bool_t AVI_StreamStart ( input_thread_t *, demux_sys_t *, int );
-static int AVI_StreamSeek ( input_thread_t *, demux_sys_t *, int, mtime_t );
-static void AVI_StreamStop ( input_thread_t *, demux_sys_t *, int );
-static int AVI_StreamStopFinishedStreams( input_thread_t *, demux_sys_t * );
-
-static vlc_bool_t AVI_StreamStart( input_thread_t *p_input,
- demux_sys_t *p_avi, int i_stream )
+typedef struct avi_stream_toread_s
{
-#define p_stream p_avi->pp_info[i_stream]
- if( !p_stream->p_es )
- {
- msg_Warn( p_input, "stream[%d] unselectable", i_stream );
- return VLC_FALSE;
- }
- if( p_stream->b_activated )
- {
- msg_Warn( p_input, "stream[%d] already selected", i_stream );
- return VLC_TRUE;
- }
+ vlc_bool_t b_ok;
- if( !p_stream->p_es->p_decoder_fifo )
- {
- vlc_mutex_lock( &p_input->stream.stream_lock );
- input_SelectES( p_input, p_stream->p_es );
- vlc_mutex_unlock( &p_input->stream.stream_lock );
- }
- p_stream->b_activated = p_stream->p_es->p_decoder_fifo ? VLC_TRUE
- : VLC_FALSE;
- if( p_stream->b_activated && p_avi->b_seekable)
- {
- AVI_StreamSeek( p_input, p_avi, i_stream, p_avi->i_time );
- }
+ int i_toread;
- return p_stream->b_activated;
-#undef p_stream
-}
+ off_t i_posf; // where we will read :
+ // if i_idxposb == 0 : begining of chunk (+8 to acces data)
+ // else : point on data directly
+} avi_stream_toread_t;
-static void AVI_StreamStop( input_thread_t *p_input,
- demux_sys_t *p_avi, int i_stream )
+static int Demux_Seekable( input_thread_t *p_input )
{
-#define p_stream p_avi->pp_info[i_stream]
+ unsigned int i_stream_count;
+ unsigned int i_stream;
+ vlc_bool_t b_stream;
+ vlc_bool_t b_play_audio;
+ vlc_bool_t b_video; /* is there some video track selected */
+ // cannot be more than 100 stream (dcXX or wbXX)
+ avi_stream_toread_t toread[100];
+
+ demux_sys_t *p_avi = p_input->p_demux_data;
+
- if( !p_stream->b_activated )
+ /* detect new selected/unselected streams */
+ for( i_stream = 0,i_stream_count= 0, b_video = VLC_FALSE;
+ i_stream < p_avi->i_streams; i_stream++ )
{
- msg_Warn( p_input, "stream[%d] already unselected", i_stream );
- return;
+#define p_stream p_avi->pp_info[i_stream]
+ if( p_stream->p_es )
+ {
+ if( p_stream->p_es->p_decoder_fifo &&
+ !p_stream->b_activated )
+ {
+ AVI_StreamStart( p_input, i_stream );
+ }
+ else
+ if( !p_stream->p_es->p_decoder_fifo &&
+ p_stream->b_activated )
+ {
+ AVI_StreamStop( p_input, i_stream );
+ }
+ }
+ if( p_stream->b_activated )
+ {
+ i_stream_count++;
+ if( p_stream->i_cat == VIDEO_ES )
+ {
+ b_video = VLC_TRUE;
+ }
+ }
+#undef p_stream
}
- if( p_stream->p_es->p_decoder_fifo )
+ if( i_stream_count <= 0 )
{
- vlc_mutex_lock( &p_input->stream.stream_lock );
- input_UnselectES( p_input, p_stream->p_es );
- vlc_mutex_unlock( &p_input->stream.stream_lock );
+ msg_Warn( p_input, "no track selected, exiting..." );
+ return( 0 );
}
+ /* wait for the good time */
+ p_avi->i_pcr = p_avi->i_time * 9 / 100;
+
+ input_ClockManageRef( p_input,
+ p_input->stream.p_selected_program,
+ p_avi->i_pcr );
- p_stream->b_activated = VLC_FALSE;
-#undef p_stream
-}
+ p_avi->i_time += 25*1000; /* read 25ms */
-static int AVI_StreamStopFinishedStreams( input_thread_t *p_input,
- demux_sys_t *p_avi )
-{
- int i_stream;
- int b_end;
+ /* Check if we need to send the audio data to decoder */
+ b_play_audio = !p_input->stream.control.b_mute;
- for( i_stream = 0,b_end = VLC_TRUE;
- i_stream < p_avi->i_streams; i_stream++ )
+ /* init toread */
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
{
#define p_stream p_avi->pp_info[i_stream]
- if( p_stream->i_idxposc >= p_stream->i_idxnb )
+ mtime_t i_dpts;
+
+ toread[i_stream].b_ok = p_stream->b_activated;
+ if( p_stream->i_idxposc < p_stream->i_idxnb )
{
- AVI_StreamStop( p_input, p_avi, i_stream );
+ toread[i_stream].i_posf =
+ p_stream->p_index[p_stream->i_idxposc].i_pos;
+ if( p_stream->i_idxposb > 0 )
+ {
+ toread[i_stream].i_posf += 8 + p_stream->i_idxposb;
+ }
}
else
{
- b_end = VLC_FALSE;
+ toread[i_stream].i_posf = -1;
}
-#undef p_stream
- }
- return( b_end );
-}
-/****************************************************************************
- * AVI_MovieGetLength give max streams length in second
- ****************************************************************************/
-static mtime_t AVI_MovieGetLength( input_thread_t *p_input, demux_sys_t *p_avi )
-{
- unsigned int i_stream;
- mtime_t i_maxlength;
- i_maxlength = 0;
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
- {
-#define p_stream p_avi->pp_info[i_stream]
- mtime_t i_length;
- /* fix length for each stream */
- if( p_stream->i_idxnb < 1 || !p_stream->p_index )
- {
- continue;
- }
+ i_dpts = p_avi->i_time - AVI_GetPTS( p_stream );
if( p_stream->i_samplesize )
{
- i_length =
- (mtime_t)( p_stream->p_index[p_stream->i_idxnb-1].i_lengthtotal +
- p_stream->p_index[p_stream->i_idxnb-1].i_length ) *
- (mtime_t)p_stream->i_scale /
- (mtime_t)p_stream->i_rate /
- (mtime_t)p_stream->i_samplesize;
+ toread[i_stream].i_toread = AVI_PTSToByte( p_stream,
+ __ABS( i_dpts ) );
}
else
{
- i_length = (mtime_t)p_stream->i_idxnb *
- (mtime_t)p_stream->i_scale /
- (mtime_t)p_stream->i_rate;
+ toread[i_stream].i_toread = AVI_PTSToChunk( p_stream,
+ __ABS( i_dpts ) );
}
- msg_Dbg( p_input,
- "stream[%d] length:"I64Fd" (based on index)",
- i_stream,
- i_length );
- i_maxlength = __MAX( i_maxlength, i_length );
-#undef p_stream
+ if( i_dpts < 0 )
+ {
+ toread[i_stream].i_toread *= -1;
+ }
+#undef p_stream
}
- return i_maxlength;
-}
-
-/*****************************************************************************
- * AVIEnd: frees unused data
- *****************************************************************************/
-static void __AVIEnd ( vlc_object_t * p_this )
-{
- input_thread_t * p_input = (input_thread_t *)p_this;
- unsigned int i;
- demux_sys_t *p_avi = p_input->p_demux_data ;
+ b_stream = VLC_FALSE;
- if( p_avi->pp_info )
+ for( ;; )
{
- for( i = 0; i < p_avi->i_streams; i++ )
+#define p_stream p_avi->pp_info[i_stream]
+ vlc_bool_t b_done;
+ pes_packet_t *p_pes;
+ off_t i_pos;
+ unsigned int i;
+ size_t i_size;
+
+ /* search for first chunk to be read */
+ for( i = 0, b_done = VLC_TRUE, i_pos = -1; i < p_avi->i_streams; i++ )
{
- if( p_avi->pp_info[i] )
+ if( !toread[i].b_ok ||
+ AVI_GetDPTS( p_avi->pp_info[i],
+ toread[i].i_toread ) <= -25 * 1000 )
+ {
+ continue;
+ }
+
+ if( toread[i].i_toread > 0 )
+ {
+ b_done = VLC_FALSE; // not yet finished
+ }
+
+ if( toread[i].i_posf > 0 )
{
- if( p_avi->pp_info[i]->p_index )
+ if( i_pos == -1 || i_pos > toread[i_stream].i_posf )
{
- free( p_avi->pp_info[i]->p_index );
+ i_stream = i;
+ i_pos = toread[i].i_posf;
}
- free( p_avi->pp_info[i] );
}
}
- free( p_avi->pp_info );
- }
-#ifdef __AVI_SUBTITLE__
- if( p_avi->p_sub )
- {
- subtitle_Close( p_avi->p_sub );
- p_avi->p_sub = NULL;
- }
-#endif
- AVI_ChunkFreeRoot( p_input, &p_avi->ck_root );
-}
-/*****************************************************************************
- * AVIInit: check file and initializes AVI structures
- *****************************************************************************/
-static int AVIInit( vlc_object_t * p_this )
-{
- input_thread_t * p_input = (input_thread_t *)p_this;
- avi_chunk_t ck_riff;
- avi_chunk_list_t *p_riff = (avi_chunk_list_t*)&ck_riff;
- avi_chunk_list_t *p_hdrl, *p_movi;
-#if 0
- avi_chunk_list_t *p_INFO;
- avi_chunk_strz_t *p_name;
-#endif
- avi_chunk_avih_t *p_avih;
- demux_sys_t *p_avi;
- es_descriptor_t *p_es = NULL; /* avoid warning */
- unsigned int i;
-#ifdef __AVI_SUBTITLE__
- mtime_t i_microsecperframe = 0; // for some subtitle format
-#endif
+ if( b_done )
+ {
+// return( b_stream ? 1 : 0 );
+ return( 1 );
+ }
- vlc_bool_t b_stream_audio, b_stream_video;
+ if( i_pos == -1 )
+ {
+ /* no valid index, we will parse directly the stream
+ * in case we fail we will disable all finished stream */
+ if( p_avi->i_movi_lastchunk_pos >= p_avi->i_movi_begin + 12 )
+ {
+ stream_Seek( p_input->s, p_avi->i_movi_lastchunk_pos );
+ if( AVI_PacketNext( p_input ) )
+ {
+ return( AVI_StreamStopFinishedStreams( p_input ) ? 0 : 1 );
+ }
+ }
+ else
+ {
+ stream_Seek( p_input->s, p_avi->i_movi_begin + 12 );
+ }
- p_input->pf_demux = AVIDemux_Seekable;
- if( AVI_TestFile( p_input ) )
- {
- msg_Warn( p_input, "avi module discarded (invalid headr)" );
- return VLC_EGENERIC;
- }
+ for( ;; )
+ {
+ avi_packet_t avi_pk;
- /* Initialize access plug-in structures. */
- if( p_input->i_mtu == 0 )
- {
- /* Improve speed. */
- p_input->i_bufsize = INPUT_DEFAULT_BUFSIZE;
- }
+ if( AVI_PacketGetHeader( p_input, &avi_pk ) )
+ {
+ msg_Warn( p_input,
+ "cannot get packet header, track disabled" );
+ return( AVI_StreamStopFinishedStreams( p_input ) ? 0 : 1 );
+ }
+ if( avi_pk.i_stream >= p_avi->i_streams ||
+ ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
+ {
+ if( AVI_PacketNext( p_input ) )
+ {
+ msg_Warn( p_input,
+ "cannot skip packet, track disabled" );
+ return( AVI_StreamStopFinishedStreams( p_input ) ? 0 : 1 );
+ }
+ continue;
+ }
+ else
+ {
+ /* add this chunk to the index */
+ AVIIndexEntry_t index;
- if( !( p_input->p_demux_data =
- p_avi = malloc( sizeof(demux_sys_t) ) ) )
- {
- msg_Err( p_input, "out of memory" );
- return VLC_ENOMEM;
- }
- memset( p_avi, 0, sizeof( demux_sys_t ) );
- p_avi->i_time = 0;
- p_avi->i_pcr = 0;
- p_avi->b_seekable = ( ( p_input->stream.b_seekable )
- &&( p_input->stream.i_method == INPUT_METHOD_FILE ) );
- p_avi->i_movi_lastchunk_pos = 0;
+ index.i_id = avi_pk.i_fourcc;
+ index.i_flags =
+ AVI_GetKeyFlag(p_avi->pp_info[avi_pk.i_stream]->i_codec,
+ avi_pk.i_peek);
+ index.i_pos = avi_pk.i_pos;
+ index.i_length = avi_pk.i_size;
+ AVI_IndexAddEntry( p_avi, avi_pk.i_stream, &index );
- /* *** for unseekable stream, automaticaly use AVIDemux_interleaved *** */
- if( !p_avi->b_seekable || config_GetInt( p_input, "avi-interleaved" ) )
- {
- p_input->pf_demux = AVIDemux_UnSeekable;
- }
+ i_stream = avi_pk.i_stream;
+ /* do we will read this data ? */
+ if( AVI_GetDPTS( p_stream,
+ toread[i_stream].i_toread ) > -25 * 1000 )
+ {
+ break;
+ }
+ else
+ {
+ if( AVI_PacketNext( p_input ) )
+ {
+ msg_Warn( p_input,
+ "cannot skip packet, track disabled" );
+ return( AVI_StreamStopFinishedStreams( p_input ) ? 0 : 1 );
+ }
+ }
+ }
+ }
- if( AVI_ChunkReadRoot( p_input, &p_avi->ck_root, p_avi->b_seekable ) )
- {
- msg_Err( p_input, "avi module discarded (invalid file)" );
- return VLC_EGENERIC;
- }
- AVI_ChunkDumpDebug( p_input, &p_avi->ck_root );
+ }
+ else
+ {
+ stream_Seek( p_input->s, i_pos );
+ }
+ /* read thoses data */
+ if( p_stream->i_samplesize )
+ {
+ unsigned int i_toread;
- p_riff = (avi_chunk_list_t*)AVI_ChunkFind( &p_avi->ck_root,
- AVIFOURCC_RIFF, 0 );
- p_hdrl = (avi_chunk_list_t*)AVI_ChunkFind( p_riff,
- AVIFOURCC_hdrl, 0 );
- p_movi = (avi_chunk_list_t*)AVI_ChunkFind( p_riff,
- AVIFOURCC_movi, 0 );
-#if 0
- p_INFO = (avi_chunk_list_t*)AVI_ChunkFind( p_riff,
- AVIFOURCC_INFO, 0 );
- p_name = (avi_chunk_strz_t*)AVI_ChunkFind( p_INFO,
- AVIFOURCC_INAM, 0 );
- if( p_name )
- {
+ if( ( i_toread = toread[i_stream].i_toread ) <= 0 )
+ {
+ if( p_stream->i_samplesize > 1 )
+ {
+ i_toread = p_stream->i_samplesize;
+ }
+ else
+ {
+ i_toread = __MAX( AVI_PTSToByte( p_stream, 20 * 1000 ), 100 );
+ }
+ }
+ i_size = __MIN( p_stream->p_index[p_stream->i_idxposc].i_length -
+ p_stream->i_idxposb,
+ i_toread );
+ }
+ else
+ {
+ i_size = p_stream->p_index[p_stream->i_idxposc].i_length;
+ }
- }
+ if( p_stream->i_idxposb == 0 )
+ {
+ i_size += 8; // need to read and skip header
+ }
+
+ if( ( p_pes = stream_PesPacket( p_input->s, __EVEN( i_size ) ) )==NULL )
+ {
+ msg_Warn( p_input, "failled reading data" );
+ AVI_StreamStop( p_input, i_stream );
+ toread[i_stream].b_ok = VLC_FALSE;
+ continue;
+ }
+ if( i_size % 2 ) // read was padded on word boundary
+ {
+ p_pes->p_last->p_payload_end--;
+ p_pes->i_pes_size--;
+ }
+ // skip header
+ if( p_stream->i_idxposb == 0 )
+ {
+ p_pes->p_first->p_payload_start += 8;
+ p_pes->i_pes_size -= 8;
+ }
+
+ p_pes->i_pts = AVI_GetPTS( p_stream );
+
+#if 0
+ /* fix pts for audio: ie pts sould be for the first byte of the first frame */
+ if( p_stream->i_samplesize == 1 )
+ {
+ AVI_FixPTS( p_stream, p_pes );
+ }
#endif
- if( !p_hdrl || !p_movi )
- {
- msg_Err( p_input, "avi module discarded (invalid file)" );
- return VLC_EGENERIC;
- }
+ /* read data */
+ if( p_stream->i_samplesize )
+ {
+ if( p_stream->i_idxposb == 0 )
+ {
+ i_size -= 8;
+ }
+ toread[i_stream].i_toread -= i_size;
+ p_stream->i_idxposb += i_size;
+ if( p_stream->i_idxposb >=
+ p_stream->p_index[p_stream->i_idxposc].i_length )
+ {
+ p_stream->i_idxposb = 0;
+ p_stream->i_idxposc++;
+ }
+ }
+ else
+ {
+ toread[i_stream].i_toread--;
+ p_stream->i_idxposc++;
+ }
- if( !( p_avih = (avi_chunk_avih_t*)AVI_ChunkFind( p_hdrl,
- AVIFOURCC_avih, 0 ) ) )
- {
- msg_Err( p_input, "cannot find avih chunk" );
- return VLC_EGENERIC;
- }
- p_avi->i_streams = AVI_ChunkCount( p_hdrl, AVIFOURCC_strl );
- if( p_avih->i_streams != p_avi->i_streams )
- {
- msg_Warn( p_input,
- "found %d stream but %d are declared",
- p_avi->i_streams,
- p_avih->i_streams );
- }
- if( p_avi->i_streams == 0 )
- {
- AVIEnd( p_input );
- msg_Err( p_input, "no stream defined!" );
- return VLC_EGENERIC;
+ if( p_stream->i_idxposc < p_stream->i_idxnb)
+ {
+ toread[i_stream].i_posf =
+ p_stream->p_index[p_stream->i_idxposc].i_pos;
+ if( p_stream->i_idxposb > 0 )
+ {
+ toread[i_stream].i_posf += 8 + p_stream->i_idxposb;
+ }
+
+ }
+ else
+ {
+ toread[i_stream].i_posf = -1;
+ }
+
+ b_stream = VLC_TRUE; // at least one read succeed
+
+ if( p_stream->p_es && p_stream->p_es->p_decoder_fifo &&
+ ( b_play_audio || p_stream->i_cat != AUDIO_ES ) )
+ {
+ p_pes->i_dts =
+ p_pes->i_pts =
+ input_ClockGetTS( p_input,
+ p_input->stream.p_selected_program,
+ p_pes->i_pts * 9/100);
+
+ p_pes->i_rate = p_input->stream.control.i_rate;
+
+ input_DecodePES( p_stream->p_es->p_decoder_fifo, p_pes );
+ }
+ else
+ {
+ input_DeletePES( p_input->p_method_data, p_pes );
+ }
}
+}
- /* create one program */
- vlc_mutex_lock( &p_input->stream.stream_lock );
- if( input_InitStream( p_input, 0 ) == -1)
+
+/*****************************************************************************
+ * Demux_UnSeekable: reads and demuxes data packets for unseekable file
+ *****************************************************************************
+ * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
+ *****************************************************************************/
+static int Demux_UnSeekable( input_thread_t *p_input )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ avi_stream_t *p_stream_master;
+ vlc_bool_t b_audio;
+ unsigned int i_stream;
+ unsigned int i_packet;
+
+ /* Check if we need to send the audio data to decoder */
+ b_audio = !p_input->stream.control.b_mute;
+
+ input_ClockManageRef( p_input,
+ p_input->stream.p_selected_program,
+ p_avi->i_pcr );
+ /* *** find master stream for data packet skipping algo *** */
+ /* *** -> first video, if any, or first audio ES *** */
+ for( i_stream = 0, p_stream_master = NULL;
+ i_stream < p_avi->i_streams; i_stream++ )
{
- vlc_mutex_unlock( &p_input->stream.stream_lock );
- AVIEnd( p_input );
- msg_Err( p_input, "cannot init stream" );
- return VLC_EGENERIC;
+#define p_stream p_avi->pp_info[i_stream]
+ if( p_stream->p_es &&
+ p_stream->p_es->p_decoder_fifo )
+ {
+ if( p_stream->i_cat == VIDEO_ES )
+ {
+ p_stream_master = p_stream;
+ break;
+ }
+ if( p_stream->i_cat == AUDIO_ES && !p_stream_master )
+ {
+ p_stream_master = p_stream;
+ }
+ }
+#undef p_stream
}
- if( input_AddProgram( p_input, 0, 0) == NULL )
+ if( !p_stream_master )
{
- vlc_mutex_unlock( &p_input->stream.stream_lock );
- AVIEnd( p_input );
- msg_Err( p_input, "cannot add program" );
- return VLC_EGENERIC;
+ msg_Warn( p_input, "no more stream selected" );
+ return( 0 );
}
- p_input->stream.p_selected_program = p_input->stream.pp_programs[0];
- vlc_mutex_unlock( &p_input->stream.stream_lock );
- /* print informations on streams */
- msg_Dbg( p_input, "AVIH: %d stream, flags %s%s%s%s ",
- p_avi->i_streams,
- p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
- p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
- p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
- p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
- {
- input_info_category_t *p_cat = input_InfoCategory( p_input, "Avi" );
- input_AddInfo( p_cat, "Number of streams", "%d", p_avi->i_streams );
- input_AddInfo( p_cat, "Flags", "%s%s%s%s",
- p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
- p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
- p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
- p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
- }
-
- /* now read info on each stream and create ES */
- p_avi->pp_info = calloc( p_avi->i_streams,
- sizeof( avi_stream_t* ) );
- memset( p_avi->pp_info,
- 0,
- sizeof( avi_stream_t* ) * p_avi->i_streams );
+ p_avi->i_pcr = AVI_GetPTS( p_stream_master ) * 9 / 100;
- for( i = 0 ; i < p_avi->i_streams; i++ )
+ for( i_packet = 0; i_packet < 10; i_packet++)
{
- avi_chunk_list_t *p_avi_strl;
- avi_chunk_strh_t *p_avi_strh;
- avi_chunk_strf_auds_t *p_avi_strf_auds;
- avi_chunk_strf_vids_t *p_avi_strf_vids;
- int i_init_size;
- void *p_init_data;
-#define p_info p_avi->pp_info[i]
- p_info = malloc( sizeof(avi_stream_t ) );
- memset( p_info, 0, sizeof( avi_stream_t ) );
+#define p_stream p_avi->pp_info[avi_pk.i_stream]
+
+ avi_packet_t avi_pk;
+
+ if( AVI_PacketGetHeader( p_input, &avi_pk ) )
+ {
+ return( 0 );
+ }
+
+ if( avi_pk.i_stream >= p_avi->i_streams ||
+ ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
+ {
+ /* we haven't found an audio or video packet:
+ * - we have seek, found first next packet
+ * - others packets could be found, skip them
+ */
+ switch( avi_pk.i_fourcc )
+ {
+ case AVIFOURCC_JUNK:
+ case AVIFOURCC_LIST:
+ case AVIFOURCC_RIFF:
+ return( !AVI_PacketNext( p_input ) ? 1 : 0 );
+ case AVIFOURCC_idx1:
+ if( p_avi->b_odml )
+ {
+ return( !AVI_PacketNext( p_input ) ? 1 : 0 );
+ }
+ return( 0 ); // eof
+ default:
+ msg_Warn( p_input,
+ "seems to have lost position, resync" );
+ if( AVI_PacketSearch( p_input ) )
+ {
+ msg_Err( p_input, "resync failed" );
+ return( -1 );
+ }
+ }
+ }
+ else
+ {
+ /* do will send this packet to decoder ? */
+ if( ( !b_audio && avi_pk.i_cat == AUDIO_ES )||
+ !p_stream->p_es ||
+ !p_stream->p_es->p_decoder_fifo )
+ {
+ if( AVI_PacketNext( p_input ) )
+ {
+ return( 0 );
+ }
+ }
+ else
+ {
+ /* it's a selected stream, check for time */
+ if( __ABS( AVI_GetPTS( p_stream ) -
+ AVI_GetPTS( p_stream_master ) )< 600*1000 )
+ {
+ /* load it and send to decoder */
+ pes_packet_t *p_pes;
+ if( AVI_PacketRead( p_input, &avi_pk, &p_pes ) || !p_pes)
+ {
+ return( -1 );
+ }
+ p_pes->i_dts =
+ p_pes->i_pts =
+ input_ClockGetTS( p_input,
+ p_input->stream.p_selected_program,
+ AVI_GetPTS( p_stream ) * 9/100);
+
+ p_pes->i_rate = p_input->stream.control.i_rate;
+
+ input_DecodePES( p_stream->p_es->p_decoder_fifo, p_pes );
+ }
+ else
+ {
+ if( AVI_PacketNext( p_input ) )
+ {
+ return( 0 );
+ }
+ }
+ }
- p_avi_strl = (avi_chunk_list_t*)AVI_ChunkFind( p_hdrl,
- AVIFOURCC_strl, i );
- p_avi_strh = (avi_chunk_strh_t*)AVI_ChunkFind( p_avi_strl,
- AVIFOURCC_strh, 0 );
- p_avi_strf_auds = (avi_chunk_strf_auds_t*)
- p_avi_strf_vids = (avi_chunk_strf_vids_t*)
- AVI_ChunkFind( p_avi_strl, AVIFOURCC_strf, 0 );
+ /* *** update stream time position *** */
+ if( p_stream->i_samplesize )
+ {
+ p_stream->i_idxposb += avi_pk.i_size;
+ }
+ else
+ {
+ p_stream->i_idxposc++;
+ }
- if( !p_avi_strl || !p_avi_strh ||
- ( !p_avi_strf_auds && !p_avi_strf_vids ) )
- {
- msg_Warn( p_input, "stream[%d] incomlete", i );
- continue;
}
- /* *** Init p_info *** */
- p_info->i_rate = p_avi_strh->i_rate;
- p_info->i_scale = p_avi_strh->i_scale;
- p_info->i_samplesize = p_avi_strh->i_samplesize;
- msg_Dbg( p_input, "stream[%d] rate:%d scale:%d samplesize:%d",
- i,
- p_info->i_rate, p_info->i_scale, p_info->i_samplesize );
- switch( p_avi_strh->i_type )
+#undef p_stream
+ }
+
+ return( 1 );
+}
+
+/*****************************************************************************
+ * Seek: goto to i_date or i_percent
+ *****************************************************************************
+ * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
+ *****************************************************************************/
+static int Seek( input_thread_t *p_input, mtime_t i_date, int i_percent )
+{
+
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ unsigned int i_stream;
+ msg_Dbg( p_input,
+ "seek requested: "I64Fd" secondes %d%%",
+ i_date / 1000000,
+ i_percent );
+
+ if( p_avi->b_seekable )
+ {
+ if( !p_avi->i_length || p_avi->b_interleaved )
{
- case( AVIFOURCC_auds ):
- p_info->i_cat = AUDIO_ES;
- p_info->i_fourcc =
- AVI_FourccGetCodec( AUDIO_ES,
- p_avi_strf_auds->p_wf->wFormatTag );
- p_info->i_codec = p_info->i_fourcc;
- i_init_size = p_avi_strf_auds->i_chunk_size;
- p_init_data = p_avi_strf_auds->p_wf;
- msg_Dbg( p_input, "stream[%d] audio(0x%x) %d channels %dHz %dbits",
- i,
- p_avi_strf_auds->p_wf->wFormatTag,
- p_avi_strf_auds->p_wf->nChannels,
- p_avi_strf_auds->p_wf->nSamplesPerSec,
- p_avi_strf_auds->p_wf->wBitsPerSample );
- {
- char hepp[sizeof("Stream") + 10];
- input_info_category_t *p_cat;
- sprintf(hepp, "Stream %d", i);
- p_cat = input_InfoCategory( p_input, hepp);
- input_AddInfo( p_cat, "Type", "audio(0x%x)",
- p_avi_strf_auds->p_wf->wFormatTag );
- input_AddInfo( p_cat, "Codec", "%4.4s",
- (const char*)&(p_info->i_codec) );
- input_AddInfo( p_cat, "Channels", "%d",
- p_avi_strf_auds->p_wf->nChannels );
- input_AddInfo( p_cat, "Samplerate", "%d",
- p_avi_strf_auds->p_wf->nSamplesPerSec );
- input_AddInfo( p_cat, "Bits Per Sample", "%d",
- p_avi_strf_auds->p_wf->wBitsPerSample );
- }
- break;
+ avi_stream_t *p_stream;
+ int64_t i_pos;
- case( AVIFOURCC_vids ):
- p_info->i_cat = VIDEO_ES;
- /* XXX quick hack for playing ffmpeg video, I don't know
- who is doing something wrong */
- p_info->i_samplesize = 0;
- p_info->i_fourcc = p_avi_strf_vids->p_bih->biCompression;
- p_info->i_codec =
- AVI_FourccGetCodec( VIDEO_ES, p_info->i_fourcc );
- i_init_size = p_avi_strf_vids->i_chunk_size;
- p_init_data = p_avi_strf_vids->p_bih;
- msg_Dbg( p_input, "stream[%d] video(%4.4s) %dx%d %dbpp %ffps",
- i,
- (char*)&p_avi_strf_vids->p_bih->biCompression,
- p_avi_strf_vids->p_bih->biWidth,
- p_avi_strf_vids->p_bih->biHeight,
- p_avi_strf_vids->p_bih->biBitCount,
- (float)p_info->i_rate /
- (float)p_info->i_scale );
- {
- char hepp[sizeof("Stream") + 10];
- input_info_category_t *p_cat;
- sprintf(hepp, "stream %d", i);
- p_cat = input_InfoCategory( p_input, hepp);
- input_AddInfo( p_cat, "Type", "video" );
- input_AddInfo( p_cat, "Codec", "%4.4s",
- (const char*)&(p_info->i_codec) );
- input_AddInfo( p_cat, "Resolution", "%dx%d",
- p_avi_strf_vids->p_bih->biWidth,
- p_avi_strf_vids->p_bih->biHeight );
- input_AddInfo( p_cat, "Frame Rate", "%f",
- (float)p_info->i_rate /
- (float)p_info->i_scale );
- input_AddInfo( p_cat, "Bits Per Pixel", "%d",
- p_avi_strf_vids->p_bih->biBitCount );
- }
-#ifdef __AVI_SUBTITLE__
- if( i_microsecperframe == 0 )
+ /* use i_percent to create a true i_date */
+ if( !p_avi->b_interleaved )
+ {
+ msg_Warn( p_input,
+ "mmh, seeking without index at %d%%"
+ " work only for interleaved file", i_percent );
+ }
+
+ if( i_percent >= 100 )
+ {
+ msg_Warn( p_input, "cannot seek so far !" );
+ return( -1 );
+ }
+ i_percent = __MAX( i_percent, 0 );
+
+ /* try to find chunk that is at i_percent or the file */
+ i_pos = __MAX( i_percent *
+ stream_Size( p_input->s ) / 100,
+ p_avi->i_movi_begin );
+ /* search first selected stream */
+ for( i_stream = 0, p_stream = NULL;
+ i_stream < p_avi->i_streams; i_stream++ )
+ {
+ p_stream = p_avi->pp_info[i_stream];
+ if( p_stream->b_activated )
{
- i_microsecperframe = (mtime_t)1000000 *
- (mtime_t)p_info->i_scale /
- (mtime_t)p_info->i_rate;
+ break;
}
-#endif
- break;
- default:
- msg_Warn( p_input, "stream[%d] unknown type", i );
- p_info->i_cat = UNKNOWN_ES;
- i_init_size = 0;
- p_init_data = NULL;
+ }
+ if( !p_stream || !p_stream->b_activated )
+ {
+ msg_Warn( p_input, "cannot find any selected stream" );
+ return( -1 );
+ }
+
+ /* be sure that the index exist */
+ if( AVI_StreamChunkSet( p_input,
+ i_stream,
+ 0 ) )
+ {
+ msg_Warn( p_input, "cannot seek" );
+ return( -1 );
+ }
+
+ while( i_pos >= p_stream->p_index[p_stream->i_idxposc].i_pos +
+ p_stream->p_index[p_stream->i_idxposc].i_length + 8 )
+ {
+ /* search after i_idxposc */
+ if( AVI_StreamChunkSet( p_input,
+ i_stream, p_stream->i_idxposc + 1 ) )
{
- char psz_cat[32]; /* We'll clip i just in case */
- input_info_category_t *p_cat;
- sprintf( psz_cat, "stream %d", __MIN( i, 100000 ) );
- p_cat = input_InfoCategory( p_input, psz_cat );
- input_AddInfo( p_cat, "Type", "unknown" );
+ msg_Warn( p_input, "cannot seek" );
+ return( -1 );
}
- break;
+ }
+ i_date = AVI_GetPTS( p_stream );
+ /* TODO better support for i_samplesize != 0 */
+ msg_Dbg( p_input, "estimate date "I64Fd, i_date );
}
- p_info->b_activated = VLC_FALSE;
- /* add one ES */
- vlc_mutex_lock( &p_input->stream.stream_lock );
- p_info->p_es =
- p_es = input_AddES( p_input,
- p_input->stream.p_selected_program, 1+i,
- 0 );
- vlc_mutex_unlock( &p_input->stream.stream_lock );
- p_es->i_stream_id =i; /* XXX: i don't use it */
- p_es->i_fourcc = p_info->i_fourcc;
- p_es->i_cat = p_info->i_cat;
- if( p_es->i_cat == AUDIO_ES )
+
+#define p_stream p_avi->pp_info[i_stream]
+ p_avi->i_time = 0;
+ /* seek for chunk based streams */
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
{
- p_es->p_waveformatex = malloc( i_init_size );
- memcpy( p_es->p_waveformatex, p_init_data, i_init_size );
+ if( p_stream->b_activated && !p_stream->i_samplesize )
+// if( p_stream->b_activated )
+ {
+ AVI_StreamSeek( p_input, i_stream, i_date );
+ p_avi->i_time = __MAX( AVI_GetPTS( p_stream ),
+ p_avi->i_time );
+ }
}
- else if( p_es->i_cat == VIDEO_ES )
+#if 1
+ if( p_avi->i_time )
{
- p_es->p_bitmapinfoheader = malloc( i_init_size );
- memcpy( p_es->p_bitmapinfoheader, p_init_data, i_init_size );
+ i_date = p_avi->i_time;
}
-#undef p_info
- }
-
-#ifdef __AVI_SUBTITLE__
- if( ( p_avi->p_sub = subtitle_New( p_input, NULL, i_microsecperframe ) ) )
- {
- subtitle_Select( p_avi->p_sub );
- }
-#endif
-
- if( config_GetInt( p_input, "avi-index" ) )
- {
- if( p_avi->b_seekable )
+ /* seek for bytes based streams */
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
{
- AVI_IndexCreate( p_input );
+ if( p_stream->b_activated && p_stream->i_samplesize )
+ {
+ AVI_StreamSeek( p_input, i_stream, i_date );
+// p_avi->i_time = __MAX( AVI_GetPTS( p_stream ), p_avi->i_time );
+ }
}
- else
+ msg_Dbg( p_input, "seek: "I64Fd" secondes", p_avi->i_time /1000000 );
+ /* set true movie time */
+#endif
+ if( !p_avi->i_time )
{
- msg_Warn( p_input, "cannot create index (unseekable stream)" );
- AVI_IndexLoad( p_input );
+ p_avi->i_time = i_date;
}
+#undef p_stream
+ return( 1 );
}
else
{
- AVI_IndexLoad( p_input );
+ msg_Err( p_input, "shouldn't yet be executed" );
+ return( -1 );
}
+}
- /* *** movie length in sec *** */
- p_avi->i_length = AVI_MovieGetLength( p_input, p_avi );
- if( p_avi->i_length < (mtime_t)p_avih->i_totalframes *
- (mtime_t)p_avih->i_microsecperframe /
- (mtime_t)1000000 )
- {
- msg_Warn( p_input, "broken or missing index, 'seek' will be axproximative or will have strange behavour" );
- }
+/*****************************************************************************
+ * Control:
+ *****************************************************************************
+ *
+ *****************************************************************************/
+static int Control( input_thread_t *p_input, int i_query, va_list args )
+{
+ demux_sys_t *p_sys = p_input->p_demux_data;
+ int i;
+ double f, *pf;
+ int64_t i64, *pi64;
- vlc_mutex_lock( &p_input->stream.stream_lock );
- if( p_avi->i_length )
- {
- p_input->stream.i_mux_rate =
- p_input->stream.p_selected_area->i_size / 50 / p_avi->i_length;
- }
- else
+ switch( i_query )
{
- p_input->stream.i_mux_rate = 0;
- }
- vlc_mutex_unlock( &p_input->stream.stream_lock );
+ case DEMUX_GET_POSITION:
+ pf = (double*)va_arg( args, double * );
+ if( p_sys->i_length > 0 )
+ {
+ *pf = (double)p_sys->i_time / (double)( p_sys->i_length * (mtime_t)1000000 );
+ return VLC_SUCCESS;
+ }
+ else if( stream_Size( p_input->s ) > 0 )
+ {
+ unsigned int i;
+ int64_t i_tmp;
- b_stream_audio = VLC_FALSE;
- b_stream_video = VLC_FALSE;
+ i64 = 0;
+ /* search the more advanced selected es */
+ for( i = 0; i < p_sys->i_streams; i++ )
+ {
+#define tk p_sys->pp_info[i]
+ if( tk->b_activated && tk->i_idxposc < tk->i_idxnb )
+ {
+ i_tmp = tk->p_index[tk->i_idxposc].i_pos +
+ tk->p_index[tk->i_idxposc].i_length + 8;
+ if( i_tmp > i64 )
+ {
+ i64 = i_tmp;
+ }
+ }
+#undef tk
+ }
+ *pf = (double)i64 / (double)stream_Size( p_input->s );
+ return VLC_SUCCESS;
+ }
+ else
+ {
+ *pf = 0.0;
+ return VLC_SUCCESS;
+ }
+ case DEMUX_SET_POSITION:
+ if( p_sys->b_seekable )
+ {
+ int i_ret;
- for( i = 0; i < p_avi->i_streams; i++ )
- {
-#define p_info p_avi->pp_info[i]
- switch( p_info->p_es->i_cat )
- {
- case( VIDEO_ES ):
+ f = (double)va_arg( args, double );
+ i64 = (mtime_t)(1000000.0 * p_sys->i_length * f );
+ i_ret = Seek( p_input, i64, (int)(f * 100) );
+ return i_ret;
+ }
+ else
+ {
+ return demux_vaControlDefault( p_input, i_query, args );
+ }
+ case DEMUX_GET_TIME:
+ pi64 = (int64_t*)va_arg( args, int64_t * );
+ *pi64 = p_sys->i_time;
+ return VLC_SUCCESS;
- if( !b_stream_video )
- {
- b_stream_video = AVI_StreamStart( p_input, p_avi, i );
- }
- break;
+ case DEMUX_SET_TIME:
+ msg_Err( p_input, "FIXME DEMUX_SET_TIME to be implemented" );
+ return VLC_EGENERIC;
+ /* return demux_vaControlDefault( p_input, i_query, args ); */
+
+ case DEMUX_GET_LENGTH:
+ pi64 = (int64_t*)va_arg( args, int64_t * );
+ *pi64 = p_sys->i_length * (mtime_t)1000000;
+ return VLC_SUCCESS;
- case( AUDIO_ES ):
- if( !b_stream_audio )
+ case DEMUX_GET_FPS:
+ pf = (double*)va_arg( args, double * );
+ *pf = 0.0;
+ for( i = 0; i < (int)p_sys->i_streams; i++ )
+ {
+#define tk p_sys->pp_info[i]
+ if( tk->i_cat == VIDEO_ES && tk->i_scale > 0)
{
- b_stream_audio = AVI_StreamStart( p_input, p_avi, i );
+ *pf = (float)tk->i_rate / (float)tk->i_scale;
+ break;
}
- break;
- default:
- break;
- }
-#undef p_info
- }
-
- if( !b_stream_video )
- {
- msg_Warn( p_input, "no video stream found" );
- }
- if( !b_stream_audio )
- {
- msg_Warn( p_input, "no audio stream found!" );
- }
-
- vlc_mutex_lock( &p_input->stream.stream_lock );
- p_input->stream.p_selected_program->b_is_ok = 1;
- vlc_mutex_unlock( &p_input->stream.stream_lock );
+#undef tk
+ }
+ return VLC_SUCCESS;
- if( p_avi->b_seekable )
- {
- AVI_ChunkGoto( p_input, p_movi );
- }
- else
- {
- // already at begining of p_movi
+ default:
+ return demux_vaControlDefault( p_input, i_query, args );
}
- AVI_SkipBytes( p_input, 12 ); // enter in p_movi
-
- p_avi->i_movi_begin = p_movi->i_chunk_pos;
- return VLC_SUCCESS;
+ return VLC_EGENERIC;
}
-
-
-
/*****************************************************************************
* Function to convert pts to chunk or byte
*****************************************************************************/
-static inline mtime_t AVI_PTSToChunk( avi_stream_t *p_info,
- mtime_t i_pts )
+static mtime_t AVI_PTSToChunk( avi_stream_t *tk, mtime_t i_pts )
{
return (mtime_t)((int64_t)i_pts *
- (int64_t)p_info->i_rate /
- (int64_t)p_info->i_scale /
+ (int64_t)tk->i_rate /
+ (int64_t)tk->i_scale /
(int64_t)1000000 );
}
-static inline mtime_t AVI_PTSToByte( avi_stream_t *p_info,
- mtime_t i_pts )
+static mtime_t AVI_PTSToByte( avi_stream_t *tk, mtime_t i_pts )
{
return (mtime_t)((int64_t)i_pts *
- (int64_t)p_info->i_rate /
- (int64_t)p_info->i_scale /
+ (int64_t)tk->i_rate /
+ (int64_t)tk->i_scale /
(int64_t)1000000 *
- (int64_t)p_info->i_samplesize );
+ (int64_t)tk->i_samplesize );
}
-static mtime_t AVI_GetDPTS( avi_stream_t *p_stream, int i_count )
+static mtime_t AVI_GetDPTS( avi_stream_t *tk, int64_t i_count )
{
- if( p_stream->i_samplesize )
- {
- return (mtime_t)( (int64_t)1000000 *
- (int64_t)i_count *
- (int64_t)p_stream->i_scale /
- (int64_t)p_stream->i_rate /
- (int64_t)p_stream->i_samplesize );
- }
- else
+ mtime_t i_dpts;
+
+ i_dpts = (mtime_t)( (int64_t)1000000 *
+ (int64_t)i_count *
+ (int64_t)tk->i_scale /
+ (int64_t)tk->i_rate );
+
+ if( tk->i_samplesize )
{
- return (mtime_t)( (int64_t)1000000 *
- (int64_t)i_count *
- (int64_t)p_stream->i_scale /
- (int64_t)p_stream->i_rate);
+ return i_dpts / tk->i_samplesize;
}
-
+ return i_dpts;
}
-static mtime_t AVI_GetPTS( avi_stream_t *p_info )
+static mtime_t AVI_GetPTS( avi_stream_t *tk )
{
-
- if( p_info->i_samplesize )
+ if( tk->i_samplesize )
{
+ int64_t i_count = 0;
+
/* we need a valid entry we will emulate one */
- int64_t i_len;
- if( p_info->i_idxposc == p_info->i_idxnb )
+ if( tk->i_idxposc == tk->i_idxnb )
{
- if( p_info->i_idxposc )
+ if( tk->i_idxposc )
{
/* use the last entry */
- i_len = p_info->p_index[p_info->i_idxnb - 1].i_lengthtotal
- + p_info->p_index[p_info->i_idxnb - 1].i_length
- + p_info->i_idxposb; /* should be 0 */
- }
- else
- {
- i_len = p_info->i_idxposb;
- /* no valid entry use only offset*/
+ i_count = tk->p_index[tk->i_idxnb - 1].i_lengthtotal
+ + tk->p_index[tk->i_idxnb - 1].i_length;
}
}
else
{
- i_len = p_info->p_index[p_info->i_idxposc].i_lengthtotal
- + p_info->i_idxposb;
+ i_count = tk->p_index[tk->i_idxposc].i_lengthtotal;
}
- return (mtime_t)( (int64_t)1000000 *
- (int64_t)i_len *
- (int64_t)p_info->i_scale /
- (int64_t)p_info->i_rate /
- (int64_t)p_info->i_samplesize );
+ return AVI_GetDPTS( tk, i_count + tk->i_idxposb );
}
else
{
- /* even if p_info->i_idxposc isn't valid, there isn't any problem */
- return (mtime_t)( (int64_t)1000000 *
- (int64_t)(p_info->i_idxposc ) *
- (int64_t)p_info->i_scale /
- (int64_t)p_info->i_rate);
+ return AVI_GetDPTS( tk, tk->i_idxposc );
+ }
+}
+
+#if 0
+static void AVI_FixPTS( avi_stream_t *p_stream, pes_packet_t *p_pes )
+{
+ data_packet_t *p_data;
+ uint8_t *p;
+ int i_pos = 0;
+
+ switch( p_stream->i_fourcc )
+ {
+ case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
+ p_data = p_pes->p_first;
+ while( p_data )
+ {
+ p = p_data->p_payload_start;
+ while( p < p_data->p_payload_end - 2 )
+ {
+ if( p[0] == 0xff && ( p[1]&0xe0) == 0xe0 )
+ {
+ mtime_t i_diff = AVI_GetDPTS( p_stream, i_pos );
+ p_pes->i_dts += i_diff;
+ p_pes->i_pts += i_diff;
+ return;
+ }
+ p++; i_pos++;
+ }
+ p_data = p_data->p_next;
+ }
+ return;
+ case VLC_FOURCC( 'a', '5', '2', ' ' ):
+ p_data = p_pes->p_first;
+ while( p_data )
+ {
+ p = p_data->p_payload_start;
+ while( p < p_data->p_payload_end - 2 )
+ {
+ if( p[0] == 0x0b && p[1] == 0x77 )
+ {
+ mtime_t i_diff = AVI_GetDPTS( p_stream, i_pos );
+ p_pes->i_dts += i_diff;
+ p_pes->i_pts += i_diff;
+ }
+ p++; i_pos++;
+ }
+ p_data = p_data->p_next;
+ }
+ return;
+ default:
+ /* we can't fix :( */
+ return;
}
}
+#endif
static int AVI_StreamChunkFind( input_thread_t *p_input,
unsigned int i_stream )
/* find first chunk of i_stream that isn't in index */
- if( p_avi->i_movi_lastchunk_pos >= p_avi->i_movi_begin )
+ if( p_avi->i_movi_lastchunk_pos >= p_avi->i_movi_begin + 12 )
{
- AVI_SeekAbsolute( p_input, p_avi->i_movi_lastchunk_pos );
+ stream_Seek( p_input->s, p_avi->i_movi_lastchunk_pos );
if( AVI_PacketNext( p_input ) )
{
return VLC_EGENERIC;
}
else
{
- AVI_SeekAbsolute( p_input, p_avi->i_movi_begin );
+ stream_Seek( p_input->s, p_avi->i_movi_begin + 12 );
}
for( ;; )
if( avi_pk.i_stream >= p_avi->i_streams ||
( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
{
- switch( avi_pk.i_fourcc )
+ if( AVI_PacketNext( p_input ) )
{
- case AVIFOURCC_LIST:
- AVI_SkipBytes( p_input, 12 );
- break;
- default:
- if( AVI_PacketNext( p_input ) )
- {
- return VLC_EGENERIC;
- }
- break;
+ return VLC_EGENERIC;
}
}
else
/* be sure that i_ck will be a valid index entry */
-static int AVI_SetStreamChunk( input_thread_t *p_input,
+static int AVI_StreamChunkSet( input_thread_t *p_input,
unsigned int i_stream,
unsigned int i_ck )
{
/* XXX FIXME up to now, we assume that all chunk are one after one */
-static int AVI_SetStreamBytes( input_thread_t *p_input,
+static int AVI_StreamBytesSet( input_thread_t *p_input,
unsigned int i_stream,
off_t i_byte )
{
}
static int AVI_StreamSeek( input_thread_t *p_input,
- demux_sys_t *p_avi,
int i_stream,
mtime_t i_date )
{
+ demux_sys_t *p_avi = p_input->p_demux_data;
#define p_stream p_avi->pp_info[i_stream]
mtime_t i_oldpts;
if( !p_stream->i_samplesize )
{
- if( AVI_SetStreamChunk( p_input,
+ if( AVI_StreamChunkSet( p_input,
i_stream,
AVI_PTSToChunk( p_stream, i_date ) ) )
{
return VLC_EGENERIC;
}
- /* search key frame */
msg_Dbg( p_input,
"old:"I64Fd" %s new "I64Fd,
i_oldpts,
i_oldpts > i_date ? ">" : "<",
i_date );
- if( i_date < i_oldpts )
+ if( p_stream->i_cat == VIDEO_ES )
{
- while( p_stream->i_idxposc > 0 &&
- !( p_stream->p_index[p_stream->i_idxposc].i_flags &
- AVIIF_KEYFRAME ) )
+ /* search key frame */
+ if( i_date < i_oldpts )
{
- if( AVI_SetStreamChunk( p_input,
- i_stream,
- p_stream->i_idxposc - 1 ) )
+ while( p_stream->i_idxposc > 0 &&
+ !( p_stream->p_index[p_stream->i_idxposc].i_flags &
+ AVIIF_KEYFRAME ) )
{
- return VLC_EGENERIC;
+ if( AVI_StreamChunkSet( p_input,
+ i_stream,
+ p_stream->i_idxposc - 1 ) )
+ {
+ return VLC_EGENERIC;
+ }
}
}
- }
- else
- {
- while( p_stream->i_idxposc < p_stream->i_idxnb &&
- !( p_stream->p_index[p_stream->i_idxposc].i_flags &
- AVIIF_KEYFRAME ) )
+ else
{
- if( AVI_SetStreamChunk( p_input,
- i_stream,
- p_stream->i_idxposc + 1 ) )
+ while( p_stream->i_idxposc < p_stream->i_idxnb &&
+ !( p_stream->p_index[p_stream->i_idxposc].i_flags &
+ AVIIF_KEYFRAME ) )
{
- return VLC_EGENERIC;
+ if( AVI_StreamChunkSet( p_input,
+ i_stream,
+ p_stream->i_idxposc + 1 ) )
+ {
+ return VLC_EGENERIC;
+ }
}
}
}
}
else
{
- if( AVI_SetStreamBytes( p_input,
+ if( AVI_StreamBytesSet( p_input,
i_stream,
AVI_PTSToByte( p_stream, i_date ) ) )
{
- return VLC_EGENERIC;
+ return VLC_EGENERIC;
+ }
+ }
+ return VLC_SUCCESS;
+#undef p_stream
+}
+
+/*****************************************************************************
+ * AVI_Interleaved: check weither a file is interleaved or not.
+ *****************************************************************************/
+static vlc_bool_t AVI_Interleaved( input_thread_t *p_input )
+{
+ demux_sys_t *p_sys = p_input->p_demux_data;
+ unsigned int i;
+ mtime_t i_time = 0;
+ vlc_bool_t b_ret = VLC_TRUE;
+
+ int64_t i_max;
+
+ if( stream_Size( p_input->s ) <= 100 )
+ {
+ return VLC_FALSE;
+ }
+
+ i_max = __MIN( 2000000, stream_Size( p_input->s ) / 100 );
+
+#define tk p_sys->pp_info[i]
+ while( i_time < p_sys->i_length * (mtime_t)1000000)
+ {
+ int64_t i_ref;
+
+ i_ref = -1;
+ i_time += 50000;
+ for( i = 0; i < p_sys->i_streams; i++ )
+ {
+ while( AVI_GetPTS( tk ) < i_time && tk->i_idxposc < tk->i_idxnb - 1 )
+ {
+ tk->i_idxposc++;
+ }
+
+ if( i_ref == -1 )
+ {
+ i_ref = tk->p_index[tk->i_idxposc].i_pos;
+ }
+ if( __ABS( tk->p_index[tk->i_idxposc].i_pos - i_ref ) > i_max ||
+ tk->p_index[tk->i_idxposc].i_length > i_max )
+ {
+ msg_Dbg( p_input, "interleaved=no because ref=%lld pos=%lld length=%d (max=%lld)",
+ i_ref, tk->p_index[tk->i_idxposc].i_pos, tk->p_index[tk->i_idxposc].i_length, i_max );
+ b_ret = VLC_FALSE;
+ goto exit;
+ }
+ }
+ }
+
+exit:
+ for( i = 0; i < p_sys->i_streams; i++ )
+ {
+ tk->i_idxposc = 0;
+ tk->i_idxposb = 0;
+ }
+#undef tk
+ return b_ret;
+}
+
+/****************************************************************************
+ * Return VLC_TRUE if it's a key frame
+ ****************************************************************************/
+static int AVI_GetKeyFlag( vlc_fourcc_t i_fourcc, uint8_t *p_byte )
+{
+ switch( i_fourcc )
+ {
+ case FOURCC_DIV1:
+ /* we have:
+ * startcode: 0x00000100 32bits
+ * framenumber ? 5bits
+ * piture type 0(I),1(P) 2bits
+ */
+ if( GetDWBE( p_byte ) != 0x00000100 )
+ {
+ /* it's not an msmpegv1 stream, strange...*/
+ return AVIIF_KEYFRAME;
+ }
+ else
+ {
+ return p_byte[4] & 0x06 ? 0 : AVIIF_KEYFRAME;
+ }
+ case FOURCC_DIV2:
+ case FOURCC_DIV3: // wmv1 also
+ /* we have
+ * picture type 0(I),1(P) 2bits
+ */
+ return p_byte[0] & 0xC0 ? 0 : AVIIF_KEYFRAME;
+ case FOURCC_mp4v:
+ /* we should find first occurence of 0x000001b6 (32bits)
+ * startcode: 0x000001b6 32bits
+ * piture type 0(I),1(P) 2bits
+ */
+ if( GetDWBE( p_byte ) != 0x000001b6 )
+ {
+ /* not true , need to find the first VOP header */
+ return AVIIF_KEYFRAME;
+ }
+ else
+ {
+ return p_byte[4] & 0xC0 ? 0 : AVIIF_KEYFRAME;
+ }
+ default:
+ /* I can't do it, so say yes */
+ return AVIIF_KEYFRAME;
+ }
+}
+
+vlc_fourcc_t AVI_FourccGetCodec( unsigned int i_cat, vlc_fourcc_t i_codec )
+{
+ switch( i_cat )
+ {
+ case AUDIO_ES:
+ wf_tag_to_fourcc( i_codec, &i_codec, NULL );
+ return i_codec;
+
+ case VIDEO_ES:
+ // XXX DIV1 <- msmpeg4v1, DIV2 <- msmpeg4v2, DIV3 <- msmpeg4v3, mp4v for mpeg4
+ switch( i_codec )
+ {
+ case FOURCC_DIV1:
+ case FOURCC_div1:
+ case FOURCC_MPG4:
+ case FOURCC_mpg4:
+ return FOURCC_DIV1;
+ case FOURCC_DIV2:
+ case FOURCC_div2:
+ case FOURCC_MP42:
+ case FOURCC_mp42:
+ case FOURCC_MPG3:
+ case FOURCC_mpg3:
+ return FOURCC_DIV2;
+ case FOURCC_div3:
+ case FOURCC_MP43:
+ case FOURCC_mp43:
+ case FOURCC_DIV3:
+ case FOURCC_DIV4:
+ case FOURCC_div4:
+ case FOURCC_DIV5:
+ case FOURCC_div5:
+ case FOURCC_DIV6:
+ case FOURCC_div6:
+ case FOURCC_AP41:
+ case FOURCC_3IV1:
+ case FOURCC_3iv1:
+ case FOURCC_3IVD:
+ case FOURCC_3ivd:
+ case FOURCC_3VID:
+ case FOURCC_3vid:
+ return FOURCC_DIV3;
+ case FOURCC_DIVX:
+ case FOURCC_divx:
+ case FOURCC_MP4S:
+ case FOURCC_mp4s:
+ case FOURCC_M4S2:
+ case FOURCC_m4s2:
+ case FOURCC_xvid:
+ case FOURCC_XVID:
+ case FOURCC_XviD:
+ case FOURCC_DX50:
+ case FOURCC_mp4v:
+ case FOURCC_4:
+ case FOURCC_3IV2:
+ case FOURCC_3iv2:
+ return FOURCC_mp4v;
+ }
+ default:
+ return VLC_FOURCC( 'u', 'n', 'd', 'f' );
+ }
+}
+
+/****************************************************************************
+ *
+ ****************************************************************************/
+static void AVI_ParseStreamHeader( vlc_fourcc_t i_id,
+ int *pi_number, int *pi_type )
+{
+#define SET_PTR( p, v ) if( p ) *(p) = (v);
+ int c1, c2;
+
+ c1 = ((uint8_t *)&i_id)[0];
+ c2 = ((uint8_t *)&i_id)[1];
+
+ if( c1 < '0' || c1 > '9' || c2 < '0' || c2 > '9' )
+ {
+ SET_PTR( pi_number, 100 ); /* > max stream number */
+ SET_PTR( pi_type, UNKNOWN_ES );
+ }
+ else
+ {
+ SET_PTR( pi_number, (c1 - '0') * 10 + (c2 - '0' ) );
+ switch( VLC_TWOCC( ((uint8_t *)&i_id)[2], ((uint8_t *)&i_id)[3] ) )
+ {
+ case AVITWOCC_wb:
+ SET_PTR( pi_type, AUDIO_ES );
+ break;
+ case AVITWOCC_dc:
+ case AVITWOCC_db:
+ SET_PTR( pi_type, VIDEO_ES );
+ break;
+ default:
+ SET_PTR( pi_type, UNKNOWN_ES );
+ break;
}
}
- return VLC_SUCCESS;
-#undef p_stream
+#undef SET_PTR
}
-/*****************************************************************************
- * AVISeek: goto to i_date or i_percent
- *****************************************************************************
- * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
- *****************************************************************************/
-static int AVISeek ( input_thread_t *p_input,
- mtime_t i_date, int i_percent )
+/****************************************************************************
+ *
+ ****************************************************************************/
+static int AVI_PacketGetHeader( input_thread_t *p_input, avi_packet_t *p_pk )
{
+ uint8_t *p_peek;
- demux_sys_t *p_avi = p_input->p_demux_data;
- unsigned int i_stream;
- msg_Dbg( p_input,
- "seek requested: "I64Fd" secondes %d%%",
- i_date / 1000000,
- i_percent );
-
- if( p_avi->b_seekable )
+ if( stream_Peek( p_input->s, &p_peek, 16 ) < 16 )
{
- if( !p_avi->i_length )
- {
- avi_stream_t *p_stream;
- uint64_t i_pos;
-
- /* use i_percent to create a true i_date */
- msg_Warn( p_input,
- "mmh, seeking without index at %d%%"
- " work only for interleaved file", i_percent );
+ return VLC_EGENERIC;
+ }
+ p_pk->i_fourcc = VLC_FOURCC( p_peek[0], p_peek[1], p_peek[2], p_peek[3] );
+ p_pk->i_size = GetDWLE( p_peek + 4 );
+ p_pk->i_pos = stream_Tell( p_input->s );
+ if( p_pk->i_fourcc == AVIFOURCC_LIST || p_pk->i_fourcc == AVIFOURCC_RIFF )
+ {
+ p_pk->i_type = VLC_FOURCC( p_peek[8], p_peek[9],
+ p_peek[10], p_peek[11] );
+ }
+ else
+ {
+ p_pk->i_type = 0;
+ }
- if( i_percent >= 100 )
- {
- msg_Warn( p_input, "cannot seek so far !" );
- return( -1 );
- }
- i_percent = __MAX( i_percent, 0 );
+ memcpy( p_pk->i_peek, p_peek + 8, 8 );
- /* try to find chunk that is at i_percent or the file */
- i_pos = __MAX( i_percent *
- p_input->stream.p_selected_area->i_size / 100,
- p_avi->i_movi_begin );
- /* search first selected stream */
- for( i_stream = 0, p_stream = NULL;
- i_stream < p_avi->i_streams; i_stream++ )
- {
- p_stream = p_avi->pp_info[i_stream];
- if( p_stream->b_activated )
- {
- break;
- }
- }
- if( !p_stream || !p_stream->b_activated )
- {
- msg_Warn( p_input, "cannot find any selected stream" );
- return( -1 );
- }
+ AVI_ParseStreamHeader( p_pk->i_fourcc, &p_pk->i_stream, &p_pk->i_cat );
+ return VLC_SUCCESS;
+}
- /* be sure that the index exit */
- if( AVI_SetStreamChunk( p_input,
- i_stream,
- 0 ) )
- {
- msg_Warn( p_input, "cannot seek" );
- return( -1 );
- }
+static int AVI_PacketNext( input_thread_t *p_input )
+{
+ avi_packet_t avi_ck;
+ int i_skip = 0;
- while( i_pos >= p_stream->p_index[p_stream->i_idxposc].i_pos +
- p_stream->p_index[p_stream->i_idxposc].i_length + 8 )
- {
- /* search after i_idxposc */
- if( AVI_SetStreamChunk( p_input,
- i_stream, p_stream->i_idxposc + 1 ) )
- {
- msg_Warn( p_input, "cannot seek" );
- return( -1 );
- }
- }
- i_date = AVI_GetPTS( p_stream );
- /* TODO better support for i_samplesize != 0 */
- msg_Dbg( p_input, "estimate date "I64Fd, i_date );
- }
+ if( AVI_PacketGetHeader( p_input, &avi_ck ) )
+ {
+ return VLC_EGENERIC;
+ }
-#define p_stream p_avi->pp_info[i_stream]
- p_avi->i_time = 0;
- /* seek for chunk based streams */
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
- {
- if( p_stream->b_activated && !p_stream->i_samplesize )
-// if( p_stream->b_activated )
- {
- AVI_StreamSeek( p_input, p_avi, i_stream, i_date );
- p_avi->i_time = __MAX( AVI_GetPTS( p_stream ),
- p_avi->i_time );
- }
- }
-#if 1
- if( p_avi->i_time )
- {
- i_date = p_avi->i_time;
- }
- /* seek for bytes based streams */
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
- {
- if( p_stream->b_activated && p_stream->i_samplesize )
- {
- AVI_StreamSeek( p_input, p_avi, i_stream, i_date );
-// p_avi->i_time = __MAX( AVI_GetPTS( p_stream ), p_avi->i_time );
- }
- }
- msg_Dbg( p_input, "seek: "I64Fd" secondes", p_avi->i_time /1000000 );
- /* set true movie time */
-#endif
- if( !p_avi->i_time )
- {
- p_avi->i_time = i_date;
- }
-#undef p_stream
- return( 1 );
+ if( avi_ck.i_fourcc == AVIFOURCC_LIST &&
+ ( avi_ck.i_type == AVIFOURCC_rec || avi_ck.i_type == AVIFOURCC_movi ) )
+ {
+ i_skip = 12;
+ }
+ else if( avi_ck.i_fourcc == AVIFOURCC_RIFF &&
+ avi_ck.i_type == AVIFOURCC_AVIX )
+ {
+ i_skip = 24;
}
else
{
- msg_Err( p_input, "shouldn't yet be executed" );
- return( -1 );
+ i_skip = __EVEN( avi_ck.i_size ) + 8;
}
-}
-#if 0
-static pes_packet_t *PES_split( input_thread_t *p_input, avi_stream_t *p_stream, pes_packet_t *p_pes )
-{
- pes_packet_t *p_pes2;
- data_packet_t *p_data;
- int i_nb_data;
-
- if( p_pes->i_nb_data < 2 )
- {
- return( NULL );
- }
- p_pes2 = input_NewPES( p_input->p_method_data );
- p_pes2->i_pts = p_pes->i_pts;
- p_pes2->i_dts = p_pes->i_dts;
- p_pes2->i_nb_data = p_pes->i_nb_data/2;
- p_pes2->i_pes_size = 0;
- for( i_nb_data = 0, p_data = p_pes->p_first;
- i_nb_data < p_pes2->i_nb_data;
- i_nb_data++, p_data = p_data->p_next )
- {
- p_pes2->i_pes_size +=
- p_data->p_payload_end - p_data->p_payload_start;
- p_pes2->p_last = p_data;
- }
- p_pes2->p_first = p_pes->p_first;
- p_pes2->p_last->p_next = NULL;
-
- p_pes->p_first = p_data;
- p_pes->i_pes_size -= p_pes2->i_pes_size;
- p_pes->i_nb_data -= p_pes2->i_nb_data;
-// p_pes->i_pts += AVI_GetDPTS( p_stream, p_pes2->i_pes_size );
-// p_pes->i_dts += AVI_GetDPTS( p_stream, p_pes2->i_pes_size );
- p_pes->i_pts = 0;
- p_pes->i_dts = 0;
- return( p_pes2 );
+ if( stream_Read( p_input->s, NULL, i_skip ) != i_skip )
+ {
+ return VLC_EGENERIC;
+ }
+ return VLC_SUCCESS;
}
-#endif
-
-/*****************************************************************************
- * AVIDemux_Seekable: reads and demuxes data packets for stream seekable
- *****************************************************************************
- * AVIDemux: reads and demuxes data packets
- *****************************************************************************
- * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
- *****************************************************************************/
-typedef struct avi_stream_toread_s
+static int AVI_PacketRead( input_thread_t *p_input,
+ avi_packet_t *p_pk,
+ pes_packet_t **pp_pes )
{
- vlc_bool_t b_ok;
+ size_t i_size;
- int i_toread;
+ i_size = __EVEN( p_pk->i_size + 8 );
- off_t i_posf; // where we will read :
- // if i_idxposb == 0 : begining of chunk (+8 to acces data)
- // else : point on data directly
-} avi_stream_toread_t;
+ if( ( *pp_pes = stream_PesPacket( p_input->s, i_size ) ) == NULL )
+ {
+ return VLC_EGENERIC;
+ }
+ (*pp_pes)->p_first->p_payload_start += 8;
+ (*pp_pes)->i_pes_size -= 8;
-static int AVIDemux_Seekable( input_thread_t *p_input )
-{
- unsigned int i_stream_count;
- unsigned int i_stream;
- vlc_bool_t b_stream;
+ if( i_size != p_pk->i_size + 8 )
+ {
+ (*pp_pes)->p_last->p_payload_end--;
+ (*pp_pes)->i_pes_size--;
+ }
- // cannot be more than 100 stream (dcXX or wbXX)
- avi_stream_toread_t toread[100];
+ return VLC_SUCCESS;
+}
- demux_sys_t *p_avi = p_input->p_demux_data;
+static int AVI_PacketSearch( input_thread_t *p_input )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
- /* detect new selected/unselected streams */
- for( i_stream = 0,i_stream_count= 0;
- i_stream < p_avi->i_streams; i_stream++ )
+ avi_packet_t avi_pk;
+ for( ;; )
{
-#define p_stream p_avi->pp_info[i_stream]
- if( p_stream->p_es )
+ if( stream_Read( p_input->s, NULL, 1 ) != 1 )
{
- if( p_stream->p_es->p_decoder_fifo &&
- !p_stream->b_activated )
- {
- AVI_StreamStart( p_input, p_avi, i_stream );
- }
- else
- if( !p_stream->p_es->p_decoder_fifo &&
- p_stream->b_activated )
- {
- AVI_StreamStop( p_input, p_avi, i_stream );
- }
+ return VLC_EGENERIC;
}
- if( p_stream->b_activated )
+ AVI_PacketGetHeader( p_input, &avi_pk );
+ if( avi_pk.i_stream < p_avi->i_streams &&
+ ( avi_pk.i_cat == AUDIO_ES || avi_pk.i_cat == VIDEO_ES ) )
{
- i_stream_count++;
+ return VLC_SUCCESS;
+ }
+ switch( avi_pk.i_fourcc )
+ {
+ case AVIFOURCC_JUNK:
+ case AVIFOURCC_LIST:
+ case AVIFOURCC_RIFF:
+ case AVIFOURCC_idx1:
+ return VLC_SUCCESS;
}
-#undef p_stream
}
+}
- if( i_stream_count <= 0 )
+/****************************************************************************
+ * Index stuff.
+ ****************************************************************************/
+static void AVI_IndexAddEntry( demux_sys_t *p_avi,
+ int i_stream,
+ AVIIndexEntry_t *p_index)
+{
+ avi_stream_t *tk = p_avi->pp_info[i_stream];
+
+ /* Update i_movi_lastchunk_pos */
+ if( p_avi->i_movi_lastchunk_pos < p_index->i_pos )
{
- msg_Warn( p_input, "no track selected, exiting..." );
- return( 0 );
+ p_avi->i_movi_lastchunk_pos = p_index->i_pos;
}
- if( p_input->stream.p_selected_program->i_synchro_state == SYNCHRO_REINIT )
+ /* add the entry */
+ if( tk->i_idxnb >= tk->i_idxmax )
{
- mtime_t i_date;
- int i_percent;
- /* first wait for empty buffer, arbitrary time FIXME */
-// msleep( DEFAULT_PTS_DELAY );
-
- i_date = (mtime_t)1000000 *
- (mtime_t)p_avi->i_length *
- (mtime_t)AVI_TellAbsolute( p_input ) /
- (mtime_t)p_input->stream.p_selected_area->i_size;
- i_percent = 100 * AVI_TellAbsolute( p_input ) /
- p_input->stream.p_selected_area->i_size;
-
-// input_ClockInit( p_input->stream.p_selected_program );
- AVISeek( p_input, i_date, i_percent);
-
-#ifdef __AVI_SUBTITLE__
- if( p_avi->p_sub )
+ tk->i_idxmax += 16384;
+ tk->p_index = realloc( tk->p_index,
+ tk->i_idxmax * sizeof( AVIIndexEntry_t ) );
+ if( tk->p_index == NULL )
{
- subtitle_Seek( p_avi->p_sub, p_avi->i_time );
+ return;
}
-#endif
+ }
+ /* calculate cumulate length */
+ if( tk->i_idxnb > 0 )
+ {
+ p_index->i_lengthtotal =
+ tk->p_index[tk->i_idxnb - 1].i_length +
+ tk->p_index[tk->i_idxnb - 1].i_lengthtotal;
+ }
+ else
+ {
+ p_index->i_lengthtotal = 0;
}
+ tk->p_index[tk->i_idxnb++] = *p_index;
+}
- /* wait for the good time */
-
- p_avi->i_pcr = p_avi->i_time * 9 / 100;
+static int AVI_IndexLoad_idx1( input_thread_t *p_input )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
- input_ClockManageRef( p_input,
- p_input->stream.p_selected_program,
- p_avi->i_pcr );
+ avi_chunk_list_t *p_riff;
+ avi_chunk_list_t *p_movi;
+ avi_chunk_idx1_t *p_idx1;
+ unsigned int i_stream;
+ unsigned int i_index;
+ off_t i_offset;
+ unsigned int i;
- p_avi->i_time += 25*1000; /* read 25ms */
+ p_riff = AVI_ChunkFind( &p_avi->ck_root, AVIFOURCC_RIFF, 0);
+ p_idx1 = AVI_ChunkFind( p_riff, AVIFOURCC_idx1, 0);
+ p_movi = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
-#ifdef __AVI_SUBTITLE__
- if( p_avi->p_sub )
+ if( !p_idx1 )
{
- subtitle_Demux( p_avi->p_sub, p_avi->i_time );
+ msg_Warn( p_input, "cannot find idx1 chunk, no index defined" );
+ return VLC_EGENERIC;
}
-#endif
- /* init toread */
- for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
- {
-#define p_stream p_avi->pp_info[i_stream]
- mtime_t i_dpts;
- toread[i_stream].b_ok = p_stream->b_activated;
-
- if( p_stream->i_idxposc < p_stream->i_idxnb )
- {
- toread[i_stream].i_posf =
- p_stream->p_index[p_stream->i_idxposc].i_pos;
- if( p_stream->i_idxposb > 0 )
- {
- toread[i_stream].i_posf += 8 + p_stream->i_idxposb;
- }
- }
- else
+ /* *** calculate offset *** */
+ /* Well, avi is __SHIT__ so test more than one entry
+ * (needed for some avi files) */
+ i_offset = 0;
+ for( i = 0; i < __MIN( p_idx1->i_entry_count, 10 ); i++ )
+ {
+ if( p_idx1->entry[i].i_pos < p_movi->i_chunk_pos )
{
- toread[i_stream].i_posf = -1;
+ i_offset = p_movi->i_chunk_pos + 8;
+ break;
}
+ }
- i_dpts = p_avi->i_time - AVI_GetPTS( p_stream );
-
- if( p_stream->i_samplesize )
- {
- toread[i_stream].i_toread = AVI_PTSToByte( p_stream,
- __ABS( i_dpts ) );
- }
- else
- {
- toread[i_stream].i_toread = AVI_PTSToChunk( p_stream,
- __ABS( i_dpts ) );
- }
+ for( i_index = 0; i_index < p_idx1->i_entry_count; i_index++ )
+ {
+ unsigned int i_cat;
- if( i_dpts < 0 )
+ AVI_ParseStreamHeader( p_idx1->entry[i_index].i_fourcc,
+ &i_stream,
+ &i_cat );
+ if( i_stream < p_avi->i_streams &&
+ i_cat == p_avi->pp_info[i_stream]->i_cat )
{
- toread[i_stream].i_toread *= -1;
+ AVIIndexEntry_t index;
+ index.i_id = p_idx1->entry[i_index].i_fourcc;
+ index.i_flags =
+ p_idx1->entry[i_index].i_flags&(~AVIIF_FIXKEYFRAME);
+ index.i_pos = p_idx1->entry[i_index].i_pos + i_offset;
+ index.i_length = p_idx1->entry[i_index].i_length;
+ AVI_IndexAddEntry( p_avi, i_stream, &index );
}
-#undef p_stream
}
+ return VLC_SUCCESS;
+}
- b_stream = VLC_FALSE;
+static void __Parse_indx( input_thread_t *p_input,
+ int i_stream,
+ avi_chunk_indx_t *p_indx )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ AVIIndexEntry_t index;
+ int32_t i;
- for( ;; )
+ msg_Dbg( p_input, "loading subindex(0x%x) %d entries", p_indx->i_indextype, p_indx->i_entriesinuse );
+ if( p_indx->i_indexsubtype == 0 )
{
-#define p_stream p_avi->pp_info[i_stream]
- vlc_bool_t b_done;
- pes_packet_t *p_pes;
- off_t i_pos;
- unsigned int i;
- size_t i_size;
-
- /* search for first chunk to be read */
- for( i = 0, b_done = VLC_TRUE, i_pos = -1; i < p_avi->i_streams; i++ )
+ for( i = 0; i < p_indx->i_entriesinuse; i++ )
{
- if( !toread[i].b_ok ||
- AVI_GetDPTS( p_avi->pp_info[i],
- toread[i].i_toread ) <= -25 * 1000 )
- {
- continue;
- }
-
- if( toread[i].i_toread > 0 )
- {
- b_done = VLC_FALSE; // not yet finished
- }
+ index.i_id = p_indx->i_id;
+ index.i_flags = p_indx->idx.std[i].i_size & 0x80000000 ? 0 : AVIIF_KEYFRAME;
+ index.i_pos = p_indx->i_baseoffset + p_indx->idx.std[i].i_offset - 8;
+ index.i_length = p_indx->idx.std[i].i_size&0x7fffffff;
- if( toread[i].i_posf > 0 )
- {
- if( i_pos == -1 || i_pos > toread[i_stream].i_posf )
- {
- i_stream = i;
- i_pos = toread[i].i_posf;
- }
- }
+ AVI_IndexAddEntry( p_avi, i_stream, &index );
}
-
- if( b_done )
+ }
+ else if( p_indx->i_indexsubtype == AVI_INDEX_2FIELD )
+ {
+ for( i = 0; i < p_indx->i_entriesinuse; i++ )
{
-// return( b_stream ? 1 : 0 );
- return( 1 );
+ index.i_id = p_indx->i_id;
+ index.i_flags = p_indx->idx.field[i].i_size & 0x80000000 ? 0 : AVIIF_KEYFRAME;
+ index.i_pos = p_indx->i_baseoffset + p_indx->idx.field[i].i_offset - 8;
+ index.i_length = p_indx->idx.field[i].i_size;
+
+ AVI_IndexAddEntry( p_avi, i_stream, &index );
}
+ }
+ else
+ {
+ msg_Warn( p_input, "unknow subtype index(0x%x)", p_indx->i_indexsubtype );
+ }
+}
- if( i_pos == -1 )
- {
- /* no valid index, we will parse directly the stream
- * in case we fail we will disable all finished stream */
- if( p_avi->i_movi_lastchunk_pos >= p_avi->i_movi_begin )
- {
- AVI_SeekAbsolute( p_input, p_avi->i_movi_lastchunk_pos );
- if( AVI_PacketNext( p_input ) )
- {
- return( AVI_StreamStopFinishedStreams( p_input, p_avi ) ? 0 : 1 );
- }
- }
- else
- {
- AVI_SeekAbsolute( p_input, p_avi->i_movi_begin );
- }
+static void AVI_IndexLoad_indx( input_thread_t *p_input )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ unsigned int i_stream;
+ int32_t i;
- for( ;; )
- {
- avi_packet_t avi_pk;
+ avi_chunk_list_t *p_riff;
+ avi_chunk_list_t *p_hdrl;
- if( AVI_PacketGetHeader( p_input, &avi_pk ) )
- {
- msg_Warn( p_input,
- "cannot get packet header, track disabled" );
- return( AVI_StreamStopFinishedStreams( p_input, p_avi ) ? 0 : 1 );
- }
- if( avi_pk.i_stream >= p_avi->i_streams ||
- ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
- {
- switch( avi_pk.i_fourcc )
- {
- case AVIFOURCC_LIST:
- AVI_SkipBytes( p_input, 12 );
- break;
- default:
- if( AVI_PacketNext( p_input ) )
- {
- msg_Warn( p_input,
- "cannot skip packet, track disabled" );
- return( AVI_StreamStopFinishedStreams( p_input, p_avi ) ? 0 : 1 );
- }
- break;
- }
- continue;
- }
- else
- {
- /* add this chunk to the index */
- AVIIndexEntry_t index;
+ p_riff = AVI_ChunkFind( &p_avi->ck_root, AVIFOURCC_RIFF, 0);
+ p_hdrl = AVI_ChunkFind( p_riff, AVIFOURCC_hdrl, 0 );
- index.i_id = avi_pk.i_fourcc;
- index.i_flags =
- AVI_GetKeyFlag(p_avi->pp_info[avi_pk.i_stream]->i_codec,
- avi_pk.i_peek);
- index.i_pos = avi_pk.i_pos;
- index.i_length = avi_pk.i_size;
- AVI_IndexAddEntry( p_avi, avi_pk.i_stream, &index );
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
+ {
+ avi_chunk_list_t *p_strl;
+ avi_chunk_indx_t *p_indx;
- i_stream = avi_pk.i_stream;
- /* do we will read this data ? */
- if( AVI_GetDPTS( p_stream,
- toread[i_stream].i_toread ) > -25 * 1000 )
- {
- break;
- }
- else
- {
- if( AVI_PacketNext( p_input ) )
- {
- msg_Warn( p_input,
- "cannot skip packet, track disabled" );
- return( AVI_StreamStopFinishedStreams( p_input, p_avi ) ? 0 : 1 );
- }
- }
- }
- }
+#define p_stream p_avi->pp_info[i_stream]
+ p_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i_stream );
+ p_indx = AVI_ChunkFind( p_strl, AVIFOURCC_indx, 0 );
- }
- else
+ if( !p_indx )
{
- AVI_SeekAbsolute( p_input, i_pos );
+ msg_Warn( p_input, "cannot find indx (misdetect/broken OpenDML file?)" );
+ continue;
}
- /* read thoses data */
- if( p_stream->i_samplesize )
+ if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS )
{
- int i_toread;
-
- if( ( i_toread = toread[i_stream].i_toread ) <= 0 )
+ __Parse_indx( p_input, i_stream, p_indx );
+ }
+ else if( p_indx->i_indextype == AVI_INDEX_OF_INDEXES )
+ {
+ avi_chunk_indx_t ck_sub;
+ for( i = 0; i < p_indx->i_entriesinuse; i++ )
{
- if( p_stream->i_samplesize > 1 )
+ if( stream_Seek( p_input->s, p_indx->idx.super[i].i_offset )||
+ AVI_ChunkRead( p_input->s, &ck_sub, NULL ) )
{
- i_toread = p_stream->i_samplesize;
- }
- else
- {
- i_toread = __MAX( AVI_PTSToByte( p_stream, 20 * 1000 ), 100 );
+ break;
}
+ __Parse_indx( p_input, i_stream, &ck_sub );
}
- i_size = __MIN( p_stream->p_index[p_stream->i_idxposc].i_length -
- p_stream->i_idxposb,
- i_toread );
}
else
{
- i_size = p_stream->p_index[p_stream->i_idxposc].i_length;
+ msg_Warn( p_input, "unknow type index(0x%x)", p_indx->i_indextype );
}
+#undef p_stream
+ }
+}
- if( p_stream->i_idxposb == 0 )
- {
- i_size += 8; // need to read and skip header
- }
+static void AVI_IndexLoad( input_thread_t *p_input )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ unsigned int i_stream;
- if( input_ReadInPES( p_input, &p_pes, __EVEN( i_size ) ) < 0 )
- {
- msg_Warn( p_input, "failled reading data" );
- toread[i_stream].b_ok = VLC_FALSE;
- continue;
- }
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
+ {
+ p_avi->pp_info[i_stream]->i_idxnb = 0;
+ p_avi->pp_info[i_stream]->i_idxmax = 0;
+ p_avi->pp_info[i_stream]->p_index = NULL;
+ }
- if( i_size % 2 ) // read was padded on word boundary
- {
- p_pes->p_last->p_payload_end--;
- p_pes->i_pes_size--;
- }
- // skip header
- if( p_stream->i_idxposb == 0 )
- {
- p_pes->p_first->p_payload_start += 8;
- p_pes->i_pes_size -= 8;
- }
+ if( p_avi->b_odml )
+ {
+ AVI_IndexLoad_indx( p_input );
+ }
+ else if( AVI_IndexLoad_idx1( p_input ) )
+ {
+ /* try indx if idx1 failed as some "normal" file have indx too */
+ AVI_IndexLoad_indx( p_input );
+ }
- p_pes->i_pts = AVI_GetPTS( p_stream );
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
+ {
+ msg_Dbg( p_input, "stream[%d] created %d index entries",
+ i_stream, p_avi->pp_info[i_stream]->i_idxnb );
+ }
+}
+
+static void AVI_IndexCreate( input_thread_t *p_input )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
+
+ avi_chunk_list_t *p_riff;
+ avi_chunk_list_t *p_movi;
+
+ unsigned int i_stream;
+ off_t i_movi_end;
+
+ p_riff = AVI_ChunkFind( &p_avi->ck_root, AVIFOURCC_RIFF, 0);
+ p_movi = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
+
+ if( !p_movi )
+ {
+ msg_Err( p_input, "cannot find p_movi" );
+ return;
+ }
+
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
+ {
+ p_avi->pp_info[i_stream]->i_idxnb = 0;
+ p_avi->pp_info[i_stream]->i_idxmax = 0;
+ p_avi->pp_info[i_stream]->p_index = NULL;
+ }
+ i_movi_end = __MIN( (off_t)(p_movi->i_chunk_pos + p_movi->i_chunk_size),
+ stream_Size( p_input->s ) );
+
+ stream_Seek( p_input->s, p_movi->i_chunk_pos + 12 );
+ msg_Warn( p_input, "creating index from LIST-movi, will take time !" );
+ for( ;; )
+ {
+ avi_packet_t pk;
- /* read data */
- if( p_stream->i_samplesize )
+ if( AVI_PacketGetHeader( p_input, &pk ) )
{
- if( p_stream->i_idxposb == 0 )
- {
- i_size -= 8;
- }
- toread[i_stream].i_toread -= i_size;
- p_stream->i_idxposb += i_size;
- if( p_stream->i_idxposb >=
- p_stream->p_index[p_stream->i_idxposc].i_length )
- {
- p_stream->i_idxposb = 0;
- p_stream->i_idxposc++;
- }
+ break;
}
- else
+ if( pk.i_stream < p_avi->i_streams &&
+ pk.i_cat == p_avi->pp_info[pk.i_stream]->i_cat )
{
- toread[i_stream].i_toread--;
- p_stream->i_idxposc++;
+ AVIIndexEntry_t index;
+ index.i_id = pk.i_fourcc;
+ index.i_flags =
+ AVI_GetKeyFlag(p_avi->pp_info[pk.i_stream]->i_codec, pk.i_peek);
+ index.i_pos = pk.i_pos;
+ index.i_length = pk.i_size;
+ AVI_IndexAddEntry( p_avi, pk.i_stream, &index );
}
-
- if( p_stream->i_idxposc < p_stream->i_idxnb)
+ else
{
- toread[i_stream].i_posf =
- p_stream->p_index[p_stream->i_idxposc].i_pos;
- if( p_stream->i_idxposb > 0 )
+ switch( pk.i_fourcc )
{
- toread[i_stream].i_posf += 8 + p_stream->i_idxposb;
- }
+ case AVIFOURCC_idx1:
+ if( p_avi->b_odml )
+ {
+ avi_chunk_list_t *p_avix;
+ p_avix = AVI_ChunkFind( &p_avi->ck_root,
+ AVIFOURCC_RIFF, 1 );
+ msg_Dbg( p_input, "looking for new RIFF chunk" );
+ if( stream_Seek( p_input->s, p_avix->i_chunk_pos + 24))
+ {
+ goto print_stat;
+ }
+ break;
+ }
+ goto print_stat;
+ case AVIFOURCC_RIFF:
+ msg_Dbg( p_input, "new RIFF chunk found" );
+ case AVIFOURCC_rec:
+ case AVIFOURCC_JUNK:
+ break;
+ default:
+ msg_Warn( p_input, "need resync, probably broken avi" );
+ if( AVI_PacketSearch( p_input ) )
+ {
+ msg_Warn( p_input, "lost sync, abord index creation" );
+ goto print_stat;
+ }
+ }
}
- else
- {
- toread[i_stream].i_posf = -1;
- }
-
- b_stream = VLC_TRUE; // at least one read succeed
-
- if( p_stream->p_es && p_stream->p_es->p_decoder_fifo )
- {
- p_pes->i_dts =
- p_pes->i_pts =
- input_ClockGetTS( p_input,
- p_input->stream.p_selected_program,
- p_pes->i_pts * 9/100);
- input_DecodePES( p_stream->p_es->p_decoder_fifo, p_pes );
- }
- else
+ if( ( !p_avi->b_odml && pk.i_pos + pk.i_size >= i_movi_end ) ||
+ AVI_PacketNext( p_input ) )
{
- input_DeletePES( p_input->p_method_data, p_pes );
+ break;
}
}
-}
+print_stat:
+ for( i_stream = 0; i_stream < p_avi->i_streams; i_stream++ )
+ {
+ msg_Dbg( p_input,
+ "stream[%d] creating %d index entries",
+ i_stream,
+ p_avi->pp_info[i_stream]->i_idxnb );
+ }
+}
/*****************************************************************************
- * AVIDemux_UnSeekable: reads and demuxes data packets for unseekable file
- *****************************************************************************
- * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
+ * Stream management
*****************************************************************************/
-static int AVIDemux_UnSeekable( input_thread_t *p_input )
+static vlc_bool_t AVI_StreamStart( input_thread_t *p_input, int i_stream )
{
- demux_sys_t *p_avi = p_input->p_demux_data;
- avi_stream_t *p_stream_master;
- vlc_bool_t b_audio;
- unsigned int i_stream;
- unsigned int i_packet;
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ avi_stream_t *tk = p_avi->pp_info[i_stream];
- /* *** send audio data to decoder only if rate == DEFAULT_RATE *** */
- vlc_mutex_lock( &p_input->stream.stream_lock );
- b_audio = p_input->stream.control.i_rate == DEFAULT_RATE;
- vlc_mutex_unlock( &p_input->stream.stream_lock );
+ if( !tk->p_es )
+ {
+ msg_Warn( p_input, "stream[%d] unselectable", i_stream );
+ return VLC_FALSE;
+ }
+ if( tk->b_activated )
+ {
+ msg_Warn( p_input, "stream[%d] already selected", i_stream );
+ return VLC_TRUE;
+ }
- input_ClockManageRef( p_input,
- p_input->stream.p_selected_program,
- p_avi->i_pcr );
- /* *** find master stream for data packet skipping algo *** */
- /* *** -> first video, if any, or first audio ES *** */
- for( i_stream = 0, p_stream_master = NULL;
- i_stream < p_avi->i_streams; i_stream++ )
+ if( !tk->p_es->p_decoder_fifo )
{
-#define p_stream p_avi->pp_info[i_stream]
- if( p_stream->p_es &&
- p_stream->p_es->p_decoder_fifo )
- {
- if( p_stream->i_cat == VIDEO_ES )
- {
- p_stream_master = p_stream;
- break;
- }
- if( p_stream->i_cat == AUDIO_ES && !p_stream_master )
- {
- p_stream_master = p_stream;
- }
- }
-#undef p_stream
+ vlc_mutex_lock( &p_input->stream.stream_lock );
+ input_SelectES( p_input, tk->p_es );
+ vlc_mutex_unlock( &p_input->stream.stream_lock );
}
- if( !p_stream_master )
+ tk->b_activated = tk->p_es->p_decoder_fifo ? VLC_TRUE : VLC_FALSE;
+ if( tk->b_activated && p_avi->b_seekable)
{
- msg_Warn( p_input, "no more stream selected" );
- return( 0 );
+ AVI_StreamSeek( p_input, i_stream, p_avi->i_time );
}
- p_avi->i_pcr = AVI_GetPTS( p_stream_master ) * 9 / 100;
+ return tk->b_activated;
+}
- for( i_packet = 0; i_packet < 10; i_packet++)
+static void AVI_StreamStop( input_thread_t *p_input, int i_stream )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ avi_stream_t *tk = p_avi->pp_info[i_stream];
+
+ if( !tk->b_activated )
{
-#define p_stream p_avi->pp_info[avi_pk.i_stream]
+ msg_Warn( p_input, "stream[%d] already unselected", i_stream );
+ return;
+ }
- avi_packet_t avi_pk;
+ if( tk->p_es->p_decoder_fifo )
+ {
+ vlc_mutex_lock( &p_input->stream.stream_lock );
+ input_UnselectES( p_input, tk->p_es );
+ vlc_mutex_unlock( &p_input->stream.stream_lock );
+ }
- if( AVI_PacketGetHeader( p_input, &avi_pk ) )
- {
- return( 0 );
- }
-// AVI_ParseStreamHeader( avi_pk.i_fourcc, &i_stream, &i_cat );
+ tk->b_activated = VLC_FALSE;
+}
- if( avi_pk.i_stream >= p_avi->i_streams ||
- ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
+static int AVI_StreamStopFinishedStreams( input_thread_t *p_input )
+{
+ demux_sys_t *p_avi = p_input->p_demux_data;
+ unsigned int i;
+ int b_end = VLC_TRUE;
+
+ for( i = 0; i < p_avi->i_streams; i++ )
+ {
+#define tk p_avi->pp_info[i]
+ if( tk->i_idxposc >= tk->i_idxnb )
{
- /* we haven't found an audio or video packet:
- * - we have seek, found first next packet
- * - others packets could be found, skip them
- */
- switch( avi_pk.i_fourcc )
- {
- case AVIFOURCC_JUNK:
- case AVIFOURCC_LIST:
- return( !AVI_PacketNext( p_input ) ? 1 : 0 );
- case AVIFOURCC_idx1:
- return( 0 ); // eof
- default:
- msg_Warn( p_input,
- "seems to have lost position, resync" );
- if( AVI_PacketSearch( p_input ) )
- {
- msg_Err( p_input, "resync failed" );
- return( -1 );
- }
- }
+ AVI_StreamStop( p_input, i );
}
else
{
- /* do will send this packet to decoder ? */
- if( ( !b_audio && avi_pk.i_cat == AUDIO_ES )||
- !p_stream->p_es ||
- !p_stream->p_es->p_decoder_fifo )
- {
- if( AVI_PacketNext( p_input ) )
- {
- return( 0 );
- }
- }
- else
- {
- /* it's a selected stream, check for time */
- if( __ABS( AVI_GetPTS( p_stream ) -
- AVI_GetPTS( p_stream_master ) )< 600*1000 )
- {
- /* load it and send to decoder */
- pes_packet_t *p_pes;
- if( AVI_PacketRead( p_input, &avi_pk, &p_pes ) || !p_pes)
- {
- return( -1 );
- }
- p_pes->i_dts =
- p_pes->i_pts =
- input_ClockGetTS( p_input,
- p_input->stream.p_selected_program,
- AVI_GetPTS( p_stream ) * 9/100);
- input_DecodePES( p_stream->p_es->p_decoder_fifo, p_pes );
- }
- else
- {
- if( AVI_PacketNext( p_input ) )
- {
- return( 0 );
- }
- }
- }
+ b_end = VLC_FALSE;
+ }
+#undef tk
+ }
+ return( b_end );
+}
- /* *** update stream time position *** */
- if( p_stream->i_samplesize )
- {
- p_stream->i_idxposb += avi_pk.i_size;
- }
- else
- {
- p_stream->i_idxposc++;
- }
+/****************************************************************************
+ * AVI_MovieGetLength give max streams length in second
+ ****************************************************************************/
+static mtime_t AVI_MovieGetLength( input_thread_t *p_input )
+{
+ demux_sys_t *p_sys = p_input->p_demux_data;
+ mtime_t i_maxlength = 0;
+ unsigned int i;
+
+ for( i = 0; i < p_sys->i_streams; i++ )
+ {
+#define tk p_sys->pp_info[i]
+ mtime_t i_length;
+
+ /* fix length for each stream */
+ if( tk->i_idxnb < 1 || !tk->p_index )
+ {
+ continue;
+ }
+ if( tk->i_samplesize )
+ {
+ i_length = AVI_GetDPTS( tk,
+ tk->p_index[tk->i_idxnb-1].i_lengthtotal +
+ tk->p_index[tk->i_idxnb-1].i_length );
}
+ else
+ {
+ i_length = AVI_GetDPTS( tk, tk->i_idxnb );
+ }
+ i_length /= (mtime_t)1000000; /* in seconds */
-#undef p_stream
+ msg_Dbg( p_input,
+ "stream[%d] length:"I64Fd" (based on index)",
+ i,
+ i_length );
+ i_maxlength = __MAX( i_maxlength, i_length );
+#undef tk
}
- return( 1 );
+ return i_maxlength;
}
+