/* SPDX-License-Identifier: GPL-2.0 */
/*
 * PiSP Back End driver image format definitions.
 *
 * Copyright (c) 2021-2024 Raspberry Pi Ltd
 */

#ifndef _PISP_BE_FORMATS_
#define _PISP_BE_FORMATS_

#include <linux/bits.h>
#include <linux/videodev2.h>

#define PISPBE_MAX_PLANES       3
#define P3(x)                   ((x) * 8)

struct pisp_be_format {
        unsigned int fourcc;
        unsigned int align;
        unsigned int bit_depth;
        /* 0P3 factor for plane sizing */
        unsigned int plane_factor[PISPBE_MAX_PLANES];
        unsigned int num_planes;
        unsigned int colorspace_mask;
        enum v4l2_colorspace colorspace_default;
};

#define V4L2_COLORSPACE_MASK(colorspace) BIT(colorspace)

#define V4L2_COLORSPACE_MASK_JPEG       \
        V4L2_COLORSPACE_MASK(V4L2_COLORSPACE_JPEG)
#define V4L2_COLORSPACE_MASK_SMPTE170M  \
        V4L2_COLORSPACE_MASK(V4L2_COLORSPACE_SMPTE170M)
#define V4L2_COLORSPACE_MASK_REC709     \
        V4L2_COLORSPACE_MASK(V4L2_COLORSPACE_REC709)
#define V4L2_COLORSPACE_MASK_SRGB       \
        V4L2_COLORSPACE_MASK(V4L2_COLORSPACE_SRGB)
#define V4L2_COLORSPACE_MASK_RAW        \
        V4L2_COLORSPACE_MASK(V4L2_COLORSPACE_RAW)

/*
 * All three colour spaces SRGB, SMPTE170M and REC709 are fundamentally sRGB
 * underneath (as near as makes no difference to us), just with different YCbCr
 * encodings. Therefore the ISP can generate sRGB on its main output and any of
 * the others on its low resolution output. Applications should, when using both
 * outputs, program the colour spaces on them to be the same, matching whatever
 * is requested for the low resolution output, even if the main output is
 * producing an RGB format. In turn this requires us to allow all these colour
 * spaces for every YUV/RGB output format.
 */
#define V4L2_COLORSPACE_MASK_ALL_SRGB (V4L2_COLORSPACE_MASK_JPEG        | \
                                       V4L2_COLORSPACE_MASK_SRGB        | \
                                       V4L2_COLORSPACE_MASK_SMPTE170M   | \
                                       V4L2_COLORSPACE_MASK_REC709)

static const struct pisp_be_format supported_formats[] = {
        /* Single plane YUV formats */
        {
                .fourcc             = V4L2_PIX_FMT_YUV420,
                /* 128 alignment to ensure U/V planes are 64 byte aligned. */
                .align              = 128,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.25), P3(0.25) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YVU420,
                /* 128 alignment to ensure U/V planes are 64 byte aligned. */
                .align              = 128,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.25), P3(0.25) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_NV12,
                .align              = 32,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.5) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_NV21,
                .align              = 32,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.5) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YUYV,
                .align              = 64,
                .bit_depth          = 16,
                .plane_factor       = { P3(1) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_UYVY,
                .align              = 64,
                .bit_depth          = 16,
                .plane_factor       = { P3(1) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YVYU,
                .align              = 64,
                .bit_depth          = 16,
                .plane_factor       = { P3(1) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_VYUY,
                .align              = 64,
                .bit_depth          = 16,
                .plane_factor       = { P3(1) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        /* Multiplane YUV formats */
        {
                .fourcc             = V4L2_PIX_FMT_YUV420M,
                .align              = 64,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.25), P3(0.25) },
                .num_planes         = 3,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_NV12M,
                .align              = 32,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.5) },
                .num_planes         = 2,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_NV21M,
                .align              = 32,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.5) },
                .num_planes         = 2,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YVU420M,
                .align              = 64,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.25), P3(0.25) },
                .num_planes         = 3,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YUV422M,
                .align              = 64,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.5), P3(0.5) },
                .num_planes         = 3,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YVU422M,
                .align              = 64,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(0.5), P3(0.5) },
                .num_planes         = 3,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YUV444M,
                .align              = 64,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(1), P3(1) },
                .num_planes         = 3,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        {
                .fourcc             = V4L2_PIX_FMT_YVU444M,
                .align              = 64,
                .bit_depth          = 8,
                .plane_factor       = { P3(1), P3(1), P3(1) },
                .num_planes         = 3,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SMPTE170M,
        },
        /* RGB formats */
        {
                .fourcc             = V4L2_PIX_FMT_RGB24,
                .align              = 32,
                .bit_depth          = 24,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SRGB,
        },
        {
                .fourcc             = V4L2_PIX_FMT_BGR24,
                .align              = 32,
                .bit_depth          = 24,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SRGB,
        },
        {
                .fourcc             = V4L2_PIX_FMT_XBGR32,
                .align              = 64,
                .bit_depth          = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SRGB,
        },
        {
                .fourcc             = V4L2_PIX_FMT_RGBX32,
                .align              = 64,
                .bit_depth          = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SRGB,
        },
        {
                .fourcc             = V4L2_PIX_FMT_RGB48,
                .align              = 64,
                .bit_depth          = 48,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SRGB,
        },
        {
                .fourcc             = V4L2_PIX_FMT_BGR48,
                .align              = 64,
                .bit_depth          = 48,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_ALL_SRGB,
                .colorspace_default = V4L2_COLORSPACE_SRGB,
        },
        /* Bayer formats - 8-bit */
        {
                .fourcc             = V4L2_PIX_FMT_SRGGB8,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SBGGR8,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGRBG8,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGBRG8,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        /* Bayer formats - 16-bit */
        {
                .fourcc             = V4L2_PIX_FMT_SRGGB16,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SBGGR16,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGRBG16,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGBRG16,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                /* Bayer formats unpacked to 16bpp */
                /* 10 bit */
                .fourcc             = V4L2_PIX_FMT_SRGGB10,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SBGGR10,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGRBG10,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGBRG10,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                /* 12 bit */
                .fourcc             = V4L2_PIX_FMT_SRGGB12,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SBGGR12,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGRBG12,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGBRG12,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                /* 14 bit */
                .fourcc             = V4L2_PIX_FMT_SRGGB14,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SBGGR14,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGRBG14,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_SGBRG14,
                .bit_depth          = 16,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        /* Bayer formats - 16-bit PiSP Compressed */
        {
                .fourcc             = V4L2_PIX_FMT_PISP_COMP1_BGGR,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_PISP_COMP1_RGGB,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_PISP_COMP1_GRBG,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc             = V4L2_PIX_FMT_PISP_COMP1_GBRG,
                .bit_depth          = 8,
                .align              = 32,
                .plane_factor       = { P3(1.0) },
                .num_planes         = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        /* Greyscale Formats */
        {
                .fourcc         = V4L2_PIX_FMT_GREY,
                .bit_depth      = 8,
                .align          = 32,
                .num_planes     = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc         = V4L2_PIX_FMT_Y16,
                .bit_depth      = 16,
                .align          = 32,
                .plane_factor   = { P3(1.0) },
                .num_planes     = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
        {
                .fourcc         = V4L2_PIX_FMT_PISP_COMP1_MONO,
                .bit_depth      = 8,
                .align          = 32,
                .plane_factor   = { P3(1.0) },
                .num_planes     = 1,
                .colorspace_mask    = V4L2_COLORSPACE_MASK_RAW,
                .colorspace_default = V4L2_COLORSPACE_RAW,
        },
};

static const struct pisp_be_format meta_out_supported_formats[] = {
        /* Configuration buffer format. */
        {
                .fourcc             = V4L2_META_FMT_RPI_BE_CFG,
        },
};

#endif /* _PISP_BE_FORMATS_ */