/************************************************************************** * * Copyright 2017 Advanced Micro Devices, Inc. * All Rights Reserved. * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the * "Software"), to deal in the Software without restriction, including * without limitation the rights to use, copy, modify, merge, publish, * distribute, sub license, and/or sell copies of the Software, and to * permit persons to whom the Software is furnished to do so, subject to * the following conditions: * * The above copyright notice and this permission notice (including the * next paragraph) shall be included in all copies or substantial portions * of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. * IN NO EVENT SHALL THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. * **************************************************************************/ #ifndef _RADEON_VCN_ENC_H #define _RADEON_VCN_ENC_H #define RENCODE_FW_INTERFACE_MAJOR_VERSION 1 #define RENCODE_FW_INTERFACE_MINOR_VERSION 2 #define RENCODE_IB_PARAM_SESSION_INFO 0x00000001 #define RENCODE_IB_PARAM_TASK_INFO 0x00000002 #define RENCODE_IB_PARAM_SESSION_INIT 0x00000003 #define RENCODE_IB_PARAM_LAYER_CONTROL 0x00000004 #define RENCODE_IB_PARAM_LAYER_SELECT 0x00000005 #define RENCODE_IB_PARAM_RATE_CONTROL_SESSION_INIT 0x00000006 #define RENCODE_IB_PARAM_RATE_CONTROL_LAYER_INIT 0x00000007 #define RENCODE_IB_PARAM_RATE_CONTROL_PER_PICTURE 0x00000008 #define RENCODE_IB_PARAM_QUALITY_PARAMS 0x00000009 #define RENCODE_IB_PARAM_SLICE_HEADER 0x0000000a #define RENCODE_IB_PARAM_ENCODE_PARAMS 0x0000000b #define RENCODE_IB_PARAM_INTRA_REFRESH 0x0000000c #define RENCODE_IB_PARAM_ENCODE_CONTEXT_BUFFER 0x0000000d #define RENCODE_IB_PARAM_VIDEO_BITSTREAM_BUFFER 0x0000000e #define RENCODE_IB_PARAM_FEEDBACK_BUFFER 0x00000010 #define RENCODE_IB_PARAM_DIRECT_OUTPUT_NALU 0x00000020 #define RENCODE_HEVC_IB_PARAM_SLICE_CONTROL 0x00100001 #define RENCODE_HEVC_IB_PARAM_SPEC_MISC 0x00100002 #define RENCODE_HEVC_IB_PARAM_DEBLOCKING_FILTER 0x00100003 #define RENCODE_H264_IB_PARAM_SLICE_CONTROL 0x00200001 #define RENCODE_H264_IB_PARAM_SPEC_MISC 0x00200002 #define RENCODE_H264_IB_PARAM_ENCODE_PARAMS 0x00200003 #define RENCODE_H264_IB_PARAM_DEBLOCKING_FILTER 0x00200004 #define RENCODE_IB_OP_INITIALIZE 0x01000001 #define RENCODE_IB_OP_CLOSE_SESSION 0x01000002 #define RENCODE_IB_OP_ENCODE 0x01000003 #define RENCODE_IB_OP_INIT_RC 0x01000004 #define RENCODE_IB_OP_INIT_RC_VBV_BUFFER_LEVEL 0x01000005 #define RENCODE_IB_OP_SET_SPEED_ENCODING_MODE 0x01000006 #define RENCODE_IB_OP_SET_BALANCE_ENCODING_MODE 0x01000007 #define RENCODE_IB_OP_SET_QUALITY_ENCODING_MODE 0x01000008 #define RENCODE_IF_MAJOR_VERSION_MASK 0xFFFF0000 #define RENCODE_IF_MAJOR_VERSION_SHIFT 16 #define RENCODE_IF_MINOR_VERSION_MASK 0x0000FFFF #define RENCODE_IF_MINOR_VERSION_SHIFT 0 #define RENCODE_ENCODE_STANDARD_HEVC 0 #define RENCODE_ENCODE_STANDARD_H264 1 #define RENCODE_PREENCODE_MODE_NONE 0x00000000 #define RENCODE_PREENCODE_MODE_1X 0x00000001 #define RENCODE_PREENCODE_MODE_2X 0x00000002 #define RENCODE_PREENCODE_MODE_4X 0x00000004 #define RENCODE_H264_SLICE_CONTROL_MODE_FIXED_MBS 0x00000000 #define RENCODE_H264_SLICE_CONTROL_MODE_FIXED_BITS 0x00000001 #define RENCODE_HEVC_SLICE_CONTROL_MODE_FIXED_CTBS 0x00000000 #define RENCODE_HEVC_SLICE_CONTROL_MODE_FIXED_BITS 0x00000001 #define RENCODE_RATE_CONTROL_METHOD_NONE 0x00000000 #define RENCODE_RATE_CONTROL_METHOD_LATENCY_CONSTRAINED_VBR 0x00000001 #define RENCODE_RATE_CONTROL_METHOD_PEAK_CONSTRAINED_VBR 0x00000002 #define RENCODE_RATE_CONTROL_METHOD_CBR 0x00000003 #define RENCODE_DIRECT_OUTPUT_NALU_TYPE_AUD 0x00000000 #define RENCODE_DIRECT_OUTPUT_NALU_TYPE_VPS 0x00000001 #define RENCODE_DIRECT_OUTPUT_NALU_TYPE_SPS 0x00000002 #define RENCODE_DIRECT_OUTPUT_NALU_TYPE_PPS 0x00000003 #define RENCODE_DIRECT_OUTPUT_NALU_TYPE_PREFIX 0x00000004 #define RENCODE_DIRECT_OUTPUT_NALU_TYPE_END_OF_SEQUENCE 0x00000005 #define RENCODE_SLICE_HEADER_TEMPLATE_MAX_TEMPLATE_SIZE_IN_DWORDS 16 #define RENCODE_SLICE_HEADER_TEMPLATE_MAX_NUM_INSTRUCTIONS 16 #define RENCODE_HEADER_INSTRUCTION_END 0x00000000 #define RENCODE_HEADER_INSTRUCTION_COPY 0x00000001 #define RENCODE_HEVC_HEADER_INSTRUCTION_DEPENDENT_SLICE_END 0x00010000 #define RENCODE_HEVC_HEADER_INSTRUCTION_FIRST_SLICE 0x00010001 #define RENCODE_HEVC_HEADER_INSTRUCTION_SLICE_SEGMENT 0x00010002 #define RENCODE_HEVC_HEADER_INSTRUCTION_SLICE_QP_DELTA 0x00010003 #define RENCODE_H264_HEADER_INSTRUCTION_FIRST_MB 0x00020000 #define RENCODE_H264_HEADER_INSTRUCTION_SLICE_QP_DELTA 0x00020001 #define RENCODE_PICTURE_TYPE_B 0 #define RENCODE_PICTURE_TYPE_P 1 #define RENCODE_PICTURE_TYPE_I 2 #define RENCODE_PICTURE_TYPE_P_SKIP 3 #define RENCODE_INPUT_SWIZZLE_MODE_LINEAR 0 #define RENCODE_INPUT_SWIZZLE_MODE_256B_S 1 #define RENCODE_INPUT_SWIZZLE_MODE_4kB_S 5 #define RENCODE_INPUT_SWIZZLE_MODE_64kB_S 9 #define RENCODE_H264_PICTURE_STRUCTURE_FRAME 0 #define RENCODE_H264_PICTURE_STRUCTURE_TOP_FIELD 1 #define RENCODE_H264_PICTURE_STRUCTURE_BOTTOM_FIELD 2 #define RENCODE_H264_INTERLACING_MODE_PROGRESSIVE 0 #define RENCODE_H264_INTERLACING_MODE_INTERLACED_STACKED 1 #define RENCODE_H264_INTERLACING_MODE_INTERLACED_INTERLEAVED 2 #define RENCODE_H264_DISABLE_DEBLOCKING_FILTER_IDC_ENABLE 0 #define RENCODE_H264_DISABLE_DEBLOCKING_FILTER_IDC_DISABLE 1 #define RENCODE_H264_DISABLE_DEBLOCKING_FILTER_IDC_DISALBE_ACROSS_SLICE_BOUNDARY 2 #define RENCODE_INTRA_REFRESH_MODE_NONE 0 #define RENCODE_INTRA_REFRESH_MODE_CTB_MB_ROWS 1 #define RENCODE_INTRA_REFRESH_MODE_CTB_MB_COLUMNS 2 #define RENCODE_MAX_NUM_RECONSTRUCTED_PICTURES 34 #define RENCODE_REC_SWIZZLE_MODE_LINEAR 0 #define RENCODE_REC_SWIZZLE_MODE_256B_S 1 #define RENCODE_VIDEO_BITSTREAM_BUFFER_MODE_LINEAR 0 #define RENCODE_VIDEO_BITSTREAM_BUFFER_MODE_CIRCULAR 1 #define RENCODE_FEEDBACK_BUFFER_MODE_LINEAR 0 #define RENCODE_FEEDBACK_BUFFER_MODE_CIRCULAR 1 typedef struct rvcn_enc_session_info_s { uint32_t interface_version; uint32_t sw_context_address_hi; uint32_t sw_context_address_lo; } rvcn_enc_session_info_t; typedef struct rvcn_enc_task_info_s { uint32_t total_size_of_all_packages; uint32_t task_id; uint32_t allowed_max_num_feedbacks; } rvcn_enc_task_info_t; typedef struct rvcn_enc_session_init_s { uint32_t encode_standard; uint32_t aligned_picture_width; uint32_t aligned_picture_height; uint32_t padding_width; uint32_t padding_height; uint32_t pre_encode_mode; uint32_t pre_encode_chroma_enabled; } rvcn_enc_session_init_t; typedef struct rvcn_enc_layer_control_s { uint32_t max_num_temporal_layers; uint32_t num_temporal_layers; } rvcn_enc_layer_control_t; typedef struct rvcn_enc_layer_select_s { uint32_t temporal_layer_index; } rvcn_enc_layer_select_t; typedef struct rvcn_enc_h264_slice_control_s { uint32_t slice_control_mode; union { uint32_t num_mbs_per_slice; uint32_t num_bits_per_slice; }; } rvcn_enc_h264_slice_control_t; typedef struct rvcn_enc_hevc_slice_control_s { uint32_t slice_control_mode; union { struct { uint32_t num_ctbs_per_slice; uint32_t num_ctbs_per_slice_segment; } fixed_ctbs_per_slice; struct { uint32_t num_bits_per_slice; uint32_t num_bits_per_slice_segment; } fixed_bits_per_slice; }; } rvcn_enc_hevc_slice_control_t; typedef struct rvcn_enc_h264_spec_misc_s { uint32_t constrained_intra_pred_flag; uint32_t cabac_enable; uint32_t cabac_init_idc; uint32_t half_pel_enabled; uint32_t quarter_pel_enabled; uint32_t profile_idc; uint32_t level_idc; } rvcn_enc_h264_spec_misc_t; typedef struct rvcn_enc_hevc_spec_misc_s { uint32_t log2_min_luma_coding_block_size_minus3; uint32_t amp_disabled; uint32_t strong_intra_smoothing_enabled; uint32_t constrained_intra_pred_flag; uint32_t cabac_init_flag; uint32_t half_pel_enabled; uint32_t quarter_pel_enabled; } rvcn_enc_hevc_spec_misc_t; typedef struct rvcn_enc_rate_ctl_session_init_s { uint32_t rate_control_method; uint32_t vbv_buffer_level; } rvcn_enc_rate_ctl_session_init_t; typedef struct rvcn_enc_rate_ctl_layer_init_s { uint32_t target_bit_rate; uint32_t peak_bit_rate; uint32_t frame_rate_num; uint32_t frame_rate_den; uint32_t vbv_buffer_size; uint32_t avg_target_bits_per_picture; uint32_t peak_bits_per_picture_integer; uint32_t peak_bits_per_picture_fractional; } rvcn_enc_rate_ctl_layer_init_t; typedef struct rvcn_enc_rate_ctl_per_picture_s { uint32_t qp; uint32_t min_qp_app; uint32_t max_qp_app; uint32_t max_au_size; uint32_t enabled_filler_data; uint32_t skip_frame_enable; uint32_t enforce_hrd; } rvcn_enc_rate_ctl_per_picture_t; typedef struct rvcn_enc_quality_params_s { uint32_t vbaq_mode; uint32_t scene_change_sensitivity; uint32_t scene_change_min_idr_interval; } rvcn_enc_quality_params_t; typedef struct rvcn_enc_direct_output_nalu_s { uint32_t type; uint32_t size; uint32_t data[1]; } rvcn_enc_direct_output_nalu_t; typedef struct rvcn_enc_slice_header_s { uint32_t bitstream_template[RENCODE_SLICE_HEADER_TEMPLATE_MAX_TEMPLATE_SIZE_IN_DWORDS]; struct { uint32_t instruction; uint32_t num_bits; } instructions[RENCODE_SLICE_HEADER_TEMPLATE_MAX_NUM_INSTRUCTIONS]; } rvcn_enc_slice_header_t; typedef struct rvcn_enc_encode_params_s { uint32_t pic_type; uint32_t allowed_max_bitstream_size; uint32_t input_picture_luma_address_hi; uint32_t input_picture_luma_address_lo; uint32_t input_picture_chroma_address_hi; uint32_t input_picture_chroma_address_lo; uint32_t input_pic_luma_pitch; uint32_t input_pic_chroma_pitch; uint8_t input_pic_swizzle_mode; uint32_t reference_picture_index; uint32_t reconstructed_picture_index; } rvcn_enc_encode_params_t; typedef struct rvcn_enc_h264_encode_params_s { uint32_t input_picture_structure; uint32_t interlaced_mode; uint32_t reference_picture_structure; uint32_t reference_picture1_index; } rvcn_enc_h264_encode_params_t; typedef struct rvcn_enc_h264_deblocking_filter_s { uint32_t disable_deblocking_filter_idc; int32_t alpha_c0_offset_div2; int32_t beta_offset_div2; int32_t cb_qp_offset; int32_t cr_qp_offset; } rvcn_enc_h264_deblocking_filter_t; typedef struct rvcn_enc_hevc_deblocking_filter_s { uint32_t loop_filter_across_slices_enabled; int32_t deblocking_filter_disabled; int32_t beta_offset_div2; int32_t tc_offset_div2; int32_t cb_qp_offset; int32_t cr_qp_offset; } rvcn_enc_hevc_deblocking_filter_t; typedef struct rvcn_enc_intra_refresh_s { uint32_t intra_refresh_mode; uint32_t offset; uint32_t region_size; } rvcn_enc_intra_refresh_t; typedef struct rvcn_enc_reconstructed_picture_s { uint32_t luma_offset; uint32_t chroma_offset; } rvcn_enc_reconstructed_picture_t; typedef struct rvcn_enc_encode_context_buffer_s { uint32_t encode_context_address_hi; uint32_t encode_context_address_lo; uint32_t swizzle_mode; uint32_t rec_luma_pitch; uint32_t rec_chroma_pitch; uint32_t num_reconstructed_pictures; rvcn_enc_reconstructed_picture_t reconstructed_pictures[RENCODE_MAX_NUM_RECONSTRUCTED_PICTURES]; uint32_t pre_encode_picture_luma_pitch; uint32_t pre_encode_picture_chroma_pitch; rvcn_enc_reconstructed_picture_t pre_encode_reconstructed_pictures[RENCODE_MAX_NUM_RECONSTRUCTED_PICTURES]; rvcn_enc_reconstructed_picture_t pre_encode_input_picture; } rvcn_enc_encode_context_buffer_t; typedef struct rvcn_enc_video_bitstream_buffer_s { uint32_t mode; uint32_t video_bitstream_buffer_address_hi; uint32_t video_bitstream_buffer_address_lo; uint32_t video_bitstream_buffer_size; uint32_t video_bitstream_data_offset; } rvcn_enc_video_bitstream_buffer_t; typedef struct rvcn_enc_feedback_buffer_s { uint32_t mode; uint32_t feedback_buffer_address_hi; uint32_t feedback_buffer_address_lo; uint32_t feedback_buffer_size; uint32_t feedback_data_size; } rvcn_enc_feedback_buffer_t; typedef void (*radeon_enc_get_buffer)(struct pipe_resource *resource, struct pb_buffer **handle, struct radeon_surf **surface); struct pipe_video_codec *radeon_create_encoder(struct pipe_context *context, const struct pipe_video_codec *templat, struct radeon_winsys* ws, radeon_enc_get_buffer get_buffer); struct radeon_enc_pic { enum pipe_h264_enc_picture_type picture_type; unsigned frame_num; unsigned pic_order_cnt; unsigned pic_order_cnt_type; unsigned ref_idx_l0; unsigned ref_idx_l1; unsigned crop_left; unsigned crop_right; unsigned crop_top; unsigned crop_bottom; unsigned general_tier_flag; unsigned general_profile_idc; unsigned general_level_idc; unsigned max_poc; unsigned log2_max_poc; unsigned chroma_format_idc; unsigned pic_width_in_luma_samples; unsigned pic_height_in_luma_samples; unsigned log2_diff_max_min_luma_coding_block_size; unsigned log2_min_transform_block_size_minus2; unsigned log2_diff_max_min_transform_block_size; unsigned max_transform_hierarchy_depth_inter; unsigned max_transform_hierarchy_depth_intra; unsigned log2_parallel_merge_level_minus2; unsigned bit_depth_luma_minus8; unsigned bit_depth_chroma_minus8; unsigned nal_unit_type; unsigned max_num_merge_cand; bool not_referenced; bool is_idr; bool is_even_frame; bool sample_adaptive_offset_enabled_flag; bool pcm_enabled_flag; bool sps_temporal_mvp_enabled_flag; rvcn_enc_task_info_t task_info; rvcn_enc_session_init_t session_init; rvcn_enc_layer_control_t layer_ctrl; rvcn_enc_layer_select_t layer_sel; rvcn_enc_h264_slice_control_t slice_ctrl; rvcn_enc_hevc_slice_control_t hevc_slice_ctrl; rvcn_enc_h264_spec_misc_t spec_misc; rvcn_enc_hevc_spec_misc_t hevc_spec_misc; rvcn_enc_rate_ctl_session_init_t rc_session_init; rvcn_enc_rate_ctl_layer_init_t rc_layer_init; rvcn_enc_h264_encode_params_t h264_enc_params; rvcn_enc_h264_deblocking_filter_t h264_deblock; rvcn_enc_hevc_deblocking_filter_t hevc_deblock; rvcn_enc_rate_ctl_per_picture_t rc_per_pic; rvcn_enc_quality_params_t quality_params; rvcn_enc_encode_context_buffer_t ctx_buf; rvcn_enc_video_bitstream_buffer_t bit_buf; rvcn_enc_feedback_buffer_t fb_buf; rvcn_enc_intra_refresh_t intra_ref; rvcn_enc_encode_params_t enc_params; }; struct radeon_encoder { struct pipe_video_codec base; void (*begin)(struct radeon_encoder *enc, struct pipe_picture_desc *pic); void (*encode)(struct radeon_encoder *enc); void (*destroy)(struct radeon_encoder *enc); unsigned stream_handle; struct pipe_screen *screen; struct radeon_winsys* ws; struct radeon_winsys_cs* cs; radeon_enc_get_buffer get_buffer; struct pb_buffer* handle; struct radeon_surf* luma; struct radeon_surf* chroma; struct pb_buffer* bs_handle; unsigned bs_size; unsigned cpb_num; struct rvid_buffer *si; struct rvid_buffer *fb; struct rvid_buffer cpb; struct radeon_enc_pic enc_pic; unsigned alignment; unsigned shifter; unsigned bits_in_shifter; unsigned num_zeros; unsigned byte_index; unsigned bits_output; uint32_t total_task_size; uint32_t* p_task_size; bool emulation_prevention; bool need_feedback; }; void radeon_enc_1_2_init(struct radeon_encoder *enc); #endif // _RADEON_VCN_ENC_H