2016-02-22 23:32:22 +00:00
|
|
|
// Copyright 2016 Google Inc. All rights reserved.
|
|
|
|
//
|
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// license that can be found in the LICENSE file or at
|
|
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
|
2016-05-25 18:03:17 +00:00
|
|
|
#include "packager/media/codecs/h265_parser.h"
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
#include <math.h>
|
2016-08-17 17:41:40 +00:00
|
|
|
#include <algorithm>
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
#include "packager/base/logging.h"
|
2016-04-13 21:46:13 +00:00
|
|
|
#include "packager/media/base/macros.h"
|
2016-05-25 18:03:17 +00:00
|
|
|
#include "packager/media/codecs/nalu_reader.h"
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
#define TRUE_OR_RETURN(a) \
|
|
|
|
do { \
|
|
|
|
if (!(a)) { \
|
|
|
|
DVLOG(1) << "Failure while processing " << #a; \
|
|
|
|
return kInvalidStream; \
|
|
|
|
} \
|
|
|
|
} while (0)
|
|
|
|
|
|
|
|
#define OK_OR_RETURN(a) \
|
|
|
|
do { \
|
|
|
|
Result status = (a); \
|
|
|
|
if (status != kOk) \
|
|
|
|
return status; \
|
|
|
|
} while (false)
|
|
|
|
|
2016-05-20 21:19:33 +00:00
|
|
|
namespace shaka {
|
2016-02-22 23:32:22 +00:00
|
|
|
namespace media {
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
int GetNumPicTotalCurr(const H265SliceHeader& slice_header,
|
|
|
|
const H265Sps& sps) {
|
|
|
|
int num_pic_total_curr = 0;
|
|
|
|
const H265ReferencePictureSet& ref_pic_set =
|
|
|
|
slice_header.short_term_ref_pic_set_sps_flag
|
|
|
|
? sps.st_ref_pic_sets[slice_header.short_term_ref_pic_set_idx]
|
|
|
|
: slice_header.st_ref_pic_set;
|
|
|
|
|
|
|
|
for (int i = 0; i < ref_pic_set.num_negative_pics; i++) {
|
|
|
|
if (ref_pic_set.used_by_curr_pic_s0[i])
|
|
|
|
num_pic_total_curr++;
|
|
|
|
}
|
|
|
|
for (int i = 0; i < ref_pic_set.num_positive_pics; i++) {
|
|
|
|
if (ref_pic_set.used_by_curr_pic_s1[i])
|
|
|
|
num_pic_total_curr++;
|
|
|
|
}
|
|
|
|
|
|
|
|
return num_pic_total_curr + slice_header.used_by_curr_pic_lt;
|
|
|
|
}
|
2016-04-13 21:46:13 +00:00
|
|
|
|
|
|
|
void GetAspectRatioInfo(const H265Sps& sps,
|
|
|
|
uint32_t* pixel_width,
|
|
|
|
uint32_t* pixel_height) {
|
|
|
|
// The default value is 0; so if this is not in the SPS, it will correctly
|
|
|
|
// assume unspecified.
|
|
|
|
int aspect_ratio_idc = sps.vui_parameters.aspect_ratio_idc;
|
|
|
|
|
|
|
|
// Table E.1
|
|
|
|
switch (aspect_ratio_idc) {
|
|
|
|
case 1: *pixel_width = 1; *pixel_height = 1; break;
|
|
|
|
case 2: *pixel_width = 12; *pixel_height = 11; break;
|
|
|
|
case 3: *pixel_width = 10; *pixel_height = 11; break;
|
|
|
|
case 4: *pixel_width = 16; *pixel_height = 11; break;
|
|
|
|
case 5: *pixel_width = 40; *pixel_height = 33; break;
|
|
|
|
case 6: *pixel_width = 24; *pixel_height = 11; break;
|
|
|
|
case 7: *pixel_width = 20; *pixel_height = 11; break;
|
|
|
|
case 8: *pixel_width = 32; *pixel_height = 11; break;
|
|
|
|
case 9: *pixel_width = 80; *pixel_height = 33; break;
|
|
|
|
case 10: *pixel_width = 18; *pixel_height = 11; break;
|
|
|
|
case 11: *pixel_width = 15; *pixel_height = 11; break;
|
|
|
|
case 12: *pixel_width = 64; *pixel_height = 33; break;
|
|
|
|
case 13: *pixel_width = 160; *pixel_height = 99; break;
|
|
|
|
case 14: *pixel_width = 4; *pixel_height = 3; break;
|
|
|
|
case 15: *pixel_width = 3; *pixel_height = 2; break;
|
|
|
|
case 16: *pixel_width = 2; *pixel_height = 1; break;
|
|
|
|
|
|
|
|
case 255:
|
|
|
|
*pixel_width = sps.vui_parameters.sar_width;
|
|
|
|
*pixel_height = sps.vui_parameters.sar_height;
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
// Section E.3.1 specifies that other values should be interpreted as 0.
|
|
|
|
LOG(WARNING) << "Unknown aspect_ratio_idc " << aspect_ratio_idc;
|
|
|
|
FALLTHROUGH_INTENDED;
|
|
|
|
case 0:
|
|
|
|
// Unlike the spec, assume 1:1 if not specified.
|
|
|
|
*pixel_width = 1;
|
|
|
|
*pixel_height = 1;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2016-02-22 23:32:22 +00:00
|
|
|
} // namespace
|
|
|
|
|
2016-04-13 21:46:13 +00:00
|
|
|
bool ExtractResolutionFromSps(const H265Sps& sps,
|
|
|
|
uint32_t* coded_width,
|
|
|
|
uint32_t* coded_height,
|
|
|
|
uint32_t* pixel_width,
|
|
|
|
uint32_t* pixel_height) {
|
|
|
|
int crop_x = 0;
|
|
|
|
int crop_y = 0;
|
|
|
|
if (sps.conformance_window_flag) {
|
|
|
|
int sub_width_c = 0;
|
|
|
|
int sub_height_c = 0;
|
|
|
|
|
|
|
|
// Table 6-1
|
|
|
|
switch (sps.chroma_format_idc) {
|
|
|
|
case 0: // Monochrome
|
|
|
|
sub_width_c = 1;
|
|
|
|
sub_height_c = 1;
|
|
|
|
break;
|
|
|
|
case 1: // 4:2:0
|
|
|
|
sub_width_c = 2;
|
|
|
|
sub_height_c = 2;
|
|
|
|
break;
|
|
|
|
case 2: // 4:2:2
|
|
|
|
sub_width_c = 2;
|
|
|
|
sub_height_c = 1;
|
|
|
|
break;
|
|
|
|
case 3: // 4:4:4
|
|
|
|
sub_width_c = 1;
|
|
|
|
sub_height_c = 1;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
LOG(ERROR) << "Unexpected chroma_format_idc " << sps.chroma_format_idc;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Formula D-28, D-29
|
|
|
|
crop_x =
|
|
|
|
sub_width_c * (sps.conf_win_right_offset + sps.conf_win_left_offset);
|
|
|
|
crop_y =
|
|
|
|
sub_height_c * (sps.conf_win_bottom_offset + sps.conf_win_top_offset);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Formula D-28, D-29
|
|
|
|
*coded_width = sps.pic_width_in_luma_samples - crop_x;
|
|
|
|
*coded_height = sps.pic_height_in_luma_samples - crop_y;
|
|
|
|
GetAspectRatioInfo(sps, pixel_width, pixel_height);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2016-02-22 23:32:22 +00:00
|
|
|
H265Pps::H265Pps() {}
|
|
|
|
H265Pps::~H265Pps() {}
|
|
|
|
|
|
|
|
H265Sps::H265Sps() {}
|
|
|
|
H265Sps::~H265Sps() {}
|
|
|
|
|
|
|
|
int H265Sps::GetPicSizeInCtbsY() const {
|
|
|
|
int min_cb_log2_size_y = log2_min_luma_coding_block_size_minus3 + 3;
|
|
|
|
int ctb_log2_size_y =
|
|
|
|
min_cb_log2_size_y + log2_diff_max_min_luma_coding_block_size;
|
|
|
|
int ctb_size_y = 1 << ctb_log2_size_y;
|
|
|
|
|
|
|
|
// Round-up division.
|
|
|
|
int pic_width_in_ctbs_y = (pic_width_in_luma_samples - 1) / ctb_size_y + 1;
|
|
|
|
int pic_height_in_ctbs_y = (pic_height_in_luma_samples - 1) / ctb_size_y + 1;
|
|
|
|
return pic_width_in_ctbs_y * pic_height_in_ctbs_y;
|
|
|
|
}
|
|
|
|
|
|
|
|
int H265Sps::GetChromaArrayType() const {
|
|
|
|
if (!separate_colour_plane_flag)
|
|
|
|
return chroma_format_idc;
|
|
|
|
else
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265ReferencePictureListModifications::H265ReferencePictureListModifications() {
|
|
|
|
}
|
|
|
|
H265ReferencePictureListModifications::
|
|
|
|
~H265ReferencePictureListModifications() {}
|
|
|
|
|
|
|
|
H265SliceHeader::H265SliceHeader() {}
|
|
|
|
H265SliceHeader::~H265SliceHeader() {}
|
|
|
|
|
|
|
|
H265Parser::H265Parser() {}
|
2016-08-30 23:01:19 +00:00
|
|
|
H265Parser::~H265Parser() {}
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
H265Parser::Result H265Parser::ParseSliceHeader(const Nalu& nalu,
|
|
|
|
H265SliceHeader* slice_header) {
|
|
|
|
DCHECK(nalu.is_video_slice());
|
|
|
|
*slice_header = H265SliceHeader();
|
|
|
|
|
|
|
|
// Parses whole element.
|
|
|
|
H26xBitReader reader;
|
|
|
|
reader.Initialize(nalu.data() + nalu.header_size(), nalu.payload_size());
|
|
|
|
H26xBitReader* br = &reader;
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->first_slice_segment_in_pic_flag));
|
|
|
|
if (nalu.type() >= Nalu::H265_BLA_W_LP &&
|
|
|
|
nalu.type() <= Nalu::H265_RSV_IRAP_VCL23) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->no_output_of_prior_pics_flag));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->pic_parameter_set_id));
|
|
|
|
const H265Pps* pps = GetPps(slice_header->pic_parameter_set_id);
|
|
|
|
TRUE_OR_RETURN(pps);
|
|
|
|
|
|
|
|
const H265Sps* sps = GetSps(pps->seq_parameter_set_id);
|
|
|
|
TRUE_OR_RETURN(sps);
|
|
|
|
|
|
|
|
if (!slice_header->first_slice_segment_in_pic_flag) {
|
|
|
|
if (pps->dependent_slice_segments_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->dependent_slice_segment_flag));
|
|
|
|
}
|
|
|
|
const int bit_length = ceil(log2(sps->GetPicSizeInCtbsY()));
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(bit_length, &slice_header->segment_address));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!slice_header->dependent_slice_segment_flag) {
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(pps->num_extra_slice_header_bits));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->slice_type));
|
|
|
|
if (pps->output_flag_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->pic_output_flag));
|
|
|
|
}
|
|
|
|
if (sps->separate_colour_plane_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(2, &slice_header->colour_plane_id));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (nalu.type() != Nalu::H265_IDR_W_RADL &&
|
|
|
|
nalu.type() != Nalu::H265_IDR_N_LP) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(sps->log2_max_pic_order_cnt_lsb_minus4 + 4,
|
|
|
|
&slice_header->slice_pic_order_cnt_lsb));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBool(&slice_header->short_term_ref_pic_set_sps_flag));
|
|
|
|
if (!slice_header->short_term_ref_pic_set_sps_flag) {
|
|
|
|
OK_OR_RETURN(ParseReferencePictureSet(
|
|
|
|
sps->num_short_term_ref_pic_sets, sps->num_short_term_ref_pic_sets,
|
|
|
|
sps->st_ref_pic_sets, br, &slice_header->st_ref_pic_set));
|
|
|
|
} else if (sps->num_short_term_ref_pic_sets > 1) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBits(ceil(log2(sps->num_short_term_ref_pic_sets)),
|
|
|
|
&slice_header->short_term_ref_pic_set_idx));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (sps->long_term_ref_pic_present_flag) {
|
|
|
|
if (sps->num_long_term_ref_pics > 0) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->num_long_term_sps));
|
|
|
|
}
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->num_long_term_pics));
|
|
|
|
|
|
|
|
const int pic_count =
|
|
|
|
slice_header->num_long_term_sps + slice_header->num_long_term_pics;
|
|
|
|
slice_header->long_term_pics_info.resize(pic_count);
|
|
|
|
for (int i = 0; i < pic_count; i++) {
|
|
|
|
if (i < slice_header->num_long_term_sps) {
|
|
|
|
int lt_idx_sps = 0;
|
|
|
|
if (sps->num_long_term_ref_pics > 1) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(
|
|
|
|
ceil(log2(sps->num_long_term_ref_pics)), <_idx_sps));
|
|
|
|
}
|
|
|
|
if (sps->used_by_curr_pic_lt_flag[lt_idx_sps])
|
|
|
|
slice_header->used_by_curr_pic_lt++;
|
|
|
|
} else {
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(sps->log2_max_pic_order_cnt_lsb_minus4 +
|
|
|
|
4)); // poc_lsb_lt
|
|
|
|
bool used_by_curr_pic_lt_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&used_by_curr_pic_lt_flag));
|
|
|
|
if (used_by_curr_pic_lt_flag)
|
|
|
|
slice_header->used_by_curr_pic_lt++;
|
|
|
|
}
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->long_term_pics_info[i]
|
|
|
|
.delta_poc_msb_present_flag));
|
|
|
|
if (slice_header->long_term_pics_info[i].delta_poc_msb_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(
|
|
|
|
&slice_header->long_term_pics_info[i].delta_poc_msb_cycle_lt));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (sps->temporal_mvp_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBool(&slice_header->slice_temporal_mvp_enabled_flag));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (nalu.nuh_layer_id() != 0) {
|
|
|
|
NOTIMPLEMENTED() << "Multi-layer streams are not supported.";
|
|
|
|
return kUnsupportedStream;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (sps->sample_adaptive_offset_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->slice_sao_luma_flag));
|
|
|
|
if (sps->GetChromaArrayType() != 0) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->slice_sao_chroma_flag));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
slice_header->num_ref_idx_l0_active_minus1 =
|
|
|
|
pps->num_ref_idx_l0_default_active_minus1;
|
|
|
|
slice_header->num_ref_idx_l1_active_minus1 =
|
|
|
|
pps->num_ref_idx_l1_default_active_minus1;
|
|
|
|
if (slice_header->slice_type == kPSlice ||
|
|
|
|
slice_header->slice_type == kBSlice) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBool(&slice_header->num_ref_idx_active_override_flag));
|
|
|
|
if (slice_header->num_ref_idx_active_override_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->num_ref_idx_l0_active_minus1));
|
|
|
|
if (slice_header->slice_type == kBSlice) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadUE(&slice_header->num_ref_idx_l1_active_minus1));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
const int num_pic_total_curr = GetNumPicTotalCurr(*slice_header, *sps);
|
|
|
|
if (pps->lists_modification_present_flag && num_pic_total_curr > 1) {
|
|
|
|
OK_OR_RETURN(SkipReferencePictureListModification(
|
|
|
|
*slice_header, *pps, num_pic_total_curr, br));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (slice_header->slice_type == kBSlice) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->mvd_l1_zero_flag));
|
|
|
|
}
|
|
|
|
if (pps->cabac_init_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->cabac_init_flag));
|
|
|
|
}
|
|
|
|
if (slice_header->slice_temporal_mvp_enabled_flag) {
|
|
|
|
if (slice_header->slice_type == kBSlice) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&slice_header->collocated_from_l0));
|
|
|
|
}
|
|
|
|
bool l0_greater_than_0 = slice_header->num_ref_idx_l0_active_minus1 > 0;
|
|
|
|
bool l1_greater_than_0 = slice_header->num_ref_idx_l1_active_minus1 > 0;
|
|
|
|
if (slice_header->collocated_from_l0 ? l0_greater_than_0
|
|
|
|
: l1_greater_than_0) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->collocated_ref_idx));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if ((pps->weighted_pred_flag && slice_header->slice_type == kPSlice) ||
|
|
|
|
(pps->weighted_bipred_flag && slice_header->slice_type == kBSlice)) {
|
|
|
|
OK_OR_RETURN(SkipPredictionWeightTable(
|
|
|
|
slice_header->slice_type == kBSlice, *sps, *slice_header, br));
|
|
|
|
}
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->five_minus_max_num_merge_cand));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_qp_delta));
|
|
|
|
if (pps->slice_chroma_qp_offsets_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_cb_qp_offset));
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_cr_qp_offset));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (pps->chroma_qp_offset_list_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBool(&slice_header->cu_chroma_qp_offset_enabled_flag));
|
|
|
|
}
|
|
|
|
if (pps->deblocking_filter_override_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBool(&slice_header->deblocking_filter_override_flag));
|
|
|
|
}
|
|
|
|
if (slice_header->deblocking_filter_override_flag) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBool(&slice_header->slice_deblocking_filter_disabled_flag));
|
|
|
|
if (!slice_header->slice_deblocking_filter_disabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_beta_offset_div2));
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_tc_offset_div2));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (pps->loop_filter_across_slices_enabled_flag &&
|
|
|
|
(slice_header->slice_sao_luma_flag ||
|
|
|
|
slice_header->slice_sao_chroma_flag ||
|
|
|
|
!slice_header->slice_deblocking_filter_disabled_flag)) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(
|
|
|
|
&slice_header->slice_loop_filter_across_slices_enabled_flag));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (pps->tiles_enabled_flag || pps->entropy_coding_sync_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->num_entry_point_offsets));
|
|
|
|
if (slice_header->num_entry_point_offsets > 0) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&slice_header->offset_len_minus1));
|
|
|
|
slice_header->entry_point_offset_minus1.resize(
|
|
|
|
slice_header->num_entry_point_offsets);
|
|
|
|
for (int i = 0; i < slice_header->num_entry_point_offsets; i++) {
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBits(slice_header->offset_len_minus1 + 1,
|
|
|
|
&slice_header->entry_point_offset_minus1[i]));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (pps->slice_segment_header_extension_present_flag) {
|
|
|
|
int extension_length;
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&extension_length));
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(extension_length * 8));
|
|
|
|
}
|
|
|
|
|
2016-12-06 20:23:37 +00:00
|
|
|
slice_header->header_bit_size = nalu.payload_size() * 8 - br->NumBitsLeft();
|
2016-02-22 23:32:22 +00:00
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265Parser::Result H265Parser::ParsePps(const Nalu& nalu, int* pps_id) {
|
|
|
|
DCHECK_EQ(Nalu::H265_PPS, nalu.type());
|
|
|
|
|
|
|
|
// Reads most of the element, not reading the extension data.
|
|
|
|
H26xBitReader reader;
|
|
|
|
reader.Initialize(nalu.data() + nalu.header_size(), nalu.payload_size());
|
|
|
|
H26xBitReader* br = &reader;
|
|
|
|
|
|
|
|
*pps_id = -1;
|
2016-08-17 17:41:40 +00:00
|
|
|
std::unique_ptr<H265Pps> pps(new H265Pps);
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->pic_parameter_set_id));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->seq_parameter_set_id));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->dependent_slice_segments_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->output_flag_present_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(3, &pps->num_extra_slice_header_bits));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->sign_data_hiding_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->cabac_init_present_flag));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->num_ref_idx_l0_default_active_minus1));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->num_ref_idx_l1_default_active_minus1));
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&pps->init_qp_minus26));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->constrained_intra_pred_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->transform_skip_enabled_flag));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->cu_qp_delta_enabled_flag));
|
|
|
|
if (pps->cu_qp_delta_enabled_flag)
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->diff_cu_qp_delta_depth));
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&pps->cb_qp_offset));
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&pps->cr_qp_offset));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->slice_chroma_qp_offsets_present_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->weighted_pred_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->weighted_bipred_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->transquant_bypass_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->tiles_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->entropy_coding_sync_enabled_flag));
|
|
|
|
|
|
|
|
if (pps->tiles_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->num_tile_columns_minus1));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->num_tile_rows_minus1));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->uniform_spacing_flag));
|
|
|
|
if (!pps->uniform_spacing_flag) {
|
|
|
|
pps->column_width_minus1.resize(pps->num_tile_columns_minus1);
|
|
|
|
for (int i = 0; i < pps->num_tile_columns_minus1; i++) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->column_width_minus1[i]));
|
|
|
|
}
|
|
|
|
pps->row_height_minus1.resize(pps->num_tile_rows_minus1);
|
|
|
|
for (int i = 0; i < pps->num_tile_rows_minus1; i++) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->row_height_minus1[i]));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->loop_filter_across_tiles_enabled_flag));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->loop_filter_across_slices_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->deblocking_filter_control_present_flag));
|
|
|
|
if (pps->deblocking_filter_control_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->deblocking_filter_override_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->deblocking_filter_disabled_flag));
|
|
|
|
if (!pps->deblocking_filter_disabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&pps->beta_offset_div2));
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&pps->tc_offset_div2));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->scaling_list_data_present_flag));
|
|
|
|
if (pps->scaling_list_data_present_flag) {
|
|
|
|
OK_OR_RETURN(SkipScalingListData(br));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->lists_modification_present_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&pps->log2_parallel_merge_level_minus2));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadBool(&pps->slice_segment_header_extension_present_flag));
|
|
|
|
|
|
|
|
bool pps_extension_present_flag;
|
|
|
|
bool pps_range_extension_flag = false;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps_extension_present_flag));
|
|
|
|
if (pps_extension_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps_range_extension_flag));
|
|
|
|
// pps_multilayer_extension_flag, pps_3d_extension_flag, pps_extension_5bits
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(1 + 1 + 5));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (pps_range_extension_flag) {
|
|
|
|
if (pps->transform_skip_enabled_flag) {
|
|
|
|
// log2_max_transform_skip_block_size_minus2
|
|
|
|
int ignored;
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(1)); // cross_component_prediction_enabled_flag
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&pps->chroma_qp_offset_list_enabled_flag));
|
|
|
|
// Incomplete
|
|
|
|
}
|
|
|
|
|
|
|
|
// Ignore remaining extension data.
|
|
|
|
|
2016-03-29 22:39:53 +00:00
|
|
|
// This will replace any existing PPS instance.
|
2016-02-22 23:32:22 +00:00
|
|
|
*pps_id = pps->pic_parameter_set_id;
|
2016-08-30 23:01:19 +00:00
|
|
|
active_ppses_[*pps_id] = std::move(pps);
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265Parser::Result H265Parser::ParseSps(const Nalu& nalu, int* sps_id) {
|
|
|
|
DCHECK_EQ(Nalu::H265_SPS, nalu.type());
|
|
|
|
|
|
|
|
// Reads most of the element, not reading the extension data.
|
|
|
|
H26xBitReader reader;
|
|
|
|
reader.Initialize(nalu.data() + nalu.header_size(), nalu.payload_size());
|
|
|
|
H26xBitReader* br = &reader;
|
|
|
|
|
|
|
|
*sps_id = -1;
|
|
|
|
|
2016-08-17 17:41:40 +00:00
|
|
|
std::unique_ptr<H265Sps> sps(new H265Sps);
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(4, &sps->video_parameter_set_id));
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(3, &sps->max_sub_layers_minus1));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->temporal_id_nesting_flag));
|
|
|
|
|
2018-01-06 04:57:51 +00:00
|
|
|
OK_OR_RETURN(
|
|
|
|
ReadProfileTierLevel(true, sps->max_sub_layers_minus1, br, sps.get()));
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->seq_parameter_set_id));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->chroma_format_idc));
|
|
|
|
if (sps->chroma_format_idc == 3) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->separate_colour_plane_flag));
|
|
|
|
}
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->pic_width_in_luma_samples));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->pic_height_in_luma_samples));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->conformance_window_flag));
|
|
|
|
if (sps->conformance_window_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_left_offset));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_right_offset));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_top_offset));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_bottom_offset));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->bit_depth_luma_minus8));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->bit_depth_chroma_minus8));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->log2_max_pic_order_cnt_lsb_minus4));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->sub_layer_ordering_info_present_flag));
|
|
|
|
int start = sps->sub_layer_ordering_info_present_flag
|
|
|
|
? 0
|
|
|
|
: sps->max_sub_layers_minus1;
|
|
|
|
for (int i = start; i <= sps->max_sub_layers_minus1; i++) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->max_dec_pic_buffering_minus1[i]));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->max_num_reorder_pics[i]));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->max_latency_increase_plus1[i]));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->log2_min_luma_coding_block_size_minus3));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->log2_diff_max_min_luma_coding_block_size));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->log2_min_luma_transform_block_size_minus2));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->log2_diff_max_min_luma_transform_block_size));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->max_transform_hierarchy_depth_inter));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->max_transform_hierarchy_depth_intra));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->scaling_list_enabled_flag));
|
|
|
|
if (sps->scaling_list_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->scaling_list_data_present_flag));
|
|
|
|
if (sps->scaling_list_data_present_flag) {
|
|
|
|
OK_OR_RETURN(SkipScalingListData(br));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->amp_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->sample_adaptive_offset_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->pcm_enabled_flag));
|
|
|
|
if (sps->pcm_enabled_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(4, &sps->pcm_sample_bit_depth_luma_minus1));
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(4, &sps->pcm_sample_bit_depth_chroma_minus1));
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadUE(&sps->log2_min_pcm_luma_coding_block_size_minus3));
|
|
|
|
TRUE_OR_RETURN(
|
|
|
|
br->ReadUE(&sps->log2_diff_max_min_pcm_luma_coding_block_size));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->pcm_loop_filter_disabled_flag));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->num_short_term_ref_pic_sets));
|
|
|
|
sps->st_ref_pic_sets.resize(sps->num_short_term_ref_pic_sets);
|
|
|
|
for (int i = 0; i < sps->num_short_term_ref_pic_sets; i++) {
|
|
|
|
OK_OR_RETURN(ParseReferencePictureSet(sps->num_short_term_ref_pic_sets, i,
|
|
|
|
sps->st_ref_pic_sets, br,
|
|
|
|
&sps->st_ref_pic_sets[i]));
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->long_term_ref_pic_present_flag));
|
|
|
|
if (sps->long_term_ref_pic_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&sps->num_long_term_ref_pics));
|
|
|
|
sps->lt_ref_pic_poc_lsb.resize(sps->num_long_term_ref_pics);
|
|
|
|
sps->used_by_curr_pic_lt_flag.resize(sps->num_long_term_ref_pics);
|
|
|
|
for (int i = 0; i < sps->num_long_term_ref_pics; i++) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(sps->log2_max_pic_order_cnt_lsb_minus4 + 4,
|
|
|
|
&sps->lt_ref_pic_poc_lsb[i]));
|
|
|
|
bool temp;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&temp));
|
|
|
|
sps->used_by_curr_pic_lt_flag[i] = temp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->temporal_mvp_enabled_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->strong_intra_smoothing_enabled_flag));
|
|
|
|
|
2016-04-13 21:46:13 +00:00
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sps->vui_parameters_present));
|
|
|
|
if (sps->vui_parameters_present) {
|
|
|
|
OK_OR_RETURN(ParseVuiParameters(sps->max_sub_layers_minus1, br,
|
|
|
|
&sps->vui_parameters));
|
|
|
|
}
|
|
|
|
|
2016-02-22 23:32:22 +00:00
|
|
|
// Ignore remaining extension data.
|
|
|
|
|
2016-03-29 22:39:53 +00:00
|
|
|
// This will replace any existing SPS instance.
|
2016-02-22 23:32:22 +00:00
|
|
|
*sps_id = sps->seq_parameter_set_id;
|
2016-08-30 23:01:19 +00:00
|
|
|
active_spses_[*sps_id] = std::move(sps);
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
const H265Pps* H265Parser::GetPps(int pps_id) {
|
2016-08-30 23:01:19 +00:00
|
|
|
return active_ppses_[pps_id].get();
|
2016-02-22 23:32:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
const H265Sps* H265Parser::GetSps(int sps_id) {
|
2016-08-30 23:01:19 +00:00
|
|
|
return active_spses_[sps_id].get();
|
2016-02-22 23:32:22 +00:00
|
|
|
}
|
|
|
|
|
2016-04-13 21:46:13 +00:00
|
|
|
H265Parser::Result H265Parser::ParseVuiParameters(int max_num_sub_layers_minus1,
|
|
|
|
H26xBitReader* br,
|
|
|
|
H265VuiParameters* vui) {
|
|
|
|
// Reads whole element but ignores most of it.
|
|
|
|
int ignored;
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&vui->aspect_ratio_info_present_flag));
|
|
|
|
if (vui->aspect_ratio_info_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(8, &vui->aspect_ratio_idc));
|
|
|
|
if (vui->aspect_ratio_idc == H265VuiParameters::kExtendedSar) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(16, &vui->sar_width));
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(16, &vui->sar_height));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool overscan_info_present_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&overscan_info_present_flag));
|
|
|
|
if (overscan_info_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(1)); // overscan_appropriate_flag
|
|
|
|
}
|
|
|
|
|
|
|
|
bool video_signal_type_present_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&video_signal_type_present_flag));
|
|
|
|
if (video_signal_type_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(3)); // video_format
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(1)); // video_full_range_flag
|
|
|
|
|
|
|
|
bool colour_description_present_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&colour_description_present_flag));
|
|
|
|
if (colour_description_present_flag) {
|
|
|
|
// colour_primaries, transfer_characteristics, matrix_coeffs
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(8 + 8 + 8));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool chroma_loc_info_present_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&chroma_loc_info_present_flag));
|
|
|
|
if (chroma_loc_info_present_flag) {
|
|
|
|
// chroma_sample_log_type_top_field, chroma_sample_log_type_bottom_field
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored));
|
|
|
|
}
|
|
|
|
|
|
|
|
// neutral_chroma_indication_flag, field_seq_flag,
|
|
|
|
// frame_field_info_present_flag.
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(3));
|
|
|
|
|
|
|
|
bool default_display_window_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&default_display_window_flag));
|
|
|
|
if (default_display_window_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_left_offset
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_right_offset
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_top_offset
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_bottom_offset
|
|
|
|
}
|
|
|
|
|
|
|
|
bool vui_timing_info_present_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&vui_timing_info_present_flag));
|
|
|
|
if (vui_timing_info_present_flag) {
|
|
|
|
// vui_num_units_in_tick, vui_time_scale
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(32 + 32));
|
|
|
|
|
|
|
|
bool vui_poc_proportional_to_timing_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&vui_poc_proportional_to_timing_flag));
|
|
|
|
if (vui_poc_proportional_to_timing_flag) {
|
|
|
|
// vui_num_ticks_poc_diff_one_minus1
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored));
|
|
|
|
}
|
|
|
|
|
|
|
|
bool vui_hdr_parameters_present_flag;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&vui_hdr_parameters_present_flag));
|
|
|
|
if (vui_hdr_parameters_present_flag) {
|
|
|
|
OK_OR_RETURN(SkipHrdParameters(max_num_sub_layers_minus1, br));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&vui->bitstream_restriction_flag));
|
|
|
|
if (vui->bitstream_restriction_flag) {
|
|
|
|
// tiles_fixed_structure_flag, motion_vectors_over_pic_boundaries_flag,
|
|
|
|
// restricted_ref_pic_lists_flag.
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(3));
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&vui->min_spatial_segmentation_idc));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // max_bytes_per_pic_denom
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // max_bits_per_min_cu_denum
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // log2_max_mv_length_horizontal
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // log2_max_mv_length_vertical
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
2016-02-22 23:32:22 +00:00
|
|
|
H265Parser::Result H265Parser::ParseReferencePictureSet(
|
|
|
|
int num_short_term_ref_pic_sets,
|
|
|
|
int st_rps_idx,
|
|
|
|
const std::vector<H265ReferencePictureSet>& ref_pic_sets,
|
|
|
|
H26xBitReader* br,
|
|
|
|
H265ReferencePictureSet* out_ref_pic_set) {
|
|
|
|
// Parses and processess a short-term reference picture set. This needs to
|
|
|
|
// be done since the size of this element may be dependent on previous
|
|
|
|
// reference picture sets.
|
|
|
|
|
|
|
|
bool inter_ref_pic_set_prediction = false;
|
|
|
|
if (st_rps_idx != 0) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&inter_ref_pic_set_prediction));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (inter_ref_pic_set_prediction) {
|
|
|
|
int delta_idx = 1;
|
|
|
|
if (st_rps_idx == num_short_term_ref_pic_sets) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&delta_idx));
|
|
|
|
delta_idx++;
|
|
|
|
TRUE_OR_RETURN(delta_idx <= st_rps_idx);
|
|
|
|
}
|
|
|
|
|
|
|
|
int ref_rps_idx = st_rps_idx - delta_idx;
|
|
|
|
DCHECK_LE(0, ref_rps_idx);
|
|
|
|
DCHECK_LT(ref_rps_idx, st_rps_idx);
|
|
|
|
|
|
|
|
bool delta_rps_sign;
|
|
|
|
int abs_delta_rps_minus1;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&delta_rps_sign));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&abs_delta_rps_minus1));
|
|
|
|
int delta_rps =
|
|
|
|
delta_rps_sign ? -(abs_delta_rps_minus1 + 1) : abs_delta_rps_minus1 + 1;
|
|
|
|
|
|
|
|
int ref_num_delta_pocs = ref_pic_sets[ref_rps_idx].num_delta_pocs;
|
|
|
|
std::vector<bool> used_by_curr_pic(ref_num_delta_pocs + 1);
|
|
|
|
std::vector<bool> use_delta(ref_num_delta_pocs + 1);
|
|
|
|
for (int j = 0; j <= ref_num_delta_pocs; j++) {
|
|
|
|
bool temp;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&temp));
|
|
|
|
used_by_curr_pic[j] = temp;
|
|
|
|
|
|
|
|
if (!used_by_curr_pic[j]) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&temp));
|
|
|
|
use_delta[j] = temp;
|
|
|
|
} else {
|
|
|
|
use_delta[j] = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
int ref_num_positive_pics = ref_pic_sets[ref_rps_idx].num_positive_pics;
|
|
|
|
int ref_num_negative_pics = ref_pic_sets[ref_rps_idx].num_negative_pics;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
// Update list 0.
|
|
|
|
{
|
|
|
|
i = 0;
|
|
|
|
for (int j = ref_num_positive_pics - 1; j >= 0; j--) {
|
|
|
|
int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s1[j] + delta_rps;
|
|
|
|
if (d_poc < 0 && use_delta[ref_num_negative_pics + j]) {
|
|
|
|
out_ref_pic_set->delta_poc_s0[i] = d_poc;
|
|
|
|
out_ref_pic_set->used_by_curr_pic_s0[i] =
|
|
|
|
used_by_curr_pic[ref_num_negative_pics + j];
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (delta_rps < 0 && use_delta[ref_num_delta_pocs]) {
|
|
|
|
out_ref_pic_set->delta_poc_s0[i] = delta_rps;
|
|
|
|
out_ref_pic_set->used_by_curr_pic_s0[i] =
|
|
|
|
used_by_curr_pic[ref_num_delta_pocs];
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
for (int j = 0; j < ref_num_negative_pics; j++) {
|
|
|
|
int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s0[j] + delta_rps;
|
|
|
|
if (d_poc < 0 && use_delta[j]) {
|
|
|
|
out_ref_pic_set->delta_poc_s0[i] = d_poc;
|
|
|
|
out_ref_pic_set->used_by_curr_pic_s0[i] = used_by_curr_pic[j];
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
out_ref_pic_set->num_negative_pics = i;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Update list 1.
|
|
|
|
{
|
|
|
|
i = 0;
|
|
|
|
for (int j = ref_num_negative_pics - 1; j >= 0; j--) {
|
|
|
|
int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s0[j] + delta_rps;
|
|
|
|
if (d_poc > 0 && use_delta[j]) {
|
|
|
|
out_ref_pic_set->delta_poc_s1[i] = d_poc;
|
|
|
|
out_ref_pic_set->used_by_curr_pic_s1[i] = used_by_curr_pic[j];
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (delta_rps > 0 && use_delta[ref_num_delta_pocs]) {
|
|
|
|
out_ref_pic_set->delta_poc_s1[i] = delta_rps;
|
|
|
|
out_ref_pic_set->used_by_curr_pic_s1[i] =
|
|
|
|
used_by_curr_pic[ref_num_delta_pocs];
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
for (int j = 0; j < ref_num_positive_pics; j++) {
|
|
|
|
int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s1[j] + delta_rps;
|
|
|
|
if (d_poc > 0 && use_delta[ref_num_negative_pics + j]) {
|
|
|
|
out_ref_pic_set->delta_poc_s1[i] = d_poc;
|
|
|
|
out_ref_pic_set->used_by_curr_pic_s1[i] =
|
|
|
|
used_by_curr_pic[ref_num_negative_pics + j];
|
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
out_ref_pic_set->num_positive_pics = i;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&out_ref_pic_set->num_negative_pics));
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&out_ref_pic_set->num_positive_pics));
|
|
|
|
|
|
|
|
int prev_poc = 0;
|
|
|
|
for (int i = 0; i < out_ref_pic_set->num_negative_pics; i++) {
|
|
|
|
int delta_poc_s0_minus1;
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&delta_poc_s0_minus1));
|
|
|
|
out_ref_pic_set->delta_poc_s0[i] = prev_poc - (delta_poc_s0_minus1 + 1);
|
|
|
|
prev_poc = out_ref_pic_set->delta_poc_s0[i];
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&out_ref_pic_set->used_by_curr_pic_s0[i]));
|
|
|
|
}
|
|
|
|
|
|
|
|
prev_poc = 0;
|
|
|
|
for (int i = 0; i < out_ref_pic_set->num_positive_pics; i++) {
|
|
|
|
int delta_poc_s1_minus1;
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&delta_poc_s1_minus1));
|
|
|
|
out_ref_pic_set->delta_poc_s1[i] = prev_poc + delta_poc_s1_minus1 + 1;
|
|
|
|
prev_poc = out_ref_pic_set->delta_poc_s1[i];
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&out_ref_pic_set->used_by_curr_pic_s1[i]));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
out_ref_pic_set->num_delta_pocs =
|
|
|
|
out_ref_pic_set->num_positive_pics + out_ref_pic_set->num_negative_pics;
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265Parser::Result H265Parser::SkipReferencePictureListModification(
|
|
|
|
const H265SliceHeader& slice_header,
|
|
|
|
const H265Pps& pps,
|
|
|
|
int num_pic_total_curr,
|
|
|
|
H26xBitReader* br) {
|
|
|
|
// Reads whole element but ignores it all.
|
|
|
|
|
|
|
|
bool ref_pic_list_modification_flag_l0;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&ref_pic_list_modification_flag_l0));
|
|
|
|
if (ref_pic_list_modification_flag_l0) {
|
|
|
|
for (int i = 0; i <= pps.num_ref_idx_l0_default_active_minus1; i++) {
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(ceil(log2(num_pic_total_curr))));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (slice_header.slice_type == kBSlice) {
|
|
|
|
bool ref_pic_list_modification_flag_l1;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&ref_pic_list_modification_flag_l1));
|
|
|
|
if (ref_pic_list_modification_flag_l1) {
|
|
|
|
for (int i = 0; i <= pps.num_ref_idx_l1_default_active_minus1; i++) {
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(ceil(log2(num_pic_total_curr))));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265Parser::Result H265Parser::SkipPredictionWeightTablePart(
|
|
|
|
int num_ref_idx_minus1,
|
|
|
|
int chroma_array_type,
|
|
|
|
H26xBitReader* br) {
|
|
|
|
// Reads whole element, ignores it.
|
|
|
|
int ignored;
|
|
|
|
std::vector<bool> luma_weight_flag(num_ref_idx_minus1 + 1);
|
|
|
|
std::vector<bool> chroma_weight_flag(num_ref_idx_minus1 + 1);
|
|
|
|
|
|
|
|
for (int i = 0; i <= num_ref_idx_minus1; i++) {
|
|
|
|
bool temp;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&temp));
|
|
|
|
luma_weight_flag[i] = temp;
|
|
|
|
}
|
|
|
|
if (chroma_array_type != 0) {
|
|
|
|
for (int i = 0; i <= num_ref_idx_minus1; i++) {
|
|
|
|
bool temp;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&temp));
|
|
|
|
chroma_weight_flag[i] = temp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for (int i = 0; i <= num_ref_idx_minus1; i++) {
|
|
|
|
if (luma_weight_flag[i]) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_luma_weight_l#
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&ignored)); // luma_offset_l#
|
|
|
|
}
|
|
|
|
if (chroma_weight_flag[i]) {
|
|
|
|
for (int j = 0; j < 2; j++) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_chroma_weight_l#
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_chroma_offset_l#
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265Parser::Result H265Parser::SkipPredictionWeightTable(
|
|
|
|
bool is_b_slice,
|
|
|
|
const H265Sps& sps,
|
|
|
|
const H265SliceHeader& slice_header,
|
|
|
|
H26xBitReader* br) {
|
|
|
|
// Reads whole element, ignores it.
|
|
|
|
int ignored;
|
|
|
|
int chroma_array_type = sps.GetChromaArrayType();
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // luma_log2_weight_denom
|
|
|
|
if (chroma_array_type != 0) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_chroma_log2_weight_denom
|
|
|
|
}
|
|
|
|
OK_OR_RETURN(SkipPredictionWeightTablePart(
|
|
|
|
slice_header.num_ref_idx_l0_active_minus1, chroma_array_type, br));
|
|
|
|
if (is_b_slice) {
|
|
|
|
OK_OR_RETURN(SkipPredictionWeightTablePart(
|
|
|
|
slice_header.num_ref_idx_l1_active_minus1, chroma_array_type, br));
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
2018-01-06 04:57:51 +00:00
|
|
|
H265Parser::Result H265Parser::ReadProfileTierLevel(
|
2016-02-22 23:32:22 +00:00
|
|
|
bool profile_present,
|
|
|
|
int max_num_sub_layers_minus1,
|
2018-01-06 04:57:51 +00:00
|
|
|
H26xBitReader* br,
|
|
|
|
H265Sps* sps) {
|
2016-02-22 23:32:22 +00:00
|
|
|
// Reads whole element, ignores it.
|
|
|
|
|
|
|
|
if (profile_present) {
|
2018-01-06 04:57:51 +00:00
|
|
|
// 11 bytes of general_profile_tier flags:
|
|
|
|
// general_profile_space, general_tier_flag, general_profile_idc
|
|
|
|
// general_profile_compativility_flag
|
|
|
|
// general_progressive_source_flag
|
|
|
|
// general_interlaced_source_flag
|
|
|
|
// general_non_packed_constraint_flag
|
|
|
|
// general_frame_only_constraint_flag
|
|
|
|
// 44-bits of other flags
|
|
|
|
for (int i = 0; i < 11; i++)
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(8, &sps->general_profile_tier_level_data[i]));
|
|
|
|
}
|
|
|
|
// general_level_idc
|
|
|
|
TRUE_OR_RETURN(br->ReadBits(8, &sps->general_profile_tier_level_data[11]));
|
2016-02-22 23:32:22 +00:00
|
|
|
|
|
|
|
std::vector<bool> sub_layer_profile_present(max_num_sub_layers_minus1);
|
|
|
|
std::vector<bool> sub_layer_level_present(max_num_sub_layers_minus1);
|
|
|
|
for (int i = 0; i < max_num_sub_layers_minus1; i++) {
|
|
|
|
bool profile, level;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&profile));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&level));
|
|
|
|
sub_layer_profile_present[i] = profile;
|
|
|
|
sub_layer_level_present[i] = level;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (max_num_sub_layers_minus1 > 0) {
|
|
|
|
for (int i = max_num_sub_layers_minus1; i < 8; i++)
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(2)); // reserved_zero_2bits
|
|
|
|
}
|
|
|
|
|
|
|
|
for (int i = 0; i < max_num_sub_layers_minus1; i++) {
|
|
|
|
if (sub_layer_profile_present[i]) {
|
|
|
|
// sub_layer_profile_space, sub_layer_tier_flag, sub_layer_profile_idc
|
|
|
|
// sub_layer_profile_compatibility
|
|
|
|
// sub_layer_reserved_zero_43bits
|
|
|
|
// sub_layer_reserved_zero_bit
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(2 + 1 + 5 + 32 + 4 + 43 + 1));
|
|
|
|
}
|
|
|
|
if (sub_layer_level_present[i]) {
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(8));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265Parser::Result H265Parser::SkipScalingListData(H26xBitReader* br) {
|
|
|
|
// Reads whole element, ignores it.
|
|
|
|
int ignored;
|
|
|
|
for (int size_id = 0; size_id < 4; size_id++) {
|
|
|
|
for (int matrix_id = 0; matrix_id < 6;
|
|
|
|
matrix_id += ((size_id == 3) ? 3 : 1)) {
|
|
|
|
bool scaling_list_pred_mode;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&scaling_list_pred_mode));
|
|
|
|
if (!scaling_list_pred_mode) {
|
|
|
|
// scaling_list_pred_matrix_id_delta
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored));
|
|
|
|
} else {
|
|
|
|
int coefNum = std::min(64, (1 << (4 + (size_id << 1))));
|
|
|
|
if (size_id > 1) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&ignored)); // scaling_list_dc_coef_minus8
|
|
|
|
}
|
|
|
|
|
|
|
|
for (int i = 0; i < coefNum; i++) {
|
|
|
|
TRUE_OR_RETURN(br->ReadSE(&ignored)); // scaling_list_delta_coef
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
2016-04-13 21:46:13 +00:00
|
|
|
H265Parser::Result H265Parser::SkipHrdParameters(int max_num_sub_layers_minus1,
|
|
|
|
H26xBitReader* br) {
|
|
|
|
// common_inf_present_flag is always 1 when parsing vui_parameters.
|
|
|
|
const bool common_inf_present_flag = true;
|
|
|
|
|
|
|
|
int ignored;
|
|
|
|
bool nal_hdr_parameters_present_flag;
|
|
|
|
bool vcl_hdr_parameters_present_flag;
|
|
|
|
bool sub_pic_hdr_params_present_flag = false;
|
|
|
|
if (common_inf_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&nal_hdr_parameters_present_flag));
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&vcl_hdr_parameters_present_flag));
|
|
|
|
if (nal_hdr_parameters_present_flag || vcl_hdr_parameters_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&sub_pic_hdr_params_present_flag));
|
|
|
|
if (sub_pic_hdr_params_present_flag) {
|
|
|
|
// tick_divisor_minus2, du_cpb_removal_delay_increment_length_minus1,
|
|
|
|
// sub_pic_cpb_params_in_pic_timing_sei_flag
|
|
|
|
// dpb_output_delay_du_length_minus1
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(8 + 5 + 1 + 5));
|
|
|
|
}
|
|
|
|
|
|
|
|
// bit_rate_scale, cpb_size_scale
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(4 + 4));
|
|
|
|
if (sub_pic_hdr_params_present_flag)
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(4)); // cpb_size_du_scale
|
|
|
|
|
|
|
|
// initial_cpb_removal_delay_length_minus1,
|
|
|
|
// au_cpb_removal_delay_length_minus1, dpb_output_delay_length_minus1
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(5 + 5 + 5));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (int i = 0; i <= max_num_sub_layers_minus1; i++) {
|
|
|
|
bool fixed_pic_rate_general_flag;
|
|
|
|
bool fixed_pic_rate_within_cvs_flag = true;
|
|
|
|
bool low_delay_hdr_flag = false;
|
|
|
|
int cpb_cnt_minus1 = 0;
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&fixed_pic_rate_general_flag));
|
|
|
|
if (!fixed_pic_rate_general_flag)
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&fixed_pic_rate_within_cvs_flag));
|
|
|
|
if (fixed_pic_rate_within_cvs_flag)
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // elemental_duration_ic_tc_minus1
|
|
|
|
else
|
|
|
|
TRUE_OR_RETURN(br->ReadBool(&low_delay_hdr_flag));
|
|
|
|
if (!low_delay_hdr_flag)
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&cpb_cnt_minus1));
|
|
|
|
|
|
|
|
if (nal_hdr_parameters_present_flag) {
|
|
|
|
OK_OR_RETURN(SkipSubLayerHrdParameters(
|
|
|
|
cpb_cnt_minus1, sub_pic_hdr_params_present_flag, br));
|
|
|
|
}
|
|
|
|
if (vcl_hdr_parameters_present_flag) {
|
|
|
|
OK_OR_RETURN(SkipSubLayerHrdParameters(
|
|
|
|
cpb_cnt_minus1, sub_pic_hdr_params_present_flag, br));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
|
|
|
H265Parser::Result H265Parser::SkipSubLayerHrdParameters(
|
|
|
|
int cpb_cnt_minus1,
|
|
|
|
bool sub_pic_hdr_params_present_flag,
|
|
|
|
H26xBitReader* br) {
|
|
|
|
int ignored;
|
|
|
|
for (int i = 0; i <= cpb_cnt_minus1; i++) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // bit_rate_value_minus1
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // cpb_size_value_minus1
|
|
|
|
if (sub_pic_hdr_params_present_flag) {
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // cpb_size_du_value_minus1
|
|
|
|
TRUE_OR_RETURN(br->ReadUE(&ignored)); // bit_rate_du_value_minus1
|
|
|
|
}
|
|
|
|
|
|
|
|
TRUE_OR_RETURN(br->SkipBits(1)); // cbr_flag
|
|
|
|
}
|
|
|
|
|
|
|
|
return kOk;
|
|
|
|
}
|
|
|
|
|
2016-02-22 23:32:22 +00:00
|
|
|
} // namespace media
|
2016-05-20 21:19:33 +00:00
|
|
|
} // namespace shaka
|