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