// SPDX-License-Identifier: GPL-2.0
/*
 * Cedrus VPU driver
 *
 * Copyright (C) 2016 Florent Revest <florent.revest@free-electrons.com>
 * Copyright (C) 2018 Paul Kocialkowski <paul.kocialkowski@bootlin.com>
 * Copyright (C) 2018 Bootlin
 */

#include <media/videobuf2-dma-contig.h>

#include "cedrus.h"
#include "cedrus_hw.h"
#include "cedrus_regs.h"

static enum cedrus_irq_status cedrus_mpeg2_irq_status(struct cedrus_ctx *ctx)
{
        struct cedrus_dev *dev = ctx->dev;
        u32 reg;

        reg = cedrus_read(dev, VE_DEC_MPEG_STATUS);
        reg &= VE_DEC_MPEG_STATUS_CHECK_MASK;

        if (!reg)
                return CEDRUS_IRQ_NONE;

        if (reg & VE_DEC_MPEG_STATUS_CHECK_ERROR ||
            !(reg & VE_DEC_MPEG_STATUS_SUCCESS))
                return CEDRUS_IRQ_ERROR;

        return CEDRUS_IRQ_OK;
}

static void cedrus_mpeg2_irq_clear(struct cedrus_ctx *ctx)
{
        struct cedrus_dev *dev = ctx->dev;

        cedrus_write(dev, VE_DEC_MPEG_STATUS, VE_DEC_MPEG_STATUS_CHECK_MASK);
}

static void cedrus_mpeg2_irq_disable(struct cedrus_ctx *ctx)
{
        struct cedrus_dev *dev = ctx->dev;
        u32 reg = cedrus_read(dev, VE_DEC_MPEG_CTRL);

        reg &= ~VE_DEC_MPEG_CTRL_IRQ_MASK;

        cedrus_write(dev, VE_DEC_MPEG_CTRL, reg);
}

static void cedrus_mpeg2_setup(struct cedrus_ctx *ctx, struct cedrus_run *run)
{
        const struct v4l2_ctrl_mpeg2_sequence *seq;
        const struct v4l2_ctrl_mpeg2_picture *pic;
        const struct v4l2_ctrl_mpeg2_quantisation *quantisation;
        dma_addr_t src_buf_addr, dst_luma_addr, dst_chroma_addr;
        dma_addr_t fwd_luma_addr, fwd_chroma_addr;
        dma_addr_t bwd_luma_addr, bwd_chroma_addr;
        struct cedrus_dev *dev = ctx->dev;
        struct vb2_queue *vq;
        const u8 *matrix;
        int forward_idx;
        int backward_idx;
        unsigned int i;
        u32 reg;

        seq = run->mpeg2.sequence;
        pic = run->mpeg2.picture;

        quantisation = run->mpeg2.quantisation;

        /* Activate MPEG engine. */
        cedrus_engine_enable(ctx, CEDRUS_CODEC_MPEG2);

        /* Set intra quantisation matrix. */
        matrix = quantisation->intra_quantiser_matrix;
        for (i = 0; i < 64; i++) {
                reg = VE_DEC_MPEG_IQMINPUT_WEIGHT(i, matrix[i]);
                reg |= VE_DEC_MPEG_IQMINPUT_FLAG_INTRA;

                cedrus_write(dev, VE_DEC_MPEG_IQMINPUT, reg);
        }

        /* Set non-intra quantisation matrix. */
        matrix = quantisation->non_intra_quantiser_matrix;
        for (i = 0; i < 64; i++) {
                reg = VE_DEC_MPEG_IQMINPUT_WEIGHT(i, matrix[i]);
                reg |= VE_DEC_MPEG_IQMINPUT_FLAG_NON_INTRA;

                cedrus_write(dev, VE_DEC_MPEG_IQMINPUT, reg);
        }

        /* Set MPEG picture header. */

        reg = VE_DEC_MPEG_MP12HDR_SLICE_TYPE(pic->picture_coding_type);
        reg |= VE_DEC_MPEG_MP12HDR_F_CODE(0, 0, pic->f_code[0][0]);
        reg |= VE_DEC_MPEG_MP12HDR_F_CODE(0, 1, pic->f_code[0][1]);
        reg |= VE_DEC_MPEG_MP12HDR_F_CODE(1, 0, pic->f_code[1][0]);
        reg |= VE_DEC_MPEG_MP12HDR_F_CODE(1, 1, pic->f_code[1][1]);
        reg |= VE_DEC_MPEG_MP12HDR_INTRA_DC_PRECISION(pic->intra_dc_precision);
        reg |= VE_DEC_MPEG_MP12HDR_INTRA_PICTURE_STRUCTURE(pic->picture_structure);
        reg |= VE_DEC_MPEG_MP12HDR_TOP_FIELD_FIRST(pic->flags & V4L2_MPEG2_PIC_FLAG_TOP_FIELD_FIRST);
        reg |= VE_DEC_MPEG_MP12HDR_FRAME_PRED_FRAME_DCT(pic->flags & V4L2_MPEG2_PIC_FLAG_FRAME_PRED_DCT);
        reg |= VE_DEC_MPEG_MP12HDR_CONCEALMENT_MOTION_VECTORS(pic->flags & V4L2_MPEG2_PIC_FLAG_CONCEALMENT_MV);
        reg |= VE_DEC_MPEG_MP12HDR_Q_SCALE_TYPE(pic->flags & V4L2_MPEG2_PIC_FLAG_Q_SCALE_TYPE);
        reg |= VE_DEC_MPEG_MP12HDR_INTRA_VLC_FORMAT(pic->flags & V4L2_MPEG2_PIC_FLAG_INTRA_VLC);
        reg |= VE_DEC_MPEG_MP12HDR_ALTERNATE_SCAN(pic->flags & V4L2_MPEG2_PIC_FLAG_ALT_SCAN);
        reg |= VE_DEC_MPEG_MP12HDR_FULL_PEL_FORWARD_VECTOR(0);
        reg |= VE_DEC_MPEG_MP12HDR_FULL_PEL_BACKWARD_VECTOR(0);

        cedrus_write(dev, VE_DEC_MPEG_MP12HDR, reg);

        /* Set frame dimensions. */

        reg = VE_DEC_MPEG_PICCODEDSIZE_WIDTH(seq->horizontal_size);
        reg |= VE_DEC_MPEG_PICCODEDSIZE_HEIGHT(seq->vertical_size);

        cedrus_write(dev, VE_DEC_MPEG_PICCODEDSIZE, reg);

        reg = VE_DEC_MPEG_PICBOUNDSIZE_WIDTH(ctx->src_fmt.width);
        reg |= VE_DEC_MPEG_PICBOUNDSIZE_HEIGHT(ctx->src_fmt.height);

        cedrus_write(dev, VE_DEC_MPEG_PICBOUNDSIZE, reg);

        /* Forward and backward prediction reference buffers. */

        vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);

        forward_idx = vb2_find_timestamp(vq, pic->forward_ref_ts, 0);
        fwd_luma_addr = cedrus_dst_buf_addr(ctx, forward_idx, 0);
        fwd_chroma_addr = cedrus_dst_buf_addr(ctx, forward_idx, 1);

        cedrus_write(dev, VE_DEC_MPEG_FWD_REF_LUMA_ADDR, fwd_luma_addr);
        cedrus_write(dev, VE_DEC_MPEG_FWD_REF_CHROMA_ADDR, fwd_chroma_addr);

        backward_idx = vb2_find_timestamp(vq, pic->backward_ref_ts, 0);
        bwd_luma_addr = cedrus_dst_buf_addr(ctx, backward_idx, 0);
        bwd_chroma_addr = cedrus_dst_buf_addr(ctx, backward_idx, 1);

        cedrus_write(dev, VE_DEC_MPEG_BWD_REF_LUMA_ADDR, bwd_luma_addr);
        cedrus_write(dev, VE_DEC_MPEG_BWD_REF_CHROMA_ADDR, bwd_chroma_addr);

        /* Destination luma and chroma buffers. */

        dst_luma_addr = cedrus_dst_buf_addr(ctx, run->dst->vb2_buf.index, 0);
        dst_chroma_addr = cedrus_dst_buf_addr(ctx, run->dst->vb2_buf.index, 1);

        cedrus_write(dev, VE_DEC_MPEG_REC_LUMA, dst_luma_addr);
        cedrus_write(dev, VE_DEC_MPEG_REC_CHROMA, dst_chroma_addr);

        /* Source offset and length in bits. */

        cedrus_write(dev, VE_DEC_MPEG_VLD_OFFSET, 0);

        reg = vb2_get_plane_payload(&run->src->vb2_buf, 0) * 8;
        cedrus_write(dev, VE_DEC_MPEG_VLD_LEN, reg);

        /* Source beginning and end addresses. */

        src_buf_addr = vb2_dma_contig_plane_dma_addr(&run->src->vb2_buf, 0);

        reg = VE_DEC_MPEG_VLD_ADDR_BASE(src_buf_addr);
        reg |= VE_DEC_MPEG_VLD_ADDR_VALID_PIC_DATA;
        reg |= VE_DEC_MPEG_VLD_ADDR_LAST_PIC_DATA;
        reg |= VE_DEC_MPEG_VLD_ADDR_FIRST_PIC_DATA;

        cedrus_write(dev, VE_DEC_MPEG_VLD_ADDR, reg);

        reg = src_buf_addr + vb2_get_plane_payload(&run->src->vb2_buf, 0);
        cedrus_write(dev, VE_DEC_MPEG_VLD_END_ADDR, reg);

        /* Macroblock address: start at the beginning. */
        reg = VE_DEC_MPEG_MBADDR_Y(0) | VE_DEC_MPEG_MBADDR_X(0);
        cedrus_write(dev, VE_DEC_MPEG_MBADDR, reg);

        /* Clear previous errors. */
        cedrus_write(dev, VE_DEC_MPEG_ERROR, 0);

        /* Clear correct macroblocks register. */
        cedrus_write(dev, VE_DEC_MPEG_CRTMBADDR, 0);

        /* Enable appropriate interruptions and components. */

        reg = VE_DEC_MPEG_CTRL_IRQ_MASK | VE_DEC_MPEG_CTRL_MC_NO_WRITEBACK |
              VE_DEC_MPEG_CTRL_MC_CACHE_EN;

        cedrus_write(dev, VE_DEC_MPEG_CTRL, reg);
}

static void cedrus_mpeg2_trigger(struct cedrus_ctx *ctx)
{
        struct cedrus_dev *dev = ctx->dev;
        u32 reg;

        /* Trigger MPEG engine. */
        reg = VE_DEC_MPEG_TRIGGER_HW_MPEG_VLD | VE_DEC_MPEG_TRIGGER_MPEG2 |
              VE_DEC_MPEG_TRIGGER_MB_BOUNDARY;

        cedrus_write(dev, VE_DEC_MPEG_TRIGGER, reg);
}

struct cedrus_dec_ops cedrus_dec_ops_mpeg2 = {
        .irq_clear      = cedrus_mpeg2_irq_clear,
        .irq_disable    = cedrus_mpeg2_irq_disable,
        .irq_status     = cedrus_mpeg2_irq_status,
        .setup          = cedrus_mpeg2_setup,
        .trigger        = cedrus_mpeg2_trigger,
};