drivers/media/platform/ti-vpe/csc.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Color space converter library
   4  *
   5  * Copyright (c) 2013 Texas Instruments Inc.
   6  *
   7  * David Griego, <dagriego@biglakesoftware.com>
   8  * Dale Farnsworth, <dale@farnsworth.org>
   9  * Archit Taneja, <archit@ti.com>
  10  */
  11
  12 #include <linux/err.h>
  13 #include <linux/io.h>
  14 #include <linux/module.h>
  15 #include <linux/platform_device.h>
  16 #include <linux/slab.h>
  17 #include <linux/videodev2.h>
  18 #include <media/v4l2-common.h>
  19
  20 #include "csc.h"
  21
  22 /*
  23  * 12 coefficients in the order:
  24  * a0, b0, c0, a1, b1, c1, a2, b2, c2, d0, d1, d2
  25  */
  26 struct quantization {
  27         u16     coeff[12];
  28 };
  29
  30 struct colorspace {
  31         struct quantization limited;
  32         struct quantization full;
  33 };
  34
  35 struct encoding_direction {
  36         struct colorspace r601;
  37         struct colorspace r709;
  38 };
  39
  40 struct csc_coeffs {
  41         struct encoding_direction y2r;
  42         struct encoding_direction r2y;
  43 };
  44
  45 /* default colorspace coefficients */
  46 static struct csc_coeffs csc_coeffs = {
  47         .y2r = {
  48                 .r601 = {
  49                         .limited = {
  50                                 {       /* SDTV */
  51                                 0x0400, 0x0000, 0x057D, 0x0400, 0x1EA7, 0x1D35,
  52                                 0x0400, 0x06EF, 0x1FFE, 0x0D40, 0x0210, 0x0C88,
  53                                 }
  54                         },
  55                         .full = {
  56                                 {       /* SDTV */
  57                                 0x04A8, 0x1FFE, 0x0662, 0x04A8, 0x1E6F, 0x1CBF,
  58                                 0x04A8, 0x0812, 0x1FFF, 0x0C84, 0x0220, 0x0BAC,
  59                                 }
  60                         },
  61                 },
  62                 .r709 = {
  63                         .limited = {
  64                                 {       /* HDTV */
  65                                 0x0400, 0x0000, 0x0629, 0x0400, 0x1F45, 0x1E2B,
  66                                 0x0400, 0x0742, 0x0000, 0x0CEC, 0x0148, 0x0C60,
  67                                 }
  68                         },
  69                         .full = {
  70                                 {       /* HDTV */
  71                                 0x04A8, 0x0000, 0x072C, 0x04A8, 0x1F26, 0x1DDE,
  72                                 0x04A8, 0x0873, 0x0000, 0x0C20, 0x0134, 0x0B7C,
  73                                 }
  74                         },
  75                 },
  76         },
  77         .r2y = {
  78                 .r601 = {
  79                         .limited = {
  80                                 {       /* SDTV */
  81                                 0x0132, 0x0259, 0x0075, 0x1F50, 0x1EA5, 0x020B,
  82                                 0x020B, 0x1E4A, 0x1FAB, 0x0000, 0x0200, 0x0200,
  83                                 }
  84                         },
  85                         .full = {
  86                                 {       /* SDTV */
  87                                 0x0107, 0x0204, 0x0064, 0x1F68, 0x1ED6, 0x01C2,
  88                                 0x01C2, 0x1E87, 0x1FB7, 0x0040, 0x0200, 0x0200,
  89                                 }
  90                         },
  91                 },
  92                 .r709 = {
  93                         .limited = {
  94                                 {       /* HDTV */
  95                                 0x00DA, 0x02DC, 0x004A, 0x1F88, 0x1E6C, 0x020C,
  96                                 0x020C, 0x1E24, 0x1FD0, 0x0000, 0x0200, 0x0200,
  97                                 }
  98                         },
  99                         .full = {
 100                                 {       /* HDTV */
 101                                 0x00bb, 0x0275, 0x003f, 0x1f99, 0x1ea5, 0x01c2,
 102                                 0x01c2, 0x1e67, 0x1fd7, 0x0040, 0x0200, 0x0200,
 103                                 }
 104                         },
 105                 },
 106         },
 107
 108 };
 109
 110 void csc_dump_regs(struct csc_data *csc)
 111 {
 112         struct device *dev = &csc->pdev->dev;
 113
 114 #define DUMPREG(r) dev_dbg(dev, "%-35s %08x\n", #r, \
 115         ioread32(csc->base + CSC_##r))
 116
 117         dev_dbg(dev, "CSC Registers @ %pa:\n", &csc->res->start);
 118
 119         DUMPREG(CSC00);
 120         DUMPREG(CSC01);
 121         DUMPREG(CSC02);
 122         DUMPREG(CSC03);
 123         DUMPREG(CSC04);
 124         DUMPREG(CSC05);
 125
 126 #undef DUMPREG
 127 }
 128 EXPORT_SYMBOL(csc_dump_regs);
 129
 130 void csc_set_coeff_bypass(struct csc_data *csc, u32 *csc_reg5)
 131 {
 132         *csc_reg5 |= CSC_BYPASS;
 133 }
 134 EXPORT_SYMBOL(csc_set_coeff_bypass);
 135
 136 /*
 137  * set the color space converter coefficient shadow register values
 138  */
 139 void csc_set_coeff(struct csc_data *csc, u32 *csc_reg0,
 140                    struct v4l2_format *src_fmt, struct v4l2_format *dst_fmt)
 141 {
 142         u32 *csc_reg5 = csc_reg0 + 5;
 143         u32 *shadow_csc = csc_reg0;
 144         u16 *coeff, *end_coeff;
 145         const struct v4l2_pix_format *pix;
 146         const struct v4l2_pix_format_mplane *mp;
 147         const struct v4l2_format_info *src_finfo, *dst_finfo;
 148         enum v4l2_ycbcr_encoding src_ycbcr_enc, dst_ycbcr_enc;
 149         enum v4l2_quantization src_quantization, dst_quantization;
 150         u32 src_pixelformat, dst_pixelformat;
 151
 152         if (V4L2_TYPE_IS_MULTIPLANAR(src_fmt->type)) {
 153                 mp = &src_fmt->fmt.pix_mp;
 154                 src_pixelformat = mp->pixelformat;
 155                 src_ycbcr_enc = mp->ycbcr_enc;
 156                 src_quantization = mp->quantization;
 157         } else {
 158                 pix = &src_fmt->fmt.pix;
 159                 src_pixelformat = pix->pixelformat;
 160                 src_ycbcr_enc = pix->ycbcr_enc;
 161                 src_quantization = pix->quantization;
 162         }
 163
 164         if (V4L2_TYPE_IS_MULTIPLANAR(dst_fmt->type)) {
 165                 mp = &dst_fmt->fmt.pix_mp;
 166                 dst_pixelformat = mp->pixelformat;
 167                 dst_ycbcr_enc = mp->ycbcr_enc;
 168                 dst_quantization = mp->quantization;
 169         } else {
 170                 pix = &dst_fmt->fmt.pix;
 171                 dst_pixelformat = pix->pixelformat;
 172                 dst_ycbcr_enc = pix->ycbcr_enc;
 173                 dst_quantization = pix->quantization;
 174         }
 175
 176         src_finfo = v4l2_format_info(src_pixelformat);
 177         dst_finfo = v4l2_format_info(dst_pixelformat);
 178
 179         if (v4l2_is_format_yuv(src_finfo) &&
 180             v4l2_is_format_rgb(dst_finfo)) {
 181                 /* Y2R */
 182
 183                 /*
 184                  * These are not the standard default values but are
 185                  * set this way for historical compatibility
 186                  */
 187                 if (src_ycbcr_enc == V4L2_YCBCR_ENC_DEFAULT)
 188                         src_ycbcr_enc = V4L2_YCBCR_ENC_601;
 189
 190                 if (src_quantization == V4L2_QUANTIZATION_DEFAULT)
 191                         src_quantization = V4L2_QUANTIZATION_FULL_RANGE;
 192
 193                 if (src_ycbcr_enc == V4L2_YCBCR_ENC_601) {
 194                         if (src_quantization == V4L2_QUANTIZATION_FULL_RANGE)
 195                                 coeff = csc_coeffs.y2r.r601.full.coeff;
 196                         else
 197                                 coeff = csc_coeffs.y2r.r601.limited.coeff;
 198                 } else if (src_ycbcr_enc == V4L2_YCBCR_ENC_709) {
 199                         if (src_quantization == V4L2_QUANTIZATION_FULL_RANGE)
 200                                 coeff = csc_coeffs.y2r.r709.full.coeff;
 201                         else
 202                                 coeff = csc_coeffs.y2r.r709.limited.coeff;
 203                 } else {
 204                         /* Should never reach this, but it keeps gcc happy */
 205                         coeff = csc_coeffs.y2r.r601.full.coeff;
 206                 }
 207         } else if (v4l2_is_format_rgb(src_finfo) &&
 208                    v4l2_is_format_yuv(dst_finfo)) {
 209                 /* R2Y */
 210
 211                 /*
 212                  * These are not the standard default values but are
 213                  * set this way for historical compatibility
 214                  */
 215                 if (dst_ycbcr_enc == V4L2_YCBCR_ENC_DEFAULT)
 216                         dst_ycbcr_enc = V4L2_YCBCR_ENC_601;
 217
 218                 if (dst_quantization == V4L2_QUANTIZATION_DEFAULT)
 219                         dst_quantization = V4L2_QUANTIZATION_FULL_RANGE;
 220
 221                 if (dst_ycbcr_enc == V4L2_YCBCR_ENC_601) {
 222                         if (dst_quantization == V4L2_QUANTIZATION_FULL_RANGE)
 223                                 coeff = csc_coeffs.r2y.r601.full.coeff;
 224                         else
 225                                 coeff = csc_coeffs.r2y.r601.limited.coeff;
 226                 } else if (dst_ycbcr_enc == V4L2_YCBCR_ENC_709) {
 227                         if (dst_quantization == V4L2_QUANTIZATION_FULL_RANGE)
 228                                 coeff = csc_coeffs.r2y.r709.full.coeff;
 229                         else
 230                                 coeff = csc_coeffs.r2y.r709.limited.coeff;
 231                 } else {
 232                         /* Should never reach this, but it keeps gcc happy */
 233                         coeff = csc_coeffs.r2y.r601.full.coeff;
 234                 }
 235         } else {
 236                 *csc_reg5 |= CSC_BYPASS;
 237                 return;
 238         }
 239
 240         end_coeff = coeff + 12;
 241
 242         for (; coeff < end_coeff; coeff += 2)
 243                 *shadow_csc++ = (*(coeff + 1) << 16) | *coeff;
 244 }
 245 EXPORT_SYMBOL(csc_set_coeff);
 246
 247 struct csc_data *csc_create(struct platform_device *pdev, const char *res_name)
 248 {
 249         struct csc_data *csc;
 250
 251         dev_dbg(&pdev->dev, "csc_create\n");
 252
 253         csc = devm_kzalloc(&pdev->dev, sizeof(*csc), GFP_KERNEL);
 254         if (!csc) {
 255                 dev_err(&pdev->dev, "couldn't alloc csc_data\n");
 256                 return ERR_PTR(-ENOMEM);
 257         }
 258
 259         csc->pdev = pdev;
 260
 261         csc->res = platform_get_resource_byname(pdev, IORESOURCE_MEM,
 262                                                 res_name);
 263         if (csc->res == NULL) {
 264                 dev_err(&pdev->dev, "missing '%s' platform resources data\n",
 265                         res_name);
 266                 return ERR_PTR(-ENODEV);
 267         }
 268
 269         csc->base = devm_ioremap_resource(&pdev->dev, csc->res);
 270         if (IS_ERR(csc->base)) {
 271                 dev_err(&pdev->dev, "failed to ioremap\n");
 272                 return ERR_CAST(csc->base);
 273         }
 274
 275         return csc;
 276 }
 277 EXPORT_SYMBOL(csc_create);
 278
 279 MODULE_DESCRIPTION("TI VIP/VPE Color Space Converter");
 280 MODULE_AUTHOR("Texas Instruments Inc.");
 281 MODULE_LICENSE("GPL v2");