void bs_write_ue(bs_t* b, uint32_t v)
{
static const int len_table[256] =
{
1,
1,
2,2,
3,3,3,3,
4,4,4,4,4,4,4,4,
5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,
6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
};
int len;
if (v == 0)
{
bs_write_u1(b, 1);
}
else
{
v++;
if (v >= 0x01000000)
{
len = 24 + len_table[ v >> 24 ];
}
else if(v >= 0x00010000)
int pes_header_write(pes_header_t *ph, bs_t *b) {
// TODO: add support for PES-level stuffing
int start_pos = bs_pos(b);
bs_write_u24(b, PES_PACKET_START_CODE_PREFIX);
bs_write_u8(b, ph->stream_id);
bs_write_u16(b, ph->PES_packet_length);
if (HAS_PES_HEADER(ph->stream_id)) {
bs_write_u(b, 2, 0x02);
bs_write_u(b, 2, ph->PES_scrambling_control);
bs_write_u1(b, ph->PES_priority);
bs_write_u1(b, ph->data_alignment_indicator);
bs_write_u1(b, ph->copyright);
bs_write_u1(b, ph->original_or_copy);
bs_write_u(b, 2, ph->PTS_DTS_flags);
bs_write_u1(b, ph->ESCR_flag);
bs_write_u1(b, ph->ES_rate_flag);
bs_write_u1(b, ph->DSM_trick_mode_flag);
bs_write_u1(b, ph->additional_copy_info_flag);
bs_write_u1(b, ph->PES_CRC_flag);
bs_write_u1(b, ph->PES_extension_flag);
bs_write_u8(b, ph->PES_header_data_length);
if (ph->PTS_DTS_flags & PES_PTS_FLAG) {
bs_write_u(b, 4, 0x02);
bs_write_90khz_timestamp(b, ph->PTS);
}
if (ph->PTS_DTS_flags & PES_DTS_FLAG) {
bs_write_u(b, 4, 0x01);
bs_write_90khz_timestamp(b, ph->DTS);
}
if (ph->ESCR_flag) {
bs_write_reserved(b, 2);
bs_write_90khz_timestamp(b, ph->ESCR_base);
bs_write_u(b, 9, ph->ESCR_extension);
bs_write_marker_bit(b);
}
if (ph->ES_rate_flag) {
bs_write_marker_bit(b);
bs_write_u(b, 22, ph->ES_rate);
bs_write_marker_bit(b);
}
if (ph->DSM_trick_mode_flag) {
bs_write_u(b, 3, ph->trick_mode_control);
switch (ph->trick_mode_control) {
case PES_DSM_TRICK_MODE_CTL_FAST_FORWARD:
case PES_DSM_TRICK_MODE_CTL_FAST_REVERSE:
bs_write_u(b, 2, ph->field_id);
bs_write_u1(b, ph->intra_slice_refresh);
bs_write_u(b, 2, ph->frequency_truncation);
break;
case PES_DSM_TRICK_MODE_CTL_SLOW_MOTION:
case PES_DSM_TRICK_MODE_CTL_SLOW_REVERSE:
bs_write_u(b, 5, ph->rep_cntrl);
break;
case PES_DSM_TRICK_MODE_CTL_FREEZE_FRAME:
bs_write_u(b, 2, ph->field_id);
bs_write_reserved(b, 3);
break;
default:
bs_write_reserved(b, 5);
break;
}
}
if (ph->additional_copy_info_flag) {
bs_write_marker_bit(b);
bs_write_u(b, 7, ph->additional_copy_info);
}
if (ph->PES_CRC_flag) {
bs_write_u16(b, ph->previous_PES_packet_CRC);
}
if (ph->PES_extension_flag) {
bs_write_u1(b, ph->PES_private_data_flag);
bs_write_u1(b, 0); // pack_header not supported
bs_write_u1(b, ph->program_packet_sequence_counter_flag);
bs_write_u1(b, ph->PSTD_buffer_flag);
bs_write_reserved(b, 3);
bs_write_u1(b, ph->PES_extension_flag_2);
if (ph->PES_private_data_flag) {
bs_write_bytes(b, ph->PES_private_data, 16);
}
if (ph->program_packet_sequence_counter_flag) {
bs_write_marker_bit(b);
bs_write_u(b, 7, ph->program_packet_sequence_counter);
bs_write_marker_bit(b);
bs_write_u1(b, ph->MPEG1_MPEG2_identifier);
bs_write_u(b, 6, ph->original_stuff_length);
}
if (ph->PSTD_buffer_flag) {
bs_write_u(b, 2, 0x01);
bs_write_u1(b, ph->PSTD_buffer_scale);
bs_write_u(b, 13, ph->PSTD_buffer_size);
}
if (ph->PES_extension_flag_2) {
bs_write_marker_bit(b);
bs_write_u(b, 7, ph->PES_extension_field_length);
bs_write_u1(b, ph->stream_id_extension_flag);
if (!ph->stream_id_extension_flag) {
bs_write_u(b, 7, ph->stream_id_extension);
} else {
bs_write_reserved(b, 6);
bs_write_u1(b, ph->tref_extension_flag);
if (ph->tref_extension_flag) {
bs_write_reserved(b, 4);
bs_write_90khz_timestamp(b, ph->TREF);
}
}
}
}
}
return (bs_pos(b) - start_pos);
}
// Appendix G.13.1.1 Scalability information SEI message syntax
void write_sei_scalability_info( h264_stream_t* h, bs_t* b )
{
sei_scalability_info_t* sei_svc = h->sei->sei_svc;
bs_write_u1(b, sei_svc->temporal_id_nesting_flag);
bs_write_u1(b, sei_svc->priority_layer_info_present_flag);
bs_write_u1(b, sei_svc->priority_id_setting_flag);
bs_write_ue(b, sei_svc->num_layers_minus1);
for( int i = 0; i <= sei_svc->num_layers_minus1; i++ ) {
bs_write_ue(b, sei_svc->layers[i].layer_id);
bs_write_u(b, 6, sei_svc->layers[i].priority_id);
bs_write_u1(b, sei_svc->layers[i].discardable_flag);
bs_write_u(b, 3, sei_svc->layers[i].dependency_id);
bs_write_u(b, 4, sei_svc->layers[i].quality_id);
bs_write_u(b, 3, sei_svc->layers[i].temporal_id);
bs_write_u1(b, sei_svc->layers[i].sub_pic_layer_flag);
bs_write_u1(b, sei_svc->layers[i].sub_region_layer_flag);
bs_write_u1(b, sei_svc->layers[i].iroi_division_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].profile_level_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].bitrate_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].frm_rate_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].frm_size_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].layer_dependency_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].parameter_sets_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].bitstream_restriction_info_present_flag);
bs_write_u1(b, sei_svc->layers[i].exact_inter_layer_pred_flag);
if( sei_svc->layers[i].sub_pic_layer_flag ||
sei_svc->layers[i].iroi_division_info_present_flag )
{
bs_write_u1(b, sei_svc->layers[i].exact_sample_value_match_flag);
}
bs_write_u1(b, sei_svc->layers[i].layer_conversion_flag);
bs_write_u1(b, sei_svc->layers[i].layer_output_flag);
if( sei_svc->layers[i].profile_level_info_present_flag )
{
bs_write_u(b, 24, sei_svc->layers[i].layer_profile_level_idc);
}
if( sei_svc->layers[i].bitrate_info_present_flag )
{
bs_write_u(b, 16, sei_svc->layers[i].avg_bitrate);
bs_write_u(b, 16, sei_svc->layers[i].max_bitrate_layer);
bs_write_u(b, 16, sei_svc->layers[i].max_bitrate_layer_representation);
bs_write_u(b, 16, sei_svc->layers[i].max_bitrate_calc_window);
}
if( sei_svc->layers[i].frm_rate_info_present_flag )
{
bs_write_u(b, 2, sei_svc->layers[i].constant_frm_rate_idc);
bs_write_u(b, 16, sei_svc->layers[i].avg_frm_rate);
}
if( sei_svc->layers[i].frm_size_info_present_flag ||
sei_svc->layers[i].iroi_division_info_present_flag )
{
bs_write_ue(b, sei_svc->layers[i].frm_width_in_mbs_minus1);
bs_write_ue(b, sei_svc->layers[i].frm_height_in_mbs_minus1);
}
if( sei_svc->layers[i].sub_region_layer_flag )
{
bs_write_ue(b, sei_svc->layers[i].base_region_layer_id);
bs_write_u1(b, sei_svc->layers[i].dynamic_rect_flag);
if( sei_svc->layers[i].dynamic_rect_flag )
{
bs_write_u(b, 16, sei_svc->layers[i].horizontal_offset);
bs_write_u(b, 16, sei_svc->layers[i].vertical_offset);
bs_write_u(b, 16, sei_svc->layers[i].region_width);
bs_write_u(b, 16, sei_svc->layers[i].region_height);
}
}
if( sei_svc->layers[i].sub_pic_layer_flag )
{
bs_write_ue(b, sei_svc->layers[i].roi_id);
}
if( sei_svc->layers[i].iroi_division_info_present_flag )
{
bs_write_u1(b, sei_svc->layers[i].iroi_grid_flag);
if( sei_svc->layers[i].iroi_grid_flag )
{
bs_write_ue(b, sei_svc->layers[i].grid_width_in_mbs_minus1);
bs_write_ue(b, sei_svc->layers[i].grid_height_in_mbs_minus1);
}
else
{
bs_write_ue(b, sei_svc->layers[i].num_rois_minus1);
for( int j = 0; j <= sei_svc->layers[i].num_rois_minus1; j++ )
{
bs_write_ue(b, sei_svc->layers[i].roi[j].first_mb_in_roi);
bs_write_ue(b, sei_svc->layers[i].roi[j].roi_width_in_mbs_minus1);
bs_write_ue(b, sei_svc->layers[i].roi[j].roi_height_in_mbs_minus1);
}
}
}
if( sei_svc->layers[i].layer_dependency_info_present_flag )
{
bs_write_ue(b, sei_svc->layers[i].num_directly_dependent_layers);
for( int j = 0; j < sei_svc->layers[i].num_directly_dependent_layers; j++ )
{
bs_write_ue(b, sei_svc->layers[i].directly_dependent_layer_id_delta_minus1[j]);
}
}
else
{
bs_write_ue(b, sei_svc->layers[i].layer_dependency_info_src_layer_id_delta);
}
if( sei_svc->layers[i].parameter_sets_info_present_flag )
{
bs_write_ue(b, sei_svc->layers[i].num_seq_parameter_sets);
for( int j = 0; j < sei_svc->layers[i].num_seq_parameter_sets; j++ )
{
bs_write_ue(b, sei_svc->layers[i].seq_parameter_set_id_delta[j]);
}
bs_write_ue(b, sei_svc->layers[i].num_subset_seq_parameter_sets);
for( int j = 0; j < sei_svc->layers[i].num_subset_seq_parameter_sets; j++ )
{
bs_write_ue(b, sei_svc->layers[i].subset_seq_parameter_set_id_delta[j]);
}
bs_write_ue(b, sei_svc->layers[i].num_pic_parameter_sets_minus1);
for( int j = 0; j < sei_svc->layers[i].num_pic_parameter_sets_minus1; j++ )
{
bs_write_ue(b, sei_svc->layers[i].pic_parameter_set_id_delta[j]);
}
}
else
{
bs_write_ue(b, sei_svc->layers[i].parameter_sets_info_src_layer_id_delta);
}
if( sei_svc->layers[i].bitstream_restriction_info_present_flag )
{
bs_write_u1(b, sei_svc->layers[i].motion_vectors_over_pic_boundaries_flag);
bs_write_ue(b, sei_svc->layers[i].max_bytes_per_pic_denom);
bs_write_ue(b, sei_svc->layers[i].max_bits_per_mb_denom);
bs_write_ue(b, sei_svc->layers[i].log2_max_mv_length_horizontal);
bs_write_ue(b, sei_svc->layers[i].log2_max_mv_length_vertical);
bs_write_ue(b, sei_svc->layers[i].max_num_reorder_frames);
bs_write_ue(b, sei_svc->layers[i].max_dec_frame_buffering);
}
if( sei_svc->layers[i].layer_conversion_flag )
{
bs_write_ue(b, sei_svc->layers[i].conversion_type_idc);
for( int j = 0; j < 2; j++ )
{
bs_write_u(b, 1, sei_svc->layers[i].rewriting_info_flag[j]);
if( sei_svc->layers[i].rewriting_info_flag[j] )
{
bs_write_u(b, 24, sei_svc->layers[i].rewriting_profile_level_idc[j]);
bs_write_u(b, 16, sei_svc->layers[i].rewriting_avg_bitrate[j]);
bs_write_u(b, 16, sei_svc->layers[i].rewriting_max_bitrate[j]);
}
}
}
}
if( sei_svc->priority_layer_info_present_flag )
{
bs_write_ue(b, sei_svc->pr_num_dIds_minus1);
for( int i = 0; i <= sei_svc->pr_num_dIds_minus1; i++ ) {
bs_write_u(b, 3, sei_svc->pr[i].pr_dependency_id);
bs_write_ue(b, sei_svc->pr[i].pr_num_minus1);
for( int j = 0; j <= sei_svc->pr[i].pr_num_minus1; j++ )
{
bs_write_ue(b, sei_svc->pr[i].pr_info[j].pr_id);
bs_write_u(b, 24, sei_svc->pr[i].pr_info[j].pr_profile_level_idc);
bs_write_u(b, 16, sei_svc->pr[i].pr_info[j].pr_avg_bitrate);
bs_write_u(b, 16, sei_svc->pr[i].pr_info[j].pr_max_bitrate);
}
}
}
}
