avcodec/nvenc: support external context in sw mode
[ffmpeg.git] / libavcodec / hevc_sei.h
1 /*
2  * HEVC Supplementary Enhancement Information messages
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #ifndef AVCODEC_HEVC_SEI_H
22 #define AVCODEC_HEVC_SEI_H
23
24 #include <stdint.h>
25
26 #include "libavutil/md5.h"
27
28 #include "get_bits.h"
29
30 /**
31  * SEI message types
32  */
33 typedef enum {
34     HEVC_SEI_TYPE_BUFFERING_PERIOD                     = 0,
35     HEVC_SEI_TYPE_PICTURE_TIMING                       = 1,
36     HEVC_SEI_TYPE_PAN_SCAN_RECT                        = 2,
37     HEVC_SEI_TYPE_FILLER_PAYLOAD                       = 3,
38     HEVC_SEI_TYPE_USER_DATA_REGISTERED_ITU_T_T35       = 4,
39     HEVC_SEI_TYPE_USER_DATA_UNREGISTERED               = 5,
40     HEVC_SEI_TYPE_RECOVERY_POINT                       = 6,
41     HEVC_SEI_TYPE_SCENE_INFO                           = 9,
42     HEVC_SEI_TYPE_FULL_FRAME_SNAPSHOT                  = 15,
43     HEVC_SEI_TYPE_PROGRESSIVE_REFINEMENT_SEGMENT_START = 16,
44     HEVC_SEI_TYPE_PROGRESSIVE_REFINEMENT_SEGMENT_END   = 17,
45     HEVC_SEI_TYPE_FILM_GRAIN_CHARACTERISTICS           = 19,
46     HEVC_SEI_TYPE_POST_FILTER_HINT                     = 22,
47     HEVC_SEI_TYPE_TONE_MAPPING_INFO                    = 23,
48     HEVC_SEI_TYPE_FRAME_PACKING                        = 45,
49     HEVC_SEI_TYPE_DISPLAY_ORIENTATION                  = 47,
50     HEVC_SEI_TYPE_SOP_DESCRIPTION                      = 128,
51     HEVC_SEI_TYPE_ACTIVE_PARAMETER_SETS                = 129,
52     HEVC_SEI_TYPE_DECODING_UNIT_INFO                   = 130,
53     HEVC_SEI_TYPE_TEMPORAL_LEVEL0_INDEX                = 131,
54     HEVC_SEI_TYPE_DECODED_PICTURE_HASH                 = 132,
55     HEVC_SEI_TYPE_SCALABLE_NESTING                     = 133,
56     HEVC_SEI_TYPE_REGION_REFRESH_INFO                  = 134,
57     HEVC_SEI_TYPE_MASTERING_DISPLAY_INFO               = 137,
58     HEVC_SEI_TYPE_CONTENT_LIGHT_LEVEL_INFO             = 144,
59 } HEVC_SEI_Type;
60
61 typedef struct HEVCSEIPictureHash {
62     struct AVMD5 *md5_ctx;
63     uint8_t       md5[3][16];
64     uint8_t is_md5;
65 } HEVCSEIPictureHash;
66
67 typedef struct HEVCSEIFramePacking {
68     int present;
69     int arrangement_type;
70     int content_interpretation_type;
71     int quincunx_subsampling;
72 } HEVCSEIFramePacking;
73
74 typedef struct HEVCSEIDisplayOrientation {
75     int present;
76     int anticlockwise_rotation;
77     int hflip, vflip;
78 } HEVCSEIDisplayOrientation;
79
80 typedef struct HEVCSEIPictureTiming {
81     int picture_struct;
82 } HEVCSEIPictureTiming;
83
84 typedef struct HEVCSEIA53Caption {
85     int a53_caption_size;
86     uint8_t *a53_caption;
87 } HEVCSEIA53Caption;
88
89 typedef struct HEVCSEIMasteringDisplay {
90     int present;
91     uint16_t display_primaries[3][2];
92     uint16_t white_point[2];
93     uint32_t max_luminance;
94     uint32_t min_luminance;
95 } HEVCSEIMasteringDisplay;
96
97 typedef struct HEVCSEIContentLight {
98     int present;
99     uint16_t max_content_light_level;
100     uint16_t max_pic_average_light_level;
101 } HEVCSEIContentLight;
102
103 typedef struct HEVCSEIContext {
104     HEVCSEIPictureHash picture_hash;
105     HEVCSEIFramePacking frame_packing;
106     HEVCSEIDisplayOrientation display_orientation;
107     HEVCSEIPictureTiming picture_timing;
108     HEVCSEIA53Caption a53_caption;
109     HEVCSEIMasteringDisplay mastering_display;
110     HEVCSEIContentLight content_light;
111     int active_seq_parameter_set_id;
112 } HEVCSEIContext;
113
114 struct HEVCParamSets;
115
116 int ff_hevc_decode_nal_sei(GetBitContext *gb, void *logctx, HEVCSEIContext *s,
117                            const struct HEVCParamSets *ps, int type);
118
119 /**
120  * Reset SEI values that are stored on the Context.
121  * e.g. Caption data that was extracted during NAL
122  * parsing.
123  *
124  * @param s HEVCContext.
125  */
126 void ff_hevc_reset_sei(HEVCSEIContext *s);
127
128 #endif /* AVCODEC_HEVC_SEI_H */