hevc: Add support for alternative transfer characterics SEI
[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 Libav.
9  *
10  * Libav 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  * Libav 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 Libav; 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_legacy.h"
26 #include "hevc.h"
27 #include "hevc_sei.h"
28
29 static int decode_nal_sei_decoded_picture_hash(HEVCSEIPictureHash *s, GetBitContext *gb)
30 {
31     int cIdx, i;
32     uint8_t hash_type = get_bits(gb, 8);
33
34     for (cIdx = 0; cIdx < 3; cIdx++) {
35         if (hash_type == 0) {
36             s->is_md5 = 1;
37             for (i = 0; i < 16; i++)
38                 s->md5[cIdx][i] = get_bits(gb, 8);
39         } else if (hash_type == 1) {
40             // picture_crc = get_bits(gb, 16);
41             skip_bits(gb, 16);
42         } else if (hash_type == 2) {
43             // picture_checksum = get_bits(gb, 32);
44             skip_bits(gb, 32);
45         }
46     }
47     return 0;
48 }
49
50 static int decode_nal_sei_frame_packing_arrangement(HEVCSEIFramePacking *s, GetBitContext *gb)
51 {
52     get_ue_golomb(gb);                  // frame_packing_arrangement_id
53     s->present = !get_bits1(gb);
54
55     if (s->present) {
56         s->arrangement_type               = get_bits(gb, 7);
57         s->quincunx_subsampling           = get_bits1(gb);
58         s->content_interpretation_type    = get_bits(gb, 6);
59
60         // the following skips spatial_flipping_flag frame0_flipped_flag
61         // field_views_flag current_frame_is_frame0_flag
62         // frame0_self_contained_flag frame1_self_contained_flag
63         skip_bits(gb, 6);
64
65         if (!s->quincunx_subsampling && s->arrangement_type != 5)
66             skip_bits(gb, 16);  // frame[01]_grid_position_[xy]
67         skip_bits(gb, 8);       // frame_packing_arrangement_reserved_byte
68         skip_bits1(gb);         // frame_packing_arrangement_persistence_flag
69     }
70     skip_bits1(gb);             // upsampled_aspect_ratio_flag
71     return 0;
72 }
73
74 static int decode_nal_sei_display_orientation(HEVCSEIDisplayOrientation *s, GetBitContext *gb)
75 {
76     s->present = !get_bits1(gb);
77
78     if (s->present) {
79         s->hflip = get_bits1(gb);     // hor_flip
80         s->vflip = get_bits1(gb);     // ver_flip
81
82         s->anticlockwise_rotation = get_bits(gb, 16);
83         skip_bits1(gb);     // display_orientation_persistence_flag
84     }
85
86     return 0;
87 }
88
89 static int decode_nal_sei_alternative_transfer(HEVCSEIAlternativeTransfer *s, GetBitContext *gb)
90 {
91     s->present = 1;
92     s->preferred_transfer_characteristics = get_bits(gb, 8);
93     return 0;
94 }
95
96 static int decode_nal_sei_prefix(GetBitContext *gb, void *logctx, HEVCSEI *s,
97                                  int type, int size)
98 {
99     switch (type) {
100     case 256:  // Mismatched value from HM 8.1
101         return decode_nal_sei_decoded_picture_hash(&s->picture_hash, gb);
102     case HEVC_SEI_TYPE_FRAME_PACKING:
103         return decode_nal_sei_frame_packing_arrangement(&s->frame_packing, gb);
104     case HEVC_SEI_TYPE_DISPLAY_ORIENTATION:
105         return decode_nal_sei_display_orientation(&s->display_orientation, gb);
106     case HEVC_SEI_TYPE_ALTERNATIVE_TRANSFER_CHARACTERISTICS:
107         return decode_nal_sei_alternative_transfer(&s->alternative_transfer, gb);
108     default:
109         av_log(logctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", type);
110         skip_bits_long(gb, 8 * size);
111         return 0;
112     }
113 }
114
115 static int decode_nal_sei_suffix(GetBitContext *gb, void *logctx, HEVCSEI *s,
116                                  int type, int size)
117 {
118     switch (type) {
119     case HEVC_SEI_TYPE_DECODED_PICTURE_HASH:
120         return decode_nal_sei_decoded_picture_hash(&s->picture_hash, gb);
121     default:
122         av_log(logctx, AV_LOG_DEBUG, "Skipped SUFFIX SEI %d\n", type);
123         skip_bits_long(gb, 8 * size);
124         return 0;
125     }
126 }
127
128 static int decode_nal_sei_message(GetBitContext *gb, void *logctx, HEVCSEI *s,
129                                   int type)
130 {
131     int payload_type = 0;
132     int payload_size = 0;
133     int byte = 0xFF;
134     av_log(logctx, 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 (type == HEVC_NAL_SEI_PREFIX) {
146         return decode_nal_sei_prefix(gb, logctx, s, payload_type, payload_size);
147     } else { /* nal_unit_type == NAL_SEI_SUFFIX */
148         return decode_nal_sei_suffix(gb, logctx, s, payload_type, payload_size);
149     }
150 }
151
152 static int more_rbsp_data(GetBitContext *gb)
153 {
154     return get_bits_left(gb) > 0 && show_bits(gb, 8) != 0x80;
155 }
156
157 int ff_hevc_decode_nal_sei(GetBitContext *gb, void *logctx, HEVCSEI *s,
158                            int type)
159 {
160     do {
161         decode_nal_sei_message(gb, logctx, s, type);
162     } while (more_rbsp_data(gb));
163     return 0;
164 }