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

   1 /*****************************************************************************
   2  * mp4.c: mp4/mov muxer
   3  *****************************************************************************
   4  * Copyright (C) 2001, 2002, 2003 VideoLAN
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Gildas Bazin <gbazin at videolan dot org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28 #include <stdlib.h>
  29 #include <string.h>
  30
  31 #include <vlc/vlc.h>
  32 #include <vlc/input.h>
  33 #include <vlc/sout.h>
  34
  35 #ifdef HAVE_TIME_H
  36 #include <time.h>
  37 #endif
  38
  39 #include "iso_lang.h"
  40 #include "vlc_meta.h"
  41
  42 /*****************************************************************************
  43  * Module descriptor
  44  *****************************************************************************/
  45 #define FASTSTART_TEXT N_("Create \"Fast start\" files")
  46 #define FASTSTART_LONGTEXT N_( \
  47     "When this option is turned on, \"Fast start\" files will be created. " \
  48     "(\"Fast start\" files are optimized for download, allowing the user " \
  49     "to start previewing the file while it is downloading).")
  50 static int  Open   ( vlc_object_t * );
  51 static void Close  ( vlc_object_t * );
  52
  53 vlc_module_begin();
  54     set_description( _("MP4/MOV muxer") );
  55
  56     add_bool( "mp4-faststart", 1, NULL, FASTSTART_TEXT, FASTSTART_LONGTEXT,
  57               VLC_TRUE );
  58
  59     set_capability( "sout mux", 5 );
  60     add_shortcut( "mp4" );
  61     add_shortcut( "mov" );
  62     set_callbacks( Open, Close );
  63 vlc_module_end();
  64
  65 /*****************************************************************************
  66  * Exported prototypes
  67  *****************************************************************************/
  68 static int Capability(sout_mux_t *, int, void *, void * );
  69 static int AddStream( sout_mux_t *, sout_input_t * );
  70 static int DelStream( sout_mux_t *, sout_input_t * );
  71 static int Mux      ( sout_mux_t * );
  72
  73 /*****************************************************************************
  74  * Local prototypes
  75  *****************************************************************************/
  76 typedef struct
  77 {
  78     uint64_t i_pos;
  79     int      i_size;
  80
  81     mtime_t  i_pts_dts;
  82     mtime_t  i_length;
  83     unsigned int i_flags;
  84
  85 } mp4_entry_t;
  86
  87 typedef struct
  88 {
  89     es_format_t   fmt;
  90     int           i_track_id;
  91
  92     /* index */
  93     unsigned int i_entry_count;
  94     unsigned int i_entry_max;
  95     mp4_entry_t  *entry;
  96     int64_t      i_length_neg;
  97
  98     /* stats */
  99     int64_t      i_dts_start;
 100     int64_t      i_duration;
 101
 102     /* for later stco fix-up (fast start files) */
 103     uint64_t i_stco_pos;
 104     vlc_bool_t b_stco64;
 105
 106 } mp4_stream_t;
 107
 108 struct sout_mux_sys_t
 109 {
 110     vlc_bool_t b_mov;
 111     vlc_bool_t b_64_ext;
 112     vlc_bool_t b_fast_start;
 113
 114     uint64_t i_mdat_pos;
 115     uint64_t i_pos;
 116
 117     int64_t  i_dts_start;
 118
 119     int          i_nb_streams;
 120     mp4_stream_t **pp_streams;
 121 };
 122
 123 typedef struct bo_t
 124 {
 125     vlc_bool_t b_grow;
 126
 127     int        i_buffer_size;
 128     int        i_buffer;
 129     uint8_t    *p_buffer;
 130
 131 } bo_t;
 132
 133 static void bo_init     ( bo_t *, int , uint8_t *, vlc_bool_t  );
 134 static void bo_add_8    ( bo_t *, uint8_t );
 135 static void bo_add_16be ( bo_t *, uint16_t );
 136 static void bo_add_24be ( bo_t *, uint32_t );
 137 static void bo_add_32be ( bo_t *, uint32_t );
 138 static void bo_add_64be ( bo_t *, uint64_t );
 139 static void bo_add_fourcc(bo_t *, char * );
 140 static void bo_add_bo   ( bo_t *, bo_t * );
 141 static void bo_add_mem  ( bo_t *, int , uint8_t * );
 142 static void bo_add_descr( bo_t *, uint8_t , uint32_t );
 143
 144 static void bo_fix_32be ( bo_t *, int , uint32_t );
 145
 146 static bo_t *box_new     ( char *fcc );
 147 static bo_t *box_full_new( char *fcc, uint8_t v, uint32_t f );
 148 static void  box_fix     ( bo_t *box );
 149 static void  box_free    ( bo_t *box );
 150 static void  box_gather  ( bo_t *box, bo_t *box2 );
 151
 152 static void box_send( sout_mux_t *p_mux,  bo_t *box );
 153
 154 static block_t *bo_to_sout( sout_instance_t *p_sout,  bo_t *box );
 155
 156 static bo_t *GetMoovBox( sout_mux_t *p_mux );
 157
 158 /*****************************************************************************
 159  * Open:
 160  *****************************************************************************/
 161 static int Open( vlc_object_t *p_this )
 162 {
 163     sout_mux_t      *p_mux = (sout_mux_t*)p_this;
 164     sout_mux_sys_t  *p_sys;
 165     bo_t            *box;
 166
 167     p_sys = malloc( sizeof( sout_mux_sys_t ) );
 168     p_sys->i_pos        = 0;
 169     p_sys->i_nb_streams = 0;
 170     p_sys->pp_streams   = NULL;
 171     p_sys->i_mdat_pos   = 0;
 172     p_sys->b_mov        = p_mux->psz_mux && !strcmp( p_mux->psz_mux, "mov" );
 173     p_sys->i_dts_start  = 0;
 174
 175     msg_Dbg( p_mux, "Open" );
 176
 177     p_mux->pf_capacity  = Capability;
 178     p_mux->pf_addstream = AddStream;
 179     p_mux->pf_delstream = DelStream;
 180     p_mux->pf_mux       = Mux;
 181     p_mux->p_sys        = p_sys;
 182
 183     if( !p_sys->b_mov )
 184     {
 185         /* Now add ftyp header */
 186         box = box_new( "ftyp" );
 187         bo_add_fourcc( box, "isom" );
 188         bo_add_32be  ( box, 0 );
 189         bo_add_fourcc( box, "mp41" );
 190         box_fix( box );
 191
 192         p_sys->i_pos += box->i_buffer;
 193         p_sys->i_mdat_pos = p_sys->i_pos;
 194
 195         box_send( p_mux, box );
 196     }
 197
 198     /* FIXME FIXME
 199      * Quicktime actually doesn't like the 64 bits extensions !!! */
 200     p_sys->b_64_ext = VLC_FALSE;
 201
 202     /* Now add mdat header */
 203     box = box_new( "mdat" );
 204     bo_add_64be  ( box, 0 ); // enough to store an extended size
 205
 206     p_sys->i_pos += box->i_buffer;
 207
 208     box_send( p_mux, box );
 209
 210     return VLC_SUCCESS;
 211 }
 212
 213 /*****************************************************************************
 214  * Close:
 215  *****************************************************************************/
 216 static void Close( vlc_object_t * p_this )
 217 {
 218     sout_mux_t      *p_mux = (sout_mux_t*)p_this;
 219     sout_mux_sys_t  *p_sys = p_mux->p_sys;
 220     block_t   *p_hdr;
 221     bo_t            bo, *moov;
 222     vlc_value_t     val;
 223
 224     int             i_trak;
 225     uint64_t        i_moov_pos;
 226
 227     msg_Dbg( p_mux, "Close" );
 228
 229     /* Update mdat size */
 230     bo_init( &bo, 0, NULL, VLC_TRUE );
 231     if( p_sys->i_pos - p_sys->i_mdat_pos >= (((uint64_t)1)<<32) )
 232     {
 233         /* Extended size */
 234         bo_add_32be  ( &bo, 1 );
 235         bo_add_fourcc( &bo, "mdat" );
 236         bo_add_64be  ( &bo, p_sys->i_pos - p_sys->i_mdat_pos );
 237     }
 238     else
 239     {
 240         bo_add_32be  ( &bo, 8 );
 241         bo_add_fourcc( &bo, "wide" );
 242         bo_add_32be  ( &bo, p_sys->i_pos - p_sys->i_mdat_pos - 8 );
 243         bo_add_fourcc( &bo, "mdat" );
 244     }
 245     p_hdr = bo_to_sout( p_mux->p_sout, &bo );
 246     free( bo.p_buffer );
 247
 248     sout_AccessOutSeek( p_mux->p_access, p_sys->i_mdat_pos );
 249     sout_AccessOutWrite( p_mux->p_access, p_hdr );
 250
 251     /* Create MOOV header */
 252     i_moov_pos = p_sys->i_pos;
 253     moov = GetMoovBox( p_mux );
 254
 255     /* Check we need to create "fast start" files */
 256     var_Create( p_this, "mp4-faststart", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
 257     var_Get( p_this, "mp4-faststart", &val );
 258     p_sys->b_fast_start = val.b_bool;
 259     while( p_sys->b_fast_start )
 260     {
 261         /* Move data to the end of the file so we can fit the moov header
 262          * at the start */
 263         block_t *p_buf;
 264         int64_t i_chunk, i_size = p_sys->i_pos - p_sys->i_mdat_pos;
 265         int i_moov_size = moov->i_buffer;
 266
 267         while( i_size > 0 )
 268         {
 269             i_chunk = __MIN( 32768, i_size );
 270             p_buf = block_New( p_mux, i_chunk );
 271             sout_AccessOutSeek( p_mux->p_access,
 272                                 p_sys->i_mdat_pos + i_size - i_chunk );
 273             if( sout_AccessOutRead( p_mux->p_access, p_buf ) < i_chunk )
 274             {
 275                 msg_Warn( p_this, "read() not supported by acces output, "
 276                           "won't create a fast start file" );
 277                 p_sys->b_fast_start = VLC_FALSE;
 278                 break;
 279             }
 280             sout_AccessOutSeek( p_mux->p_access, p_sys->i_mdat_pos + i_size +
 281                                 i_moov_size - i_chunk );
 282             sout_AccessOutWrite( p_mux->p_access, p_buf );
 283             i_size -= i_chunk;
 284         }
 285
 286         if( !p_sys->b_fast_start ) break;
 287
 288         /* Fix-up samples to chunks table in MOOV header */
 289         for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
 290         {
 291             mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
 292             unsigned int i;
 293             int i_chunk;
 294
 295             moov->i_buffer = p_stream->i_stco_pos;
 296             for( i_chunk = 0, i = 0; i < p_stream->i_entry_count; i_chunk++ )
 297             {
 298                 if( p_stream->b_stco64 )
 299                     bo_add_64be( moov, p_stream->entry[i].i_pos + i_moov_size);
 300                 else
 301                     bo_add_32be( moov, p_stream->entry[i].i_pos + i_moov_size);
 302
 303                 while( i < p_stream->i_entry_count )
 304                 {
 305                     if( i + 1 < p_stream->i_entry_count &&
 306                         p_stream->entry[i].i_pos + p_stream->entry[i].i_size
 307                         != p_stream->entry[i + 1].i_pos )
 308                     {
 309                         i++;
 310                         break;
 311                     }
 312
 313                     i++;
 314                 }
 315             }
 316         }
 317
 318         moov->i_buffer = i_moov_size;
 319         i_moov_pos = p_sys->i_mdat_pos;
 320         p_sys->b_fast_start = VLC_FALSE;
 321     }
 322
 323     /* Write MOOV header */
 324     sout_AccessOutSeek( p_mux->p_access, i_moov_pos );
 325     box_send( p_mux, moov );
 326
 327     /* Clean-up */
 328     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
 329     {
 330         mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
 331
 332         es_format_Clean( &p_stream->fmt );
 333         free( p_stream->entry );
 334         free( p_stream );
 335     }
 336     if( p_sys->i_nb_streams ) free( p_sys->pp_streams );
 337     free( p_sys );
 338 }
 339
 340 /*****************************************************************************
 341  * Capability:
 342  *****************************************************************************/
 343 static int Capability( sout_mux_t *p_mux, int i_query, void *p_args,
 344                        void *p_answer )
 345 {
 346    switch( i_query )
 347    {
 348         case SOUT_MUX_CAP_GET_ADD_STREAM_ANY_TIME:
 349             *(vlc_bool_t*)p_answer = VLC_TRUE;
 350             return SOUT_MUX_CAP_ERR_OK;
 351
 352         case SOUT_MUX_CAP_GET_ADD_STREAM_WAIT:
 353             *(vlc_bool_t*)p_answer = VLC_TRUE;
 354             return( SOUT_MUX_CAP_ERR_OK );
 355
 356         default:
 357             return SOUT_MUX_CAP_ERR_UNIMPLEMENTED;
 358    }
 359 }
 360
 361 /*****************************************************************************
 362  * AddStream:
 363  *****************************************************************************/
 364 static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input )
 365 {
 366     sout_mux_sys_t  *p_sys = p_mux->p_sys;
 367     mp4_stream_t    *p_stream;
 368
 369     switch( p_input->p_fmt->i_codec )
 370     {
 371         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
 372         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
 373         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
 374         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
 375         case VLC_FOURCC( 'M', 'J', 'P', 'G' ):
 376         case VLC_FOURCC( 'm', 'j', 'p', 'b' ):
 377         case VLC_FOURCC( 'S', 'V', 'Q', '1' ):
 378         case VLC_FOURCC( 'S', 'V', 'Q', '3' ):
 379         case VLC_FOURCC( 'h', '2', '6', '4' ):
 380             break;
 381         default:
 382             msg_Err( p_mux, "unsupported codec %4.4s in mp4",
 383                      (char*)&p_input->p_fmt->i_codec );
 384             return VLC_EGENERIC;
 385     }
 386
 387     p_stream                = malloc( sizeof( mp4_stream_t ) );
 388     es_format_Copy( &p_stream->fmt, p_input->p_fmt );
 389     p_stream->i_track_id    = p_sys->i_nb_streams + 1;
 390     p_stream->i_length_neg  = 0;
 391     p_stream->i_entry_count = 0;
 392     p_stream->i_entry_max   = 1000;
 393     p_stream->entry         =
 394         calloc( p_stream->i_entry_max, sizeof( mp4_entry_t ) );
 395     p_stream->i_dts_start   = 0;
 396     p_stream->i_duration    = 0;
 397
 398     p_input->p_sys          = p_stream;
 399
 400     msg_Dbg( p_mux, "adding input" );
 401
 402     TAB_APPEND( p_sys->i_nb_streams, p_sys->pp_streams, p_stream );
 403     return VLC_SUCCESS;
 404 }
 405
 406 /*****************************************************************************
 407  * DelStream:
 408  *****************************************************************************/
 409 static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input )
 410 {
 411     msg_Dbg( p_mux, "removing input" );
 412     return VLC_SUCCESS;
 413 }
 414
 415 static int MuxGetStream( sout_mux_t *p_mux, int *pi_stream, mtime_t *pi_dts )
 416 {
 417     mtime_t i_dts;
 418     int     i_stream;
 419     int     i;
 420
 421     for( i = 0, i_dts = 0, i_stream = -1; i < p_mux->i_nb_inputs; i++ )
 422     {
 423         block_fifo_t   *p_fifo = p_mux->pp_inputs[i]->p_fifo;
 424         block_t *p_buf;
 425
 426         if( p_fifo->i_depth <= 1 )
 427         {
 428             return -1; // wait that all fifo have at least 2 packets
 429         }
 430
 431         p_buf = block_FifoShow( p_fifo );
 432         if( i_stream < 0 || p_buf->i_dts < i_dts )
 433         {
 434             i_dts = p_buf->i_dts;
 435             i_stream = i;
 436         }
 437     }
 438     if( pi_stream )
 439     {
 440         *pi_stream = i_stream;
 441     }
 442     if( pi_dts )
 443     {
 444         *pi_dts = i_dts;
 445     }
 446     return i_stream;
 447 }
 448
 449 /*****************************************************************************
 450  * Mux:
 451  *****************************************************************************/
 452 static int Mux( sout_mux_t *p_mux )
 453 {
 454     sout_mux_sys_t *p_sys = p_mux->p_sys;
 455
 456     for( ;; )
 457     {
 458         sout_input_t    *p_input;
 459         int             i_stream;
 460         mp4_stream_t    *p_stream;
 461         block_t   *p_data;
 462         mtime_t         i_dts;
 463
 464         if( MuxGetStream( p_mux, &i_stream, &i_dts) < 0 )
 465         {
 466             return( VLC_SUCCESS );
 467         }
 468
 469         p_input  = p_mux->pp_inputs[i_stream];
 470         p_stream = (mp4_stream_t*)p_input->p_sys;
 471
 472         p_data  = block_FifoGet( p_input->p_fifo );
 473         if( p_input->p_fifo->i_depth > 0 )
 474         {
 475             block_t *p_next = block_FifoShow( p_input->p_fifo );
 476             int64_t       i_diff  = p_next->i_dts - p_data->i_dts;
 477
 478             if( i_diff < I64C(1000000 ) )   /* protection */
 479             {
 480                 p_data->i_length = i_diff;
 481             }
 482         }
 483         if( p_data->i_length <= 0 )
 484         {
 485             msg_Warn( p_mux, "i_length <= 0" );
 486             p_stream->i_length_neg += p_data->i_length - 1;
 487             p_data->i_length = 1;
 488         }
 489         else if( p_stream->i_length_neg < 0 )
 490         {
 491             int64_t i_recover = __MIN( p_data->i_length / 4, - p_stream->i_length_neg );
 492
 493             p_data->i_length -= i_recover;
 494             p_stream->i_length_neg += i_recover;
 495         }
 496
 497         /* Save starting time */
 498         if( p_stream->i_entry_count == 0 )
 499         {
 500             p_stream->i_dts_start = p_data->i_dts;
 501
 502             /* Update global dts_start */
 503             if( p_sys->i_dts_start <= 0 ||
 504                 p_stream->i_dts_start < p_sys->i_dts_start )
 505             {
 506                 p_sys->i_dts_start = p_stream->i_dts_start;
 507             }
 508         }
 509
 510         /* add index entry */
 511         p_stream->entry[p_stream->i_entry_count].i_pos    = p_sys->i_pos;
 512         p_stream->entry[p_stream->i_entry_count].i_size   = p_data->i_buffer;
 513         p_stream->entry[p_stream->i_entry_count].i_pts_dts=
 514             __MAX( p_data->i_pts - p_data->i_dts, 0 );
 515         p_stream->entry[p_stream->i_entry_count].i_length = p_data->i_length;
 516         p_stream->entry[p_stream->i_entry_count].i_flags  = p_data->i_flags;
 517
 518         p_stream->i_entry_count++;
 519         if( p_stream->i_entry_count >= p_stream->i_entry_max )
 520         {
 521             p_stream->i_entry_max += 1000;
 522             p_stream->entry =
 523                 realloc( p_stream->entry,
 524                          p_stream->i_entry_max * sizeof( mp4_entry_t ) );
 525         }
 526
 527         /* update */
 528         p_stream->i_duration += p_data->i_length;
 529         p_sys->i_pos += p_data->i_buffer;
 530
 531         /* write data */
 532         sout_AccessOutWrite( p_mux->p_access, p_data );
 533     }
 534
 535     return( VLC_SUCCESS );
 536 }
 537
 538 /*****************************************************************************
 539  *
 540  *****************************************************************************/
 541 static int GetDescrLength( int i_size )
 542 {
 543     if( i_size < 0x00000080 )
 544         return 2 + i_size;
 545     else if( i_size < 0x00004000 )
 546         return 3 + i_size;
 547     else if( i_size < 0x00200000 )
 548         return 4 + i_size;
 549     else
 550         return 5 + i_size;
 551 }
 552
 553 static bo_t *GetESDS( mp4_stream_t *p_stream )
 554 {
 555     bo_t *esds;
 556     int  i_stream_type;
 557     int  i_object_type_indication;
 558     int  i_decoder_specific_info_size;
 559
 560     if( p_stream->fmt.i_extra > 0 )
 561     {
 562         i_decoder_specific_info_size =
 563             GetDescrLength( p_stream->fmt.i_extra );
 564     }
 565     else
 566     {
 567         i_decoder_specific_info_size = 0;
 568     }
 569
 570     esds = box_full_new( "esds", 0, 0 );
 571
 572     /* ES_Descr */
 573     bo_add_descr( esds, 0x03, 3 +
 574                   GetDescrLength( 13 + i_decoder_specific_info_size ) +
 575                   GetDescrLength( 1 ) );
 576     bo_add_16be( esds, p_stream->i_track_id );
 577     bo_add_8   ( esds, 0x1f );      // flags=0|streamPriority=0x1f
 578
 579     /* DecoderConfigDescr */
 580     bo_add_descr( esds, 0x04, 13 + i_decoder_specific_info_size );
 581
 582     switch( p_stream->fmt.i_codec )
 583     {
 584         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
 585             i_object_type_indication = 0x20;
 586             break;
 587         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
 588             /* FIXME MPEG-I=0x6b, MPEG-II = 0x60 -> 0x65 */
 589             i_object_type_indication = 0x60;
 590             break;
 591         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
 592             /* FIXME for mpeg2-aac == 0x66->0x68 */
 593             i_object_type_indication = 0x40;
 594             break;
 595         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
 596             i_object_type_indication =
 597                 p_stream->fmt.audio.i_rate < 32000 ? 0x69 : 0x6b;
 598             break;
 599         default:
 600             i_object_type_indication = 0x00;
 601             break;
 602     }
 603     i_stream_type = p_stream->fmt.i_cat == VIDEO_ES ? 0x04 : 0x05;
 604
 605     bo_add_8   ( esds, i_object_type_indication );
 606     bo_add_8   ( esds, ( i_stream_type << 2 ) | 1 );
 607     bo_add_24be( esds, 1024 * 1024 );       // bufferSizeDB
 608     bo_add_32be( esds, 0x7fffffff );        // maxBitrate
 609     bo_add_32be( esds, 0 );                 // avgBitrate
 610
 611     if( p_stream->fmt.i_extra > 0 )
 612     {
 613         int i;
 614
 615         /* DecoderSpecificInfo */
 616         bo_add_descr( esds, 0x05, p_stream->fmt.i_extra );
 617
 618         for( i = 0; i < p_stream->fmt.i_extra; i++ )
 619         {
 620             bo_add_8( esds, ((uint8_t*)p_stream->fmt.p_extra)[i] );
 621         }
 622     }
 623
 624     /* SL_Descr mandatory */
 625     bo_add_descr( esds, 0x06, 1 );
 626     bo_add_8    ( esds, 0x02 );  // sl_predefined
 627
 628     box_fix( esds );
 629
 630     return esds;
 631 }
 632
 633 static bo_t *GetWaveTag( mp4_stream_t *p_stream )
 634 {
 635     bo_t *wave;
 636     bo_t *box;
 637
 638     wave = box_new( "wave" );
 639
 640     box = box_new( "frma" );
 641     bo_add_fourcc( box, "mp4a" );
 642     box_fix( box );
 643     box_gather( wave, box );
 644
 645     box = box_new( "mp4a" );
 646     bo_add_32be( box, 0 );
 647     box_fix( box );
 648     box_gather( wave, box );
 649
 650     box = GetESDS( p_stream );
 651     box_fix( box );
 652     box_gather( wave, box );
 653
 654     box = box_new( "srcq" );
 655     bo_add_32be( box, 0x40 );
 656     box_fix( box );
 657     box_gather( wave, box );
 658
 659     /* wazza ? */
 660     bo_add_32be( wave, 8 ); /* new empty box */
 661     bo_add_32be( wave, 0 ); /* box label */
 662
 663     box_fix( wave );
 664
 665     return wave;
 666 }
 667
 668 /* TODO: No idea about these values */
 669 static bo_t *GetSVQ3Tag( mp4_stream_t *p_stream )
 670 {
 671     bo_t *smi = box_new( "SMI " );
 672
 673     if( p_stream->fmt.i_extra > 0x4e )
 674     {
 675         uint8_t *p_end = &((uint8_t*)p_stream->fmt.p_extra)[p_stream->fmt.i_extra];
 676         uint8_t *p     = &((uint8_t*)p_stream->fmt.p_extra)[0x46];
 677
 678         while( p + 8 < p_end )
 679         {
 680             int i_size = GetDWBE( p );
 681             if( i_size <= 1 )
 682             {
 683                 /* FIXME handle 1 as long size */
 684                 break;
 685             }
 686             if( !strncmp( &p[4], "SMI ", 4 ) )
 687             {
 688                 bo_add_mem( smi, p_end - p - 8, &p[8] );
 689                 return smi;
 690             }
 691             p += i_size;
 692         }
 693     }
 694
 695     /* Create a dummy one in fallback */
 696     bo_add_fourcc( smi, "SEQH" );
 697     bo_add_32be( smi, 0x5 );
 698     bo_add_32be( smi, 0xe2c0211d );
 699     bo_add_8( smi, 0xc0 );
 700     box_fix( smi );
 701
 702     return smi;
 703 }
 704
 705 static bo_t *GetUdtaTag( sout_mux_t *p_mux )
 706 {
 707     sout_mux_sys_t *p_sys = p_mux->p_sys;
 708     bo_t *udta = box_new( "udta" );
 709     vlc_meta_t *p_meta = p_mux->p_sout->p_meta;
 710     int i_track;
 711
 712     /* Requirements */
 713     for( i_track = 0; i_track < p_sys->i_nb_streams; i_track++ )
 714     {
 715         mp4_stream_t *p_stream = p_sys->pp_streams[i_track];
 716
 717         if( p_stream->fmt.i_codec == VLC_FOURCC('m','p','4','v') ||
 718             p_stream->fmt.i_codec == VLC_FOURCC('m','p','4','a') )
 719         {
 720             bo_t *box = box_new( "\251req" );
 721             /* String length */
 722             bo_add_16be( box, sizeof("QuickTime 6.0 or greater") - 1);
 723             bo_add_16be( box, 0 );
 724             bo_add_mem( box, sizeof("QuickTime 6.0 or greater") - 1,
 725                         "QuickTime 6.0 or greater" );
 726             box_fix( box );
 727             box_gather( udta, box );
 728             break;
 729         }
 730     }
 731
 732     /* Encoder */
 733     {
 734         bo_t *box = box_new( "\251enc" );
 735         /* String length */
 736         bo_add_16be( box, sizeof(PACKAGE_STRING " stream output") - 1);
 737         bo_add_16be( box, 0 );
 738         bo_add_mem( box, sizeof(PACKAGE_STRING " stream output") - 1,
 739                     PACKAGE_STRING " stream output" );
 740         box_fix( box );
 741         box_gather( udta, box );
 742     }
 743
 744     /* Misc atoms */
 745     if( p_meta )
 746     {
 747         int i;
 748         for( i = 0; i < p_meta->i_meta; i++ )
 749         {
 750             bo_t *box = NULL;
 751
 752             if( !strcmp( p_meta->name[i], VLC_META_TITLE ) )
 753                 box = box_new( "\251nam" );
 754             else if( !strcmp( p_meta->name[i], VLC_META_AUTHOR ) )
 755                 box = box_new( "\251aut" );
 756             else if( !strcmp( p_meta->name[i], VLC_META_ARTIST ) )
 757                 box = box_new( "\251ART" );
 758             else if( !strcmp( p_meta->name[i], VLC_META_GENRE ) )
 759                 box = box_new( "\251gen" );
 760             else if( !strcmp( p_meta->name[i], VLC_META_COPYRIGHT ) )
 761                 box = box_new( "\251cpy" );
 762             else if( !strcmp( p_meta->name[i], VLC_META_DESCRIPTION ) )
 763                 box = box_new( "\251des" );
 764             else if( !strcmp( p_meta->name[i], VLC_META_DATE ) )
 765                 box = box_new( "\251day" );
 766             else if( !strcmp( p_meta->name[i], VLC_META_URL ) )
 767                 box = box_new( "\251url" );
 768
 769             if( box )
 770             {
 771                 bo_add_16be( box, strlen( p_meta->value[i] ) );
 772                 bo_add_16be( box, 0 );
 773                 bo_add_mem( box, strlen( p_meta->value[i] ),
 774                             p_meta->value[i] );
 775                 box_fix( box );
 776                 box_gather( udta, box );
 777             }
 778         }
 779     }
 780
 781     box_fix( udta );
 782     return udta;
 783 }
 784
 785 static bo_t *GetSounBox( sout_mux_t *p_mux, mp4_stream_t *p_stream )
 786 {
 787     sout_mux_sys_t *p_sys = p_mux->p_sys;
 788     vlc_bool_t b_descr = VLC_FALSE;
 789     bo_t *soun;
 790     char fcc[4] = "    ";
 791     int  i;
 792
 793     switch( p_stream->fmt.i_codec )
 794     {
 795     case VLC_FOURCC('m','p','4','a'):
 796         memcpy( fcc, "mp4a", 4 );
 797         b_descr = VLC_TRUE;
 798         break;
 799
 800     case VLC_FOURCC('m','p','g','a'):
 801         if( p_sys->b_mov )
 802             memcpy( fcc, ".mp3", 4 );
 803         else
 804         {
 805             memcpy( fcc, "mp4a", 4 );
 806             b_descr = VLC_TRUE;
 807         }
 808         break;
 809
 810     default:
 811         memcpy( fcc, (char*)&p_stream->fmt.i_codec, 4 );
 812         break;
 813     }
 814
 815     soun = box_new( fcc );
 816     for( i = 0; i < 6; i++ )
 817     {
 818         bo_add_8( soun, 0 );        // reserved;
 819     }
 820     bo_add_16be( soun, 1 );         // data-reference-index
 821
 822     /* SoundDescription */
 823     if( p_sys->b_mov &&
 824         p_stream->fmt.i_codec == VLC_FOURCC('m','p','4','a') )
 825     {
 826         bo_add_16be( soun, 1 );     // version 1;
 827     }
 828     else
 829     {
 830         bo_add_16be( soun, 0 );     // version 0;
 831     }
 832     bo_add_16be( soun, 0 );         // revision level (0)
 833     bo_add_32be( soun, 0 );         // vendor
 834     // channel-count
 835     bo_add_16be( soun, p_stream->fmt.audio.i_channels );
 836     // sample size
 837     bo_add_16be( soun, p_stream->fmt.audio.i_bitspersample ?
 838                  p_stream->fmt.audio.i_bitspersample : 16 );
 839     bo_add_16be( soun, -2 );        // compression id
 840     bo_add_16be( soun, 0 );         // packet size (0)
 841     bo_add_16be( soun, p_stream->fmt.audio.i_rate ); // sampleratehi
 842     bo_add_16be( soun, 0 );                             // sampleratelo
 843
 844     /* Extended data for SoundDescription V1 */
 845     if( p_sys->b_mov &&
 846         p_stream->fmt.i_codec == VLC_FOURCC('m','p','4','a') )
 847     {
 848         /* samples per packet */
 849         bo_add_32be( soun, p_stream->fmt.audio.i_frame_length );
 850         bo_add_32be( soun, 1536 ); /* bytes per packet */
 851         bo_add_32be( soun, 2 );    /* bytes per frame */
 852         /* bytes per sample */
 853         bo_add_32be( soun, 2 /*p_stream->fmt.audio.i_bitspersample/8 */);
 854     }
 855
 856     /* Add an ES Descriptor */
 857     if( b_descr )
 858     {
 859         bo_t *box;
 860
 861         if( p_sys->b_mov &&
 862             p_stream->fmt.i_codec == VLC_FOURCC('m','p','4','a') )
 863         {
 864             box = GetWaveTag( p_stream );
 865         }
 866         else
 867         {
 868             box = GetESDS( p_stream );
 869         }
 870         box_fix( box );
 871         box_gather( soun, box );
 872     }
 873
 874     box_fix( soun );
 875
 876     return soun;
 877 }
 878
 879 static bo_t *GetVideBox( sout_mux_t *p_mux, mp4_stream_t *p_stream )
 880 {
 881
 882     bo_t *vide;
 883     char fcc[4] = "    ";
 884     int  i;
 885
 886     switch( p_stream->fmt.i_codec )
 887     {
 888     case VLC_FOURCC('m','p','4','v'):
 889     case VLC_FOURCC('m','p','g','v'):
 890         memcpy( fcc, "mp4v", 4 );
 891         break;
 892
 893     case VLC_FOURCC('M','J','P','G'):
 894         memcpy( fcc, "mjpa", 4 );
 895         break;
 896
 897     case VLC_FOURCC('S','V','Q','1'):
 898         memcpy( fcc, "SVQ1", 4 );
 899         break;
 900
 901     case VLC_FOURCC('S','V','Q','3'):
 902         memcpy( fcc, "SVQ3", 4 );
 903         break;
 904
 905     default:
 906         memcpy( fcc, (char*)&p_stream->fmt.i_codec, 4 );
 907         break;
 908     }
 909
 910     vide = box_new( fcc );
 911     for( i = 0; i < 6; i++ )
 912     {
 913         bo_add_8( vide, 0 );        // reserved;
 914     }
 915     bo_add_16be( vide, 1 );         // data-reference-index
 916
 917     bo_add_16be( vide, 0 );         // predefined;
 918     bo_add_16be( vide, 0 );         // reserved;
 919     for( i = 0; i < 3; i++ )
 920     {
 921         bo_add_32be( vide, 0 );     // predefined;
 922     }
 923
 924     bo_add_16be( vide, p_stream->fmt.video.i_width );  // i_width
 925     bo_add_16be( vide, p_stream->fmt.video.i_height ); // i_height
 926
 927     bo_add_32be( vide, 0x00480000 );                // h 72dpi
 928     bo_add_32be( vide, 0x00480000 );                // v 72dpi
 929
 930     bo_add_32be( vide, 0 );         // data size, always 0
 931     bo_add_16be( vide, 1 );         // frames count per sample
 932
 933     // compressor name;
 934     for( i = 0; i < 32; i++ )
 935     {
 936         bo_add_8( vide, 0 );
 937     }
 938
 939     bo_add_16be( vide, 0x18 );      // depth
 940     bo_add_16be( vide, 0xffff );    // predefined
 941
 942     /* add an ES Descriptor */
 943     switch( p_stream->fmt.i_codec )
 944     {
 945     case VLC_FOURCC('m','p','4','v'):
 946     case VLC_FOURCC('m','p','g','v'):
 947         {
 948             bo_t *esds = GetESDS( p_stream );
 949
 950             box_fix( esds );
 951             box_gather( vide, esds );
 952         }
 953         break;
 954
 955     case VLC_FOURCC('S','V','Q','3'):
 956         {
 957             bo_t *esds = GetSVQ3Tag( p_stream );
 958
 959             box_fix( esds );
 960             box_gather( vide, esds );
 961         }
 962         break;
 963
 964     default:
 965         break;
 966     }
 967
 968     box_fix( vide );
 969
 970     return vide;
 971 }
 972
 973 static bo_t *GetStblBox( sout_mux_t *p_mux, mp4_stream_t *p_stream )
 974 {
 975     sout_mux_sys_t *p_sys = p_mux->p_sys;
 976     unsigned int i_chunk, i_stsc_last_val, i_stsc_entries, i, i_index;
 977     bo_t *stbl, *stsd, *stts, *stco, *stsc, *stsz, *stss;
 978     uint32_t i_timescale;
 979     int64_t i_dts, i_dts_q;
 980
 981     stbl = box_new( "stbl" );
 982
 983     /* sample description */
 984     stsd = box_full_new( "stsd", 0, 0 );
 985     bo_add_32be( stsd, 1 );
 986     if( p_stream->fmt.i_cat == AUDIO_ES )
 987     {
 988         bo_t *soun = GetSounBox( p_mux, p_stream );
 989         box_gather( stsd, soun );
 990     }
 991     else if( p_stream->fmt.i_cat == VIDEO_ES )
 992     {
 993         bo_t *vide = GetVideBox( p_mux, p_stream );
 994         box_gather( stsd, vide );
 995     }
 996     box_fix( stsd );
 997
 998     /* chunk offset table */
 999     if( p_sys->i_pos >= (((uint64_t)0x1) << 32) )
1000     {
1001         /* 64 bits version */
1002         p_stream->b_stco64 = VLC_TRUE;
1003         stco = box_full_new( "co64", 0, 0 );
1004     }
1005     else
1006     {
1007         /* 32 bits version */
1008         p_stream->b_stco64 = VLC_FALSE;
1009         stco = box_full_new( "stco", 0, 0 );
1010     }
1011     bo_add_32be( stco, 0 );     // entry-count (fixed latter)
1012
1013     /* sample to chunk table */
1014     stsc = box_full_new( "stsc", 0, 0 );
1015     bo_add_32be( stsc, 0 );     // entry-count (fixed latter)
1016
1017     for( i_chunk = 0, i_stsc_last_val = 0, i_stsc_entries = 0, i = 0;
1018          i < p_stream->i_entry_count; i_chunk++ )
1019     {
1020         int i_first = i;
1021
1022         if( p_stream->b_stco64 )
1023             bo_add_64be( stco, p_stream->entry[i].i_pos );
1024         else
1025             bo_add_32be( stco, p_stream->entry[i].i_pos );
1026
1027         while( i < p_stream->i_entry_count )
1028         {
1029             if( i + 1 < p_stream->i_entry_count &&
1030                 p_stream->entry[i].i_pos + p_stream->entry[i].i_size
1031                 != p_stream->entry[i + 1].i_pos )
1032             {
1033                 i++;
1034                 break;
1035             }
1036
1037             i++;
1038         }
1039
1040         /* Add entry to the stsc table */
1041         if( i_stsc_last_val != i - i_first )
1042         {
1043             bo_add_32be( stsc, 1 + i_chunk );   // first-chunk
1044             bo_add_32be( stsc, i - i_first ) ;  // samples-per-chunk
1045             bo_add_32be( stsc, 1 );             // sample-descr-index
1046             i_stsc_last_val = i - i_first;
1047             i_stsc_entries++;
1048         }
1049     }
1050
1051     /* Fix stco entry count */
1052     bo_fix_32be( stco, 12, i_chunk );
1053     msg_Dbg( p_mux, "created %d chunks (stco)", i_chunk );
1054     box_fix( stco );
1055
1056     /* Fix stsc entry count */
1057     bo_fix_32be( stsc, 12, i_stsc_entries  );
1058     box_fix( stsc );
1059
1060     /* add stts */
1061     stts = box_full_new( "stts", 0, 0 );
1062     bo_add_32be( stts, 0 );     // entry-count (fixed latter)
1063
1064     if( p_stream->fmt.i_cat == AUDIO_ES )
1065         i_timescale = p_stream->fmt.audio.i_rate;
1066     else
1067         i_timescale = 1001;
1068
1069     /* first, create quantified length */
1070     for( i = 0, i_dts = 0, i_dts_q = 0; i < p_stream->i_entry_count; i++ )
1071     {
1072         int64_t i_dts_deq = i_dts_q * I64C(1000000) / (int64_t)i_timescale;
1073         int64_t i_delta = p_stream->entry[i].i_length + i_dts - i_dts_deq;
1074
1075         i_dts += p_stream->entry[i].i_length;
1076
1077         p_stream->entry[i].i_length =
1078             i_delta * (int64_t)i_timescale / I64C(1000000);
1079
1080         i_dts_q += p_stream->entry[i].i_length;
1081     }
1082     /* then write encoded table */
1083     for( i = 0, i_index = 0; i < p_stream->i_entry_count; i_index++)
1084     {
1085         int     i_first = i;
1086         int64_t i_delta = p_stream->entry[i].i_length;
1087
1088         while( i < p_stream->i_entry_count )
1089         {
1090             i++;
1091             if( i >= p_stream->i_entry_count ||
1092                 p_stream->entry[i].i_length != i_delta )
1093             {
1094                 break;
1095             }
1096         }
1097
1098         bo_add_32be( stts, i - i_first ); // sample-count
1099         bo_add_32be( stts, i_delta );     // sample-delta
1100     }
1101     bo_fix_32be( stts, 12, i_index );
1102     box_fix( stts );
1103
1104     /* FIXME add ctts ?? FIXME */
1105
1106     stsz = box_full_new( "stsz", 0, 0 );
1107     bo_add_32be( stsz, 0 );                             // sample-size
1108     bo_add_32be( stsz, p_stream->i_entry_count );       // sample-count
1109     for( i = 0; i < p_stream->i_entry_count; i++ )
1110     {
1111         bo_add_32be( stsz, p_stream->entry[i].i_size ); // sample-size
1112     }
1113     box_fix( stsz );
1114
1115     /* create stss table */
1116     stss = NULL;
1117     for( i = 0, i_index = 0; i < p_stream->i_entry_count; i++ )
1118     {
1119         if( p_stream->entry[i].i_flags & BLOCK_FLAG_TYPE_I )
1120         {
1121             if( stss == NULL )
1122             {
1123                 stss = box_full_new( "stss", 0, 0 );
1124                 bo_add_32be( stss, 0 ); /* fixed later */
1125             }
1126             bo_add_32be( stss, 1 + i );
1127             i_index++;
1128         }
1129     }
1130     if( stss )
1131     {
1132         bo_fix_32be( stss, 12, i_index );
1133         box_fix( stss );
1134     }
1135
1136     /* Now gather all boxes into stbl */
1137     box_gather( stbl, stsd );
1138     box_gather( stbl, stts );
1139     if( stss )
1140     {
1141         box_gather( stbl, stss );
1142     }
1143     box_gather( stbl, stsc );
1144     box_gather( stbl, stsz );
1145     p_stream->i_stco_pos = stbl->i_buffer + 16;
1146     box_gather( stbl, stco );
1147
1148     /* finish stbl */
1149     box_fix( stbl );
1150
1151     return stbl;
1152 }
1153
1154 static int64_t get_timestamp();
1155
1156 static uint32_t mvhd_matrix[9] =
1157     { 0x10000, 0, 0, 0, 0x10000, 0, 0, 0, 0x40000000 };
1158
1159 static bo_t *GetMoovBox( sout_mux_t *p_mux )
1160 {
1161     sout_mux_sys_t *p_sys = p_mux->p_sys;
1162
1163     bo_t            *moov, *mvhd;
1164     int             i_trak, i;
1165
1166     uint32_t        i_movie_timescale = 90000;
1167     int64_t         i_movie_duration  = 0;
1168
1169     moov = box_new( "moov" );
1170
1171     /* Create general info */
1172     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1173     {
1174         mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
1175         i_movie_duration = __MAX( i_movie_duration, p_stream->i_duration );
1176     }
1177     msg_Dbg( p_mux, "movie duration %ds",
1178              (uint32_t)( i_movie_duration / (mtime_t)1000000 ) );
1179
1180     i_movie_duration = i_movie_duration * i_movie_timescale / 1000000;
1181
1182     /* *** add /moov/mvhd *** */
1183     if( !p_sys->b_64_ext )
1184     {
1185         mvhd = box_full_new( "mvhd", 0, 0 );
1186         bo_add_32be( mvhd, get_timestamp() );   // creation time
1187         bo_add_32be( mvhd, get_timestamp() );   // modification time
1188         bo_add_32be( mvhd, i_movie_timescale);  // timescale
1189         bo_add_32be( mvhd, i_movie_duration );  // duration
1190     }
1191     else
1192     {
1193         mvhd = box_full_new( "mvhd", 1, 0 );
1194         bo_add_64be( mvhd, get_timestamp() );   // creation time
1195         bo_add_64be( mvhd, get_timestamp() );   // modification time
1196         bo_add_32be( mvhd, i_movie_timescale);  // timescale
1197         bo_add_64be( mvhd, i_movie_duration );  // duration
1198     }
1199     bo_add_32be( mvhd, 0x10000 );           // rate
1200     bo_add_16be( mvhd, 0x100 );             // volume
1201     bo_add_16be( mvhd, 0 );                 // reserved
1202     for( i = 0; i < 2; i++ )
1203     {
1204         bo_add_32be( mvhd, 0 );             // reserved
1205     }
1206     for( i = 0; i < 9; i++ )
1207     {
1208         bo_add_32be( mvhd, mvhd_matrix[i] );// matrix
1209     }
1210     for( i = 0; i < 6; i++ )
1211     {
1212         bo_add_32be( mvhd, 0 );             // pre-defined
1213     }
1214
1215     /* Next available track id */
1216     bo_add_32be( mvhd, p_sys->i_nb_streams + 1 ); // next-track-id
1217
1218     box_fix( mvhd );
1219     box_gather( moov, mvhd );
1220
1221     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1222     {
1223         mp4_stream_t *p_stream;
1224         uint32_t     i_timescale;
1225
1226         bo_t *trak, *tkhd, *edts, *elst, *mdia, *mdhd, *hdlr;
1227         bo_t *minf, *dinf, *dref, *url, *stbl;
1228
1229         p_stream = p_sys->pp_streams[i_trak];
1230
1231         if( p_stream->fmt.i_cat != AUDIO_ES &&
1232             p_stream->fmt.i_cat != VIDEO_ES )
1233         {
1234             msg_Err( p_mux, "FIXME ignoring trak (noaudio&&novideo)" );
1235             continue;
1236         }
1237
1238         if( p_stream->fmt.i_cat == AUDIO_ES )
1239             i_timescale = p_stream->fmt.audio.i_rate;
1240         else
1241             i_timescale = 1001;
1242
1243         /* *** add /moov/trak *** */
1244         trak = box_new( "trak" );
1245
1246         /* *** add /moov/trak/tkhd *** */
1247         if( !p_sys->b_64_ext )
1248         {
1249             if( p_sys->b_mov )
1250                 tkhd = box_full_new( "tkhd", 0, 0x0f );
1251             else
1252                 tkhd = box_full_new( "tkhd", 0, 1 );
1253
1254             bo_add_32be( tkhd, get_timestamp() );       // creation time
1255             bo_add_32be( tkhd, get_timestamp() );       // modification time
1256             bo_add_32be( tkhd, p_stream->i_track_id );
1257             bo_add_32be( tkhd, 0 );                     // reserved 0
1258             bo_add_32be( tkhd, p_stream->i_duration *
1259                          (int64_t)i_movie_timescale /
1260                          (mtime_t)1000000 );            // duration
1261         }
1262         else
1263         {
1264             if( p_sys->b_mov )
1265                 tkhd = box_full_new( "tkhd", 1, 0x0f );
1266             else
1267                 tkhd = box_full_new( "tkhd", 1, 1 );
1268
1269             bo_add_64be( tkhd, get_timestamp() );       // creation time
1270             bo_add_64be( tkhd, get_timestamp() );       // modification time
1271             bo_add_32be( tkhd, p_stream->i_track_id );
1272             bo_add_32be( tkhd, 0 );                     // reserved 0
1273             bo_add_64be( tkhd, p_stream->i_duration *
1274                          (int64_t)i_movie_timescale /
1275                          (mtime_t)1000000 );            // duration
1276         }
1277
1278         for( i = 0; i < 2; i++ )
1279         {
1280             bo_add_32be( tkhd, 0 );                 // reserved
1281         }
1282         bo_add_16be( tkhd, 0 );                     // layer
1283         bo_add_16be( tkhd, 0 );                     // pre-defined
1284         // volume
1285         bo_add_16be( tkhd, p_stream->fmt.i_cat == AUDIO_ES ? 0x100 : 0 );
1286         bo_add_16be( tkhd, 0 );                     // reserved
1287         for( i = 0; i < 9; i++ )
1288         {
1289             bo_add_32be( tkhd, mvhd_matrix[i] );    // matrix
1290         }
1291         if( p_stream->fmt.i_cat == AUDIO_ES )
1292         {
1293             bo_add_32be( tkhd, 0 );                 // width (presentation)
1294             bo_add_32be( tkhd, 0 );                 // height(presentation)
1295         }
1296         else
1297         {
1298             // width (presentation)
1299             bo_add_32be( tkhd, p_stream->fmt.video.i_aspect *
1300                          p_stream->fmt.video.i_height /
1301                          VOUT_ASPECT_FACTOR << 16 );
1302             // height(presentation)
1303             bo_add_32be( tkhd, p_stream->fmt.video.i_height << 16 );
1304         }
1305         box_fix( tkhd );
1306         box_gather( trak, tkhd );
1307
1308         /* *** add /moov/trak/edts and elst */
1309         edts = box_new( "edts" );
1310         elst = box_full_new( "elst", p_sys->b_64_ext ? 1 : 0, 0 );
1311         if( p_stream->i_dts_start > p_sys->i_dts_start )
1312         {
1313             bo_add_32be( elst, 2 );
1314
1315             if( p_sys->b_64_ext )
1316             {
1317                 bo_add_64be( elst, (p_stream->i_dts_start-p_sys->i_dts_start) *
1318                              i_movie_timescale / I64C(1000000) );
1319                 bo_add_64be( elst, -1 );
1320             }
1321             else
1322             {
1323                 bo_add_32be( elst, (p_stream->i_dts_start-p_sys->i_dts_start) *
1324                              i_movie_timescale / I64C(1000000) );
1325                 bo_add_32be( elst, -1 );
1326             }
1327             bo_add_16be( elst, 1 );
1328             bo_add_16be( elst, 0 );
1329         }
1330         else
1331         {
1332             bo_add_32be( elst, 1 );
1333         }
1334         if( p_sys->b_64_ext )
1335         {
1336             bo_add_64be( elst, p_stream->i_duration *
1337                          i_movie_timescale / I64C(1000000) );
1338             bo_add_64be( elst, 0 );
1339         }
1340         else
1341         {
1342             bo_add_32be( elst, p_stream->i_duration *
1343                          i_movie_timescale / I64C(1000000) );
1344             bo_add_32be( elst, 0 );
1345         }
1346         bo_add_16be( elst, 1 );
1347         bo_add_16be( elst, 0 );
1348
1349         box_fix( elst );
1350         box_gather( edts, elst );
1351         box_fix( edts );
1352         box_gather( trak, edts );
1353
1354         /* *** add /moov/trak/mdia *** */
1355         mdia = box_new( "mdia" );
1356
1357         /* media header */
1358         if( !p_sys->b_64_ext )
1359         {
1360             mdhd = box_full_new( "mdhd", 0, 0 );
1361             bo_add_32be( mdhd, get_timestamp() );   // creation time
1362             bo_add_32be( mdhd, get_timestamp() );   // modification time
1363             bo_add_32be( mdhd, i_timescale);        // timescale
1364             bo_add_32be( mdhd, p_stream->i_duration * (int64_t)i_timescale /
1365                                (mtime_t)1000000 );  // duration
1366         }
1367         else
1368         {
1369             mdhd = box_full_new( "mdhd", 1, 0 );
1370             bo_add_64be( mdhd, get_timestamp() );   // creation time
1371             bo_add_64be( mdhd, get_timestamp() );   // modification time
1372             bo_add_32be( mdhd, i_timescale);        // timescale
1373             bo_add_64be( mdhd, p_stream->i_duration * (int64_t)i_timescale /
1374                                (mtime_t)1000000 );  // duration
1375         }
1376
1377         if( p_stream->fmt.psz_language )
1378         {
1379             char *psz = p_stream->fmt.psz_language;
1380             const iso639_lang_t *pl = NULL;
1381             uint16_t lang = 0x0;
1382
1383             if( strlen( psz ) == 2 )
1384             {
1385                 pl = GetLang_1( psz );
1386             }
1387             else if( strlen( psz ) == 3 )
1388             {
1389                 pl = GetLang_2B( psz );
1390                 if( !strcmp( pl->psz_iso639_1, "??" ) )
1391                 {
1392                     pl = GetLang_2T( psz );
1393                 }
1394             }
1395             if( pl && strcmp( pl->psz_iso639_1, "??" ) )
1396             {
1397                 lang = ( ( pl->psz_iso639_2T[0] - 0x60 ) << 10 ) |
1398                        ( ( pl->psz_iso639_2T[1] - 0x60 ) <<  5 ) |
1399                        ( ( pl->psz_iso639_2T[2] - 0x60 ) );
1400             }
1401             bo_add_16be( mdhd, lang );          // language
1402         }
1403         else
1404         {
1405             bo_add_16be( mdhd, 0    );          // language
1406         }
1407         bo_add_16be( mdhd, 0    );              // predefined
1408         box_fix( mdhd );
1409         box_gather( mdia, mdhd );
1410
1411         /* handler reference */
1412         hdlr = box_full_new( "hdlr", 0, 0 );
1413
1414         bo_add_fourcc( hdlr, "mhlr" );         // media handler
1415         if( p_stream->fmt.i_cat == AUDIO_ES )
1416         {
1417             bo_add_fourcc( hdlr, "soun" );
1418         }
1419         else
1420         {
1421             bo_add_fourcc( hdlr, "vide" );
1422         }
1423
1424         bo_add_32be( hdlr, 0 );         // reserved
1425         bo_add_32be( hdlr, 0 );         // reserved
1426         bo_add_32be( hdlr, 0 );         // reserved
1427
1428         bo_add_8( hdlr, 12 );
1429         if( p_stream->fmt.i_cat == AUDIO_ES )
1430             bo_add_mem( hdlr, 12, "SoundHandler" );
1431         else
1432             bo_add_mem( hdlr, 12, "VideoHandler" );
1433
1434         box_fix( hdlr );
1435         box_gather( mdia, hdlr );
1436
1437         /* minf*/
1438         minf = box_new( "minf" );
1439
1440         /* add smhd|vmhd */
1441         if( p_stream->fmt.i_cat == AUDIO_ES )
1442         {
1443             bo_t *smhd;
1444
1445             smhd = box_full_new( "smhd", 0, 0 );
1446             bo_add_16be( smhd, 0 );     // balance
1447             bo_add_16be( smhd, 0 );     // reserved
1448             box_fix( smhd );
1449
1450             box_gather( minf, smhd );
1451         }
1452         else if( p_stream->fmt.i_cat == VIDEO_ES )
1453         {
1454             bo_t *vmhd;
1455
1456             vmhd = box_full_new( "vmhd", 0, 1 );
1457             bo_add_16be( vmhd, 0 );     // graphicsmode
1458             for( i = 0; i < 3; i++ )
1459             {
1460                 bo_add_16be( vmhd, 0 ); // opcolor
1461             }
1462             box_fix( vmhd );
1463
1464             box_gather( minf, vmhd );
1465         }
1466
1467         /* dinf */
1468         dinf = box_new( "dinf" );
1469         dref = box_full_new( "dref", 0, 0 );
1470         bo_add_32be( dref, 1 );
1471         url = box_full_new( "url ", 0, 0x01 );
1472         box_fix( url );
1473         box_gather( dref, url );
1474         box_fix( dref );
1475         box_gather( dinf, dref );
1476
1477         /* append dinf to mdia */
1478         box_fix( dinf );
1479         box_gather( minf, dinf );
1480
1481         /* add stbl */
1482         stbl = GetStblBox( p_mux, p_stream );
1483
1484         /* append stbl to minf */
1485         p_stream->i_stco_pos += minf->i_buffer;
1486         box_gather( minf, stbl );
1487
1488         /* append minf to mdia */
1489         box_fix( minf );
1490         p_stream->i_stco_pos += mdia->i_buffer;
1491         box_gather( mdia, minf );
1492
1493         /* append mdia to trak */
1494         box_fix( mdia );
1495         p_stream->i_stco_pos += trak->i_buffer;
1496         box_gather( trak, mdia );
1497
1498         /* append trak to moov */
1499         box_fix( trak );
1500         p_stream->i_stco_pos += moov->i_buffer;
1501         box_gather( moov, trak );
1502     }
1503
1504     /* Add user data tags */
1505     box_gather( moov, GetUdtaTag( p_mux ) );
1506
1507     box_fix( moov );
1508     return moov;
1509 }
1510
1511 /****************************************************************************/
1512
1513 static void bo_init( bo_t *p_bo, int i_size, uint8_t *p_buffer,
1514                      vlc_bool_t b_grow )
1515 {
1516     if( !p_buffer )
1517     {
1518         p_bo->i_buffer_size = __MAX( i_size, 1024 );
1519         p_bo->p_buffer = malloc( p_bo->i_buffer_size );
1520     }
1521     else
1522     {
1523         p_bo->i_buffer_size = i_size;
1524         p_bo->p_buffer = p_buffer;
1525     }
1526
1527     p_bo->b_grow = b_grow;
1528     p_bo->i_buffer = 0;
1529 }
1530
1531 static void bo_add_8( bo_t *p_bo, uint8_t i )
1532 {
1533     if( p_bo->i_buffer < p_bo->i_buffer_size )
1534     {
1535         p_bo->p_buffer[p_bo->i_buffer] = i;
1536     }
1537     else if( p_bo->b_grow )
1538     {
1539         p_bo->i_buffer_size += 1024;
1540         p_bo->p_buffer = realloc( p_bo->p_buffer, p_bo->i_buffer_size );
1541
1542         p_bo->p_buffer[p_bo->i_buffer] = i;
1543     }
1544
1545     p_bo->i_buffer++;
1546 }
1547
1548 static void bo_add_16be( bo_t *p_bo, uint16_t i )
1549 {
1550     bo_add_8( p_bo, ( ( i >> 8) &0xff ) );
1551     bo_add_8( p_bo, i &0xff );
1552 }
1553
1554 static void bo_add_24be( bo_t *p_bo, uint32_t i )
1555 {
1556     bo_add_8( p_bo, ( ( i >> 16) &0xff ) );
1557     bo_add_8( p_bo, ( ( i >> 8) &0xff ) );
1558     bo_add_8( p_bo, (   i &0xff ) );
1559 }
1560 static void bo_add_32be( bo_t *p_bo, uint32_t i )
1561 {
1562     bo_add_16be( p_bo, ( ( i >> 16) &0xffff ) );
1563     bo_add_16be( p_bo, i &0xffff );
1564 }
1565
1566 static void bo_fix_32be ( bo_t *p_bo, int i_pos, uint32_t i)
1567 {
1568     p_bo->p_buffer[i_pos    ] = ( i >> 24 )&0xff;
1569     p_bo->p_buffer[i_pos + 1] = ( i >> 16 )&0xff;
1570     p_bo->p_buffer[i_pos + 2] = ( i >>  8 )&0xff;
1571     p_bo->p_buffer[i_pos + 3] = ( i       )&0xff;
1572 }
1573
1574 static void bo_add_64be( bo_t *p_bo, uint64_t i )
1575 {
1576     bo_add_32be( p_bo, ( ( i >> 32) &0xffffffff ) );
1577     bo_add_32be( p_bo, i &0xffffffff );
1578 }
1579
1580 static void bo_add_fourcc( bo_t *p_bo, char *fcc )
1581 {
1582     bo_add_8( p_bo, fcc[0] );
1583     bo_add_8( p_bo, fcc[1] );
1584     bo_add_8( p_bo, fcc[2] );
1585     bo_add_8( p_bo, fcc[3] );
1586 }
1587
1588 static void bo_add_mem( bo_t *p_bo, int i_size, uint8_t *p_mem )
1589 {
1590     int i;
1591
1592     for( i = 0; i < i_size; i++ )
1593     {
1594         bo_add_8( p_bo, p_mem[i] );
1595     }
1596 }
1597
1598 static void bo_add_descr( bo_t *p_bo, uint8_t tag, uint32_t i_size )
1599 {
1600     uint32_t i_length;
1601     uint8_t  vals[4];
1602
1603     i_length = i_size;
1604     vals[3] = (unsigned char)(i_length & 0x7f);
1605     i_length >>= 7;
1606     vals[2] = (unsigned char)((i_length & 0x7f) | 0x80);
1607     i_length >>= 7;
1608     vals[1] = (unsigned char)((i_length & 0x7f) | 0x80);
1609     i_length >>= 7;
1610     vals[0] = (unsigned char)((i_length & 0x7f) | 0x80);
1611
1612     bo_add_8( p_bo, tag );
1613
1614     if( i_size < 0x00000080 )
1615     {
1616         bo_add_8( p_bo, vals[3] );
1617     }
1618     else if( i_size < 0x00004000 )
1619     {
1620         bo_add_8( p_bo, vals[2] );
1621         bo_add_8( p_bo, vals[3] );
1622     }
1623     else if( i_size < 0x00200000 )
1624     {
1625         bo_add_8( p_bo, vals[1] );
1626         bo_add_8( p_bo, vals[2] );
1627         bo_add_8( p_bo, vals[3] );
1628     }
1629     else if( i_size < 0x10000000 )
1630     {
1631         bo_add_8( p_bo, vals[0] );
1632         bo_add_8( p_bo, vals[1] );
1633         bo_add_8( p_bo, vals[2] );
1634         bo_add_8( p_bo, vals[3] );
1635     }
1636 }
1637
1638 static void bo_add_bo( bo_t *p_bo, bo_t *p_bo2 )
1639 {
1640     int i;
1641
1642     for( i = 0; i < p_bo2->i_buffer; i++ )
1643     {
1644         bo_add_8( p_bo, p_bo2->p_buffer[i] );
1645     }
1646 }
1647
1648 static bo_t * box_new( char *fcc )
1649 {
1650     bo_t *box;
1651
1652     if( ( box = malloc( sizeof( bo_t ) ) ) )
1653     {
1654         bo_init( box, 0, NULL, VLC_TRUE );
1655
1656         bo_add_32be  ( box, 0 );
1657         bo_add_fourcc( box, fcc );
1658     }
1659
1660     return box;
1661 }
1662
1663 static bo_t * box_full_new( char *fcc, uint8_t v, uint32_t f )
1664 {
1665     bo_t *box;
1666
1667     if( ( box = malloc( sizeof( bo_t ) ) ) )
1668     {
1669         bo_init( box, 0, NULL, VLC_TRUE );
1670
1671         bo_add_32be  ( box, 0 );
1672         bo_add_fourcc( box, fcc );
1673         bo_add_8     ( box, v );
1674         bo_add_24be  ( box, f );
1675     }
1676
1677     return box;
1678 }
1679
1680 static void box_fix( bo_t *box )
1681 {
1682     bo_t box_tmp;
1683
1684     memcpy( &box_tmp, box, sizeof( bo_t ) );
1685
1686     box_tmp.i_buffer = 0;
1687     bo_add_32be( &box_tmp, box->i_buffer );
1688 }
1689
1690 static void box_free( bo_t *box )
1691 {
1692     if( box->p_buffer )
1693     {
1694         free( box->p_buffer );
1695     }
1696
1697     free( box );
1698 }
1699
1700 static void box_gather ( bo_t *box, bo_t *box2 )
1701 {
1702     bo_add_bo( box, box2 );
1703     box_free( box2 );
1704 }
1705
1706 static block_t * bo_to_sout( sout_instance_t *p_sout,  bo_t *box )
1707 {
1708     block_t *p_buf;
1709
1710     p_buf = block_New( p_sout, box->i_buffer );
1711     if( box->i_buffer > 0 )
1712     {
1713         memcpy( p_buf->p_buffer, box->p_buffer, box->i_buffer );
1714     }
1715
1716     return p_buf;
1717 }
1718
1719 static void box_send( sout_mux_t *p_mux,  bo_t *box )
1720 {
1721     block_t *p_buf;
1722
1723     p_buf = bo_to_sout( p_mux->p_sout, box );
1724     box_free( box );
1725
1726     sout_AccessOutWrite( p_mux->p_access, p_buf );
1727 }
1728
1729 static int64_t get_timestamp()
1730 {
1731     int64_t i_timestamp = 0;
1732
1733 #ifdef HAVE_TIME_H
1734     i_timestamp = time(NULL);
1735     i_timestamp += 2082844800; // MOV/MP4 start date is 1/1/1904
1736     // 208284480 is (((1970 - 1904) * 365) + 17) * 24 * 60 * 60
1737 #endif
1738
1739     return i_timestamp;
1740 }