static int h265_metadata_update_sps(AVBSFContext *bsf, H265RawSPS *sps) { H265MetadataContext *ctx = bsf->priv_data; int need_vui = 0; int crop_unit_x, crop_unit_y; if (ctx->sample_aspect_ratio.num && ctx->sample_aspect_ratio.den) { // Table E-1. static const AVRational sar_idc[] = { { 0, 0 }, // Unspecified (never written here). { 1, 1 }, { 12, 11 }, { 10, 11 }, { 16, 11 }, { 40, 33 }, { 24, 11 }, { 20, 11 }, { 32, 11 }, { 80, 33 }, { 18, 11 }, { 15, 11 }, { 64, 33 }, { 160, 99 }, { 4, 3 }, { 3, 2 }, { 2, 1 }, }; int num, den, i; av_reduce(&num, &den, ctx->sample_aspect_ratio.num, ctx->sample_aspect_ratio.den, 65535); for (i = 1; i < FF_ARRAY_ELEMS(sar_idc); i++) { if (num == sar_idc[i].num && den == sar_idc[i].den) break; } if (i == FF_ARRAY_ELEMS(sar_idc)) { sps->vui.aspect_ratio_idc = 255; sps->vui.sar_width = num; sps->vui.sar_height = den; } else { sps->vui.aspect_ratio_idc = i; } sps->vui.aspect_ratio_info_present_flag = 1; need_vui = 1; } #define SET_OR_INFER(field, value, present_flag, infer) do { \ if (value >= 0) { \ field = value; \ need_vui = 1; \ } else if (!present_flag) \ field = infer; \ } while (0) if (ctx->video_format >= 0 || ctx->video_full_range_flag >= 0 || ctx->colour_primaries >= 0 || ctx->transfer_characteristics >= 0 || ctx->matrix_coefficients >= 0) { SET_OR_INFER(sps->vui.video_format, ctx->video_format, sps->vui.video_signal_type_present_flag, 5); SET_OR_INFER(sps->vui.video_full_range_flag, ctx->video_full_range_flag, sps->vui.video_signal_type_present_flag, 0); if (ctx->colour_primaries >= 0 || ctx->transfer_characteristics >= 0 || ctx->matrix_coefficients >= 0) { SET_OR_INFER(sps->vui.colour_primaries, ctx->colour_primaries, sps->vui.colour_description_present_flag, 2); SET_OR_INFER(sps->vui.transfer_characteristics, ctx->transfer_characteristics, sps->vui.colour_description_present_flag, 2); SET_OR_INFER(sps->vui.matrix_coefficients, ctx->matrix_coefficients, sps->vui.colour_description_present_flag, 2); sps->vui.colour_description_present_flag = 1; } sps->vui.video_signal_type_present_flag = 1; need_vui = 1; } if (ctx->chroma_sample_loc_type >= 0) { sps->vui.chroma_sample_loc_type_top_field = ctx->chroma_sample_loc_type; sps->vui.chroma_sample_loc_type_bottom_field = ctx->chroma_sample_loc_type; sps->vui.chroma_loc_info_present_flag = 1; need_vui = 1; } if (ctx->tick_rate.num && ctx->tick_rate.den) { int num, den; av_reduce(&num, &den, ctx->tick_rate.num, ctx->tick_rate.den, UINT32_MAX > INT_MAX ? UINT32_MAX : INT_MAX); sps->vui.vui_time_scale = num; sps->vui.vui_num_units_in_tick = den; sps->vui.vui_timing_info_present_flag = 1; need_vui = 1; if (ctx->num_ticks_poc_diff_one > 0) { sps->vui.vui_num_ticks_poc_diff_one_minus1 = ctx->num_ticks_poc_diff_one - 1; sps->vui.vui_poc_proportional_to_timing_flag = 1; } else if (ctx->num_ticks_poc_diff_one == 0) { sps->vui.vui_poc_proportional_to_timing_flag = 0; } } if (sps->separate_colour_plane_flag || sps->chroma_format_idc == 0) { crop_unit_x = 1; crop_unit_y = 1; } else { crop_unit_x = 1 + (sps->chroma_format_idc < 3); crop_unit_y = 1 + (sps->chroma_format_idc < 2); } #define CROP(border, unit) do { \ if (ctx->crop_ ## border >= 0) { \ if (ctx->crop_ ## border % unit != 0) { \ av_log(bsf, AV_LOG_ERROR, "Invalid value for crop_%s: " \ "must be a multiple of %d.\n", #border, unit); \ return AVERROR(EINVAL); \ } \ sps->conf_win_ ## border ## _offset = \ ctx->crop_ ## border / unit; \ sps->conformance_window_flag = 1; \ } \ } while (0) CROP(left, crop_unit_x); CROP(right, crop_unit_x); CROP(top, crop_unit_y); CROP(bottom, crop_unit_y); #undef CROP if (need_vui) sps->vui_parameters_present_flag = 1; return 0; }
static int h264_metadata_update_sps(AVBSFContext *bsf, H264RawSPS *sps) { H264MetadataContext *ctx = bsf->priv_data; int need_vui = 0; int crop_unit_x, crop_unit_y; if (ctx->sample_aspect_ratio.num && ctx->sample_aspect_ratio.den) { // Table E-1. static const AVRational sar_idc[] = { { 0, 0 }, // Unspecified (never written here). { 1, 1 }, { 12, 11 }, { 10, 11 }, { 16, 11 }, { 40, 33 }, { 24, 11 }, { 20, 11 }, { 32, 11 }, { 80, 33 }, { 18, 11 }, { 15, 11 }, { 64, 33 }, { 160, 99 }, { 4, 3 }, { 3, 2 }, { 2, 1 }, }; int num, den, i; av_reduce(&num, &den, ctx->sample_aspect_ratio.num, ctx->sample_aspect_ratio.den, 65535); for (i = 1; i < FF_ARRAY_ELEMS(sar_idc); i++) { if (num == sar_idc[i].num && den == sar_idc[i].den) break; } if (i == FF_ARRAY_ELEMS(sar_idc)) { sps->vui.aspect_ratio_idc = 255; sps->vui.sar_width = num; sps->vui.sar_height = den; } else { sps->vui.aspect_ratio_idc = i; } sps->vui.aspect_ratio_info_present_flag = 1; need_vui = 1; } #define SET_OR_INFER(field, value, present_flag, infer) do { \ if (value >= 0) { \ field = value; \ need_vui = 1; \ } else if (!present_flag) \ field = infer; \ } while (0) if (ctx->video_format >= 0 || ctx->video_full_range_flag >= 0 || ctx->colour_primaries >= 0 || ctx->transfer_characteristics >= 0 || ctx->matrix_coefficients >= 0) { SET_OR_INFER(sps->vui.video_format, ctx->video_format, sps->vui.video_signal_type_present_flag, 5); SET_OR_INFER(sps->vui.video_full_range_flag, ctx->video_full_range_flag, sps->vui.video_signal_type_present_flag, 0); if (ctx->colour_primaries >= 0 || ctx->transfer_characteristics >= 0 || ctx->matrix_coefficients >= 0) { SET_OR_INFER(sps->vui.colour_primaries, ctx->colour_primaries, sps->vui.colour_description_present_flag, 2); SET_OR_INFER(sps->vui.transfer_characteristics, ctx->transfer_characteristics, sps->vui.colour_description_present_flag, 2); SET_OR_INFER(sps->vui.matrix_coefficients, ctx->matrix_coefficients, sps->vui.colour_description_present_flag, 2); sps->vui.colour_description_present_flag = 1; } sps->vui.video_signal_type_present_flag = 1; need_vui = 1; } if (ctx->chroma_sample_loc_type >= 0) { sps->vui.chroma_sample_loc_type_top_field = ctx->chroma_sample_loc_type; sps->vui.chroma_sample_loc_type_bottom_field = ctx->chroma_sample_loc_type; sps->vui.chroma_loc_info_present_flag = 1; need_vui = 1; } if (ctx->tick_rate.num && ctx->tick_rate.den) { int num, den; av_reduce(&num, &den, ctx->tick_rate.num, ctx->tick_rate.den, UINT32_MAX > INT_MAX ? UINT32_MAX : INT_MAX); sps->vui.time_scale = num; sps->vui.num_units_in_tick = den; sps->vui.timing_info_present_flag = 1; need_vui = 1; } SET_OR_INFER(sps->vui.fixed_frame_rate_flag, ctx->fixed_frame_rate_flag, sps->vui.timing_info_present_flag, 0); if (sps->separate_colour_plane_flag || sps->chroma_format_idc == 0) { crop_unit_x = 1; crop_unit_y = 2 - sps->frame_mbs_only_flag; } else { crop_unit_x = 1 + (sps->chroma_format_idc < 3); crop_unit_y = (1 + (sps->chroma_format_idc < 2)) * (2 - sps->frame_mbs_only_flag); } #define CROP(border, unit) do { \ if (ctx->crop_ ## border >= 0) { \ if (ctx->crop_ ## border % unit != 0) { \ av_log(bsf, AV_LOG_ERROR, "Invalid value for crop_%s: " \ "must be a multiple of %d.\n", #border, unit); \ return AVERROR(EINVAL); \ } \ sps->frame_crop_ ## border ## _offset = \ ctx->crop_ ## border / unit; \ sps->frame_cropping_flag = 1; \ } \ } while (0) CROP(left, crop_unit_x); CROP(right, crop_unit_x); CROP(top, crop_unit_y); CROP(bottom, crop_unit_y); #undef CROP if (ctx->level != LEVEL_UNSET) { int level_idc; if (ctx->level == LEVEL_AUTO) { const H264LevelDescriptor *desc; int64_t bit_rate; int width, height, dpb_frames; if (sps->vui.nal_hrd_parameters_present_flag) { bit_rate = (sps->vui.nal_hrd_parameters.bit_rate_value_minus1[0] + 1) * (INT64_C(1) << (sps->vui.nal_hrd_parameters.bit_rate_scale + 6)); } else if (sps->vui.vcl_hrd_parameters_present_flag) { bit_rate = (sps->vui.vcl_hrd_parameters.bit_rate_value_minus1[0] + 1) * (INT64_C(1) << (sps->vui.vcl_hrd_parameters.bit_rate_scale + 6)); // Adjust for VCL vs. NAL limits. bit_rate = bit_rate * 6 / 5; } else { bit_rate = 0; } // Don't use max_dec_frame_buffering if it is only inferred. dpb_frames = sps->vui.bitstream_restriction_flag ? sps->vui.max_dec_frame_buffering : H264_MAX_DPB_FRAMES; width = 16 * (sps->pic_width_in_mbs_minus1 + 1); height = 16 * (sps->pic_height_in_map_units_minus1 + 1) * (2 - sps->frame_mbs_only_flag); desc = ff_h264_guess_level(sps->profile_idc, bit_rate, width, height, dpb_frames); if (desc) { level_idc = desc->level_idc; } else { av_log(bsf, AV_LOG_WARNING, "Stream does not appear to " "conform to any level: using level 6.2.\n"); level_idc = 62; } } else { level_idc = ctx->level; } if (level_idc == 9) { if (sps->profile_idc == 66 || sps->profile_idc == 77 || sps->profile_idc == 88) { sps->level_idc = 11; sps->constraint_set3_flag = 1; } else { sps->level_idc = 9; } } else { sps->level_idc = level_idc; } } if (need_vui) sps->vui_parameters_present_flag = 1; return 0; }