2835b870caac2880479f05e4d79c7f3820df2b82
[ffmpeg.git] / libavcodec / h264_ps.h
1 /*
2  * This file is part of Libav.
3  *
4  * Libav is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * Libav is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with Libav; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 /**
20  * @file
21  * H.264 parameter set handling
22  */
23
24 #ifndef AVCODEC_H264_PS_H
25 #define AVCODEC_H264_PS_H
26
27 #include <stdint.h>
28
29 #include "libavutil/buffer.h"
30 #include "libavutil/pixfmt.h"
31 #include "libavutil/rational.h"
32
33 #include "avcodec.h"
34 #include "get_bits.h"
35 #include "h264.h"
36
37 #define MAX_SPS_COUNT          32
38 #define MAX_PPS_COUNT         256
39
40 /**
41  * Sequence parameter set
42  */
43 typedef struct SPS {
44     unsigned int sps_id;
45     int profile_idc;
46     int level_idc;
47     int chroma_format_idc;
48     int transform_bypass;              ///< qpprime_y_zero_transform_bypass_flag
49     int log2_max_frame_num;            ///< log2_max_frame_num_minus4 + 4
50     int poc_type;                      ///< pic_order_cnt_type
51     int log2_max_poc_lsb;              ///< log2_max_pic_order_cnt_lsb_minus4
52     int delta_pic_order_always_zero_flag;
53     int offset_for_non_ref_pic;
54     int offset_for_top_to_bottom_field;
55     int poc_cycle_length;              ///< num_ref_frames_in_pic_order_cnt_cycle
56     int ref_frame_count;               ///< num_ref_frames
57     int gaps_in_frame_num_allowed_flag;
58     int mb_width;                      ///< pic_width_in_mbs_minus1 + 1
59     int mb_height;                     ///< pic_height_in_map_units_minus1 + 1
60     int frame_mbs_only_flag;
61     int mb_aff;                        ///< mb_adaptive_frame_field_flag
62     int direct_8x8_inference_flag;
63     int crop;                          ///< frame_cropping_flag
64
65     /* those 4 are already in luma samples */
66     unsigned int crop_left;            ///< frame_cropping_rect_left_offset
67     unsigned int crop_right;           ///< frame_cropping_rect_right_offset
68     unsigned int crop_top;             ///< frame_cropping_rect_top_offset
69     unsigned int crop_bottom;          ///< frame_cropping_rect_bottom_offset
70     int vui_parameters_present_flag;
71     AVRational sar;
72     int video_signal_type_present_flag;
73     int full_range;
74     int colour_description_present_flag;
75     enum AVColorPrimaries color_primaries;
76     enum AVColorTransferCharacteristic color_trc;
77     enum AVColorSpace colorspace;
78     int timing_info_present_flag;
79     uint32_t num_units_in_tick;
80     uint32_t time_scale;
81     int fixed_frame_rate_flag;
82     short offset_for_ref_frame[256]; // FIXME dyn aloc?
83     int bitstream_restriction_flag;
84     int num_reorder_frames;
85     int scaling_matrix_present;
86     uint8_t scaling_matrix4[6][16];
87     uint8_t scaling_matrix8[6][64];
88     int nal_hrd_parameters_present_flag;
89     int vcl_hrd_parameters_present_flag;
90     int pic_struct_present_flag;
91     int time_offset_length;
92     int cpb_cnt;                          ///< See H.264 E.1.2
93     int initial_cpb_removal_delay_length; ///< initial_cpb_removal_delay_length_minus1 + 1
94     int cpb_removal_delay_length;         ///< cpb_removal_delay_length_minus1 + 1
95     int dpb_output_delay_length;          ///< dpb_output_delay_length_minus1 + 1
96     int bit_depth_luma;                   ///< bit_depth_luma_minus8 + 8
97     int bit_depth_chroma;                 ///< bit_depth_chroma_minus8 + 8
98     int residual_color_transform_flag;    ///< residual_colour_transform_flag
99     int constraint_set_flags;             ///< constraint_set[0-3]_flag
100 } SPS;
101
102 /**
103  * Picture parameter set
104  */
105 typedef struct PPS {
106     unsigned int sps_id;
107     int cabac;                  ///< entropy_coding_mode_flag
108     int pic_order_present;      ///< pic_order_present_flag
109     int slice_group_count;      ///< num_slice_groups_minus1 + 1
110     int mb_slice_group_map_type;
111     unsigned int ref_count[2];  ///< num_ref_idx_l0/1_active_minus1 + 1
112     int weighted_pred;          ///< weighted_pred_flag
113     int weighted_bipred_idc;
114     int init_qp;                ///< pic_init_qp_minus26 + 26
115     int init_qs;                ///< pic_init_qs_minus26 + 26
116     int chroma_qp_index_offset[2];
117     int deblocking_filter_parameters_present; ///< deblocking_filter_parameters_present_flag
118     int constrained_intra_pred;     ///< constrained_intra_pred_flag
119     int redundant_pic_cnt_present;  ///< redundant_pic_cnt_present_flag
120     int transform_8x8_mode;         ///< transform_8x8_mode_flag
121     uint8_t scaling_matrix4[6][16];
122     uint8_t scaling_matrix8[6][64];
123     uint8_t chroma_qp_table[2][64]; ///< pre-scaled (with chroma_qp_index_offset) version of qp_table
124     int chroma_qp_diff;
125
126     uint32_t dequant4_buffer[6][QP_MAX_NUM + 1][16];
127     uint32_t dequant8_buffer[6][QP_MAX_NUM + 1][64];
128     uint32_t(*dequant4_coeff[6])[16];
129     uint32_t(*dequant8_coeff[6])[64];
130 } PPS;
131
132 typedef struct H264ParamSets {
133     AVBufferRef *sps_list[MAX_SPS_COUNT];
134     AVBufferRef *pps_list[MAX_PPS_COUNT];
135
136     /* currently active parameters sets */
137     const PPS *pps;
138     // FIXME this should properly be const
139     SPS *sps;
140 } H264ParamSets;
141
142 /**
143  * Decode SPS
144  */
145 int ff_h264_decode_seq_parameter_set(GetBitContext *gb, AVCodecContext *avctx,
146                                      H264ParamSets *ps);
147
148 /**
149  * Decode PPS
150  */
151 int ff_h264_decode_picture_parameter_set(GetBitContext *gb, AVCodecContext *avctx,
152                                          H264ParamSets *ps, int bit_length);
153
154 #endif /* AVCODEC_H264_PS_H */