root/drivers/media/platform/amlogic/c3/isp/c3-isp-capture.c
// SPDX-License-Identifier: (GPL-2.0-only OR MIT)
/*
 * Copyright (C) 2024 Amlogic, Inc. All rights reserved
 */

#include <linux/cleanup.h>
#include <linux/pm_runtime.h>

#include <media/v4l2-ctrls.h>
#include <media/v4l2-event.h>
#include <media/v4l2-ioctl.h>
#include <media/v4l2-mc.h>
#include <media/videobuf2-dma-contig.h>

#include "c3-isp-common.h"
#include "c3-isp-regs.h"

#define C3_ISP_WRMIFX3_REG(addr, id)    ((addr) + (id) * 0x100)

static const struct c3_isp_cap_format_info cap_formats[] = {
        /* YUV formats */
        {
                .mbus_code = MEDIA_BUS_FMT_YUV10_1X30,
                .fourcc = V4L2_PIX_FMT_GREY,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_Y_ONLY,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X1,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_8BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_VU,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_8BIT,
                .hdiv = 1,
                .vdiv = 1,
        }, {
                .mbus_code = MEDIA_BUS_FMT_YUV10_1X30,
                .fourcc = V4L2_PIX_FMT_NV12M,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_YUV420,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X2,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_8BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_UV,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_8BIT,
                .hdiv = 2,
                .vdiv = 2,
        }, {
                .mbus_code = MEDIA_BUS_FMT_YUV10_1X30,
                .fourcc = V4L2_PIX_FMT_NV21M,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_YUV420,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X2,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_8BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_VU,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_8BIT,
                .hdiv = 2,
                .vdiv = 2,
        }, {
                .mbus_code = MEDIA_BUS_FMT_YUV10_1X30,
                .fourcc = V4L2_PIX_FMT_NV16M,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_YUV422,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X2,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_8BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_UV,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_8BIT,
                .hdiv = 1,
                .vdiv = 2
        }, {
                .mbus_code = MEDIA_BUS_FMT_YUV10_1X30,
                .fourcc = V4L2_PIX_FMT_NV61M,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_YUV422,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X2,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_8BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_VU,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_8BIT,
                .hdiv = 1,
                .vdiv = 2,
        },
        /* RAW formats */
        {
                .mbus_code = MEDIA_BUS_FMT_SRGGB16_1X16,
                .fourcc = V4L2_PIX_FMT_SRGGB12,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_RAW,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X1,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_16BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_VU,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_16BIT,
                .hdiv = 1,
                .vdiv = 1,
        }, {
                .mbus_code = MEDIA_BUS_FMT_SBGGR16_1X16,
                .fourcc = V4L2_PIX_FMT_SBGGR12,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_RAW,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X1,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_16BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_VU,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_16BIT,
                .hdiv = 1,
                .vdiv = 1,
        }, {
                .mbus_code = MEDIA_BUS_FMT_SGRBG16_1X16,
                .fourcc = V4L2_PIX_FMT_SGRBG12,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_RAW,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X1,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_16BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_VU,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_16BIT,
                .hdiv = 1,
                .vdiv = 1,
        }, {
                .mbus_code = MEDIA_BUS_FMT_SGBRG16_1X16,
                .fourcc = V4L2_PIX_FMT_SGBRG12,
                .format = ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_RAW,
                .planes = ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_X1,
                .ch0_pix_bits = ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_16BITS,
                .uv_swap = ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_VU,
                .in_bits = ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_16BIT,
                .hdiv = 1,
                .vdiv = 1,
        },
};

/* Hardware configuration */

/* Set the address of wrmifx3(write memory interface) */
static void c3_isp_cap_wrmifx3_buff(struct c3_isp_capture *cap)
{
        dma_addr_t y_dma_addr;
        dma_addr_t uv_dma_addr;

        if (cap->buff) {
                y_dma_addr = cap->buff->dma_addr[C3_ISP_PLANE_Y];
                uv_dma_addr = cap->buff->dma_addr[C3_ISP_PLANE_UV];
        } else {
                y_dma_addr = cap->dummy_buff.dma_addr;
                uv_dma_addr = cap->dummy_buff.dma_addr;
        }

        c3_isp_write(cap->isp,
                     C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_CH0_BADDR, cap->id),
                     ISP_WRMIFX3_0_CH0_BASE_ADDR(y_dma_addr));

        c3_isp_write(cap->isp,
                     C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_CH1_BADDR, cap->id),
                     ISP_WRMIFX3_0_CH1_BASE_ADDR(uv_dma_addr));
}

static void c3_isp_cap_wrmifx3_format(struct c3_isp_capture *cap)
{
        struct v4l2_pix_format_mplane *pix_mp = &cap->format.pix_mp;
        const struct c3_isp_cap_format_info *info = cap->format.info;
        u32 stride;
        u32 chrom_h;
        u32 chrom_v;

        c3_isp_write(cap->isp,
                     C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_FMT_SIZE, cap->id),
                     ISP_WRMIFX3_0_FMT_SIZE_HSIZE(pix_mp->width) |
                     ISP_WRMIFX3_0_FMT_SIZE_VSIZE(pix_mp->height));

        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_FMT_CTRL, cap->id),
                           ISP_WRMIFX3_0_FMT_CTRL_MODE_OUT_MASK, info->format);

        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_FMT_CTRL, cap->id),
                           ISP_WRMIFX3_0_FMT_CTRL_MTX_IBITS_MASK,
                           info->in_bits);

        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_FMT_CTRL, cap->id),
                           ISP_WRMIFX3_0_FMT_CTRL_MTX_PLANE_MASK, info->planes);

        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_FMT_CTRL, cap->id),
                           ISP_WRMIFX3_0_FMT_CTRL_MTX_UV_SWAP_MASK,
                           info->uv_swap);

        stride = DIV_ROUND_UP(pix_mp->plane_fmt[C3_ISP_PLANE_Y].bytesperline,
                              C3_ISP_DMA_SIZE_ALIGN_BYTES);
        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_CH0_CTRL0, cap->id),
                           ISP_WRMIFX3_0_CH0_CTRL0_STRIDE_MASK,
                           ISP_WRMIFX3_0_CH0_CTRL0_STRIDE(stride));

        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_CH0_CTRL1, cap->id),
                           ISP_WRMIFX3_0_CH0_CTRL1_PIX_BITS_MODE_MASK,
                           info->ch0_pix_bits);

        c3_isp_write(cap->isp,
                     C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_WIN_LUMA_H, cap->id),
                     ISP_WRMIFX3_0_WIN_LUMA_H_LUMA_HEND(pix_mp->width));

        c3_isp_write(cap->isp,
                     C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_WIN_LUMA_V, cap->id),
                     ISP_WRMIFX3_0_WIN_LUMA_V_LUMA_VEND(pix_mp->height));

        stride = DIV_ROUND_UP(pix_mp->plane_fmt[C3_ISP_PLANE_UV].bytesperline,
                              C3_ISP_DMA_SIZE_ALIGN_BYTES);
        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_CH1_CTRL0, cap->id),
                           ISP_WRMIFX3_0_CH1_CTRL0_STRIDE_MASK,
                           ISP_WRMIFX3_0_CH1_CTRL0_STRIDE(stride));

        c3_isp_update_bits(cap->isp,
                           C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_CH1_CTRL1, cap->id),
                           ISP_WRMIFX3_0_CH1_CTRL1_PIX_BITS_MODE_MASK,
                           ISP_WRMIFX3_0_CH1_CTRL1_PIX_BITS_16BITS);

        chrom_h = DIV_ROUND_UP(pix_mp->width, info->hdiv);
        c3_isp_write(cap->isp,
                     C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_WIN_CHROM_H, cap->id),
                     ISP_WRMIFX3_0_WIN_CHROM_H_CHROM_HEND(chrom_h));

        chrom_v = DIV_ROUND_UP(pix_mp->height, info->vdiv);
        c3_isp_write(cap->isp,
                     C3_ISP_WRMIFX3_REG(ISP_WRMIFX3_0_WIN_CHROM_V, cap->id),
                     ISP_WRMIFX3_0_WIN_CHROM_V_CHROM_VEND(chrom_v));
}

static int c3_isp_cap_dummy_buff_create(struct c3_isp_capture *cap)
{
        struct c3_isp_dummy_buffer *dummy_buff = &cap->dummy_buff;
        struct v4l2_pix_format_mplane *pix_mp = &cap->format.pix_mp;

        if (pix_mp->num_planes == 1)
                dummy_buff->size = pix_mp->plane_fmt[C3_ISP_PLANE_Y].sizeimage;
        else
                dummy_buff->size =
                        max(pix_mp->plane_fmt[C3_ISP_PLANE_Y].sizeimage,
                            pix_mp->plane_fmt[C3_ISP_PLANE_UV].sizeimage);

        /* The driver never access vaddr, no mapping is required */
        dummy_buff->vaddr = dma_alloc_attrs(cap->isp->dev, dummy_buff->size,
                                            &dummy_buff->dma_addr, GFP_KERNEL,
                                            DMA_ATTR_NO_KERNEL_MAPPING);
        if (!dummy_buff->vaddr)
                return -ENOMEM;

        return 0;
}

static void c3_isp_cap_dummy_buff_destroy(struct c3_isp_capture *cap)
{
        dma_free_attrs(cap->isp->dev, cap->dummy_buff.size,
                       cap->dummy_buff.vaddr, cap->dummy_buff.dma_addr,
                       DMA_ATTR_NO_KERNEL_MAPPING);
}

static void c3_isp_cap_cfg_buff(struct c3_isp_capture *cap)
{
        cap->buff = list_first_entry_or_null(&cap->pending,
                                             struct c3_isp_cap_buffer, list);

        c3_isp_cap_wrmifx3_buff(cap);

        if (cap->buff)
                list_del(&cap->buff->list);
}

static void c3_isp_cap_start(struct c3_isp_capture *cap)
{
        u32 mask;
        u32 val;

        scoped_guard(spinlock_irqsave, &cap->buff_lock)
                c3_isp_cap_cfg_buff(cap);

        c3_isp_cap_wrmifx3_format(cap);

        if (cap->id == C3_ISP_CAP_DEV_0) {
                mask = ISP_TOP_PATH_EN_WRMIF0_EN_MASK;
                val = ISP_TOP_PATH_EN_WRMIF0_EN;
        } else if (cap->id == C3_ISP_CAP_DEV_1) {
                mask = ISP_TOP_PATH_EN_WRMIF1_EN_MASK;
                val = ISP_TOP_PATH_EN_WRMIF1_EN;
        } else {
                mask = ISP_TOP_PATH_EN_WRMIF2_EN_MASK;
                val = ISP_TOP_PATH_EN_WRMIF2_EN;
        }

        c3_isp_update_bits(cap->isp, ISP_TOP_PATH_EN, mask, val);
}

static void c3_isp_cap_stop(struct c3_isp_capture *cap)
{
        u32 mask;
        u32 val;

        if (cap->id == C3_ISP_CAP_DEV_0) {
                mask = ISP_TOP_PATH_EN_WRMIF0_EN_MASK;
                val = ISP_TOP_PATH_EN_WRMIF0_DIS;
        } else if (cap->id == C3_ISP_CAP_DEV_1) {
                mask = ISP_TOP_PATH_EN_WRMIF1_EN_MASK;
                val = ISP_TOP_PATH_EN_WRMIF1_DIS;
        } else {
                mask = ISP_TOP_PATH_EN_WRMIF2_EN_MASK;
                val = ISP_TOP_PATH_EN_WRMIF2_DIS;
        }

        c3_isp_update_bits(cap->isp, ISP_TOP_PATH_EN, mask, val);
}

static void c3_isp_cap_done(struct c3_isp_capture *cap)
{
        struct c3_isp_cap_buffer *buff = cap->buff;

        guard(spinlock_irqsave)(&cap->buff_lock);

        if (buff) {
                buff->vb.sequence = cap->isp->frm_sequence;
                buff->vb.vb2_buf.timestamp = ktime_get();
                buff->vb.field = V4L2_FIELD_NONE;
                vb2_buffer_done(&buff->vb.vb2_buf, VB2_BUF_STATE_DONE);
        }

        c3_isp_cap_cfg_buff(cap);
}

/* V4L2 video operations */

static const struct c3_isp_cap_format_info *c3_cap_find_fmt(u32 fourcc)
{
        for (unsigned int i = 0; i < ARRAY_SIZE(cap_formats); i++) {
                if (cap_formats[i].fourcc == fourcc)
                        return &cap_formats[i];
        }

        return NULL;
}

static void c3_cap_try_fmt(struct v4l2_pix_format_mplane *pix_mp)
{
        const struct c3_isp_cap_format_info *fmt;
        const struct v4l2_format_info *info;
        struct v4l2_plane_pix_format *plane;

        fmt = c3_cap_find_fmt(pix_mp->pixelformat);
        if (!fmt)
                fmt = &cap_formats[0];

        pix_mp->width = clamp(pix_mp->width, C3_ISP_MIN_WIDTH,
                              C3_ISP_MAX_WIDTH);
        pix_mp->height = clamp(pix_mp->height, C3_ISP_MIN_HEIGHT,
                               C3_ISP_MAX_HEIGHT);
        pix_mp->pixelformat = fmt->fourcc;
        pix_mp->field = V4L2_FIELD_NONE;
        pix_mp->colorspace = V4L2_COLORSPACE_SRGB;
        pix_mp->ycbcr_enc = V4L2_YCBCR_ENC_601;
        pix_mp->quantization = V4L2_QUANTIZATION_LIM_RANGE;

        info = v4l2_format_info(fmt->fourcc);
        pix_mp->num_planes = info->mem_planes;
        memset(pix_mp->plane_fmt, 0, sizeof(pix_mp->plane_fmt));

        for (unsigned int i = 0; i < info->comp_planes; i++) {
                unsigned int hdiv = (i == 0) ? 1 : info->hdiv;
                unsigned int vdiv = (i == 0) ? 1 : info->vdiv;

                plane = &pix_mp->plane_fmt[i];

                plane->bytesperline = DIV_ROUND_UP(pix_mp->width, hdiv) *
                                      info->bpp[i] / info->bpp_div[i];
                plane->bytesperline = ALIGN(plane->bytesperline,
                                            C3_ISP_DMA_SIZE_ALIGN_BYTES);
                plane->sizeimage = plane->bytesperline *
                                   DIV_ROUND_UP(pix_mp->height, vdiv);
        }
}

static void c3_isp_cap_return_buffers(struct c3_isp_capture *cap,
                                      enum vb2_buffer_state state)
{
        struct c3_isp_cap_buffer *buff;

        guard(spinlock_irqsave)(&cap->buff_lock);

        if (cap->buff) {
                vb2_buffer_done(&cap->buff->vb.vb2_buf, state);
                cap->buff = NULL;
        }

        while (!list_empty(&cap->pending)) {
                buff = list_first_entry(&cap->pending,
                                        struct c3_isp_cap_buffer, list);
                list_del(&buff->list);
                vb2_buffer_done(&buff->vb.vb2_buf, state);
        }
}

static int c3_isp_cap_querycap(struct file *file, void *fh,
                               struct v4l2_capability *cap)
{
        strscpy(cap->driver, C3_ISP_DRIVER_NAME, sizeof(cap->driver));
        strscpy(cap->card, "AML C3 ISP", sizeof(cap->card));

        return 0;
}

static int c3_isp_cap_enum_fmt(struct file *file, void *fh,
                               struct v4l2_fmtdesc *f)
{
        const struct c3_isp_cap_format_info *fmt;
        unsigned int index = 0;
        unsigned int i;

        if (!f->mbus_code) {
                if (f->index >= ARRAY_SIZE(cap_formats))
                        return -EINVAL;

                fmt = &cap_formats[f->index];
                f->pixelformat = fmt->fourcc;
                return 0;
        }

        for (i = 0; i < ARRAY_SIZE(cap_formats); i++) {
                fmt = &cap_formats[i];
                if (f->mbus_code != fmt->mbus_code)
                        continue;

                if (index++ == f->index) {
                        f->pixelformat = cap_formats[i].fourcc;
                        return 0;
                }
        }

        return -EINVAL;
}

static int c3_isp_cap_g_fmt_mplane(struct file *file, void *fh,
                                   struct v4l2_format *f)
{
        struct c3_isp_capture *cap = video_drvdata(file);

        f->fmt.pix_mp = cap->format.pix_mp;

        return 0;
}

static int c3_isp_cap_s_fmt_mplane(struct file *file, void *fh,
                                   struct v4l2_format *f)
{
        struct c3_isp_capture *cap = video_drvdata(file);

        c3_cap_try_fmt(&f->fmt.pix_mp);

        cap->format.pix_mp = f->fmt.pix_mp;
        cap->format.info = c3_cap_find_fmt(f->fmt.pix_mp.pixelformat);

        return 0;
}

static int c3_isp_cap_try_fmt_mplane(struct file *file, void *fh,
                                     struct v4l2_format *f)
{
        c3_cap_try_fmt(&f->fmt.pix_mp);

        return 0;
}

static int c3_isp_cap_enum_frmsize(struct file *file, void *fh,
                                   struct v4l2_frmsizeenum *fsize)
{
        const struct c3_isp_cap_format_info *fmt;

        if (fsize->index)
                return -EINVAL;

        fmt = c3_cap_find_fmt(fsize->pixel_format);
        if (!fmt)
                return -EINVAL;

        fsize->type = V4L2_FRMSIZE_TYPE_STEPWISE;
        fsize->stepwise.min_width = C3_ISP_MIN_WIDTH;
        fsize->stepwise.min_height = C3_ISP_MIN_HEIGHT;
        fsize->stepwise.max_width = C3_ISP_MAX_WIDTH;
        fsize->stepwise.max_height = C3_ISP_MAX_HEIGHT;
        fsize->stepwise.step_width = 2;
        fsize->stepwise.step_height = 2;

        return 0;
}

static const struct v4l2_ioctl_ops isp_cap_v4l2_ioctl_ops = {
        .vidioc_querycap                = c3_isp_cap_querycap,
        .vidioc_enum_fmt_vid_cap        = c3_isp_cap_enum_fmt,
        .vidioc_g_fmt_vid_cap_mplane    = c3_isp_cap_g_fmt_mplane,
        .vidioc_s_fmt_vid_cap_mplane    = c3_isp_cap_s_fmt_mplane,
        .vidioc_try_fmt_vid_cap_mplane  = c3_isp_cap_try_fmt_mplane,
        .vidioc_reqbufs                 = vb2_ioctl_reqbufs,
        .vidioc_querybuf                = vb2_ioctl_querybuf,
        .vidioc_qbuf                    = vb2_ioctl_qbuf,
        .vidioc_expbuf                  = vb2_ioctl_expbuf,
        .vidioc_dqbuf                   = vb2_ioctl_dqbuf,
        .vidioc_prepare_buf             = vb2_ioctl_prepare_buf,
        .vidioc_create_bufs             = vb2_ioctl_create_bufs,
        .vidioc_streamon                = vb2_ioctl_streamon,
        .vidioc_streamoff               = vb2_ioctl_streamoff,
        .vidioc_enum_framesizes         = c3_isp_cap_enum_frmsize,
        .vidioc_subscribe_event         = v4l2_ctrl_subscribe_event,
        .vidioc_unsubscribe_event       = v4l2_event_unsubscribe,
};

static const struct v4l2_file_operations isp_cap_v4l2_fops = {
        .open = v4l2_fh_open,
        .release = vb2_fop_release,
        .poll = vb2_fop_poll,
        .unlocked_ioctl = video_ioctl2,
        .mmap = vb2_fop_mmap,
};

static int c3_isp_cap_link_validate(struct media_link *link)
{
        struct video_device *vdev =
                media_entity_to_video_device(link->sink->entity);
        struct v4l2_subdev *sd =
                media_entity_to_v4l2_subdev(link->source->entity);
        struct c3_isp_capture *cap = video_get_drvdata(vdev);
        struct v4l2_subdev_format src_fmt = {
                .which = V4L2_SUBDEV_FORMAT_ACTIVE,
                .pad = link->source->index,
        };
        int ret;

        ret = v4l2_subdev_call_state_active(sd, pad, get_fmt, &src_fmt);
        if (ret)
                return ret;

        if (src_fmt.format.width != cap->format.pix_mp.width ||
            src_fmt.format.height != cap->format.pix_mp.height ||
            src_fmt.format.code != cap->format.info->mbus_code) {
                dev_err(cap->isp->dev,
                        "link %s: %u -> %s: %u not valid: 0x%04x/%ux%u not match 0x%04x/%ux%u\n",
                        link->source->entity->name, link->source->index,
                        link->sink->entity->name, link->sink->index,
                        src_fmt.format.code, src_fmt.format.width,
                        src_fmt.format.height, cap->format.info->mbus_code,
                        cap->format.pix_mp.width, cap->format.pix_mp.height);

                return -EPIPE;
        }

        return 0;
}

static const struct media_entity_operations isp_cap_entity_ops = {
        .link_validate = c3_isp_cap_link_validate,
};

static int c3_isp_vb2_queue_setup(struct vb2_queue *q,
                                  unsigned int *num_buffers,
                                  unsigned int *num_planes,
                                  unsigned int sizes[],
                                  struct device *alloc_devs[])
{
        struct c3_isp_capture *cap = vb2_get_drv_priv(q);
        const struct v4l2_pix_format_mplane *pix_mp = &cap->format.pix_mp;
        unsigned int i;

        if (*num_planes) {
                if (*num_planes != pix_mp->num_planes)
                        return -EINVAL;

                for (i = 0; i < pix_mp->num_planes; i++)
                        if (sizes[i] < pix_mp->plane_fmt[i].sizeimage)
                                return -EINVAL;

                return 0;
        }

        *num_planes = pix_mp->num_planes;
        for (i = 0; i < pix_mp->num_planes; i++)
                sizes[i] = pix_mp->plane_fmt[i].sizeimage;

        return 0;
}

static void c3_isp_vb2_buf_queue(struct vb2_buffer *vb)
{
        struct vb2_v4l2_buffer *v4l2_buf = to_vb2_v4l2_buffer(vb);
        struct c3_isp_cap_buffer *buf =
                        container_of(v4l2_buf, struct c3_isp_cap_buffer, vb);
        struct c3_isp_capture *cap = vb2_get_drv_priv(vb->vb2_queue);

        guard(spinlock_irqsave)(&cap->buff_lock);

        list_add_tail(&buf->list, &cap->pending);
}

static int c3_isp_vb2_buf_prepare(struct vb2_buffer *vb)
{
        struct c3_isp_capture *cap = vb2_get_drv_priv(vb->vb2_queue);
        unsigned long size;

        for (unsigned int i = 0; i < cap->format.pix_mp.num_planes; i++) {
                size = cap->format.pix_mp.plane_fmt[i].sizeimage;
                if (vb2_plane_size(vb, i) < size) {
                        dev_err(cap->isp->dev,
                                "User buffer too small (%ld < %lu)\n",
                                vb2_plane_size(vb, i), size);
                        return -EINVAL;
                }

                vb2_set_plane_payload(vb, i, size);
        }

        return 0;
}

static int c3_isp_vb2_buf_init(struct vb2_buffer *vb)
{
        struct c3_isp_capture *cap = vb2_get_drv_priv(vb->vb2_queue);
        struct vb2_v4l2_buffer *v4l2_buf = to_vb2_v4l2_buffer(vb);
        struct c3_isp_cap_buffer *buf =
                container_of(v4l2_buf, struct c3_isp_cap_buffer, vb);

        for (unsigned int i = 0; i < cap->format.pix_mp.num_planes; i++)
                buf->dma_addr[i] = vb2_dma_contig_plane_dma_addr(vb, i);

        return 0;
}

static int c3_isp_vb2_start_streaming(struct vb2_queue *q,
                                      unsigned int count)
{
        struct c3_isp_capture *cap = vb2_get_drv_priv(q);
        int ret;

        ret = video_device_pipeline_start(&cap->vdev, &cap->isp->pipe);
        if (ret) {
                dev_err(cap->isp->dev,
                        "Failed to start cap%u pipeline: %d\n", cap->id, ret);
                goto err_return_buffers;
        }

        ret = c3_isp_cap_dummy_buff_create(cap);
        if (ret)
                goto err_pipeline_stop;

        ret = pm_runtime_resume_and_get(cap->isp->dev);
        if (ret)
                goto err_dummy_destroy;

        c3_isp_cap_start(cap);

        ret = v4l2_subdev_enable_streams(&cap->rsz->sd, C3_ISP_RSZ_PAD_SOURCE,
                                         BIT(0));
        if (ret)
                goto err_pm_put;

        return 0;

err_pm_put:
        pm_runtime_put(cap->isp->dev);
err_dummy_destroy:
        c3_isp_cap_dummy_buff_destroy(cap);
err_pipeline_stop:
        video_device_pipeline_stop(&cap->vdev);
err_return_buffers:
        c3_isp_cap_return_buffers(cap, VB2_BUF_STATE_QUEUED);
        return ret;
}

static void c3_isp_vb2_stop_streaming(struct vb2_queue *q)
{
        struct c3_isp_capture *cap = vb2_get_drv_priv(q);

        c3_isp_cap_stop(cap);

        c3_isp_cap_return_buffers(cap, VB2_BUF_STATE_ERROR);

        v4l2_subdev_disable_streams(&cap->rsz->sd, C3_ISP_RSZ_PAD_SOURCE,
                                    BIT(0));

        pm_runtime_put(cap->isp->dev);

        c3_isp_cap_dummy_buff_destroy(cap);

        video_device_pipeline_stop(&cap->vdev);
}

static const struct vb2_ops isp_video_vb2_ops = {
        .queue_setup = c3_isp_vb2_queue_setup,
        .buf_queue = c3_isp_vb2_buf_queue,
        .buf_prepare = c3_isp_vb2_buf_prepare,
        .buf_init = c3_isp_vb2_buf_init,
        .start_streaming = c3_isp_vb2_start_streaming,
        .stop_streaming = c3_isp_vb2_stop_streaming,
};

static int c3_isp_register_capture(struct c3_isp_capture *cap)
{
        struct video_device *vdev = &cap->vdev;
        struct vb2_queue *vb2_q = &cap->vb2_q;
        int ret;

        snprintf(vdev->name, sizeof(vdev->name), "c3-isp-cap%u", cap->id);
        vdev->fops = &isp_cap_v4l2_fops;
        vdev->ioctl_ops = &isp_cap_v4l2_ioctl_ops;
        vdev->v4l2_dev = &cap->isp->v4l2_dev;
        vdev->entity.ops = &isp_cap_entity_ops;
        vdev->lock = &cap->lock;
        vdev->minor = -1;
        vdev->queue = vb2_q;
        vdev->release = video_device_release_empty;
        vdev->device_caps = V4L2_CAP_VIDEO_CAPTURE_MPLANE |
                            V4L2_CAP_STREAMING;
        vdev->vfl_dir = VFL_DIR_RX;
        video_set_drvdata(vdev, cap);

        vb2_q->drv_priv = cap;
        vb2_q->mem_ops = &vb2_dma_contig_memops;
        vb2_q->ops = &isp_video_vb2_ops;
        vb2_q->type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE;
        vb2_q->io_modes = VB2_DMABUF | VB2_MMAP;
        vb2_q->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
        vb2_q->buf_struct_size = sizeof(struct c3_isp_cap_buffer);
        vb2_q->dev = cap->isp->dev;
        vb2_q->lock = &cap->lock;

        ret = vb2_queue_init(vb2_q);
        if (ret)
                goto err_destroy;

        cap->pad.flags = MEDIA_PAD_FL_SINK;
        ret = media_entity_pads_init(&vdev->entity, 1, &cap->pad);
        if (ret)
                goto err_queue_release;

        ret = video_register_device(vdev, VFL_TYPE_VIDEO, -1);
        if (ret) {
                dev_err(cap->isp->dev,
                        "Failed to register %s: %d\n", vdev->name, ret);
                goto err_entity_cleanup;
        }

        return 0;

err_entity_cleanup:
        media_entity_cleanup(&vdev->entity);
err_queue_release:
        vb2_queue_release(vb2_q);
err_destroy:
        mutex_destroy(&cap->lock);
        return ret;
}

int c3_isp_captures_register(struct c3_isp_device *isp)
{
        int ret;
        unsigned int i;
        struct c3_isp_capture *cap;

        for (i = C3_ISP_CAP_DEV_0; i < C3_ISP_NUM_CAP_DEVS; i++) {
                cap = &isp->caps[i];
                memset(cap, 0, sizeof(*cap));

                cap->format.pix_mp.width = C3_ISP_DEFAULT_WIDTH;
                cap->format.pix_mp.height = C3_ISP_DEFAULT_HEIGHT;
                cap->format.pix_mp.field = V4L2_FIELD_NONE;
                cap->format.pix_mp.pixelformat = V4L2_PIX_FMT_NV12M;
                cap->format.pix_mp.colorspace = V4L2_COLORSPACE_SRGB;
                cap->format.info =
                        c3_cap_find_fmt(cap->format.pix_mp.pixelformat);

                c3_cap_try_fmt(&cap->format.pix_mp);

                cap->id = i;
                cap->rsz = &isp->resizers[i];
                cap->isp = isp;
                INIT_LIST_HEAD(&cap->pending);
                spin_lock_init(&cap->buff_lock);
                mutex_init(&cap->lock);

                ret = c3_isp_register_capture(cap);
                if (ret) {
                        cap->isp = NULL;
                        mutex_destroy(&cap->lock);
                        c3_isp_captures_unregister(isp);
                        return ret;
                }
        }

        return 0;
}

void c3_isp_captures_unregister(struct c3_isp_device *isp)
{
        unsigned int i;
        struct c3_isp_capture *cap;

        for (i = C3_ISP_CAP_DEV_0; i < C3_ISP_NUM_CAP_DEVS; i++) {
                cap = &isp->caps[i];

                if (!cap->isp)
                        continue;
                vb2_queue_release(&cap->vb2_q);
                media_entity_cleanup(&cap->vdev.entity);
                video_unregister_device(&cap->vdev);
                mutex_destroy(&cap->lock);
        }
}

void c3_isp_captures_isr(struct c3_isp_device *isp)
{
        c3_isp_cap_done(&isp->caps[C3_ISP_CAP_DEV_0]);
        c3_isp_cap_done(&isp->caps[C3_ISP_CAP_DEV_1]);
        c3_isp_cap_done(&isp->caps[C3_ISP_CAP_DEV_2]);
}