Shaka Packager SDK
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
h265_parser.cc
1 // Copyright 2016 Google Inc. All rights reserved.
2 //
3 // Use of this source code is governed by a BSD-style
4 // license that can be found in the LICENSE file or at
5 // https://developers.google.com/open-source/licenses/bsd
6 
7 #include "packager/media/codecs/h265_parser.h"
8 
9 #include <math.h>
10 #include <algorithm>
11 
12 #include "packager/base/logging.h"
13 #include "packager/media/base/macros.h"
14 #include "packager/media/codecs/nalu_reader.h"
15 
16 #define TRUE_OR_RETURN(a) \
17  do { \
18  if (!(a)) { \
19  DVLOG(1) << "Failure while processing " << #a; \
20  return kInvalidStream; \
21  } \
22  } while (0)
23 
24 #define OK_OR_RETURN(a) \
25  do { \
26  Result status = (a); \
27  if (status != kOk) \
28  return status; \
29  } while (false)
30 
31 namespace shaka {
32 namespace media {
33 
34 namespace {
35 int GetNumPicTotalCurr(const H265SliceHeader& slice_header,
36  const H265Sps& sps) {
37  int num_pic_total_curr = 0;
38  const H265ReferencePictureSet& ref_pic_set =
39  slice_header.short_term_ref_pic_set_sps_flag
40  ? sps.st_ref_pic_sets[slice_header.short_term_ref_pic_set_idx]
41  : slice_header.st_ref_pic_set;
42 
43  for (int i = 0; i < ref_pic_set.num_negative_pics; i++) {
44  if (ref_pic_set.used_by_curr_pic_s0[i])
45  num_pic_total_curr++;
46  }
47  for (int i = 0; i < ref_pic_set.num_positive_pics; i++) {
48  if (ref_pic_set.used_by_curr_pic_s1[i])
49  num_pic_total_curr++;
50  }
51 
52  return num_pic_total_curr + slice_header.used_by_curr_pic_lt;
53 }
54 
55 void GetAspectRatioInfo(const H265Sps& sps,
56  uint32_t* pixel_width,
57  uint32_t* pixel_height) {
58  // The default value is 0; so if this is not in the SPS, it will correctly
59  // assume unspecified.
60  int aspect_ratio_idc = sps.vui_parameters.aspect_ratio_idc;
61 
62  // Table E.1
63  switch (aspect_ratio_idc) {
64  case 1: *pixel_width = 1; *pixel_height = 1; break;
65  case 2: *pixel_width = 12; *pixel_height = 11; break;
66  case 3: *pixel_width = 10; *pixel_height = 11; break;
67  case 4: *pixel_width = 16; *pixel_height = 11; break;
68  case 5: *pixel_width = 40; *pixel_height = 33; break;
69  case 6: *pixel_width = 24; *pixel_height = 11; break;
70  case 7: *pixel_width = 20; *pixel_height = 11; break;
71  case 8: *pixel_width = 32; *pixel_height = 11; break;
72  case 9: *pixel_width = 80; *pixel_height = 33; break;
73  case 10: *pixel_width = 18; *pixel_height = 11; break;
74  case 11: *pixel_width = 15; *pixel_height = 11; break;
75  case 12: *pixel_width = 64; *pixel_height = 33; break;
76  case 13: *pixel_width = 160; *pixel_height = 99; break;
77  case 14: *pixel_width = 4; *pixel_height = 3; break;
78  case 15: *pixel_width = 3; *pixel_height = 2; break;
79  case 16: *pixel_width = 2; *pixel_height = 1; break;
80 
81  case 255:
82  *pixel_width = sps.vui_parameters.sar_width;
83  *pixel_height = sps.vui_parameters.sar_height;
84  break;
85 
86  default:
87  // Section E.3.1 specifies that other values should be interpreted as 0.
88  LOG(WARNING) << "Unknown aspect_ratio_idc " << aspect_ratio_idc;
89  FALLTHROUGH_INTENDED;
90  case 0:
91  // Unlike the spec, assume 1:1 if not specified.
92  *pixel_width = 1;
93  *pixel_height = 1;
94  break;
95  }
96 }
97 } // namespace
98 
99 bool ExtractResolutionFromSps(const H265Sps& sps,
100  uint32_t* coded_width,
101  uint32_t* coded_height,
102  uint32_t* pixel_width,
103  uint32_t* pixel_height) {
104  int crop_x = 0;
105  int crop_y = 0;
106  if (sps.conformance_window_flag) {
107  int sub_width_c = 0;
108  int sub_height_c = 0;
109 
110  // Table 6-1
111  switch (sps.chroma_format_idc) {
112  case 0: // Monochrome
113  sub_width_c = 1;
114  sub_height_c = 1;
115  break;
116  case 1: // 4:2:0
117  sub_width_c = 2;
118  sub_height_c = 2;
119  break;
120  case 2: // 4:2:2
121  sub_width_c = 2;
122  sub_height_c = 1;
123  break;
124  case 3: // 4:4:4
125  sub_width_c = 1;
126  sub_height_c = 1;
127  break;
128  default:
129  LOG(ERROR) << "Unexpected chroma_format_idc " << sps.chroma_format_idc;
130  return false;
131  }
132 
133  // Formula D-28, D-29
134  crop_x =
135  sub_width_c * (sps.conf_win_right_offset + sps.conf_win_left_offset);
136  crop_y =
137  sub_height_c * (sps.conf_win_bottom_offset + sps.conf_win_top_offset);
138  }
139 
140  // Formula D-28, D-29
141  *coded_width = sps.pic_width_in_luma_samples - crop_x;
142  *coded_height = sps.pic_height_in_luma_samples - crop_y;
143  GetAspectRatioInfo(sps, pixel_width, pixel_height);
144  return true;
145 }
146 
147 H265Pps::H265Pps() {}
148 H265Pps::~H265Pps() {}
149 
150 H265Sps::H265Sps() {}
151 H265Sps::~H265Sps() {}
152 
153 int H265Sps::GetPicSizeInCtbsY() const {
154  int min_cb_log2_size_y = log2_min_luma_coding_block_size_minus3 + 3;
155  int ctb_log2_size_y =
156  min_cb_log2_size_y + log2_diff_max_min_luma_coding_block_size;
157  int ctb_size_y = 1 << ctb_log2_size_y;
158 
159  // Round-up division.
160  int pic_width_in_ctbs_y = (pic_width_in_luma_samples - 1) / ctb_size_y + 1;
161  int pic_height_in_ctbs_y = (pic_height_in_luma_samples - 1) / ctb_size_y + 1;
162  return pic_width_in_ctbs_y * pic_height_in_ctbs_y;
163 }
164 
165 int H265Sps::GetChromaArrayType() const {
166  if (!separate_colour_plane_flag)
167  return chroma_format_idc;
168  else
169  return 0;
170 }
171 
172 H265ReferencePictureListModifications::H265ReferencePictureListModifications() {
173 }
174 H265ReferencePictureListModifications::
175  ~H265ReferencePictureListModifications() {}
176 
177 H265SliceHeader::H265SliceHeader() {}
178 H265SliceHeader::~H265SliceHeader() {}
179 
180 H265Parser::H265Parser() {}
181 H265Parser::~H265Parser() {}
182 
183 H265Parser::Result H265Parser::ParseSliceHeader(const Nalu& nalu,
184  H265SliceHeader* slice_header) {
185  DCHECK(nalu.is_video_slice());
186  *slice_header = H265SliceHeader();
187 
188  // Parses whole element.
189  H26xBitReader reader;
190  reader.Initialize(nalu.data() + nalu.header_size(), nalu.payload_size());
191  H26xBitReader* br = &reader;
192 
193  TRUE_OR_RETURN(br->ReadBool(&slice_header->first_slice_segment_in_pic_flag));
194  if (nalu.type() >= Nalu::H265_BLA_W_LP &&
195  nalu.type() <= Nalu::H265_RSV_IRAP_VCL23) {
196  TRUE_OR_RETURN(br->ReadBool(&slice_header->no_output_of_prior_pics_flag));
197  }
198 
199  TRUE_OR_RETURN(br->ReadUE(&slice_header->pic_parameter_set_id));
200  const H265Pps* pps = GetPps(slice_header->pic_parameter_set_id);
201  TRUE_OR_RETURN(pps);
202 
203  const H265Sps* sps = GetSps(pps->seq_parameter_set_id);
204  TRUE_OR_RETURN(sps);
205 
206  if (!slice_header->first_slice_segment_in_pic_flag) {
207  if (pps->dependent_slice_segments_enabled_flag) {
208  TRUE_OR_RETURN(br->ReadBool(&slice_header->dependent_slice_segment_flag));
209  }
210  const int bit_length = ceil(log2(sps->GetPicSizeInCtbsY()));
211  TRUE_OR_RETURN(br->ReadBits(bit_length, &slice_header->segment_address));
212  }
213 
214  if (!slice_header->dependent_slice_segment_flag) {
215  TRUE_OR_RETURN(br->SkipBits(pps->num_extra_slice_header_bits));
216  TRUE_OR_RETURN(br->ReadUE(&slice_header->slice_type));
217  if (pps->output_flag_present_flag) {
218  TRUE_OR_RETURN(br->ReadBool(&slice_header->pic_output_flag));
219  }
220  if (sps->separate_colour_plane_flag) {
221  TRUE_OR_RETURN(br->ReadBits(2, &slice_header->colour_plane_id));
222  }
223 
224  if (nalu.type() != Nalu::H265_IDR_W_RADL &&
225  nalu.type() != Nalu::H265_IDR_N_LP) {
226  TRUE_OR_RETURN(br->ReadBits(sps->log2_max_pic_order_cnt_lsb_minus4 + 4,
227  &slice_header->slice_pic_order_cnt_lsb));
228 
229  TRUE_OR_RETURN(
230  br->ReadBool(&slice_header->short_term_ref_pic_set_sps_flag));
231  if (!slice_header->short_term_ref_pic_set_sps_flag) {
232  OK_OR_RETURN(ParseReferencePictureSet(
233  sps->num_short_term_ref_pic_sets, sps->num_short_term_ref_pic_sets,
234  sps->st_ref_pic_sets, br, &slice_header->st_ref_pic_set));
235  } else if (sps->num_short_term_ref_pic_sets > 1) {
236  TRUE_OR_RETURN(
237  br->ReadBits(ceil(log2(sps->num_short_term_ref_pic_sets)),
238  &slice_header->short_term_ref_pic_set_idx));
239  }
240 
241  if (sps->long_term_ref_pic_present_flag) {
242  if (sps->num_long_term_ref_pics > 0) {
243  TRUE_OR_RETURN(br->ReadUE(&slice_header->num_long_term_sps));
244  }
245  TRUE_OR_RETURN(br->ReadUE(&slice_header->num_long_term_pics));
246 
247  const int pic_count =
248  slice_header->num_long_term_sps + slice_header->num_long_term_pics;
249  slice_header->long_term_pics_info.resize(pic_count);
250  for (int i = 0; i < pic_count; i++) {
251  if (i < slice_header->num_long_term_sps) {
252  int lt_idx_sps = 0;
253  if (sps->num_long_term_ref_pics > 1) {
254  TRUE_OR_RETURN(br->ReadBits(
255  ceil(log2(sps->num_long_term_ref_pics)), &lt_idx_sps));
256  }
257  if (sps->used_by_curr_pic_lt_flag[lt_idx_sps])
258  slice_header->used_by_curr_pic_lt++;
259  } else {
260  TRUE_OR_RETURN(br->SkipBits(sps->log2_max_pic_order_cnt_lsb_minus4 +
261  4)); // poc_lsb_lt
262  bool used_by_curr_pic_lt_flag;
263  TRUE_OR_RETURN(br->ReadBool(&used_by_curr_pic_lt_flag));
264  if (used_by_curr_pic_lt_flag)
265  slice_header->used_by_curr_pic_lt++;
266  }
267  TRUE_OR_RETURN(br->ReadBool(&slice_header->long_term_pics_info[i]
268  .delta_poc_msb_present_flag));
269  if (slice_header->long_term_pics_info[i].delta_poc_msb_present_flag) {
270  TRUE_OR_RETURN(br->ReadUE(
271  &slice_header->long_term_pics_info[i].delta_poc_msb_cycle_lt));
272  }
273  }
274  }
275 
276  if (sps->temporal_mvp_enabled_flag) {
277  TRUE_OR_RETURN(
278  br->ReadBool(&slice_header->slice_temporal_mvp_enabled_flag));
279  }
280  }
281 
282  if (nalu.nuh_layer_id() != 0) {
283  NOTIMPLEMENTED() << "Multi-layer streams are not supported.";
284  return kUnsupportedStream;
285  }
286 
287  if (sps->sample_adaptive_offset_enabled_flag) {
288  TRUE_OR_RETURN(br->ReadBool(&slice_header->slice_sao_luma_flag));
289  if (sps->GetChromaArrayType() != 0) {
290  TRUE_OR_RETURN(br->ReadBool(&slice_header->slice_sao_chroma_flag));
291  }
292  }
293 
294  slice_header->num_ref_idx_l0_active_minus1 =
295  pps->num_ref_idx_l0_default_active_minus1;
296  slice_header->num_ref_idx_l1_active_minus1 =
297  pps->num_ref_idx_l1_default_active_minus1;
298  if (slice_header->slice_type == kPSlice ||
299  slice_header->slice_type == kBSlice) {
300  TRUE_OR_RETURN(
301  br->ReadBool(&slice_header->num_ref_idx_active_override_flag));
302  if (slice_header->num_ref_idx_active_override_flag) {
303  TRUE_OR_RETURN(br->ReadUE(&slice_header->num_ref_idx_l0_active_minus1));
304  if (slice_header->slice_type == kBSlice) {
305  TRUE_OR_RETURN(
306  br->ReadUE(&slice_header->num_ref_idx_l1_active_minus1));
307  }
308  }
309 
310  const int num_pic_total_curr = GetNumPicTotalCurr(*slice_header, *sps);
311  if (pps->lists_modification_present_flag && num_pic_total_curr > 1) {
312  OK_OR_RETURN(SkipReferencePictureListModification(
313  *slice_header, *pps, num_pic_total_curr, br));
314  }
315 
316  if (slice_header->slice_type == kBSlice) {
317  TRUE_OR_RETURN(br->ReadBool(&slice_header->mvd_l1_zero_flag));
318  }
319  if (pps->cabac_init_present_flag) {
320  TRUE_OR_RETURN(br->ReadBool(&slice_header->cabac_init_flag));
321  }
322  if (slice_header->slice_temporal_mvp_enabled_flag) {
323  if (slice_header->slice_type == kBSlice) {
324  TRUE_OR_RETURN(br->ReadBool(&slice_header->collocated_from_l0));
325  }
326  bool l0_greater_than_0 = slice_header->num_ref_idx_l0_active_minus1 > 0;
327  bool l1_greater_than_0 = slice_header->num_ref_idx_l1_active_minus1 > 0;
328  if (slice_header->collocated_from_l0 ? l0_greater_than_0
329  : l1_greater_than_0) {
330  TRUE_OR_RETURN(br->ReadUE(&slice_header->collocated_ref_idx));
331  }
332  }
333 
334  if ((pps->weighted_pred_flag && slice_header->slice_type == kPSlice) ||
335  (pps->weighted_bipred_flag && slice_header->slice_type == kBSlice)) {
336  OK_OR_RETURN(SkipPredictionWeightTable(
337  slice_header->slice_type == kBSlice, *sps, *slice_header, br));
338  }
339  TRUE_OR_RETURN(br->ReadUE(&slice_header->five_minus_max_num_merge_cand));
340  }
341 
342  TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_qp_delta));
343  if (pps->slice_chroma_qp_offsets_present_flag) {
344  TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_cb_qp_offset));
345  TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_cr_qp_offset));
346  }
347 
348  if (pps->chroma_qp_offset_list_enabled_flag) {
349  TRUE_OR_RETURN(
350  br->ReadBool(&slice_header->cu_chroma_qp_offset_enabled_flag));
351  }
352  if (pps->deblocking_filter_override_enabled_flag) {
353  TRUE_OR_RETURN(
354  br->ReadBool(&slice_header->deblocking_filter_override_flag));
355  }
356  if (slice_header->deblocking_filter_override_flag) {
357  TRUE_OR_RETURN(
358  br->ReadBool(&slice_header->slice_deblocking_filter_disabled_flag));
359  if (!slice_header->slice_deblocking_filter_disabled_flag) {
360  TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_beta_offset_div2));
361  TRUE_OR_RETURN(br->ReadSE(&slice_header->slice_tc_offset_div2));
362  }
363  }
364  if (pps->loop_filter_across_slices_enabled_flag &&
365  (slice_header->slice_sao_luma_flag ||
366  slice_header->slice_sao_chroma_flag ||
367  !slice_header->slice_deblocking_filter_disabled_flag)) {
368  TRUE_OR_RETURN(br->ReadBool(
369  &slice_header->slice_loop_filter_across_slices_enabled_flag));
370  }
371  }
372 
373  if (pps->tiles_enabled_flag || pps->entropy_coding_sync_enabled_flag) {
374  TRUE_OR_RETURN(br->ReadUE(&slice_header->num_entry_point_offsets));
375  if (slice_header->num_entry_point_offsets > 0) {
376  TRUE_OR_RETURN(br->ReadUE(&slice_header->offset_len_minus1));
377  slice_header->entry_point_offset_minus1.resize(
378  slice_header->num_entry_point_offsets);
379  for (int i = 0; i < slice_header->num_entry_point_offsets; i++) {
380  TRUE_OR_RETURN(
381  br->ReadBits(slice_header->offset_len_minus1 + 1,
382  &slice_header->entry_point_offset_minus1[i]));
383  }
384  }
385  }
386 
387  if (pps->slice_segment_header_extension_present_flag) {
388  int extension_length;
389  TRUE_OR_RETURN(br->ReadUE(&extension_length));
390  TRUE_OR_RETURN(br->SkipBits(extension_length * 8));
391  }
392 
393  slice_header->header_bit_size = nalu.payload_size() * 8 - br->NumBitsLeft();
394  return kOk;
395 }
396 
397 H265Parser::Result H265Parser::ParsePps(const Nalu& nalu, int* pps_id) {
398  DCHECK_EQ(Nalu::H265_PPS, nalu.type());
399 
400  // Reads most of the element, not reading the extension data.
401  H26xBitReader reader;
402  reader.Initialize(nalu.data() + nalu.header_size(), nalu.payload_size());
403  H26xBitReader* br = &reader;
404 
405  *pps_id = -1;
406  std::unique_ptr<H265Pps> pps(new H265Pps);
407 
408  TRUE_OR_RETURN(br->ReadUE(&pps->pic_parameter_set_id));
409  TRUE_OR_RETURN(br->ReadUE(&pps->seq_parameter_set_id));
410 
411  TRUE_OR_RETURN(br->ReadBool(&pps->dependent_slice_segments_enabled_flag));
412  TRUE_OR_RETURN(br->ReadBool(&pps->output_flag_present_flag));
413  TRUE_OR_RETURN(br->ReadBits(3, &pps->num_extra_slice_header_bits));
414  TRUE_OR_RETURN(br->ReadBool(&pps->sign_data_hiding_enabled_flag));
415  TRUE_OR_RETURN(br->ReadBool(&pps->cabac_init_present_flag));
416 
417  TRUE_OR_RETURN(br->ReadUE(&pps->num_ref_idx_l0_default_active_minus1));
418  TRUE_OR_RETURN(br->ReadUE(&pps->num_ref_idx_l1_default_active_minus1));
419  TRUE_OR_RETURN(br->ReadSE(&pps->init_qp_minus26));
420  TRUE_OR_RETURN(br->ReadBool(&pps->constrained_intra_pred_flag));
421  TRUE_OR_RETURN(br->ReadBool(&pps->transform_skip_enabled_flag));
422 
423  TRUE_OR_RETURN(br->ReadBool(&pps->cu_qp_delta_enabled_flag));
424  if (pps->cu_qp_delta_enabled_flag)
425  TRUE_OR_RETURN(br->ReadUE(&pps->diff_cu_qp_delta_depth));
426  TRUE_OR_RETURN(br->ReadSE(&pps->cb_qp_offset));
427  TRUE_OR_RETURN(br->ReadSE(&pps->cr_qp_offset));
428 
429  TRUE_OR_RETURN(br->ReadBool(&pps->slice_chroma_qp_offsets_present_flag));
430  TRUE_OR_RETURN(br->ReadBool(&pps->weighted_pred_flag));
431  TRUE_OR_RETURN(br->ReadBool(&pps->weighted_bipred_flag));
432  TRUE_OR_RETURN(br->ReadBool(&pps->transquant_bypass_enabled_flag));
433  TRUE_OR_RETURN(br->ReadBool(&pps->tiles_enabled_flag));
434  TRUE_OR_RETURN(br->ReadBool(&pps->entropy_coding_sync_enabled_flag));
435 
436  if (pps->tiles_enabled_flag) {
437  TRUE_OR_RETURN(br->ReadUE(&pps->num_tile_columns_minus1));
438  TRUE_OR_RETURN(br->ReadUE(&pps->num_tile_rows_minus1));
439  TRUE_OR_RETURN(br->ReadBool(&pps->uniform_spacing_flag));
440  if (!pps->uniform_spacing_flag) {
441  pps->column_width_minus1.resize(pps->num_tile_columns_minus1);
442  for (int i = 0; i < pps->num_tile_columns_minus1; i++) {
443  TRUE_OR_RETURN(br->ReadUE(&pps->column_width_minus1[i]));
444  }
445  pps->row_height_minus1.resize(pps->num_tile_rows_minus1);
446  for (int i = 0; i < pps->num_tile_rows_minus1; i++) {
447  TRUE_OR_RETURN(br->ReadUE(&pps->row_height_minus1[i]));
448  }
449  }
450  TRUE_OR_RETURN(br->ReadBool(&pps->loop_filter_across_tiles_enabled_flag));
451  }
452 
453  TRUE_OR_RETURN(br->ReadBool(&pps->loop_filter_across_slices_enabled_flag));
454  TRUE_OR_RETURN(br->ReadBool(&pps->deblocking_filter_control_present_flag));
455  if (pps->deblocking_filter_control_present_flag) {
456  TRUE_OR_RETURN(br->ReadBool(&pps->deblocking_filter_override_enabled_flag));
457  TRUE_OR_RETURN(br->ReadBool(&pps->deblocking_filter_disabled_flag));
458  if (!pps->deblocking_filter_disabled_flag) {
459  TRUE_OR_RETURN(br->ReadSE(&pps->beta_offset_div2));
460  TRUE_OR_RETURN(br->ReadSE(&pps->tc_offset_div2));
461  }
462  }
463 
464  TRUE_OR_RETURN(br->ReadBool(&pps->scaling_list_data_present_flag));
465  if (pps->scaling_list_data_present_flag) {
466  OK_OR_RETURN(SkipScalingListData(br));
467  }
468 
469  TRUE_OR_RETURN(br->ReadBool(&pps->lists_modification_present_flag));
470  TRUE_OR_RETURN(br->ReadUE(&pps->log2_parallel_merge_level_minus2));
471 
472  TRUE_OR_RETURN(
473  br->ReadBool(&pps->slice_segment_header_extension_present_flag));
474 
475  bool pps_extension_present_flag;
476  bool pps_range_extension_flag = false;
477  TRUE_OR_RETURN(br->ReadBool(&pps_extension_present_flag));
478  if (pps_extension_present_flag) {
479  TRUE_OR_RETURN(br->ReadBool(&pps_range_extension_flag));
480  // pps_multilayer_extension_flag, pps_3d_extension_flag, pps_extension_5bits
481  TRUE_OR_RETURN(br->SkipBits(1 + 1 + 5));
482  }
483 
484  if (pps_range_extension_flag) {
485  if (pps->transform_skip_enabled_flag) {
486  // log2_max_transform_skip_block_size_minus2
487  int ignored;
488  TRUE_OR_RETURN(br->ReadUE(&ignored));
489  }
490 
491  TRUE_OR_RETURN(br->SkipBits(1)); // cross_component_prediction_enabled_flag
492  TRUE_OR_RETURN(br->ReadBool(&pps->chroma_qp_offset_list_enabled_flag));
493  // Incomplete
494  }
495 
496  // Ignore remaining extension data.
497 
498  // This will replace any existing PPS instance.
499  *pps_id = pps->pic_parameter_set_id;
500  active_ppses_[*pps_id] = std::move(pps);
501 
502  return kOk;
503 }
504 
505 H265Parser::Result H265Parser::ParseSps(const Nalu& nalu, int* sps_id) {
506  DCHECK_EQ(Nalu::H265_SPS, nalu.type());
507 
508  // Reads most of the element, not reading the extension data.
509  H26xBitReader reader;
510  reader.Initialize(nalu.data() + nalu.header_size(), nalu.payload_size());
511  H26xBitReader* br = &reader;
512 
513  *sps_id = -1;
514 
515  std::unique_ptr<H265Sps> sps(new H265Sps);
516 
517  TRUE_OR_RETURN(br->ReadBits(4, &sps->video_parameter_set_id));
518  TRUE_OR_RETURN(br->ReadBits(3, &sps->max_sub_layers_minus1));
519  TRUE_OR_RETURN(br->ReadBool(&sps->temporal_id_nesting_flag));
520 
521  OK_OR_RETURN(SkipProfileTierLevel(true, sps->max_sub_layers_minus1, br));
522 
523  TRUE_OR_RETURN(br->ReadUE(&sps->seq_parameter_set_id));
524  TRUE_OR_RETURN(br->ReadUE(&sps->chroma_format_idc));
525  if (sps->chroma_format_idc == 3) {
526  TRUE_OR_RETURN(br->ReadBool(&sps->separate_colour_plane_flag));
527  }
528  TRUE_OR_RETURN(br->ReadUE(&sps->pic_width_in_luma_samples));
529  TRUE_OR_RETURN(br->ReadUE(&sps->pic_height_in_luma_samples));
530 
531  TRUE_OR_RETURN(br->ReadBool(&sps->conformance_window_flag));
532  if (sps->conformance_window_flag) {
533  TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_left_offset));
534  TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_right_offset));
535  TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_top_offset));
536  TRUE_OR_RETURN(br->ReadUE(&sps->conf_win_bottom_offset));
537  }
538 
539  TRUE_OR_RETURN(br->ReadUE(&sps->bit_depth_luma_minus8));
540  TRUE_OR_RETURN(br->ReadUE(&sps->bit_depth_chroma_minus8));
541  TRUE_OR_RETURN(br->ReadUE(&sps->log2_max_pic_order_cnt_lsb_minus4));
542 
543  TRUE_OR_RETURN(br->ReadBool(&sps->sub_layer_ordering_info_present_flag));
544  int start = sps->sub_layer_ordering_info_present_flag
545  ? 0
546  : sps->max_sub_layers_minus1;
547  for (int i = start; i <= sps->max_sub_layers_minus1; i++) {
548  TRUE_OR_RETURN(br->ReadUE(&sps->max_dec_pic_buffering_minus1[i]));
549  TRUE_OR_RETURN(br->ReadUE(&sps->max_num_reorder_pics[i]));
550  TRUE_OR_RETURN(br->ReadUE(&sps->max_latency_increase_plus1[i]));
551  }
552 
553  TRUE_OR_RETURN(br->ReadUE(&sps->log2_min_luma_coding_block_size_minus3));
554  TRUE_OR_RETURN(br->ReadUE(&sps->log2_diff_max_min_luma_coding_block_size));
555  TRUE_OR_RETURN(br->ReadUE(&sps->log2_min_luma_transform_block_size_minus2));
556  TRUE_OR_RETURN(br->ReadUE(&sps->log2_diff_max_min_luma_transform_block_size));
557  TRUE_OR_RETURN(br->ReadUE(&sps->max_transform_hierarchy_depth_inter));
558  TRUE_OR_RETURN(br->ReadUE(&sps->max_transform_hierarchy_depth_intra));
559 
560  TRUE_OR_RETURN(br->ReadBool(&sps->scaling_list_enabled_flag));
561  if (sps->scaling_list_enabled_flag) {
562  TRUE_OR_RETURN(br->ReadBool(&sps->scaling_list_data_present_flag));
563  if (sps->scaling_list_data_present_flag) {
564  OK_OR_RETURN(SkipScalingListData(br));
565  }
566  }
567 
568  TRUE_OR_RETURN(br->ReadBool(&sps->amp_enabled_flag));
569  TRUE_OR_RETURN(br->ReadBool(&sps->sample_adaptive_offset_enabled_flag));
570  TRUE_OR_RETURN(br->ReadBool(&sps->pcm_enabled_flag));
571  if (sps->pcm_enabled_flag) {
572  TRUE_OR_RETURN(br->ReadBits(4, &sps->pcm_sample_bit_depth_luma_minus1));
573  TRUE_OR_RETURN(br->ReadBits(4, &sps->pcm_sample_bit_depth_chroma_minus1));
574  TRUE_OR_RETURN(
575  br->ReadUE(&sps->log2_min_pcm_luma_coding_block_size_minus3));
576  TRUE_OR_RETURN(
577  br->ReadUE(&sps->log2_diff_max_min_pcm_luma_coding_block_size));
578  TRUE_OR_RETURN(br->ReadBool(&sps->pcm_loop_filter_disabled_flag));
579  }
580 
581  TRUE_OR_RETURN(br->ReadUE(&sps->num_short_term_ref_pic_sets));
582  sps->st_ref_pic_sets.resize(sps->num_short_term_ref_pic_sets);
583  for (int i = 0; i < sps->num_short_term_ref_pic_sets; i++) {
584  OK_OR_RETURN(ParseReferencePictureSet(sps->num_short_term_ref_pic_sets, i,
585  sps->st_ref_pic_sets, br,
586  &sps->st_ref_pic_sets[i]));
587  }
588 
589  TRUE_OR_RETURN(br->ReadBool(&sps->long_term_ref_pic_present_flag));
590  if (sps->long_term_ref_pic_present_flag) {
591  TRUE_OR_RETURN(br->ReadUE(&sps->num_long_term_ref_pics));
592  sps->lt_ref_pic_poc_lsb.resize(sps->num_long_term_ref_pics);
593  sps->used_by_curr_pic_lt_flag.resize(sps->num_long_term_ref_pics);
594  for (int i = 0; i < sps->num_long_term_ref_pics; i++) {
595  TRUE_OR_RETURN(br->ReadBits(sps->log2_max_pic_order_cnt_lsb_minus4 + 4,
596  &sps->lt_ref_pic_poc_lsb[i]));
597  bool temp;
598  TRUE_OR_RETURN(br->ReadBool(&temp));
599  sps->used_by_curr_pic_lt_flag[i] = temp;
600  }
601  }
602 
603  TRUE_OR_RETURN(br->ReadBool(&sps->temporal_mvp_enabled_flag));
604  TRUE_OR_RETURN(br->ReadBool(&sps->strong_intra_smoothing_enabled_flag));
605 
606  TRUE_OR_RETURN(br->ReadBool(&sps->vui_parameters_present));
607  if (sps->vui_parameters_present) {
608  OK_OR_RETURN(ParseVuiParameters(sps->max_sub_layers_minus1, br,
609  &sps->vui_parameters));
610  }
611 
612  // Ignore remaining extension data.
613 
614  // This will replace any existing SPS instance.
615  *sps_id = sps->seq_parameter_set_id;
616  active_spses_[*sps_id] = std::move(sps);
617 
618  return kOk;
619 }
620 
621 const H265Pps* H265Parser::GetPps(int pps_id) {
622  return active_ppses_[pps_id].get();
623 }
624 
625 const H265Sps* H265Parser::GetSps(int sps_id) {
626  return active_spses_[sps_id].get();
627 }
628 
629 H265Parser::Result H265Parser::ParseVuiParameters(int max_num_sub_layers_minus1,
630  H26xBitReader* br,
631  H265VuiParameters* vui) {
632  // Reads whole element but ignores most of it.
633  int ignored;
634 
635  TRUE_OR_RETURN(br->ReadBool(&vui->aspect_ratio_info_present_flag));
636  if (vui->aspect_ratio_info_present_flag) {
637  TRUE_OR_RETURN(br->ReadBits(8, &vui->aspect_ratio_idc));
638  if (vui->aspect_ratio_idc == H265VuiParameters::kExtendedSar) {
639  TRUE_OR_RETURN(br->ReadBits(16, &vui->sar_width));
640  TRUE_OR_RETURN(br->ReadBits(16, &vui->sar_height));
641  }
642  }
643 
644  bool overscan_info_present_flag;
645  TRUE_OR_RETURN(br->ReadBool(&overscan_info_present_flag));
646  if (overscan_info_present_flag) {
647  TRUE_OR_RETURN(br->SkipBits(1)); // overscan_appropriate_flag
648  }
649 
650  bool video_signal_type_present_flag;
651  TRUE_OR_RETURN(br->ReadBool(&video_signal_type_present_flag));
652  if (video_signal_type_present_flag) {
653  TRUE_OR_RETURN(br->SkipBits(3)); // video_format
654  TRUE_OR_RETURN(br->SkipBits(1)); // video_full_range_flag
655 
656  bool colour_description_present_flag;
657  TRUE_OR_RETURN(br->ReadBool(&colour_description_present_flag));
658  if (colour_description_present_flag) {
659  // colour_primaries, transfer_characteristics, matrix_coeffs
660  TRUE_OR_RETURN(br->SkipBits(8 + 8 + 8));
661  }
662  }
663 
664  bool chroma_loc_info_present_flag;
665  TRUE_OR_RETURN(br->ReadBool(&chroma_loc_info_present_flag));
666  if (chroma_loc_info_present_flag) {
667  // chroma_sample_log_type_top_field, chroma_sample_log_type_bottom_field
668  TRUE_OR_RETURN(br->ReadUE(&ignored));
669  TRUE_OR_RETURN(br->ReadUE(&ignored));
670  }
671 
672  // neutral_chroma_indication_flag, field_seq_flag,
673  // frame_field_info_present_flag.
674  TRUE_OR_RETURN(br->SkipBits(3));
675 
676  bool default_display_window_flag;
677  TRUE_OR_RETURN(br->ReadBool(&default_display_window_flag));
678  if (default_display_window_flag) {
679  TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_left_offset
680  TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_right_offset
681  TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_top_offset
682  TRUE_OR_RETURN(br->ReadUE(&ignored)); // def_disp_win_bottom_offset
683  }
684 
685  bool vui_timing_info_present_flag;
686  TRUE_OR_RETURN(br->ReadBool(&vui_timing_info_present_flag));
687  if (vui_timing_info_present_flag) {
688  // vui_num_units_in_tick, vui_time_scale
689  TRUE_OR_RETURN(br->SkipBits(32 + 32));
690 
691  bool vui_poc_proportional_to_timing_flag;
692  TRUE_OR_RETURN(br->ReadBool(&vui_poc_proportional_to_timing_flag));
693  if (vui_poc_proportional_to_timing_flag) {
694  // vui_num_ticks_poc_diff_one_minus1
695  TRUE_OR_RETURN(br->ReadUE(&ignored));
696  }
697 
698  bool vui_hdr_parameters_present_flag;
699  TRUE_OR_RETURN(br->ReadBool(&vui_hdr_parameters_present_flag));
700  if (vui_hdr_parameters_present_flag) {
701  OK_OR_RETURN(SkipHrdParameters(max_num_sub_layers_minus1, br));
702  }
703  }
704 
705  TRUE_OR_RETURN(br->ReadBool(&vui->bitstream_restriction_flag));
706  if (vui->bitstream_restriction_flag) {
707  // tiles_fixed_structure_flag, motion_vectors_over_pic_boundaries_flag,
708  // restricted_ref_pic_lists_flag.
709  TRUE_OR_RETURN(br->SkipBits(3));
710 
711  TRUE_OR_RETURN(br->ReadUE(&vui->min_spatial_segmentation_idc));
712  TRUE_OR_RETURN(br->ReadUE(&ignored)); // max_bytes_per_pic_denom
713  TRUE_OR_RETURN(br->ReadUE(&ignored)); // max_bits_per_min_cu_denum
714  TRUE_OR_RETURN(br->ReadUE(&ignored)); // log2_max_mv_length_horizontal
715  TRUE_OR_RETURN(br->ReadUE(&ignored)); // log2_max_mv_length_vertical
716  }
717 
718  return kOk;
719 }
720 
721 H265Parser::Result H265Parser::ParseReferencePictureSet(
722  int num_short_term_ref_pic_sets,
723  int st_rps_idx,
724  const std::vector<H265ReferencePictureSet>& ref_pic_sets,
725  H26xBitReader* br,
726  H265ReferencePictureSet* out_ref_pic_set) {
727  // Parses and processess a short-term reference picture set. This needs to
728  // be done since the size of this element may be dependent on previous
729  // reference picture sets.
730 
731  bool inter_ref_pic_set_prediction = false;
732  if (st_rps_idx != 0) {
733  TRUE_OR_RETURN(br->ReadBool(&inter_ref_pic_set_prediction));
734  }
735 
736  if (inter_ref_pic_set_prediction) {
737  int delta_idx = 1;
738  if (st_rps_idx == num_short_term_ref_pic_sets) {
739  TRUE_OR_RETURN(br->ReadUE(&delta_idx));
740  delta_idx++;
741  TRUE_OR_RETURN(delta_idx <= st_rps_idx);
742  }
743 
744  int ref_rps_idx = st_rps_idx - delta_idx;
745  DCHECK_LE(0, ref_rps_idx);
746  DCHECK_LT(ref_rps_idx, st_rps_idx);
747 
748  bool delta_rps_sign;
749  int abs_delta_rps_minus1;
750  TRUE_OR_RETURN(br->ReadBool(&delta_rps_sign));
751  TRUE_OR_RETURN(br->ReadUE(&abs_delta_rps_minus1));
752  int delta_rps =
753  delta_rps_sign ? -(abs_delta_rps_minus1 + 1) : abs_delta_rps_minus1 + 1;
754 
755  int ref_num_delta_pocs = ref_pic_sets[ref_rps_idx].num_delta_pocs;
756  std::vector<bool> used_by_curr_pic(ref_num_delta_pocs + 1);
757  std::vector<bool> use_delta(ref_num_delta_pocs + 1);
758  for (int j = 0; j <= ref_num_delta_pocs; j++) {
759  bool temp;
760  TRUE_OR_RETURN(br->ReadBool(&temp));
761  used_by_curr_pic[j] = temp;
762 
763  if (!used_by_curr_pic[j]) {
764  TRUE_OR_RETURN(br->ReadBool(&temp));
765  use_delta[j] = temp;
766  } else {
767  use_delta[j] = true;
768  }
769  }
770 
771  int ref_num_positive_pics = ref_pic_sets[ref_rps_idx].num_positive_pics;
772  int ref_num_negative_pics = ref_pic_sets[ref_rps_idx].num_negative_pics;
773  int i;
774 
775  // Update list 0.
776  {
777  i = 0;
778  for (int j = ref_num_positive_pics - 1; j >= 0; j--) {
779  int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s1[j] + delta_rps;
780  if (d_poc < 0 && use_delta[ref_num_negative_pics + j]) {
781  out_ref_pic_set->delta_poc_s0[i] = d_poc;
782  out_ref_pic_set->used_by_curr_pic_s0[i] =
783  used_by_curr_pic[ref_num_negative_pics + j];
784  i++;
785  }
786  }
787  if (delta_rps < 0 && use_delta[ref_num_delta_pocs]) {
788  out_ref_pic_set->delta_poc_s0[i] = delta_rps;
789  out_ref_pic_set->used_by_curr_pic_s0[i] =
790  used_by_curr_pic[ref_num_delta_pocs];
791  i++;
792  }
793  for (int j = 0; j < ref_num_negative_pics; j++) {
794  int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s0[j] + delta_rps;
795  if (d_poc < 0 && use_delta[j]) {
796  out_ref_pic_set->delta_poc_s0[i] = d_poc;
797  out_ref_pic_set->used_by_curr_pic_s0[i] = used_by_curr_pic[j];
798  i++;
799  }
800  }
801  out_ref_pic_set->num_negative_pics = i;
802  }
803 
804  // Update list 1.
805  {
806  i = 0;
807  for (int j = ref_num_negative_pics - 1; j >= 0; j--) {
808  int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s0[j] + delta_rps;
809  if (d_poc > 0 && use_delta[j]) {
810  out_ref_pic_set->delta_poc_s1[i] = d_poc;
811  out_ref_pic_set->used_by_curr_pic_s1[i] = used_by_curr_pic[j];
812  i++;
813  }
814  }
815  if (delta_rps > 0 && use_delta[ref_num_delta_pocs]) {
816  out_ref_pic_set->delta_poc_s1[i] = delta_rps;
817  out_ref_pic_set->used_by_curr_pic_s1[i] =
818  used_by_curr_pic[ref_num_delta_pocs];
819  i++;
820  }
821  for (int j = 0; j < ref_num_positive_pics; j++) {
822  int d_poc = ref_pic_sets[ref_rps_idx].delta_poc_s1[j] + delta_rps;
823  if (d_poc > 0 && use_delta[ref_num_negative_pics + j]) {
824  out_ref_pic_set->delta_poc_s1[i] = d_poc;
825  out_ref_pic_set->used_by_curr_pic_s1[i] =
826  used_by_curr_pic[ref_num_negative_pics + j];
827  i++;
828  }
829  }
830  out_ref_pic_set->num_positive_pics = i;
831  }
832  } else {
833  TRUE_OR_RETURN(br->ReadUE(&out_ref_pic_set->num_negative_pics));
834  TRUE_OR_RETURN(br->ReadUE(&out_ref_pic_set->num_positive_pics));
835 
836  int prev_poc = 0;
837  for (int i = 0; i < out_ref_pic_set->num_negative_pics; i++) {
838  int delta_poc_s0_minus1;
839  TRUE_OR_RETURN(br->ReadUE(&delta_poc_s0_minus1));
840  out_ref_pic_set->delta_poc_s0[i] = prev_poc - (delta_poc_s0_minus1 + 1);
841  prev_poc = out_ref_pic_set->delta_poc_s0[i];
842 
843  TRUE_OR_RETURN(br->ReadBool(&out_ref_pic_set->used_by_curr_pic_s0[i]));
844  }
845 
846  prev_poc = 0;
847  for (int i = 0; i < out_ref_pic_set->num_positive_pics; i++) {
848  int delta_poc_s1_minus1;
849  TRUE_OR_RETURN(br->ReadUE(&delta_poc_s1_minus1));
850  out_ref_pic_set->delta_poc_s1[i] = prev_poc + delta_poc_s1_minus1 + 1;
851  prev_poc = out_ref_pic_set->delta_poc_s1[i];
852 
853  TRUE_OR_RETURN(br->ReadBool(&out_ref_pic_set->used_by_curr_pic_s1[i]));
854  }
855  }
856 
857  out_ref_pic_set->num_delta_pocs =
858  out_ref_pic_set->num_positive_pics + out_ref_pic_set->num_negative_pics;
859  return kOk;
860 }
861 
862 H265Parser::Result H265Parser::SkipReferencePictureListModification(
863  const H265SliceHeader& slice_header,
864  const H265Pps& pps,
865  int num_pic_total_curr,
866  H26xBitReader* br) {
867  // Reads whole element but ignores it all.
868 
869  bool ref_pic_list_modification_flag_l0;
870  TRUE_OR_RETURN(br->ReadBool(&ref_pic_list_modification_flag_l0));
871  if (ref_pic_list_modification_flag_l0) {
872  for (int i = 0; i <= pps.num_ref_idx_l0_default_active_minus1; i++) {
873  TRUE_OR_RETURN(br->SkipBits(ceil(log2(num_pic_total_curr))));
874  }
875  }
876 
877  if (slice_header.slice_type == kBSlice) {
878  bool ref_pic_list_modification_flag_l1;
879  TRUE_OR_RETURN(br->ReadBool(&ref_pic_list_modification_flag_l1));
880  if (ref_pic_list_modification_flag_l1) {
881  for (int i = 0; i <= pps.num_ref_idx_l1_default_active_minus1; i++) {
882  TRUE_OR_RETURN(br->SkipBits(ceil(log2(num_pic_total_curr))));
883  }
884  }
885  }
886 
887  return kOk;
888 }
889 
890 H265Parser::Result H265Parser::SkipPredictionWeightTablePart(
891  int num_ref_idx_minus1,
892  int chroma_array_type,
893  H26xBitReader* br) {
894  // Reads whole element, ignores it.
895  int ignored;
896  std::vector<bool> luma_weight_flag(num_ref_idx_minus1 + 1);
897  std::vector<bool> chroma_weight_flag(num_ref_idx_minus1 + 1);
898 
899  for (int i = 0; i <= num_ref_idx_minus1; i++) {
900  bool temp;
901  TRUE_OR_RETURN(br->ReadBool(&temp));
902  luma_weight_flag[i] = temp;
903  }
904  if (chroma_array_type != 0) {
905  for (int i = 0; i <= num_ref_idx_minus1; i++) {
906  bool temp;
907  TRUE_OR_RETURN(br->ReadBool(&temp));
908  chroma_weight_flag[i] = temp;
909  }
910  }
911  for (int i = 0; i <= num_ref_idx_minus1; i++) {
912  if (luma_weight_flag[i]) {
913  TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_luma_weight_l#
914  TRUE_OR_RETURN(br->ReadSE(&ignored)); // luma_offset_l#
915  }
916  if (chroma_weight_flag[i]) {
917  for (int j = 0; j < 2; j++) {
918  TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_chroma_weight_l#
919  TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_chroma_offset_l#
920  }
921  }
922  }
923 
924  return kOk;
925 }
926 
927 H265Parser::Result H265Parser::SkipPredictionWeightTable(
928  bool is_b_slice,
929  const H265Sps& sps,
930  const H265SliceHeader& slice_header,
931  H26xBitReader* br) {
932  // Reads whole element, ignores it.
933  int ignored;
934  int chroma_array_type = sps.GetChromaArrayType();
935 
936  TRUE_OR_RETURN(br->ReadUE(&ignored)); // luma_log2_weight_denom
937  if (chroma_array_type != 0) {
938  TRUE_OR_RETURN(br->ReadSE(&ignored)); // delta_chroma_log2_weight_denom
939  }
940  OK_OR_RETURN(SkipPredictionWeightTablePart(
941  slice_header.num_ref_idx_l0_active_minus1, chroma_array_type, br));
942  if (is_b_slice) {
943  OK_OR_RETURN(SkipPredictionWeightTablePart(
944  slice_header.num_ref_idx_l1_active_minus1, chroma_array_type, br));
945  }
946 
947  return kOk;
948 }
949 
950 H265Parser::Result H265Parser::SkipProfileTierLevel(
951  bool profile_present,
952  int max_num_sub_layers_minus1,
953  H26xBitReader* br) {
954  // Reads whole element, ignores it.
955 
956  if (profile_present) {
957  // general_profile_space, general_tier_flag, general_profile_idc
958  // general_profile_compativility_flag
959  // general_progressive_source_flag
960  // general_interlaced_source_flag
961  // general_non_packed_constraint_flag
962  // general_frame_only_constraint_flag
963  // 44-bits of other flags
964  TRUE_OR_RETURN(br->SkipBits(2 + 1 + 5 + 32 + 4 + 44));
965  }
966 
967  TRUE_OR_RETURN(br->SkipBits(8)); // general_level_idc
968 
969  std::vector<bool> sub_layer_profile_present(max_num_sub_layers_minus1);
970  std::vector<bool> sub_layer_level_present(max_num_sub_layers_minus1);
971  for (int i = 0; i < max_num_sub_layers_minus1; i++) {
972  bool profile, level;
973  TRUE_OR_RETURN(br->ReadBool(&profile));
974  TRUE_OR_RETURN(br->ReadBool(&level));
975  sub_layer_profile_present[i] = profile;
976  sub_layer_level_present[i] = level;
977  }
978 
979  if (max_num_sub_layers_minus1 > 0) {
980  for (int i = max_num_sub_layers_minus1; i < 8; i++)
981  TRUE_OR_RETURN(br->SkipBits(2)); // reserved_zero_2bits
982  }
983 
984  for (int i = 0; i < max_num_sub_layers_minus1; i++) {
985  if (sub_layer_profile_present[i]) {
986  // sub_layer_profile_space, sub_layer_tier_flag, sub_layer_profile_idc
987  // sub_layer_profile_compatibility
988  // sub_layer_reserved_zero_43bits
989  // sub_layer_reserved_zero_bit
990  TRUE_OR_RETURN(br->SkipBits(2 + 1 + 5 + 32 + 4 + 43 + 1));
991  }
992  if (sub_layer_level_present[i]) {
993  TRUE_OR_RETURN(br->SkipBits(8));
994  }
995  }
996 
997  return kOk;
998 }
999 
1000 H265Parser::Result H265Parser::SkipScalingListData(H26xBitReader* br) {
1001  // Reads whole element, ignores it.
1002  int ignored;
1003  for (int size_id = 0; size_id < 4; size_id++) {
1004  for (int matrix_id = 0; matrix_id < 6;
1005  matrix_id += ((size_id == 3) ? 3 : 1)) {
1006  bool scaling_list_pred_mode;
1007  TRUE_OR_RETURN(br->ReadBool(&scaling_list_pred_mode));
1008  if (!scaling_list_pred_mode) {
1009  // scaling_list_pred_matrix_id_delta
1010  TRUE_OR_RETURN(br->ReadUE(&ignored));
1011  } else {
1012  int coefNum = std::min(64, (1 << (4 + (size_id << 1))));
1013  if (size_id > 1) {
1014  TRUE_OR_RETURN(br->ReadSE(&ignored)); // scaling_list_dc_coef_minus8
1015  }
1016 
1017  for (int i = 0; i < coefNum; i++) {
1018  TRUE_OR_RETURN(br->ReadSE(&ignored)); // scaling_list_delta_coef
1019  }
1020  }
1021  }
1022  }
1023 
1024  return kOk;
1025 }
1026 
1027 H265Parser::Result H265Parser::SkipHrdParameters(int max_num_sub_layers_minus1,
1028  H26xBitReader* br) {
1029  // common_inf_present_flag is always 1 when parsing vui_parameters.
1030  const bool common_inf_present_flag = true;
1031 
1032  int ignored;
1033  bool nal_hdr_parameters_present_flag;
1034  bool vcl_hdr_parameters_present_flag;
1035  bool sub_pic_hdr_params_present_flag = false;
1036  if (common_inf_present_flag) {
1037  TRUE_OR_RETURN(br->ReadBool(&nal_hdr_parameters_present_flag));
1038  TRUE_OR_RETURN(br->ReadBool(&vcl_hdr_parameters_present_flag));
1039  if (nal_hdr_parameters_present_flag || vcl_hdr_parameters_present_flag) {
1040  TRUE_OR_RETURN(br->ReadBool(&sub_pic_hdr_params_present_flag));
1041  if (sub_pic_hdr_params_present_flag) {
1042  // tick_divisor_minus2, du_cpb_removal_delay_increment_length_minus1,
1043  // sub_pic_cpb_params_in_pic_timing_sei_flag
1044  // dpb_output_delay_du_length_minus1
1045  TRUE_OR_RETURN(br->SkipBits(8 + 5 + 1 + 5));
1046  }
1047 
1048  // bit_rate_scale, cpb_size_scale
1049  TRUE_OR_RETURN(br->SkipBits(4 + 4));
1050  if (sub_pic_hdr_params_present_flag)
1051  TRUE_OR_RETURN(br->SkipBits(4)); // cpb_size_du_scale
1052 
1053  // initial_cpb_removal_delay_length_minus1,
1054  // au_cpb_removal_delay_length_minus1, dpb_output_delay_length_minus1
1055  TRUE_OR_RETURN(br->SkipBits(5 + 5 + 5));
1056  }
1057  }
1058 
1059  for (int i = 0; i <= max_num_sub_layers_minus1; i++) {
1060  bool fixed_pic_rate_general_flag;
1061  bool fixed_pic_rate_within_cvs_flag = true;
1062  bool low_delay_hdr_flag = false;
1063  int cpb_cnt_minus1 = 0;
1064  TRUE_OR_RETURN(br->ReadBool(&fixed_pic_rate_general_flag));
1065  if (!fixed_pic_rate_general_flag)
1066  TRUE_OR_RETURN(br->ReadBool(&fixed_pic_rate_within_cvs_flag));
1067  if (fixed_pic_rate_within_cvs_flag)
1068  TRUE_OR_RETURN(br->ReadUE(&ignored)); // elemental_duration_ic_tc_minus1
1069  else
1070  TRUE_OR_RETURN(br->ReadBool(&low_delay_hdr_flag));
1071  if (!low_delay_hdr_flag)
1072  TRUE_OR_RETURN(br->ReadUE(&cpb_cnt_minus1));
1073 
1074  if (nal_hdr_parameters_present_flag) {
1075  OK_OR_RETURN(SkipSubLayerHrdParameters(
1076  cpb_cnt_minus1, sub_pic_hdr_params_present_flag, br));
1077  }
1078  if (vcl_hdr_parameters_present_flag) {
1079  OK_OR_RETURN(SkipSubLayerHrdParameters(
1080  cpb_cnt_minus1, sub_pic_hdr_params_present_flag, br));
1081  }
1082  }
1083 
1084  return kOk;
1085 }
1086 
1087 H265Parser::Result H265Parser::SkipSubLayerHrdParameters(
1088  int cpb_cnt_minus1,
1089  bool sub_pic_hdr_params_present_flag,
1090  H26xBitReader* br) {
1091  int ignored;
1092  for (int i = 0; i <= cpb_cnt_minus1; i++) {
1093  TRUE_OR_RETURN(br->ReadUE(&ignored)); // bit_rate_value_minus1
1094  TRUE_OR_RETURN(br->ReadUE(&ignored)); // cpb_size_value_minus1
1095  if (sub_pic_hdr_params_present_flag) {
1096  TRUE_OR_RETURN(br->ReadUE(&ignored)); // cpb_size_du_value_minus1
1097  TRUE_OR_RETURN(br->ReadUE(&ignored)); // bit_rate_du_value_minus1
1098  }
1099 
1100  TRUE_OR_RETURN(br->SkipBits(1)); // cbr_flag
1101  }
1102 
1103  return kOk;
1104 }
1105 
1106 } // namespace media
1107 } // namespace shaka
uint64_t payload_size() const
Size of this Nalu minus header_size().
Definition: nalu_reader.h:102
bool is_video_slice() const
Slice data partition NALs are not considered as slice NALs.
Definition: nalu_reader.h:117
Result ParseSliceHeader(const Nalu &nalu, H265SliceHeader *slice_header)
Definition: h265_parser.cc:183
Result ParsePps(const Nalu &nalu, int *pps_id)
Definition: h265_parser.cc:397
int type() const
Definition: nalu_reader.h:113
uint64_t header_size() const
The size of the header, e.g. 1 for H.264.
Definition: nalu_reader.h:100
const H265Sps * GetSps(int sps_id)
Definition: h265_parser.cc:625
const H265Pps * GetPps(int pps_id)
Definition: h265_parser.cc:621
Result ParseSps(const Nalu &nalu, int *sps_id)
Definition: h265_parser.cc:505
const uint8_t * data() const
This is the pointer to the Nalu data, pointing to the header.
Definition: nalu_reader.h:97