1 /*****************************************************************************
3 *****************************************************************************
4 * Copyright (C) 2001-2009 VLC authors and VideoLAN
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * This program is free software; you can redistribute it and/or modify it
10 * under the terms of the GNU Lesser General Public License as published by
11 * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public License
20 * along with this program; if not, write to the Free Software Foundation,
21 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
27 /* TODO: add OpenDML write support */
34 #include <vlc_common.h>
35 #include <vlc_plugin.h>
37 #include <vlc_block.h>
38 #include <vlc_codecs.h>
40 /*****************************************************************************
42 *****************************************************************************/
43 static int Open ( vlc_object_t
* );
44 static void Close ( vlc_object_t
* );
46 #define SOUT_CFG_PREFIX "sout-avi-"
48 #define CFG_ARTIST_TEXT N_("Artist")
49 #define CFG_DATE_TEXT N_("Date")
50 #define CFG_GENRE_TEXT N_("Genre")
51 #define CFG_COPYRIGHT_TEXT N_("Copyright")
52 #define CFG_COMMENT_TEXT N_("Comment")
53 #define CFG_NAME_TEXT N_("Name")
54 #define CFG_SUBJECT_TEXT N_("Subject")
55 #define CFG_ENCODER_TEXT N_("Encoder")
56 #define CFG_KEYWORDS_TEXT N_("Keywords")
59 set_description( N_("AVI muxer") )
60 set_category( CAT_SOUT
)
61 set_subcategory( SUBCAT_SOUT_MUX
)
62 set_capability( "sout mux", 5 )
65 add_string( SOUT_CFG_PREFIX
"artist", NULL
, CFG_ARTIST_TEXT
, NULL
, true )
66 add_string( SOUT_CFG_PREFIX
"date", NULL
, CFG_DATE_TEXT
, NULL
, true )
67 add_string( SOUT_CFG_PREFIX
"genre", NULL
, CFG_GENRE_TEXT
, NULL
, true )
68 add_string( SOUT_CFG_PREFIX
"copyright", NULL
, CFG_COPYRIGHT_TEXT
, NULL
, true )
69 add_string( SOUT_CFG_PREFIX
"comment", NULL
, CFG_COMMENT_TEXT
, NULL
, true )
70 add_string( SOUT_CFG_PREFIX
"name", NULL
, CFG_NAME_TEXT
, NULL
, true )
71 add_string( SOUT_CFG_PREFIX
"subject", NULL
, CFG_SUBJECT_TEXT
, NULL
, true )
72 add_string( SOUT_CFG_PREFIX
"encoder",
73 "VLC Media Player - " VERSION_MESSAGE
,
74 CFG_ENCODER_TEXT
, NULL
, true )
75 add_string( SOUT_CFG_PREFIX
"keywords", NULL
, CFG_KEYWORDS_TEXT
, NULL
, true )
77 set_callbacks( Open
, Close
)
81 /*****************************************************************************
83 *****************************************************************************/
84 static int Control( sout_mux_t
*, int, va_list );
85 static int AddStream( sout_mux_t
*, sout_input_t
* );
86 static int DelStream( sout_mux_t
*, sout_input_t
* );
87 static int Mux ( sout_mux_t
* );
89 typedef struct avi_stream_s
95 mtime_t i_duration
; // in µs
97 int i_frames
; // total frame count
98 int64_t i_totalsize
; // total stream size
103 VLC_BITMAPINFOHEADER
*p_bih
;
108 typedef struct avi_idx1_entry_s
117 typedef struct avi_idx1_s
119 unsigned int i_entry_count
;
120 unsigned int i_entry_max
;
122 avi_idx1_entry_t
*entry
;
125 struct sout_mux_sys_t
133 avi_stream_t stream
[100];
140 #define HDR_BASE_SIZE 512 /* single video&audio ~ 400 bytes header */
143 #define AVIF_HASINDEX 0x00000010 // Index at end of file?
144 #define AVIF_ISINTERLEAVED 0x00000100
145 #define AVIF_TRUSTCKTYPE 0x00000800 // Use CKType to find key frames?
147 /* Flags for index */
148 #define AVIIF_KEYFRAME 0x00000010L /* this frame is a key frame.*/
151 static block_t
*avi_HeaderCreateRIFF( sout_mux_t
* );
152 static block_t
*avi_HeaderCreateidx1( sout_mux_t
* );
154 static void SetFCC( uint8_t *p
, char *fcc
)
159 /*****************************************************************************
161 *****************************************************************************/
162 static int Open( vlc_object_t
*p_this
)
164 sout_mux_t
*p_mux
= (sout_mux_t
*)p_this
;
165 sout_mux_sys_t
*p_sys
;
167 msg_Dbg( p_mux
, "AVI muxer opened" );
169 p_sys
= malloc( sizeof( sout_mux_sys_t
) );
172 p_sys
->i_streams
= 0;
173 p_sys
->i_stream_video
= -1;
174 p_sys
->i_movi_size
= 0;
176 p_sys
->idx1
.i_entry_count
= 0;
177 p_sys
->idx1
.i_entry_max
= 10000;
178 p_sys
->idx1
.entry
= calloc( p_sys
->idx1
.i_entry_max
,
179 sizeof( avi_idx1_entry_t
) );
180 if( !p_sys
->idx1
.entry
)
185 p_sys
->b_write_header
= true;
188 p_mux
->pf_control
= Control
;
189 p_mux
->pf_addstream
= AddStream
;
190 p_mux
->pf_delstream
= DelStream
;
192 p_mux
->p_sys
= p_sys
;
197 /*****************************************************************************
199 *****************************************************************************/
200 static void Close( vlc_object_t
* p_this
)
202 sout_mux_t
*p_mux
= (sout_mux_t
*)p_this
;
203 sout_mux_sys_t
*p_sys
= p_mux
->p_sys
;
205 block_t
*p_hdr
, *p_idx1
;
208 msg_Dbg( p_mux
, "AVI muxer closed" );
210 /* first create idx1 chunk (write at the end of the stream */
211 p_idx1
= avi_HeaderCreateidx1( p_mux
);
212 p_sys
->i_idx1_size
= p_idx1
->i_buffer
;
213 sout_AccessOutWrite( p_mux
->p_access
, p_idx1
);
215 /* calculate some value for headers creations */
216 for( i_stream
= 0; i_stream
< p_sys
->i_streams
; i_stream
++ )
218 avi_stream_t
*p_stream
;
220 p_stream
= &p_sys
->stream
[i_stream
];
222 p_stream
->f_fps
= 25;
223 if( p_stream
->i_duration
> 0 )
225 p_stream
->f_fps
= (float)p_stream
->i_frames
/
226 ( (float)p_stream
->i_duration
/
229 p_stream
->i_bitrate
= 128 * 1024;
230 if( p_stream
->i_duration
> 0 )
232 p_stream
->i_bitrate
=
233 8 * (uint64_t)1000000 *
234 (uint64_t)p_stream
->i_totalsize
/
235 (uint64_t)p_stream
->i_duration
;
237 msg_Info( p_mux
, "stream[%d] duration:%"PRId64
" totalsize:%"PRId64
238 " frames:%d fps:%f KiB/s:%d",
240 (int64_t)p_stream
->i_duration
/ (int64_t)1000000,
241 p_stream
->i_totalsize
,
243 p_stream
->f_fps
, p_stream
->i_bitrate
/1024 );
246 p_hdr
= avi_HeaderCreateRIFF( p_mux
);
247 sout_AccessOutSeek( p_mux
->p_access
, 0 );
248 sout_AccessOutWrite( p_mux
->p_access
, p_hdr
);
250 for( i_stream
= 0; i_stream
< p_sys
->i_streams
; i_stream
++ )
252 avi_stream_t
*p_stream
;
253 p_stream
= &p_sys
->stream
[i_stream
];
254 free( p_stream
->p_bih
);
255 free( p_stream
->p_wf
);
257 free( p_sys
->idx1
.entry
);
261 static int Control( sout_mux_t
*p_mux
, int i_query
, va_list args
)
269 case MUX_CAN_ADD_STREAM_WHILE_MUXING
:
270 pb_bool
= (bool*)va_arg( args
, bool * );
274 case MUX_GET_ADD_STREAM_WAIT
:
275 pb_bool
= (bool*)va_arg( args
, bool * );
280 ppsz
= (char**)va_arg( args
, char ** );
281 *ppsz
= strdup( "video/avi" );
289 static int AddStream( sout_mux_t
*p_mux
, sout_input_t
*p_input
)
291 sout_mux_sys_t
*p_sys
= p_mux
->p_sys
;
292 avi_stream_t
*p_stream
;
294 if( p_sys
->i_streams
>= 100 )
296 msg_Err( p_mux
, "too many streams" );
300 msg_Dbg( p_mux
, "adding input" );
301 p_input
->p_sys
= malloc( sizeof( int ) );
302 if( !p_input
->p_sys
)
305 *((int*)p_input
->p_sys
) = p_sys
->i_streams
;
306 p_stream
= &p_sys
->stream
[p_sys
->i_streams
];
308 switch( p_input
->p_fmt
->i_cat
)
311 p_stream
->i_cat
= AUDIO_ES
;
312 p_stream
->fcc
[0] = '0' + p_sys
->i_streams
/ 10;
313 p_stream
->fcc
[1] = '0' + p_sys
->i_streams
% 10;
314 p_stream
->fcc
[2] = 'w';
315 p_stream
->fcc
[3] = 'b';
317 p_stream
->p_bih
= NULL
;
319 p_stream
->p_wf
= malloc( sizeof( WAVEFORMATEX
) +
320 p_input
->p_fmt
->i_extra
);
321 if( !p_stream
->p_wf
)
323 free( p_input
->p_sys
);
326 #define p_wf p_stream->p_wf
327 p_wf
->cbSize
= p_input
->p_fmt
->i_extra
;
328 if( p_wf
->cbSize
> 0 )
331 p_input
->p_fmt
->p_extra
,
332 p_input
->p_fmt
->i_extra
);
334 p_wf
->nChannels
= p_input
->p_fmt
->audio
.i_channels
;
335 p_wf
->nSamplesPerSec
= p_input
->p_fmt
->audio
.i_rate
;
336 p_wf
->nBlockAlign
= p_input
->p_fmt
->audio
.i_blockalign
;
337 p_wf
->nAvgBytesPerSec
= p_input
->p_fmt
->i_bitrate
/ 8;
338 p_wf
->wBitsPerSample
= 0;
340 switch( p_input
->p_fmt
->i_codec
)
343 p_wf
->wFormatTag
= WAVE_FORMAT_A52
;
344 p_wf
->nBlockAlign
= 1;
347 p_wf
->wFormatTag
= WAVE_FORMAT_MPEGLAYER3
;
348 p_wf
->nBlockAlign
= 1;
351 p_wf
->wFormatTag
= WAVE_FORMAT_WMA1
;
354 p_wf
->wFormatTag
= WAVE_FORMAT_WMA2
;
357 p_wf
->wFormatTag
= WAVE_FORMAT_WMAP
;
360 p_wf
->wFormatTag
= WAVE_FORMAT_WMAL
;
364 p_wf
->wFormatTag
= WAVE_FORMAT_PCM
;
365 p_wf
->nBlockAlign
= p_wf
->nChannels
;
366 p_wf
->wBitsPerSample
= 8;
367 p_wf
->nAvgBytesPerSec
= (p_wf
->wBitsPerSample
/8) *
368 p_wf
->nSamplesPerSec
* p_wf
->nChannels
;
371 p_wf
->wFormatTag
= WAVE_FORMAT_PCM
;
372 p_wf
->nBlockAlign
= 2 * p_wf
->nChannels
;
373 p_wf
->wBitsPerSample
= 16;
374 p_wf
->nAvgBytesPerSec
= (p_wf
->wBitsPerSample
/8) *
375 p_wf
->nSamplesPerSec
* p_wf
->nChannels
;
378 p_wf
->wFormatTag
= WAVE_FORMAT_PCM
;
379 p_wf
->nBlockAlign
= 3 * p_wf
->nChannels
;
380 p_wf
->wBitsPerSample
= 24;
381 p_wf
->nAvgBytesPerSec
= (p_wf
->wBitsPerSample
/8) *
382 p_wf
->nSamplesPerSec
* p_wf
->nChannels
;
385 p_wf
->wFormatTag
= WAVE_FORMAT_PCM
;
386 p_wf
->nBlockAlign
= 4 * p_wf
->nChannels
;
387 p_wf
->wBitsPerSample
= 32;
388 p_wf
->nAvgBytesPerSec
= (p_wf
->wBitsPerSample
/8) *
389 p_wf
->nSamplesPerSec
* p_wf
->nChannels
;
397 p_stream
->i_cat
= VIDEO_ES
;
398 p_stream
->fcc
[0] = '0' + p_sys
->i_streams
/ 10;
399 p_stream
->fcc
[1] = '0' + p_sys
->i_streams
% 10;
400 p_stream
->fcc
[2] = 'd';
401 p_stream
->fcc
[3] = 'c';
402 if( p_sys
->i_stream_video
< 0 )
404 p_sys
->i_stream_video
= p_sys
->i_streams
;
406 p_stream
->p_wf
= NULL
;
407 p_stream
->p_bih
= malloc( sizeof( VLC_BITMAPINFOHEADER
) +
408 p_input
->p_fmt
->i_extra
);
409 if( !p_stream
->p_bih
)
411 free( p_input
->p_sys
);
414 #define p_bih p_stream->p_bih
415 p_bih
->biSize
= sizeof( VLC_BITMAPINFOHEADER
) +
416 p_input
->p_fmt
->i_extra
;
417 if( p_input
->p_fmt
->i_extra
> 0 )
420 p_input
->p_fmt
->p_extra
,
421 p_input
->p_fmt
->i_extra
);
423 p_bih
->biWidth
= p_input
->p_fmt
->video
.i_width
;
424 p_bih
->biHeight
= p_input
->p_fmt
->video
.i_height
;
426 p_bih
->biBitCount
= 24;
427 p_bih
->biSizeImage
= 0;
428 p_bih
->biXPelsPerMeter
= 0;
429 p_bih
->biYPelsPerMeter
= 0;
430 p_bih
->biClrUsed
= 0;
431 p_bih
->biClrImportant
= 0;
432 switch( p_input
->p_fmt
->i_codec
)
435 p_bih
->biCompression
= VLC_FOURCC( 'X', 'V', 'I', 'D' );
438 p_bih
->biCompression
= p_input
->p_fmt
->i_original_fourcc
?: p_input
->p_fmt
->i_codec
;
444 return( VLC_EGENERIC
);
446 p_stream
->i_totalsize
= 0;
447 p_stream
->i_frames
= 0;
448 p_stream
->i_duration
= 0;
451 p_stream
->f_fps
= 25;
452 p_stream
->i_bitrate
= 128 * 1024;
455 return( VLC_SUCCESS
);
458 static int DelStream( sout_mux_t
*p_mux
, sout_input_t
*p_input
)
460 msg_Dbg( p_mux
, "removing input" );
462 free( p_input
->p_sys
);
466 static int Mux ( sout_mux_t
*p_mux
)
468 sout_mux_sys_t
*p_sys
= p_mux
->p_sys
;
469 avi_stream_t
*p_stream
;
472 if( p_sys
->b_write_header
)
476 msg_Dbg( p_mux
, "writing header" );
478 p_hdr
= avi_HeaderCreateRIFF( p_mux
);
479 sout_AccessOutWrite( p_mux
->p_access
, p_hdr
);
481 p_sys
->b_write_header
= false;
484 for( i
= 0; i
< p_mux
->i_nb_inputs
; i
++ )
487 block_fifo_t
*p_fifo
;
489 i_stream
= *((int*)p_mux
->pp_inputs
[i
]->p_sys
);
490 p_stream
= &p_sys
->stream
[i_stream
];
492 p_fifo
= p_mux
->pp_inputs
[i
]->p_fifo
;
493 i_count
= block_FifoCount( p_fifo
);
496 avi_idx1_entry_t
*p_idx
;
499 p_data
= block_FifoGet( p_fifo
);
500 if( block_FifoCount( p_fifo
) > 0 )
502 block_t
*p_next
= block_FifoShow( p_fifo
);
503 p_data
->i_length
= p_next
->i_dts
- p_data
->i_dts
;
507 if( p_stream
->i_frames
== 0 &&p_stream
->i_cat
== VIDEO_ES
)
509 /* Add header present at the end of BITMAP info header
510 to first frame in case of XVID */
511 if( p_stream
->p_bih
->biCompression
512 == VLC_FOURCC( 'X', 'V', 'I', 'D' ) )
514 int i_header_length
=
515 p_stream
->p_bih
->biSize
- sizeof(VLC_BITMAPINFOHEADER
);
516 p_data
= block_Realloc( p_data
,
517 i_header_length
, p_data
->i_buffer
);
520 memcpy(p_data
->p_buffer
,&p_stream
->p_bih
[1], i_header_length
);
524 p_stream
->i_frames
++;
525 if( p_data
->i_length
< 0 )
527 msg_Warn( p_mux
, "argg length < 0 l" );
528 block_Release( p_data
);
532 p_stream
->i_duration
+= p_data
->i_length
;
533 p_stream
->i_totalsize
+= p_data
->i_buffer
;
535 /* add idx1 entry for this frame */
536 p_idx
= &p_sys
->idx1
.entry
[p_sys
->idx1
.i_entry_count
];
537 memcpy( p_idx
->fcc
, p_stream
->fcc
, 4 );
539 if( ( p_data
->i_flags
& BLOCK_FLAG_TYPE_MASK
) == 0 || ( p_data
->i_flags
& BLOCK_FLAG_TYPE_I
) )
540 p_idx
->i_flags
= AVIIF_KEYFRAME
;
541 p_idx
->i_pos
= p_sys
->i_movi_size
+ 4;
542 p_idx
->i_length
= p_data
->i_buffer
;
543 p_sys
->idx1
.i_entry_count
++;
544 if( p_sys
->idx1
.i_entry_count
>= p_sys
->idx1
.i_entry_max
)
546 p_sys
->idx1
.i_entry_max
+= 10000;
547 p_sys
->idx1
.entry
= xrealloc( p_sys
->idx1
.entry
,
548 p_sys
->idx1
.i_entry_max
* sizeof( avi_idx1_entry_t
) );
551 p_data
= block_Realloc( p_data
, 8, p_data
->i_buffer
);
554 SetFCC( p_data
->p_buffer
, p_stream
->fcc
);
555 SetDWLE( p_data
->p_buffer
+ 4, p_data
->i_buffer
- 8 );
557 if( p_data
->i_buffer
& 0x01 )
559 p_data
= block_Realloc( p_data
, 0, p_data
->i_buffer
+ 1 );
560 p_data
->p_buffer
[ p_data
->i_buffer
- 1 ] = '\0';
563 p_sys
->i_movi_size
+= p_data
->i_buffer
;
564 sout_AccessOutWrite( p_mux
->p_access
, p_data
);
574 /****************************************************************************/
575 /****************************************************************************/
576 /****************************************************************************/
577 /****************************************************************************/
579 typedef struct buffer_out_s
585 static void bo_Init( buffer_out_t
*p_bo
, int i_size
)
587 p_bo
->p_block
= block_Alloc( i_size
);
588 p_bo
->p_block
->i_buffer
= 0;
589 p_bo
->i_buffer_size
= i_size
;
592 static void bo_SetByte( buffer_out_t
*p_bo
, int i_offset
, uint8_t i
)
594 if( i_offset
>= p_bo
->i_buffer_size
)
596 int i_growth
= HDR_BASE_SIZE
;
597 while( i_offset
>= p_bo
->i_buffer_size
+ i_growth
)
599 i_growth
+= HDR_BASE_SIZE
;
601 int i
= p_bo
->p_block
->i_buffer
; /* Realloc would set payload size == buffer size */
602 p_bo
->p_block
= block_Realloc( p_bo
->p_block
, 0, p_bo
->i_buffer_size
+ i_growth
);
603 p_bo
->p_block
->i_buffer
= i
;
604 p_bo
->i_buffer_size
+= i_growth
;
606 p_bo
->p_block
->p_buffer
[i_offset
] = i
;
608 static void bo_AddByte( buffer_out_t
*p_bo
, uint8_t i
)
610 bo_SetByte( p_bo
, p_bo
->p_block
->i_buffer
, i
);
611 p_bo
->p_block
->i_buffer
++;
613 static void bo_SetWordLE( buffer_out_t
*p_bo
, int i_offset
, uint16_t i
)
615 bo_SetByte( p_bo
, i_offset
, i
&0xff );
616 bo_SetByte( p_bo
, i_offset
+ 1, ( ( i
>> 8) &0xff ) );
618 static void bo_AddWordLE( buffer_out_t
*p_bo
, uint16_t i
)
620 bo_AddByte( p_bo
, i
&0xff );
621 bo_AddByte( p_bo
, ( ( i
>> 8) &0xff ) );
623 static void bo_AddWordBE( buffer_out_t
*p_bo
, uint16_t i
)
625 bo_AddByte( p_bo
, ( ( i
>> 8) &0xff ) );
626 bo_AddByte( p_bo
, i
&0xff );
628 static void bo_SetDWordLE( buffer_out_t
*p_bo
, int i_offset
, uint32_t i
)
630 bo_SetWordLE( p_bo
, i_offset
, i
&0xffff );
631 bo_SetWordLE( p_bo
, i_offset
+ 2, ( ( i
>> 16) &0xffff ) );
633 static void bo_AddDWordLE( buffer_out_t
*p_bo
, uint32_t i
)
635 bo_AddWordLE( p_bo
, i
&0xffff );
636 bo_AddWordLE( p_bo
, ( ( i
>> 16) &0xffff ) );
638 static void bo_AddDWordBE( buffer_out_t
*p_bo
, uint32_t i
)
640 bo_AddWordBE( p_bo
, ( ( i
>> 16) &0xffff ) );
641 bo_AddWordBE( p_bo
, i
&0xffff );
644 static void bo_AddLWordLE( buffer_out_t
*p_bo
, uint64_t i
)
646 bo_AddDWordLE( p_bo
, i
&0xffffffff );
647 bo_AddDWordLE( p_bo
, ( ( i
>> 32) &0xffffffff ) );
649 static void bo_AddLWordBE( buffer_out_t
*p_bo
, uint64_t i
)
651 bo_AddDWordBE( p_bo
, ( ( i
>> 32) &0xffffffff ) );
652 bo_AddDWordBE( p_bo
, i
&0xffffffff );
656 static void bo_AddFCC( buffer_out_t
*p_bo
, const char *fcc
)
658 bo_AddByte( p_bo
, fcc
[0] );
659 bo_AddByte( p_bo
, fcc
[1] );
660 bo_AddByte( p_bo
, fcc
[2] );
661 bo_AddByte( p_bo
, fcc
[3] );
664 static void bo_AddMem( buffer_out_t
*p_bo
, int i_size
, uint8_t *p_mem
)
668 for( i
= 0; i
< i_size
; i
++ )
670 bo_AddByte( p_bo
, p_mem
[i
] );
674 /****************************************************************************
675 ****************************************************************************
677 ** avi header generation
679 ****************************************************************************
680 ****************************************************************************/
681 #define AVI_BOX_ENTER( fcc ) \
682 int i_datasize_offset; \
683 bo_AddFCC( p_bo, fcc ); \
684 i_datasize_offset = p_bo->p_block->i_buffer; \
685 bo_AddDWordLE( p_bo, 0 )
687 #define AVI_BOX_ENTER_LIST( fcc ) \
688 AVI_BOX_ENTER( "LIST" ); \
689 bo_AddFCC( p_bo, fcc )
691 #define AVI_BOX_EXIT( i_err ) \
692 if( p_bo->p_block->i_buffer&0x01 ) bo_AddByte( p_bo, 0 ); \
693 bo_SetDWordLE( p_bo, i_datasize_offset, p_bo->p_block->i_buffer - i_datasize_offset - 4 ); \
696 static int avi_HeaderAdd_avih( sout_mux_t
*p_mux
,
699 sout_mux_sys_t
*p_sys
= p_mux
->p_sys
;
700 avi_stream_t
*p_video
= NULL
;
702 uint32_t i_microsecperframe
;
703 int i_maxbytespersec
;
705 AVI_BOX_ENTER( "avih" );
707 if( p_sys
->i_stream_video
>= 0 )
709 p_video
= &p_sys
->stream
[p_sys
->i_stream_video
];
710 if( p_video
->i_frames
<= 0 )
719 (uint32_t)( (float)1000000 /
720 (float)p_sys
->stream
[p_sys
->i_stream_video
].f_fps
);
721 i_totalframes
= p_sys
->stream
[p_sys
->i_stream_video
].i_frames
;
725 msg_Warn( p_mux
, "avi file without video track isn't a good idea..." );
726 i_microsecperframe
= 0;
730 for( i_stream
= 0,i_maxbytespersec
= 0; i_stream
< p_sys
->i_streams
; i_stream
++ )
732 if( p_sys
->stream
[i_stream
].i_duration
> 0 )
735 p_sys
->stream
[i_stream
].i_totalsize
/
736 p_sys
->stream
[i_stream
].i_duration
;
740 bo_AddDWordLE( p_bo
, i_microsecperframe
);
741 bo_AddDWordLE( p_bo
, i_maxbytespersec
);
742 bo_AddDWordLE( p_bo
, 0 ); /* padding */
743 bo_AddDWordLE( p_bo
, AVIF_TRUSTCKTYPE
|
745 AVIF_ISINTERLEAVED
); /* flags */
746 bo_AddDWordLE( p_bo
, i_totalframes
);
747 bo_AddDWordLE( p_bo
, 0 ); /* initial frame */
748 bo_AddDWordLE( p_bo
, p_sys
->i_streams
); /* streams count */
749 bo_AddDWordLE( p_bo
, 1024 * 1024 ); /* suggested buffer size */
752 bo_AddDWordLE( p_bo
, p_video
->p_bih
->biWidth
);
753 bo_AddDWordLE( p_bo
, p_video
->p_bih
->biHeight
);
757 bo_AddDWordLE( p_bo
, 0 );
758 bo_AddDWordLE( p_bo
, 0 );
760 bo_AddDWordLE( p_bo
, 0 ); /* ???? */
761 bo_AddDWordLE( p_bo
, 0 ); /* ???? */
762 bo_AddDWordLE( p_bo
, 0 ); /* ???? */
763 bo_AddDWordLE( p_bo
, 0 ); /* ???? */
767 static int avi_HeaderAdd_strh( buffer_out_t
*p_bo
, avi_stream_t
*p_stream
)
769 AVI_BOX_ENTER( "strh" );
771 switch( p_stream
->i_cat
)
775 bo_AddFCC( p_bo
, "vids" );
776 bo_AddDWordBE( p_bo
, p_stream
->p_bih
->biCompression
);
777 bo_AddDWordLE( p_bo
, 0 ); /* flags */
778 bo_AddWordLE( p_bo
, 0 ); /* priority */
779 bo_AddWordLE( p_bo
, 0 ); /* langage */
780 bo_AddDWordLE( p_bo
, 0 ); /* initial frame */
781 bo_AddDWordLE( p_bo
, 1000 );/* scale */
782 bo_AddDWordLE( p_bo
, (uint32_t)( 1000 * p_stream
->f_fps
));
783 bo_AddDWordLE( p_bo
, 0 ); /* start */
784 bo_AddDWordLE( p_bo
, p_stream
->i_frames
);
785 bo_AddDWordLE( p_bo
, 1024 * 1024 );
786 bo_AddDWordLE( p_bo
, -1 ); /* quality */
787 bo_AddDWordLE( p_bo
, 0 ); /* samplesize */
788 bo_AddWordLE( p_bo
, 0 ); /* ??? */
789 bo_AddWordLE( p_bo
, 0 ); /* ??? */
790 bo_AddWordLE( p_bo
, p_stream
->p_bih
->biWidth
);
791 bo_AddWordLE( p_bo
, p_stream
->p_bih
->biHeight
);
796 int i_rate
, i_scale
, i_samplesize
;
798 i_samplesize
= p_stream
->p_wf
->nBlockAlign
;
799 if( i_samplesize
> 1 )
801 i_scale
= i_samplesize
;
802 i_rate
= /*i_scale **/ p_stream
->i_bitrate
/ 8;
808 i_rate
= 1000 * p_stream
->i_bitrate
/ 8;
810 bo_AddFCC( p_bo
, "auds" );
811 bo_AddDWordLE( p_bo
, 0 ); /* tag */
812 bo_AddDWordLE( p_bo
, 0 ); /* flags */
813 bo_AddWordLE( p_bo
, 0 ); /* priority */
814 bo_AddWordLE( p_bo
, 0 ); /* langage */
815 bo_AddDWordLE( p_bo
, 0 ); /* initial frame */
816 bo_AddDWordLE( p_bo
, i_scale
);/* scale */
817 bo_AddDWordLE( p_bo
, i_rate
);
818 bo_AddDWordLE( p_bo
, 0 ); /* start */
819 bo_AddDWordLE( p_bo
, p_stream
->i_frames
);
820 bo_AddDWordLE( p_bo
, 10 * 1024 );
821 bo_AddDWordLE( p_bo
, -1 ); /* quality */
822 bo_AddDWordLE( p_bo
, i_samplesize
);
823 bo_AddWordLE( p_bo
, 0 ); /* ??? */
824 bo_AddWordLE( p_bo
, 0 ); /* ??? */
825 bo_AddWordLE( p_bo
, 0 );
826 bo_AddWordLE( p_bo
, 0 );
834 static int avi_HeaderAdd_strf( buffer_out_t
*p_bo
, avi_stream_t
*p_stream
)
836 AVI_BOX_ENTER( "strf" );
838 switch( p_stream
->i_cat
)
841 bo_AddWordLE( p_bo
, p_stream
->p_wf
->wFormatTag
);
842 bo_AddWordLE( p_bo
, p_stream
->p_wf
->nChannels
);
843 bo_AddDWordLE( p_bo
, p_stream
->p_wf
->nSamplesPerSec
);
844 bo_AddDWordLE( p_bo
, p_stream
->p_wf
->nAvgBytesPerSec
);
845 bo_AddWordLE( p_bo
, p_stream
->p_wf
->nBlockAlign
);
846 bo_AddWordLE( p_bo
, p_stream
->p_wf
->wBitsPerSample
);
847 bo_AddWordLE( p_bo
, p_stream
->p_wf
->cbSize
);
848 bo_AddMem( p_bo
, p_stream
->p_wf
->cbSize
, (uint8_t*)&p_stream
->p_wf
[1] );
851 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biSize
);
852 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biWidth
);
853 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biHeight
);
854 bo_AddWordLE( p_bo
, p_stream
->p_bih
->biPlanes
);
855 bo_AddWordLE( p_bo
, p_stream
->p_bih
->biBitCount
);
856 if( VLC_FOURCC( 0, 0, 0, 1 ) == 0x00000001 )
858 bo_AddDWordBE( p_bo
, p_stream
->p_bih
->biCompression
);
862 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biCompression
);
864 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biSizeImage
);
865 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biXPelsPerMeter
);
866 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biYPelsPerMeter
);
867 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biClrUsed
);
868 bo_AddDWordLE( p_bo
, p_stream
->p_bih
->biClrImportant
);
870 p_stream
->p_bih
->biSize
- sizeof( VLC_BITMAPINFOHEADER
),
871 (uint8_t*)&p_stream
->p_bih
[1] );
878 static int avi_HeaderAdd_strl( buffer_out_t
*p_bo
, avi_stream_t
*p_stream
)
880 AVI_BOX_ENTER_LIST( "strl" );
882 avi_HeaderAdd_strh( p_bo
, p_stream
);
883 avi_HeaderAdd_strf( p_bo
, p_stream
);
888 static int avi_HeaderAdd_meta( buffer_out_t
*p_bo
, const char psz_meta
[4],
889 const char *psz_data
)
891 if ( psz_data
== NULL
) return 1;
892 const char *psz
= psz_data
;
893 AVI_BOX_ENTER( psz_meta
);
894 while (*psz
) bo_AddByte( p_bo
, *psz
++ );
895 bo_AddByte( p_bo
, 0 );
899 static int avi_HeaderAdd_INFO( sout_mux_t
*p_mux
, buffer_out_t
*p_bo
)
903 #define APPLY_META(var, fourcc) \
904 psz = var_InheritString( p_mux, SOUT_CFG_PREFIX var );\
907 avi_HeaderAdd_meta( p_bo, fourcc, psz );\
911 AVI_BOX_ENTER_LIST( "INFO" );
913 APPLY_META( "artist", "IART")
914 APPLY_META( "comment", "ICMT")
915 APPLY_META( "copyright","ICOP")
916 APPLY_META( "date", "ICRD")
917 APPLY_META( "genre", "IGNR")
918 APPLY_META( "name", "INAM")
919 APPLY_META( "keywords", "IKEY")
920 APPLY_META( "subject", "ISBJ")
921 APPLY_META( "encoder", "ISFT")
922 /* Some are missing, but are they really useful ?? */
929 static block_t
*avi_HeaderCreateRIFF( sout_mux_t
*p_mux
)
931 sout_mux_sys_t
*p_sys
= p_mux
->p_sys
;
943 bo_Init( &bo
, HDR_BASE_SIZE
);
945 bo_AddFCC( &bo
, "RIFF" );
946 offsets
.i_riffsize
= bo
.p_block
->i_buffer
;
947 bo_AddDWordLE( &bo
, 0xEFBEADDE );
948 bo_AddFCC( &bo
, "AVI " );
950 bo_AddFCC( &bo
, "LIST" );
951 /* HDRL List size should exclude following data in HDR buffer
952 * -12 (RIFF, RIFF size, 'AVI ' tag),
953 * - 8 (hdr1 LIST tag and its size)
954 * - 12 (movi LIST tag, size, 'movi' listType )
956 offsets
.i_hdrllistsize
= bo
.p_block
->i_buffer
;
957 bo_AddDWordLE( &bo
, 0xEFBEADDE );
958 bo_AddFCC( &bo
, "hdrl" );
959 offsets
.i_hdrldatastart
= bo
.p_block
->i_buffer
;
961 avi_HeaderAdd_avih( p_mux
, &bo
);
962 for( i_stream
= 0; i_stream
< p_sys
->i_streams
; i_stream
++ )
964 avi_HeaderAdd_strl( &bo
, &p_sys
->stream
[i_stream
] );
967 /* align on 16 bytes */
968 int i_align
= ( ( bo
.p_block
->i_buffer
+ 12 + 0xE ) & ~ 0xF );
969 i_junk
= i_align
- bo
.p_block
->i_buffer
;
970 bo_AddFCC( &bo
, "JUNK" );
971 bo_AddDWordLE( &bo
, i_junk
);
972 for( int i
=0; i
< i_junk
; i
++ )
974 bo_AddByte( &bo
, 0 );
977 /* Now set hdrl size */
978 bo_SetDWordLE( &bo
, offsets
.i_hdrllistsize
,
979 bo
.p_block
->i_buffer
- offsets
.i_hdrldatastart
);
981 avi_HeaderAdd_INFO( p_mux
, &bo
);
983 bo_AddFCC( &bo
, "LIST" );
984 bo_AddDWordLE( &bo
, p_sys
->i_movi_size
+ 4 );
985 bo_AddFCC( &bo
, "movi" );
987 /* Now set RIFF size */
988 bo_SetDWordLE( &bo
, offsets
.i_riffsize
, bo
.p_block
->i_buffer
- 8
989 + p_sys
->i_movi_size
+ p_sys
->i_idx1_size
);
991 return( bo
.p_block
);
994 static block_t
* avi_HeaderCreateidx1( sout_mux_t
*p_mux
)
996 sout_mux_sys_t
*p_sys
= p_mux
->p_sys
;
997 uint32_t i_idx1_size
;
1000 i_idx1_size
= 16 * p_sys
->idx1
.i_entry_count
+ 8;
1002 bo_Init( &bo
, i_idx1_size
);
1003 memset( bo
.p_block
->p_buffer
, 0, i_idx1_size
);
1005 bo_AddFCC( &bo
, "idx1" );
1006 bo_AddDWordLE( &bo
, i_idx1_size
- 8);
1008 for( unsigned i
= 0; i
< p_sys
->idx1
.i_entry_count
; i
++ )
1010 bo_AddFCC( &bo
, p_sys
->idx1
.entry
[i
].fcc
);
1011 bo_AddDWordLE( &bo
, p_sys
->idx1
.entry
[i
].i_flags
);
1012 bo_AddDWordLE( &bo
, p_sys
->idx1
.entry
[i
].i_pos
);
1013 bo_AddDWordLE( &bo
, p_sys
->idx1
.entry
[i
].i_length
);
1016 return( bo
.p_block
);