libavcodec/h264_ps.h

   1 /*
   2  * This file is part of FFmpeg.
   3  *
   4  * FFmpeg 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  * FFmpeg 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 FFmpeg; 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/pixfmt.h"
  30 #include "libavutil/rational.h"
  31
  32 #include "avcodec.h"
  33 #include "get_bits.h"
  34 #include "h264.h"
  35 #include "h2645_vui.h"
  36
  37 #define MAX_SPS_COUNT          32
  38 #define MAX_PPS_COUNT         256
  39 #define MAX_LOG2_MAX_FRAME_NUM    (12 + 4)
  40
  41 /**
  42  * Sequence parameter set
  43  */
  44 typedef struct SPS {
  45     unsigned int sps_id;
  46     int profile_idc;
  47     int level_idc;
  48     int chroma_format_idc;
  49     int transform_bypass;              ///< qpprime_y_zero_transform_bypass_flag
  50     int log2_max_frame_num;            ///< log2_max_frame_num_minus4 + 4
  51     int poc_type;                      ///< pic_order_cnt_type
  52     int log2_max_poc_lsb;              ///< log2_max_pic_order_cnt_lsb_minus4
  53     int delta_pic_order_always_zero_flag;
  54     int offset_for_non_ref_pic;
  55     int offset_for_top_to_bottom_field;
  56     int poc_cycle_length;              ///< num_ref_frames_in_pic_order_cnt_cycle
  57     int ref_frame_count;               ///< num_ref_frames
  58     int gaps_in_frame_num_allowed_flag;
  59     int mb_width;                      ///< pic_width_in_mbs_minus1 + 1
  60     ///< (pic_height_in_map_units_minus1 + 1) * (2 - frame_mbs_only_flag)
  61     int mb_height;
  62     int frame_mbs_only_flag;
  63     int mb_aff;                        ///< mb_adaptive_frame_field_flag
  64     int direct_8x8_inference_flag;
  65     int crop;                          ///< frame_cropping_flag
  66
  67     /* those 4 are already in luma samples */
  68     unsigned int crop_left;            ///< frame_cropping_rect_left_offset
  69     unsigned int crop_right;           ///< frame_cropping_rect_right_offset
  70     unsigned int crop_top;             ///< frame_cropping_rect_top_offset
  71     unsigned int crop_bottom;          ///< frame_cropping_rect_bottom_offset
  72     int vui_parameters_present_flag;
  73     H2645VUI vui;
  74
  75     int timing_info_present_flag;
  76     uint32_t num_units_in_tick;
  77     uint32_t time_scale;
  78     int fixed_frame_rate_flag;
  79     int32_t offset_for_ref_frame[256];
  80     int bitstream_restriction_flag;
  81     int num_reorder_frames;
  82     int max_dec_frame_buffering;
  83     int scaling_matrix_present;
  84     uint16_t scaling_matrix_present_mask;
  85     uint8_t scaling_matrix4[6][16];
  86     uint8_t scaling_matrix8[6][64];
  87     int nal_hrd_parameters_present_flag;
  88     int vcl_hrd_parameters_present_flag;
  89     int pic_struct_present_flag;
  90     int time_offset_length;
  91     int cpb_cnt;                          ///< See H.264 E.1.2
  92     int bit_rate_scale;
  93     uint32_t bit_rate_value[32];          ///< bit_rate_value_minus1 + 1
  94     uint32_t cpb_size_value[32];          ///< cpb_size_value_minus1 + 1
  95     uint32_t cpr_flag;
  96     int initial_cpb_removal_delay_length; ///< initial_cpb_removal_delay_length_minus1 + 1
  97     int cpb_removal_delay_length;         ///< cpb_removal_delay_length_minus1 + 1
  98     int dpb_output_delay_length;          ///< dpb_output_delay_length_minus1 + 1
  99     int bit_depth_luma;                   ///< bit_depth_luma_minus8 + 8
 100     int bit_depth_chroma;                 ///< bit_depth_chroma_minus8 + 8
 101     int residual_color_transform_flag;    ///< residual_colour_transform_flag
 102     int constraint_set_flags;             ///< constraint_set[0-3]_flag
 103     uint8_t data[4096];
 104     size_t data_size;
 105 } SPS;
 106
 107 /**
 108  * Picture parameter set
 109  */
 110 typedef struct PPS {
 111     unsigned int pps_id;
 112     unsigned int sps_id;
 113     int cabac;                  ///< entropy_coding_mode_flag
 114     int pic_order_present;      ///< bottom_field_pic_order_in_frame_present_flag
 115     int slice_group_count;      ///< num_slice_groups_minus1 + 1
 116     int mb_slice_group_map_type;
 117     unsigned int ref_count[2];  ///< num_ref_idx_l0/1_active_minus1 + 1
 118     int weighted_pred;          ///< weighted_pred_flag
 119     int weighted_bipred_idc;
 120     int init_qp;                ///< pic_init_qp_minus26 + 26
 121     int init_qs;                ///< pic_init_qs_minus26 + 26
 122     int chroma_qp_index_offset[2];
 123     int deblocking_filter_parameters_present; ///< deblocking_filter_parameters_present_flag
 124     int constrained_intra_pred;     ///< constrained_intra_pred_flag
 125     int redundant_pic_cnt_present;  ///< redundant_pic_cnt_present_flag
 126     int transform_8x8_mode;         ///< transform_8x8_mode_flag
 127     int pic_scaling_matrix_present_flag;
 128     uint16_t pic_scaling_matrix_present_mask;
 129     uint8_t scaling_matrix4[6][16];
 130     uint8_t scaling_matrix8[6][64];
 131     uint8_t chroma_qp_table[2][QP_MAX_NUM+1];  ///< pre-scaled (with chroma_qp_index_offset) version of qp_table
 132     int chroma_qp_diff;
 133     uint8_t data[4096];
 134     size_t data_size;
 135
 136     uint32_t dequant4_buffer[6][QP_MAX_NUM + 1][16];
 137     uint32_t dequant8_buffer[6][QP_MAX_NUM + 1][64];
 138     uint32_t(*dequant4_coeff[6])[16];
 139     uint32_t(*dequant8_coeff[6])[64];
 140
 141     const SPS   *sps; ///< RefStruct reference
 142 } PPS;
 143
 144 typedef struct H264ParamSets {
 145     const SPS *sps_list[MAX_SPS_COUNT]; ///< RefStruct references
 146     const PPS *pps_list[MAX_PPS_COUNT]; ///< RefStruct references
 147
 148     /* currently active parameters sets */
 149     const PPS *pps; ///< RefStruct reference
 150     const SPS *sps; ///< ordinary pointer, no RefStruct reference
 151
 152     int overread_warning_printed[2];
 153 } H264ParamSets;
 154
 155 /**
 156  * compute profile from sps
 157  */
 158 int ff_h264_get_profile(const SPS *sps);
 159
 160 /**
 161  * Decode SPS
 162  */
 163 int ff_h264_decode_seq_parameter_set(GetBitContext *gb, AVCodecContext *avctx,
 164                                      H264ParamSets *ps, int ignore_truncation);
 165
 166 /**
 167  * Decode PPS
 168  */
 169 int ff_h264_decode_picture_parameter_set(GetBitContext *gb, AVCodecContext *avctx,
 170                                          H264ParamSets *ps, int bit_length);
 171
 172 /**
 173  * Uninit H264 param sets structure.
 174  */
 175 void ff_h264_ps_uninit(H264ParamSets *ps);
 176
 177 #endif /* AVCODEC_H264_PS_H */