/*****************************************************************************
* Preamble
*****************************************************************************/
-#include <stdlib.h>
-#include <string.h>
#include <vlc/vlc.h>
-#include <vlc/input.h>
-#include <vlc/sout.h>
+#include <vlc_sout.h>
+#include <vlc_block.h>
+#include <vlc_codecs.h>
#ifdef HAVE_TIME_H
#include <time.h>
uint64_t i_stco_pos;
vlc_bool_t b_stco64;
- /* for h264 */
- struct
- {
- int i_profile;
- int i_profile_compat;
- int i_level;
-
- int i_sps;
- uint8_t *sps;
- int i_pps;
- uint8_t *pps;
- } avc;
-
/* for spu */
int64_t i_last_dts;
static void bo_add_24be ( bo_t *, uint32_t );
static void bo_add_32be ( bo_t *, uint32_t );
static void bo_add_64be ( bo_t *, uint64_t );
-static void bo_add_fourcc(bo_t *, char * );
+static void bo_add_fourcc(bo_t *, const char * );
static void bo_add_bo ( bo_t *, bo_t * );
static void bo_add_mem ( bo_t *, int , uint8_t * );
static void bo_add_descr( bo_t *, uint8_t , uint32_t );
static void bo_fix_32be ( bo_t *, int , uint32_t );
-static bo_t *box_new ( char *fcc );
-static bo_t *box_full_new( char *fcc, uint8_t v, uint32_t f );
+static bo_t *box_new ( const char *fcc );
+static bo_t *box_full_new( const char *fcc, uint8_t v, uint32_t f );
static void box_fix ( bo_t *box );
static void box_free ( bo_t *box );
static void box_gather ( bo_t *box, bo_t *box2 );
static bo_t *GetMoovBox( sout_mux_t *p_mux );
static block_t *ConvertSUBT( sout_mux_t *, mp4_stream_t *, block_t *);
-static void ConvertAVC1( sout_mux_t *, mp4_stream_t *, block_t * );
+static block_t *ConvertAVC1( sout_mux_t *, mp4_stream_t *, block_t * );
/*****************************************************************************
* Open:
bo_t *box;
msg_Dbg( p_mux, "Mp4 muxer opend" );
- sout_CfgParse( p_mux, SOUT_CFG_PREFIX, ppsz_sout_options, p_mux->p_cfg );
+ config_ChainParse( p_mux, SOUT_CFG_PREFIX, ppsz_sout_options, p_mux->p_cfg );
p_mux->pf_control = Control;
p_mux->pf_addstream = AddStream;
mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
es_format_Clean( &p_stream->fmt );
- if( p_stream->avc.i_sps ) free( p_stream->avc.sps );
- if( p_stream->avc.i_pps ) free( p_stream->avc.pps );
free( p_stream->entry );
free( p_stream );
}
case VLC_FOURCC( 'h', '2', '6', '4' ):
case VLC_FOURCC( 's', 'a', 'm', 'r' ):
case VLC_FOURCC( 's', 'a', 'w', 'b' ):
+ case VLC_FOURCC( 'Y', 'V', '1', '2' ):
+ case VLC_FOURCC( 'Y', 'U', 'Y', '2' ):
break;
case VLC_FOURCC( 's', 'u', 'b', 't' ):
msg_Warn( p_mux, "subtitle track added like in .mov (even when creating .mp4)" );
calloc( p_stream->i_entry_max, sizeof( mp4_entry_t ) );
p_stream->i_dts_start = 0;
p_stream->i_duration = 0;
- p_stream->avc.i_profile = 77;
- p_stream->avc.i_profile_compat = 64;
- p_stream->avc.i_level = 30;
- p_stream->avc.i_sps = 0;
- p_stream->avc.sps = NULL;
- p_stream->avc.i_pps = 0;
- p_stream->avc.pps = NULL;
p_input->p_sys = p_stream;
block_fifo_t *p_fifo = p_mux->pp_inputs[i]->p_fifo;
block_t *p_buf;
- if( p_fifo->i_depth <= 1 )
+ if( block_FifoCount( p_fifo ) <= 1 )
{
if( p_mux->pp_inputs[i]->p_fmt->i_cat != SPU_ES )
{
p_input = p_mux->pp_inputs[i_stream];
p_stream = (mp4_stream_t*)p_input->p_sys;
+again:
p_data = block_FifoGet( p_input->p_fifo );
if( p_stream->fmt.i_codec == VLC_FOURCC( 'h', '2', '6', '4' ) )
{
- ConvertAVC1( p_mux, p_stream, p_data );
+ p_data = ConvertAVC1( p_mux, p_stream, p_data );
}
else if( p_stream->fmt.i_codec == VLC_FOURCC( 's', 'u', 'b', 't' ) )
{
p_data = ConvertSUBT( p_mux, p_stream, p_data );
}
+ if( p_data == NULL ) goto again;
if( p_stream->fmt.i_cat != SPU_ES )
{
/* Fix length of the sample */
- if( p_input->p_fifo->i_depth > 0 )
+ if( block_FifoCount( p_input->p_fifo ) > 0 )
{
block_t *p_next = block_FifoShow( p_input->p_fifo );
int64_t i_diff = p_next->i_dts - p_data->i_dts;
return p_block;
}
-static void ConvertAVC1( sout_mux_t *p_mux, mp4_stream_t *tk, block_t *p_block )
+static block_t *ConvertAVC1( sout_mux_t *p_mux, mp4_stream_t *tk, block_t *p_block )
{
uint8_t *last = p_block->p_buffer; /* Assume it starts with 0x00000001 */
uint8_t *dat = &p_block->p_buffer[4];
last[2] = ( i_size >> 8 )&0xff;
last[3] = ( i_size )&0xff;
- if( (last[4]&0x1f) == 7 && tk->avc.i_sps <= 0 ) /* SPS */
+ /* Skip blocks with SPS/PPS */
+ if( (last[4]&0x1f) == 7 || (last[4]&0x1f) == 8 )
{
- tk->avc.i_sps = i_size;
- tk->avc.sps = malloc( i_size );
- memcpy( tk->avc.sps, &last[4], i_size );
-
- tk->avc.i_profile = tk->avc.sps[1];
- tk->avc.i_profile = tk->avc.sps[2];
- tk->avc.i_level = tk->avc.sps[3];
+ ; // FIXME Find a way to skip dat without frelling everything
}
- else if( (last[4]&0x1f) == 8 && tk->avc.i_pps <= 0 ) /* PPS */
- {
- tk->avc.i_pps = i_size;
- tk->avc.pps = malloc( i_size );
- memcpy( tk->avc.pps, &last[4], i_size );
- }
-
last = dat;
-
dat += 4;
}
+ return p_block;
}
static int GetDescrLength( int i_size )
static bo_t *GetAvcCTag( mp4_stream_t *p_stream )
{
- bo_t *avcC;
+ bo_t *avcC = NULL;
+ uint8_t *p_sps = NULL;
+ uint8_t *p_pps = NULL;
+ int i_sps_size = 0;
+ int i_pps_size = 0;
+ if( p_stream->fmt.i_extra > 0 )
+ {
+ /* FIXME: take into account multiple sps/pps */
+ uint8_t *p_buffer = p_stream->fmt.p_extra;
+ int i_buffer = p_stream->fmt.i_extra;
+
+ while( i_buffer > 4 &&
+ p_buffer[0] == 0 && p_buffer[1] == 0 &&
+ p_buffer[2] == 0 && p_buffer[3] == 1 )
+ {
+ const int i_nal_type = p_buffer[4]&0x1f;
+ int i_offset = 1;
+ int i_size = 0;
+ int i_startcode = 0;
+
+ //msg_Dbg( p_stream, "we found a startcode for NAL with TYPE:%d", i_nal_type );
+
+ for( i_offset = 1; i_offset+3 < i_buffer ; i_offset++)
+ {
+ if( p_buffer[i_offset] == 0 && p_buffer[i_offset+1] == 0 &&
+ p_buffer[i_offset+2] == 0 && p_buffer[i_offset+3] == 1 )
+ {
+ /* we found another startcode */
+ i_startcode = i_offset;
+ break;
+ }
+ }
+ i_size = i_startcode ? i_startcode : i_buffer;
+ if( i_nal_type == 7 )
+ {
+ p_sps = &p_buffer[4];
+ i_sps_size = i_size - 4;
+ }
+ if( i_nal_type == 8 )
+ {
+ p_pps = &p_buffer[4];
+ i_pps_size = i_size - 4;
+ }
+ i_buffer -= i_size;
+ p_buffer += i_size;
+ }
+ }
+
/* FIXME use better value */
avcC = box_new( "avcC" );
bo_add_8( avcC, 1 ); /* configuration version */
- bo_add_8( avcC, p_stream->avc.i_profile );
- bo_add_8( avcC, p_stream->avc.i_profile_compat );
- bo_add_8( avcC, p_stream->avc.i_level ); /* level, 5.1 */
+ bo_add_8( avcC, i_sps_size ? p_sps[1] : 77 );
+ bo_add_8( avcC, i_sps_size ? p_sps[2] : 64 );
+ bo_add_8( avcC, i_sps_size ? p_sps[3] : 30 ); /* level, 5.1 */
bo_add_8( avcC, 0xff ); /* 0b11111100 | lengthsize = 0x11 */
- bo_add_8( avcC, 0xe0 | (p_stream->avc.i_sps > 0 ? 1 : 0) ); /* 0b11100000 | sps_count */
- if( p_stream->avc.i_sps > 0 )
+ bo_add_8( avcC, 0xe0 | (i_sps_size > 0 ? 1 : 0) ); /* 0b11100000 | sps_count */
+ if( i_sps_size > 0 )
{
- bo_add_16be( avcC, p_stream->avc.i_sps );
- bo_add_mem( avcC, p_stream->avc.i_sps, p_stream->avc.sps );
+ bo_add_16be( avcC, i_sps_size );
+ bo_add_mem( avcC, i_sps_size, p_sps );
}
- bo_add_8( avcC, (p_stream->avc.i_pps > 0 ? 1 : 0) ); /* pps_count */
- if( p_stream->avc.i_pps > 0 )
+ bo_add_8( avcC, (i_pps_size > 0 ? 1 : 0) ); /* pps_count */
+ if( i_pps_size > 0 )
{
- bo_add_16be( avcC, p_stream->avc.i_pps );
- bo_add_mem( avcC, p_stream->avc.i_pps, p_stream->avc.pps );
+ bo_add_16be( avcC, i_pps_size );
+ bo_add_mem( avcC, i_pps_size, p_pps );
}
box_fix( avcC );
{
#define ADD_META_BOX( type, box_string ) { \
bo_t *box = NULL; \
- if( p_meta->psz_##type ) box = box_new( "\251" box_string ); \
+ if( vlc_meta_Get( p_meta, vlc_meta_##type ) ) box = box_new( "\251" box_string ); \
if( box ) \
{ \
- bo_add_16be( box, strlen( p_meta->psz_##type ) ); \
+ bo_add_16be( box, strlen( vlc_meta_Get( p_meta, vlc_meta_##type ) )); \
bo_add_16be( box, 0 ); \
- bo_add_mem( box, strlen( p_meta->psz_##type ), \
- (uint8_t*)(p_meta->psz_##type ) ); \
+ bo_add_mem( box, strlen( vlc_meta_Get( p_meta, vlc_meta_##type ) ), \
+ (uint8_t*)(vlc_meta_Get( p_meta, vlc_meta_##type ) ) ); \
box_fix( box ); \
box_gather( udta, box ); \
} }
- ADD_META_BOX( title, "nam" );
- ADD_META_BOX( author, "aut" );
- ADD_META_BOX( artist, "ART" );
- ADD_META_BOX( genre, "gen" );
- ADD_META_BOX( copyright, "cpy" );
- ADD_META_BOX( description, "des" );
- ADD_META_BOX( date, "day" );
- ADD_META_BOX( url, "url" );
+ ADD_META_BOX( Title, "nam" );
+ ADD_META_BOX( Artist, "ART" );
+ ADD_META_BOX( Genre, "gen" );
+ ADD_META_BOX( Copyright, "cpy" );
+ ADD_META_BOX( Description, "des" );
+ ADD_META_BOX( Date, "day" );
+ ADD_META_BOX( URL, "url" );
#undef ADD_META_BOX
}
memcpy( fcc, "avc1", 4 );
break;
+ case VLC_FOURCC('Y','V','1','2'):
+ memcpy( fcc, "yv12", 4 );
+ break;
+
+ case VLC_FOURCC('Y','U','Y','2'):
+ memcpy( fcc, "yuy2", 4 );
+ break;
+
default:
memcpy( fcc, (char*)&p_stream->fmt.i_codec, 4 );
break;
return stbl;
}
-static int64_t get_timestamp();
+static int64_t get_timestamp(void);
static uint32_t mvhd_matrix[9] =
{ 0x10000, 0, 0, 0, 0x10000, 0, 0, 0, 0x40000000 };
bo_add_32be( p_bo, i &0xffffffff );
}
-static void bo_add_fourcc( bo_t *p_bo, char *fcc )
+static void bo_add_fourcc( bo_t *p_bo, const char *fcc )
{
bo_add_8( p_bo, fcc[0] );
bo_add_8( p_bo, fcc[1] );
i_length = i_size;
vals[3] = (unsigned char)(i_length & 0x7f);
i_length >>= 7;
- vals[2] = (unsigned char)((i_length & 0x7f) | 0x80);
+ vals[2] = (unsigned char)((i_length & 0x7f) | 0x80);
i_length >>= 7;
- vals[1] = (unsigned char)((i_length & 0x7f) | 0x80);
+ vals[1] = (unsigned char)((i_length & 0x7f) | 0x80);
i_length >>= 7;
vals[0] = (unsigned char)((i_length & 0x7f) | 0x80);
}
}
-static bo_t * box_new( char *fcc )
+static bo_t * box_new( const char *fcc )
{
bo_t *box;
return box;
}
-static bo_t * box_full_new( char *fcc, uint8_t v, uint32_t f )
+static bo_t * box_full_new( const char *fcc, uint8_t v, uint32_t f )
{
bo_t *box;
sout_AccessOutWrite( p_mux->p_access, p_buf );
}
-static int64_t get_timestamp()
+static int64_t get_timestamp(void)
{
int64_t i_timestamp = 0;