| /* |
| * drivers/amvdec_ports/decoder/aml_h264_parser.h |
| * |
| * Copyright (C) 2015 Amlogic, Inc. All rights reserved. |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 2 of the License, or |
| * (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for |
| * more details. |
| * |
| */ |
| |
| #ifndef AML_H264_PARSER_H |
| #define AML_H264_PARSER_H |
| |
| #include "../aml_vcodec_drv.h" |
| #ifdef CONFIG_AMLOGIC_MEDIA_V4L_SOFTWARE_PARSER |
| #include "../utils/pixfmt.h" |
| #endif |
| |
| #define QP_MAX_NUM (51 + 6 * 6) // The maximum supported qp |
| |
| /* NAL unit types */ |
| enum { |
| H264_NAL_SLICE = 1, |
| H264_NAL_DPA = 2, |
| H264_NAL_DPB = 3, |
| H264_NAL_DPC = 4, |
| H264_NAL_IDR_SLICE = 5, |
| H264_NAL_SEI = 6, |
| H264_NAL_SPS = 7, |
| H264_NAL_PPS = 8, |
| H264_NAL_AUD = 9, |
| H264_NAL_END_SEQUENCE = 10, |
| H264_NAL_END_STREAM = 11, |
| H264_NAL_FILLER_DATA = 12, |
| H264_NAL_SPS_EXT = 13, |
| H264_NAL_AUXILIARY_SLICE = 19, |
| }; |
| |
| enum { |
| // 7.4.2.1.1: seq_parameter_set_id is in [0, 31]. |
| H264_MAX_SPS_COUNT = 32, |
| // 7.4.2.2: pic_parameter_set_id is in [0, 255]. |
| H264_MAX_PPS_COUNT = 256, |
| |
| // A.3: MaxDpbFrames is bounded above by 16. |
| H264_MAX_DPB_FRAMES = 16, |
| // 7.4.2.1.1: max_num_ref_frames is in [0, MaxDpbFrames], and |
| // each reference frame can have two fields. |
| H264_MAX_REFS = 2 * H264_MAX_DPB_FRAMES, |
| |
| // 7.4.3.1: modification_of_pic_nums_idc is not equal to 3 at most |
| // num_ref_idx_lN_active_minus1 + 1 times (that is, once for each |
| // possible reference), then equal to 3 once. |
| H264_MAX_RPLM_COUNT = H264_MAX_REFS + 1, |
| |
| // 7.4.3.3: in the worst case, we begin with a full short-term |
| // reference picture list. Each picture in turn is moved to the |
| // long-term list (type 3) and then discarded from there (type 2). |
| // Then, we set the length of the long-term list (type 4), mark |
| // the current picture as long-term (type 6) and terminate the |
| // process (type 0). |
| H264_MAX_MMCO_COUNT = H264_MAX_REFS * 2 + 3, |
| |
| // A.2.1, A.2.3: profiles supporting FMO constrain |
| // num_slice_groups_minus1 to be in [0, 7]. |
| H264_MAX_SLICE_GROUPS = 8, |
| |
| // E.2.2: cpb_cnt_minus1 is in [0, 31]. |
| H264_MAX_CPB_CNT = 32, |
| |
| // A.3: in table A-1 the highest level allows a MaxFS of 139264. |
| H264_MAX_MB_PIC_SIZE = 139264, |
| // A.3.1, A.3.2: PicWidthInMbs and PicHeightInMbs are constrained |
| // to be not greater than sqrt(MaxFS * 8). Hence height/width are |
| // bounded above by sqrt(139264 * 8) = 1055.5 macroblocks. |
| H264_MAX_MB_WIDTH = 1055, |
| H264_MAX_MB_HEIGHT = 1055, |
| H264_MAX_WIDTH = H264_MAX_MB_WIDTH * 16, |
| H264_MAX_HEIGHT = H264_MAX_MB_HEIGHT * 16, |
| }; |
| |
| /** |
| * Rational number (pair of numerator and denominator). |
| */ |
| struct rational{ |
| int num; ///< Numerator |
| int den; ///< Denominator |
| }; |
| |
| /** |
| * Sequence parameter set |
| */ |
| struct h264_SPS_t { |
| u32 sps_id; |
| int profile_idc; |
| int level_idc; |
| int chroma_format_idc; |
| int transform_bypass; ///< qpprime_y_zero_transform_bypass_flag |
| int log2_max_frame_num; ///< log2_max_frame_num_minus4 + 4 |
| int poc_type; ///< pic_order_cnt_type |
| int log2_max_poc_lsb; ///< log2_max_pic_order_cnt_lsb_minus4 |
| int delta_pic_order_always_zero_flag; |
| int offset_for_non_ref_pic; |
| int offset_for_top_to_bottom_field; |
| int poc_cycle_length; ///< num_ref_frames_in_pic_order_cnt_cycle |
| int ref_frame_count; ///< num_ref_frames |
| int gaps_in_frame_num_allowed_flag; |
| int mb_width; ///< pic_width_in_mbs_minus1 + 1 |
| ///< (pic_height_in_map_units_minus1 + 1) * (2 - frame_mbs_only_flag) |
| int mb_height; |
| int frame_mbs_only_flag; |
| int mb_aff; ///< mb_adaptive_frame_field_flag |
| int direct_8x8_inference_flag; |
| int crop; ///< frame_cropping_flag |
| |
| /* those 4 are already in luma samples */ |
| u32 crop_left; ///< frame_cropping_rect_left_offset |
| u32 crop_right; ///< frame_cropping_rect_right_offset |
| u32 crop_top; ///< frame_cropping_rect_top_offset |
| u32 crop_bottom; ///< frame_cropping_rect_bottom_offset |
| int vui_parameters_present_flag; |
| struct rational sar; |
| int video_signal_type_present_flag; |
| int full_range; |
| int colour_description_present_flag; |
| #ifdef CONFIG_AMLOGIC_MEDIA_V4L_SOFTWARE_PARSER |
| enum AVColorPrimaries color_primaries; |
| enum AVColorTransferCharacteristic color_trc; |
| enum AVColorSpace colorspace; |
| #endif |
| int timing_info_present_flag; |
| u32 num_units_in_tick; |
| u32 time_scale; |
| int fixed_frame_rate_flag; |
| int32_t offset_for_ref_frame[256]; |
| int bitstream_restriction_flag; |
| int num_reorder_frames; |
| int max_dec_frame_buffering; |
| int scaling_matrix_present; |
| u8 scaling_matrix4[6][16]; |
| u8 scaling_matrix8[6][64]; |
| int nal_hrd_parameters_present_flag; |
| int vcl_hrd_parameters_present_flag; |
| int pic_struct_present_flag; |
| int time_offset_length; |
| int cpb_cnt; ///< See H.264 E.1.2 |
| int initial_cpb_removal_delay_length; ///< initial_cpb_removal_delay_length_minus1 + 1 |
| int cpb_removal_delay_length; ///< cpb_removal_delay_length_minus1 + 1 |
| int dpb_output_delay_length; ///< dpb_output_delay_length_minus1 + 1 |
| int bit_depth_luma; ///< bit_depth_luma_minus8 + 8 |
| int bit_depth_chroma; ///< bit_depth_chroma_minus8 + 8 |
| int residual_color_transform_flag; ///< residual_colour_transform_flag |
| int constraint_set_flags; ///< constraint_set[0-3]_flag |
| } ; |
| |
| /** |
| * Picture parameter set |
| */ |
| struct h264_PPS_t { |
| u32 sps_id; |
| int cabac; ///< entropy_coding_mode_flag |
| int pic_order_present; ///< pic_order_present_flag |
| int slice_group_count; ///< num_slice_groups_minus1 + 1 |
| int mb_slice_group_map_type; |
| u32 ref_count[2]; ///< num_ref_idx_l0/1_active_minus1 + 1 |
| int weighted_pred; ///< weighted_pred_flag |
| int weighted_bipred_idc; |
| int init_qp; ///< pic_init_qp_minus26 + 26 |
| int init_qs; ///< pic_init_qs_minus26 + 26 |
| int chroma_qp_index_offset[2]; |
| int deblocking_filter_parameters_present; ///< deblocking_filter_parameters_present_flag |
| int constrained_intra_pred; ///< constrained_intra_pred_flag |
| int redundant_pic_cnt_present; ///< redundant_pic_cnt_present_flag |
| int transform_8x8_mode; ///< transform_8x8_mode_flag |
| u8 scaling_matrix4[6][16]; |
| u8 scaling_matrix8[6][64]; |
| u8 chroma_qp_table[2][87+1]; ///< pre-scaled (with chroma_qp_index_offset) version of qp_table |
| int chroma_qp_diff; |
| u8 data[4096]; |
| int data_size; |
| |
| u32 dequant4_buffer[6][87 + 1][16]; |
| u32 dequant8_buffer[6][87 + 1][64]; |
| u32(*dequant4_coeff[6])[16]; |
| u32(*dequant8_coeff[6])[64]; |
| } ; |
| |
| struct h264_param_sets { |
| bool sps_parsed; |
| bool pps_parsed; |
| struct h264_SPS_t sps; |
| struct h264_PPS_t pps; |
| }; |
| |
| |
| #ifdef CONFIG_AMLOGIC_MEDIA_V4L_SOFTWARE_PARSER |
| int h264_decode_extradata_ps(u8 *data, int size, struct h264_param_sets *ps); |
| #else |
| inline int h264_decode_extradata_ps(u8 *data, int size, struct h264_param_sets *ps) { return -1; } |
| #endif |
| |
| #endif /* AML_H264_PARSER_H */ |
| |