/* SPDX-License-Identifier: GPL-2.0 */ /* * Hantro VPU codec driver * * Copyright 2018 Google LLC. * Tomasz Figa <[email protected]> */ #ifndef HANTRO_HW_H_ #define HANTRO_HW_H_ #include <linux/interrupt.h> #include <linux/v4l2-controls.h> #include <media/v4l2-ctrls.h> #include <media/v4l2-vp9.h> #include <media/videobuf2-core.h> #include "rockchip_av1_entropymode.h" #include "rockchip_av1_filmgrain.h" #define DEC_8190_ALIGN_MASK … #define MB_DIM … #define TILE_MB_DIM … #define MB_WIDTH(w) … #define MB_HEIGHT(h) … #define FMT_MIN_WIDTH … #define FMT_MIN_HEIGHT … #define FMT_HD_WIDTH … #define FMT_HD_HEIGHT … #define FMT_FHD_WIDTH … #define FMT_FHD_HEIGHT … #define FMT_UHD_WIDTH … #define FMT_UHD_HEIGHT … #define FMT_4K_WIDTH … #define FMT_4K_HEIGHT … #define NUM_REF_PICTURES … #define AV1_MAX_FRAME_BUF_COUNT … #define MAX_POSTPROC_BUFFERS … struct hantro_dev; struct hantro_ctx; struct hantro_buf; struct hantro_variant; /** * struct hantro_aux_buf - auxiliary DMA buffer for hardware data * * @cpu: CPU pointer to the buffer. * @dma: DMA address of the buffer. * @size: Size of the buffer. * @attrs: Attributes of the DMA mapping. */ struct hantro_aux_buf { … }; /* Max. number of entries in the DPB (HW limitation). */ #define HANTRO_H264_DPB_SIZE … /** * struct hantro_h264_dec_ctrls * * @decode: Decode params * @scaling: Scaling info * @sps: SPS info * @pps: PPS info */ struct hantro_h264_dec_ctrls { … }; /** * struct hantro_h264_dec_reflists * * @p: P reflist * @b0: B0 reflist * @b1: B1 reflist */ struct hantro_h264_dec_reflists { … }; /** * struct hantro_h264_dec_hw_ctx * * @priv: Private auxiliary buffer for hardware. * @dpb: DPB * @reflists: P/B0/B1 reflists * @ctrls: V4L2 controls attached to a run * @dpb_longterm: DPB long-term * @dpb_valid: DPB valid * @cur_poc: Current picture order count */ struct hantro_h264_dec_hw_ctx { … }; /** * struct hantro_hevc_dec_ctrls * @decode_params: Decode params * @scaling: Scaling matrix * @sps: SPS info * @pps: PPS info * @hevc_hdr_skip_length: the number of data (in bits) to skip in the * slice segment header syntax after 'slice type' * token */ struct hantro_hevc_dec_ctrls { … }; /** * struct hantro_hevc_dec_hw_ctx * @tile_sizes: Tile sizes buffer * @tile_filter: Tile vertical filter buffer * @tile_sao: Tile SAO buffer * @tile_bsd: Tile BSD control buffer * @ref_bufs: Internal reference buffers * @scaling_lists: Scaling lists buffer * @ref_bufs_poc: Internal reference buffers picture order count * @ref_bufs_used: Bitfield of used reference buffers * @ctrls: V4L2 controls attached to a run * @num_tile_cols_allocated: number of allocated tiles */ struct hantro_hevc_dec_hw_ctx { … }; /** * struct hantro_mpeg2_dec_hw_ctx * * @qtable: Quantization table */ struct hantro_mpeg2_dec_hw_ctx { … }; /** * struct hantro_vp8_dec_hw_ctx * * @segment_map: Segment map buffer. * @prob_tbl: Probability table buffer. */ struct hantro_vp8_dec_hw_ctx { … }; /** * struct hantro_vp9_frame_info * * @valid: frame info valid flag * @frame_context_idx: index of frame context * @reference_mode: inter prediction type * @tx_mode: transform mode * @interpolation_filter: filter selection for inter prediction * @flags: frame flags * @timestamp: frame timestamp */ struct hantro_vp9_frame_info { … }; #define MAX_SB_COLS … #define MAX_SB_ROWS … /** * struct hantro_vp9_dec_hw_ctx * * @tile_edge: auxiliary DMA buffer for tile edge processing * @segment_map: auxiliary DMA buffer for segment map * @misc: auxiliary DMA buffer for tile info, probabilities and hw counters * @cnts: vp9 library struct for abstracting hw counters access * @probability_tables: VP9 probability tables implied by the spec * @frame_context: VP9 frame contexts * @cur: current frame information * @last: last frame information * @bsd_ctrl_offset: bsd offset into tile_edge * @segment_map_size: size of segment map * @ctx_counters_offset: hw counters offset into misc * @tile_info_offset: tile info offset into misc * @tile_r_info: per-tile information array * @tile_c_info: per-tile information array * @last_tile_r: last number of tile rows * @last_tile_c: last number of tile cols * @last_sbs_r: last number of superblock rows * @last_sbs_c: last number of superblock cols * @active_segment: number of active segment (alternating between 0 and 1) * @feature_enabled: segmentation feature enabled flags * @feature_data: segmentation feature data */ struct hantro_vp9_dec_hw_ctx { … }; /** * struct hantro_av1_dec_ctrls * @sequence: AV1 Sequence * @tile_group_entry: AV1 Tile Group entry * @frame: AV1 Frame Header OBU * @film_grain: AV1 Film Grain */ struct hantro_av1_dec_ctrls { … }; struct hantro_av1_frame_ref { … }; /** * struct hantro_av1_dec_hw_ctx * @db_data_col: db tile col data buffer * @db_ctrl_col: db tile col ctrl buffer * @cdef_col: cdef tile col buffer * @sr_col: sr tile col buffer * @lr_col: lr tile col buffer * @global_model: global model buffer * @tile_info: tile info buffer * @segment: segmentation info buffer * @film_grain: film grain buffer * @prob_tbl: probability table * @prob_tbl_out: probability table output * @tile_buf: tile buffer * @ctrls: V4L2 controls attached to a run * @frame_refs: reference frames info slots * @ref_frame_sign_bias: array of sign bias * @num_tile_cols_allocated: number of allocated tiles * @cdfs: current probabilities structure * @cdfs_ndvc: current mv probabilities structure * @default_cdfs: default probabilities structure * @default_cdfs_ndvc: default mv probabilties structure * @cdfs_last: stored probabilities structures * @cdfs_last_ndvc: stored mv probabilities structures * @current_frame_index: index of the current in frame_refs array */ struct hantro_av1_dec_hw_ctx { … }; /** * struct hantro_postproc_ctx * * @dec_q: References buffers, in decoder format. */ struct hantro_postproc_ctx { … }; /** * struct hantro_postproc_ops - post-processor operations * * @enable: Enable the post-processor block. Optional. * @disable: Disable the post-processor block. Optional. * @enum_framesizes: Enumerate possible scaled output formats. * Returns zero if OK, a negative value in error cases. * Optional. */ struct hantro_postproc_ops { … }; /** * struct hantro_codec_ops - codec mode specific operations * * @init: If needed, can be used for initialization. * Optional and called from process context. * @exit: If needed, can be used to undo the .init phase. * Optional and called from process context. * @run: Start single {en,de)coding job. Called from atomic context * to indicate that a pair of buffers is ready and the hardware * should be programmed and started. Returns zero if OK, a * negative value in error cases. * @done: Read back processing results and additional data from hardware. * @reset: Reset the hardware in case of a timeout. */ struct hantro_codec_ops { … }; /** * enum hantro_enc_fmt - source format ID for hardware registers. * * @ROCKCHIP_VPU_ENC_FMT_YUV420P: Y/CbCr 4:2:0 planar format * @ROCKCHIP_VPU_ENC_FMT_YUV420SP: Y/CbCr 4:2:0 semi-planar format * @ROCKCHIP_VPU_ENC_FMT_YUYV422: YUV 4:2:2 packed format (YUYV) * @ROCKCHIP_VPU_ENC_FMT_UYVY422: YUV 4:2:2 packed format (UYVY) */ enum hantro_enc_fmt { … }; extern const struct hantro_variant imx8mm_vpu_g1_variant; extern const struct hantro_variant imx8mq_vpu_g1_variant; extern const struct hantro_variant imx8mq_vpu_g2_variant; extern const struct hantro_variant imx8mq_vpu_variant; extern const struct hantro_variant px30_vpu_variant; extern const struct hantro_variant rk3036_vpu_variant; extern const struct hantro_variant rk3066_vpu_variant; extern const struct hantro_variant rk3288_vpu_variant; extern const struct hantro_variant rk3328_vpu_variant; extern const struct hantro_variant rk3399_vpu_variant; extern const struct hantro_variant rk3568_vepu_variant; extern const struct hantro_variant rk3568_vpu_variant; extern const struct hantro_variant rk3588_vpu981_variant; extern const struct hantro_variant sama5d4_vdec_variant; extern const struct hantro_variant sunxi_vpu_variant; extern const struct hantro_variant stm32mp25_vdec_variant; extern const struct hantro_variant stm32mp25_venc_variant; extern const struct hantro_postproc_ops hantro_g1_postproc_ops; extern const struct hantro_postproc_ops hantro_g2_postproc_ops; extern const struct hantro_postproc_ops rockchip_vpu981_postproc_ops; extern const u32 hantro_vp8_dec_mc_filter[8][6]; void hantro_watchdog(struct work_struct *work); void hantro_run(struct hantro_ctx *ctx); void hantro_irq_done(struct hantro_dev *vpu, enum vb2_buffer_state result); void hantro_start_prepare_run(struct hantro_ctx *ctx); void hantro_end_prepare_run(struct hantro_ctx *ctx); irqreturn_t hantro_g1_irq(int irq, void *dev_id); void hantro_g1_reset(struct hantro_ctx *ctx); int hantro_h1_jpeg_enc_run(struct hantro_ctx *ctx); int rockchip_vpu2_jpeg_enc_run(struct hantro_ctx *ctx); void hantro_h1_jpeg_enc_done(struct hantro_ctx *ctx); void rockchip_vpu2_jpeg_enc_done(struct hantro_ctx *ctx); dma_addr_t hantro_h264_get_ref_buf(struct hantro_ctx *ctx, unsigned int dpb_idx); u16 hantro_h264_get_ref_nbr(struct hantro_ctx *ctx, unsigned int dpb_idx); int hantro_h264_dec_prepare_run(struct hantro_ctx *ctx); int rockchip_vpu2_h264_dec_run(struct hantro_ctx *ctx); int hantro_g1_h264_dec_run(struct hantro_ctx *ctx); int hantro_h264_dec_init(struct hantro_ctx *ctx); void hantro_h264_dec_exit(struct hantro_ctx *ctx); int hantro_hevc_dec_init(struct hantro_ctx *ctx); void hantro_hevc_dec_exit(struct hantro_ctx *ctx); int hantro_g2_hevc_dec_run(struct hantro_ctx *ctx); int hantro_hevc_dec_prepare_run(struct hantro_ctx *ctx); void hantro_hevc_ref_init(struct hantro_ctx *ctx); dma_addr_t hantro_hevc_get_ref_buf(struct hantro_ctx *ctx, s32 poc); int hantro_hevc_add_ref_buf(struct hantro_ctx *ctx, int poc, dma_addr_t addr); int rockchip_vpu981_av1_dec_init(struct hantro_ctx *ctx); void rockchip_vpu981_av1_dec_exit(struct hantro_ctx *ctx); int rockchip_vpu981_av1_dec_run(struct hantro_ctx *ctx); void rockchip_vpu981_av1_dec_done(struct hantro_ctx *ctx); static inline unsigned short hantro_vp9_num_sbs(unsigned short dimension) { … } static inline size_t hantro_vp9_mv_size(unsigned int width, unsigned int height) { … } static inline size_t hantro_h264_mv_size(unsigned int width, unsigned int height) { … } static inline size_t hantro_hevc_mv_size(unsigned int width, unsigned int height) { … } static inline unsigned short hantro_av1_num_sbs(unsigned short dimension) { … } static inline size_t hantro_av1_mv_size(unsigned int width, unsigned int height) { … } size_t hantro_g2_chroma_offset(struct hantro_ctx *ctx); size_t hantro_g2_motion_vectors_offset(struct hantro_ctx *ctx); int hantro_g1_mpeg2_dec_run(struct hantro_ctx *ctx); int rockchip_vpu2_mpeg2_dec_run(struct hantro_ctx *ctx); void hantro_mpeg2_dec_copy_qtable(u8 *qtable, const struct v4l2_ctrl_mpeg2_quantisation *ctrl); int hantro_mpeg2_dec_init(struct hantro_ctx *ctx); void hantro_mpeg2_dec_exit(struct hantro_ctx *ctx); int hantro_g1_vp8_dec_run(struct hantro_ctx *ctx); int rockchip_vpu2_vp8_dec_run(struct hantro_ctx *ctx); int hantro_vp8_dec_init(struct hantro_ctx *ctx); void hantro_vp8_dec_exit(struct hantro_ctx *ctx); void hantro_vp8_prob_update(struct hantro_ctx *ctx, const struct v4l2_ctrl_vp8_frame *hdr); int hantro_g2_vp9_dec_run(struct hantro_ctx *ctx); void hantro_g2_vp9_dec_done(struct hantro_ctx *ctx); int hantro_vp9_dec_init(struct hantro_ctx *ctx); void hantro_vp9_dec_exit(struct hantro_ctx *ctx); void hantro_g2_check_idle(struct hantro_dev *vpu); irqreturn_t hantro_g2_irq(int irq, void *dev_id); #endif /* HANTRO_HW_H_ */
Codebase Browser
linux