1 /*****************************************************************************
3 *****************************************************************************
4 * Copyright (C) 2001, 2002 the VideoLAN team
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
27 /* TODO: add OpenDML write support */
36 #include <vlc_block.h>
37 #include <vlc_codecs.h>
39 /*****************************************************************************
41 *****************************************************************************/
42 static int Open ( vlc_object_t * );
43 static void Close ( vlc_object_t * );
46 set_description( _("AVI muxer") );
47 set_category( CAT_SOUT );
48 set_subcategory( SUBCAT_SOUT_MUX );
49 set_capability( "sout mux", 5 );
50 add_shortcut( "avi" );
51 set_callbacks( Open, Close );
55 /*****************************************************************************
57 *****************************************************************************/
58 static int Control( sout_mux_t *, int, va_list );
59 static int AddStream( sout_mux_t *, sout_input_t * );
60 static int DelStream( sout_mux_t *, sout_input_t * );
61 static int Mux ( sout_mux_t * );
63 typedef struct avi_stream_s
69 mtime_t i_duration; // in µs
71 int i_frames; // total frame count
72 int64_t i_totalsize; // total stream size
77 BITMAPINFOHEADER *p_bih;
82 typedef struct avi_idx1_entry_s
91 typedef struct avi_idx1_s
93 unsigned int i_entry_count;
94 unsigned int i_entry_max;
96 avi_idx1_entry_t *entry;
101 vlc_bool_t b_write_header;
107 avi_stream_t stream[100];
115 #define HDR_SIZE 10240
118 #define AVIF_HASINDEX 0x00000010 // Index at end of file?
119 #define AVIF_ISINTERLEAVED 0x00000100
120 #define AVIF_TRUSTCKTYPE 0x00000800 // Use CKType to find key frames?
122 /* Flags for index */
123 #define AVIIF_KEYFRAME 0x00000010L /* this frame is a key frame.*/
126 static block_t *avi_HeaderCreateRIFF( sout_mux_t * );
127 static block_t *avi_HeaderCreateidx1( sout_mux_t * );
129 static void SetFCC( uint8_t *p, char *fcc )
134 /*****************************************************************************
136 *****************************************************************************/
137 static int Open( vlc_object_t *p_this )
139 sout_mux_t *p_mux = (sout_mux_t*)p_this;
140 sout_mux_sys_t *p_sys = p_mux->p_sys;
142 msg_Dbg( p_mux, "AVI muxer opened" );
144 p_sys = malloc( sizeof( sout_mux_sys_t ) );
145 p_sys->i_streams = 0;
146 p_sys->i_stream_video = -1;
147 p_sys->i_movi_size = 0;
149 p_sys->idx1.i_entry_count = 0;
150 p_sys->idx1.i_entry_max = 10000;
151 p_sys->idx1.entry = calloc( p_sys->idx1.i_entry_max,
152 sizeof( avi_idx1_entry_t ) );
153 p_sys->b_write_header = VLC_TRUE;
156 p_mux->pf_control = Control;
157 p_mux->pf_addstream = AddStream;
158 p_mux->pf_delstream = DelStream;
160 p_mux->p_sys = p_sys;
165 /*****************************************************************************
167 *****************************************************************************/
168 static void Close( vlc_object_t * p_this )
170 sout_mux_t *p_mux = (sout_mux_t*)p_this;
171 sout_mux_sys_t *p_sys = p_mux->p_sys;
173 block_t *p_hdr, *p_idx1;
176 msg_Dbg( p_mux, "AVI muxer closed" );
178 /* first create idx1 chunk (write at the end of the stream */
179 p_idx1 = avi_HeaderCreateidx1( p_mux );
180 p_sys->i_idx1_size = p_idx1->i_buffer;
181 sout_AccessOutWrite( p_mux->p_access, p_idx1 );
183 /* calculate some value for headers creations */
184 for( i_stream = 0; i_stream < p_sys->i_streams; i_stream++ )
186 avi_stream_t *p_stream;
188 p_stream = &p_sys->stream[i_stream];
190 p_stream->f_fps = 25;
191 if( p_stream->i_duration > 0 )
193 p_stream->f_fps = (float)p_stream->i_frames /
194 ( (float)p_stream->i_duration /
197 p_stream->i_bitrate = 128 * 1024;
198 if( p_stream->i_duration > 0 )
200 p_stream->i_bitrate =
201 8 * (uint64_t)1000000 *
202 (uint64_t)p_stream->i_totalsize /
203 (uint64_t)p_stream->i_duration;
205 msg_Info( p_mux, "stream[%d] duration:"I64Fd" totalsize:"I64Fd
206 " frames:%d fps:%f kb/s:%d",
208 (int64_t)p_stream->i_duration / (int64_t)1000000,
209 p_stream->i_totalsize,
211 p_stream->f_fps, p_stream->i_bitrate/1024 );
214 p_hdr = avi_HeaderCreateRIFF( p_mux );
215 sout_AccessOutSeek( p_mux->p_access, 0 );
216 sout_AccessOutWrite( p_mux->p_access, p_hdr );
219 static int Control( sout_mux_t *p_mux, int i_query, va_list args )
227 case MUX_CAN_ADD_STREAM_WHILE_MUXING:
228 pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t * );
229 *pb_bool = VLC_FALSE;
232 case MUX_GET_ADD_STREAM_WAIT:
233 pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t * );
238 ppsz = (char**)va_arg( args, char ** );
239 *ppsz = strdup( "video/avi" );
247 static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input )
249 sout_mux_sys_t *p_sys = p_mux->p_sys;
250 avi_stream_t *p_stream;
252 if( p_sys->i_streams >= 100 )
254 msg_Err( p_mux, "too many streams" );
258 msg_Dbg( p_mux, "adding input" );
259 p_input->p_sys = malloc( sizeof( int ) );
261 *((int*)p_input->p_sys) = p_sys->i_streams;
262 p_stream = &p_sys->stream[p_sys->i_streams];
264 switch( p_input->p_fmt->i_cat )
267 p_stream->i_cat = AUDIO_ES;
268 p_stream->fcc[0] = '0' + p_sys->i_streams / 10;
269 p_stream->fcc[1] = '0' + p_sys->i_streams % 10;
270 p_stream->fcc[2] = 'w';
271 p_stream->fcc[3] = 'b';
273 p_stream->p_bih = NULL;
275 p_stream->p_wf = malloc( sizeof( WAVEFORMATEX ) +
276 p_input->p_fmt->i_extra );
277 #define p_wf p_stream->p_wf
278 p_wf->cbSize = p_input->p_fmt->i_extra;
279 if( p_wf->cbSize > 0 )
282 p_input->p_fmt->p_extra,
283 p_input->p_fmt->i_extra );
285 p_wf->nChannels = p_input->p_fmt->audio.i_channels;
286 p_wf->nSamplesPerSec = p_input->p_fmt->audio.i_rate;
287 p_wf->nBlockAlign = p_input->p_fmt->audio.i_blockalign;
288 p_wf->nAvgBytesPerSec= p_input->p_fmt->i_bitrate / 8;
289 p_wf->wBitsPerSample = 0;
291 switch( p_input->p_fmt->i_codec )
293 case VLC_FOURCC( 'a', '5', '2', ' ' ):
294 p_wf->wFormatTag = WAVE_FORMAT_A52;
296 case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
297 p_wf->wFormatTag = WAVE_FORMAT_MPEGLAYER3;
299 case VLC_FOURCC( 'w', 'm', 'a', '1' ):
300 p_wf->wFormatTag = WAVE_FORMAT_WMA1;
302 case VLC_FOURCC( 'w', 'm', 'a', ' ' ):
303 case VLC_FOURCC( 'w', 'm', 'a', '2' ):
304 p_wf->wFormatTag = WAVE_FORMAT_WMA2;
306 case VLC_FOURCC( 'w', 'm', 'a', 'p' ):
307 p_wf->wFormatTag = WAVE_FORMAT_WMAP;
309 case VLC_FOURCC( 'w', 'm', 'a', 'l' ):
310 p_wf->wFormatTag = WAVE_FORMAT_WMAL;
313 case VLC_FOURCC( 'u', '8', ' ', ' ' ):
314 p_wf->wFormatTag = WAVE_FORMAT_PCM;
315 p_wf->nBlockAlign= p_wf->nChannels;
316 p_wf->wBitsPerSample = 8;
318 case VLC_FOURCC( 's', '1', '6', 'l' ):
319 p_wf->wFormatTag = WAVE_FORMAT_PCM;
320 p_wf->nBlockAlign= 2 * p_wf->nChannels;
321 p_wf->wBitsPerSample = 16;
323 case VLC_FOURCC( 's', '2', '4', 'l' ):
324 p_wf->wFormatTag = WAVE_FORMAT_PCM;
325 p_wf->nBlockAlign= 3 * p_wf->nChannels;
326 p_wf->wBitsPerSample = 24;
328 case VLC_FOURCC( 's', '3', '2', 'l' ):
329 p_wf->wFormatTag = WAVE_FORMAT_PCM;
330 p_wf->nBlockAlign= 4 * p_wf->nChannels;
331 p_wf->wBitsPerSample = 32;
339 p_stream->i_cat = VIDEO_ES;
340 p_stream->fcc[0] = '0' + p_sys->i_streams / 10;
341 p_stream->fcc[1] = '0' + p_sys->i_streams % 10;
342 p_stream->fcc[2] = 'd';
343 p_stream->fcc[3] = 'c';
344 if( p_sys->i_stream_video < 0 )
346 p_sys->i_stream_video = p_sys->i_streams;
348 p_stream->p_wf = NULL;
349 p_stream->p_bih = malloc( sizeof( BITMAPINFOHEADER ) +
350 p_input->p_fmt->i_extra );
351 #define p_bih p_stream->p_bih
352 p_bih->biSize = sizeof( BITMAPINFOHEADER ) +
353 p_input->p_fmt->i_extra;
354 if( p_input->p_fmt->i_extra > 0 )
357 p_input->p_fmt->p_extra,
358 p_input->p_fmt->i_extra );
360 p_bih->biWidth = p_input->p_fmt->video.i_width;
361 p_bih->biHeight= p_input->p_fmt->video.i_height;
363 p_bih->biBitCount = 24;
364 p_bih->biSizeImage = 0;
365 p_bih->biXPelsPerMeter = 0;
366 p_bih->biYPelsPerMeter = 0;
367 p_bih->biClrUsed = 0;
368 p_bih->biClrImportant = 0;
369 switch( p_input->p_fmt->i_codec )
371 case VLC_FOURCC( 'm', 'p', '4', 'v' ):
372 p_bih->biCompression = VLC_FOURCC( 'X', 'V', 'I', 'D' );
375 p_bih->biCompression = p_input->p_fmt->i_codec;
381 return( VLC_EGENERIC );
383 p_stream->i_totalsize = 0;
384 p_stream->i_frames = 0;
385 p_stream->i_duration = 0;
388 p_stream->f_fps = 25;
389 p_stream->i_bitrate = 128 * 1024;
392 return( VLC_SUCCESS );
395 static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input )
398 msg_Dbg( p_mux, "removing input" );
400 free( p_input->p_sys );
404 static int Mux ( sout_mux_t *p_mux )
406 sout_mux_sys_t *p_sys = p_mux->p_sys;
407 avi_stream_t *p_stream;
410 if( p_sys->b_write_header )
414 msg_Dbg( p_mux, "writing header" );
416 p_hdr = avi_HeaderCreateRIFF( p_mux );
417 sout_AccessOutWrite( p_mux->p_access, p_hdr );
419 p_sys->b_write_header = VLC_FALSE;
422 for( i = 0; i < p_mux->i_nb_inputs; i++ )
425 block_fifo_t *p_fifo;
427 i_stream = *((int*)p_mux->pp_inputs[i]->p_sys );
428 p_stream = &p_sys->stream[i_stream];
430 p_fifo = p_mux->pp_inputs[i]->p_fifo;
431 i_count = block_FifoCount( p_fifo );
434 avi_idx1_entry_t *p_idx;
437 p_data = block_FifoGet( p_fifo );
438 if( block_FifoCount( p_fifo ) > 0 )
440 block_t *p_next = block_FifoShow( p_fifo );
441 p_data->i_length = p_next->i_dts - p_data->i_dts;
444 p_stream->i_frames++;
445 if( p_data->i_length < 0 )
447 msg_Warn( p_mux, "argg length < 0 l" );
448 block_Release( p_data );
452 p_stream->i_duration += p_data->i_length;
453 p_stream->i_totalsize += p_data->i_buffer;
455 /* add idx1 entry for this frame */
456 p_idx = &p_sys->idx1.entry[p_sys->idx1.i_entry_count];
457 memcpy( p_idx->fcc, p_stream->fcc, 4 );
458 p_idx->i_flags = AVIIF_KEYFRAME;
459 p_idx->i_pos = p_sys->i_movi_size + 4;
460 p_idx->i_length= p_data->i_buffer;
461 p_sys->idx1.i_entry_count++;
462 if( p_sys->idx1.i_entry_count >= p_sys->idx1.i_entry_max )
464 p_sys->idx1.i_entry_max += 10000;
465 p_sys->idx1.entry = realloc( p_sys->idx1.entry,
466 p_sys->idx1.i_entry_max * sizeof( avi_idx1_entry_t ) );
469 p_data = block_Realloc( p_data, 8, p_data->i_buffer );
472 SetFCC( p_data->p_buffer, p_stream->fcc );
473 SetDWLE( p_data->p_buffer + 4, p_data->i_buffer - 8 );
475 if( p_data->i_buffer & 0x01 )
477 p_data = block_Realloc( p_data, 0, p_data->i_buffer + 1 );
478 p_data->p_buffer[ p_data->i_buffer - 1 ] = '\0';
481 p_sys->i_movi_size += p_data->i_buffer;
482 sout_AccessOutWrite( p_mux->p_access, p_data );
492 /****************************************************************************/
493 /****************************************************************************/
494 /****************************************************************************/
495 /****************************************************************************/
497 typedef struct buffer_out_s
505 static void bo_Init( buffer_out_t *p_bo, int i_size, uint8_t *p_buffer )
507 p_bo->i_buffer_size = i_size;
509 p_bo->p_buffer = p_buffer;
511 static void bo_AddByte( buffer_out_t *p_bo, uint8_t i )
513 if( p_bo->i_buffer < p_bo->i_buffer_size )
515 p_bo->p_buffer[p_bo->i_buffer] = i;
519 static void bo_AddWordLE( buffer_out_t *p_bo, uint16_t i )
521 bo_AddByte( p_bo, i &0xff );
522 bo_AddByte( p_bo, ( ( i >> 8) &0xff ) );
524 static void bo_AddWordBE( buffer_out_t *p_bo, uint16_t i )
526 bo_AddByte( p_bo, ( ( i >> 8) &0xff ) );
527 bo_AddByte( p_bo, i &0xff );
529 static void bo_AddDWordLE( buffer_out_t *p_bo, uint32_t i )
531 bo_AddWordLE( p_bo, i &0xffff );
532 bo_AddWordLE( p_bo, ( ( i >> 16) &0xffff ) );
534 static void bo_AddDWordBE( buffer_out_t *p_bo, uint32_t i )
536 bo_AddWordBE( p_bo, ( ( i >> 16) &0xffff ) );
537 bo_AddWordBE( p_bo, i &0xffff );
540 static void bo_AddLWordLE( buffer_out_t *p_bo, uint64_t i )
542 bo_AddDWordLE( p_bo, i &0xffffffff );
543 bo_AddDWordLE( p_bo, ( ( i >> 32) &0xffffffff ) );
545 static void bo_AddLWordBE( buffer_out_t *p_bo, uint64_t i )
547 bo_AddDWordBE( p_bo, ( ( i >> 32) &0xffffffff ) );
548 bo_AddDWordBE( p_bo, i &0xffffffff );
552 static void bo_AddFCC( buffer_out_t *p_bo, const char *fcc )
554 bo_AddByte( p_bo, fcc[0] );
555 bo_AddByte( p_bo, fcc[1] );
556 bo_AddByte( p_bo, fcc[2] );
557 bo_AddByte( p_bo, fcc[3] );
560 static void bo_AddMem( buffer_out_t *p_bo, int i_size, uint8_t *p_mem )
564 for( i = 0; i < i_size; i++ )
566 bo_AddByte( p_bo, p_mem[i] );
570 /****************************************************************************
571 ****************************************************************************
573 ** avi header generation
575 ****************************************************************************
576 ****************************************************************************/
577 #define AVI_BOX_ENTER( fcc ) \
578 buffer_out_t _bo_sav_; \
579 bo_AddFCC( p_bo, fcc ); \
581 bo_AddDWordLE( p_bo, 0 )
583 #define AVI_BOX_ENTER_LIST( fcc ) \
584 AVI_BOX_ENTER( "LIST" ); \
585 bo_AddFCC( p_bo, fcc )
587 #define AVI_BOX_EXIT( i_err ) \
588 if( p_bo->i_buffer&0x01 ) bo_AddByte( p_bo, 0 ); \
589 bo_AddDWordLE( &_bo_sav_, p_bo->i_buffer - _bo_sav_.i_buffer - 4 ); \
592 static int avi_HeaderAdd_avih( sout_mux_t *p_mux,
595 sout_mux_sys_t *p_sys = p_mux->p_sys;
596 avi_stream_t *p_video = NULL;
598 uint32_t i_microsecperframe;
599 int i_maxbytespersec;
601 AVI_BOX_ENTER( "avih" );
603 if( p_sys->i_stream_video >= 0 )
605 p_video = &p_sys->stream[p_sys->i_stream_video];
606 if( p_video->i_frames <= 0 )
615 (uint32_t)( (float)1000000 /
616 (float)p_sys->stream[p_sys->i_stream_video].f_fps );
617 i_totalframes = p_sys->stream[p_sys->i_stream_video].i_frames;
621 msg_Warn( p_mux, "avi file without video track isn't a good idea..." );
622 i_microsecperframe = 0;
626 for( i_stream = 0,i_maxbytespersec = 0; i_stream < p_sys->i_streams; i_stream++ )
628 if( p_sys->stream[i_stream].i_duration > 0 )
631 p_sys->stream[i_stream].i_totalsize /
632 p_sys->stream[i_stream].i_duration;
636 bo_AddDWordLE( p_bo, i_microsecperframe );
637 bo_AddDWordLE( p_bo, i_maxbytespersec );
638 bo_AddDWordLE( p_bo, 0 ); /* padding */
639 bo_AddDWordLE( p_bo, AVIF_TRUSTCKTYPE |
641 AVIF_ISINTERLEAVED ); /* flags */
642 bo_AddDWordLE( p_bo, i_totalframes );
643 bo_AddDWordLE( p_bo, 0 ); /* initial frame */
644 bo_AddDWordLE( p_bo, p_sys->i_streams ); /* streams count */
645 bo_AddDWordLE( p_bo, 1024 * 1024 ); /* suggested buffer size */
648 bo_AddDWordLE( p_bo, p_video->p_bih->biWidth );
649 bo_AddDWordLE( p_bo, p_video->p_bih->biHeight );
653 bo_AddDWordLE( p_bo, 0 );
654 bo_AddDWordLE( p_bo, 0 );
656 bo_AddDWordLE( p_bo, 0 ); /* ???? */
657 bo_AddDWordLE( p_bo, 0 ); /* ???? */
658 bo_AddDWordLE( p_bo, 0 ); /* ???? */
659 bo_AddDWordLE( p_bo, 0 ); /* ???? */
663 static int avi_HeaderAdd_strh( buffer_out_t *p_bo, avi_stream_t *p_stream )
665 AVI_BOX_ENTER( "strh" );
667 switch( p_stream->i_cat )
671 bo_AddFCC( p_bo, "vids" );
672 bo_AddDWordBE( p_bo, p_stream->p_bih->biCompression );
673 bo_AddDWordLE( p_bo, 0 ); /* flags */
674 bo_AddWordLE( p_bo, 0 ); /* priority */
675 bo_AddWordLE( p_bo, 0 ); /* langage */
676 bo_AddDWordLE( p_bo, 0 ); /* initial frame */
677 bo_AddDWordLE( p_bo, 1000 );/* scale */
678 bo_AddDWordLE( p_bo, (uint32_t)( 1000 * p_stream->f_fps ));
679 bo_AddDWordLE( p_bo, 0 ); /* start */
680 bo_AddDWordLE( p_bo, p_stream->i_frames );
681 bo_AddDWordLE( p_bo, 1024 * 1024 );
682 bo_AddDWordLE( p_bo, -1 ); /* quality */
683 bo_AddDWordLE( p_bo, 0 ); /* samplesize */
684 bo_AddWordLE( p_bo, 0 ); /* ??? */
685 bo_AddWordLE( p_bo, 0 ); /* ??? */
686 bo_AddWordLE( p_bo, p_stream->p_bih->biWidth );
687 bo_AddWordLE( p_bo, p_stream->p_bih->biHeight );
692 int i_rate, i_scale, i_samplesize;
694 i_samplesize = p_stream->p_wf->nBlockAlign;
695 if( i_samplesize > 1 )
697 i_scale = i_samplesize;
698 i_rate = /*i_scale **/ p_stream->i_bitrate / 8;
704 i_rate = 1000 * p_stream->i_bitrate / 8;
706 bo_AddFCC( p_bo, "auds" );
707 bo_AddDWordLE( p_bo, 0 ); /* tag */
708 bo_AddDWordLE( p_bo, 0 ); /* flags */
709 bo_AddWordLE( p_bo, 0 ); /* priority */
710 bo_AddWordLE( p_bo, 0 ); /* langage */
711 bo_AddDWordLE( p_bo, 0 ); /* initial frame */
712 bo_AddDWordLE( p_bo, i_scale );/* scale */
713 bo_AddDWordLE( p_bo, i_rate );
714 bo_AddDWordLE( p_bo, 0 ); /* start */
715 bo_AddDWordLE( p_bo, p_stream->i_frames );
716 bo_AddDWordLE( p_bo, 10 * 1024 );
717 bo_AddDWordLE( p_bo, -1 ); /* quality */
718 bo_AddDWordLE( p_bo, i_samplesize );
719 bo_AddWordLE( p_bo, 0 ); /* ??? */
720 bo_AddWordLE( p_bo, 0 ); /* ??? */
721 bo_AddWordLE( p_bo, 0 );
722 bo_AddWordLE( p_bo, 0 );
730 static int avi_HeaderAdd_strf( buffer_out_t *p_bo, avi_stream_t *p_stream )
732 AVI_BOX_ENTER( "strf" );
734 switch( p_stream->i_cat )
737 bo_AddWordLE( p_bo, p_stream->p_wf->wFormatTag );
738 bo_AddWordLE( p_bo, p_stream->p_wf->nChannels );
739 bo_AddDWordLE( p_bo, p_stream->p_wf->nSamplesPerSec );
740 bo_AddDWordLE( p_bo, p_stream->p_wf->nAvgBytesPerSec );
741 bo_AddWordLE( p_bo, p_stream->p_wf->nBlockAlign );
742 bo_AddWordLE( p_bo, p_stream->p_wf->wBitsPerSample );
743 bo_AddWordLE( p_bo, p_stream->p_wf->cbSize );
744 bo_AddMem( p_bo, p_stream->p_wf->cbSize, (uint8_t*)&p_stream->p_wf[1] );
747 bo_AddDWordLE( p_bo, p_stream->p_bih->biSize );
748 bo_AddDWordLE( p_bo, p_stream->p_bih->biWidth );
749 bo_AddDWordLE( p_bo, p_stream->p_bih->biHeight );
750 bo_AddWordLE( p_bo, p_stream->p_bih->biPlanes );
751 bo_AddWordLE( p_bo, p_stream->p_bih->biBitCount );
752 if( VLC_FOURCC( 0, 0, 0, 1 ) == 0x00000001 )
754 bo_AddDWordBE( p_bo, p_stream->p_bih->biCompression );
758 bo_AddDWordLE( p_bo, p_stream->p_bih->biCompression );
760 bo_AddDWordLE( p_bo, p_stream->p_bih->biSizeImage );
761 bo_AddDWordLE( p_bo, p_stream->p_bih->biXPelsPerMeter );
762 bo_AddDWordLE( p_bo, p_stream->p_bih->biYPelsPerMeter );
763 bo_AddDWordLE( p_bo, p_stream->p_bih->biClrUsed );
764 bo_AddDWordLE( p_bo, p_stream->p_bih->biClrImportant );
766 p_stream->p_bih->biSize - sizeof( BITMAPINFOHEADER ),
767 (uint8_t*)&p_stream->p_bih[1] );
774 static int avi_HeaderAdd_strl( buffer_out_t *p_bo, avi_stream_t *p_stream )
776 AVI_BOX_ENTER_LIST( "strl" );
778 avi_HeaderAdd_strh( p_bo, p_stream );
779 avi_HeaderAdd_strf( p_bo, p_stream );
784 static block_t *avi_HeaderCreateRIFF( sout_mux_t *p_mux )
786 sout_mux_sys_t *p_sys = p_mux->p_sys;
789 int i_maxbytespersec;
793 p_hdr = block_New( p_mux, HDR_SIZE );
794 memset( p_hdr->p_buffer, 0, HDR_SIZE );
796 bo_Init( &bo, HDR_SIZE, p_hdr->p_buffer );
798 bo_AddFCC( &bo, "RIFF" );
799 bo_AddDWordLE( &bo, p_sys->i_movi_size + HDR_SIZE - 8 + p_sys->i_idx1_size );
800 bo_AddFCC( &bo, "AVI " );
802 bo_AddFCC( &bo, "LIST" );
803 bo_AddDWordLE( &bo, HDR_SIZE - 8);
804 bo_AddFCC( &bo, "hdrl" );
806 avi_HeaderAdd_avih( p_mux, &bo );
807 for( i_stream = 0,i_maxbytespersec = 0; i_stream < p_sys->i_streams; i_stream++ )
809 avi_HeaderAdd_strl( &bo, &p_sys->stream[i_stream] );
812 i_junk = HDR_SIZE - bo.i_buffer - 8 - 12;
813 bo_AddFCC( &bo, "JUNK" );
814 bo_AddDWordLE( &bo, i_junk );
816 bo.i_buffer += i_junk;
817 bo_AddFCC( &bo, "LIST" );
818 bo_AddDWordLE( &bo, p_sys->i_movi_size + 4 );
819 bo_AddFCC( &bo, "movi" );
824 static block_t * avi_HeaderCreateidx1( sout_mux_t *p_mux )
826 sout_mux_sys_t *p_sys = p_mux->p_sys;
828 uint32_t i_idx1_size;
832 i_idx1_size = 16 * p_sys->idx1.i_entry_count;
834 p_idx1 = block_New( p_mux, i_idx1_size + 8 );
835 memset( p_idx1->p_buffer, 0, i_idx1_size );
837 bo_Init( &bo, i_idx1_size, p_idx1->p_buffer );
838 bo_AddFCC( &bo, "idx1" );
839 bo_AddDWordLE( &bo, i_idx1_size );
841 for( i = 0; i < p_sys->idx1.i_entry_count; i++ )
843 bo_AddFCC( &bo, p_sys->idx1.entry[i].fcc );
844 bo_AddDWordLE( &bo, p_sys->idx1.entry[i].i_flags );
845 bo_AddDWordLE( &bo, p_sys->idx1.entry[i].i_pos );
846 bo_AddDWordLE( &bo, p_sys->idx1.entry[i].i_length );