开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > midi > 查看源码
avi.c
资源名称：vlc-1.0.5.zip　[点击查看]
上传用户：kjfoods
上传日期：2020-07-06
资源大小：29949k
文件大小：81k
源码类别：

midi

开发平台：

Unix_Linux

avi.c：源码内容
  1. /*****************************************************************************
  2.  * avi.c : AVI file Stream input module for vlc
  3.  *****************************************************************************
  4.  * Copyright (C) 2001-2004 the VideoLAN team
  5.  * $Id: 47a97e4c57c9c35a2115f09e1dbebd326902a788 $
  6.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  7.  *
  8.  * This program is free software; you can redistribute it and/or modify
  9.  * it under the terms of the GNU General Public License as published by
  10.  * the Free Software Foundation; either version 2 of the License, or
  11.  * (at your option) any later version.
  12.  *
  13.  * This program is distributed in the hope that it will be useful,
  14.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16.  * GNU General Public License for more details.
  17.  *
  18.  * You should have received a copy of the GNU General Public License
  19.  * along with this program; if not, write to the Free Software
  20.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  21.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  27. #ifdef HAVE_CONFIG_H
  28. # include "config.h"
  29. #endif
  31. #include <vlc_common.h>
  32. #include <vlc_plugin.h>
  33. #include <vlc_demux.h>
  35. #include <vlc_dialog.h>
  37. #include <vlc_meta.h>
  38. #include <vlc_codecs.h>
  39. #include <vlc_charset.h>
  41. #include "libavi.h"
  43. /*****************************************************************************
  44.  * Module descriptor
  45.  *****************************************************************************/
  47. #define INTERLEAVE_TEXT N_("Force interleaved method" )
  48. #define INTERLEAVE_LONGTEXT N_( "Force interleaved method." )
  50. #define INDEX_TEXT N_("Force index creation")
  51. #define INDEX_LONGTEXT N_( 
  52.     "Recreate a index for the AVI file. Use this if your AVI file is damaged "
  53.     "or incomplete (not seekable)." )
  55. static int  Open ( vlc_object_t * );
  56. static void Close( vlc_object_t * );
  58. static const int pi_index[] = {0,1,2};
  60. static const char *const ppsz_indexes[] = { N_("Ask"), N_("Always fix"),
  61.                                 N_("Never fix") };
  63. vlc_module_begin ()
  64.     set_shortname( "AVI" )
  65.     set_description( N_("AVI demuxer") )
  66.     set_capability( "demux", 212 )
  67.     set_category( CAT_INPUT )
  68.     set_subcategory( SUBCAT_INPUT_DEMUX )
  70.     add_bool( "avi-interleaved", 0, NULL,
  71.               INTERLEAVE_TEXT, INTERLEAVE_LONGTEXT, true )
  72.     add_integer( "avi-index", 0, NULL,
  73.               INDEX_TEXT, INDEX_LONGTEXT, false )
  74.         change_integer_list( pi_index, ppsz_indexes, NULL )
  76.     set_callbacks( Open, Close )
  77. vlc_module_end ()
  79. /*****************************************************************************
  80.  * Local prototypes
  81.  *****************************************************************************/
  82. static int Control         ( demux_t *, int, va_list );
  83. static int Seek            ( demux_t *, mtime_t, int );
  84. static int Demux_Seekable  ( demux_t * );
  85. static int Demux_UnSeekable( demux_t * );
  87. #define __ABS( x ) ( (x) < 0 ? (-(x)) : (x) )
  89. typedef struct
  90. {
  91.     vlc_fourcc_t i_fourcc;
  92.     off_t        i_pos;
  93.     uint32_t     i_size;
  94.     vlc_fourcc_t i_type;     /* only for AVIFOURCC_LIST */
  96.     uint8_t      i_peek[8];  /* first 8 bytes */
  98.     unsigned int i_stream;
  99.     unsigned int i_cat;
  100. } avi_packet_t;
  103. typedef struct
  104. {
  105.     vlc_fourcc_t i_id;
  106.     uint32_t     i_flags;
  107.     off_t        i_pos;
  108.     uint32_t     i_length;
  109.     int64_t      i_lengthtotal;
  111. } avi_entry_t;
  113. typedef struct
  114. {
  115.     bool            b_activated;
  116.     bool            b_eof;
  118.     unsigned int    i_cat; /* AUDIO_ES, VIDEO_ES */
  119.     vlc_fourcc_t    i_codec;
  121.     int             i_rate;
  122.     int             i_scale;
  123.     int             i_samplesize;
  125.     es_out_id_t     *p_es;
  127.     /* Avi Index */
  128.     avi_entry_t     *p_index;
  129.     unsigned int    i_idxnb;
  130.     unsigned int    i_idxmax;
  132.     unsigned int    i_idxposc;  /* numero of chunk */
  133.     unsigned int    i_idxposb;  /* byte in the current chunk */
  135.     /* extra information given to the decoder */
  136.     void            *p_extra;
  138.     /* For VBR audio only */
  139.     unsigned int    i_blockno;
  140.     unsigned int    i_blocksize;
  142.     /* For muxed streams */
  143.     stream_t        *p_out_muxed;
  144. } avi_track_t;
  146. struct demux_sys_t
  147. {
  148.     mtime_t i_time;
  149.     mtime_t i_length;
  151.     bool  b_seekable;
  152.     bool  b_muxed;
  153.     avi_chunk_t ck_root;
  155.     bool  b_odml;
  157.     off_t   i_movi_begin;
  158.     off_t   i_movi_lastchunk_pos;   /* XXX position of last valid chunk */
  160.     /* number of streams and information */
  161.     unsigned int i_track;
  162.     avi_track_t  **track;
  164.     /* meta */
  165.     vlc_meta_t  *meta;
  166. };
  168. static inline off_t __EVEN( off_t i )
  169. {
  170.     return (i & 1) ? i + 1 : i;
  171. }
  173. static mtime_t AVI_PTSToChunk( avi_track_t *, mtime_t i_pts );
  174. static mtime_t AVI_PTSToByte ( avi_track_t *, mtime_t i_pts );
  175. static mtime_t AVI_GetDPTS   ( avi_track_t *, int64_t i_count );
  176. static mtime_t AVI_GetPTS    ( avi_track_t * );
  179. static int AVI_StreamChunkFind( demux_t *, unsigned int i_stream );
  180. static int AVI_StreamChunkSet ( demux_t *,
  181.                                 unsigned int i_stream, unsigned int i_ck );
  182. static int AVI_StreamBytesSet ( demux_t *,
  183.                                 unsigned int i_stream, off_t   i_byte );
  185. vlc_fourcc_t AVI_FourccGetCodec( unsigned int i_cat, vlc_fourcc_t );
  186. static int   AVI_GetKeyFlag    ( vlc_fourcc_t , uint8_t * );
  188. static int AVI_PacketGetHeader( demux_t *, avi_packet_t *p_pk );
  189. static int AVI_PacketNext     ( demux_t * );
  190. static int AVI_PacketRead     ( demux_t *, avi_packet_t *, block_t **);
  191. static int AVI_PacketSearch   ( demux_t * );
  193. static void AVI_IndexLoad    ( demux_t * );
  194. static void AVI_IndexCreate  ( demux_t * );
  195. static void AVI_IndexAddEntry( demux_sys_t *, int, avi_entry_t * );
  197. static mtime_t  AVI_MovieGetLength( demux_t * );
  199. /*****************************************************************************
  200.  * Stream management
  201.  *****************************************************************************/
  202. static int        AVI_TrackSeek  ( demux_t *, int, mtime_t );
  203. static int        AVI_TrackStopFinishedStreams( demux_t *);
  205. /* Remarks:
  206.  - For VBR mp3 stream:
  207.     count blocks by rounded-up chunksizes instead of chunks
  208.     we need full emulation of dshow avi demuxer bugs :(
  209.     fixes silly nandub-style a-v delaying in avi with vbr mp3...
  210.     (from mplayer 2002/08/02)
  211.  - to complete....
  212.  */
  214. /*****************************************************************************
  215.  * Open: check file and initializes AVI structures
  216.  *****************************************************************************/
  217. static int Open( vlc_object_t * p_this )
  218. {
  219.     demux_t  *p_demux = (demux_t *)p_this;
  220.     demux_sys_t     *p_sys;
  222.     bool       b_index = false;
  223.     int              i_do_index;
  225.     avi_chunk_t         ck_riff;
  226.     avi_chunk_list_t    *p_riff = (avi_chunk_list_t*)&ck_riff;
  227.     avi_chunk_list_t    *p_hdrl, *p_movi;
  228.     avi_chunk_avih_t    *p_avih;
  230.     unsigned int i_track;
  231.     unsigned int i, i_peeker;
  233.     const uint8_t *p_peek;
  235.     /* Is it an avi file ? */
  236.     if( stream_Peek( p_demux->s, &p_peek, 200 ) < 200 ) return VLC_EGENERIC;
  238.     for( i_peeker = 0; i_peeker < 188; i_peeker++ )
  239.     {
  240.         if( !strncmp( (char *)&p_peek[0], "RIFF", 4 ) && !strncmp( (char *)&p_peek[8], "AVI ", 4 ) )
  241.             break;
  242.         if( !strncmp( (char *)&p_peek[0], "ON2 ", 4 ) && !strncmp( (char *)&p_peek[8], "ON2f", 4 ) )
  243.             break;
  244.         p_peek++;
  245.     }
  246.     if( i_peeker == 188 )
  247.     {
  248.         return VLC_EGENERIC;
  249.     }
  251.     /* Initialize input  structures. */
  252.     p_sys = p_demux->p_sys = malloc( sizeof(demux_sys_t) );
  253.     memset( p_sys, 0, sizeof( demux_sys_t ) );
  254.     p_sys->i_time   = 0;
  255.     p_sys->i_length = 0;
  256.     p_sys->i_movi_lastchunk_pos = 0;
  257.     p_sys->b_odml   = false;
  258.     p_sys->b_muxed  = false;
  259.     p_sys->i_track  = 0;
  260.     p_sys->track    = NULL;
  261.     p_sys->meta     = NULL;
  263.     stream_Control( p_demux->s, STREAM_CAN_FASTSEEK, &p_sys->b_seekable );
  265.     p_demux->pf_control = Control;
  266.     p_demux->pf_demux = Demux_Seekable;
  268.     /* For unseekable stream, automaticaly use Demux_UnSeekable */
  269.     if( !p_sys->b_seekable || config_GetInt( p_demux, "avi-interleaved" ) )
  270.     {
  271.         p_demux->pf_demux = Demux_UnSeekable;
  272.     }
  274.     if( i_peeker > 0 )
  275.     {
  276.         stream_Read( p_demux->s, NULL, i_peeker );
  277.     }
  279.     if( AVI_ChunkReadRoot( p_demux->s, &p_sys->ck_root ) )
  280.     {
  281.         msg_Err( p_demux, "avi module discarded (invalid file)" );
  282.         return VLC_EGENERIC;
  283.     }
  285.     if( AVI_ChunkCount( &p_sys->ck_root, AVIFOURCC_RIFF ) > 1 )
  286.     {
  287.         unsigned int i_count =
  288.             AVI_ChunkCount( &p_sys->ck_root, AVIFOURCC_RIFF );
  290.         msg_Warn( p_demux, "multiple riff -> OpenDML ?" );
  291.         for( i = 1; i < i_count; i++ )
  292.         {
  293.             avi_chunk_list_t *p_sysx;
  295.             p_sysx = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, i );
  296.             if( p_sysx->i_type == AVIFOURCC_AVIX )
  297.             {
  298.                 msg_Warn( p_demux, "detected OpenDML file" );
  299.                 p_sys->b_odml = true;
  300.                 break;
  301.             }
  302.         }
  303.     }
  305.     p_riff  = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0 );
  306.     p_hdrl  = AVI_ChunkFind( p_riff, AVIFOURCC_hdrl, 0 );
  307.     p_movi  = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0 );
  309.     if( !p_hdrl || !p_movi )
  310.     {
  311.         msg_Err( p_demux, "avi module discarded (invalid file)" );
  312.         goto error;
  313.     }
  315.     if( !( p_avih = AVI_ChunkFind( p_hdrl, AVIFOURCC_avih, 0 ) ) )
  316.     {
  317.         msg_Err( p_demux, "cannot find avih chunk" );
  318.         goto error;
  319.     }
  320.     i_track = AVI_ChunkCount( p_hdrl, AVIFOURCC_strl );
  321.     if( p_avih->i_streams != i_track )
  322.     {
  323.         msg_Warn( p_demux,
  324.                   "found %d stream but %d are declared",
  325.                   i_track, p_avih->i_streams );
  326.     }
  327.     if( i_track == 0 )
  328.     {
  329.         msg_Err( p_demux, "no stream defined!" );
  330.         goto error;
  331.     }
  333.     /* print information on streams */
  334.     msg_Dbg( p_demux, "AVIH: %d stream, flags %s%s%s%s ",
  335.              i_track,
  336.              p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
  337.              p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
  338.              p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
  339.              p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
  340.     if( ( p_sys->meta = vlc_meta_New() ) )
  341.     {
  342.         char buffer[200];
  343.         snprintf( buffer, sizeof(buffer), "%s%s%s%s",
  344.                   p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
  345.                   p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
  346.                   p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
  347.                   p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
  348.         vlc_meta_SetSetting( p_sys->meta, buffer );
  349.     }
  351.     /* now read info on each stream and create ES */
  352.     for( i = 0 ; i < i_track; i++ )
  353.     {
  354.         avi_track_t           *tk     = malloc( sizeof( avi_track_t ) );
  355.         if( !tk )
  356.             goto error;
  358.         avi_chunk_list_t      *p_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i );
  359.         avi_chunk_strh_t      *p_strh = AVI_ChunkFind( p_strl, AVIFOURCC_strh, 0 );
  360.         avi_chunk_STRING_t    *p_strn = AVI_ChunkFind( p_strl, AVIFOURCC_strn, 0 );
  361.         avi_chunk_strf_auds_t *p_auds = NULL;
  362.         avi_chunk_strf_vids_t *p_vids = NULL;
  363.         es_format_t fmt;
  365.         memset( tk, 0, sizeof(*tk) );
  366.         tk->b_eof = false;
  367.         tk->b_activated = true;
  369.         p_vids = (avi_chunk_strf_vids_t*)AVI_ChunkFind( p_strl, AVIFOURCC_strf, 0 );
  370.         p_auds = (avi_chunk_strf_auds_t*)AVI_ChunkFind( p_strl, AVIFOURCC_strf, 0 );
  372.         if( p_strl == NULL || p_strh == NULL || p_auds == NULL || p_vids == NULL )
  373.         {
  374.             msg_Warn( p_demux, "stream[%d] incomplete", i );
  375.             free( tk );
  376.             continue;
  377.         }
  379.         tk->i_rate  = p_strh->i_rate;
  380.         tk->i_scale = p_strh->i_scale;
  381.         tk->i_samplesize = p_strh->i_samplesize;
  382.         msg_Dbg( p_demux, "stream[%d] rate:%d scale:%d samplesize:%d",
  383.                  i, tk->i_rate, tk->i_scale, tk->i_samplesize );
  385.         switch( p_strh->i_type )
  386.         {
  387.             case( AVIFOURCC_auds ):
  388.                 tk->i_cat   = AUDIO_ES;
  389.                 tk->i_codec = AVI_FourccGetCodec( AUDIO_ES,
  390.                                                   p_auds->p_wf->wFormatTag );
  392.                 tk->i_blocksize = p_auds->p_wf->nBlockAlign;
  393.                 if( tk->i_blocksize == 0 )
  394.                 {
  395.                     if( p_auds->p_wf->wFormatTag == 1 )
  396.                         tk->i_blocksize = p_auds->p_wf->nChannels * (p_auds->p_wf->wBitsPerSample/8);
  397.                     else
  398.                         tk->i_blocksize = 1;
  399.                 }
  400.                 else if( tk->i_samplesize != 0 && tk->i_samplesize != tk->i_blocksize )
  401.                 {
  402.                     msg_Warn( p_demux, "track[%d] samplesize=%d and blocksize=%d are not equal."
  403.                                        "Using blocksize as a workaround.",
  404.                                        i, tk->i_samplesize, tk->i_blocksize );
  405.                     tk->i_samplesize = tk->i_blocksize;
  406.                 }
  408.                 if( tk->i_codec == VLC_FOURCC( 'v', 'o', 'r', 'b' ) )
  409.                 {
  410.                     tk->i_blocksize = 0; /* fix vorbis VBR decoding */
  411.                 }
  413.                 es_format_Init( &fmt, AUDIO_ES, tk->i_codec );
  415.                 fmt.audio.i_channels        = p_auds->p_wf->nChannels;
  416.                 fmt.audio.i_rate            = p_auds->p_wf->nSamplesPerSec;
  417.                 fmt.i_bitrate               = p_auds->p_wf->nAvgBytesPerSec*8;
  418.                 fmt.audio.i_blockalign      = p_auds->p_wf->nBlockAlign;
  419.                 fmt.audio.i_bitspersample   = p_auds->p_wf->wBitsPerSample;
  420.                 fmt.b_packetized            = !tk->i_blocksize;
  422.                 msg_Dbg( p_demux,
  423.                     "stream[%d] audio(0x%x) %d channels %dHz %dbits",
  424.                     i, p_auds->p_wf->wFormatTag, p_auds->p_wf->nChannels,
  425.                     p_auds->p_wf->nSamplesPerSec, 
  426.                     p_auds->p_wf->wBitsPerSample );
  428.                 fmt.i_extra = __MIN( p_auds->p_wf->cbSize,
  429.                     p_auds->i_chunk_size - sizeof(WAVEFORMATEX) );
  430.                 fmt.p_extra = tk->p_extra = malloc( fmt.i_extra );
  431.                 if( !fmt.p_extra ) goto error;
  432.                 memcpy( fmt.p_extra, &p_auds->p_wf[1], fmt.i_extra );
  434.                 /* Rewrite the vorbis headers from Xiph-like format
  435.                  * to VLC internal format
  436.                  *
  437.                  * Xiph format:
  438.                  *  - 1st byte == N, is the number of packets - 1
  439.                  *  - Following bytes are the size of the N first packets:
  440.                  *      while( *p == 0xFF ) { size += 0xFF; p++ } size += *p;
  441.                  *      (the size of the last packet is the size of remaining
  442.                  *      data in the buffer)
  443.                  *  - Finally, all the packets concatenated
  444.                  *
  445.                  * VLC format:
  446.                  *  - Size of the packet on 16 bits (big endian) FIXME: should be 32 bits to be safe
  447.                  *  - The packet itself
  448.                  *  - Size of the next packet, and so on ...
  449.                  */
  451.                 if( tk->i_codec == VLC_FOURCC( 'v', 'o', 'r', 'b' ) )
  452.                 {
  453.                     uint8_t *p_extra = fmt.p_extra; 
  454.                     size_t i_extra = fmt.i_extra;
  456.                     if( i_extra <= 1 ) break;
  457.                     if( *p_extra++ != 2 ) break; /* 3 packets - 1 = 2 */
  458.                     i_extra--;
  460.                     size_t i_identifier_len = 0;
  461.                     while( *p_extra == 0xFF )
  462.                     {
  463.                         i_identifier_len += 0xFF;
  464.                         p_extra++;
  465.                         if( --i_extra <= 1 ) break;
  466.                     }
  467.                     i_identifier_len += *p_extra++;
  468.                     if( i_identifier_len > --i_extra ) break;
  470.                     size_t i_comment_len = 0;
  471.                     while( *p_extra == 0xFF )
  472.                     {
  473.                         i_comment_len += 0xFF;
  474.                         p_extra++;
  475.                         if( --i_extra <= 1 ) break;
  476.                     }
  477.                     i_comment_len += *p_extra++;
  478.                     if( i_comment_len > --i_extra ) break;
  479.                     size_t i_cookbook_len = i_extra;
  481.                     size_t i_headers_size = 3  * 2 + i_identifier_len +
  482.                                             i_comment_len + i_cookbook_len;
  483.                     uint8_t *p_out = malloc( i_headers_size );
  484.                     if( !p_out ) goto error;
  485.                     free( fmt.p_extra );
  486.                     fmt.p_extra = tk->p_extra = p_out;
  487.                     fmt.i_extra = i_headers_size;
  488.                     #define copy_packet( len ) 
  489.                         *p_out++ = len >> 8; 
  490.                         *p_out++ = len & 0xFF; 
  491.                         memcpy( p_out, p_extra, len ); 
  492.                         p_out += len; 
  493.                         p_extra += len;
  494.                     copy_packet( i_identifier_len );
  495.                     copy_packet( i_comment_len );
  496.                     copy_packet( i_cookbook_len );
  497.                     #undef copy_packet
  498.                     break;
  499.                 }
  500.                 break;
  502.             case( AVIFOURCC_vids ):
  503.                 tk->i_cat   = VIDEO_ES;
  504.                 tk->i_codec = AVI_FourccGetCodec( VIDEO_ES,
  505.                                                   p_vids->p_bih->biCompression );
  506.                 if( p_vids->p_bih->biCompression == VLC_FOURCC( 'D', 'X', 'S', 'B' ) )
  507.                 {
  508.                    msg_Dbg( p_demux, "stream[%d] subtitles", i );
  509.                    es_format_Init( &fmt, SPU_ES, p_vids->p_bih->biCompression );
  510.                    break;
  511.                 }
  512.                 else if( p_vids->p_bih->biCompression == 0x00 )
  513.                 {
  514.                     switch( p_vids->p_bih->biBitCount )
  515.                     {
  516.                         case 32:
  517.                             tk->i_codec = VLC_FOURCC('R','V','3','2');
  518.                             break;
  519.                         case 24:
  520.                             tk->i_codec = VLC_FOURCC('R','V','2','4');
  521.                             break;
  522.                         case 16: /* Yes it is RV15 */
  523.                         case 15:
  524.                             tk->i_codec = VLC_FOURCC('R','V','1','5');
  525.                             break;
  526.                         case 9: /* <- TODO check that */
  527.                             tk->i_codec = VLC_FOURCC( 'Y', 'V', 'U', '9' );
  528.                             break;
  529.                         case 8: /* <- TODO check that */
  530.                             tk->i_codec = VLC_FOURCC('Y','8','0','0');
  531.                             break;
  532.                     }
  533.                     es_format_Init( &fmt, VIDEO_ES, tk->i_codec );
  535.                     switch( tk->i_codec )
  536.                     {
  537.                     case VLC_FOURCC('R','V','2','4'):
  538.                     case VLC_FOURCC('R','V','3','2'):
  539.                         fmt.video.i_rmask = 0x00ff0000;
  540.                         fmt.video.i_gmask = 0x0000ff00;
  541.                         fmt.video.i_bmask = 0x000000ff;
  542.                         break;
  543.                     case VLC_FOURCC('R','V','1','5'):
  544.                         fmt.video.i_rmask = 0x7c00;
  545.                         fmt.video.i_gmask = 0x03e0;
  546.                         fmt.video.i_bmask = 0x001f;
  547.                         break;
  548.                     default:
  549.                         break;
  550.                     }
  551.                 }
  552.                 else
  553.                 {
  554.                     es_format_Init( &fmt, VIDEO_ES, p_vids->p_bih->biCompression );
  555.                     if( tk->i_codec == FOURCC_mp4v &&
  556.                         !strncasecmp( (char*)&p_strh->i_handler, "XVID", 4 ) )
  557.                     {
  558.                         fmt.i_codec = VLC_FOURCC( 'X', 'V', 'I', 'D' );
  559.                     }
  560.                 }
  561.                 tk->i_samplesize = 0;
  562.                 fmt.video.i_width  = p_vids->p_bih->biWidth;
  563.                 fmt.video.i_height = p_vids->p_bih->biHeight;
  564.                 fmt.video.i_bits_per_pixel = p_vids->p_bih->biBitCount;
  565.                 fmt.video.i_frame_rate = tk->i_rate;
  566.                 fmt.video.i_frame_rate_base = tk->i_scale;
  567.                 fmt.i_extra =
  568.                     __MIN( p_vids->p_bih->biSize - sizeof( BITMAPINFOHEADER ),
  569.                            p_vids->i_chunk_size - sizeof(BITMAPINFOHEADER) );
  570.                 fmt.p_extra = &p_vids->p_bih[1];
  571.                 msg_Dbg( p_demux, "stream[%d] video(%4.4s) %"PRIu32"x%"PRIu32" %dbpp %ffps",
  572.                          i, (char*)&p_vids->p_bih->biCompression,
  573.                          (uint32_t)p_vids->p_bih->biWidth,
  574.                          (uint32_t)p_vids->p_bih->biHeight,
  575.                          p_vids->p_bih->biBitCount,
  576.                          (float)tk->i_rate/(float)tk->i_scale );
  578.                 if( p_vids->p_bih->biCompression == 0x00 )
  579.                 {
  580.                     /* RGB DIB are coded from bottom to top */
  581.                     fmt.video.i_height =
  582.                         (unsigned int)(-(int)p_vids->p_bih->biHeight);
  583.                 }
  585.                 /* Extract palette from extradata if bpp <= 8
  586.                  * (assumes that extradata contains only palette but appears
  587.                  *  to be true for all palettized codecs we support) */
  588.                 if( fmt.video.i_bits_per_pixel > 0 && fmt.video.i_bits_per_pixel <= 8 )
  589.                 {
  590.                     /* The palette is not always included in biSize */
  591.                     fmt.i_extra = p_vids->i_chunk_size - sizeof(BITMAPINFOHEADER);
  592.                     if( fmt.i_extra > 0 )
  593.                     {
  594.                         const uint8_t *p_pal = fmt.p_extra;
  596.                         fmt.video.p_palette = calloc( 1, sizeof(video_palette_t) );
  597.                         fmt.video.p_palette->i_entries = __MIN(fmt.i_extra/4, 256);
  599.                         for( int i = 0; i < fmt.video.p_palette->i_entries; i++ )
  600.                         {
  601.                             for( int j = 0; j < 4; j++ )
  602.                                 fmt.video.p_palette->palette[i][j] = p_pal[4*i+j];
  603.                         }
  604.                     }
  605.                 }
  606.                 break;
  608.             case( AVIFOURCC_txts):
  609.                 tk->i_cat   = SPU_ES;
  610.                 tk->i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
  611.                 msg_Dbg( p_demux, "stream[%d] subtitles", i );
  612.                 es_format_Init( &fmt, SPU_ES, tk->i_codec );
  613.                 break;
  615.             case( AVIFOURCC_iavs):
  616.             case( AVIFOURCC_ivas):
  617.                 p_sys->b_muxed = true;
  618.                 msg_Dbg( p_demux, "stream[%d] iavs with handler %4.4s", i, (char *)&p_strh->i_handler );
  619.                 if( p_strh->i_handler == FOURCC_dvsd ||
  620.                     p_strh->i_handler == FOURCC_dvhd ||
  621.                     p_strh->i_handler == FOURCC_dvsl ||
  622.                     p_strh->i_handler == FOURCC_dv25 ||
  623.                     p_strh->i_handler == FOURCC_dv50 )
  624.                 {
  625.                     tk->p_out_muxed = stream_DemuxNew( p_demux, (char *)"rawdv", p_demux->out );
  626.                     if( !tk->p_out_muxed )
  627.                         msg_Err( p_demux, "could not load the DV parser" );
  628.                     else break;
  629.                 }
  630.                 free( tk );
  631.                 continue;
  633.             case( AVIFOURCC_mids):
  634.                 msg_Dbg( p_demux, "stream[%d] midi is UNSUPPORTED", i );
  636.             default:
  637.                 msg_Warn( p_demux, "stream[%d] unknown type %4.4s", i, (char *)&p_strh->i_type );
  638.                 free( tk );
  639.                 continue;
  640.         }
  641.         if( p_strn )
  642.             fmt.psz_description = FromLatin1( p_strn->p_str );
  643.         if( tk->p_out_muxed == NULL )
  644.             tk->p_es = es_out_Add( p_demux->out, &fmt );
  645.         TAB_APPEND( p_sys->i_track, p_sys->track, tk );
  646.     }
  648.     if( p_sys->i_track <= 0 )
  649.     {
  650.         msg_Err( p_demux, "no valid track" );
  651.         goto error;
  652.     }
  654.     i_do_index =  config_GetInt( p_demux, "avi-index" );
  655.     if( i_do_index == 1 ) /* Always fix */
  656.     {
  657. aviindex:
  658.         if( p_sys->b_seekable )
  659.         {
  660.             AVI_IndexCreate( p_demux );
  661.         }
  662.         else
  663.         {
  664.             msg_Warn( p_demux, "cannot create index (unseekable stream)" );
  665.             AVI_IndexLoad( p_demux );
  666.         }
  667.     }
  668.     else
  669.     {
  670.         AVI_IndexLoad( p_demux );
  671.     }
  673.     /* *** movie length in sec *** */
  674.     p_sys->i_length = AVI_MovieGetLength( p_demux );
  676.     /* Check the index completeness */
  677.     unsigned int i_idx_totalframes = 0;
  678.     for( unsigned int i = 0; i < p_sys->i_track; i++ )
  679.     {
  680.         const avi_track_t *tk = p_sys->track[i];
  681.         if( tk->i_cat == VIDEO_ES && tk->p_index )
  682.             i_idx_totalframes = __MAX(i_idx_totalframes, tk->i_idxnb);
  683.             continue;
  684.     }
  685.     if( i_idx_totalframes != p_avih->i_totalframes &&
  686.         p_sys->i_length < (mtime_t)p_avih->i_totalframes *
  687.                           (mtime_t)p_avih->i_microsecperframe /
  688.                           (mtime_t)1000000 )
  689.     {
  690.         if( !vlc_object_alive( p_demux) )
  691.             goto error;
  693.         msg_Warn( p_demux, "broken or missing index, 'seek' will be "
  694.                            "approximative or will exhibit strange behavior" );
  695.         if( i_do_index == 0 && !b_index )
  696.         {
  697.             if( !p_sys->b_seekable ) {
  698.                 b_index = true;
  699.                 goto aviindex;
  700.             }
  701.             switch( dialog_Question( p_demux, _("AVI Index") ,
  702.                _( "This AVI file is broken. Seeking will not work correctly.n"
  703.                   "Do you want to try to fix it?nn"
  704.                   "This might take a long time." ),
  705.                   _( "Repair" ), _( "Don't repair" ), _( "Cancel") ) )
  706.             {
  707.                 case 1:
  708.                     b_index = true;
  709.                     msg_Dbg( p_demux, "Fixing AVI index" );
  710.                     goto aviindex;
  711.                 case 3:
  712.                     /* Kill input */
  713.                     vlc_object_kill( p_demux->p_parent );
  714.                     goto error;
  715.             }
  716.         }
  717.     }
  719.     /* fix some BeOS MediaKit generated file */
  720.     for( i = 0 ; i < p_sys->i_track; i++ )
  721.     {
  722.         avi_track_t         *tk = p_sys->track[i];
  723.         avi_chunk_list_t    *p_strl;
  724.         avi_chunk_strh_t    *p_strh;
  725.         avi_chunk_strf_auds_t    *p_auds;
  727.         if( tk->i_cat != AUDIO_ES )
  728.         {
  729.             continue;
  730.         }
  731.         if( tk->i_idxnb < 1 ||
  732.             tk->i_scale != 1 ||
  733.             tk->i_samplesize != 0 )
  734.         {
  735.             continue;
  736.         }
  737.         p_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i );
  738.         p_strh = AVI_ChunkFind( p_strl, AVIFOURCC_strh, 0 );
  739.         p_auds = AVI_ChunkFind( p_strl, AVIFOURCC_strf, 0 );
  741.         if( p_auds->p_wf->wFormatTag != WAVE_FORMAT_PCM &&
  742.             (unsigned int)tk->i_rate == p_auds->p_wf->nSamplesPerSec )
  743.         {
  744.             int64_t i_track_length =
  745.                 tk->p_index[tk->i_idxnb-1].i_length +
  746.                 tk->p_index[tk->i_idxnb-1].i_lengthtotal;
  747.             mtime_t i_length = (mtime_t)p_avih->i_totalframes *
  748.                                (mtime_t)p_avih->i_microsecperframe;
  750.             if( i_length == 0 )
  751.             {
  752.                 msg_Warn( p_demux, "track[%d] cannot be fixed (BeOS MediaKit generated)", i );
  753.                 continue;
  754.             }
  755.             tk->i_samplesize = 1;
  756.             tk->i_rate       = i_track_length  * (int64_t)1000000/ i_length;
  757.             msg_Warn( p_demux, "track[%d] fixed with rate=%d scale=%d (BeOS MediaKit generated)", i, tk->i_rate, tk->i_scale );
  758.         }
  759.     }
  761.     if( p_sys->b_seekable )
  762.     {
  763.         /* we have read all chunk so go back to movi */
  764.         stream_Seek( p_demux->s, p_movi->i_chunk_pos );
  765.     }
  766.     /* Skip movi header */
  767.     stream_Read( p_demux->s, NULL, 12 );
  769.     p_sys->i_movi_begin = p_movi->i_chunk_pos;
  770.     return VLC_SUCCESS;
  772. error:
  773.     if( p_sys->meta )
  774.     {
  775.         vlc_meta_Delete( p_sys->meta );
  776.     }
  777.     AVI_ChunkFreeRoot( p_demux->s, &p_sys->ck_root );
  778.     free( p_sys );
  779.     return VLC_EGENERIC;
  780. }
  782. /*****************************************************************************
  783.  * Close: frees unused data
  784.  *****************************************************************************/
  785. static void Close ( vlc_object_t * p_this )
  786. {
  787.     demux_t *    p_demux = (demux_t *)p_this;
  788.     unsigned int i;
  789.     demux_sys_t *p_sys = p_demux->p_sys  ;
  791.     for( i = 0; i < p_sys->i_track; i++ )
  792.     {
  793.         if( p_sys->track[i] )
  794.         {
  795.             if( p_sys->track[i]->p_out_muxed )
  796.                 stream_Delete( p_sys->track[i]->p_out_muxed );
  797.             free( p_sys->track[i]->p_index );
  798.             free( p_sys->track[i]->p_extra );
  799.             free( p_sys->track[i] );
  800.         }
  801.     }
  802.     free( p_sys->track );
  803.     AVI_ChunkFreeRoot( p_demux->s, &p_sys->ck_root );
  804.     vlc_meta_Delete( p_sys->meta );
  806.     free( p_sys );
  807. }
  809. /*****************************************************************************
  810.  * Demux_Seekable: reads and demuxes data packets for stream seekable
  811.  *****************************************************************************
  812.  * AVIDemux: reads and demuxes data packets
  813.  *****************************************************************************
  814.  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
  815.  *****************************************************************************/
  816. typedef struct
  817. {
  818.     bool b_ok;
  820.     int i_toread;
  822.     off_t i_posf; /* where we will read :
  823.                    if i_idxposb == 0 : begining of chunk (+8 to acces data)
  824.                    else : point on data directly */
  825. } avi_track_toread_t;
  827. static int Demux_Seekable( demux_t *p_demux )
  828. {
  829.     demux_sys_t *p_sys = p_demux->p_sys;
  831.     unsigned int i_track_count = 0;
  832.     unsigned int i_track;
  833.     bool b_stream;
  834.     /* cannot be more than 100 stream (dcXX or wbXX) */
  835.     avi_track_toread_t toread[100];
  838.     /* detect new selected/unselected streams */
  839.     for( i_track = 0; i_track < p_sys->i_track; i_track++ )
  840.     {
  841.         avi_track_t *tk = p_sys->track[i_track];
  842.         bool  b;
  844.         if( p_sys->b_muxed && tk->p_out_muxed )
  845.         {
  846.             i_track_count++;
  847.             tk->b_activated = true;
  848.             continue;
  849.         }
  851.         es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b );
  852.         if( b && !tk->b_activated )
  853.         {
  854.             if( p_sys->b_seekable)
  855.             {
  856.                 AVI_TrackSeek( p_demux, i_track, p_sys->i_time );
  857.             }
  858.             tk->b_activated = true;
  859.         }
  860.         else if( !b && tk->b_activated )
  861.         {
  862.             tk->b_activated = false;
  863.         }
  864.         if( b )
  865.         {
  866.             i_track_count++;
  867.         }
  868.     }
  870.     if( i_track_count <= 0 )
  871.     {
  872.         int64_t i_length = p_sys->i_length * (mtime_t)1000000;
  874.         p_sys->i_time += 25*1000;  /* read 25ms */
  875.         if( i_length > 0 )
  876.         {
  877.             if( p_sys->i_time >= i_length )
  878.                 return 0;
  879.             return 1;
  880.         }
  881.         msg_Warn( p_demux, "no track selected, exiting..." );
  882.         return 0;
  883.     }
  885.     /* wait for the good time */
  886.     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_time + 1 );
  887.     p_sys->i_time += 25*1000;  /* read 25ms */
  889.     /* init toread */
  890.     for( i_track = 0; i_track < p_sys->i_track; i_track++ )
  891.     {
  892.         avi_track_t *tk = p_sys->track[i_track];
  893.         mtime_t i_dpts;
  895.         toread[i_track].b_ok = tk->b_activated && !tk->b_eof;
  896.         if( tk->i_idxposc < tk->i_idxnb )
  897.         {
  898.             toread[i_track].i_posf = tk->p_index[tk->i_idxposc].i_pos;
  899.            if( tk->i_idxposb > 0 )
  900.            {
  901.                 toread[i_track].i_posf += 8 + tk->i_idxposb;
  902.            }
  903.         }
  904.         else
  905.         {
  906.             toread[i_track].i_posf = -1;
  907.         }
  909.         i_dpts = p_sys->i_time - AVI_GetPTS( tk  );
  911.         if( tk->i_samplesize )
  912.         {
  913.             toread[i_track].i_toread = AVI_PTSToByte( tk, __ABS( i_dpts ) );
  914.         }
  915.         else
  916.         {
  917.             toread[i_track].i_toread = AVI_PTSToChunk( tk, __ABS( i_dpts ) );
  918.         }
  920.         if( i_dpts < 0 )
  921.         {
  922.             toread[i_track].i_toread *= -1;
  923.         }
  924.     }
  926.     b_stream = false;
  928.     for( ;; )
  929.     {
  930.         avi_track_t     *tk;
  931.         bool       b_done;
  932.         block_t         *p_frame;
  933.         off_t i_pos;
  934.         unsigned int i;
  935.         size_t i_size;
  937.         /* search for first chunk to be read */
  938.         for( i = 0, b_done = true, i_pos = -1; i < p_sys->i_track; i++ )
  939.         {
  940.             if( !toread[i].b_ok ||
  941.                 AVI_GetDPTS( p_sys->track[i],
  942.                              toread[i].i_toread ) <= -25 * 1000 )
  943.             {
  944.                 continue;
  945.             }
  947.             if( toread[i].i_toread > 0 )
  948.             {
  949.                 b_done = false; /* not yet finished */
  950.             }
  951.             if( toread[i].i_posf > 0 )
  952.             {
  953.                 if( i_pos == -1 || i_pos > toread[i].i_posf )
  954.                 {
  955.                     i_track = i;
  956.                     i_pos = toread[i].i_posf;
  957.                 }
  958.             }
  959.         }
  961.         if( b_done )
  962.         {
  963.             for( i = 0; i < p_sys->i_track; i++ )
  964.             {
  965.                 if( toread[i].b_ok )
  966.                     return 1;
  967.             }
  968.             msg_Warn( p_demux, "all tracks have failed, exiting..." );
  969.             return 0;
  970.         }
  972.         if( i_pos == -1 )
  973.         {
  974.             int i_loop_count = 0;
  976.             /* no valid index, we will parse directly the stream
  977.              * in case we fail we will disable all finished stream */
  978.             if( p_sys->i_movi_lastchunk_pos >= p_sys->i_movi_begin + 12 )
  979.             {
  980.                 stream_Seek( p_demux->s, p_sys->i_movi_lastchunk_pos );
  981.                 if( AVI_PacketNext( p_demux ) )
  982.                 {
  983.                     return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  984.                 }
  985.             }
  986.             else
  987.             {
  988.                 stream_Seek( p_demux->s, p_sys->i_movi_begin + 12 );
  989.             }
  991.             for( ;; )
  992.             {
  993.                 avi_packet_t avi_pk;
  995.                 if( AVI_PacketGetHeader( p_demux, &avi_pk ) )
  996.                 {
  997.                     msg_Warn( p_demux,
  998.                              "cannot get packet header, track disabled" );
  999.                     return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  1000.                 }
  1001.                 if( avi_pk.i_stream >= p_sys->i_track ||
  1002.                     ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
  1003.                 {
  1004.                     if( AVI_PacketNext( p_demux ) )
  1005.                     {
  1006.                         msg_Warn( p_demux,
  1007.                                   "cannot skip packet, track disabled" );
  1008.                         return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  1009.                     }
  1011.                     /* Prevents from eating all the CPU with broken files.
  1012.                      * This value should be low enough so that it doesn't
  1013.                      * affect the reading speed too much. */
  1014.                     if( !(++i_loop_count % 1024) )
  1015.                     {
  1016.                         if( !vlc_object_alive (p_demux) ) return -1;
  1017.                         msleep( 10000 );
  1019.                         if( !(i_loop_count % (1024 * 10)) )
  1020.                             msg_Warn( p_demux,
  1021.                                       "don't seem to find any data..." );
  1022.                     }
  1023.                     continue;
  1024.                 }
  1025.                 else
  1026.                 {
  1027.                     /* add this chunk to the index */
  1028.                     avi_entry_t index;
  1030.                     index.i_id = avi_pk.i_fourcc;
  1031.                     index.i_flags =
  1032.                        AVI_GetKeyFlag(p_sys->track[avi_pk.i_stream]->i_codec,
  1033.                                       avi_pk.i_peek);
  1034.                     index.i_pos = avi_pk.i_pos;
  1035.                     index.i_length = avi_pk.i_size;
  1036.                     AVI_IndexAddEntry( p_sys, avi_pk.i_stream, &index );
  1038.                     i_track = avi_pk.i_stream;
  1039.                     tk = p_sys->track[i_track];
  1040.                     /* do we will read this data ? */
  1041.                     if( AVI_GetDPTS( tk, toread[i_track].i_toread ) > -25*1000 )
  1042.                     {
  1043.                         break;
  1044.                     }
  1045.                     else
  1046.                     {
  1047.                         if( AVI_PacketNext( p_demux ) )
  1048.                         {
  1049.                             msg_Warn( p_demux,
  1050.                                       "cannot skip packet, track disabled" );
  1051.                             return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  1052.                         }
  1053.                     }
  1054.                 }
  1055.             }
  1057.         }
  1058.         else
  1059.         {
  1060.             stream_Seek( p_demux->s, i_pos );
  1061.         }
  1063.         /* Set the track to use */
  1064.         tk = p_sys->track[i_track];
  1066.         /* read thoses data */
  1067.         if( tk->i_samplesize )
  1068.         {
  1069.             unsigned int i_toread;
  1071.             if( ( i_toread = toread[i_track].i_toread ) <= 0 )
  1072.             {
  1073.                 if( tk->i_samplesize > 1 )
  1074.                 {
  1075.                     i_toread = tk->i_samplesize;
  1076.                 }
  1077.                 else
  1078.                 {
  1079.                     i_toread = AVI_PTSToByte( tk, 20 * 1000 );
  1080.                     i_toread = __MAX( i_toread, 100 );
  1081.                 }
  1082.             }
  1083.             i_size = __MIN( tk->p_index[tk->i_idxposc].i_length -
  1084.                                 tk->i_idxposb,
  1085.                             i_toread );
  1086.         }
  1087.         else
  1088.         {
  1089.             i_size = tk->p_index[tk->i_idxposc].i_length;
  1090.         }
  1092.         if( tk->i_idxposb == 0 )
  1093.         {
  1094.             i_size += 8; /* need to read and skip header */
  1095.         }
  1097.         if( ( p_frame = stream_Block( p_demux->s, __EVEN( i_size ) ) )==NULL )
  1098.         {
  1099.             msg_Warn( p_demux, "failed reading data" );
  1100.             tk->b_eof = false;
  1101.             toread[i_track].b_ok = false;
  1102.             continue;
  1103.         }
  1104.         if( i_size % 2 )    /* read was padded on word boundary */
  1105.         {
  1106.             p_frame->i_buffer--;
  1107.         }
  1108.         /* skip header */
  1109.         if( tk->i_idxposb == 0 )
  1110.         {
  1111.             p_frame->p_buffer += 8;
  1112.             p_frame->i_buffer -= 8;
  1113.         }
  1114.         p_frame->i_pts = AVI_GetPTS( tk ) + 1;
  1115.         if( tk->p_index[tk->i_idxposc].i_flags&AVIIF_KEYFRAME )
  1116.         {
  1117.             p_frame->i_flags = BLOCK_FLAG_TYPE_I;
  1118.         }
  1119.         else
  1120.         {
  1121.             p_frame->i_flags = BLOCK_FLAG_TYPE_PB;
  1122.         }
  1124.         /* read data */
  1125.         if( tk->i_samplesize )
  1126.         {
  1127.             if( tk->i_idxposb == 0 )
  1128.             {
  1129.                 i_size -= 8;
  1130.             }
  1131.             toread[i_track].i_toread -= i_size;
  1132.             tk->i_idxposb += i_size;
  1133.             if( tk->i_idxposb >=
  1134.                     tk->p_index[tk->i_idxposc].i_length )
  1135.             {
  1136.                 tk->i_idxposb = 0;
  1137.                 tk->i_idxposc++;
  1138.             }
  1139.         }
  1140.         else
  1141.         {
  1142.             int i_length = tk->p_index[tk->i_idxposc].i_length;
  1144.             tk->i_idxposc++;
  1145.             if( tk->i_cat == AUDIO_ES )
  1146.             {
  1147.                 tk->i_blockno += tk->i_blocksize > 0 ? ( i_length + tk->i_blocksize - 1 ) / tk->i_blocksize : 1;
  1148.             }
  1149.             toread[i_track].i_toread--;
  1150.         }
  1152.         if( tk->i_idxposc < tk->i_idxnb)
  1153.         {
  1154.             toread[i_track].i_posf =
  1155.                 tk->p_index[tk->i_idxposc].i_pos;
  1156.             if( tk->i_idxposb > 0 )
  1157.             {
  1158.                 toread[i_track].i_posf += 8 + tk->i_idxposb;
  1159.             }
  1161.         }
  1162.         else
  1163.         {
  1164.             toread[i_track].i_posf = -1;
  1165.         }
  1167.         b_stream = true; /* at least one read succeed */
  1169.         if( tk->i_cat != VIDEO_ES )
  1170.             p_frame->i_dts = p_frame->i_pts;
  1171.         else
  1172.         {
  1173.             p_frame->i_dts = p_frame->i_pts;
  1174.             p_frame->i_pts = 0;
  1175.         }
  1177.         //p_pes->i_rate = p_demux->stream.control.i_rate;
  1178.         if( tk->p_out_muxed )
  1179.             stream_DemuxSend( tk->p_out_muxed, p_frame );
  1180.         else
  1181.             es_out_Send( p_demux->out, tk->p_es, p_frame );
  1182.     }
  1183. }
  1186. /*****************************************************************************
  1187.  * Demux_UnSeekable: reads and demuxes data packets for unseekable file
  1188.  *****************************************************************************
  1189.  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
  1190.  *****************************************************************************/
  1191. static int Demux_UnSeekable( demux_t *p_demux )
  1192. {
  1193.     demux_sys_t     *p_sys = p_demux->p_sys;
  1194.     avi_track_t *p_stream_master = NULL;
  1195.     unsigned int i_stream;
  1196.     unsigned int i_packet;
  1198.     if( p_sys->b_muxed )
  1199.     {
  1200.         msg_Err( p_demux, "Can not yet process muxed avi substreams without seeking" );
  1201.         return VLC_EGENERIC;
  1202.     }
  1204.     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_time + 1 );
  1206.     /* *** find master stream for data packet skipping algo *** */
  1207.     /* *** -> first video, if any, or first audio ES *** */
  1208.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  1209.     {
  1210.         avi_track_t *tk = p_sys->track[i_stream];
  1211.         bool  b;
  1213.         es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b );
  1215.         if( b && tk->i_cat == VIDEO_ES )
  1216.         {
  1217.             p_stream_master = tk;
  1218.         }
  1219.         else if( b )
  1220.         {
  1221.             p_stream_master = tk;
  1222.         }
  1223.     }
  1225.     if( !p_stream_master )
  1226.     {
  1227.         msg_Warn( p_demux, "no more stream selected" );
  1228.         return( 0 );
  1229.     }
  1231.     p_sys->i_time = AVI_GetPTS( p_stream_master );
  1233.     for( i_packet = 0; i_packet < 10; i_packet++)
  1234.     {
  1235. #define p_stream    p_sys->track[avi_pk.i_stream]
  1237.         avi_packet_t    avi_pk;
  1239.         if( AVI_PacketGetHeader( p_demux, &avi_pk ) )
  1240.         {
  1241.             return( 0 );
  1242.         }
  1244.         if( avi_pk.i_stream >= p_sys->i_track ||
  1245.             ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
  1246.         {
  1247.             /* we haven't found an audio or video packet:
  1248.              *  - we have seek, found first next packet
  1249.              *  - others packets could be found, skip them
  1250.              */
  1251.             switch( avi_pk.i_fourcc )
  1252.             {
  1253.                 case AVIFOURCC_JUNK:
  1254.                 case AVIFOURCC_LIST:
  1255.                 case AVIFOURCC_RIFF:
  1256.                     return( !AVI_PacketNext( p_demux ) ? 1 : 0 );
  1257.                 case AVIFOURCC_idx1:
  1258.                     if( p_sys->b_odml )
  1259.                     {
  1260.                         return( !AVI_PacketNext( p_demux ) ? 1 : 0 );
  1261.                     }
  1262.                     return( 0 );    /* eof */
  1263.                 default:
  1264.                     msg_Warn( p_demux,
  1265.                               "seems to have lost position, resync" );
  1266.                     if( AVI_PacketSearch( p_demux ) )
  1267.                     {
  1268.                         msg_Err( p_demux, "resync failed" );
  1269.                         return( -1 );
  1270.                     }
  1271.             }
  1272.         }
  1273.         else
  1274.         {
  1275.             /* check for time */
  1276.             if( __ABS( AVI_GetPTS( p_stream ) -
  1277.                         AVI_GetPTS( p_stream_master ) )< 600*1000 )
  1278.             {
  1279.                 /* load it and send to decoder */
  1280.                 block_t *p_frame;
  1281.                 if( AVI_PacketRead( p_demux, &avi_pk, &p_frame ) || p_frame == NULL )
  1282.                 {
  1283.                     return( -1 );
  1284.                 }
  1285.                 p_frame->i_pts = AVI_GetPTS( p_stream ) + 1;
  1287.                 if( avi_pk.i_cat != VIDEO_ES )
  1288.                     p_frame->i_dts = p_frame->i_pts;
  1289.                 else
  1290.                 {
  1291.                     p_frame->i_dts = p_frame->i_pts;
  1292.                     p_frame->i_pts = 0;
  1293.                 }
  1295.                 //p_pes->i_rate = p_demux->stream.control.i_rate;
  1296.                 es_out_Send( p_demux->out, p_stream->p_es, p_frame );
  1297.             }
  1298.             else
  1299.             {
  1300.                 if( AVI_PacketNext( p_demux ) )
  1301.                 {
  1302.                     return( 0 );
  1303.                 }
  1304.             }
  1306.             /* *** update stream time position *** */
  1307.             if( p_stream->i_samplesize )
  1308.             {
  1309.                 p_stream->i_idxposb += avi_pk.i_size;
  1310.             }
  1311.             else
  1312.             {
  1313.                 if( p_stream->i_cat == AUDIO_ES )
  1314.                 {
  1315.                     p_stream->i_blockno += p_stream->i_blocksize > 0 ? ( avi_pk.i_size + p_stream->i_blocksize - 1 ) / p_stream->i_blocksize : 1;
  1316.                 }
  1317.                 p_stream->i_idxposc++;
  1318.             }
  1320.         }
  1321. #undef p_stream
  1322.     }
  1324.     return( 1 );
  1325. }
  1327. /*****************************************************************************
  1328.  * Seek: goto to i_date or i_percent
  1329.  *****************************************************************************/
  1330. static int Seek( demux_t *p_demux, mtime_t i_date, int i_percent )
  1331. {
  1333.     demux_sys_t *p_sys = p_demux->p_sys;
  1334.     unsigned int i_stream;
  1335.     msg_Dbg( p_demux, "seek requested: %"PRId64" seconds %d%%",
  1336.              i_date / 1000000, i_percent );
  1338.     if( p_sys->b_seekable )
  1339.     {
  1340.         if( !p_sys->i_length )
  1341.         {
  1342.             avi_track_t *p_stream;
  1343.             int64_t i_pos;
  1345.             /* use i_percent to create a true i_date */
  1346.             msg_Warn( p_demux, "seeking without index at %d%%"
  1347.                       " only works for interleaved files", i_percent );
  1348.             if( i_percent >= 100 )
  1349.             {
  1350.                 msg_Warn( p_demux, "cannot seek so far !" );
  1351.                 return VLC_EGENERIC;
  1352.             }
  1353.             i_percent = __MAX( i_percent, 0 );
  1355.             /* try to find chunk that is at i_percent or the file */
  1356.             i_pos = __MAX( i_percent * stream_Size( p_demux->s ) / 100,
  1357.                            p_sys->i_movi_begin );
  1358.             /* search first selected stream (and prefer non eof ones) */
  1359.             for( i_stream = 0, p_stream = NULL;
  1360.                         i_stream < p_sys->i_track; i_stream++ )
  1361.             {
  1362.                 if( !p_stream || p_stream->b_eof )
  1363.                     p_stream = p_sys->track[i_stream];
  1365.                 if( p_stream->b_activated && !p_stream->b_eof )
  1366.                     break;
  1367.             }
  1368.             if( !p_stream || !p_stream->b_activated )
  1369.             {
  1370.                 msg_Warn( p_demux, "cannot find any selected stream" );
  1371.                 return VLC_EGENERIC;
  1372.             }
  1374.             /* be sure that the index exist */
  1375.             if( AVI_StreamChunkSet( p_demux, i_stream, 0 ) )
  1376.             {
  1377.                 msg_Warn( p_demux, "cannot seek" );
  1378.                 return VLC_EGENERIC;
  1379.             }
  1381.             while( i_pos >= p_stream->p_index[p_stream->i_idxposc].i_pos +
  1382.                p_stream->p_index[p_stream->i_idxposc].i_length + 8 )
  1383.             {
  1384.                 /* search after i_idxposc */
  1385.                 if( AVI_StreamChunkSet( p_demux,
  1386.                                         i_stream, p_stream->i_idxposc + 1 ) )
  1387.                 {
  1388.                     msg_Warn( p_demux, "cannot seek" );
  1389.                     return VLC_EGENERIC;
  1390.                 }
  1391.             }
  1393.             i_date = AVI_GetPTS( p_stream );
  1394.             /* TODO better support for i_samplesize != 0 */
  1395.             msg_Dbg( p_demux, "estimate date %"PRId64, i_date );
  1396.         }
  1398.         /* */
  1399.         for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  1400.         {
  1401.             avi_track_t *p_stream = p_sys->track[i_stream];
  1403.             if( !p_stream->b_activated )
  1404.                 continue;
  1406.             p_stream->b_eof = AVI_TrackSeek( p_demux, i_stream, i_date ) != 0;
  1407.         }
  1408.         es_out_Control( p_demux->out, ES_OUT_SET_NEXT_DISPLAY_TIME, i_date );
  1409.         p_sys->i_time = i_date;
  1410.         msg_Dbg( p_demux, "seek: %"PRId64" seconds", p_sys->i_time /1000000 );
  1411.         return VLC_SUCCESS;
  1412.     }
  1413.     else
  1414.     {
  1415.         msg_Err( p_demux, "shouldn't yet be executed" );
  1416.         return VLC_EGENERIC;
  1417.     }
  1418. }
  1420. /*****************************************************************************
  1421.  * Control:
  1422.  *****************************************************************************/
  1423. static double ControlGetPosition( demux_t *p_demux )
  1424. {
  1425.     demux_sys_t *p_sys = p_demux->p_sys;
  1427.     if( p_sys->i_length > 0 )
  1428.     {
  1429.         return (double)p_sys->i_time / (double)( p_sys->i_length * (mtime_t)1000000 );
  1430.     }
  1431.     else if( stream_Size( p_demux->s ) > 0 )
  1432.     {
  1433.         unsigned int i;
  1434.         int64_t i_tmp;
  1435.         int64_t i64 = 0;
  1437.         /* search the more advanced selected es */
  1438.         for( i = 0; i < p_sys->i_track; i++ )
  1439.         {
  1440.             avi_track_t *tk = p_sys->track[i];
  1441.             if( tk->b_activated && tk->i_idxposc < tk->i_idxnb )
  1442.             {
  1443.                 i_tmp = tk->p_index[tk->i_idxposc].i_pos +
  1444.                         tk->p_index[tk->i_idxposc].i_length + 8;
  1445.                 if( i_tmp > i64 )
  1446.                 {
  1447.                     i64 = i_tmp;
  1448.                 }
  1449.             }
  1450.         }
  1451.         return (double)i64 / stream_Size( p_demux->s );
  1452.     }
  1453.     return 0.0;
  1454. }
  1456. static int Control( demux_t *p_demux, int i_query, va_list args )
  1457. {
  1458.     demux_sys_t *p_sys = p_demux->p_sys;
  1459.     int i;
  1460.     double   f, *pf;
  1461.     int64_t i64, *pi64;
  1462.     vlc_meta_t *p_meta;
  1464.     switch( i_query )
  1465.     {
  1466.         case DEMUX_GET_POSITION:
  1467.             pf = (double*)va_arg( args, double * );
  1468.             *pf = ControlGetPosition( p_demux );
  1469.             return VLC_SUCCESS;
  1470.         case DEMUX_SET_POSITION:
  1471.             f = (double)va_arg( args, double );
  1472.             if( p_sys->b_seekable )
  1473.             {
  1474.                 i64 = (mtime_t)(1000000.0 * p_sys->i_length * f );
  1475.                 return Seek( p_demux, i64, (int)(f * 100) );
  1476.             }
  1477.             else
  1478.             {
  1479.                 int64_t i_pos = stream_Size( p_demux->s ) * f;
  1480.                 return stream_Seek( p_demux->s, i_pos );
  1481.             }
  1483.         case DEMUX_GET_TIME:
  1484.             pi64 = (int64_t*)va_arg( args, int64_t * );
  1485.             *pi64 = p_sys->i_time;
  1486.             return VLC_SUCCESS;
  1488.         case DEMUX_SET_TIME:
  1489.         {
  1490.             int i_percent = 0;
  1492.             i64 = (int64_t)va_arg( args, int64_t );
  1493.             if( p_sys->i_length > 0 )
  1494.             {
  1495.                 i_percent = 100 * i64 / (p_sys->i_length*1000000);
  1496.             }
  1497.             else if( p_sys->i_time > 0 )
  1498.             {
  1499.                 i_percent = (int)( 100.0 * ControlGetPosition( p_demux ) *
  1500.                                    (double)i64 / (double)p_sys->i_time );
  1501.             }
  1502.             return Seek( p_demux, i64, i_percent );
  1503.         }
  1504.         case DEMUX_GET_LENGTH:
  1505.             pi64 = (int64_t*)va_arg( args, int64_t * );
  1506.             *pi64 = p_sys->i_length * (mtime_t)1000000;
  1507.             return VLC_SUCCESS;
  1509.         case DEMUX_GET_FPS:
  1510.             pf = (double*)va_arg( args, double * );
  1511.             *pf = 0.0;
  1512.             for( i = 0; i < (int)p_sys->i_track; i++ )
  1513.             {
  1514.                 avi_track_t *tk = p_sys->track[i];
  1515.                 if( tk->i_cat == VIDEO_ES && tk->i_scale > 0)
  1516.                 {
  1517.                     *pf = (float)tk->i_rate / (float)tk->i_scale;
  1518.                     break;
  1519.                 }
  1520.             }
  1521.             return VLC_SUCCESS;
  1522.         case DEMUX_GET_META:
  1523.             p_meta = (vlc_meta_t*)va_arg( args, vlc_meta_t* );
  1524.             vlc_meta_Merge( p_meta,  p_sys->meta );
  1525.             return VLC_SUCCESS;
  1527.         default:
  1528.             return VLC_EGENERIC;
  1529.     }
  1530. }
  1532. /*****************************************************************************
  1533.  * Function to convert pts to chunk or byte
  1534.  *****************************************************************************/
  1536. static mtime_t AVI_PTSToChunk( avi_track_t *tk, mtime_t i_pts )
  1537. {
  1538.     if( !tk->i_scale )
  1539.         return (mtime_t)0;
  1541.     return (mtime_t)((int64_t)i_pts *
  1542.                      (int64_t)tk->i_rate /
  1543.                      (int64_t)tk->i_scale /
  1544.                      (int64_t)1000000 );
  1545. }
  1546. static mtime_t AVI_PTSToByte( avi_track_t *tk, mtime_t i_pts )
  1547. {
  1548.     if( !tk->i_scale || !tk->i_samplesize )
  1549.         return (mtime_t)0;
  1551.     return (mtime_t)((int64_t)i_pts *
  1552.                      (int64_t)tk->i_rate /
  1553.                      (int64_t)tk->i_scale /
  1554.                      (int64_t)1000000 *
  1555.                      (int64_t)tk->i_samplesize );
  1556. }
  1558. static mtime_t AVI_GetDPTS( avi_track_t *tk, int64_t i_count )
  1559. {
  1560.     mtime_t i_dpts = 0;
  1562.     if( !tk->i_rate )
  1563.         return i_dpts;
  1565.     i_dpts = (mtime_t)( (int64_t)1000000 *
  1566.                         (int64_t)i_count *
  1567.                         (int64_t)tk->i_scale /
  1568.                         (int64_t)tk->i_rate );
  1570.     if( tk->i_samplesize )
  1571.     {
  1572.         return i_dpts / tk->i_samplesize;
  1573.     }
  1574.     return i_dpts;
  1575. }
  1577. static mtime_t AVI_GetPTS( avi_track_t *tk )
  1578. {
  1579.     if( tk->i_samplesize )
  1580.     {
  1581.         int64_t i_count = 0;
  1583.         /* we need a valid entry we will emulate one */
  1584.         if( tk->i_idxposc == tk->i_idxnb )
  1585.         {
  1586.             if( tk->i_idxposc )
  1587.             {
  1588.                 /* use the last entry */
  1589.                 i_count = tk->p_index[tk->i_idxnb - 1].i_lengthtotal
  1590.                             + tk->p_index[tk->i_idxnb - 1].i_length;
  1591.             }
  1592.         }
  1593.         else
  1594.         {
  1595.             i_count = tk->p_index[tk->i_idxposc].i_lengthtotal;
  1596.         }
  1597.         return AVI_GetDPTS( tk, i_count + tk->i_idxposb );
  1598.     }
  1599.     else
  1600.     {
  1601.         if( tk->i_cat == AUDIO_ES )
  1602.         {
  1603.             return AVI_GetDPTS( tk, tk->i_blockno );
  1604.         }
  1605.         else
  1606.         {
  1607.             return AVI_GetDPTS( tk, tk->i_idxposc );
  1608.         }
  1609.     }
  1610. }
  1612. static int AVI_StreamChunkFind( demux_t *p_demux, unsigned int i_stream )
  1613. {
  1614.     demux_sys_t *p_sys = p_demux->p_sys;
  1615.     avi_packet_t avi_pk;
  1616.     int i_loop_count = 0;
  1618.     /* find first chunk of i_stream that isn't in index */
  1620.     if( p_sys->i_movi_lastchunk_pos >= p_sys->i_movi_begin + 12 )
  1621.     {
  1622.         stream_Seek( p_demux->s, p_sys->i_movi_lastchunk_pos );
  1623.         if( AVI_PacketNext( p_demux ) )
  1624.         {
  1625.             return VLC_EGENERIC;
  1626.         }
  1627.     }
  1628.     else
  1629.     {
  1630.         stream_Seek( p_demux->s, p_sys->i_movi_begin + 12 );
  1631.     }
  1633.     for( ;; )
  1634.     {
  1635.         if( !vlc_object_alive (p_demux) ) return VLC_EGENERIC;
  1637.         if( AVI_PacketGetHeader( p_demux, &avi_pk ) )
  1638.         {
  1639.             msg_Warn( p_demux, "cannot get packet header" );
  1640.             return VLC_EGENERIC;
  1641.         }
  1642.         if( avi_pk.i_stream >= p_sys->i_track ||
  1643.             ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
  1644.         {
  1645.             if( AVI_PacketNext( p_demux ) )
  1646.             {
  1647.                 return VLC_EGENERIC;
  1648.             }
  1650.             /* Prevents from eating all the CPU with broken files.
  1651.              * This value should be low enough so that it doesn't
  1652.              * affect the reading speed too much. */
  1653.             if( !(++i_loop_count % 1024) )
  1654.             {
  1655.                 if( !vlc_object_alive (p_demux) ) return VLC_EGENERIC;
  1656.                 msleep( 10000 );
  1658.                 if( !(i_loop_count % (1024 * 10)) )
  1659.                     msg_Warn( p_demux, "don't seem to find any data..." );
  1660.             }
  1661.         }
  1662.         else
  1663.         {
  1664.             /* add this chunk to the index */
  1665.             avi_entry_t index;
  1667.             index.i_id = avi_pk.i_fourcc;
  1668.             index.i_flags =
  1669.                AVI_GetKeyFlag(p_sys->track[avi_pk.i_stream]->i_codec,
  1670.                               avi_pk.i_peek);
  1671.             index.i_pos = avi_pk.i_pos;
  1672.             index.i_length = avi_pk.i_size;
  1673.             AVI_IndexAddEntry( p_sys, avi_pk.i_stream, &index );
  1675.             if( avi_pk.i_stream == i_stream  )
  1676.             {
  1677.                 return VLC_SUCCESS;
  1678.             }
  1680.             if( AVI_PacketNext( p_demux ) )
  1681.             {
  1682.                 return VLC_EGENERIC;
  1683.             }
  1684.         }
  1685.     }
  1686. }
  1688. /* be sure that i_ck will be a valid index entry */
  1689. static int AVI_StreamChunkSet( demux_t *p_demux, unsigned int i_stream,
  1690.                                unsigned int i_ck )
  1691. {
  1692.     demux_sys_t *p_sys = p_demux->p_sys;
  1693.     avi_track_t *p_stream = p_sys->track[i_stream];
  1695.     p_stream->i_idxposc = i_ck;
  1696.     p_stream->i_idxposb = 0;
  1698.     if(  i_ck >= p_stream->i_idxnb )
  1699.     {
  1700.         p_stream->i_idxposc = p_stream->i_idxnb - 1;
  1701.         do
  1702.         {
  1703.             p_stream->i_idxposc++;
  1704.             if( AVI_StreamChunkFind( p_demux, i_stream ) )
  1705.             {
  1706.                 return VLC_EGENERIC;
  1707.             }
  1709.         } while( p_stream->i_idxposc < i_ck );
  1710.     }
  1712.     return VLC_SUCCESS;
  1713. }
  1715. /* XXX FIXME up to now, we assume that all chunk are one after one */
  1716. static int AVI_StreamBytesSet( demux_t    *p_demux,
  1717.                                unsigned int i_stream,
  1718.                                off_t   i_byte )
  1719. {
  1720.     demux_sys_t *p_sys = p_demux->p_sys;
  1721.     avi_track_t *p_stream = p_sys->track[i_stream];
  1723.     if( ( p_stream->i_idxnb > 0 )
  1724.         &&( i_byte < p_stream->p_index[p_stream->i_idxnb - 1].i_lengthtotal +
  1725.                 p_stream->p_index[p_stream->i_idxnb - 1].i_length ) )
  1726.     {
  1727.         /* index is valid to find the ck */
  1728.         /* uses dichototmie to be fast enougth */
  1729.         int i_idxposc = __MIN( p_stream->i_idxposc, p_stream->i_idxnb - 1 );
  1730.         int i_idxmax  = p_stream->i_idxnb;
  1731.         int i_idxmin  = 0;
  1732.         for( ;; )
  1733.         {
  1734.             if( p_stream->p_index[i_idxposc].i_lengthtotal > i_byte )
  1735.             {
  1736.                 i_idxmax  = i_idxposc ;
  1737.                 i_idxposc = ( i_idxmin + i_idxposc ) / 2 ;
  1738.             }
  1739.             else
  1740.             {
  1741.                 if( p_stream->p_index[i_idxposc].i_lengthtotal +
  1742.                         p_stream->p_index[i_idxposc].i_length <= i_byte)
  1743.                 {
  1744.                     i_idxmin  = i_idxposc ;
  1745.                     i_idxposc = (i_idxmax + i_idxposc ) / 2 ;
  1746.                 }
  1747.                 else
  1748.                 {
  1749.                     p_stream->i_idxposc = i_idxposc;
  1750.                     p_stream->i_idxposb = i_byte -
  1751.                             p_stream->p_index[i_idxposc].i_lengthtotal;
  1752.                     return VLC_SUCCESS;
  1753.                 }
  1754.             }
  1755.         }
  1757.     }
  1758.     else
  1759.     {
  1760.         p_stream->i_idxposc = p_stream->i_idxnb - 1;
  1761.         p_stream->i_idxposb = 0;
  1762.         do
  1763.         {
  1764.             p_stream->i_idxposc++;
  1765.             if( AVI_StreamChunkFind( p_demux, i_stream ) )
  1766.             {
  1767.                 return VLC_EGENERIC;
  1768.             }
  1770.         } while( p_stream->p_index[p_stream->i_idxposc].i_lengthtotal +
  1771.                     p_stream->p_index[p_stream->i_idxposc].i_length <= i_byte );
  1773.         p_stream->i_idxposb = i_byte -
  1774.                        p_stream->p_index[p_stream->i_idxposc].i_lengthtotal;
  1775.         return VLC_SUCCESS;
  1776.     }
  1777. }
  1779. static int AVI_TrackSeek( demux_t *p_demux,
  1780.                            int i_stream,
  1781.                            mtime_t i_date )
  1782. {
  1783.     demux_sys_t  *p_sys = p_demux->p_sys;
  1784.     avi_track_t  *tk = p_sys->track[i_stream];
  1786. #define p_stream    p_sys->track[i_stream]
  1787.     mtime_t i_oldpts;
  1789.     i_oldpts = AVI_GetPTS( p_stream );
  1791.     if( !p_stream->i_samplesize )
  1792.     {
  1793.         if( AVI_StreamChunkSet( p_demux,
  1794.                                 i_stream,
  1795.                                 AVI_PTSToChunk( p_stream, i_date ) ) )
  1796.         {
  1797.             return VLC_EGENERIC;
  1798.         }
  1800.         if( p_stream->i_cat == AUDIO_ES )
  1801.         {
  1802.             unsigned int i;
  1803.             tk->i_blockno = 0;
  1804.             for( i = 0; i < tk->i_idxposc; i++ )
  1805.             {
  1806.                 if( tk->i_blocksize > 0 )
  1807.                 {
  1808.                     tk->i_blockno += ( tk->p_index[i].i_length + tk->i_blocksize - 1 ) / tk->i_blocksize;
  1809.                 }
  1810.                 else
  1811.                 {
  1812.                     tk->i_blockno++;
  1813.                 }
  1814.             }
  1815.         }
  1817.         msg_Dbg( p_demux,
  1818.                  "old:%"PRId64" %s new %"PRId64,
  1819.                  i_oldpts,
  1820.                  i_oldpts > i_date ? ">" : "<",
  1821.                  i_date );
  1823.         if( p_stream->i_cat == VIDEO_ES )
  1824.         {
  1825.             /* search key frame */
  1826.             //if( i_date < i_oldpts || 1 )
  1827.             {
  1828.                 while( p_stream->i_idxposc > 0 &&
  1829.                    !( p_stream->p_index[p_stream->i_idxposc].i_flags &
  1830.                                                                 AVIIF_KEYFRAME ) )
  1831.                 {
  1832.                     if( AVI_StreamChunkSet( p_demux,
  1833.                                             i_stream,
  1834.                                             p_stream->i_idxposc - 1 ) )
  1835.                     {
  1836.                         return VLC_EGENERIC;
  1837.                     }
  1838.                 }
  1839.             }
  1840. #if 0
  1841.             else
  1842.             {
  1843.                 while( p_stream->i_idxposc < p_stream->i_idxnb &&
  1844.                         !( p_stream->p_index[p_stream->i_idxposc].i_flags &
  1845.                                                                 AVIIF_KEYFRAME ) )
  1846.                 {
  1847.                     if( AVI_StreamChunkSet( p_demux,
  1848.                                             i_stream,
  1849.                                             p_stream->i_idxposc + 1 ) )
  1850.                     {
  1851.                         return VLC_EGENERIC;
  1852.                     }
  1853.                 }
  1854.             }
  1855. #endif
  1856.         }
  1857.     }
  1858.     else
  1859.     {
  1860.         if( AVI_StreamBytesSet( p_demux,
  1861.                                 i_stream,
  1862.                                 AVI_PTSToByte( p_stream, i_date ) ) )
  1863.         {
  1864.             return VLC_EGENERIC;
  1865.         }
  1866.     }
  1867.     return VLC_SUCCESS;
  1868. #undef p_stream
  1869. }
  1871. /****************************************************************************
  1872.  * Return true if it's a key frame
  1873.  ****************************************************************************/
  1874. static int AVI_GetKeyFlag( vlc_fourcc_t i_fourcc, uint8_t *p_byte )
  1875. {
  1876.     switch( i_fourcc )
  1877.     {
  1878.         case FOURCC_DIV1:
  1879.             /* we have:
  1880.              *  startcode:      0x00000100   32bits
  1881.              *  framenumber     ?             5bits
  1882.              *  piture type     0(I),1(P)     2bits
  1883.              */
  1884.             if( GetDWBE( p_byte ) != 0x00000100 )
  1885.             {
  1886.                 /* it's not an msmpegv1 stream, strange...*/
  1887.                 return AVIIF_KEYFRAME;
  1888.             }
  1889.             return p_byte[4] & 0x06 ? 0 : AVIIF_KEYFRAME;
  1891.         case FOURCC_DIV2:
  1892.         case FOURCC_DIV3:   /* wmv1 also */
  1893.             /* we have
  1894.              *  picture type    0(I),1(P)     2bits
  1895.              */
  1896.             return p_byte[0] & 0xC0 ? 0 : AVIIF_KEYFRAME;
  1897.         case FOURCC_mp4v:
  1898.             /* we should find first occurrence of 0x000001b6 (32bits)
  1899.              *  startcode:      0x000001b6   32bits
  1900.              *  piture type     0(I),1(P)     2bits
  1901.              */
  1902.             if( GetDWBE( p_byte ) != 0x000001b6 )
  1903.             {
  1904.                 /* not true , need to find the first VOP header */
  1905.                 return AVIIF_KEYFRAME;
  1906.             }
  1907.             return p_byte[4] & 0xC0 ? 0 : AVIIF_KEYFRAME;
  1909.         default:
  1910.             /* I can't do it, so say yes */
  1911.             return AVIIF_KEYFRAME;
  1912.     }
  1913. }
  1915. vlc_fourcc_t AVI_FourccGetCodec( unsigned int i_cat, vlc_fourcc_t i_codec )
  1916. {
  1917.     switch( i_cat )
  1918.     {
  1919.         case AUDIO_ES:
  1920.             wf_tag_to_fourcc( i_codec, &i_codec, NULL );
  1921.             return i_codec;
  1923.         case VIDEO_ES:
  1924.             /* XXX DIV1 <- msmpeg4v1, DIV2 <- msmpeg4v2, DIV3 <- msmpeg4v3, mp4v for mpeg4 */
  1925.             switch( i_codec )
  1926.             {
  1927.                 case FOURCC_1:
  1928.                     return VLC_FOURCC('m','r','l','e');
  1929.                 case FOURCC_DIV1:
  1930.                 case FOURCC_div1:
  1931.                 case FOURCC_MPG4:
  1932.                 case FOURCC_mpg4:
  1933.                     return FOURCC_DIV1;
  1934.                 case FOURCC_DIV2:
  1935.                 case FOURCC_div2:
  1936.                 case FOURCC_MP42:
  1937.                 case FOURCC_mp42:
  1938.                 case FOURCC_MPG3:
  1939.                 case FOURCC_mpg3:
  1940.                     return FOURCC_DIV2;
  1941.                 case FOURCC_div3:
  1942.                 case FOURCC_MP43:
  1943.                 case FOURCC_mp43:
  1944.                 case FOURCC_DIV3:
  1945.                 case FOURCC_DIV4:
  1946.                 case FOURCC_div4:
  1947.                 case FOURCC_DIV5:
  1948.                 case FOURCC_div5:
  1949.                 case FOURCC_DIV6:
  1950.                 case FOURCC_div6:
  1951.                 case FOURCC_AP41:
  1952.                 case FOURCC_3IV1:
  1953.                 case FOURCC_3iv1:
  1954.                 case FOURCC_3IVD:
  1955.                 case FOURCC_3ivd:
  1956.                 case FOURCC_3VID:
  1957.                 case FOURCC_3vid:
  1958.                     return FOURCC_DIV3;
  1959.                 case FOURCC_DIVX:
  1960.                 case FOURCC_divx:
  1961.                 case FOURCC_MP4S:
  1962.                 case FOURCC_mp4s:
  1963.                 case FOURCC_M4S2:
  1964.                 case FOURCC_m4s2:
  1965.                 case FOURCC_xvid:
  1966.                 case FOURCC_XVID:
  1967.                 case FOURCC_XviD:
  1968.                 case FOURCC_DX50:
  1969.                 case FOURCC_dx50:
  1970.                 case FOURCC_mp4v:
  1971.                 case FOURCC_4:
  1972.                 case FOURCC_3IV2:
  1973.                 case FOURCC_3iv2:
  1974.                     return FOURCC_mp4v;
  1975.             }
  1976.         default:
  1977.             return VLC_FOURCC( 'u', 'n', 'd', 'f' );
  1978.     }
  1979. }
  1981. /****************************************************************************
  1982.  *
  1983.  ****************************************************************************/
  1984. static void AVI_ParseStreamHeader( vlc_fourcc_t i_id,
  1985.                                    unsigned int *pi_number, unsigned int *pi_type )
  1986. {
  1987. #define SET_PTR( p, v ) if( p ) *(p) = (v);
  1988.     int c1, c2;
  1990.     c1 = ((uint8_t *)&i_id)[0];
  1991.     c2 = ((uint8_t *)&i_id)[1];
  1993.     if( c1 < '0' || c1 > '9' || c2 < '0' || c2 > '9' )
  1994.     {
  1995.         SET_PTR( pi_number, 100 ); /* > max stream number */
  1996.         SET_PTR( pi_type, UNKNOWN_ES );
  1997.     }
  1998.     else
  1999.     {
  2000.         SET_PTR( pi_number, (c1 - '0') * 10 + (c2 - '0' ) );
  2001.         switch( VLC_TWOCC( ((uint8_t *)&i_id)[2], ((uint8_t *)&i_id)[3] ) )
  2002.         {
  2003.             case AVITWOCC_wb:
  2004.                 SET_PTR( pi_type, AUDIO_ES );
  2005.                 break;
  2006.             case AVITWOCC_dc:
  2007.             case AVITWOCC_db:
  2008.             case AVITWOCC_AC:
  2009.                 SET_PTR( pi_type, VIDEO_ES );
  2010.                 break;
  2011.             default:
  2012.                 SET_PTR( pi_type, UNKNOWN_ES );
  2013.                 break;
  2014.         }
  2015.     }
  2016. #undef SET_PTR
  2017. }
  2019. /****************************************************************************
  2020.  *
  2021.  ****************************************************************************/
  2022. static int AVI_PacketGetHeader( demux_t *p_demux, avi_packet_t *p_pk )
  2023. {
  2024.     const uint8_t *p_peek;
  2026.     if( stream_Peek( p_demux->s, &p_peek, 16 ) < 16 )
  2027.     {
  2028.         return VLC_EGENERIC;
  2029.     }
  2030.     p_pk->i_fourcc  = VLC_FOURCC( p_peek[0], p_peek[1], p_peek[2], p_peek[3] );
  2031.     p_pk->i_size    = GetDWLE( p_peek + 4 );
  2032.     p_pk->i_pos     = stream_Tell( p_demux->s );
  2033.     if( p_pk->i_fourcc == AVIFOURCC_LIST || p_pk->i_fourcc == AVIFOURCC_RIFF )
  2034.     {
  2035.         p_pk->i_type = VLC_FOURCC( p_peek[8],  p_peek[9],
  2036.                                    p_peek[10], p_peek[11] );
  2037.     }
  2038.     else
  2039.     {
  2040.         p_pk->i_type = 0;
  2041.     }
  2043.     memcpy( p_pk->i_peek, p_peek + 8, 8 );
  2045.     AVI_ParseStreamHeader( p_pk->i_fourcc, &p_pk->i_stream, &p_pk->i_cat );
  2046.     return VLC_SUCCESS;
  2047. }
  2049. static int AVI_PacketNext( demux_t *p_demux )
  2050. {
  2051.     avi_packet_t    avi_ck;
  2052.     int             i_skip = 0;
  2054.     if( AVI_PacketGetHeader( p_demux, &avi_ck ) )
  2055.     {
  2056.         return VLC_EGENERIC;
  2057.     }
  2059.     if( avi_ck.i_fourcc == AVIFOURCC_LIST &&
  2060.         ( avi_ck.i_type == AVIFOURCC_rec || avi_ck.i_type == AVIFOURCC_movi ) )
  2061.     {
  2062.         i_skip = 12;
  2063.     }
  2064.     else if( avi_ck.i_fourcc == AVIFOURCC_RIFF &&
  2065.              avi_ck.i_type == AVIFOURCC_AVIX )
  2066.     {
  2067.         i_skip = 24;
  2068.     }
  2069.     else
  2070.     {
  2071.         i_skip = __EVEN( avi_ck.i_size ) + 8;
  2072.     }
  2074.     if( stream_Read( p_demux->s, NULL, i_skip ) != i_skip )
  2075.     {
  2076.         return VLC_EGENERIC;
  2077.     }
  2078.     return VLC_SUCCESS;
  2079. }
  2081. static int AVI_PacketRead( demux_t   *p_demux,
  2082.                            avi_packet_t     *p_pk,
  2083.                            block_t          **pp_frame )
  2084. {
  2085.     size_t i_size;
  2087.     i_size = __EVEN( p_pk->i_size + 8 );
  2089.     if( ( *pp_frame = stream_Block( p_demux->s, i_size ) ) == NULL )
  2090.     {
  2091.         return VLC_EGENERIC;
  2092.     }
  2093.     (*pp_frame)->p_buffer += 8;
  2094.     (*pp_frame)->i_buffer -= 8;
  2096.     if( i_size != p_pk->i_size + 8 )
  2097.     {
  2098.         (*pp_frame)->i_buffer--;
  2099.     }
  2101.     return VLC_SUCCESS;
  2102. }
  2104. static int AVI_PacketSearch( demux_t *p_demux )
  2105. {
  2106.     demux_sys_t     *p_sys = p_demux->p_sys;
  2107.     avi_packet_t    avi_pk;
  2108.     int             i_count = 0;
  2110.     for( ;; )
  2111.     {
  2112.         if( stream_Read( p_demux->s, NULL, 1 ) != 1 )
  2113.         {
  2114.             return VLC_EGENERIC;
  2115.         }
  2116.         AVI_PacketGetHeader( p_demux, &avi_pk );
  2117.         if( avi_pk.i_stream < p_sys->i_track &&
  2118.             ( avi_pk.i_cat == AUDIO_ES || avi_pk.i_cat == VIDEO_ES ) )
  2119.         {
  2120.             return VLC_SUCCESS;
  2121.         }
  2122.         switch( avi_pk.i_fourcc )
  2123.         {
  2124.             case AVIFOURCC_JUNK:
  2125.             case AVIFOURCC_LIST:
  2126.             case AVIFOURCC_RIFF:
  2127.             case AVIFOURCC_idx1:
  2128.                 return VLC_SUCCESS;
  2129.         }
  2131.         /* Prevents from eating all the CPU with broken files.
  2132.          * This value should be low enough so that it doesn't affect the
  2133.          * reading speed too much (not that we care much anyway because
  2134.          * this code is called only on broken files). */
  2135.         if( !(++i_count % 1024) )
  2136.         {
  2137.             if( !vlc_object_alive (p_demux) ) return VLC_EGENERIC;
  2139.             msleep( 10000 );
  2140.             if( !(i_count % (1024 * 10)) )
  2141.                 msg_Warn( p_demux, "trying to resync..." );
  2142.         }
  2143.     }
  2144. }
  2146. /****************************************************************************
  2147.  * Index stuff.
  2148.  ****************************************************************************/
  2149. static void AVI_IndexAddEntry( demux_sys_t *p_sys,
  2150.                                int i_stream,
  2151.                                avi_entry_t *p_index)
  2152. {
  2153.     avi_track_t *tk = p_sys->track[i_stream];
  2155.     /* Update i_movi_lastchunk_pos */
  2156.     if( p_sys->i_movi_lastchunk_pos < p_index->i_pos )
  2157.     {
  2158.         p_sys->i_movi_lastchunk_pos = p_index->i_pos;
  2159.     }
  2161.     /* add the entry */
  2162.     if( tk->i_idxnb >= tk->i_idxmax )
  2163.     {
  2164.         tk->i_idxmax += 16384;
  2165.         tk->p_index = realloc( tk->p_index,
  2166.                                tk->i_idxmax * sizeof( avi_entry_t ) );
  2167.         if( tk->p_index == NULL )
  2168.         {
  2169.             return;
  2170.         }
  2171.     }
  2172.     /* calculate cumulate length */
  2173.     if( tk->i_idxnb > 0 )
  2174.     {
  2175.         p_index->i_lengthtotal =
  2176.             tk->p_index[tk->i_idxnb - 1].i_length +
  2177.                 tk->p_index[tk->i_idxnb - 1].i_lengthtotal;
  2178.     }
  2179.     else
  2180.     {
  2181.         p_index->i_lengthtotal = 0;
  2182.     }
  2184.     tk->p_index[tk->i_idxnb++] = *p_index;
  2185. }
  2187. static int AVI_IndexLoad_idx1( demux_t *p_demux )
  2188. {
  2189.     demux_sys_t *p_sys = p_demux->p_sys;
  2191.     avi_chunk_list_t    *p_riff;
  2192.     avi_chunk_list_t    *p_movi;
  2193.     avi_chunk_idx1_t    *p_idx1;
  2195.     unsigned int i_stream;
  2196.     unsigned int i_index;
  2197.     off_t        i_offset;
  2198.     unsigned int i;
  2200.     bool b_keyset[100];
  2202.     p_riff = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0);
  2203.     p_idx1 = AVI_ChunkFind( p_riff, AVIFOURCC_idx1, 0);
  2204.     p_movi = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
  2206.     if( !p_idx1 )
  2207.     {
  2208.         msg_Warn( p_demux, "cannot find idx1 chunk, no index defined" );
  2209.         return VLC_EGENERIC;
  2210.     }
  2212.     /* *** calculate offset *** */
  2213.     /* Well, avi is __SHIT__ so test more than one entry
  2214.      * (needed for some avi files) */
  2215.     i_offset = 0;
  2216.     for( i = 0; i < __MIN( p_idx1->i_entry_count, 10 ); i++ )
  2217.     {
  2218.         if( p_idx1->entry[i].i_pos < p_movi->i_chunk_pos )
  2219.         {
  2220.             i_offset = p_movi->i_chunk_pos + 8;
  2221.             break;
  2222.         }
  2223.     }
  2225.     /* Reset b_keyset */
  2226.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2227.         b_keyset[i_stream] = false;
  2229.     for( i_index = 0; i_index < p_idx1->i_entry_count; i_index++ )
  2230.     {
  2231.         unsigned int i_cat;
  2233.         AVI_ParseStreamHeader( p_idx1->entry[i_index].i_fourcc,
  2234.                                &i_stream,
  2235.                                &i_cat );
  2236.         if( i_stream < p_sys->i_track &&
  2237.             i_cat == p_sys->track[i_stream]->i_cat )
  2238.         {
  2239.             avi_entry_t index;
  2240.             index.i_id      = p_idx1->entry[i_index].i_fourcc;
  2241.             index.i_flags   =
  2242.                 p_idx1->entry[i_index].i_flags&(~AVIIF_FIXKEYFRAME);
  2243.             index.i_pos     = p_idx1->entry[i_index].i_pos + i_offset;
  2244.             index.i_length  = p_idx1->entry[i_index].i_length;
  2245.             AVI_IndexAddEntry( p_sys, i_stream, &index );
  2247.             if( index.i_flags&AVIIF_KEYFRAME )
  2248.                 b_keyset[i_stream] = true;
  2249.         }
  2250.     }
  2252.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2253.     {
  2254.         if( !b_keyset[i_stream] )
  2255.         {
  2256.             avi_track_t *tk = p_sys->track[i_stream];
  2258.             msg_Dbg( p_demux, "no key frame set for track %d", i_stream );
  2259.             for( i_index = 0; i_index < tk->i_idxnb; i_index++ )
  2260.                 tk->p_index[i_index].i_flags |= AVIIF_KEYFRAME;
  2261.         }
  2262.     }
  2263.     return VLC_SUCCESS;
  2264. }
  2266. static void __Parse_indx( demux_t    *p_demux,
  2267.                           int               i_stream,
  2268.                           avi_chunk_indx_t  *p_indx )
  2269. {
  2270.     demux_sys_t         *p_sys    = p_demux->p_sys;
  2271.     avi_entry_t     index;
  2272.     int32_t             i;
  2274.     msg_Dbg( p_demux, "loading subindex(0x%x) %d entries", p_indx->i_indextype, p_indx->i_entriesinuse );
  2275.     if( p_indx->i_indexsubtype == 0 )
  2276.     {
  2277.         for( i = 0; i < p_indx->i_entriesinuse; i++ )
  2278.         {
  2279.             index.i_id      = p_indx->i_id;
  2280.             index.i_flags   = p_indx->idx.std[i].i_size & 0x80000000 ? 0 : AVIIF_KEYFRAME;
  2281.             index.i_pos     = p_indx->i_baseoffset + p_indx->idx.std[i].i_offset - 8;
  2282.             index.i_length  = p_indx->idx.std[i].i_size&0x7fffffff;
  2284.             AVI_IndexAddEntry( p_sys, i_stream, &index );
  2285.         }
  2286.     }
  2287.     else if( p_indx->i_indexsubtype == AVI_INDEX_2FIELD )
  2288.     {
  2289.         for( i = 0; i < p_indx->i_entriesinuse; i++ )
  2290.         {
  2291.             index.i_id      = p_indx->i_id;
  2292.             index.i_flags   = p_indx->idx.field[i].i_size & 0x80000000 ? 0 : AVIIF_KEYFRAME;
  2293.             index.i_pos     = p_indx->i_baseoffset + p_indx->idx.field[i].i_offset - 8;
  2294.             index.i_length  = p_indx->idx.field[i].i_size;
  2296.             AVI_IndexAddEntry( p_sys, i_stream, &index );
  2297.         }
  2298.     }
  2299.     else
  2300.     {
  2301.         msg_Warn( p_demux, "unknown subtype index(0x%x)", p_indx->i_indexsubtype );
  2302.     }
  2303. }
  2305. static void AVI_IndexLoad_indx( demux_t *p_demux )
  2306. {
  2307.     demux_sys_t         *p_sys = p_demux->p_sys;
  2308.     unsigned int        i_stream;
  2309.     int32_t             i;
  2311.     avi_chunk_list_t    *p_riff;
  2312.     avi_chunk_list_t    *p_hdrl;
  2314.     p_riff = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0);
  2315.     p_hdrl = AVI_ChunkFind( p_riff, AVIFOURCC_hdrl, 0 );
  2317.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2318.     {
  2319.         avi_chunk_list_t    *p_strl;
  2320.         avi_chunk_indx_t    *p_indx;
  2322. #define p_stream  p_sys->track[i_stream]
  2323.         p_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i_stream );
  2324.         p_indx = AVI_ChunkFind( p_strl, AVIFOURCC_indx, 0 );
  2326.         if( !p_indx )
  2327.         {
  2328.             msg_Warn( p_demux, "cannot find indx (misdetect/broken OpenDML "
  2329.                                "file?)" );
  2330.             continue;
  2331.         }
  2333.         if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS )
  2334.         {
  2335.             __Parse_indx( p_demux, i_stream, p_indx );
  2336.         }
  2337.         else if( p_indx->i_indextype == AVI_INDEX_OF_INDEXES )
  2338.         {
  2339.             avi_chunk_t    ck_sub;
  2340.             for( i = 0; i < p_indx->i_entriesinuse; i++ )
  2341.             {
  2342.                 if( stream_Seek( p_demux->s, p_indx->idx.super[i].i_offset )||
  2343.                     AVI_ChunkRead( p_demux->s, &ck_sub, NULL  ) )
  2344.                 {
  2345.                     break;
  2346.                 }
  2347.                 __Parse_indx( p_demux, i_stream, &ck_sub.indx );
  2348.             }
  2349.         }
  2350.         else
  2351.         {
  2352.             msg_Warn( p_demux, "unknown type index(0x%x)", p_indx->i_indextype );
  2353.         }
  2354. #undef p_stream
  2355.     }
  2356. }
  2358. static void AVI_IndexLoad( demux_t *p_demux )
  2359. {
  2360.     demux_sys_t *p_sys = p_demux->p_sys;
  2361.     unsigned int i_stream;
  2363.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2364.     {
  2365.         p_sys->track[i_stream]->i_idxnb  = 0;
  2366.         p_sys->track[i_stream]->i_idxmax = 0;
  2367.         p_sys->track[i_stream]->p_index  = NULL;
  2368.     }
  2370.     if( p_sys->b_odml )
  2371.     {
  2372.         AVI_IndexLoad_indx( p_demux );
  2373.     }
  2374.     else  if( AVI_IndexLoad_idx1( p_demux ) )
  2375.     {
  2376.         /* try indx if idx1 failed as some "normal" file have indx too */
  2377.         AVI_IndexLoad_indx( p_demux );
  2378.     }
  2380.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2381.     {
  2382.         msg_Dbg( p_demux, "stream[%d] created %d index entries",
  2383.                 i_stream, p_sys->track[i_stream]->i_idxnb );
  2384.     }
  2385. }
  2387. static void AVI_IndexCreate( demux_t *p_demux )
  2388. {
  2389.     demux_sys_t *p_sys = p_demux->p_sys;
  2391.     avi_chunk_list_t *p_riff;
  2392.     avi_chunk_list_t *p_movi;
  2394.     unsigned int i_stream;
  2395.     off_t i_movi_end;
  2397.     mtime_t i_dialog_update;
  2398.     dialog_progress_bar_t *p_dialog = NULL;
  2400.     p_riff = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0);
  2401.     p_movi = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
  2403.     if( !p_movi )
  2404.     {
  2405.         msg_Err( p_demux, "cannot find p_movi" );
  2406.         return;
  2407.     }
  2409.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2410.     {
  2411.         p_sys->track[i_stream]->i_idxnb  = 0;
  2412.         p_sys->track[i_stream]->i_idxmax = 0;
  2413.         p_sys->track[i_stream]->p_index  = NULL;
  2414.     }
  2415.     i_movi_end = __MIN( (off_t)(p_movi->i_chunk_pos + p_movi->i_chunk_size),
  2416.                         stream_Size( p_demux->s ) );
  2418.     stream_Seek( p_demux->s, p_movi->i_chunk_pos + 12 );
  2419.     msg_Warn( p_demux, "creating index from LIST-movi, will take time !" );
  2422.     /* Only show dialog if AVI is > 10MB */
  2423.     i_dialog_update = mdate();
  2424.     if( stream_Size( p_demux->s ) > 10000000 )
  2425.         p_dialog = dialog_ProgressCreate( p_demux, _("Fixing AVI Index..."),
  2426.                                        NULL, _("Cancel") );
  2428.     for( ;; )
  2429.     {
  2430.         avi_packet_t pk;
  2432.         if( !vlc_object_alive (p_demux) )
  2433.             break;
  2435.         /* Don't update/check dialog too often */
  2436.         if( p_dialog && mdate() - i_dialog_update > 100000 )
  2437.         {
  2438.             if( dialog_ProgressCancelled( p_dialog ) )
  2439.                 break;
  2441.             float f_pos = (float)stream_Tell( p_demux->s ) /
  2442.                           (float)stream_Size( p_demux->s );
  2443.             dialog_ProgressSet( p_dialog, NULL, f_pos );
  2445.             i_dialog_update = mdate();
  2446.         }
  2448.         if( AVI_PacketGetHeader( p_demux, &pk ) )
  2449.             break;
  2451.         if( pk.i_stream < p_sys->i_track &&
  2452.             pk.i_cat == p_sys->track[pk.i_stream]->i_cat )
  2453.         {
  2454.             avi_entry_t index;
  2455.             index.i_id      = pk.i_fourcc;
  2456.             index.i_flags   =
  2457.                AVI_GetKeyFlag(p_sys->track[pk.i_stream]->i_codec, pk.i_peek);
  2458.             index.i_pos     = pk.i_pos;
  2459.             index.i_length  = pk.i_size;
  2460.             AVI_IndexAddEntry( p_sys, pk.i_stream, &index );
  2461.         }
  2462.         else
  2463.         {
  2464.             switch( pk.i_fourcc )
  2465.             {
  2466.             case AVIFOURCC_idx1:
  2467.                 if( p_sys->b_odml )
  2468.                 {
  2469.                     avi_chunk_list_t *p_sysx;
  2470.                     p_sysx = AVI_ChunkFind( &p_sys->ck_root,
  2471.                                             AVIFOURCC_RIFF, 1 );
  2473.                     msg_Dbg( p_demux, "looking for new RIFF chunk" );
  2474.                     if( stream_Seek( p_demux->s, p_sysx->i_chunk_pos + 24 ) )
  2475.                         goto print_stat;
  2476.                     break;
  2477.                 }
  2478.                 goto print_stat;
  2480.             case AVIFOURCC_RIFF:
  2481.                     msg_Dbg( p_demux, "new RIFF chunk found" );
  2482.                     break;
  2484.             case AVIFOURCC_rec:
  2485.             case AVIFOURCC_JUNK:
  2486.                 break;
  2488.             default:
  2489.                 msg_Warn( p_demux, "need resync, probably broken avi" );
  2490.                 if( AVI_PacketSearch( p_demux ) )
  2491.                 {
  2492.                     msg_Warn( p_demux, "lost sync, abord index creation" );
  2493.                     goto print_stat;
  2494.                 }
  2495.             }
  2496.         }
  2498.         if( ( !p_sys->b_odml && pk.i_pos + pk.i_size >= i_movi_end ) ||
  2499.             AVI_PacketNext( p_demux ) )
  2500.         {
  2501.             break;
  2502.         }
  2503.     }
  2505. print_stat:
  2506.     if( p_dialog != NULL )
  2507.         dialog_ProgressDestroy( p_dialog );
  2509.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2510.     {
  2511.         msg_Dbg( p_demux, "stream[%d] creating %d index entries",
  2512.                 i_stream, p_sys->track[i_stream]->i_idxnb );
  2513.     }
  2514. }
  2516. /*****************************************************************************
  2517.  * Stream management
  2518.  *****************************************************************************/
  2519. static int AVI_TrackStopFinishedStreams( demux_t *p_demux )
  2520. {
  2521.     demux_sys_t *p_sys = p_demux->p_sys;
  2522.     unsigned int i;
  2523.     int b_end = true;
  2525.     for( i = 0; i < p_sys->i_track; i++ )
  2526.     {
  2527.         avi_track_t *tk = p_sys->track[i];
  2528.         if( tk->i_idxposc >= tk->i_idxnb )
  2529.         {
  2530.             tk->b_eof = true;
  2531.         }
  2532.         else
  2533.         {
  2534.             b_end = false;
  2535.         }
  2536.     }
  2537.     return( b_end );
  2538. }
  2540. /****************************************************************************
  2541.  * AVI_MovieGetLength give max streams length in second
  2542.  ****************************************************************************/
  2543. static mtime_t  AVI_MovieGetLength( demux_t *p_demux )
  2544. {
  2545.     demux_sys_t  *p_sys = p_demux->p_sys;
  2546.     mtime_t      i_maxlength = 0;
  2547.     unsigned int i;
  2549.     for( i = 0; i < p_sys->i_track; i++ )
  2550.     {
  2551.         avi_track_t *tk = p_sys->track[i];
  2552.         mtime_t i_length;
  2554.         /* fix length for each stream */
  2555.         if( tk->i_idxnb < 1 || !tk->p_index )
  2556.         {
  2557.             continue;
  2558.         }
  2560.         if( tk->i_samplesize )
  2561.         {
  2562.             i_length = AVI_GetDPTS( tk,
  2563.                                     tk->p_index[tk->i_idxnb-1].i_lengthtotal +
  2564.                                         tk->p_index[tk->i_idxnb-1].i_length );
  2565.         }
  2566.         else
  2567.         {
  2568.             i_length = AVI_GetDPTS( tk, tk->i_idxnb );
  2569.         }
  2570.         i_length /= (mtime_t)1000000;    /* in seconds */
  2572.         msg_Dbg( p_demux,
  2573.                  "stream[%d] length:%"PRId64" (based on index)",
  2574.                  i,
  2575.                  i_length );
  2576.         i_maxlength = __MAX( i_maxlength, i_length );
  2577.     }
  2579.     return i_maxlength;
  2580. }