static int write_frame( hnd_t handle, uint8_t *p_nalu, int i_size, x264_picture_t *p_picture, int i_ref_idc )
{
mkv_hnd_t *p_mkv = handle;
if( !p_mkv->b_writing_frame )
{
if( mk_start_frame( p_mkv->w ) < 0 )
return -1;
p_mkv->b_writing_frame = 1;
}
if( mk_add_frame_data( p_mkv->w, p_nalu, i_size ) < 0 )
return -1;
int64_t i_stamp = (int64_t)((p_picture->i_pts * 1e9 * p_mkv->i_timebase_num / p_mkv->i_timebase_den) + 0.5);
p_mkv->b_writing_frame = 0;
if( mk_set_frame_flags( p_mkv->w, i_stamp, p_picture->b_keyframe, p_picture->i_type == X264_TYPE_B ) < 0 )
return -1;
return i_size;
}
static int write_headers_mpeg2( hnd_t handle, x264_nal_t *p_nal )
{
mkv_hnd_t *p_mkv = handle;
int mpeg2_size = p_nal[0].i_payload;
int ret;
uint8_t *mpeg2C;
if( !p_mkv->width || !p_mkv->height ||
!p_mkv->d_width || !p_mkv->d_height )
return -1;
mpeg2C = malloc( mpeg2_size );
if( !mpeg2C )
return -1;
memcpy( mpeg2C, p_nal[0].p_payload, mpeg2_size );
ret = mk_write_header( p_mkv->w, "x264" X264_VERSION, "V_MPEG2", mpeg2C, mpeg2_size,
p_mkv->frame_duration, 50000,
p_mkv->width, p_mkv->height,
p_mkv->d_width, p_mkv->d_height, p_mkv->display_size_units );
if( ret < 0 )
return ret;
free( mpeg2C );
if( !p_mkv->b_writing_frame )
{
if( mk_start_frame( p_mkv->w ) < 0 )
return -1;
p_mkv->b_writing_frame = 1;
}
return mpeg2_size;
}
static int write_headers( hnd_t handle, x264_nal_t *p_nal )
{
mkv_hnd_t *p_mkv = handle;
int sps_size = p_nal[0].i_payload - 4;
int pps_size = p_nal[1].i_payload - 4;
int sei_size = p_nal[2].i_payload;
uint8_t *sps = p_nal[0].p_payload + 4;
uint8_t *pps = p_nal[1].p_payload + 4;
uint8_t *sei = p_nal[2].p_payload;
int ret;
uint8_t *avcC;
int avcC_len;
if( !p_mkv->width || !p_mkv->height ||
!p_mkv->d_width || !p_mkv->d_height )
return -1;
avcC_len = 5 + 1 + 2 + sps_size + 1 + 2 + pps_size;
avcC = malloc( avcC_len );
if( !avcC )
return -1;
avcC[0] = 1;
avcC[1] = sps[1];
avcC[2] = sps[2];
avcC[3] = sps[3];
avcC[4] = 0xff; // nalu size length is four bytes
avcC[5] = 0xe1; // one sps
avcC[6] = sps_size >> 8;
avcC[7] = sps_size;
memcpy( avcC+8, sps, sps_size );
avcC[8+sps_size] = 1; // one pps
avcC[9+sps_size] = pps_size >> 8;
avcC[10+sps_size] = pps_size;
memcpy( avcC+11+sps_size, pps, pps_size );
ret = mk_write_header( p_mkv->w, "x264" X264_VERSION, "V_MPEG4/ISO/AVC",
avcC, avcC_len, p_mkv->frame_duration, 50000,
p_mkv->width, p_mkv->height,
p_mkv->d_width, p_mkv->d_height, p_mkv->display_size_units );
if( ret < 0 )
return ret;
free( avcC );
// SEI
if( !p_mkv->b_writing_frame )
{
if( mk_start_frame( p_mkv->w ) < 0 )
return -1;
p_mkv->b_writing_frame = 1;
}
if( mk_add_frame_data( p_mkv->w, sei, sei_size ) < 0 )
return -1;
return sei_size + sps_size + pps_size;
}
