// D.1 SEI payload syntax void write_sei_payload(h264_stream_t* h, bs_t* b, int payloadType, int payloadSize) { sei_t* s = h->sei; int i; for ( i = 0; i < s->payloadSize; i++ ) bs_write_u(b, 8, s->payload[i]); }
//This function assumes the values are set elsewhere. static void write_sei_type_0(h264_stream_t* h, sei_t* s, bs_t* b) { int sched_sel_idx; sps_t *sps = h->sps; sei_type_0 *bp = s->sei_type_struct; if(sps->vui.nal_hrd_parameters_present_flag) { for (sched_sel_idx = 0; sched_sel_idx < sps->hrd.cpb_cnt_minus1 + 1; sched_sel_idx++) { bs_write_u(b, bp->initial_cbp_removal_delay[sched_sel_idx], h->sps->hrd.initial_cpb_removal_delay_length_minus1 + 1); bs_write_u(b, bp->initial_cbp_removal_delay_offset[sched_sel_idx], sps->hrd.initial_cpb_removal_delay_length_minus1 + 1); } } if (sps->vui.vcl_hrd_parameters_present_flag) { for (sched_sel_idx = 0; sched_sel_idx < sps->hrd.cpb_cnt_minus1 + 1; sched_sel_idx++) { bs_write_u(b, bp->initial_cbp_removal_delay[sched_sel_idx], h->sps->hrd.initial_cpb_removal_delay_length_minus1 + 1); bs_write_u(b, bp->initial_cbp_removal_delay_offset[sched_sel_idx], sps->hrd.initial_cpb_removal_delay_length_minus1 + 1); } } }
int write_avcc(avcc_t* avcc, h264_stream_t* h, bs_t* b) { bs_write_u8(b, 1); // configurationVersion = 1; bs_write_u8(b, avcc->AVCProfileIndication); bs_write_u8(b, avcc->profile_compatibility); bs_write_u8(b, avcc->AVCLevelIndication); bs_write_u(b, 6, 0x3F); // reserved = '111111'b; bs_write_u(b, 2, avcc->lengthSizeMinusOne); bs_write_u(b, 3, 0x07); // reserved = '111'b; bs_write_u(b, 5, avcc->numOfSequenceParameterSets); int i; for (i = 0; i < avcc->numOfSequenceParameterSets; i++) { int max_len = 1024; // FIXME uint8_t* buf = (uint8_t*)malloc(max_len); h->nal->nal_ref_idc = 3; // NAL_REF_IDC_PRIORITY_HIGHEST; h->nal->nal_unit_type = NAL_UNIT_TYPE_SPS; h->sps = avcc->sps_table[i]; int len = write_nal_unit(h, buf, max_len); if (len < 0) { free(buf); continue; } // TODO report errors int sequenceParameterSetLength = len; bs_write_u(b, 16, sequenceParameterSetLength); bs_write_bytes(b, buf, len); free(buf); } bs_write_u(b, 8, avcc->numOfPictureParameterSets); for (i = 0; i < avcc->numOfPictureParameterSets; i++) { int max_len = 1024; // FIXME uint8_t* buf = (uint8_t*)malloc(max_len); h->nal->nal_ref_idc = 3; // NAL_REF_IDC_PRIORITY_HIGHEST; h->nal->nal_unit_type = NAL_UNIT_TYPE_PPS; h->pps = avcc->pps_table[i]; int len = write_nal_unit(h, buf, max_len); if (len < 0) { free(buf); continue; } // TODO report errors int pictureParameterSetLength = len; bs_write_u(b, 16, pictureParameterSetLength); bs_write_bytes(b, buf, len); free(buf); } if (bs_overrun(b)) { return -1; } return bs_pos(b); }
// D.1 SEI payload syntax void write_sei_payload( h264_stream_t* h, bs_t* b, int payloadType, int payloadSize) { sei_t* s = h->sei; int i; switch (payloadType) { case SEI_TYPE_BUFFERING_PERIOD: write_sei_type_0(h, s, b); break; case SEI_TYPE_PIC_TIMING: //payloadSize = sizeof(sei_type_1) write_sei_type_1(h, s, b); break; default: for( i = 0; i < payloadSize; i++ ) { bs_write_u(b, 8, s->payload[i]); } //Not implemented break; } }
void bs_write_u8(bs_t* b, uint32_t v) { bs_write_u(b, 8, v); }
void bs_write_f(bs_t* b, int n, uint32_t v) { bs_write_u(b, n, v); }
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); } } } }
static void write_sei_type_1(h264_stream_t* h, sei_t* s, bs_t* b) { bs_t bs; sps_t *sps = h->sps; sei_type_1 *pt_struct = s->sei_type_struct; if(sps->vui.nal_hrd_parameters_present_flag || sps->vui.vcl_hrd_parameters_present_flag) { bs_write_u(b, pt_struct->cpb_removal_delay, sps->hrd.cpb_removal_delay_length_minus1 + 1); bs_write_u(b, pt_struct->dpb_output_delay, sps->hrd.cpb_removal_delay_length_minus1 + 1); } if (sps->vui.pic_struct_present_flag) { if (pt_struct->pic_struct > SEI_PIC_STRUCT_FRAME_TRIPLING) { return; } bs_write_u(&bs, pt_struct->pic_struct, 4); for (int i = 0; i < pt_struct->NumClockTS; i++) { sei_type_1_pic_timing *pt = &pt_struct->timings[i]; bs_write_u(&bs, pt->clock_timestamp_flag, 1); if (pt->clock_timestamp_flag) { bs_write_u(&bs, pt->ct_type, 2); bs_write_u(&bs, pt->nuit_field_based_flag, 1); bs_write_u(&bs, pt->counting_type, 5); bs_write_u(&bs, pt->full_timestamp_flag, 1); bs_write_u(&bs, pt->discontinuity_flag, 1); bs_write_u(&bs, pt->cnt_dropped_flag, 1); bs_write_u(&bs, pt->n_frames, 8); if (pt->full_timestamp_flag) { bs_write_u(&bs, pt->seconds_value, 6); bs_write_u(&bs, pt->minutes_value, 6); bs_write_u(&bs, pt->hours_flag, 5); }else { bs_write_u(&bs, pt->seconds_flag, 1); if (pt->seconds_flag) { bs_write_u(&bs, pt->seconds_value, 6); bs_write_u(&bs, pt->minutes_flag, 1); if(pt->minutes_flag) { bs_write_u(&bs, pt->minutes_value, 6); bs_write_u(&bs, pt->hours_flag, 1); if (pt->hours_flag) { bs_write_u(&bs, pt->hours_value, 5); } } } } if (sps->hrd.time_offset_length > 0) { bs_write_u(&bs, pt->time_offset, sps->hrd.time_offset_length); } } } } s->sei_type_struct = pt_struct; }