Quelle  stm32mp25_vpu_hw.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * STM32MP25 video codec driver
 *
 * Copyright (C) STMicroelectronics SA 2024
 * Authors: Hugues Fruchet <hugues.fruchet@foss.st.com>
 *          for STMicroelectronics.
 *
 */

#include "hantro.h"
#include "hantro_jpeg.h"
#include "hantro_h1_regs.h"

/*
 * Supported formats.
 */

static const struct hantro_fmt stm32mp25_vdec_fmts[] = {
 {
  .fourcc = V4L2_PIX_FMT_NV12,
  .codec_mode = HANTRO_MODE_NONE,
  .frmsize = {
   .min_width = FMT_MIN_WIDTH,
   .max_width = FMT_FHD_WIDTH,
   .step_width = MB_DIM,
   .min_height = FMT_MIN_HEIGHT,
   .max_height = FMT_FHD_HEIGHT,
   .step_height = MB_DIM,
  },
 },
 {
  .fourcc = V4L2_PIX_FMT_VP8_FRAME,
  .codec_mode = HANTRO_MODE_VP8_DEC,
  .max_depth = 2,
  .frmsize = {
   .min_width = FMT_MIN_WIDTH,
   .max_width = FMT_FHD_WIDTH,
   .step_width = MB_DIM,
   .min_height = FMT_MIN_HEIGHT,
   .max_height = FMT_FHD_HEIGHT,
   .step_height = MB_DIM,
  },
 },
 {
  .fourcc = V4L2_PIX_FMT_H264_SLICE,
  .codec_mode = HANTRO_MODE_H264_DEC,
  .max_depth = 2,
  .frmsize = {
   .min_width = FMT_MIN_WIDTH,
   .max_width = FMT_FHD_WIDTH,
   .step_width = MB_DIM,
   .min_height = FMT_MIN_HEIGHT,
   .max_height = FMT_FHD_HEIGHT,
   .step_height = MB_DIM,
  },
 },
};

static const struct hantro_fmt stm32mp25_venc_fmts[] = {
 {
  .fourcc = V4L2_PIX_FMT_YUV420M,
  .codec_mode = HANTRO_MODE_NONE,
  .enc_fmt = ROCKCHIP_VPU_ENC_FMT_YUV420P,
 },
 {
  .fourcc = V4L2_PIX_FMT_NV12M,
  .codec_mode = HANTRO_MODE_NONE,
  .enc_fmt = ROCKCHIP_VPU_ENC_FMT_YUV420SP,
 },
 {
  .fourcc = V4L2_PIX_FMT_YUYV,
  .codec_mode = HANTRO_MODE_NONE,
  .enc_fmt = ROCKCHIP_VPU_ENC_FMT_YUYV422,
 },
 {
  .fourcc = V4L2_PIX_FMT_UYVY,
  .codec_mode = HANTRO_MODE_NONE,
  .enc_fmt = ROCKCHIP_VPU_ENC_FMT_UYVY422,
 },
 {
  .fourcc = V4L2_PIX_FMT_JPEG,
  .codec_mode = HANTRO_MODE_JPEG_ENC,
  .max_depth = 2,
  .header_size = JPEG_HEADER_SIZE,
  .frmsize = {
   .min_width = 96,
   .max_width = FMT_4K_WIDTH,
   .step_width = MB_DIM,
   .min_height = 96,
   .max_height = FMT_4K_HEIGHT,
   .step_height = MB_DIM,
  },
 },
};

static irqreturn_t stm32mp25_venc_irq(int irq, void *dev_id)
{
 struct hantro_dev *vpu = dev_id;
 enum vb2_buffer_state state;
 u32 status;

 status = vepu_read(vpu, H1_REG_INTERRUPT);
 state = (status & H1_REG_INTERRUPT_FRAME_RDY) ?
  VB2_BUF_STATE_DONE : VB2_BUF_STATE_ERROR;

 vepu_write(vpu, H1_REG_INTERRUPT_BIT, H1_REG_INTERRUPT);

 hantro_irq_done(vpu, state);

 return IRQ_HANDLED;
}

static void stm32mp25_venc_reset(struct hantro_ctx *ctx)
{
 struct hantro_dev *vpu = ctx->dev;

 reset_control_reset(vpu->resets);
}

/*
 * Supported codec ops.
 */

static const struct hantro_codec_ops stm32mp25_vdec_codec_ops[] = {
 [HANTRO_MODE_VP8_DEC] = {
  .run = hantro_g1_vp8_dec_run,
  .reset = hantro_g1_reset,
  .init = hantro_vp8_dec_init,
  .exit = hantro_vp8_dec_exit,
 },
 [HANTRO_MODE_H264_DEC] = {
  .run = hantro_g1_h264_dec_run,
  .reset = hantro_g1_reset,
  .init = hantro_h264_dec_init,
  .exit = hantro_h264_dec_exit,
 },
};

static const struct hantro_codec_ops stm32mp25_venc_codec_ops[] = {
 [HANTRO_MODE_JPEG_ENC] = {
  .run = hantro_h1_jpeg_enc_run,
  .reset = stm32mp25_venc_reset,
  .done = hantro_h1_jpeg_enc_done,
 },
};

/*
 * Variants.
 */

static const struct hantro_irq stm32mp25_vdec_irqs[] = {
 { "vdec", hantro_g1_irq },
};

static const char * const stm32mp25_vdec_clk_names[] = { "vdec-clk" };

const struct hantro_variant stm32mp25_vdec_variant = {
 .dec_fmts = stm32mp25_vdec_fmts,
 .num_dec_fmts = ARRAY_SIZE(stm32mp25_vdec_fmts),
 .codec = HANTRO_VP8_DECODER | HANTRO_H264_DECODER,
 .codec_ops = stm32mp25_vdec_codec_ops,
 .irqs = stm32mp25_vdec_irqs,
 .num_irqs = ARRAY_SIZE(stm32mp25_vdec_irqs),
 .clk_names = stm32mp25_vdec_clk_names,
 .num_clocks = ARRAY_SIZE(stm32mp25_vdec_clk_names),
};

static const struct hantro_irq stm32mp25_venc_irqs[] = {
 { "venc", stm32mp25_venc_irq },
};

static const char * const stm32mp25_venc_clk_names[] = {
 "venc-clk"
};

const struct hantro_variant stm32mp25_venc_variant = {
 .enc_fmts = stm32mp25_venc_fmts,
 .num_enc_fmts = ARRAY_SIZE(stm32mp25_venc_fmts),
 .codec = HANTRO_JPEG_ENCODER,
 .codec_ops = stm32mp25_venc_codec_ops,
 .irqs = stm32mp25_venc_irqs,
 .num_irqs = ARRAY_SIZE(stm32mp25_venc_irqs),
 .clk_names = stm32mp25_venc_clk_names,
 .num_clocks = ARRAY_SIZE(stm32mp25_venc_clk_names)
};