git.sesse.net Git - vlc/blob - modules/demux/mp4/mp4.c

   1 /*****************************************************************************
   2  * mp4.c : MP4 file input module for vlc
   3  *****************************************************************************
   4  * Copyright (C) 2001-2004 VideoLAN
   5  * $Id$
   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  *****************************************************************************/
  22
  23 /*****************************************************************************
  24  * Preamble
  25  *****************************************************************************/
  26 #include <stdlib.h>                                      /* malloc(), free() */
  27
  28 #include <vlc/vlc.h>
  29 #include <vlc/input.h>
  30 #include <vlc_playlist.h>
  31 #include "iso_lang.h"
  32 #include "vlc_meta.h"
  33
  34 #include "libmp4.h"
  35 #include "drms.h"
  36
  37 /*****************************************************************************
  38  * Module descriptor
  39  *****************************************************************************/
  40 static int  Open ( vlc_object_t * );
  41 static void Close( vlc_object_t * );
  42
  43 vlc_module_begin();
  44     set_description( _("MP4 stream demuxer") );
  45     set_capability( "demux2", 242 );
  46     set_callbacks( Open, Close );
  47 vlc_module_end();
  48
  49 /*****************************************************************************
  50  * Local prototypes
  51  *****************************************************************************/
  52 static int    Demux   ( demux_t * );
  53 static int    DemuxRef( demux_t *p_demux )
  54 {
  55     return 0;
  56 }
  57 static int   Seek     ( demux_t *, mtime_t );
  58 static int   Control  ( demux_t *, int, va_list );
  59
  60 /* Contain all information about a chunk */
  61 typedef struct
  62 {
  63     uint64_t     i_offset; /* absolute position of this chunk in the file */
  64     uint32_t     i_sample_description_index; /* index for SampleEntry to use */
  65     uint32_t     i_sample_count; /* how many samples in this chunk */
  66     uint32_t     i_sample_first; /* index of the first sample in this chunk */
  67
  68     /* now provide way to calculate pts, dts, and offset without to
  69         much memory and with fast acces */
  70
  71     /* with this we can calculate dts/pts without waste memory */
  72     uint64_t     i_first_dts;
  73     uint32_t     *p_sample_count_dts;
  74     uint32_t     *p_sample_delta_dts; /* dts delta */
  75
  76     /* TODO if needed add pts
  77         but quickly *add* support for edts and seeking */
  78
  79 } mp4_chunk_t;
  80
  81
  82  /* Contain all needed information for read all track with vlc */
  83 typedef struct
  84 {
  85     int i_track_ID;     /* this should be unique */
  86
  87     int b_ok;           /* The track is usable */
  88     int b_enable;       /* is the trak enable by default */
  89     vlc_bool_t b_selected;     /* is the trak being played */
  90
  91     es_format_t fmt;
  92     es_out_id_t *p_es;
  93
  94     /* display size only ! */
  95     int i_width;
  96     int i_height;
  97
  98     /* more internal data */
  99     uint64_t        i_timescale;    /* time scale for this track only */
 100
 101     /* elst */
 102     int             i_elst;         /* current elst */
 103     int64_t         i_elst_time;    /* current elst start time (in movie time scale)*/
 104     MP4_Box_t       *p_elst;        /* elst (could be NULL) */
 105
 106     /* give the next sample to read, i_chunk is to find quickly where
 107       the sample is located */
 108     uint32_t         i_sample;       /* next sample to read */
 109     uint32_t         i_chunk;        /* chunk where next sample is stored */
 110     /* total count of chunk and sample */
 111     uint32_t         i_chunk_count;
 112     uint32_t         i_sample_count;
 113
 114     mp4_chunk_t    *chunk; /* always defined  for each chunk */
 115
 116     /* sample size, p_sample_size defined only if i_sample_size == 0
 117         else i_sample_size is size for all sample */
 118     uint32_t         i_sample_size;
 119     uint32_t         *p_sample_size; /* XXX perhaps add file offset if take
 120                                     too much time to do sumations each time*/
 121
 122     MP4_Box_t *p_stbl;  /* will contain all timing information */
 123     MP4_Box_t *p_stsd;  /* will contain all data to initialize decoder */
 124     MP4_Box_t *p_sample;/* point on actual sdsd */
 125
 126     vlc_bool_t b_drms;
 127     void      *p_drms;
 128
 129 } mp4_track_t;
 130
 131
 132 struct demux_sys_t
 133 {
 134     MP4_Box_t    *p_root;      /* container for the whole file */
 135
 136     mtime_t      i_pcr;
 137
 138     uint64_t     i_time;        /* time position of the presentation
 139                                  * in movie timescale */
 140     uint64_t     i_timescale;   /* movie time scale */
 141     uint64_t     i_duration;    /* movie duration */
 142     unsigned int i_tracks;      /* number of tracks */
 143     mp4_track_t *track;    /* array of track */
 144 };
 145
 146
 147 /*****************************************************************************
 148  * Declaration of local function
 149  *****************************************************************************/
 150 static void MP4_TrackCreate ( demux_t *, mp4_track_t *, MP4_Box_t  *);
 151 static void MP4_TrackDestroy( demux_t *, mp4_track_t * );
 152
 153 static int  MP4_TrackSelect ( demux_t *, mp4_track_t *, mtime_t );
 154 static void MP4_TrackUnselect(demux_t *, mp4_track_t * );
 155
 156 static int  MP4_TrackSeek   ( demux_t *, mp4_track_t *, mtime_t );
 157
 158 static uint64_t MP4_TrackGetPos    ( mp4_track_t * );
 159 static int      MP4_TrackSampleSize( mp4_track_t * );
 160 static int      MP4_TrackNextSample( demux_t *, mp4_track_t * );
 161 static void     MP4_TrackSetELST( demux_t *, mp4_track_t *, int64_t );
 162
 163 /* Return time in µs of a track */
 164 static inline int64_t MP4_TrackGetPTS( demux_t *p_demux, mp4_track_t *p_track )
 165 {
 166     unsigned int i_sample;
 167     unsigned int i_index;
 168     int64_t i_dts;
 169
 170     i_sample = p_track->i_sample - p_track->chunk[p_track->i_chunk].i_sample_first;
 171     i_dts = p_track->chunk[p_track->i_chunk].i_first_dts;
 172     i_index = 0;
 173     while( i_sample > 0 )
 174     {
 175         if( i_sample > p_track->chunk[p_track->i_chunk].p_sample_count_dts[i_index] )
 176         {
 177             i_dts += p_track->chunk[p_track->i_chunk].p_sample_count_dts[i_index] *
 178                         p_track->chunk[p_track->i_chunk].p_sample_delta_dts[i_index];
 179             i_sample -= p_track->chunk[p_track->i_chunk].p_sample_count_dts[i_index];
 180             i_index++;
 181         }
 182         else
 183         {
 184             i_dts += i_sample *
 185                         p_track->chunk[p_track->i_chunk].p_sample_delta_dts[i_index];
 186             i_sample = 0;
 187             break;
 188         }
 189     }
 190
 191     /* now handle elst */
 192     if( p_track->p_elst )
 193     {
 194         demux_sys_t         *p_sys = p_demux->p_sys;
 195         MP4_Box_data_elst_t *elst = p_track->p_elst->data.p_elst;
 196
 197         /* convert to offset */
 198         if( ( elst->i_media_rate_integer[p_track->i_elst] > 0 ||
 199               elst->i_media_rate_fraction[p_track->i_elst] > 0 ) &&
 200             elst->i_media_time[p_track->i_elst] > 0 )
 201         {
 202             i_dts -= elst->i_media_time[p_track->i_elst];
 203         }
 204
 205         /* add i_elst_time */
 206         i_dts += p_track->i_elst_time * p_track->i_timescale / p_sys->i_timescale;
 207
 208         if( i_dts < 0 ) i_dts = 0;
 209     }
 210
 211     return (int64_t)1000000 * i_dts / p_track->i_timescale;
 212 }
 213 static inline int64_t MP4_GetMoviePTS(demux_sys_t *p_sys )
 214 {
 215     return (int64_t)1000000 * p_sys->i_time / p_sys->i_timescale;
 216 }
 217
 218 #define FREE( p ) if( p ) { free( p ); (p) = NULL;}
 219
 220 /*****************************************************************************
 221  * Open: check file and initializes MP4 structures
 222  *****************************************************************************/
 223 static int Open( vlc_object_t * p_this )
 224 {
 225     demux_t  *p_demux = (demux_t *)p_this;
 226     demux_sys_t     *p_sys;
 227
 228     uint8_t         *p_peek;
 229
 230     MP4_Box_t       *p_ftyp;
 231     MP4_Box_t       *p_rmra;
 232     MP4_Box_t       *p_mvhd;
 233     MP4_Box_t       *p_trak;
 234
 235     unsigned int    i;
 236     vlc_bool_t      b_seekable;
 237
 238     /* a little test to see if it could be a mp4 */
 239     if( stream_Peek( p_demux->s, &p_peek, 8 ) < 8 )
 240     {
 241         msg_Warn( p_demux, "MP4 plugin discarded (cannot peek)" );
 242         return VLC_EGENERIC;
 243     }
 244     switch( VLC_FOURCC( p_peek[4], p_peek[5], p_peek[6], p_peek[7] ) )
 245     {
 246         case FOURCC_ftyp:
 247         case FOURCC_moov:
 248         case FOURCC_foov:
 249         case FOURCC_moof:
 250         case FOURCC_mdat:
 251         case FOURCC_udta:
 252         case FOURCC_free:
 253         case FOURCC_skip:
 254         case FOURCC_wide:
 255         case VLC_FOURCC( 'p', 'n', 'o', 't' ):
 256             break;
 257          default:
 258             msg_Warn( p_demux, "MP4 plugin discarded (not a valid file)" );
 259             return VLC_EGENERIC;
 260     }
 261
 262     /* I need to seek */
 263     stream_Control( p_demux->s, STREAM_CAN_SEEK, &b_seekable );
 264     if( !b_seekable )
 265     {
 266         msg_Warn( p_demux, "MP4 plugin discarded (unseekable)" );
 267         return VLC_EGENERIC;
 268     }
 269
 270     /*Set exported functions */
 271     p_demux->pf_demux = Demux;
 272     p_demux->pf_control = Control;
 273
 274     /* create our structure that will contains all data */
 275     p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
 276     memset( p_sys, 0, sizeof( demux_sys_t ) );
 277
 278     /* Now load all boxes ( except raw data ) */
 279     if( ( p_sys->p_root = MP4_BoxGetRoot( p_demux->s ) ) == NULL )
 280     {
 281         msg_Warn( p_demux, "MP4 plugin discarded (not a valid file)" );
 282         goto error;
 283     }
 284
 285     MP4_BoxDumpStructure( p_demux->s, p_sys->p_root );
 286
 287     if( ( p_ftyp = MP4_BoxGet( p_sys->p_root, "/ftyp" ) ) )
 288     {
 289         switch( p_ftyp->data.p_ftyp->i_major_brand )
 290         {
 291             case( FOURCC_isom ):
 292                 msg_Dbg( p_demux,
 293                          "ISO Media file (isom) version %d.",
 294                          p_ftyp->data.p_ftyp->i_minor_version );
 295                 break;
 296             default:
 297                 msg_Dbg( p_demux,
 298                          "unrecognized major file specification (%4.4s).",
 299                           (char*)&p_ftyp->data.p_ftyp->i_major_brand );
 300                 break;
 301         }
 302     }
 303     else
 304     {
 305         msg_Dbg( p_demux, "file type box missing (assuming ISO Media file)" );
 306     }
 307
 308     /* the file need to have one moov box */
 309     if( MP4_BoxCount( p_sys->p_root, "/moov" ) <= 0 )
 310     {
 311         MP4_Box_t *p_foov = MP4_BoxGet( p_sys->p_root, "/foov" );
 312
 313         if( !p_foov )
 314         {
 315             msg_Err( p_demux, "MP4 plugin discarded (no moov box)" );
 316             goto error;
 317         }
 318         /* we have a free box as a moov, rename it */
 319         p_foov->i_type = FOURCC_moov;
 320     }
 321
 322     if( ( p_rmra = MP4_BoxGet( p_sys->p_root,  "/moov/rmra" ) ) )
 323     {
 324         playlist_t *p_playlist;
 325         int        i_count = MP4_BoxCount( p_rmra, "rmda" );
 326         int        i;
 327
 328         msg_Dbg( p_demux, "detected playlist mov file (%d ref)", i_count );
 329
 330         p_playlist =
 331             (playlist_t *)vlc_object_find( p_demux,
 332                                            VLC_OBJECT_PLAYLIST,
 333                                            FIND_ANYWHERE );
 334         if( p_playlist )
 335         {
 336             //p_playlist->pp_items[p_playlist->i_index]->b_autodeletion = VLC_TRUE;
 337             for( i = 0; i < i_count; i++ )
 338             {
 339                 MP4_Box_t *p_rdrf = MP4_BoxGet( p_rmra, "rmda[%d]/rdrf", i );
 340                 char      *psz_ref;
 341                 uint32_t  i_ref_type;
 342
 343                 if( !p_rdrf || !( psz_ref = p_rdrf->data.p_rdrf->psz_ref ) )
 344                 {
 345                     continue;
 346                 }
 347                 i_ref_type = p_rdrf->data.p_rdrf->i_ref_type;
 348
 349                 msg_Dbg( p_demux, "new ref=`%s' type=%4.4s",
 350                          psz_ref, (char*)&i_ref_type );
 351
 352                 if( i_ref_type == VLC_FOURCC( 'u', 'r', 'l', ' ' ) )
 353                 {
 354                     if( strstr( psz_ref, "qt5gateQT" ) )
 355                     {
 356                         msg_Dbg( p_demux, "ignoring pseudo ref =`%s'", psz_ref );
 357                         continue;
 358                     }
 359                     if( !strncmp( psz_ref, "http://", 7 ) ||
 360                         !strncmp( psz_ref, "rtsp://", 7 ) )
 361                     {
 362                         msg_Dbg( p_demux, "adding ref = `%s'", psz_ref );
 363                         playlist_Add( p_playlist, psz_ref, psz_ref,
 364                                       PLAYLIST_APPEND, PLAYLIST_END );
 365                     }
 366                     else
 367                     {
 368                         /* msg dbg relative ? */
 369                         char *psz_absolute = alloca( strlen( p_demux->psz_path ) + strlen( psz_ref ) + 1);
 370                         char *end = strrchr( p_demux->psz_path, '/' );
 371
 372                         if( end )
 373                         {
 374                             int i_len = end + 1 - p_demux->psz_path;
 375
 376                             strncpy( psz_absolute, p_demux->psz_path, i_len);
 377                             psz_absolute[i_len] = '\0';
 378                         }
 379                         else
 380                         {
 381                             strcpy( psz_absolute, "" );
 382                         }
 383                         strcat( psz_absolute, psz_ref );
 384                         msg_Dbg( p_demux, "adding ref = `%s'", psz_absolute );
 385                         playlist_Add( p_playlist, psz_absolute, psz_absolute,
 386                                       PLAYLIST_APPEND, PLAYLIST_END );
 387                     }
 388                 }
 389                 else
 390                 {
 391                     msg_Err( p_demux, "unknown ref type=%4.4s FIXME (send a bug report)",
 392                              (char*)&p_rdrf->data.p_rdrf->i_ref_type );
 393                 }
 394             }
 395             vlc_object_release( p_playlist );
 396         }
 397         else
 398         {
 399             msg_Err( p_demux, "can't find playlist" );
 400         }
 401     }
 402
 403     if( !(p_mvhd = MP4_BoxGet( p_sys->p_root, "/moov/mvhd" ) ) )
 404     {
 405         if( !p_rmra )
 406         {
 407             msg_Err( p_demux, "cannot find /moov/mvhd" );
 408             goto error;
 409         }
 410         else
 411         {
 412             msg_Warn( p_demux, "cannot find /moov/mvhd (pure ref file)" );
 413             p_demux->pf_demux = DemuxRef;
 414             return VLC_SUCCESS;
 415         }
 416     }
 417     else
 418     {
 419         p_sys->i_timescale = p_mvhd->data.p_mvhd->i_timescale;
 420         p_sys->i_duration = p_mvhd->data.p_mvhd->i_duration;
 421     }
 422
 423     if( !( p_sys->i_tracks = MP4_BoxCount( p_sys->p_root, "/moov/trak" ) ) )
 424     {
 425         msg_Err( p_demux, "cannot find any /moov/trak" );
 426         goto error;
 427     }
 428     msg_Dbg( p_demux, "find %d track%c",
 429                         p_sys->i_tracks,
 430                         p_sys->i_tracks ? 's':' ' );
 431
 432     /* allocate memory */
 433     p_sys->track = calloc( p_sys->i_tracks, sizeof( mp4_track_t ) );
 434     memset( p_sys->track, 0, p_sys->i_tracks * sizeof( mp4_track_t ) );
 435
 436     /* now process each track and extract all usefull informations */
 437     for( i = 0; i < p_sys->i_tracks; i++ )
 438     {
 439         p_trak = MP4_BoxGet( p_sys->p_root, "/moov/trak[%d]", i );
 440         MP4_TrackCreate( p_demux, &p_sys->track[i], p_trak );
 441
 442         if( p_sys->track[i].b_ok )
 443         {
 444             char *psz_cat;
 445             switch( p_sys->track[i].fmt.i_cat )
 446             {
 447                 case( VIDEO_ES ):
 448                     psz_cat = "video";
 449                     break;
 450                 case( AUDIO_ES ):
 451                     psz_cat = "audio";
 452                     break;
 453                 case( SPU_ES ):
 454                     psz_cat = "subtitle";
 455                     break;
 456
 457                 default:
 458                     psz_cat = "unknown";
 459                     break;
 460             }
 461
 462             msg_Dbg( p_demux, "adding track[Id 0x%x] %s (%s) language %s",
 463                             p_sys->track[i].i_track_ID,
 464                             psz_cat,
 465                             p_sys->track[i].b_enable ? "enable":"disable",
 466                             p_sys->track[i].fmt.psz_language ? p_sys->track[i].fmt.psz_language : "undef" );
 467         }
 468         else
 469         {
 470             msg_Dbg( p_demux, "ignoring track[Id 0x%x]", p_sys->track[i].i_track_ID );
 471         }
 472
 473     }
 474     return VLC_SUCCESS;
 475
 476 error:
 477     if( p_sys->p_root )
 478     {
 479         MP4_BoxFree( p_demux->s, p_sys->p_root );
 480     }
 481     free( p_sys );
 482     return VLC_EGENERIC;
 483 }
 484
 485 /*****************************************************************************
 486  * Demux: read packet and send them to decoders
 487  *****************************************************************************
 488  * TODO check for newly selected track (ie audio upt to now )
 489  *****************************************************************************/
 490 static int Demux( demux_t *p_demux )
 491 {
 492     demux_sys_t *p_sys = p_demux->p_sys;
 493     unsigned int i_track;
 494
 495
 496     unsigned int i_track_selected;
 497
 498     /* check for newly selected/unselected track */
 499     for( i_track = 0, i_track_selected = 0; i_track < p_sys->i_tracks;
 500          i_track++ )
 501     {
 502         mp4_track_t *tk = &p_sys->track[i_track];
 503
 504         if( tk->b_selected && tk->i_sample >= tk->i_sample_count )
 505         {
 506             msg_Warn( p_demux, "track[0x%x] will be disabled", tk->i_track_ID );
 507             MP4_TrackUnselect( p_demux, tk);
 508         }
 509         else if( tk->b_ok )
 510         {
 511             vlc_bool_t b;
 512             es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b );
 513
 514             if( tk->b_selected && !b )
 515             {
 516                 MP4_TrackUnselect( p_demux, tk );
 517             }
 518             else if( !tk->b_selected && b)
 519             {
 520                 MP4_TrackSelect( p_demux, tk, MP4_GetMoviePTS( p_sys ) );
 521             }
 522
 523             if( tk->b_selected )
 524             {
 525                 i_track_selected++;
 526             }
 527         }
 528     }
 529
 530     if( i_track_selected <= 0 )
 531     {
 532         p_sys->i_time += __MAX( p_sys->i_timescale / 10 , 1 );
 533         if( p_sys->i_timescale > 0 )
 534         {
 535             int64_t i_length = (mtime_t)1000000 *
 536                                (mtime_t)p_sys->i_duration /
 537                                (mtime_t)p_sys->i_timescale;
 538             if( MP4_GetMoviePTS( p_sys ) >= i_length )
 539                 return 0;
 540             return 1;
 541         }
 542
 543         msg_Warn( p_demux, "no track selected, exiting..." );
 544         return 0;
 545     }
 546
 547     /* first wait for the good time to read a packet */
 548     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_sys->i_pcr + 1 );
 549
 550     p_sys->i_pcr = MP4_GetMoviePTS( p_sys );
 551
 552     /* we will read 100ms for each stream so ...*/
 553     p_sys->i_time += __MAX( p_sys->i_timescale / 10 , 1 );
 554
 555     for( i_track = 0; i_track < p_sys->i_tracks; i_track++ )
 556     {
 557         mp4_track_t *tk = &p_sys->track[i_track];
 558
 559         if( !tk->b_ok || !tk->b_selected )
 560         {
 561             continue;
 562         }
 563
 564         while( MP4_TrackGetPTS( p_demux, tk ) < MP4_GetMoviePTS( p_sys ) )
 565         {
 566 #if 0
 567             msg_Dbg( p_demux, "tk=%lld mv=%lld",
 568                      MP4_TrackGetPTS( p_demux, tk ),
 569                      MP4_GetMoviePTS( p_sys ) );
 570 #endif
 571
 572             if( MP4_TrackSampleSize( tk ) > 0 )
 573             {
 574                 block_t *p_block;
 575
 576                 /* go,go go ! */
 577                 if( stream_Seek( p_demux->s, MP4_TrackGetPos( tk ) ) )
 578                 {
 579                     msg_Warn( p_demux, "track[0x%x] will be disabled (eof?)", tk->i_track_ID );
 580                     MP4_TrackUnselect( p_demux, tk );
 581                     break;
 582                 }
 583
 584                 /* now read pes */
 585                 if( ( p_block = stream_Block( p_demux->s,
 586                                               MP4_TrackSampleSize( tk ) ) ) == NULL )
 587                 {
 588                     msg_Warn( p_demux, "track[0x%x] will be disabled (eof?)", tk->i_track_ID );
 589                     MP4_TrackUnselect( p_demux, tk );
 590                     break;
 591                 }
 592
 593                 if( tk->b_drms && tk->p_drms )
 594                 {
 595                     drms_decrypt( tk->p_drms,
 596                                   (uint32_t*)p_block->p_buffer,
 597                                   p_block->i_buffer );
 598                 }
 599                 else if( tk->fmt.i_cat == SPU_ES && p_block->i_buffer >= 2 )
 600                 {
 601                     uint16_t i_size = GetWBE( p_block->p_buffer );
 602
 603                     if( i_size + 2 <= p_block->i_buffer )
 604                     {
 605                         char *p;
 606                         /* remove the length field, and append a '\0' */
 607                         memmove( &p_block->p_buffer[0], &p_block->p_buffer[2], i_size );
 608                         p_block->p_buffer[i_size] = '\0';
 609                         p_block->i_buffer = i_size + 1;
 610
 611                         /* convert \r -> \n */
 612                         while( ( p = strchr( p_block->p_buffer, '\r' ) ) )
 613                         {
 614                             *p = '\n';
 615                         }
 616                     }
 617                     else
 618                     {
 619                         /* Invalid */
 620                         p_block->i_buffer = 0;
 621                     }
 622                 }
 623                 p_block->i_dts = MP4_TrackGetPTS( p_demux, tk ) + 1;
 624
 625                 p_block->i_pts = tk->fmt.i_cat == VIDEO_ES ? 0 : p_block->i_dts + 1;
 626
 627                 if( !tk->b_drms || ( tk->b_drms && tk->p_drms ) )
 628                 {
 629                     es_out_Send( p_demux->out, tk->p_es, p_block );
 630                 }
 631             }
 632
 633             /* Next sample */
 634             if( MP4_TrackNextSample( p_demux, tk ) )
 635             {
 636                 break;
 637             }
 638         }
 639     }
 640
 641     return 1;
 642 }
 643 /*****************************************************************************
 644  * Seek: Got to i_date
 645 ******************************************************************************/
 646 static int Seek( demux_t *p_demux, mtime_t i_date )
 647 {
 648     demux_sys_t *p_sys = p_demux->p_sys;
 649     unsigned int i_track;
 650
 651     /* First update update global time */
 652     p_sys->i_time = i_date * p_sys->i_timescale / 1000000;
 653     p_sys->i_pcr  = i_date;
 654
 655     /* Now for each stream try to go to this time */
 656     for( i_track = 0; i_track < p_sys->i_tracks; i_track++ )
 657     {
 658         mp4_track_t *tk = &p_sys->track[i_track];
 659         MP4_TrackSeek( p_demux, tk, i_date );
 660     }
 661     return VLC_SUCCESS;
 662 }
 663
 664 /*****************************************************************************
 665  * Control:
 666  *****************************************************************************/
 667 static int Control( demux_t *p_demux, int i_query, va_list args )
 668 {
 669     demux_sys_t *p_sys = p_demux->p_sys;
 670
 671     double f, *pf;
 672     int64_t i64, *pi64;
 673
 674     switch( i_query )
 675     {
 676         case DEMUX_GET_POSITION:
 677             pf = (double*)va_arg( args, double * );
 678             if( p_sys->i_duration > 0 )
 679             {
 680                 *pf = (double)p_sys->i_time / (double)p_sys->i_duration;
 681             }
 682             else
 683             {
 684                 *pf = 0.0;
 685             }
 686             return VLC_SUCCESS;
 687
 688         case DEMUX_SET_POSITION:
 689             f = (double)va_arg( args, double );
 690             if( p_sys->i_timescale > 0 )
 691             {
 692                 i64 = (int64_t)( f * (double)1000000 *
 693                                  (double)p_sys->i_duration /
 694                                  (double)p_sys->i_timescale );
 695                 return Seek( p_demux, i64 );
 696             }
 697             else return VLC_SUCCESS;
 698
 699         case DEMUX_GET_TIME:
 700             pi64 = (int64_t*)va_arg( args, int64_t * );
 701             if( p_sys->i_timescale > 0 )
 702             {
 703                 *pi64 = (mtime_t)1000000 *
 704                         (mtime_t)p_sys->i_time /
 705                         (mtime_t)p_sys->i_timescale;
 706             }
 707             else *pi64 = 0;
 708             return VLC_SUCCESS;
 709
 710         case DEMUX_SET_TIME:
 711             i64 = (int64_t)va_arg( args, int64_t );
 712             return Seek( p_demux, i64 );
 713
 714         case DEMUX_GET_LENGTH:
 715             pi64 = (int64_t*)va_arg( args, int64_t * );
 716             if( p_sys->i_timescale > 0 )
 717             {
 718                 *pi64 = (mtime_t)1000000 *
 719                         (mtime_t)p_sys->i_duration /
 720                         (mtime_t)p_sys->i_timescale;
 721             }
 722             else *pi64 = 0;
 723             return VLC_SUCCESS;
 724
 725         case DEMUX_GET_FPS:
 726             msg_Warn( p_demux, "DEMUX_GET_FPS unimplemented !!" );
 727             return VLC_EGENERIC;
 728
 729         case DEMUX_GET_META:
 730         {
 731             vlc_meta_t **pp_meta = (vlc_meta_t**)va_arg( args, vlc_meta_t** );
 732             vlc_meta_t *meta;
 733             MP4_Box_t  *p_udta   = MP4_BoxGet( p_sys->p_root, "/moov/udta" );
 734             MP4_Box_t  *p_0xa9xxx;
 735             if( p_udta == NULL )
 736             {
 737                 return VLC_EGENERIC;
 738             }
 739             *pp_meta = meta = vlc_meta_New();
 740             for( p_0xa9xxx = p_udta->p_first; p_0xa9xxx != NULL;
 741                  p_0xa9xxx = p_0xa9xxx->p_next )
 742             {
 743                 switch( p_0xa9xxx->i_type )
 744                 {
 745                 case FOURCC_0xa9nam:
 746                     vlc_meta_Add( meta, VLC_META_TITLE,
 747                                   p_0xa9xxx->data.p_0xa9xxx->psz_text );
 748                     break;
 749                 case FOURCC_0xa9aut:
 750                     vlc_meta_Add( meta, VLC_META_AUTHOR,
 751                                   p_0xa9xxx->data.p_0xa9xxx->psz_text );
 752                     break;
 753                 case FOURCC_0xa9ART:
 754                     vlc_meta_Add( meta, VLC_META_ARTIST,
 755                                   p_0xa9xxx->data.p_0xa9xxx->psz_text );
 756                     break;
 757                 case FOURCC_0xa9cpy:
 758                     vlc_meta_Add( meta, VLC_META_COPYRIGHT,
 759                                   p_0xa9xxx->data.p_0xa9xxx->psz_text );
 760                     break;
 761                 case FOURCC_0xa9day:
 762                     vlc_meta_Add( meta, VLC_META_DATE,
 763                                   p_0xa9xxx->data.p_0xa9xxx->psz_text );
 764                     break;
 765                 case FOURCC_0xa9des:
 766                     vlc_meta_Add( meta, VLC_META_DESCRIPTION,
 767                                   p_0xa9xxx->data.p_0xa9xxx->psz_text );
 768                     break;
 769
 770                 case FOURCC_0xa9swr:
 771                 case FOURCC_0xa9inf:
 772                 case FOURCC_0xa9dir:
 773                 case FOURCC_0xa9cmt:
 774                 case FOURCC_0xa9req:
 775                 case FOURCC_0xa9fmt:
 776                 case FOURCC_0xa9prd:
 777                 case FOURCC_0xa9prf:
 778                 case FOURCC_0xa9src:
 779                     /* TODO one day, but they aren't really meaningfull */
 780                     break;
 781
 782                 default:
 783                     break;
 784                 }
 785             }
 786             return VLC_SUCCESS;
 787         }
 788
 789         default:
 790             msg_Err( p_demux, "control query unimplemented !!!" );
 791             return VLC_EGENERIC;
 792     }
 793 }
 794
 795 /*****************************************************************************
 796  * Close: frees unused data
 797  *****************************************************************************/
 798 static void Close ( vlc_object_t * p_this )
 799 {
 800     unsigned int i_track;
 801     demux_t *  p_demux = (demux_t *)p_this;
 802     demux_sys_t *p_sys = p_demux->p_sys;
 803
 804     msg_Dbg( p_demux, "freeing all memory" );
 805
 806     MP4_BoxFree( p_demux->s, p_sys->p_root );
 807     for( i_track = 0; i_track < p_sys->i_tracks; i_track++ )
 808     {
 809         MP4_TrackDestroy( p_demux, &p_sys->track[i_track] );
 810     }
 811     FREE( p_sys->track );
 812
 813     free( p_sys );
 814 }
 815
 816
 817
 818 /****************************************************************************
 819  * Local functions, specific to vlc
 820  ****************************************************************************/
 821
 822 /* now create basic chunk data, the rest will be filled by MP4_CreateSamplesIndex */
 823 static int TrackCreateChunksIndex( demux_t *p_demux,
 824                                    mp4_track_t *p_demux_track )
 825 {
 826     MP4_Box_t *p_co64; /* give offset for each chunk, same for stco and co64 */
 827     MP4_Box_t *p_stsc;
 828
 829     unsigned int i_chunk;
 830     unsigned int i_index, i_last;
 831
 832     if( ( !(p_co64 = MP4_BoxGet( p_demux_track->p_stbl, "stco" ) )&&
 833           !(p_co64 = MP4_BoxGet( p_demux_track->p_stbl, "co64" ) ) )||
 834         ( !(p_stsc = MP4_BoxGet( p_demux_track->p_stbl, "stsc" ) ) ))
 835     {
 836         return( VLC_EGENERIC );
 837     }
 838
 839     p_demux_track->i_chunk_count = p_co64->data.p_co64->i_entry_count;
 840     if( !p_demux_track->i_chunk_count )
 841     {
 842         msg_Warn( p_demux, "no chunk defined" );
 843         return( VLC_EGENERIC );
 844     }
 845     p_demux_track->chunk = calloc( p_demux_track->i_chunk_count,
 846                                    sizeof( mp4_chunk_t ) );
 847
 848     /* first we read chunk offset */
 849     for( i_chunk = 0; i_chunk < p_demux_track->i_chunk_count; i_chunk++ )
 850     {
 851         p_demux_track->chunk[i_chunk].i_offset =
 852                 p_co64->data.p_co64->i_chunk_offset[i_chunk];
 853     }
 854
 855     /* now we read index for SampleEntry( soun vide mp4a mp4v ...)
 856         to be used for the sample XXX begin to 1
 857         We construct it begining at the end */
 858     i_last = p_demux_track->i_chunk_count; /* last chunk proceded */
 859     i_index = p_stsc->data.p_stsc->i_entry_count;
 860     if( !i_index )
 861     {
 862         msg_Warn( p_demux, "cannot read chunk table or table empty" );
 863         return( VLC_EGENERIC );
 864     }
 865
 866     while( i_index )
 867     {
 868         i_index--;
 869         for( i_chunk = p_stsc->data.p_stsc->i_first_chunk[i_index] - 1;
 870                 i_chunk < i_last; i_chunk++ )
 871         {
 872             p_demux_track->chunk[i_chunk].i_sample_description_index =
 873                     p_stsc->data.p_stsc->i_sample_description_index[i_index];
 874             p_demux_track->chunk[i_chunk].i_sample_count =
 875                     p_stsc->data.p_stsc->i_samples_per_chunk[i_index];
 876         }
 877         i_last = p_stsc->data.p_stsc->i_first_chunk[i_index] - 1;
 878     }
 879
 880     p_demux_track->chunk[0].i_sample_first = 0;
 881     for( i_chunk = 1; i_chunk < p_demux_track->i_chunk_count; i_chunk++ )
 882     {
 883         p_demux_track->chunk[i_chunk].i_sample_first =
 884             p_demux_track->chunk[i_chunk-1].i_sample_first +
 885                 p_demux_track->chunk[i_chunk-1].i_sample_count;
 886     }
 887
 888     msg_Dbg( p_demux,
 889              "track[Id 0x%x] read %d chunk",
 890              p_demux_track->i_track_ID,
 891              p_demux_track->i_chunk_count );
 892
 893     return( VLC_SUCCESS );
 894 }
 895 static int TrackCreateSamplesIndex( demux_t *p_demux,
 896                                     mp4_track_t *p_demux_track )
 897 {
 898     MP4_Box_t *p_stts; /* makes mapping between sample and decoding time,
 899                           ctts make same mapping but for composition time,
 900                           not yet used and probably not usefull */
 901     MP4_Box_t *p_stsz; /* gives sample size of each samples, there is also stz2
 902                           that uses a compressed form FIXME make them in libmp4
 903                           as a unique type */
 904     /* TODO use also stss and stsh table for seeking */
 905     /* FIXME use edit table */
 906     int64_t i_sample;
 907     int64_t i_chunk;
 908
 909     int64_t i_index;
 910     int64_t i_index_sample_used;
 911
 912     int64_t i_last_dts;
 913
 914     p_stts = MP4_BoxGet( p_demux_track->p_stbl, "stts" );
 915     p_stsz = MP4_BoxGet( p_demux_track->p_stbl, "stsz" ); /* FIXME and stz2 */
 916
 917     if( ( !p_stts )||( !p_stsz ) )
 918     {
 919         msg_Warn( p_demux, "cannot read sample table" );
 920         return( VLC_EGENERIC );
 921     }
 922
 923     p_demux_track->i_sample_count = p_stsz->data.p_stsz->i_sample_count;
 924
 925
 926     /* for sample size, there are 2 case */
 927     if( p_stsz->data.p_stsz->i_sample_size )
 928     {
 929         /* 1: all sample have the same size, so no need to construct a table */
 930         p_demux_track->i_sample_size = p_stsz->data.p_stsz->i_sample_size;
 931         p_demux_track->p_sample_size = NULL;
 932     }
 933     else
 934     {
 935         /* 2: each sample can have a different size */
 936         p_demux_track->i_sample_size = 0;
 937         p_demux_track->p_sample_size =
 938             calloc( p_demux_track->i_sample_count, sizeof( uint32_t ) );
 939
 940         for( i_sample = 0; i_sample < p_demux_track->i_sample_count; i_sample++ )
 941         {
 942             p_demux_track->p_sample_size[i_sample] =
 943                     p_stsz->data.p_stsz->i_entry_size[i_sample];
 944         }
 945     }
 946     /* we have extract all information from stsz,
 947         now use stts */
 948
 949     /* if we don't want to waste too much memory, we can't expand
 950        the box !, so each chunk will contain an "extract" of this table
 951        for fast research */
 952
 953     i_last_dts = 0;
 954     i_index = 0; i_index_sample_used =0;
 955
 956     /* create and init last data for each chunk */
 957     for(i_chunk = 0 ; i_chunk < p_demux_track->i_chunk_count; i_chunk++ )
 958     {
 959
 960         int64_t i_entry, i_sample_count, i;
 961         /* save last dts */
 962         p_demux_track->chunk[i_chunk].i_first_dts = i_last_dts;
 963     /* count how many entries needed for this chunk
 964        for p_sample_delta_dts and p_sample_count_dts */
 965
 966         i_sample_count = p_demux_track->chunk[i_chunk].i_sample_count;
 967
 968         i_entry = 0;
 969         while( i_sample_count > 0 )
 970         {
 971             i_sample_count -= p_stts->data.p_stts->i_sample_count[i_index+i_entry];
 972             if( i_entry == 0 )
 973             {
 974                 i_sample_count += i_index_sample_used; /* don't count already used sample
 975                                                    int this entry */
 976             }
 977             i_entry++;
 978         }
 979
 980         /* allocate them */
 981         p_demux_track->chunk[i_chunk].p_sample_count_dts =
 982             calloc( i_entry, sizeof( uint32_t ) );
 983         p_demux_track->chunk[i_chunk].p_sample_delta_dts =
 984             calloc( i_entry, sizeof( uint32_t ) );
 985
 986         /* now copy */
 987         i_sample_count = p_demux_track->chunk[i_chunk].i_sample_count;
 988         for( i = 0; i < i_entry; i++ )
 989         {
 990             int64_t i_used;
 991             int64_t i_rest;
 992
 993             i_rest = p_stts->data.p_stts->i_sample_count[i_index] - i_index_sample_used;
 994
 995             i_used = __MIN( i_rest, i_sample_count );
 996
 997             i_index_sample_used += i_used;
 998             i_sample_count -= i_used;
 999
1000             p_demux_track->chunk[i_chunk].p_sample_count_dts[i] = i_used;
1001
1002             p_demux_track->chunk[i_chunk].p_sample_delta_dts[i] =
1003                         p_stts->data.p_stts->i_sample_delta[i_index];
1004
1005             i_last_dts += i_used *
1006                     p_demux_track->chunk[i_chunk].p_sample_delta_dts[i];
1007
1008             if( i_index_sample_used >=
1009                              p_stts->data.p_stts->i_sample_count[i_index] )
1010             {
1011
1012                 i_index++;
1013                 i_index_sample_used = 0;
1014             }
1015         }
1016
1017     }
1018
1019     msg_Dbg( p_demux, "track[Id 0x%x] read %d samples length:"I64Fd"s",
1020              p_demux_track->i_track_ID,
1021              p_demux_track->i_sample_count,
1022              i_last_dts / p_demux_track->i_timescale );
1023
1024     return( VLC_SUCCESS );
1025 }
1026
1027 /*
1028  * TrackCreateES:
1029  * Create ES and PES to init decoder if needed, for a track starting at i_chunk
1030  */
1031 static int TrackCreateES( demux_t *p_demux, mp4_track_t *p_track,
1032                           unsigned int i_chunk, es_out_id_t **pp_es )
1033 {
1034     MP4_Box_t   *p_sample;
1035     MP4_Box_t   *p_esds;
1036
1037     *pp_es = NULL;
1038
1039     if( !p_track->chunk[i_chunk].i_sample_description_index )
1040     {
1041         msg_Warn( p_demux,
1042                   "invalid SampleEntry index (track[Id 0x%x])",
1043                   p_track->i_track_ID );
1044         return VLC_EGENERIC;
1045     }
1046
1047     p_sample = MP4_BoxGet(  p_track->p_stsd, "[%d]",
1048                 p_track->chunk[i_chunk].i_sample_description_index - 1 );
1049
1050     if( !p_sample || !p_sample->data.p_data )
1051     {
1052         msg_Warn( p_demux,
1053                   "cannot find SampleEntry (track[Id 0x%x])",
1054                   p_track->i_track_ID );
1055         return( VLC_EGENERIC );
1056     }
1057
1058     p_track->p_sample = p_sample;
1059
1060     if( p_track->i_sample_size == 1 )
1061     {
1062         MP4_Box_data_sample_soun_t *p_soun;
1063
1064         p_soun = p_sample->data.p_sample_soun;
1065
1066         if( p_soun->i_qt_version == 0 )
1067         {
1068             switch( p_sample->i_type )
1069             {
1070                 case VLC_FOURCC( 'i', 'm', 'a', '4' ):
1071                     p_soun->i_qt_version = 1;
1072                     p_soun->i_sample_per_packet = 64;
1073                     p_soun->i_bytes_per_packet  = 34;
1074                     p_soun->i_bytes_per_frame   = 34 * p_soun->i_channelcount;
1075                     p_soun->i_bytes_per_sample  = 2;
1076                     break;
1077                 case VLC_FOURCC( 'M', 'A', 'C', '3' ):
1078                     p_soun->i_qt_version = 1;
1079                     p_soun->i_sample_per_packet = 6;
1080                     p_soun->i_bytes_per_packet  = 2;
1081                     p_soun->i_bytes_per_frame   = 2 * p_soun->i_channelcount;
1082                     p_soun->i_bytes_per_sample  = 2;
1083                     break;
1084                 case VLC_FOURCC( 'M', 'A', 'C', '6' ):
1085                     p_soun->i_qt_version = 1;
1086                     p_soun->i_sample_per_packet = 12;
1087                     p_soun->i_bytes_per_packet  = 2;
1088                     p_soun->i_bytes_per_frame   = 2 * p_soun->i_channelcount;
1089                     p_soun->i_bytes_per_sample  = 2;
1090                     break;
1091                 case VLC_FOURCC( 'a', 'l', 'a', 'w' ):
1092                 case VLC_FOURCC( 'u', 'l', 'a', 'w' ):
1093                     p_soun->i_samplesize = 8;
1094                     break;
1095                 default:
1096                     break;
1097             }
1098
1099         }
1100         else if( p_soun->i_qt_version == 1 && p_soun->i_sample_per_packet <= 0 )
1101         {
1102             p_soun->i_qt_version = 0;
1103         }
1104     }
1105
1106
1107     /* It's a little ugly but .. there are special cases */
1108     switch( p_sample->i_type )
1109     {
1110         case( VLC_FOURCC( '.', 'm', 'p', '3' ) ):
1111         case( VLC_FOURCC( 'm', 's', 0x00, 0x55 ) ):
1112             p_track->fmt.i_codec = VLC_FOURCC( 'm', 'p', 'g', 'a' );
1113             break;
1114         case( VLC_FOURCC( 'r', 'a', 'w', ' ' ) ):
1115             p_track->fmt.i_codec = VLC_FOURCC( 'a', 'r', 'a', 'w' );
1116             break;
1117         case( VLC_FOURCC( 's', '2', '6', '3' ) ):
1118             p_track->fmt.i_codec = VLC_FOURCC( 'h', '2', '6', '3' );
1119             break;
1120
1121         case( VLC_FOURCC( 't', 'e', 'x', 't' ) ):
1122             p_track->fmt.i_codec = VLC_FOURCC( 's', 'u', 'b', 't' );
1123             /* FIXME: Not true, could be UTF-16 with a Byte Order Mark (0xfeff) */
1124             /* FIXME UTF-8 doesn't work here ? */
1125             /* p_track->fmt.subs.psz_encoding = strdup( "UTF-8" ); */
1126             break;
1127
1128         default:
1129             p_track->fmt.i_codec = p_sample->i_type;
1130             break;
1131     }
1132
1133     /* now see if esds is present and if so create a data packet
1134         with decoder_specific_info  */
1135 #define p_decconfig p_esds->data.p_esds->es_descriptor.p_decConfigDescr
1136     if( ( ( p_esds = MP4_BoxGet( p_sample, "esds" ) ) ||
1137           ( p_esds = MP4_BoxGet( p_sample, "wave/esds" ) ) )&&
1138         ( p_esds->data.p_esds )&&
1139         ( p_decconfig ) )
1140     {
1141         /* First update information based on i_objectTypeIndication */
1142         switch( p_decconfig->i_objectTypeIndication )
1143         {
1144             case( 0x20 ): /* MPEG4 VIDEO */
1145                 p_track->fmt.i_codec = VLC_FOURCC( 'm','p','4','v' );
1146                 break;
1147             case( 0x40):
1148                 p_track->fmt.i_codec = VLC_FOURCC( 'm','p','4','a' );
1149                 break;
1150             case( 0x60):
1151             case( 0x61):
1152             case( 0x62):
1153             case( 0x63):
1154             case( 0x64):
1155             case( 0x65): /* MPEG2 video */
1156                 p_track->fmt.i_codec = VLC_FOURCC( 'm','p','g','v' );
1157                 break;
1158             /* Theses are MPEG2-AAC */
1159             case( 0x66): /* main profile */
1160             case( 0x67): /* Low complexity profile */
1161             case( 0x68): /* Scaleable Sampling rate profile */
1162                 p_track->fmt.i_codec = VLC_FOURCC( 'm','p','4','a' );
1163                 break;
1164             /* true MPEG 2 audio */
1165             case( 0x69):
1166                 p_track->fmt.i_codec = VLC_FOURCC( 'm','p','g','a' );
1167                 break;
1168             case( 0x6a): /* MPEG1 video */
1169                 p_track->fmt.i_codec = VLC_FOURCC( 'm','p','g','v' );
1170                 break;
1171             case( 0x6b): /* MPEG1 audio */
1172                 p_track->fmt.i_codec = VLC_FOURCC( 'm','p','g','a' );
1173                 break;
1174             case( 0x6c ): /* jpeg */
1175                 p_track->fmt.i_codec = VLC_FOURCC( 'j','p','e','g' );
1176                 break;
1177             default:
1178                 /* Unknown entry, but don't touch i_fourcc */
1179                 msg_Warn( p_demux,
1180                           "unknown objectTypeIndication(0x%x) (Track[ID 0x%x])",
1181                           p_decconfig->i_objectTypeIndication,
1182                           p_track->i_track_ID );
1183                 break;
1184         }
1185         p_track->fmt.i_extra = p_decconfig->i_decoder_specific_info_len;
1186         if( p_track->fmt.i_extra > 0 )
1187         {
1188             p_track->fmt.p_extra = malloc( p_track->fmt.i_extra );
1189             memcpy( p_track->fmt.p_extra, p_decconfig->p_decoder_specific_info,
1190                     p_track->fmt.i_extra );
1191         }
1192     }
1193     else
1194     {
1195         switch( p_sample->i_type )
1196         {
1197             /* qt decoder, send the complete chunk */
1198             case VLC_FOURCC( 'S', 'V', 'Q', '3' ):
1199             case VLC_FOURCC( 'S', 'V', 'Q', '1' ):
1200             case VLC_FOURCC( 'V', 'P', '3', '1' ):
1201             case VLC_FOURCC( '3', 'I', 'V', '1' ):
1202             case VLC_FOURCC( 'Z', 'y', 'G', 'o' ):
1203                 p_track->fmt.i_extra =
1204                     p_sample->data.p_sample_vide->i_qt_image_description;
1205                 if( p_track->fmt.i_extra > 0 )
1206                 {
1207                     p_track->fmt.p_extra = malloc( p_track->fmt.i_extra );
1208                     memcpy( p_track->fmt.p_extra,
1209                             p_sample->data.p_sample_vide->p_qt_image_description,
1210                             p_track->fmt.i_extra);
1211                 }
1212                 break;
1213             case VLC_FOURCC( 'Q', 'D', 'M', 'C' ):
1214             case VLC_FOURCC( 'Q', 'D', 'M', '2' ):
1215             case VLC_FOURCC( 'Q', 'c', 'l', 'p' ):
1216             case VLC_FOURCC( 's', 'a', 'm', 'r' ):
1217                 p_track->fmt.i_extra =
1218                     p_sample->data.p_sample_soun->i_qt_description;
1219                 if( p_track->fmt.i_extra > 0 )
1220                 {
1221                     p_track->fmt.p_extra = malloc( p_track->fmt.i_extra );
1222                     memcpy( p_track->fmt.p_extra,
1223                             p_sample->data.p_sample_soun->p_qt_description,
1224                             p_track->fmt.i_extra);
1225                 }
1226                 break;
1227
1228             /* avc1: send avcC (h264 without annexe B, ie without start code)*/
1229             case VLC_FOURCC( 'a', 'v', 'c', '1' ):
1230             {
1231                 MP4_Box_t *p_avcC = MP4_BoxGet( p_sample, "avcC" );
1232
1233                 if( p_avcC )
1234                 {
1235                     /* Hack: use a packetizer to reecampsulate data in anexe B format */
1236                     msg_Dbg( p_demux, "avcC: size=%d", p_avcC->data.p_avcC->i_avcC );
1237                     p_track->fmt.i_extra = p_avcC->data.p_avcC->i_avcC;
1238                     p_track->fmt.p_extra = malloc( p_avcC->data.p_avcC->i_avcC );
1239                     memcpy( p_track->fmt.p_extra, p_avcC->data.p_avcC->p_avcC, p_track->fmt.i_extra );
1240                     p_track->fmt.b_packetized = VLC_FALSE;
1241                 }
1242                 else
1243                 {
1244                     msg_Err( p_demux, "missing avcC" );
1245                 }
1246                 break;
1247             }
1248
1249             default:
1250                 break;
1251         }
1252     }
1253
1254 #undef p_decconfig
1255
1256     /* some last initialisation */
1257     switch( p_track->fmt.i_cat )
1258     {
1259     case( VIDEO_ES ):
1260         p_track->fmt.video.i_width = p_sample->data.p_sample_vide->i_width;
1261         p_track->fmt.video.i_height = p_sample->data.p_sample_vide->i_height;
1262
1263         /* fall on display size */
1264         if( p_track->fmt.video.i_width <= 0 )
1265             p_track->fmt.video.i_width = p_track->i_width;
1266         if( p_track->fmt.video.i_height <= 0 )
1267             p_track->fmt.video.i_height = p_track->i_height;
1268
1269         /* Find out apect ratio from display size */
1270         if( p_track->i_width > 0 && p_track->i_height > 0 &&
1271             /* Work-around buggy muxed files */
1272             p_sample->data.p_sample_vide->i_width != p_track->i_width )
1273             p_track->fmt.video.i_aspect =
1274                 VOUT_ASPECT_FACTOR * p_track->i_width / p_track->i_height;
1275
1276         break;
1277
1278     case( AUDIO_ES ):
1279         p_track->fmt.audio.i_channels =
1280             p_sample->data.p_sample_soun->i_channelcount;
1281         p_track->fmt.audio.i_rate =
1282             p_sample->data.p_sample_soun->i_sampleratehi;
1283         p_track->fmt.i_bitrate = p_sample->data.p_sample_soun->i_channelcount *
1284             p_sample->data.p_sample_soun->i_sampleratehi *
1285                 p_sample->data.p_sample_soun->i_samplesize;
1286         p_track->fmt.audio.i_bitspersample =
1287             p_sample->data.p_sample_soun->i_samplesize;
1288         break;
1289
1290     default:
1291         break;
1292     }
1293
1294     *pp_es = es_out_Add( p_demux->out, &p_track->fmt );
1295
1296     return VLC_SUCCESS;
1297 }
1298
1299 /* given a time it return sample/chunk
1300  * it also update elst field of the track
1301  */
1302 static int TrackTimeToSampleChunk( demux_t *p_demux, mp4_track_t *p_track,
1303                                    int64_t i_start, uint32_t *pi_chunk,
1304                                    uint32_t *pi_sample )
1305 {
1306     demux_sys_t *p_sys = p_demux->p_sys;
1307     MP4_Box_t   *p_stss;
1308     uint64_t     i_dts;
1309     unsigned int i_sample;
1310     unsigned int i_chunk;
1311     int          i_index;
1312
1313     /* FIXME see if it's needed to check p_track->i_chunk_count */
1314     if( !p_track->b_ok || p_track->i_chunk_count == 0 )
1315     {
1316         return( VLC_EGENERIC );
1317     }
1318
1319     /* handle elst (find the correct one) */
1320     MP4_TrackSetELST( p_demux, p_track, i_start );
1321     if( p_track->p_elst && p_track->p_elst->data.p_elst->i_entry_count > 0 )
1322     {
1323         MP4_Box_data_elst_t *elst = p_track->p_elst->data.p_elst;
1324         int64_t i_mvt= i_start * p_sys->i_timescale / (int64_t)1000000;
1325
1326         /* now calculate i_start for this elst */
1327         /* offset */
1328         i_start -= p_track->i_elst_time * I64C(1000000) / p_sys->i_timescale;
1329         if( i_start < 0 )
1330         {
1331             *pi_chunk = 0;
1332             *pi_sample= 0;
1333
1334             return VLC_SUCCESS;
1335         }
1336         /* to track time scale */
1337         i_start  = i_start * p_track->i_timescale / (int64_t)1000000;
1338         /* add elst offset */
1339         if( ( elst->i_media_rate_integer[p_track->i_elst] > 0 ||
1340              elst->i_media_rate_fraction[p_track->i_elst] > 0 ) &&
1341             elst->i_media_time[p_track->i_elst] > 0 )
1342         {
1343             i_start += elst->i_media_time[p_track->i_elst];
1344         }
1345
1346         msg_Dbg( p_demux, "elst (%d) gives "I64Fd"ms (movie)-> "I64Fd
1347                  "ms (track)", p_track->i_elst,
1348                  i_mvt * 1000 / p_sys->i_timescale,
1349                  i_start * 1000 / p_track->i_timescale );
1350     }
1351     else
1352     {
1353         /* convert absolute time to in timescale unit */
1354         i_start = i_start * p_track->i_timescale / (int64_t)1000000;
1355     }
1356
1357     /* we start from sample 0/chunk 0, hope it won't take too much time */
1358     /* *** find good chunk *** */
1359     for( i_chunk = 0; ; i_chunk++ )
1360     {
1361         if( i_chunk + 1 >= p_track->i_chunk_count )
1362         {
1363             /* at the end and can't check if i_start in this chunk,
1364                it will be check while searching i_sample */
1365             i_chunk = p_track->i_chunk_count - 1;
1366             break;
1367         }
1368
1369         if( i_start >= p_track->chunk[i_chunk].i_first_dts &&
1370             i_start <  p_track->chunk[i_chunk + 1].i_first_dts )
1371         {
1372             break;
1373         }
1374     }
1375
1376     /* *** find sample in the chunk *** */
1377     i_sample = p_track->chunk[i_chunk].i_sample_first;
1378     i_dts    = p_track->chunk[i_chunk].i_first_dts;
1379     for( i_index = 0; i_sample < p_track->chunk[i_chunk].i_sample_count; )
1380     {
1381         if( i_dts +
1382             p_track->chunk[i_chunk].p_sample_count_dts[i_index] *
1383             p_track->chunk[i_chunk].p_sample_delta_dts[i_index] < i_start )
1384         {
1385             i_dts    +=
1386                 p_track->chunk[i_chunk].p_sample_count_dts[i_index] *
1387                 p_track->chunk[i_chunk].p_sample_delta_dts[i_index];
1388
1389             i_sample += p_track->chunk[i_chunk].p_sample_count_dts[i_index];
1390             i_index++;
1391         }
1392         else
1393         {
1394             if( p_track->chunk[i_chunk].p_sample_delta_dts[i_index] <= 0 )
1395             {
1396                 break;
1397             }
1398             i_sample += ( i_start - i_dts ) /
1399                 p_track->chunk[i_chunk].p_sample_delta_dts[i_index];
1400             break;
1401         }
1402     }
1403
1404     if( i_sample >= p_track->i_sample_count )
1405     {
1406         msg_Warn( p_demux, "track[Id 0x%x] will be disabled "
1407                   "(seeking too far) chunk=%d sample=%d",
1408                   p_track->i_track_ID, i_chunk, i_sample );
1409         return( VLC_EGENERIC );
1410     }
1411
1412
1413     /* *** Try to find nearest sync points *** */
1414     if( ( p_stss = MP4_BoxGet( p_track->p_stbl, "stss" ) ) )
1415     {
1416         unsigned int i_index;
1417         msg_Dbg( p_demux,
1418                     "track[Id 0x%x] using Sync Sample Box (stss)",
1419                     p_track->i_track_ID );
1420         for( i_index = 0; i_index < p_stss->data.p_stss->i_entry_count; i_index++ )
1421         {
1422             if( p_stss->data.p_stss->i_sample_number[i_index] >= i_sample )
1423             {
1424                 if( i_index > 0 )
1425                 {
1426                     msg_Dbg( p_demux, "stts gives %d --> %d (sample number)",
1427                             i_sample,
1428                             p_stss->data.p_stss->i_sample_number[i_index-1] );
1429                     i_sample = p_stss->data.p_stss->i_sample_number[i_index-1];
1430                     /* new i_sample is less than old so i_chunk can only decreased */
1431                     while( i_chunk > 0 &&
1432                             i_sample < p_track->chunk[i_chunk].i_sample_first )
1433                     {
1434                         i_chunk--;
1435                     }
1436                 }
1437                 else
1438                 {
1439                     msg_Dbg( p_demux, "stts gives %d --> %d (sample number)",
1440                             i_sample,
1441                             p_stss->data.p_stss->i_sample_number[i_index] );
1442                     i_sample = p_stss->data.p_stss->i_sample_number[i_index];
1443                     /* new i_sample is more than old so i_chunk can only increased */
1444                     while( i_chunk < p_track->i_chunk_count - 1 &&
1445                            i_sample >= p_track->chunk[i_chunk].i_sample_first +
1446                              p_track->chunk[i_chunk].i_sample_count )
1447                     {
1448                         i_chunk++;
1449                     }
1450                 }
1451                 break;
1452             }
1453         }
1454     }
1455     else
1456     {
1457         msg_Dbg( p_demux, "track[Id 0x%x] does not provide Sync "
1458                  "Sample Box (stss)", p_track->i_track_ID );
1459     }
1460
1461     *pi_chunk  = i_chunk;
1462     *pi_sample = i_sample;
1463
1464     return VLC_SUCCESS;
1465 }
1466
1467 static int TrackGotoChunkSample( demux_t *p_demux, mp4_track_t *p_track,
1468                                  unsigned int i_chunk, unsigned int i_sample )
1469 {
1470     vlc_bool_t b_reselect = VLC_FALSE;
1471
1472     /* now see if actual es is ok */
1473     if( p_track->i_chunk < 0 ||
1474         p_track->i_chunk >= p_track->i_chunk_count - 1 ||
1475         p_track->chunk[p_track->i_chunk].i_sample_description_index !=
1476             p_track->chunk[i_chunk].i_sample_description_index )
1477     {
1478         msg_Warn( p_demux, "recreate ES for track[Id 0x%x]",
1479                   p_track->i_track_ID );
1480
1481         es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE,
1482                         p_track->p_es, &b_reselect );
1483
1484         es_out_Del( p_demux->out, p_track->p_es );
1485
1486         p_track->p_es = NULL;
1487
1488         if( TrackCreateES( p_demux, p_track, i_chunk, &p_track->p_es ) )
1489         {
1490             msg_Err( p_demux, "cannot create es for track[Id 0x%x]",
1491                      p_track->i_track_ID );
1492
1493             p_track->b_ok       = VLC_FALSE;
1494             p_track->b_selected = VLC_FALSE;
1495             return VLC_EGENERIC;
1496         }
1497     }
1498
1499     /* select again the new decoder */
1500     if( b_reselect )
1501     {
1502         es_out_Control( p_demux->out, ES_OUT_SET_ES, p_track->p_es );
1503     }
1504
1505     p_track->i_chunk    = i_chunk;
1506     p_track->i_sample   = i_sample;
1507
1508     return p_track->b_selected ? VLC_SUCCESS : VLC_EGENERIC;
1509 }
1510
1511 /****************************************************************************
1512  * MP4_TrackCreate:
1513  ****************************************************************************
1514  * Parse track information and create all needed data to run a track
1515  * If it succeed b_ok is set to 1 else to 0
1516  ****************************************************************************/
1517 static void MP4_TrackCreate( demux_t *p_demux, mp4_track_t *p_track,
1518                              MP4_Box_t *p_box_trak )
1519 {
1520     demux_sys_t *p_sys = p_demux->p_sys;
1521
1522     MP4_Box_t *p_tkhd = MP4_BoxGet( p_box_trak, "tkhd" );
1523     MP4_Box_t *p_tref = MP4_BoxGet( p_box_trak, "tref" );
1524     MP4_Box_t *p_elst;
1525
1526     MP4_Box_t *p_mdhd;
1527     MP4_Box_t *p_hdlr;
1528
1529     MP4_Box_t *p_vmhd;
1530     MP4_Box_t *p_smhd;
1531
1532     MP4_Box_t *p_drms;
1533
1534     unsigned int i;
1535     char language[4];
1536
1537     /* hint track unsuported */
1538
1539     /* set default value (-> track unusable) */
1540     p_track->b_ok       = VLC_FALSE;
1541     p_track->b_enable   = VLC_FALSE;
1542     p_track->b_selected = VLC_FALSE;
1543
1544     es_format_Init( &p_track->fmt, UNKNOWN_ES, 0 );
1545
1546     if( !p_tkhd )
1547     {
1548         return;
1549     }
1550
1551     /* do we launch this track by default ? */
1552     p_track->b_enable =
1553         ( ( p_tkhd->data.p_tkhd->i_flags&MP4_TRACK_ENABLED ) != 0 );
1554
1555     p_track->i_track_ID = p_tkhd->data.p_tkhd->i_track_ID;
1556     p_track->i_width = p_tkhd->data.p_tkhd->i_width / 65536;
1557     p_track->i_height = p_tkhd->data.p_tkhd->i_height / 65536;
1558
1559     if( p_tref )
1560     {
1561 /*        msg_Warn( p_demux, "unhandled box: tref --> FIXME" ); */
1562     }
1563
1564     p_mdhd = MP4_BoxGet( p_box_trak, "mdia/mdhd" );
1565     p_hdlr = MP4_BoxGet( p_box_trak, "mdia/hdlr" );
1566
1567     if( ( !p_mdhd )||( !p_hdlr ) )
1568     {
1569         return;
1570     }
1571
1572     p_track->i_timescale = p_mdhd->data.p_mdhd->i_timescale;
1573
1574     for( i = 0; i < 3; i++ )
1575     {
1576         language[i] = p_mdhd->data.p_mdhd->i_language[i];
1577     }
1578     language[3] = '\0';
1579
1580     switch( p_hdlr->data.p_hdlr->i_handler_type )
1581     {
1582         case( FOURCC_soun ):
1583             if( !( p_smhd = MP4_BoxGet( p_box_trak, "mdia/minf/smhd" ) ) )
1584             {
1585                 return;
1586             }
1587             p_track->fmt.i_cat = AUDIO_ES;
1588             break;
1589
1590         case( FOURCC_vide ):
1591             if( !( p_vmhd = MP4_BoxGet( p_box_trak, "mdia/minf/vmhd" ) ) )
1592             {
1593                 return;
1594             }
1595             p_track->fmt.i_cat = VIDEO_ES;
1596             break;
1597
1598         case( FOURCC_text ):
1599             p_track->fmt.i_cat = SPU_ES;
1600             break;
1601
1602         default:
1603             return;
1604     }
1605
1606     p_track->i_elst = 0;
1607     p_track->i_elst_time = 0;
1608     if( ( p_track->p_elst = p_elst = MP4_BoxGet( p_box_trak, "edts/elst" ) ) )
1609     {
1610         MP4_Box_data_elst_t *elst = p_elst->data.p_elst;
1611         int i;
1612
1613         msg_Warn( p_demux, "elst box found" );
1614         for( i = 0; i < elst->i_entry_count; i++ )
1615         {
1616             msg_Dbg( p_demux, "   - [%d] duration="I64Fd"ms media time="I64Fd"ms) rate=%d.%d",
1617                      i,
1618                      elst->i_segment_duration[i] * 1000 / p_sys->i_timescale,
1619                      elst->i_media_time[i] >= 0 ?
1620                         elst->i_media_time[i] * 1000 / p_track->i_timescale : -1,
1621                      elst->i_media_rate_integer[i],
1622                      elst->i_media_rate_fraction[i] );
1623         }
1624     }
1625
1626
1627 /*  TODO
1628     add support for:
1629     p_dinf = MP4_BoxGet( p_minf, "dinf" );
1630 */
1631     if( !( p_track->p_stbl = MP4_BoxGet( p_box_trak,"mdia/minf/stbl" ) ) ||
1632         !( p_track->p_stsd = MP4_BoxGet( p_box_trak,"mdia/minf/stbl/stsd") ) )
1633     {
1634         return;
1635     }
1636
1637     p_drms = MP4_BoxGet( p_track->p_stsd, "drms" );
1638     p_track->b_drms = p_drms != NULL;
1639     p_track->p_drms = p_track->b_drms ?
1640         p_drms->data.p_sample_soun->p_drms : NULL;
1641
1642     /* Set language */
1643     if( strcmp( language, "```" ) && strcmp( language, "und" ) )
1644     {
1645         p_track->fmt.psz_language = strdup( language );
1646     }
1647
1648     /* fxi i_timescale for AUDIO_ES with i_qt_version == 0 */
1649     if( p_track->fmt.i_cat == AUDIO_ES ) //&& p_track->i_sample_size == 1 )
1650     {
1651         MP4_Box_t *p_sample;
1652
1653         p_sample = MP4_BoxGet(  p_track->p_stsd, "[0]" );
1654         if( p_sample && p_sample->data.p_sample_soun)
1655         {
1656             MP4_Box_data_sample_soun_t *p_soun = p_sample->data.p_sample_soun;
1657             if( p_soun->i_qt_version == 0 &&
1658                 p_track->i_timescale != p_soun->i_sampleratehi )
1659             {
1660                 msg_Warn( p_demux,
1661                           "i_timescale ("I64Fu") != i_sampleratehi (%u) with "
1662                           "qt_version == 0\n"
1663                           "Making both equal. (report any problem)",
1664                           p_track->i_timescale, p_soun->i_sampleratehi );
1665
1666                 if( p_soun->i_sampleratehi )
1667                     p_track->i_timescale = p_soun->i_sampleratehi;
1668                 else
1669                     p_soun->i_sampleratehi = p_track->i_timescale;
1670             }
1671         }
1672     }
1673
1674     /* Create chunk index table and sample index table */
1675     if( TrackCreateChunksIndex( p_demux,p_track  ) ||
1676         TrackCreateSamplesIndex( p_demux, p_track ) )
1677     {
1678         return; /* cannot create chunks index */
1679     }
1680
1681     p_track->i_chunk  = 0;
1682     p_track->i_sample = 0;
1683
1684     /* now create es */
1685     if( TrackCreateES( p_demux,
1686                        p_track, p_track->i_chunk,
1687                        &p_track->p_es ) )
1688     {
1689         msg_Err( p_demux, "cannot create es for track[Id 0x%x]",
1690                  p_track->i_track_ID );
1691         return;
1692     }
1693
1694 #if 0
1695     {
1696         int i;
1697         for( i = 0; i < p_track->i_chunk_count; i++ )
1698         {
1699             fprintf( stderr, "%-5d sample_count=%d pts=%lld\n",
1700                      i, p_track->chunk[i].i_sample_count,
1701                      p_track->chunk[i].i_first_dts );
1702
1703         }
1704     }
1705 #endif
1706     p_track->b_ok = VLC_TRUE;
1707 }
1708
1709 /****************************************************************************
1710  * MP4_TrackDestroy:
1711  ****************************************************************************
1712  * Destroy a track created by MP4_TrackCreate.
1713  ****************************************************************************/
1714 static void MP4_TrackDestroy( demux_t *p_demux, mp4_track_t *p_track )
1715 {
1716     unsigned int i_chunk;
1717
1718     p_track->b_ok = VLC_FALSE;
1719     p_track->b_enable   = VLC_FALSE;
1720     p_track->b_selected = VLC_FALSE;
1721
1722     es_format_Clean( &p_track->fmt );
1723
1724     for( i_chunk = 0; i_chunk < p_track->i_chunk_count; i_chunk++ )
1725     {
1726         if( p_track->chunk )
1727         {
1728            FREE(p_track->chunk[i_chunk].p_sample_count_dts);
1729            FREE(p_track->chunk[i_chunk].p_sample_delta_dts );
1730         }
1731     }
1732     FREE( p_track->chunk );
1733
1734     if( !p_track->i_sample_size )
1735     {
1736         FREE( p_track->p_sample_size );
1737     }
1738 }
1739
1740 static int MP4_TrackSelect( demux_t *p_demux, mp4_track_t *p_track,
1741                             mtime_t i_start )
1742 {
1743     if( !p_track->b_ok )
1744     {
1745         return VLC_EGENERIC;
1746     }
1747
1748     if( p_track->b_selected )
1749     {
1750         msg_Warn( p_demux, "track[Id 0x%x] already selected",
1751                   p_track->i_track_ID );
1752         return VLC_SUCCESS;
1753     }
1754
1755     return MP4_TrackSeek( p_demux, p_track, i_start );
1756 }
1757
1758 static void MP4_TrackUnselect( demux_t *p_demux, mp4_track_t *p_track )
1759 {
1760     if( !p_track->b_ok )
1761     {
1762         return;
1763     }
1764
1765     if( !p_track->b_selected )
1766     {
1767         msg_Warn( p_demux, "track[Id 0x%x] already unselected",
1768                   p_track->i_track_ID );
1769         return;
1770     }
1771     if( p_track->p_es )
1772     {
1773         es_out_Control( p_demux->out, ES_OUT_SET_ES_STATE,
1774                         p_track->p_es, VLC_FALSE );
1775     }
1776
1777     p_track->b_selected = VLC_FALSE;
1778 }
1779
1780 static int MP4_TrackSeek( demux_t *p_demux, mp4_track_t *p_track,
1781                           mtime_t i_start )
1782 {
1783     uint32_t i_chunk;
1784     uint32_t i_sample;
1785
1786     if( !p_track->b_ok )
1787     {
1788         return( VLC_EGENERIC );
1789     }
1790
1791     p_track->b_selected = VLC_FALSE;
1792
1793     if( TrackTimeToSampleChunk( p_demux, p_track, i_start,
1794                                 &i_chunk, &i_sample ) )
1795     {
1796         msg_Warn( p_demux, "cannot select track[Id 0x%x]",
1797                   p_track->i_track_ID );
1798         return( VLC_EGENERIC );
1799     }
1800
1801     p_track->b_selected = VLC_TRUE;
1802
1803     if( TrackGotoChunkSample( p_demux, p_track, i_chunk, i_sample ) ==
1804         VLC_SUCCESS )
1805     {
1806         p_track->b_selected = VLC_TRUE;
1807     }
1808     return( p_track->b_selected ? VLC_SUCCESS : VLC_EGENERIC );
1809 }
1810
1811
1812 /*
1813  * 3 types: for audio
1814  *
1815  */
1816 #define QT_V0_MAX_SAMPLES    1500
1817 static int MP4_TrackSampleSize( mp4_track_t *p_track )
1818 {
1819     int i_size;
1820     MP4_Box_data_sample_soun_t *p_soun;
1821
1822     if( p_track->i_sample_size == 0 )
1823     {
1824         /* most simple case */
1825         return( p_track->p_sample_size[p_track->i_sample] );
1826     }
1827     if( p_track->fmt.i_cat != AUDIO_ES )
1828     {
1829         return( p_track->i_sample_size );
1830     }
1831
1832     if( p_track->i_sample_size != 1 )
1833     {
1834         //msg_Warn( p_demux, "SampleSize != 1" );
1835         return( p_track->i_sample_size );
1836     }
1837
1838     p_soun = p_track->p_sample->data.p_sample_soun;
1839
1840     if( p_soun->i_qt_version == 1 )
1841     {
1842         i_size = p_track->chunk[p_track->i_chunk].i_sample_count / p_soun->i_sample_per_packet * p_soun->i_bytes_per_frame;
1843     }
1844     else
1845     {
1846         /* FIXME */
1847         int i_samples = p_track->chunk[p_track->i_chunk].i_sample_count -
1848                 ( p_track->i_sample - p_track->chunk[p_track->i_chunk].i_sample_first );
1849         if( i_samples > QT_V0_MAX_SAMPLES )
1850         {
1851             i_samples = QT_V0_MAX_SAMPLES;
1852         }
1853         i_size = i_samples * p_soun->i_channelcount * p_soun->i_samplesize / 8;
1854     }
1855
1856     //fprintf( stderr, "size=%d\n", i_size );
1857     return( i_size );
1858 }
1859
1860
1861 static uint64_t MP4_TrackGetPos( mp4_track_t *p_track )
1862 {
1863     unsigned int i_sample;
1864     uint64_t i_pos;
1865
1866
1867     i_pos = p_track->chunk[p_track->i_chunk].i_offset;
1868
1869     if( p_track->i_sample_size )
1870     {
1871         MP4_Box_data_sample_soun_t *p_soun = p_track->p_sample->data.p_sample_soun;
1872
1873         if( p_soun->i_qt_version == 0 )
1874         {
1875             i_pos += ( p_track->i_sample - p_track->chunk[p_track->i_chunk].i_sample_first ) *
1876                         p_soun->i_channelcount * p_soun->i_samplesize / 8;
1877         }
1878         else
1879         {
1880             /* we read chunk by chunk */
1881             i_pos += 0;
1882         }
1883     }
1884     else
1885     {
1886         for( i_sample = p_track->chunk[p_track->i_chunk].i_sample_first;
1887                 i_sample < p_track->i_sample; i_sample++ )
1888         {
1889             i_pos += p_track->p_sample_size[i_sample];
1890         }
1891
1892     }
1893     return( i_pos );
1894 }
1895
1896 static int  MP4_TrackNextSample( demux_t     *p_demux,
1897                                  mp4_track_t   *p_track )
1898 {
1899
1900     if( p_track->fmt.i_cat == AUDIO_ES &&
1901         p_track->i_sample_size != 0 )
1902     {
1903         MP4_Box_data_sample_soun_t *p_soun;
1904
1905         p_soun = p_track->p_sample->data.p_sample_soun;
1906
1907         if( p_soun->i_qt_version == 1 )
1908         {
1909             /* chunk by chunk */
1910             p_track->i_sample =
1911                 p_track->chunk[p_track->i_chunk].i_sample_first +
1912                 p_track->chunk[p_track->i_chunk].i_sample_count;
1913         }
1914         else
1915         {
1916             /* FIXME */
1917             p_track->i_sample += QT_V0_MAX_SAMPLES;
1918             if( p_track->i_sample > p_track->chunk[p_track->i_chunk].i_sample_first + p_track->chunk[p_track->i_chunk].i_sample_count )
1919             {
1920                 p_track->i_sample =
1921                     p_track->chunk[p_track->i_chunk].i_sample_first +
1922                     p_track->chunk[p_track->i_chunk].i_sample_count;
1923             }
1924         }
1925     }
1926     else
1927     {
1928         p_track->i_sample++;
1929     }
1930
1931     if( p_track->i_sample >= p_track->i_sample_count )
1932     {
1933         /* we have reach end of the track so free decoder stuff */
1934         msg_Warn( p_demux, "track[0x%x] will be disabled", p_track->i_track_ID );
1935         MP4_TrackUnselect( p_demux, p_track );
1936         return VLC_EGENERIC;
1937     }
1938
1939     /* Have we changed chunk ? */
1940     if( p_track->i_sample >=
1941             p_track->chunk[p_track->i_chunk].i_sample_first +
1942             p_track->chunk[p_track->i_chunk].i_sample_count )
1943     {
1944         if( TrackGotoChunkSample( p_demux, p_track, p_track->i_chunk + 1,
1945                                   p_track->i_sample ) )
1946         {
1947             msg_Warn( p_demux, "track[0x%x] will be disabled "
1948                       "(cannot restart decoder)", p_track->i_track_ID );
1949             MP4_TrackUnselect( p_demux, p_track );
1950             return VLC_EGENERIC;
1951         }
1952     }
1953
1954     /* Have we changed elst */
1955     if( p_track->p_elst && p_track->p_elst->data.p_elst->i_entry_count > 0 )
1956     {
1957         demux_sys_t *p_sys = p_demux->p_sys;
1958         MP4_Box_data_elst_t *elst = p_track->p_elst->data.p_elst;
1959         int64_t i_mvt = MP4_TrackGetPTS( p_demux, p_track ) *
1960                         p_sys->i_timescale / (int64_t)1000000;
1961
1962         if( p_track->i_elst < elst->i_entry_count &&
1963             i_mvt >= p_track->i_elst_time +
1964                      elst->i_segment_duration[p_track->i_elst] )
1965         {
1966             MP4_TrackSetELST( p_demux, p_track,
1967                               MP4_TrackGetPTS( p_demux, p_track ) );
1968         }
1969     }
1970
1971     return VLC_SUCCESS;
1972 }
1973
1974 static void MP4_TrackSetELST( demux_t *p_demux, mp4_track_t *tk,
1975                               int64_t i_time )
1976 {
1977     demux_sys_t *p_sys = p_demux->p_sys;
1978     int         i_elst_last = tk->i_elst;
1979
1980     /* handle elst (find the correct one) */
1981     tk->i_elst      = 0;
1982     tk->i_elst_time = 0;
1983     if( tk->p_elst && tk->p_elst->data.p_elst->i_entry_count > 0 )
1984     {
1985         MP4_Box_data_elst_t *elst = tk->p_elst->data.p_elst;
1986         int64_t i_mvt= i_time * p_sys->i_timescale / (int64_t)1000000;
1987
1988         for( tk->i_elst = 0; tk->i_elst < elst->i_entry_count; tk->i_elst++ )
1989         {
1990             mtime_t i_dur = elst->i_segment_duration[tk->i_elst];
1991
1992             if( tk->i_elst_time <= i_mvt && i_mvt < tk->i_elst_time + i_dur )
1993             {
1994                 break;
1995             }
1996             tk->i_elst_time += i_dur;
1997         }
1998
1999         if( tk->i_elst >= elst->i_entry_count )
2000         {
2001             /* msg_Dbg( p_demux, "invalid number of entry in elst" ); */
2002             tk->i_elst = elst->i_entry_count - 1;
2003             tk->i_elst_time -= elst->i_segment_duration[tk->i_elst];
2004         }
2005
2006         if( elst->i_media_time[tk->i_elst] < 0 )
2007         {
2008             /* track offset */
2009             tk->i_elst_time += elst->i_segment_duration[tk->i_elst];
2010         }
2011     }
2012     if( i_elst_last != tk->i_elst )
2013     {
2014         msg_Warn( p_demux, "elst old=%d new=%d", i_elst_last, tk->i_elst );
2015     }
2016 }