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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 查看源码
avi.c
资源名称：vlc-0.8.1.tar.gz　[点击查看]
上传用户：riyaled888
上传日期：2009-03-27
资源大小：7338k
文件大小：72k
源码类别：

多媒体

开发平台：

MultiPlatform

avi.c：源码内容
  1. /*****************************************************************************
  2.  * avi.c : AVI file Stream input module for vlc
  3.  *****************************************************************************
  4.  * Copyright (C) 2001-2004 VideoLAN
  5.  * $Id: avi.c 9270 2004-11-10 13:05:27Z gbazin $
  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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  21.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #include <stdlib.h>                                      /* malloc(), free() */
  28. #include <vlc/vlc.h>
  29. #include <vlc/input.h>
  31. #include "vlc_meta.h"
  32. #include "codecs.h"
  34. #include "libavi.h"
  36. /*****************************************************************************
  37.  * Module descriptor
  38.  *****************************************************************************/
  40. #define INTERLEAVE_TEXT N_("Force interleaved method" )
  41. #define INTERLEAVE_LONGTEXT N_( "Force interleaved method" )
  43. #define INDEX_TEXT N_("Force index creation")
  44. #define INDEX_LONGTEXT N_( 
  45.     "Recreate a index for the AVI file so we can seek trough it more reliably." )
  47. static int  Open ( vlc_object_t * );
  48. static void Close( vlc_object_t * );
  50. vlc_module_begin();
  51.     set_description( _("AVI demuxer") );
  52.     set_capability( "demux2", 212 );
  54.     add_bool( "avi-interleaved", 0, NULL,
  55.               INTERLEAVE_TEXT, INTERLEAVE_LONGTEXT, VLC_TRUE );
  56.     add_bool( "avi-index", 0, NULL,
  57.               INDEX_TEXT, INDEX_LONGTEXT, VLC_TRUE );
  59.     set_callbacks( Open, Close );
  60. vlc_module_end();
  62. /*****************************************************************************
  63.  * Local prototypes
  64.  *****************************************************************************/
  65. static int Control         ( demux_t *, int, va_list );
  66. static int Seek            ( demux_t *, mtime_t, int );
  67. static int Demux_Seekable  ( demux_t * );
  68. static int Demux_UnSeekable( demux_t * );
  70. #define FREE( p ) if( p ) { free( p ); (p) = NULL; }
  71. #define __ABS( x ) ( (x) < 0 ? (-(x)) : (x) )
  73. typedef struct
  74. {
  75.     vlc_fourcc_t i_fourcc;
  76.     off_t        i_pos;
  77.     uint32_t     i_size;
  78.     vlc_fourcc_t i_type;     /* only for AVIFOURCC_LIST */
  80.     uint8_t      i_peek[8];  /* first 8 bytes */
  82.     unsigned int i_stream;
  83.     unsigned int i_cat;
  84. } avi_packet_t;
  87. typedef struct
  88. {
  89.     vlc_fourcc_t i_id;
  90.     uint32_t     i_flags;
  91.     off_t        i_pos;
  92.     uint32_t     i_length;
  93.     uint32_t     i_lengthtotal;
  95. } avi_entry_t;
  97. typedef struct
  98. {
  99.     vlc_bool_t      b_activated;
  101.     unsigned int    i_cat; /* AUDIO_ES, VIDEO_ES */
  102.     vlc_fourcc_t    i_codec;
  104.     int             i_rate;
  105.     int             i_scale;
  106.     int             i_samplesize;
  108.     es_out_id_t     *p_es;
  110.     avi_entry_t     *p_index;
  111.     unsigned int        i_idxnb;
  112.     unsigned int        i_idxmax;
  114.     unsigned int        i_idxposc;  /* numero of chunk */
  115.     unsigned int        i_idxposb;  /* byte in the current chunk */
  117.     /* For VBR audio only */
  118.     unsigned int        i_blockno;
  119.     unsigned int        i_blocksize;
  120. } avi_track_t;
  122. struct demux_sys_t
  123. {
  124.     mtime_t i_time;
  125.     mtime_t i_length;
  127.     vlc_bool_t  b_seekable;
  128.     avi_chunk_t ck_root;
  130.     vlc_bool_t  b_odml;
  132.     off_t   i_movi_begin;
  133.     off_t   i_movi_lastchunk_pos;   /* XXX position of last valid chunk */
  135.     /* number of streams and information */
  136.     unsigned int i_track;
  137.     avi_track_t  **track;
  139.     /* meta */
  140.     vlc_meta_t  *meta;
  141. };
  143. static inline off_t __EVEN( off_t i )
  144. {
  145.     return (i & 1) ? i + 1 : i;
  146. }
  148. static mtime_t AVI_PTSToChunk( avi_track_t *, mtime_t i_pts );
  149. static mtime_t AVI_PTSToByte ( avi_track_t *, mtime_t i_pts );
  150. static mtime_t AVI_GetDPTS   ( avi_track_t *, int64_t i_count );
  151. static mtime_t AVI_GetPTS    ( avi_track_t * );
  154. static int AVI_StreamChunkFind( demux_t *, unsigned int i_stream );
  155. static int AVI_StreamChunkSet ( demux_t *,
  156.                                 unsigned int i_stream, unsigned int i_ck );
  157. static int AVI_StreamBytesSet ( demux_t *,
  158.                                 unsigned int i_stream, off_t   i_byte );
  160. vlc_fourcc_t AVI_FourccGetCodec( unsigned int i_cat, vlc_fourcc_t );
  161. static int   AVI_GetKeyFlag    ( vlc_fourcc_t , uint8_t * );
  163. static int AVI_PacketGetHeader( demux_t *, avi_packet_t *p_pk );
  164. static int AVI_PacketNext     ( demux_t * );
  165. static int AVI_PacketRead     ( demux_t *, avi_packet_t *, block_t **);
  166. static int AVI_PacketSearch   ( demux_t * );
  168. static void AVI_IndexLoad    ( demux_t * );
  169. static void AVI_IndexCreate  ( demux_t * );
  170. static void AVI_IndexAddEntry( demux_sys_t *, int, avi_entry_t * );
  172. static mtime_t  AVI_MovieGetLength( demux_t * );
  174. /*****************************************************************************
  175.  * Stream management
  176.  *****************************************************************************/
  177. static int        AVI_TrackSeek  ( demux_t *, int, mtime_t );
  178. static int        AVI_TrackStopFinishedStreams( demux_t *);
  180. /* Remarks:
  181.  - For VBR mp3 stream:
  182.     count blocks by rounded-up chunksizes instead of chunks
  183.     we need full emulation of dshow avi demuxer bugs :(
  184.     fixes silly nandub-style a-v delaying in avi with vbr mp3...
  185.     (from mplayer 2002/08/02)
  186.  - to complete....
  187.  */
  189. /*****************************************************************************
  190.  * Open: check file and initializes AVI structures
  191.  *****************************************************************************/
  192. static int Open( vlc_object_t * p_this )
  193. {
  194.     demux_t  *p_demux = (demux_t *)p_this;
  195.     demux_sys_t     *p_sys;
  197.     avi_chunk_t         ck_riff;
  198.     avi_chunk_list_t    *p_riff = (avi_chunk_list_t*)&ck_riff;
  199.     avi_chunk_list_t    *p_hdrl, *p_movi;
  200.     avi_chunk_avih_t    *p_avih;
  202.     unsigned int i_track;
  203.     unsigned int i, i_peeker;
  205.     uint8_t  *p_peek;
  208.     /* Is it an avi file ? */
  209.     if( stream_Peek( p_demux->s, &p_peek, 200 ) < 200 )
  210.     {
  211.         msg_Dbg( p_demux, "cannot peek()" );
  212.         return VLC_EGENERIC;
  213.     }
  214.     for( i_peeker = 0; i_peeker < 188; i_peeker++ )
  215.     {
  216.         if( !strncmp( &p_peek[0], "RIFF", 4 ) && !strncmp( &p_peek[8], "AVI ", 4 ) )
  217.         {
  218.             break;
  219.         }
  220.         p_peek++;
  221.     }
  222.     if( i_peeker == 188 )
  223.     {
  224.         msg_Warn( p_demux, "avi module discarded (invalid header)" );
  225.         return VLC_EGENERIC;
  226.     }
  228.     /* Initialize input  structures. */
  229.     p_sys = p_demux->p_sys = malloc( sizeof(demux_sys_t) );
  230.     memset( p_sys, 0, sizeof( demux_sys_t ) );
  231.     p_sys->i_time   = 0;
  232.     p_sys->i_length = 0;
  233.     p_sys->i_movi_lastchunk_pos = 0;
  234.     p_sys->b_odml   = VLC_FALSE;
  235.     p_sys->i_track  = 0;
  236.     p_sys->track    = NULL;
  237.     p_sys->meta     = NULL;
  239.     stream_Control( p_demux->s, STREAM_CAN_FASTSEEK, &p_sys->b_seekable );
  241.     p_demux->pf_control = Control;
  242.     p_demux->pf_demux = Demux_Seekable;
  243.     /* For unseekable stream, automaticaly use Demux_UnSeekable */
  244.     if( !p_sys->b_seekable || config_GetInt( p_demux, "avi-interleaved" ) )
  245.     {
  246.         p_demux->pf_demux = Demux_UnSeekable;
  247.     }
  249.     if( i_peeker > 0 )
  250.     {
  251. stream_Read( p_demux->s, NULL, i_peeker );
  252.     }
  254.     if( AVI_ChunkReadRoot( p_demux->s, &p_sys->ck_root ) )
  255.     {
  256.         msg_Err( p_demux, "avi module discarded (invalid file)" );
  257.         return VLC_EGENERIC;
  258.     }
  260.     if( AVI_ChunkCount( &p_sys->ck_root, AVIFOURCC_RIFF ) > 1 )
  261.     {
  262.         unsigned int i_count =
  263.             AVI_ChunkCount( &p_sys->ck_root, AVIFOURCC_RIFF );
  265.         msg_Warn( p_demux, "multiple riff -> OpenDML ?" );
  266.         for( i = 1; i < i_count; i++ )
  267.         {
  268.             avi_chunk_list_t *p_sysx;
  270.             p_sysx = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, i );
  271.             if( p_sysx->i_type == AVIFOURCC_AVIX )
  272.             {
  273.                 msg_Warn( p_demux, "detected OpenDML file" );
  274.                 p_sys->b_odml = VLC_TRUE;
  275.                 break;
  276.             }
  277.         }
  278.     }
  280.     p_riff  = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0 );
  281.     p_hdrl  = AVI_ChunkFind( p_riff, AVIFOURCC_hdrl, 0 );
  282.     p_movi  = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0 );
  284.     if( !p_hdrl || !p_movi )
  285.     {
  286.         msg_Err( p_demux, "avi module discarded (invalid file)" );
  287.         goto error;
  288.     }
  290.     if( !( p_avih = AVI_ChunkFind( p_hdrl, AVIFOURCC_avih, 0 ) ) )
  291.     {
  292.         msg_Err( p_demux, "cannot find avih chunk" );
  293.         goto error;
  294.     }
  295.     i_track = AVI_ChunkCount( p_hdrl, AVIFOURCC_strl );
  296.     if( p_avih->i_streams != i_track )
  297.     {
  298.         msg_Warn( p_demux,
  299.                   "found %d stream but %d are declared",
  300.                   i_track, p_avih->i_streams );
  301.     }
  302.     if( i_track == 0 )
  303.     {
  304.         msg_Err( p_demux, "no stream defined!" );
  305.         goto error;
  306.     }
  308.     /* print information on streams */
  309.     msg_Dbg( p_demux, "AVIH: %d stream, flags %s%s%s%s ",
  310.              i_track,
  311.              p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
  312.              p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
  313.              p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
  314.              p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
  315.     if( ( p_sys->meta = vlc_meta_New() ) )
  316.     {
  317.         char buffer[200];
  318.         sprintf( buffer, "%s%s%s%s",
  319.                  p_avih->i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
  320.                  p_avih->i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
  321.                  p_avih->i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
  322.                  p_avih->i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"" );
  323.         vlc_meta_Add( p_sys->meta, VLC_META_SETTING, buffer );
  324.     }
  326.     /* now read info on each stream and create ES */
  327.     for( i = 0 ; i < i_track; i++ )
  328.     {
  329.         avi_track_t      *tk = malloc( sizeof( avi_track_t ) );
  330.         avi_chunk_list_t *p_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i );
  331.         avi_chunk_strh_t *p_strh = AVI_ChunkFind( p_strl, AVIFOURCC_strh, 0 );
  332.         avi_chunk_STRING_t *p_strn = AVI_ChunkFind( p_strl, AVIFOURCC_strn, 0 );
  333.         avi_chunk_strf_auds_t *p_auds;
  334.         avi_chunk_strf_vids_t *p_vids;
  335.         es_format_t fmt;
  337.         tk->b_activated = VLC_FALSE;
  338.         tk->p_index     = 0;
  339.         tk->i_idxnb     = 0;
  340.         tk->i_idxmax    = 0;
  341.         tk->i_idxposc   = 0;
  342.         tk->i_idxposb   = 0;
  344.         tk->i_blockno   = 0;
  345.         tk->i_blocksize = 0;
  347.         p_vids = (avi_chunk_strf_vids_t*)AVI_ChunkFind( p_strl, AVIFOURCC_strf, 0 );
  348.         p_auds = (avi_chunk_strf_auds_t*)AVI_ChunkFind( p_strl, AVIFOURCC_strf, 0 );
  350.         if( p_strl == NULL || p_strh == NULL || p_auds == NULL || p_vids == NULL )
  351.         {
  352.             msg_Warn( p_demux, "stream[%d] incomplete", i );
  353.             continue;
  354.         }
  356.         tk->i_rate  = p_strh->i_rate;
  357.         tk->i_scale = p_strh->i_scale;
  358.         tk->i_samplesize = p_strh->i_samplesize;
  359.         msg_Dbg( p_demux, "stream[%d] rate:%d scale:%d samplesize:%d",
  360.                  i, tk->i_rate, tk->i_scale, tk->i_samplesize );
  362.         switch( p_strh->i_type )
  363.         {
  364.             case( AVIFOURCC_auds ):
  365.                 tk->i_cat   = AUDIO_ES;
  366.                 tk->i_codec = AVI_FourccGetCodec( AUDIO_ES,
  367.                                                   p_auds->p_wf->wFormatTag );
  368.                 if( ( tk->i_blocksize = p_auds->p_wf->nBlockAlign ) == 0 )
  369.                 {
  370.                     if( p_auds->p_wf->wFormatTag == 1 )
  371.                     {
  372.                         tk->i_blocksize = p_auds->p_wf->nChannels * (p_auds->p_wf->wBitsPerSample/8);
  373.                     }
  374.                     else
  375.                     {
  376.                         tk->i_blocksize = 1;
  377.                     }
  378.                 }
  379.                 es_format_Init( &fmt, AUDIO_ES, tk->i_codec );
  381.                 fmt.audio.i_channels        = p_auds->p_wf->nChannels;
  382.                 fmt.audio.i_rate            = p_auds->p_wf->nSamplesPerSec;
  383.                 fmt.i_bitrate               = p_auds->p_wf->nAvgBytesPerSec*8;
  384.                 fmt.audio.i_blockalign      = p_auds->p_wf->nBlockAlign;
  385.                 fmt.audio.i_bitspersample   = p_auds->p_wf->wBitsPerSample;
  386.                 fmt.i_extra = __MIN( p_auds->p_wf->cbSize,
  387.                     p_auds->i_chunk_size - sizeof(WAVEFORMATEX) );
  388.                 fmt.p_extra = &p_auds->p_wf[1];
  389.                 msg_Dbg( p_demux, "stream[%d] audio(0x%x) %d channels %dHz %dbits",
  390.                          i, p_auds->p_wf->wFormatTag, p_auds->p_wf->nChannels,
  391.                          p_auds->p_wf->nSamplesPerSec, p_auds->p_wf->wBitsPerSample);
  392.                 break;
  394.             case( AVIFOURCC_vids ):
  395.                 tk->i_cat   = VIDEO_ES;
  396.                 tk->i_codec = AVI_FourccGetCodec( VIDEO_ES,
  397.                                                   p_vids->p_bih->biCompression );
  398.                 if( p_vids->p_bih->biCompression == 0x00 )
  399.                 {
  400.                     switch( p_vids->p_bih->biBitCount )
  401.                     {
  402.                         case 32:
  403.                             tk->i_codec = VLC_FOURCC('R','V','3','2');
  404.                             break;
  405.                         case 24:
  406.                             tk->i_codec = VLC_FOURCC('R','V','2','4');
  407.                             break;
  408.                         case 16:
  409.                             /* tk->i_codec = VLC_FOURCC('R','V','1','6');*/
  410.                             /* break;*/
  411.                         case 15:
  412.                             tk->i_codec = VLC_FOURCC('R','V','1','5');
  413.                             break;
  414.                         case 9:
  415.                             tk->i_codec = VLC_FOURCC( 'Y', 'V', 'U', '9' ); /* <- TODO check that */
  416.                             break;
  417.                     }
  418.                     es_format_Init( &fmt, VIDEO_ES, tk->i_codec );
  420.                     if( p_vids->p_bih->biBitCount == 24 )
  421.                     {
  422.                         /* This is in BGR format */
  423.                         fmt.video.i_bmask = 0x00ff0000;
  424.                         fmt.video.i_gmask = 0x0000ff00;
  425.                         fmt.video.i_rmask = 0x000000ff;
  426.                     }
  427.                 }
  428.                 else
  429.                 {
  430.                     es_format_Init( &fmt, VIDEO_ES, p_vids->p_bih->biCompression );
  431.                     if( tk->i_codec == FOURCC_mp4v &&
  432.                         !strncasecmp( (char*)&p_strh->i_handler, "XVID", 4 ) )
  433.                     {
  434.                         fmt.i_codec = VLC_FOURCC( 'X', 'V', 'I', 'D' );
  435.                     }
  436.                 }
  437.                 tk->i_samplesize = 0;
  438.                 fmt.video.i_width  = p_vids->p_bih->biWidth;
  439.                 fmt.video.i_height = p_vids->p_bih->biHeight;
  440.                 fmt.video.i_bits_per_pixel = p_vids->p_bih->biBitCount;
  441.                 fmt.i_extra =
  442.                     __MIN( p_vids->p_bih->biSize - sizeof( BITMAPINFOHEADER ),
  443.                            p_vids->i_chunk_size - sizeof(BITMAPINFOHEADER) );
  444.                 fmt.p_extra = &p_vids->p_bih[1];
  445.                 msg_Dbg( p_demux, "stream[%d] video(%4.4s) %dx%d %dbpp %ffps",
  446.                         i,
  447.                          (char*)&p_vids->p_bih->biCompression,
  448.                          p_vids->p_bih->biWidth,
  449.                          p_vids->p_bih->biHeight,
  450.                          p_vids->p_bih->biBitCount,
  451.                          (float)tk->i_rate/(float)tk->i_scale );
  453.                 if( p_vids->p_bih->biCompression == 0x00 )
  454.                 {
  455.                     /* RGB DIB are coded from bottom to top */
  456.                     fmt.video.i_height =
  457.                         (unsigned int)(-(int)p_vids->p_bih->biHeight);
  458.                 }
  459.                 break;
  461.             case( AVIFOURCC_txts):
  462.                 tk->i_cat   = SPU_ES;
  463.                 tk->i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
  464.                 msg_Dbg( p_demux, "stream[%d] subtitles", i );
  465.                 es_format_Init( &fmt, SPU_ES, tk->i_codec );
  466.                 break;
  467.                 
  468.             case( AVIFOURCC_mids):
  469.                 msg_Dbg( p_demux, "stream[%d] midi is UNSUPPORTED", i );
  470.             default:
  471.                 msg_Warn( p_demux, "stream[%d] unknown type", i );
  472.                 free( tk );
  473.                 continue;
  474.         }
  475.         if( p_strn )
  476.         {
  477.             fmt.psz_description = strdup( p_strn->p_str );
  478.         }
  479.         tk->p_es = es_out_Add( p_demux->out, &fmt );
  480.         TAB_APPEND( p_sys->i_track, p_sys->track, tk );
  481.     }
  483.     if( p_sys->i_track <= 0 )
  484.     {
  485.         msg_Err( p_demux, "no valid track" );
  486.         goto error;
  487.     }
  489.     if( config_GetInt( p_demux, "avi-index" ) )
  490.     {
  491.         if( p_sys->b_seekable )
  492.         {
  493.             AVI_IndexCreate( p_demux );
  494.         }
  495.         else
  496.         {
  497.             msg_Warn( p_demux, "cannot create index (unseekable stream)" );
  498.             AVI_IndexLoad( p_demux );
  499.         }
  500.     }
  501.     else
  502.     {
  503.         AVI_IndexLoad( p_demux );
  504.     }
  506.     /* *** movie length in sec *** */
  507.     p_sys->i_length = AVI_MovieGetLength( p_demux );
  508.     if( p_sys->i_length < (mtime_t)p_avih->i_totalframes *
  509.                           (mtime_t)p_avih->i_microsecperframe /
  510.                           (mtime_t)1000000 )
  511.     {
  512.         msg_Warn( p_demux, "broken or missing index, 'seek' will be axproximative or will have strange behavour" );
  513.     }
  514.     /* fix some BeOS MediaKit generated file */
  515.     for( i = 0 ; i < p_sys->i_track; i++ )
  516.     {
  517.         avi_track_t         *tk = p_sys->track[i];
  518.         avi_chunk_list_t    *p_strl;
  519.         avi_chunk_strh_t    *p_strh;
  520.         avi_chunk_strf_auds_t    *p_auds;
  522.         if( tk->i_cat != AUDIO_ES )
  523.         {
  524.             continue;
  525.         }
  526.         if( tk->i_idxnb < 1 ||
  527.             tk->i_scale != 1 ||
  528.             tk->i_samplesize != 0 )
  529.         {
  530.             continue;
  531.         }
  532.         p_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i );
  533.         p_strh = AVI_ChunkFind( p_strl, AVIFOURCC_strh, 0 );
  534.         p_auds = AVI_ChunkFind( p_strl, AVIFOURCC_strf, 0 );
  536.         if( p_auds->p_wf->wFormatTag != WAVE_FORMAT_PCM &&
  537.             (unsigned int)tk->i_rate == p_auds->p_wf->nSamplesPerSec )
  538.         {
  539.             int64_t i_track_length =
  540.                 tk->p_index[tk->i_idxnb-1].i_length +
  541.                 tk->p_index[tk->i_idxnb-1].i_lengthtotal;
  542.             mtime_t i_length = (mtime_t)p_avih->i_totalframes *
  543.                                (mtime_t)p_avih->i_microsecperframe;
  545.             if( i_length == 0 )
  546.             {
  547.                 msg_Warn( p_demux, "track[%d] cannot be fixed (BeOS MediaKit generated)", i );
  548.                 continue;
  549.             }
  550.             tk->i_samplesize = 1;
  551.             tk->i_rate       = i_track_length  * (int64_t)1000000/ i_length;
  552.             msg_Warn( p_demux, "track[%d] fixed with rate=%d scale=%d (BeOS MediaKit generated)", i, tk->i_rate, tk->i_scale );
  553.         }
  554.     }
  556.     if( p_sys->b_seekable )
  557.     {
  558.         /* we have read all chunk so go back to movi */
  559.         stream_Seek( p_demux->s, p_movi->i_chunk_pos );
  560.     }
  561.     /* Skip movi header */
  562.     stream_Read( p_demux->s, NULL, 12 );
  564.     p_sys->i_movi_begin = p_movi->i_chunk_pos;
  565.     return VLC_SUCCESS;
  567. error:
  568.     if( p_sys->meta )
  569.     {
  570.         vlc_meta_Delete( p_sys->meta );
  571.     }
  572.     AVI_ChunkFreeRoot( p_demux->s, &p_sys->ck_root );
  573.     free( p_sys );
  574.     return VLC_EGENERIC;
  575. }
  577. /*****************************************************************************
  578.  * Close: frees unused data
  579.  *****************************************************************************/
  580. static void Close ( vlc_object_t * p_this )
  581. {
  582.     demux_t *    p_demux = (demux_t *)p_this;
  583.     unsigned int i;
  584.     demux_sys_t *p_sys = p_demux->p_sys  ;
  586.     for( i = 0; i < p_sys->i_track; i++ )
  587.     {
  588.         if( p_sys->track[i] )
  589.         {
  590.             FREE( p_sys->track[i]->p_index );
  591.             free( p_sys->track[i] );
  592.         }
  593.     }
  594.     FREE( p_sys->track );
  595.     AVI_ChunkFreeRoot( p_demux->s, &p_sys->ck_root );
  596.     vlc_meta_Delete( p_sys->meta );
  598.     free( p_sys );
  599. }
  601. /*****************************************************************************
  602.  * Demux_Seekable: reads and demuxes data packets for stream seekable
  603.  *****************************************************************************
  604.  * AVIDemux: reads and demuxes data packets
  605.  *****************************************************************************
  606.  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
  607.  *****************************************************************************/
  608. typedef struct
  609. {
  610.     vlc_bool_t b_ok;
  612.     int i_toread;
  614.     off_t i_posf; /* where we will read :
  615.                    if i_idxposb == 0 : begining of chunk (+8 to acces data)
  616.                    else : point on data directly */
  617. } avi_track_toread_t;
  619. static int Demux_Seekable( demux_t *p_demux )
  620. {
  621.     demux_sys_t *p_sys = p_demux->p_sys;
  623.     unsigned int i_track_count = 0;
  624.     unsigned int i_track;
  625.     vlc_bool_t b_stream;
  626.     /* cannot be more than 100 stream (dcXX or wbXX) */
  627.     avi_track_toread_t toread[100];
  630.     /* detect new selected/unselected streams */
  631.     for( i_track = 0; i_track < p_sys->i_track; i_track++ )
  632.     {
  633.         avi_track_t *tk = p_sys->track[i_track];
  634.         vlc_bool_t  b;
  636.         es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b );
  637.         if( b && !tk->b_activated )
  638.         {
  639.             if( p_sys->b_seekable)
  640.             {
  641.                 AVI_TrackSeek( p_demux, i_track, p_sys->i_time );
  642.             }
  643.             tk->b_activated = VLC_TRUE;
  644.         }
  645.         else if( !b && tk->b_activated )
  646.         {
  647.             tk->b_activated = VLC_FALSE;
  648.         }
  649.         if( b )
  650.         {
  651.             i_track_count++;
  652.         }
  653.     }
  655.     if( i_track_count <= 0 )
  656.     {
  657.         int64_t i_length = p_sys->i_length * (mtime_t)1000000;
  659.         p_sys->i_time += 25*1000;  /* read 25ms */
  660.         if( i_length > 0 )
  661.         {
  662.             if( p_sys->i_time >= i_length )
  663.                 return 0;
  664.             return 1;
  665.         }
  666.         msg_Warn( p_demux, "no track selected, exiting..." );
  667.         return 0;
  668.     }
  670.     /* wait for the good time */
  671.     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_time + 1 );
  672.     p_sys->i_time += 25*1000;  /* read 25ms */
  674.     /* init toread */
  675.     for( i_track = 0; i_track < p_sys->i_track; i_track++ )
  676.     {
  677.         avi_track_t *tk = p_sys->track[i_track];
  678.         mtime_t i_dpts;
  680.         toread[i_track].b_ok = tk->b_activated;
  681.         if( tk->i_idxposc < tk->i_idxnb )
  682.         {
  683.             toread[i_track].i_posf = tk->p_index[tk->i_idxposc].i_pos;
  684.            if( tk->i_idxposb > 0 )
  685.            {
  686.                 toread[i_track].i_posf += 8 + tk->i_idxposb;
  687.            }
  688.         }
  689.         else
  690.         {
  691.             toread[i_track].i_posf = -1;
  692.         }
  694.         i_dpts = p_sys->i_time - AVI_GetPTS( tk  );
  696.         if( tk->i_samplesize )
  697.         {
  698.             toread[i_track].i_toread = AVI_PTSToByte( tk, __ABS( i_dpts ) );
  699.         }
  700.         else
  701.         {
  702.             toread[i_track].i_toread = AVI_PTSToChunk( tk, __ABS( i_dpts ) );
  703.         }
  705.         if( i_dpts < 0 )
  706.         {
  707.             toread[i_track].i_toread *= -1;
  708.         }
  709.     }
  711.     b_stream = VLC_FALSE;
  713.     for( ;; )
  714.     {
  715.         avi_track_t     *tk;
  716.         vlc_bool_t       b_done;
  717.         block_t         *p_frame;
  718.         off_t i_pos;
  719.         unsigned int i;
  720.         size_t i_size;
  722.         /* search for first chunk to be read */
  723.         for( i = 0, b_done = VLC_TRUE, i_pos = -1; i < p_sys->i_track; i++ )
  724.         {
  725.             if( !toread[i].b_ok ||
  726.                 AVI_GetDPTS( p_sys->track[i],
  727.                              toread[i].i_toread ) <= -25 * 1000 )
  728.             {
  729.                 continue;
  730.             }
  732.             if( toread[i].i_toread > 0 )
  733.             {
  734.                 b_done = VLC_FALSE; /* not yet finished */
  735.             }
  736.             if( toread[i].i_posf > 0 )
  737.             {
  738.                 if( i_pos == -1 || i_pos > toread[i_track].i_posf )
  739.                 {
  740.                     i_track = i;
  741.                     i_pos = toread[i].i_posf;
  742.                 }
  743.             }
  744.         }
  746.         if( b_done )
  747.         {
  748.             return( 1 );
  749.         }
  751.         if( i_pos == -1 )
  752.         {
  753.             int i_loop_count = 0;
  755.             /* no valid index, we will parse directly the stream
  756.              * in case we fail we will disable all finished stream */
  757.             if( p_sys->i_movi_lastchunk_pos >= p_sys->i_movi_begin + 12 )
  758.             {
  759.                 stream_Seek( p_demux->s, p_sys->i_movi_lastchunk_pos );
  760.                 if( AVI_PacketNext( p_demux ) )
  761.                 {
  762.                     return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  763.                 }
  764.             }
  765.             else
  766.             {
  767.                 stream_Seek( p_demux->s, p_sys->i_movi_begin + 12 );
  768.             }
  770.             for( ;; )
  771.             {
  772.                 avi_packet_t avi_pk;
  774.                 if( AVI_PacketGetHeader( p_demux, &avi_pk ) )
  775.                 {
  776.                     msg_Warn( p_demux,
  777.                              "cannot get packet header, track disabled" );
  778.                     return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  779.                 }
  780.                 if( avi_pk.i_stream >= p_sys->i_track ||
  781.                     ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
  782.                 {
  783.                     if( AVI_PacketNext( p_demux ) )
  784.                     {
  785.                         msg_Warn( p_demux,
  786.                                   "cannot skip packet, track disabled" );
  787.                         return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  788.                     }
  790.                     /* Prevents from eating all the CPU with broken files.
  791.                      * This value should be low enough so that it doesn't
  792.                      * affect the reading speed too much. */
  793.                     if( !(++i_loop_count % 1024) )
  794.                     {
  795.                         if( p_demux->b_die ) return -1;
  796.                         msleep( 10000 );
  798.                         if( !(i_loop_count % (1024 * 10)) )
  799.                             msg_Warn( p_demux,
  800.                                       "don't seem to find any data..." );
  801.                     }
  802.                     continue;
  803.                 }
  804.                 else
  805.                 {
  806.                     /* add this chunk to the index */
  807.                     avi_entry_t index;
  809.                     index.i_id = avi_pk.i_fourcc;
  810.                     index.i_flags =
  811.                        AVI_GetKeyFlag(p_sys->track[avi_pk.i_stream]->i_codec,
  812.                                       avi_pk.i_peek);
  813.                     index.i_pos = avi_pk.i_pos;
  814.                     index.i_length = avi_pk.i_size;
  815.                     AVI_IndexAddEntry( p_sys, avi_pk.i_stream, &index );
  817.                     i_track = avi_pk.i_stream;
  818.                     tk = p_sys->track[i_track];
  819.                     /* do we will read this data ? */
  820.                     if( AVI_GetDPTS( tk, toread[i_track].i_toread ) > -25*1000 )
  821.                     {
  822.                         break;
  823.                     }
  824.                     else
  825.                     {
  826.                         if( AVI_PacketNext( p_demux ) )
  827.                         {
  828.                             msg_Warn( p_demux,
  829.                                       "cannot skip packet, track disabled" );
  830.                             return( AVI_TrackStopFinishedStreams( p_demux ) ? 0 : 1 );
  831.                         }
  832.                     }
  833.                 }
  834.             }
  836.         }
  837.         else
  838.         {
  839.             stream_Seek( p_demux->s, i_pos );
  840.         }
  842.         /* Set the track to use */
  843.         tk = p_sys->track[i_track];
  845.         /* read thoses data */
  846.         if( tk->i_samplesize )
  847.         {
  848.             unsigned int i_toread;
  850.             if( ( i_toread = toread[i_track].i_toread ) <= 0 )
  851.             {
  852.                 if( tk->i_samplesize > 1 )
  853.                 {
  854.                     i_toread = tk->i_samplesize;
  855.                 }
  856.                 else
  857.                 {
  858.                     i_toread = AVI_PTSToByte( tk, 20 * 1000 );
  859.                     i_toread = __MAX( i_toread, 100 );
  860.                 }
  861.             }
  862.             i_size = __MIN( tk->p_index[tk->i_idxposc].i_length -
  863.                                 tk->i_idxposb,
  864.                             i_toread );
  865.         }
  866.         else
  867.         {
  868.             i_size = tk->p_index[tk->i_idxposc].i_length;
  869.         }
  871.         if( tk->i_idxposb == 0 )
  872.         {
  873.             i_size += 8; /* need to read and skip header */
  874.         }
  876.         if( ( p_frame = stream_Block( p_demux->s, __EVEN( i_size ) ) )==NULL )
  877.         {
  878.             msg_Warn( p_demux, "failled reading data" );
  879.             tk->b_activated = VLC_FALSE;
  880.             toread[i_track].b_ok = VLC_FALSE;
  881.             continue;
  882.         }
  883.         if( i_size % 2 )    /* read was padded on word boundary */
  884.         {
  885.             p_frame->i_buffer--;
  886.         }
  887.         /* skip header */
  888.         if( tk->i_idxposb == 0 )
  889.         {
  890.             p_frame->p_buffer += 8;
  891.             p_frame->i_buffer -= 8;
  892.         }
  893.         p_frame->i_pts = AVI_GetPTS( tk ) + 1;
  894.         if( tk->p_index[tk->i_idxposc].i_flags&AVIIF_KEYFRAME )
  895.         {
  896.             p_frame->i_flags = BLOCK_FLAG_TYPE_I;
  897.         }
  898.         else
  899.         {
  900.             p_frame->i_flags = BLOCK_FLAG_TYPE_PB;
  901.         }
  903.         /* read data */
  904.         if( tk->i_samplesize )
  905.         {
  906.             if( tk->i_idxposb == 0 )
  907.             {
  908.                 i_size -= 8;
  909.             }
  910.             toread[i_track].i_toread -= i_size;
  911.             tk->i_idxposb += i_size;
  912.             if( tk->i_idxposb >=
  913.                     tk->p_index[tk->i_idxposc].i_length )
  914.             {
  915.                 tk->i_idxposb = 0;
  916.                 tk->i_idxposc++;
  917.             }
  918.         }
  919.         else
  920.         {
  921.             int i_length = tk->p_index[tk->i_idxposc].i_length;
  923.             tk->i_idxposc++;
  924.             if( tk->i_cat == AUDIO_ES )
  925.             {
  926.                 tk->i_blockno += tk->i_blocksize > 0 ? ( i_length + tk->i_blocksize - 1 ) / tk->i_blocksize : 1;
  927.             }
  928.             toread[i_track].i_toread--;
  929.         }
  931.         if( tk->i_idxposc < tk->i_idxnb)
  932.         {
  933.             toread[i_track].i_posf =
  934.                 tk->p_index[tk->i_idxposc].i_pos;
  935.             if( tk->i_idxposb > 0 )
  936.             {
  937.                 toread[i_track].i_posf += 8 + tk->i_idxposb;
  938.             }
  940.         }
  941.         else
  942.         {
  943.             toread[i_track].i_posf = -1;
  944.         }
  946.         b_stream = VLC_TRUE; /* at least one read succeed */
  948.         if( tk->i_cat != VIDEO_ES )
  949.             p_frame->i_dts = p_frame->i_pts;
  950.         else
  951.         {
  952.             p_frame->i_dts = p_frame->i_pts;
  953.             p_frame->i_pts = 0;
  954.         }
  956.         //p_pes->i_rate = p_demux->stream.control.i_rate;
  957.         es_out_Send( p_demux->out, tk->p_es, p_frame );
  958.     }
  959. }
  962. /*****************************************************************************
  963.  * Demux_UnSeekable: reads and demuxes data packets for unseekable file
  964.  *****************************************************************************
  965.  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
  966.  *****************************************************************************/
  967. static int Demux_UnSeekable( demux_t *p_demux )
  968. {
  969.     demux_sys_t     *p_sys = p_demux->p_sys;
  970.     avi_track_t *p_stream_master = NULL;
  971.     unsigned int i_stream;
  972.     unsigned int i_packet;
  974.     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_time + 1 );
  976.     /* *** find master stream for data packet skipping algo *** */
  977.     /* *** -> first video, if any, or first audio ES *** */
  978.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  979.     {
  980.         avi_track_t *tk = p_sys->track[i_stream];
  981.         vlc_bool_t  b;
  983.         es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b );
  985.         if( b && tk->i_cat == VIDEO_ES )
  986.         {
  987.             p_stream_master = tk;
  988.         }
  989.         else if( b )
  990.         {
  991.             p_stream_master = tk;
  992.         }
  993.     }
  995.     if( !p_stream_master )
  996.     {
  997.         msg_Warn( p_demux, "no more stream selected" );
  998.         return( 0 );
  999.     }
  1001.     p_sys->i_time = AVI_GetPTS( p_stream_master );
  1003.     for( i_packet = 0; i_packet < 10; i_packet++)
  1004.     {
  1005. #define p_stream    p_sys->track[avi_pk.i_stream]
  1007.         avi_packet_t    avi_pk;
  1009.         if( AVI_PacketGetHeader( p_demux, &avi_pk ) )
  1010.         {
  1011.             return( 0 );
  1012.         }
  1014.         if( avi_pk.i_stream >= p_sys->i_track ||
  1015.             ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
  1016.         {
  1017.             /* we haven't found an audio or video packet:
  1018.              *  - we have seek, found first next packet
  1019.              *  - others packets could be found, skip them
  1020.              */
  1021.             switch( avi_pk.i_fourcc )
  1022.             {
  1023.                 case AVIFOURCC_JUNK:
  1024.                 case AVIFOURCC_LIST:
  1025.                 case AVIFOURCC_RIFF:
  1026.                     return( !AVI_PacketNext( p_demux ) ? 1 : 0 );
  1027.                 case AVIFOURCC_idx1:
  1028.                     if( p_sys->b_odml )
  1029.                     {
  1030.                         return( !AVI_PacketNext( p_demux ) ? 1 : 0 );
  1031.                     }
  1032.                     return( 0 );    /* eof */
  1033.                 default:
  1034.                     msg_Warn( p_demux,
  1035.                               "seems to have lost position, resync" );
  1036.                     if( AVI_PacketSearch( p_demux ) )
  1037.                     {
  1038.                         msg_Err( p_demux, "resync failed" );
  1039.                         return( -1 );
  1040.                     }
  1041.             }
  1042.         }
  1043.         else
  1044.         {
  1045.             /* check for time */
  1046.             if( __ABS( AVI_GetPTS( p_stream ) -
  1047.                         AVI_GetPTS( p_stream_master ) )< 600*1000 )
  1048.             {
  1049.                 /* load it and send to decoder */
  1050.                 block_t *p_frame;
  1051.                 if( AVI_PacketRead( p_demux, &avi_pk, &p_frame ) || p_frame == NULL )
  1052.                 {
  1053.                     return( -1 );
  1054.                 }
  1055.                 p_frame->i_pts = AVI_GetPTS( p_stream ) + 1;
  1057.                 if( avi_pk.i_cat != VIDEO_ES )
  1058.                     p_frame->i_dts = p_frame->i_pts;
  1059.                 else
  1060.                 {
  1061.                     p_frame->i_dts = p_frame->i_pts;
  1062.                     p_frame->i_pts = 0;
  1063.                 }
  1065.                 //p_pes->i_rate = p_demux->stream.control.i_rate;
  1066.                 es_out_Send( p_demux->out, p_stream->p_es, p_frame );
  1067.             }
  1068.             else
  1069.             {
  1070.                 if( AVI_PacketNext( p_demux ) )
  1071.                 {
  1072.                     return( 0 );
  1073.                 }
  1074.             }
  1076.             /* *** update stream time position *** */
  1077.             if( p_stream->i_samplesize )
  1078.             {
  1079.                 p_stream->i_idxposb += avi_pk.i_size;
  1080.             }
  1081.             else
  1082.             {
  1083.                 if( p_stream->i_cat == AUDIO_ES )
  1084.                 {
  1085.                     p_stream->i_blockno += p_stream->i_blocksize > 0 ? ( avi_pk.i_size + p_stream->i_blocksize - 1 ) / p_stream->i_blocksize : 1;
  1086.                 }
  1087.                 p_stream->i_idxposc++;
  1088.             }
  1090.         }
  1091. #undef p_stream
  1092.     }
  1094.     return( 1 );
  1095. }
  1097. /*****************************************************************************
  1098.  * Seek: goto to i_date or i_percent
  1099.  *****************************************************************************/
  1100. static int Seek( demux_t *p_demux, mtime_t i_date, int i_percent )
  1101. {
  1103.     demux_sys_t *p_sys = p_demux->p_sys;
  1104.     unsigned int i_stream;
  1105.     msg_Dbg( p_demux, "seek requested: "I64Fd" secondes %d%%",
  1106.              i_date / 1000000, i_percent );
  1108.     if( p_sys->b_seekable )
  1109.     {
  1110.         if( !p_sys->i_length )
  1111.         {
  1112.             avi_track_t *p_stream;
  1113.             int64_t i_pos;
  1115.             /* use i_percent to create a true i_date */
  1116.             msg_Warn( p_demux, "mmh, seeking without index at %d%%"
  1117.                       " work only for interleaved file", i_percent );
  1118.             if( i_percent >= 100 )
  1119.             {
  1120.                 msg_Warn( p_demux, "cannot seek so far !" );
  1121.                 return VLC_EGENERIC;
  1122.             }
  1123.             i_percent = __MAX( i_percent, 0 );
  1125.             /* try to find chunk that is at i_percent or the file */
  1126.             i_pos = __MAX( i_percent * stream_Size( p_demux->s ) / 100,
  1127.                            p_sys->i_movi_begin );
  1128.             /* search first selected stream */
  1129.             for( i_stream = 0, p_stream = NULL;
  1130.                         i_stream < p_sys->i_track; i_stream++ )
  1131.             {
  1132.                 p_stream = p_sys->track[i_stream];
  1133.                 if( p_stream->b_activated )
  1134.                 {
  1135.                     break;
  1136.                 }
  1137.             }
  1138.             if( !p_stream || !p_stream->b_activated )
  1139.             {
  1140.                 msg_Warn( p_demux, "cannot find any selected stream" );
  1141.                 return VLC_EGENERIC;
  1142.             }
  1144.             /* be sure that the index exist */
  1145.             if( AVI_StreamChunkSet( p_demux, i_stream, 0 ) )
  1146.             {
  1147.                 msg_Warn( p_demux, "cannot seek" );
  1148.                 return VLC_EGENERIC;
  1149.             }
  1151.             while( i_pos >= p_stream->p_index[p_stream->i_idxposc].i_pos +
  1152.                p_stream->p_index[p_stream->i_idxposc].i_length + 8 )
  1153.             {
  1154.                 /* search after i_idxposc */
  1155.                 if( AVI_StreamChunkSet( p_demux,
  1156.                                         i_stream, p_stream->i_idxposc + 1 ) )
  1157.                 {
  1158.                     msg_Warn( p_demux, "cannot seek" );
  1159.                     return VLC_EGENERIC;
  1160.                 }
  1161.             }
  1163.             i_date = AVI_GetPTS( p_stream );
  1164.             /* TODO better support for i_samplesize != 0 */
  1165.             msg_Dbg( p_demux, "estimate date "I64Fd, i_date );
  1166.         }
  1168. #define p_stream    p_sys->track[i_stream]
  1169.         p_sys->i_time = 0;
  1170.         /* seek for chunk based streams */
  1171.         for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  1172.         {
  1173.             if( p_stream->b_activated && !p_stream->i_samplesize )
  1174. /*            if( p_stream->b_activated ) */
  1175.             {
  1176.                 AVI_TrackSeek( p_demux, i_stream, i_date );
  1177.                 p_sys->i_time = __MAX( AVI_GetPTS( p_stream ),
  1178.                                         p_sys->i_time );
  1179.             }
  1180.         }
  1181. #if 1
  1182.         if( p_sys->i_time )
  1183.         {
  1184.             i_date = p_sys->i_time;
  1185.         }
  1186.         /* seek for bytes based streams */
  1187.         for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  1188.         {
  1189.             if( p_stream->b_activated && p_stream->i_samplesize )
  1190.             {
  1191.                 AVI_TrackSeek( p_demux, i_stream, i_date );
  1192. /*                p_sys->i_time = __MAX( AVI_GetPTS( p_stream ), p_sys->i_time );*/
  1193.             }
  1194.         }
  1195.         msg_Dbg( p_demux, "seek: "I64Fd" seconds", p_sys->i_time /1000000 );
  1196.         /* set true movie time */
  1197. #endif
  1198.         if( !p_sys->i_time )
  1199.         {
  1200.             p_sys->i_time = i_date;
  1201.         }
  1202. #undef p_stream
  1203.         return VLC_SUCCESS;
  1204.     }
  1205.     else
  1206.     {
  1207.         msg_Err( p_demux, "shouldn't yet be executed" );
  1208.         return VLC_EGENERIC;
  1209.     }
  1210. }
  1212. /*****************************************************************************
  1213.  * Control:
  1214.  *****************************************************************************
  1215.  *
  1216.  *****************************************************************************/
  1217. static double ControlGetPosition( demux_t *p_demux )
  1218. {
  1219.     demux_sys_t *p_sys = p_demux->p_sys;
  1221.     if( p_sys->i_length > 0 )
  1222.     {
  1223.         return (double)p_sys->i_time / (double)( p_sys->i_length * (mtime_t)1000000 );
  1224.     }
  1225.     else if( stream_Size( p_demux->s ) > 0 )
  1226.     {
  1227.         unsigned int i;
  1228.         int64_t i_tmp;
  1229.         int64_t i64 = 0;
  1231.         /* search the more advanced selected es */
  1232.         for( i = 0; i < p_sys->i_track; i++ )
  1233.         {
  1234.             avi_track_t *tk = p_sys->track[i];
  1235.             if( tk->b_activated && tk->i_idxposc < tk->i_idxnb )
  1236.             {
  1237.                 i_tmp = tk->p_index[tk->i_idxposc].i_pos +
  1238.                         tk->p_index[tk->i_idxposc].i_length + 8;
  1239.                 if( i_tmp > i64 )
  1240.                 {
  1241.                     i64 = i_tmp;
  1242.                 }
  1243.             }
  1244.         }
  1245.         return (double)i64 / (double)stream_Size( p_demux->s );
  1246.     }
  1247.     return 0.0;
  1248. }
  1250. static int    Control( demux_t *p_demux, int i_query, va_list args )
  1251. {
  1252.     demux_sys_t *p_sys = p_demux->p_sys;
  1253.     int i;
  1254.     double   f, *pf;
  1255.     int64_t i64, *pi64;
  1256.     vlc_meta_t **pp_meta;
  1258.     switch( i_query )
  1259.     {
  1260.         case DEMUX_GET_POSITION:
  1261.             pf = (double*)va_arg( args, double * );
  1262.             *pf = ControlGetPosition( p_demux );
  1263.             return VLC_SUCCESS;
  1264.         case DEMUX_SET_POSITION:
  1265.             f = (double)va_arg( args, double );
  1266.             if( p_sys->b_seekable )
  1267.             {
  1268.                 i64 = (mtime_t)(1000000.0 * p_sys->i_length * f );
  1269.                 return Seek( p_demux, i64, (int)(f * 100) );
  1270.             }
  1271.             else
  1272.             {
  1273.                 int64_t i_pos = stream_Size( p_demux->s ) * f;
  1274.                 return stream_Seek( p_demux->s, i_pos );
  1275.             }
  1277.         case DEMUX_GET_TIME:
  1278.             pi64 = (int64_t*)va_arg( args, int64_t * );
  1279.             *pi64 = p_sys->i_time;
  1280.             return VLC_SUCCESS;
  1282.         case DEMUX_SET_TIME:
  1283.         {
  1284.             int i_percent = 0;
  1286.             i64 = (int64_t)va_arg( args, int64_t );
  1287.             if( p_sys->i_length > 0 )
  1288.             {
  1289.                 i_percent = 100 * i64 / (p_sys->i_length*1000000);
  1290.             }
  1291.             else if( p_sys->i_time > 0 )
  1292.             {
  1293.                 i_percent = (int)( 100.0 * ControlGetPosition( p_demux ) *
  1294.                                    (double)i64 / (double)p_sys->i_time );
  1295.             }
  1296.             return Seek( p_demux, i64, i_percent );
  1297.         }
  1298.         case DEMUX_GET_LENGTH:
  1299.             pi64 = (int64_t*)va_arg( args, int64_t * );
  1300.             *pi64 = p_sys->i_length * (mtime_t)1000000;
  1301.             return VLC_SUCCESS;
  1303.         case DEMUX_GET_FPS:
  1304.             pf = (double*)va_arg( args, double * );
  1305.             *pf = 0.0;
  1306.             for( i = 0; i < (int)p_sys->i_track; i++ )
  1307.             {
  1308.                 avi_track_t *tk = p_sys->track[i];
  1309.                 if( tk->i_cat == VIDEO_ES && tk->i_scale > 0)
  1310.                 {
  1311.                     *pf = (float)tk->i_rate / (float)tk->i_scale;
  1312.                     break;
  1313.                 }
  1314.             }
  1315.             return VLC_SUCCESS;
  1316.         case DEMUX_GET_META:
  1317.             pp_meta = (vlc_meta_t**)va_arg( args, vlc_meta_t** );
  1318.             *pp_meta = vlc_meta_Duplicate( p_sys->meta );
  1319.             return VLC_SUCCESS;
  1321.         default:
  1322.             return VLC_EGENERIC;
  1323.     }
  1324. }
  1326. /*****************************************************************************
  1327.  * Function to convert pts to chunk or byte
  1328.  *****************************************************************************/
  1330. static mtime_t AVI_PTSToChunk( avi_track_t *tk, mtime_t i_pts )
  1331. {
  1332.     return (mtime_t)((int64_t)i_pts *
  1333.                      (int64_t)tk->i_rate /
  1334.                      (int64_t)tk->i_scale /
  1335.                      (int64_t)1000000 );
  1336. }
  1337. static mtime_t AVI_PTSToByte( avi_track_t *tk, mtime_t i_pts )
  1338. {
  1339.     return (mtime_t)((int64_t)i_pts *
  1340.                      (int64_t)tk->i_rate /
  1341.                      (int64_t)tk->i_scale /
  1342.                      (int64_t)1000000 *
  1343.                      (int64_t)tk->i_samplesize );
  1344. }
  1346. static mtime_t AVI_GetDPTS( avi_track_t *tk, int64_t i_count )
  1347. {
  1348.     mtime_t i_dpts;
  1350.     i_dpts = (mtime_t)( (int64_t)1000000 *
  1351.                         (int64_t)i_count *
  1352.                         (int64_t)tk->i_scale /
  1353.                         (int64_t)tk->i_rate );
  1355.     if( tk->i_samplesize )
  1356.     {
  1357.         return i_dpts / tk->i_samplesize;
  1358.     }
  1359.     return i_dpts;
  1360. }
  1362. static mtime_t AVI_GetPTS( avi_track_t *tk )
  1363. {
  1364.     if( tk->i_samplesize )
  1365.     {
  1366.         int64_t i_count = 0;
  1368.         /* we need a valid entry we will emulate one */
  1369.         if( tk->i_idxposc == tk->i_idxnb )
  1370.         {
  1371.             if( tk->i_idxposc )
  1372.             {
  1373.                 /* use the last entry */
  1374.                 i_count = tk->p_index[tk->i_idxnb - 1].i_lengthtotal
  1375.                             + tk->p_index[tk->i_idxnb - 1].i_length;
  1376.             }
  1377.         }
  1378.         else
  1379.         {
  1380.             i_count = tk->p_index[tk->i_idxposc].i_lengthtotal;
  1381.         }
  1382.         return AVI_GetDPTS( tk, i_count + tk->i_idxposb );
  1383.     }
  1384.     else
  1385.     {
  1386.         if( tk->i_cat == AUDIO_ES )
  1387.         {
  1388.             return AVI_GetDPTS( tk, tk->i_blockno );
  1389.         }
  1390.         else
  1391.         {
  1392.             return AVI_GetDPTS( tk, tk->i_idxposc );
  1393.         }
  1394.     }
  1395. }
  1397. static int AVI_StreamChunkFind( demux_t *p_demux, unsigned int i_stream )
  1398. {
  1399.     demux_sys_t *p_sys = p_demux->p_sys;
  1400.     avi_packet_t avi_pk;
  1401.     int i_loop_count = 0;
  1403.     /* find first chunk of i_stream that isn't in index */
  1405.     if( p_sys->i_movi_lastchunk_pos >= p_sys->i_movi_begin + 12 )
  1406.     {
  1407.         stream_Seek( p_demux->s, p_sys->i_movi_lastchunk_pos );
  1408.         if( AVI_PacketNext( p_demux ) )
  1409.         {
  1410.             return VLC_EGENERIC;
  1411.         }
  1412.     }
  1413.     else
  1414.     {
  1415.         stream_Seek( p_demux->s, p_sys->i_movi_begin + 12 );
  1416.     }
  1418.     for( ;; )
  1419.     {
  1420.         if( p_demux->b_die ) return VLC_EGENERIC;
  1422.         if( AVI_PacketGetHeader( p_demux, &avi_pk ) )
  1423.         {
  1424.             msg_Warn( p_demux, "cannot get packet header" );
  1425.             return VLC_EGENERIC;
  1426.         }
  1427.         if( avi_pk.i_stream >= p_sys->i_track ||
  1428.             ( avi_pk.i_cat != AUDIO_ES && avi_pk.i_cat != VIDEO_ES ) )
  1429.         {
  1430.             if( AVI_PacketNext( p_demux ) )
  1431.             {
  1432.                 return VLC_EGENERIC;
  1433.             }
  1435.             /* Prevents from eating all the CPU with broken files.
  1436.              * This value should be low enough so that it doesn't
  1437.              * affect the reading speed too much. */
  1438.             if( !(++i_loop_count % 1024) )
  1439.             {
  1440.                 if( p_demux->b_die ) return VLC_EGENERIC;
  1441.                 msleep( 10000 );
  1443.                 if( !(i_loop_count % (1024 * 10)) )
  1444.                     msg_Warn( p_demux, "don't seem to find any data..." );
  1445.             }
  1446.         }
  1447.         else
  1448.         {
  1449.             /* add this chunk to the index */
  1450.             avi_entry_t index;
  1452.             index.i_id = avi_pk.i_fourcc;
  1453.             index.i_flags =
  1454.                AVI_GetKeyFlag(p_sys->track[avi_pk.i_stream]->i_codec,
  1455.                               avi_pk.i_peek);
  1456.             index.i_pos = avi_pk.i_pos;
  1457.             index.i_length = avi_pk.i_size;
  1458.             AVI_IndexAddEntry( p_sys, avi_pk.i_stream, &index );
  1460.             if( avi_pk.i_stream == i_stream  )
  1461.             {
  1462.                 return VLC_SUCCESS;
  1463.             }
  1465.             if( AVI_PacketNext( p_demux ) )
  1466.             {
  1467.                 return VLC_EGENERIC;
  1468.             }
  1469.         }
  1470.     }
  1471. }
  1474. /* be sure that i_ck will be a valid index entry */
  1475. static int AVI_StreamChunkSet( demux_t *p_demux, unsigned int i_stream,
  1476.                                unsigned int i_ck )
  1477. {
  1478.     demux_sys_t *p_sys = p_demux->p_sys;
  1479.     avi_track_t *p_stream = p_sys->track[i_stream];
  1481.     p_stream->i_idxposc = i_ck;
  1482.     p_stream->i_idxposb = 0;
  1484.     if(  i_ck >= p_stream->i_idxnb )
  1485.     {
  1486.         p_stream->i_idxposc = p_stream->i_idxnb - 1;
  1487.         do
  1488.         {
  1489.             p_stream->i_idxposc++;
  1490.             if( AVI_StreamChunkFind( p_demux, i_stream ) )
  1491.             {
  1492.                 return VLC_EGENERIC;
  1493.             }
  1495.         } while( p_stream->i_idxposc < i_ck );
  1496.     }
  1498.     return VLC_SUCCESS;
  1499. }
  1502. /* XXX FIXME up to now, we assume that all chunk are one after one */
  1503. static int AVI_StreamBytesSet( demux_t    *p_demux,
  1504.                                unsigned int i_stream,
  1505.                                off_t   i_byte )
  1506. {
  1507.     demux_sys_t *p_sys = p_demux->p_sys;
  1508.     avi_track_t *p_stream = p_sys->track[i_stream];
  1510.     if( ( p_stream->i_idxnb > 0 )
  1511.         &&( i_byte < p_stream->p_index[p_stream->i_idxnb - 1].i_lengthtotal +
  1512.                 p_stream->p_index[p_stream->i_idxnb - 1].i_length ) )
  1513.     {
  1514.         /* index is valid to find the ck */
  1515.         /* uses dichototmie to be fast enougth */
  1516.         int i_idxposc = __MIN( p_stream->i_idxposc, p_stream->i_idxnb - 1 );
  1517.         int i_idxmax  = p_stream->i_idxnb;
  1518.         int i_idxmin  = 0;
  1519.         for( ;; )
  1520.         {
  1521.             if( p_stream->p_index[i_idxposc].i_lengthtotal > i_byte )
  1522.             {
  1523.                 i_idxmax  = i_idxposc ;
  1524.                 i_idxposc = ( i_idxmin + i_idxposc ) / 2 ;
  1525.             }
  1526.             else
  1527.             {
  1528.                 if( p_stream->p_index[i_idxposc].i_lengthtotal +
  1529.                         p_stream->p_index[i_idxposc].i_length <= i_byte)
  1530.                 {
  1531.                     i_idxmin  = i_idxposc ;
  1532.                     i_idxposc = (i_idxmax + i_idxposc ) / 2 ;
  1533.                 }
  1534.                 else
  1535.                 {
  1536.                     p_stream->i_idxposc = i_idxposc;
  1537.                     p_stream->i_idxposb = i_byte -
  1538.                             p_stream->p_index[i_idxposc].i_lengthtotal;
  1539.                     return VLC_SUCCESS;
  1540.                 }
  1541.             }
  1542.         }
  1544.     }
  1545.     else
  1546.     {
  1547.         p_stream->i_idxposc = p_stream->i_idxnb - 1;
  1548.         p_stream->i_idxposb = 0;
  1549.         do
  1550.         {
  1551.             p_stream->i_idxposc++;
  1552.             if( AVI_StreamChunkFind( p_demux, i_stream ) )
  1553.             {
  1554.                 return VLC_EGENERIC;
  1555.             }
  1557.         } while( p_stream->p_index[p_stream->i_idxposc].i_lengthtotal +
  1558.                     p_stream->p_index[p_stream->i_idxposc].i_length <= i_byte );
  1560.         p_stream->i_idxposb = i_byte -
  1561.                        p_stream->p_index[p_stream->i_idxposc].i_lengthtotal;
  1562.         return VLC_SUCCESS;
  1563.     }
  1564. }
  1566. static int AVI_TrackSeek( demux_t *p_demux,
  1567.                            int i_stream,
  1568.                            mtime_t i_date )
  1569. {
  1570.     demux_sys_t  *p_sys = p_demux->p_sys;
  1571.     avi_track_t  *tk = p_sys->track[i_stream];
  1573. #define p_stream    p_sys->track[i_stream]
  1574.     mtime_t i_oldpts;
  1576.     i_oldpts = AVI_GetPTS( p_stream );
  1578.     if( !p_stream->i_samplesize )
  1579.     {
  1580.         if( AVI_StreamChunkSet( p_demux,
  1581.                                 i_stream,
  1582.                                 AVI_PTSToChunk( p_stream, i_date ) ) )
  1583.         {
  1584.             return VLC_EGENERIC;
  1585.         }
  1587.         if( p_stream->i_cat == AUDIO_ES )
  1588.         {
  1589.             unsigned int i;
  1590.             tk->i_blockno = 0;
  1591.             for( i = 0; i < tk->i_idxposc; i++ )
  1592.             {
  1593.                 if( tk->i_blocksize > 0 )
  1594.                 {
  1595.                     tk->i_blockno += ( tk->p_index[i].i_length + tk->i_blocksize - 1 ) / tk->i_blocksize;
  1596.                 }
  1597.                 else
  1598.                 {
  1599.                     tk->i_blockno++;
  1600.                 }
  1601.             }
  1602.         }
  1604.         msg_Dbg( p_demux,
  1605.                  "old:"I64Fd" %s new "I64Fd,
  1606.                  i_oldpts,
  1607.                  i_oldpts > i_date ? ">" : "<",
  1608.                  i_date );
  1610.         if( p_stream->i_cat == VIDEO_ES )
  1611.         {
  1612.             /* search key frame */
  1613.             if( i_date < i_oldpts )
  1614.             {
  1615.                 while( p_stream->i_idxposc > 0 &&
  1616.                    !( p_stream->p_index[p_stream->i_idxposc].i_flags &
  1617.                                                                 AVIIF_KEYFRAME ) )
  1618.                 {
  1619.                     if( AVI_StreamChunkSet( p_demux,
  1620.                                             i_stream,
  1621.                                             p_stream->i_idxposc - 1 ) )
  1622.                     {
  1623.                         return VLC_EGENERIC;
  1624.                     }
  1625.                 }
  1626.             }
  1627.             else
  1628.             {
  1629.                 while( p_stream->i_idxposc < p_stream->i_idxnb &&
  1630.                         !( p_stream->p_index[p_stream->i_idxposc].i_flags &
  1631.                                                                 AVIIF_KEYFRAME ) )
  1632.                 {
  1633.                     if( AVI_StreamChunkSet( p_demux,
  1634.                                             i_stream,
  1635.                                             p_stream->i_idxposc + 1 ) )
  1636.                     {
  1637.                         return VLC_EGENERIC;
  1638.                     }
  1639.                 }
  1640.             }
  1641.         }
  1642.     }
  1643.     else
  1644.     {
  1645.         if( AVI_StreamBytesSet( p_demux,
  1646.                                 i_stream,
  1647.                                 AVI_PTSToByte( p_stream, i_date ) ) )
  1648.         {
  1649.             return VLC_EGENERIC;
  1650.         }
  1651.     }
  1652.     return VLC_SUCCESS;
  1653. #undef p_stream
  1654. }
  1656. /****************************************************************************
  1657.  * Return VLC_TRUE if it's a key frame
  1658.  ****************************************************************************/
  1659. static int AVI_GetKeyFlag( vlc_fourcc_t i_fourcc, uint8_t *p_byte )
  1660. {
  1661.     switch( i_fourcc )
  1662.     {
  1663.         case FOURCC_DIV1:
  1664.             /* we have:
  1665.              *  startcode:      0x00000100   32bits
  1666.              *  framenumber     ?             5bits
  1667.              *  piture type     0(I),1(P)     2bits
  1668.              */
  1669.             if( GetDWBE( p_byte ) != 0x00000100 )
  1670.             {
  1671.                 /* it's not an msmpegv1 stream, strange...*/
  1672.                 return AVIIF_KEYFRAME;
  1673.             }
  1674.             return p_byte[4] & 0x06 ? 0 : AVIIF_KEYFRAME;
  1676.         case FOURCC_DIV2:
  1677.         case FOURCC_DIV3:   /* wmv1 also */
  1678.             /* we have
  1679.              *  picture type    0(I),1(P)     2bits
  1680.              */
  1681.             return p_byte[0] & 0xC0 ? 0 : AVIIF_KEYFRAME;
  1682.         case FOURCC_mp4v:
  1683.             /* we should find first occurrence of 0x000001b6 (32bits)
  1684.              *  startcode:      0x000001b6   32bits
  1685.              *  piture type     0(I),1(P)     2bits
  1686.              */
  1687.             if( GetDWBE( p_byte ) != 0x000001b6 )
  1688.             {
  1689.                 /* not true , need to find the first VOP header */
  1690.                 return AVIIF_KEYFRAME;
  1691.             }
  1692.             return p_byte[4] & 0xC0 ? 0 : AVIIF_KEYFRAME;
  1694.         default:
  1695.             /* I can't do it, so say yes */
  1696.             return AVIIF_KEYFRAME;
  1697.     }
  1698. }
  1700. vlc_fourcc_t AVI_FourccGetCodec( unsigned int i_cat, vlc_fourcc_t i_codec )
  1701. {
  1702.     switch( i_cat )
  1703.     {
  1704.         case AUDIO_ES:
  1705.             wf_tag_to_fourcc( i_codec, &i_codec, NULL );
  1706.             return i_codec;
  1708.         case VIDEO_ES:
  1709.             /* XXX DIV1 <- msmpeg4v1, DIV2 <- msmpeg4v2, DIV3 <- msmpeg4v3, mp4v for mpeg4 */
  1710.             switch( i_codec )
  1711.             {
  1712.                 case FOURCC_1:
  1713.                     return VLC_FOURCC('m','r','l','e');
  1714.                 case FOURCC_DIV1:
  1715.                 case FOURCC_div1:
  1716.                 case FOURCC_MPG4:
  1717.                 case FOURCC_mpg4:
  1718.                     return FOURCC_DIV1;
  1719.                 case FOURCC_DIV2:
  1720.                 case FOURCC_div2:
  1721.                 case FOURCC_MP42:
  1722.                 case FOURCC_mp42:
  1723.                 case FOURCC_MPG3:
  1724.                 case FOURCC_mpg3:
  1725.                     return FOURCC_DIV2;
  1726.                 case FOURCC_div3:
  1727.                 case FOURCC_MP43:
  1728.                 case FOURCC_mp43:
  1729.                 case FOURCC_DIV3:
  1730.                 case FOURCC_DIV4:
  1731.                 case FOURCC_div4:
  1732.                 case FOURCC_DIV5:
  1733.                 case FOURCC_div5:
  1734.                 case FOURCC_DIV6:
  1735.                 case FOURCC_div6:
  1736.                 case FOURCC_AP41:
  1737.                 case FOURCC_3IV1:
  1738.                 case FOURCC_3iv1:
  1739.                 case FOURCC_3IVD:
  1740.                 case FOURCC_3ivd:
  1741.                 case FOURCC_3VID:
  1742.                 case FOURCC_3vid:
  1743.                     return FOURCC_DIV3;
  1744.                 case FOURCC_DIVX:
  1745.                 case FOURCC_divx:
  1746.                 case FOURCC_MP4S:
  1747.                 case FOURCC_mp4s:
  1748.                 case FOURCC_M4S2:
  1749.                 case FOURCC_m4s2:
  1750.                 case FOURCC_xvid:
  1751.                 case FOURCC_XVID:
  1752.                 case FOURCC_XviD:
  1753.                 case FOURCC_DX50:
  1754.                 case FOURCC_dx50:
  1755.                 case FOURCC_mp4v:
  1756.                 case FOURCC_4:
  1757.                 case FOURCC_3IV2:
  1758.                 case FOURCC_3iv2:
  1759.                     return FOURCC_mp4v;
  1760.             }
  1761.         default:
  1762.             return VLC_FOURCC( 'u', 'n', 'd', 'f' );
  1763.     }
  1764. }
  1766. /****************************************************************************
  1767.  *
  1768.  ****************************************************************************/
  1769. static void AVI_ParseStreamHeader( vlc_fourcc_t i_id,
  1770.                                    int *pi_number, int *pi_type )
  1771. {
  1772. #define SET_PTR( p, v ) if( p ) *(p) = (v);
  1773.     int c1, c2;
  1775.     c1 = ((uint8_t *)&i_id)[0];
  1776.     c2 = ((uint8_t *)&i_id)[1];
  1778.     if( c1 < '0' || c1 > '9' || c2 < '0' || c2 > '9' )
  1779.     {
  1780.         SET_PTR( pi_number, 100 ); /* > max stream number */
  1781.         SET_PTR( pi_type, UNKNOWN_ES );
  1782.     }
  1783.     else
  1784.     {
  1785.         SET_PTR( pi_number, (c1 - '0') * 10 + (c2 - '0' ) );
  1786.         switch( VLC_TWOCC( ((uint8_t *)&i_id)[2], ((uint8_t *)&i_id)[3] ) )
  1787.         {
  1788.             case AVITWOCC_wb:
  1789.                 SET_PTR( pi_type, AUDIO_ES );
  1790.                 break;
  1791.             case AVITWOCC_dc:
  1792.             case AVITWOCC_db:
  1793.                 SET_PTR( pi_type, VIDEO_ES );
  1794.                 break;
  1795.             default:
  1796.                 SET_PTR( pi_type, UNKNOWN_ES );
  1797.                 break;
  1798.         }
  1799.     }
  1800. #undef SET_PTR
  1801. }
  1803. /****************************************************************************
  1804.  *
  1805.  ****************************************************************************/
  1806. static int AVI_PacketGetHeader( demux_t *p_demux, avi_packet_t *p_pk )
  1807. {
  1808.     uint8_t  *p_peek;
  1810.     if( stream_Peek( p_demux->s, &p_peek, 16 ) < 16 )
  1811.     {
  1812.         return VLC_EGENERIC;
  1813.     }
  1814.     p_pk->i_fourcc  = VLC_FOURCC( p_peek[0], p_peek[1], p_peek[2], p_peek[3] );
  1815.     p_pk->i_size    = GetDWLE( p_peek + 4 );
  1816.     p_pk->i_pos     = stream_Tell( p_demux->s );
  1817.     if( p_pk->i_fourcc == AVIFOURCC_LIST || p_pk->i_fourcc == AVIFOURCC_RIFF )
  1818.     {
  1819.         p_pk->i_type = VLC_FOURCC( p_peek[8],  p_peek[9],
  1820.                                    p_peek[10], p_peek[11] );
  1821.     }
  1822.     else
  1823.     {
  1824.         p_pk->i_type = 0;
  1825.     }
  1827.     memcpy( p_pk->i_peek, p_peek + 8, 8 );
  1829.     AVI_ParseStreamHeader( p_pk->i_fourcc, &p_pk->i_stream, &p_pk->i_cat );
  1830.     return VLC_SUCCESS;
  1831. }
  1833. static int AVI_PacketNext( demux_t *p_demux )
  1834. {
  1835.     avi_packet_t    avi_ck;
  1836.     int             i_skip = 0;
  1838.     if( AVI_PacketGetHeader( p_demux, &avi_ck ) )
  1839.     {
  1840.         return VLC_EGENERIC;
  1841.     }
  1843.     if( avi_ck.i_fourcc == AVIFOURCC_LIST &&
  1844.         ( avi_ck.i_type == AVIFOURCC_rec || avi_ck.i_type == AVIFOURCC_movi ) )
  1845.     {
  1846.         i_skip = 12;
  1847.     }
  1848.     else if( avi_ck.i_fourcc == AVIFOURCC_RIFF &&
  1849.              avi_ck.i_type == AVIFOURCC_AVIX )
  1850.     {
  1851.         i_skip = 24;
  1852.     }
  1853.     else
  1854.     {
  1855.         i_skip = __EVEN( avi_ck.i_size ) + 8;
  1856.     }
  1858.     if( stream_Read( p_demux->s, NULL, i_skip ) != i_skip )
  1859.     {
  1860.         return VLC_EGENERIC;
  1861.     }
  1862.     return VLC_SUCCESS;
  1863. }
  1864. static int AVI_PacketRead( demux_t   *p_demux,
  1865.                            avi_packet_t     *p_pk,
  1866.                            block_t          **pp_frame )
  1867. {
  1868.     size_t i_size;
  1870.     i_size = __EVEN( p_pk->i_size + 8 );
  1872.     if( ( *pp_frame = stream_Block( p_demux->s, i_size ) ) == NULL )
  1873.     {
  1874.         return VLC_EGENERIC;
  1875.     }
  1876.     (*pp_frame)->p_buffer += 8;
  1877.     (*pp_frame)->i_buffer -= 8;
  1879.     if( i_size != p_pk->i_size + 8 )
  1880.     {
  1881.         (*pp_frame)->i_buffer--;
  1882.     }
  1884.     return VLC_SUCCESS;
  1885. }
  1887. static int AVI_PacketSearch( demux_t *p_demux )
  1888. {
  1889.     demux_sys_t     *p_sys = p_demux->p_sys;
  1890.     avi_packet_t    avi_pk;
  1891.     int             i_count = 0;
  1893.     for( ;; )
  1894.     {
  1895.         if( stream_Read( p_demux->s, NULL, 1 ) != 1 )
  1896.         {
  1897.             return VLC_EGENERIC;
  1898.         }
  1899.         AVI_PacketGetHeader( p_demux, &avi_pk );
  1900.         if( avi_pk.i_stream < p_sys->i_track &&
  1901.             ( avi_pk.i_cat == AUDIO_ES || avi_pk.i_cat == VIDEO_ES ) )
  1902.         {
  1903.             return VLC_SUCCESS;
  1904.         }
  1905.         switch( avi_pk.i_fourcc )
  1906.         {
  1907.             case AVIFOURCC_JUNK:
  1908.             case AVIFOURCC_LIST:
  1909.             case AVIFOURCC_RIFF:
  1910.             case AVIFOURCC_idx1:
  1911.                 return VLC_SUCCESS;
  1912.         }
  1914.         /* Prevents from eating all the CPU with broken files.
  1915.          * This value should be low enough so that it doesn't affect the
  1916.          * reading speed too much (not that we care much anyway because
  1917.          * this code is called only on broken files). */
  1918.         if( !(++i_count % 1024) )
  1919.         {
  1920.             if( p_demux->b_die ) return VLC_EGENERIC;
  1922.             msleep( 10000 );
  1923.             if( !(i_count % (1024 * 10)) )
  1924.                 msg_Warn( p_demux, "trying to resync..." );
  1925.         }
  1926.     }
  1927. }
  1929. /****************************************************************************
  1930.  * Index stuff.
  1931.  ****************************************************************************/
  1932. static void AVI_IndexAddEntry( demux_sys_t *p_sys,
  1933.                                int i_stream,
  1934.                                avi_entry_t *p_index)
  1935. {
  1936.     avi_track_t *tk = p_sys->track[i_stream];
  1938.     /* Update i_movi_lastchunk_pos */
  1939.     if( p_sys->i_movi_lastchunk_pos < p_index->i_pos )
  1940.     {
  1941.         p_sys->i_movi_lastchunk_pos = p_index->i_pos;
  1942.     }
  1944.     /* add the entry */
  1945.     if( tk->i_idxnb >= tk->i_idxmax )
  1946.     {
  1947.         tk->i_idxmax += 16384;
  1948.         tk->p_index = realloc( tk->p_index,
  1949.                                tk->i_idxmax * sizeof( avi_entry_t ) );
  1950.         if( tk->p_index == NULL )
  1951.         {
  1952.             return;
  1953.         }
  1954.     }
  1955.     /* calculate cumulate length */
  1956.     if( tk->i_idxnb > 0 )
  1957.     {
  1958.         p_index->i_lengthtotal =
  1959.             tk->p_index[tk->i_idxnb - 1].i_length +
  1960.                 tk->p_index[tk->i_idxnb - 1].i_lengthtotal;
  1961.     }
  1962.     else
  1963.     {
  1964.         p_index->i_lengthtotal = 0;
  1965.     }
  1967.     tk->p_index[tk->i_idxnb++] = *p_index;
  1968. }
  1970. static int AVI_IndexLoad_idx1( demux_t *p_demux )
  1971. {
  1972.     demux_sys_t *p_sys = p_demux->p_sys;
  1974.     avi_chunk_list_t    *p_riff;
  1975.     avi_chunk_list_t    *p_movi;
  1976.     avi_chunk_idx1_t    *p_idx1;
  1978.     unsigned int i_stream;
  1979.     unsigned int i_index;
  1980.     off_t        i_offset;
  1981.     unsigned int i;
  1983.     vlc_bool_t b_keyset[100];
  1985.     p_riff = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0);
  1986.     p_idx1 = AVI_ChunkFind( p_riff, AVIFOURCC_idx1, 0);
  1987.     p_movi = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
  1989.     if( !p_idx1 )
  1990.     {
  1991.         msg_Warn( p_demux, "cannot find idx1 chunk, no index defined" );
  1992.         return VLC_EGENERIC;
  1993.     }
  1995.     /* *** calculate offset *** */
  1996.     /* Well, avi is __SHIT__ so test more than one entry
  1997.      * (needed for some avi files) */
  1998.     i_offset = 0;
  1999.     for( i = 0; i < __MIN( p_idx1->i_entry_count, 10 ); i++ )
  2000.     {
  2001.         if( p_idx1->entry[i].i_pos < p_movi->i_chunk_pos )
  2002.         {
  2003.             i_offset = p_movi->i_chunk_pos + 8;
  2004.             break;
  2005.         }
  2006.     }
  2008.     /* Reset b_keyset */
  2009.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2010.         b_keyset[i_stream] = VLC_FALSE;
  2012.     for( i_index = 0; i_index < p_idx1->i_entry_count; i_index++ )
  2013.     {
  2014.         unsigned int i_cat;
  2016.         AVI_ParseStreamHeader( p_idx1->entry[i_index].i_fourcc,
  2017.                                &i_stream,
  2018.                                &i_cat );
  2019.         if( i_stream < p_sys->i_track &&
  2020.             i_cat == p_sys->track[i_stream]->i_cat )
  2021.         {
  2022.             avi_entry_t index;
  2023.             index.i_id      = p_idx1->entry[i_index].i_fourcc;
  2024.             index.i_flags   =
  2025.                 p_idx1->entry[i_index].i_flags&(~AVIIF_FIXKEYFRAME);
  2026.             index.i_pos     = p_idx1->entry[i_index].i_pos + i_offset;
  2027.             index.i_length  = p_idx1->entry[i_index].i_length;
  2028.             AVI_IndexAddEntry( p_sys, i_stream, &index );
  2030.             if( index.i_flags&AVIIF_KEYFRAME )
  2031.                 b_keyset[i_stream] = VLC_TRUE;
  2032.         }
  2033.     }
  2035.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2036.     {
  2037.         if( !b_keyset[i_stream] )
  2038.         {
  2039.             avi_track_t *tk = p_sys->track[i_stream];
  2041.             msg_Dbg( p_demux, "no key frame set for track %d", i_stream );
  2042.             for( i_index = 0; i_index < tk->i_idxnb; i_index++ )
  2043.                 tk->p_index[i_index].i_flags |= AVIIF_KEYFRAME;
  2044.         }
  2045.     }
  2046.     return VLC_SUCCESS;
  2047. }
  2049. static void __Parse_indx( demux_t    *p_demux,
  2050.                           int               i_stream,
  2051.                           avi_chunk_indx_t  *p_indx )
  2052. {
  2053.     demux_sys_t         *p_sys    = p_demux->p_sys;
  2054.     avi_entry_t     index;
  2055.     int32_t             i;
  2057.     msg_Dbg( p_demux, "loading subindex(0x%x) %d entries", p_indx->i_indextype, p_indx->i_entriesinuse );
  2058.     if( p_indx->i_indexsubtype == 0 )
  2059.     {
  2060.         for( i = 0; i < p_indx->i_entriesinuse; i++ )
  2061.         {
  2062.             index.i_id      = p_indx->i_id;
  2063.             index.i_flags   = p_indx->idx.std[i].i_size & 0x80000000 ? 0 : AVIIF_KEYFRAME;
  2064.             index.i_pos     = p_indx->i_baseoffset + p_indx->idx.std[i].i_offset - 8;
  2065.             index.i_length  = p_indx->idx.std[i].i_size&0x7fffffff;
  2067.             AVI_IndexAddEntry( p_sys, i_stream, &index );
  2068.         }
  2069.     }
  2070.     else if( p_indx->i_indexsubtype == AVI_INDEX_2FIELD )
  2071.     {
  2072.         for( i = 0; i < p_indx->i_entriesinuse; i++ )
  2073.         {
  2074.             index.i_id      = p_indx->i_id;
  2075.             index.i_flags   = p_indx->idx.field[i].i_size & 0x80000000 ? 0 : AVIIF_KEYFRAME;
  2076.             index.i_pos     = p_indx->i_baseoffset + p_indx->idx.field[i].i_offset - 8;
  2077.             index.i_length  = p_indx->idx.field[i].i_size;
  2079.             AVI_IndexAddEntry( p_sys, i_stream, &index );
  2080.         }
  2081.     }
  2082.     else
  2083.     {
  2084.         msg_Warn( p_demux, "unknown subtype index(0x%x)", p_indx->i_indexsubtype );
  2085.     }
  2086. }
  2088. static void AVI_IndexLoad_indx( demux_t *p_demux )
  2089. {
  2090.     demux_sys_t         *p_sys = p_demux->p_sys;
  2091.     unsigned int        i_stream;
  2092.     int32_t             i;
  2094.     avi_chunk_list_t    *p_riff;
  2095.     avi_chunk_list_t    *p_hdrl;
  2097.     p_riff = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0);
  2098.     p_hdrl = AVI_ChunkFind( p_riff, AVIFOURCC_hdrl, 0 );
  2100.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2101.     {
  2102.         avi_chunk_list_t    *p_strl;
  2103.         avi_chunk_indx_t    *p_indx;
  2105. #define p_stream  p_sys->track[i_stream]
  2106.         p_strl = AVI_ChunkFind( p_hdrl, AVIFOURCC_strl, i_stream );
  2107.         p_indx = AVI_ChunkFind( p_strl, AVIFOURCC_indx, 0 );
  2109.         if( !p_indx )
  2110.         {
  2111.             msg_Warn( p_demux, "cannot find indx (misdetect/broken OpenDML file?)" );
  2112.             continue;
  2113.         }
  2115.         if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS )
  2116.         {
  2117.             __Parse_indx( p_demux, i_stream, p_indx );
  2118.         }
  2119.         else if( p_indx->i_indextype == AVI_INDEX_OF_INDEXES )
  2120.         {
  2121.             avi_chunk_t    ck_sub;
  2122.             for( i = 0; i < p_indx->i_entriesinuse; i++ )
  2123.             {
  2124.                 if( stream_Seek( p_demux->s, p_indx->idx.super[i].i_offset )||
  2125.                     AVI_ChunkRead( p_demux->s, &ck_sub, NULL  ) )
  2126.                 {
  2127.                     break;
  2128.                 }
  2129.                 __Parse_indx( p_demux, i_stream, &ck_sub.indx );
  2130.             }
  2131.         }
  2132.         else
  2133.         {
  2134.             msg_Warn( p_demux, "unknown type index(0x%x)", p_indx->i_indextype );
  2135.         }
  2136. #undef p_stream
  2137.     }
  2138. }
  2140. static void AVI_IndexLoad( demux_t *p_demux )
  2141. {
  2142.     demux_sys_t *p_sys = p_demux->p_sys;
  2143.     unsigned int i_stream;
  2145.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2146.     {
  2147.         p_sys->track[i_stream]->i_idxnb  = 0;
  2148.         p_sys->track[i_stream]->i_idxmax = 0;
  2149.         p_sys->track[i_stream]->p_index  = NULL;
  2150.     }
  2152.     if( p_sys->b_odml )
  2153.     {
  2154.         AVI_IndexLoad_indx( p_demux );
  2155.     }
  2156.     else  if( AVI_IndexLoad_idx1( p_demux ) )
  2157.     {
  2158.         /* try indx if idx1 failed as some "normal" file have indx too */
  2159.         AVI_IndexLoad_indx( p_demux );
  2160.     }
  2162.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2163.     {
  2164.         msg_Dbg( p_demux, "stream[%d] created %d index entries",
  2165.                 i_stream, p_sys->track[i_stream]->i_idxnb );
  2166.     }
  2167. }
  2169. static void AVI_IndexCreate( demux_t *p_demux )
  2170. {
  2171.     demux_sys_t *p_sys = p_demux->p_sys;
  2173.     avi_chunk_list_t    *p_riff;
  2174.     avi_chunk_list_t    *p_movi;
  2176.     unsigned int i_stream;
  2177.     off_t i_movi_end;
  2179.     p_riff = AVI_ChunkFind( &p_sys->ck_root, AVIFOURCC_RIFF, 0);
  2180.     p_movi = AVI_ChunkFind( p_riff, AVIFOURCC_movi, 0);
  2182.     if( !p_movi )
  2183.     {
  2184.         msg_Err( p_demux, "cannot find p_movi" );
  2185.         return;
  2186.     }
  2188.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2189.     {
  2190.         p_sys->track[i_stream]->i_idxnb  = 0;
  2191.         p_sys->track[i_stream]->i_idxmax = 0;
  2192.         p_sys->track[i_stream]->p_index  = NULL;
  2193.     }
  2194.     i_movi_end = __MIN( (off_t)(p_movi->i_chunk_pos + p_movi->i_chunk_size),
  2195.                         stream_Size( p_demux->s ) );
  2197.     stream_Seek( p_demux->s, p_movi->i_chunk_pos + 12 );
  2198.     msg_Warn( p_demux, "creating index from LIST-movi, will take time !" );
  2199.     for( ;; )
  2200.     {
  2201.         avi_packet_t pk;
  2203.         if( p_demux->b_die )
  2204.         {
  2205.             return;
  2206.         }
  2208.         if( AVI_PacketGetHeader( p_demux, &pk ) )
  2209.         {
  2210.             break;
  2211.         }
  2212.         if( pk.i_stream < p_sys->i_track &&
  2213.             pk.i_cat == p_sys->track[pk.i_stream]->i_cat )
  2214.         {
  2215.             avi_entry_t index;
  2216.             index.i_id      = pk.i_fourcc;
  2217.             index.i_flags   =
  2218.                AVI_GetKeyFlag(p_sys->track[pk.i_stream]->i_codec, pk.i_peek);
  2219.             index.i_pos     = pk.i_pos;
  2220.             index.i_length  = pk.i_size;
  2221.             AVI_IndexAddEntry( p_sys, pk.i_stream, &index );
  2222.         }
  2223.         else
  2224.         {
  2225.             switch( pk.i_fourcc )
  2226.             {
  2227.                 case AVIFOURCC_idx1:
  2228.                     if( p_sys->b_odml )
  2229.                     {
  2230.                         avi_chunk_list_t *p_sysx;
  2231.                         p_sysx = AVI_ChunkFind( &p_sys->ck_root,
  2232.                                                 AVIFOURCC_RIFF, 1 );
  2234.                         msg_Dbg( p_demux, "looking for new RIFF chunk" );
  2235.                         if( stream_Seek( p_demux->s, p_sysx->i_chunk_pos + 24))
  2236.                         {
  2237.                             goto print_stat;
  2238.                         }
  2239.                         break;
  2240.                     }
  2241.                     goto print_stat;
  2242.                 case AVIFOURCC_RIFF:
  2243.                         msg_Dbg( p_demux, "new RIFF chunk found" );
  2244.                 case AVIFOURCC_rec:
  2245.                 case AVIFOURCC_JUNK:
  2246.                     break;
  2247.                 default:
  2248.                     msg_Warn( p_demux, "need resync, probably broken avi" );
  2249.                     if( AVI_PacketSearch( p_demux ) )
  2250.                     {
  2251.                         msg_Warn( p_demux, "lost sync, abord index creation" );
  2252.                         goto print_stat;
  2253.                     }
  2254.             }
  2255.         }
  2257.         if( ( !p_sys->b_odml && pk.i_pos + pk.i_size >= i_movi_end ) ||
  2258.             AVI_PacketNext( p_demux ) )
  2259.         {
  2260.             break;
  2261.         }
  2262.     }
  2264. print_stat:
  2265.     for( i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
  2266.     {
  2267.         msg_Dbg( p_demux,
  2268.                 "stream[%d] creating %d index entries",
  2269.                 i_stream,
  2270.                 p_sys->track[i_stream]->i_idxnb );
  2271.     }
  2272. }
  2274. /*****************************************************************************
  2275.  * Stream management
  2276.  *****************************************************************************/
  2277. static int AVI_TrackStopFinishedStreams( demux_t *p_demux )
  2278. {
  2279.     demux_sys_t *p_sys = p_demux->p_sys;
  2280.     unsigned int i;
  2281.     int b_end = VLC_TRUE;
  2283.     for( i = 0; i < p_sys->i_track; i++ )
  2284.     {
  2285.         avi_track_t *tk = p_sys->track[i];
  2286.         if( tk->i_idxposc >= tk->i_idxnb )
  2287.         {
  2288.             tk->b_activated = VLC_FALSE;
  2289.             es_out_Control( p_demux->out, ES_OUT_SET_ES_STATE, tk->p_es, VLC_FALSE );
  2290.         }
  2291.         else
  2292.         {
  2293.             b_end = VLC_FALSE;
  2294.         }
  2295.     }
  2296.     return( b_end );
  2297. }
  2299. /****************************************************************************
  2300.  * AVI_MovieGetLength give max streams length in second
  2301.  ****************************************************************************/
  2302. static mtime_t  AVI_MovieGetLength( demux_t *p_demux )
  2303. {
  2304.     demux_sys_t  *p_sys = p_demux->p_sys;
  2305.     mtime_t      i_maxlength = 0;
  2306.     unsigned int i;
  2308.     for( i = 0; i < p_sys->i_track; i++ )
  2309.     {
  2310.         avi_track_t *tk = p_sys->track[i];
  2311.         mtime_t i_length;
  2313.         /* fix length for each stream */
  2314.         if( tk->i_idxnb < 1 || !tk->p_index )
  2315.         {
  2316.             continue;
  2317.         }
  2319.         if( tk->i_samplesize )
  2320.         {
  2321.             i_length = AVI_GetDPTS( tk,
  2322.                                     tk->p_index[tk->i_idxnb-1].i_lengthtotal +
  2323.                                         tk->p_index[tk->i_idxnb-1].i_length );
  2324.         }
  2325.         else
  2326.         {
  2327.             i_length = AVI_GetDPTS( tk, tk->i_idxnb );
  2328.         }
  2329.         i_length /= (mtime_t)1000000;    /* in seconds */
  2331.         msg_Dbg( p_demux,
  2332.                  "stream[%d] length:"I64Fd" (based on index)",
  2333.                  i,
  2334.                  i_length );
  2335.         i_maxlength = __MAX( i_maxlength, i_length );
  2336.     }
  2338.     return i_maxlength;
  2339. }