libavcodec/hevc: random cosmetics to reduce diff to 064698d381e1e7790f21b0199a8930ea0...
[ffmpeg.git] / libavcodec / hevc_sei.c
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 "hevc.h"
26 #include "golomb.h"
27
28 static void decode_nal_sei_decoded_picture_hash(HEVCContext *s,
29                                                 int payload_size)
30 {
31     int cIdx, i;
32     uint8_t hash_type;
33     //uint16_t picture_crc;
34     //uint32_t picture_checksum;
35     GetBitContext *gb = &s->HEVClc->gb;
36     hash_type = get_bits(gb, 8);
37
38     for (cIdx = 0; cIdx < 3/*((s->sps->chroma_format_idc == 0) ? 1 : 3)*/; cIdx++) {
39         if (hash_type == 0) {
40             s->is_md5 = 1;
41             for (i = 0; i < 16; i++)
42                 s->md5[cIdx][i] = get_bits(gb, 8);
43         } else if (hash_type == 1) {
44             // picture_crc = get_bits(gb, 16);
45             skip_bits(gb, 16);
46         } else if (hash_type == 2) {
47             // picture_checksum = get_bits(gb, 32);
48             skip_bits(gb, 32);
49         }
50     }
51 }
52
53 static void decode_nal_sei_frame_packing_arrangement(HEVCContext *s)
54 {
55     GetBitContext *gb = &s->HEVClc->gb;
56     int cancel, type, quincunx;
57
58     get_ue_golomb(gb);                  // frame_packing_arrangement_id
59     cancel = get_bits1(gb);             // frame_packing_cancel_flag
60     if (cancel == 0 ){
61         type     = get_bits(gb, 7);     // frame_packing_arrangement_type
62         quincunx = get_bits1(gb);       // quincunx_sampling_flag
63         skip_bits(gb, 6);               // content_interpretation_type
64
65         // the following skips spatial_flipping_flag frame0_flipped_flag
66         // field_views_flag current_frame_is_frame0_flag
67         // frame0_self_contained_flag frame1_self_contained_flag
68         skip_bits(gb, 6);
69
70         if (quincunx == 0 && type != 5)
71             skip_bits(gb, 16);  // frame[01]_grid_position_[xy]
72         skip_bits(gb, 8);       // frame_packing_arrangement_reserved_byte
73         skip_bits1(gb);         // frame_packing_arrangement_persistance_flag
74     }
75     skip_bits1(gb);             // upsampled_aspect_ratio_flag
76 }
77
78 static int decode_pic_timing(HEVCContext *s)
79 {
80     GetBitContext *gb = &s->HEVClc->gb;
81     HEVCSPS *sps = (HEVCSPS*)s->sps_list[s->active_seq_parameter_set_id]->data;
82
83     if (!sps)
84         return(AVERROR(ENOMEM));
85
86     if (sps->vui.frame_field_info_present_flag) {
87         int pic_struct = get_bits(gb, 4);
88         s->picture_struct = AV_PICTURE_STRUCTURE_UNKNOWN;
89         if (pic_struct == 2) {
90             av_log(s->avctx, AV_LOG_DEBUG, "BOTTOM Field\n");
91             s->picture_struct = AV_PICTURE_STRUCTURE_BOTTOM_FIELD;
92         } else if (pic_struct == 1) {
93             av_log(s->avctx, AV_LOG_DEBUG, "TOP Field\n");
94             s->picture_struct = AV_PICTURE_STRUCTURE_TOP_FIELD;
95         }
96         get_bits(gb, 2);                   // source_scan_type
97         get_bits(gb, 1);                   // duplicate_flag
98     }
99     return 1;
100 }
101
102 static void active_parameter_sets(HEVCContext *s)
103 {
104     GetBitContext *gb = &s->HEVClc->gb;
105     int num_sps_ids_minus1;
106     int i;
107
108     get_bits(gb, 4); // active_video_parameter_set_id
109     get_bits(gb, 1); // self_contained_cvs_flag
110     get_bits(gb, 1); // num_sps_ids_minus1
111     num_sps_ids_minus1 = get_ue_golomb_long(gb); // num_sps_ids_minus1
112
113     s->active_seq_parameter_set_id = get_ue_golomb_long(gb);
114
115     for (i = 1; i <= num_sps_ids_minus1; i++)
116         get_ue_golomb_long(gb); // active_seq_parameter_set_id[i]
117 }
118
119 static int decode_nal_sei_message(HEVCContext *s)
120 {
121     GetBitContext *gb = &s->HEVClc->gb;
122
123     int payload_type = 0;
124     int payload_size = 0;
125     int byte = 0xFF;
126     av_log(s->avctx, AV_LOG_DEBUG, "Decoding SEI\n");
127
128     while (byte == 0xFF) {
129         byte          = get_bits(gb, 8);
130         payload_type += byte;
131     }
132     byte = 0xFF;
133     while (byte == 0xFF) {
134         byte          = get_bits(gb, 8);
135         payload_size += byte;
136     }
137     if (s->nal_unit_type == NAL_SEI_PREFIX) {
138         if (payload_type == 256 /*&& s->decode_checksum_sei*/) {
139             decode_nal_sei_decoded_picture_hash(s, payload_size);
140             return 1;
141         } else if (payload_type == 45) {
142             decode_nal_sei_frame_packing_arrangement(s);
143             return 1;
144         } else if (payload_type == 1){
145             int ret = decode_pic_timing(s);
146             av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", payload_type);
147             skip_bits(gb, 8 * payload_size);
148             return ret;
149         } else if (payload_type == 129){
150             active_parameter_sets(s);
151             av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", payload_type);
152             return 1;
153         } else {
154             av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", payload_type);
155             skip_bits(gb, 8*payload_size);
156             return 1;
157         }
158     } else { /* nal_unit_type == NAL_SEI_SUFFIX */
159         if (payload_type == 132 /* && s->decode_checksum_sei */)
160             decode_nal_sei_decoded_picture_hash(s, payload_size);
161         else {
162             av_log(s->avctx, AV_LOG_DEBUG, "Skipped SUFFIX SEI %d\n", payload_type);
163             skip_bits(gb, 8 * payload_size);
164         }
165         return 1;
166     }
167 }
168
169 static int more_rbsp_data(GetBitContext *gb)
170 {
171     return get_bits_left(gb) > 0 && show_bits(gb, 8) != 0x80;
172 }
173
174 int ff_hevc_decode_nal_sei(HEVCContext *s)
175 {
176     int ret;
177
178     do {
179         ret = decode_nal_sei_message(s);
180         if (ret < 0)
181             return(AVERROR(ENOMEM));
182     } while (more_rbsp_data(&s->HEVClc->gb));
183     return 1;
184 }