FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
hevc_sei.c
Go to the documentation of this file.
1 /*
2  * HEVC Supplementary Enhancement Information messages
3  *
4  * Copyright (C) 2012 - 2013 Guillaume Martres
5  * Copyright (C) 2012 - 2013 Gildas Cocherel
6  * Copyright (C) 2013 Vittorio Giovara
7  *
8  * This file is part of FFmpeg.
9  *
10  * FFmpeg is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public
12  * License as published by the Free Software Foundation; either
13  * version 2.1 of the License, or (at your option) any later version.
14  *
15  * FFmpeg is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with FFmpeg; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23  */
24 
25 #include "golomb.h"
26 #include "hevc.h"
27 
29 {
30  int cIdx, i;
31  uint8_t hash_type;
32  //uint16_t picture_crc;
33  //uint32_t picture_checksum;
34  GetBitContext *gb = &s->HEVClc->gb;
35  hash_type = get_bits(gb, 8);
36 
37  for (cIdx = 0; cIdx < 3/*((s->sps->chroma_format_idc == 0) ? 1 : 3)*/; cIdx++) {
38  if (hash_type == 0) {
39  s->is_md5 = 1;
40  for (i = 0; i < 16; i++)
41  s->md5[cIdx][i] = get_bits(gb, 8);
42  } else if (hash_type == 1) {
43  // picture_crc = get_bits(gb, 16);
44  skip_bits(gb, 16);
45  } else if (hash_type == 2) {
46  // picture_checksum = get_bits(gb, 32);
47  skip_bits(gb, 32);
48  }
49  }
50 }
51 
53 {
54  GetBitContext *gb = &s->HEVClc->gb;
55 
56  get_ue_golomb(gb); // frame_packing_arrangement_id
58 
63 
64  // the following skips spatial_flipping_flag frame0_flipped_flag
65  // field_views_flag current_frame_is_frame0_flag
66  // frame0_self_contained_flag frame1_self_contained_flag
67  skip_bits(gb, 6);
68 
70  skip_bits(gb, 16); // frame[01]_grid_position_[xy]
71  skip_bits(gb, 8); // frame_packing_arrangement_reserved_byte
72  skip_bits1(gb); // frame_packing_arrangement_persistance_flag
73  }
74  skip_bits1(gb); // upsampled_aspect_ratio_flag
75 }
76 
78 {
79  GetBitContext *gb = &s->HEVClc->gb;
80  HEVCSPS *sps;
81 
83  return(AVERROR(ENOMEM));
85 
87  int pic_struct = get_bits(gb, 4);
89  if (pic_struct == 2) {
90  av_log(s->avctx, AV_LOG_DEBUG, "BOTTOM Field\n");
92  } else if (pic_struct == 1) {
93  av_log(s->avctx, AV_LOG_DEBUG, "TOP Field\n");
95  }
96  get_bits(gb, 2); // source_scan_type
97  get_bits(gb, 1); // duplicate_flag
98  }
99  return 1;
100 }
101 
103 {
104  GetBitContext *gb = &s->HEVClc->gb;
105  int num_sps_ids_minus1;
106  int i;
107  unsigned active_seq_parameter_set_id;
108 
109  get_bits(gb, 4); // active_video_parameter_set_id
110  get_bits(gb, 1); // self_contained_cvs_flag
111  get_bits(gb, 1); // num_sps_ids_minus1
112  num_sps_ids_minus1 = get_ue_golomb_long(gb); // num_sps_ids_minus1
113 
114  active_seq_parameter_set_id = get_ue_golomb_long(gb);
115  if (active_seq_parameter_set_id >= MAX_SPS_COUNT) {
116  av_log(s->avctx, AV_LOG_ERROR, "active_parameter_set_id %d invalid\n", active_seq_parameter_set_id);
117  return AVERROR_INVALIDDATA;
118  }
119  s->active_seq_parameter_set_id = active_seq_parameter_set_id;
120 
121  for (i = 1; i <= num_sps_ids_minus1; i++)
122  get_ue_golomb_long(gb); // active_seq_parameter_set_id[i]
123 
124  return 0;
125 }
126 
128 {
129  GetBitContext *gb = &s->HEVClc->gb;
130 
131  int payload_type = 0;
132  int payload_size = 0;
133  int byte = 0xFF;
134  av_log(s->avctx, AV_LOG_DEBUG, "Decoding SEI\n");
135 
136  while (byte == 0xFF) {
137  byte = get_bits(gb, 8);
138  payload_type += byte;
139  }
140  byte = 0xFF;
141  while (byte == 0xFF) {
142  byte = get_bits(gb, 8);
143  payload_size += byte;
144  }
145  if (s->nal_unit_type == NAL_SEI_PREFIX) {
146  if (payload_type == 256 /*&& s->decode_checksum_sei*/) {
148  return 1;
149  } else if (payload_type == 45) {
151  return 1;
152  } else if (payload_type == 1){
153  int ret = decode_pic_timing(s);
154  av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", payload_type);
155  skip_bits(gb, 8 * payload_size);
156  return ret;
157  } else if (payload_type == 129){
159  av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", payload_type);
160  return 1;
161  } else {
162  av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", payload_type);
163  skip_bits(gb, 8*payload_size);
164  return 1;
165  }
166  } else { /* nal_unit_type == NAL_SEI_SUFFIX */
167  if (payload_type == 132 /* && s->decode_checksum_sei */)
169  else {
170  av_log(s->avctx, AV_LOG_DEBUG, "Skipped SUFFIX SEI %d\n", payload_type);
171  skip_bits(gb, 8 * payload_size);
172  }
173  return 1;
174  }
175 }
176 
178 {
179  return get_bits_left(gb) > 0 && show_bits(gb, 8) != 0x80;
180 }
181 
183 {
184  int ret;
185 
186  do {
187  ret = decode_nal_sei_message(s);
188  if (ret < 0)
189  return(AVERROR(ENOMEM));
190  } while (more_rbsp_data(&s->HEVClc->gb));
191  return 1;
192 }