brintos

brintos / linux-shallow public Read only

0
0
Text · 6.8 KiB · ff15bc5 Raw
280 lines · c
1// SPDX-License-Identifier: GPL-2.0-only2/*3 * Color space converter library4 *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, d225 */26struct quantization {27	u16	coeff[12];28};29 30struct colorspace {31	struct quantization limited;32	struct quantization full;33};34 35struct encoding_direction {36	struct colorspace r601;37	struct colorspace r709;38};39 40struct csc_coeffs {41	struct encoding_direction y2r;42	struct encoding_direction r2y;43};44 45/* default colorspace coefficients */46static 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 110void 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 DUMPREG127}128EXPORT_SYMBOL(csc_dump_regs);129 130void csc_set_coeff_bypass(struct csc_data *csc, u32 *csc_reg5)131{132	*csc_reg5 |= CSC_BYPASS;133}134EXPORT_SYMBOL(csc_set_coeff_bypass);135 136/*137 * set the color space converter coefficient shadow register values138 */139void 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 are185		 * set this way for historical compatibility186		 */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			else197				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			else202				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 are213		 * set this way for historical compatibility214		 */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			else225				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			else230				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}245EXPORT_SYMBOL(csc_set_coeff);246 247struct 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		return ERR_CAST(csc->base);272 273	return csc;274}275EXPORT_SYMBOL(csc_create);276 277MODULE_DESCRIPTION("TI VIP/VPE Color Space Converter");278MODULE_AUTHOR("Texas Instruments Inc.");279MODULE_LICENSE("GPL v2");280