/***************************************************************************
 * Plug-in for HV7131R image sensor connected to the SN9C1xx PC Camera     *
 * Controllers                                                             *
 *                                                                         *
 * Copyright (C) 2007 by Luca Risolia <luca.risolia@studio.unibo.it>       *
 *                                                                         *
 * This program is free software; you can redistribute it and/or modify    *
 * it under the terms of the GNU General Public License as published by    *
 * the Free Software Foundation; either version 2 of the License, or       *
 * (at your option) any later version.                                     *
 *                                                                         *
 * This program is distributed in the hope that it will be useful,         *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of          *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           *
 * GNU General Public License for more details.                            *
 *                                                                         *
 * You should have received a copy of the GNU General Public License       *
 * along with this program; if not, write to the Free Software             *
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.               *
 ***************************************************************************/

#include "sn9c102_sensor.h"
#include "sn9c102_devtable.h"


static int hv7131r_init(struct sn9c102_device* cam)
{
	int err = 0;

	switch (sn9c102_get_bridge(cam)) {
	case BRIDGE_SN9C103:
		err = sn9c102_write_const_regs(cam, {0x00, 0x03}, {0x1a, 0x04},
					       {0x20, 0x05}, {0x20, 0x06},
					       {0x03, 0x10}, {0x00, 0x14},
					       {0x60, 0x17}, {0x0a, 0x18},
					       {0xf0, 0x19}, {0x1d, 0x1a},
					       {0x10, 0x1b}, {0x02, 0x1c},
					       {0x03, 0x1d}, {0x0f, 0x1e},
					       {0x0c, 0x1f}, {0x00, 0x20},
					       {0x10, 0x21}, {0x20, 0x22},
					       {0x30, 0x23}, {0x40, 0x24},
					       {0x50, 0x25}, {0x60, 0x26},
					       {0x70, 0x27}, {0x80, 0x28},
					       {0x90, 0x29}, {0xa0, 0x2a},
					       {0xb0, 0x2b}, {0xc0, 0x2c},
					       {0xd0, 0x2d}, {0xe0, 0x2e},
					       {0xf0, 0x2f}, {0xff, 0x30});
		break;
	case BRIDGE_SN9C105:
	case BRIDGE_SN9C120:
		err = sn9c102_write_const_regs(cam, {0x44, 0x01}, {0x40, 0x02},
					       {0x00, 0x03}, {0x1a, 0x04},
					       {0x44, 0x05}, {0x3e, 0x06},
					       {0x1a, 0x07}, {0x03, 0x10},
					       {0x08, 0x14}, {0xa3, 0x17},
					       {0x4b, 0x18}, {0x00, 0x19},
					       {0x1d, 0x1a}, {0x10, 0x1b},
					       {0x02, 0x1c}, {0x03, 0x1d},
					       {0x0f, 0x1e}, {0x0c, 0x1f},
					       {0x00, 0x20}, {0x29, 0x21},
					       {0x40, 0x22}, {0x54, 0x23},
					       {0x66, 0x24}, {0x76, 0x25},
					       {0x85, 0x26}, {0x94, 0x27},
					       {0xa1, 0x28}, {0xae, 0x29},
					       {0xbb, 0x2a}, {0xc7, 0x2b},
					       {0xd3, 0x2c}, {0xde, 0x2d},
					       {0xea, 0x2e}, {0xf4, 0x2f},
					       {0xff, 0x30}, {0x00, 0x3F},
					       {0xC7, 0x40}, {0x01, 0x41},
					       {0x44, 0x42}, {0x00, 0x43},
					       {0x44, 0x44}, {0x00, 0x45},
					       {0x44, 0x46}, {0x00, 0x47},
					       {0xC7, 0x48}, {0x01, 0x49},
					       {0xC7, 0x4A}, {0x01, 0x4B},
					       {0xC7, 0x4C}, {0x01, 0x4D},
					       {0x44, 0x4E}, {0x00, 0x4F},
					       {0x44, 0x50}, {0x00, 0x51},
					       {0x44, 0x52}, {0x00, 0x53},
					       {0xC7, 0x54}, {0x01, 0x55},
					       {0xC7, 0x56}, {0x01, 0x57},
					       {0xC7, 0x58}, {0x01, 0x59},
					       {0x44, 0x5A}, {0x00, 0x5B},
					       {0x44, 0x5C}, {0x00, 0x5D},
					       {0x44, 0x5E}, {0x00, 0x5F},
					       {0xC7, 0x60}, {0x01, 0x61},
					       {0xC7, 0x62}, {0x01, 0x63},
					       {0xC7, 0x64}, {0x01, 0x65},
					       {0x44, 0x66}, {0x00, 0x67},
					       {0x44, 0x68}, {0x00, 0x69},
					       {0x44, 0x6A}, {0x00, 0x6B},
					       {0xC7, 0x6C}, {0x01, 0x6D},
					       {0xC7, 0x6E}, {0x01, 0x6F},
					       {0xC7, 0x70}, {0x01, 0x71},
					       {0x44, 0x72}, {0x00, 0x73},
					       {0x44, 0x74}, {0x00, 0x75},
					       {0x44, 0x76}, {0x00, 0x77},
					       {0xC7, 0x78}, {0x01, 0x79},
					       {0xC7, 0x7A}, {0x01, 0x7B},
					       {0xC7, 0x7C}, {0x01, 0x7D},
					       {0x44, 0x7E}, {0x00, 0x7F},
					       {0x14, 0x84}, {0x00, 0x85},
					       {0x27, 0x86}, {0x00, 0x87},
					       {0x07, 0x88}, {0x00, 0x89},
					       {0xEC, 0x8A}, {0x0f, 0x8B},
					       {0xD8, 0x8C}, {0x0f, 0x8D},
					       {0x3D, 0x8E}, {0x00, 0x8F},
					       {0x3D, 0x90}, {0x00, 0x91},
					       {0xCD, 0x92}, {0x0f, 0x93},
					       {0xf7, 0x94}, {0x0f, 0x95},
					       {0x0C, 0x96}, {0x00, 0x97},
					       {0x00, 0x98}, {0x66, 0x99},
					       {0x05, 0x9A}, {0x00, 0x9B},
					       {0x04, 0x9C}, {0x00, 0x9D},
					       {0x08, 0x9E}, {0x00, 0x9F},
					       {0x2D, 0xC0}, {0x2D, 0xC1},
					       {0x3A, 0xC2}, {0x05, 0xC3},
					       {0x04, 0xC4}, {0x3F, 0xC5},
					       {0x00, 0xC6}, {0x00, 0xC7},
					       {0x50, 0xC8}, {0x3C, 0xC9},
					       {0x28, 0xCA}, {0xD8, 0xCB},
					       {0x14, 0xCC}, {0xEC, 0xCD},
					       {0x32, 0xCE}, {0xDD, 0xCF},
					       {0x32, 0xD0}, {0xDD, 0xD1},
					       {0x6A, 0xD2}, {0x50, 0xD3},
					       {0x00, 0xD4}, {0x00, 0xD5},
					       {0x00, 0xD6});
		break;
	default:
		break;
	}

	err += sn9c102_i2c_write(cam, 0x20, 0x00);
	err += sn9c102_i2c_write(cam, 0x21, 0xd6);
	err += sn9c102_i2c_write(cam, 0x25, 0x06);

	return err;
}


static int hv7131r_get_ctrl(struct sn9c102_device* cam,
			    struct v4l2_control* ctrl)
{
	switch (ctrl->id) {
	case V4L2_CID_GAIN:
		if ((ctrl->value = sn9c102_i2c_read(cam, 0x30)) < 0)
			return -EIO;
		return 0;
	case V4L2_CID_RED_BALANCE:
		if ((ctrl->value = sn9c102_i2c_read(cam, 0x31)) < 0)
			return -EIO;
		ctrl->value = ctrl->value & 0x3f;
		return 0;
	case V4L2_CID_BLUE_BALANCE:
		if ((ctrl->value = sn9c102_i2c_read(cam, 0x33)) < 0)
			return -EIO;
		ctrl->value = ctrl->value & 0x3f;
		return 0;
	case SN9C102_V4L2_CID_GREEN_BALANCE:
		if ((ctrl->value = sn9c102_i2c_read(cam, 0x32)) < 0)
			return -EIO;
		ctrl->value = ctrl->value & 0x3f;
		return 0;
	case V4L2_CID_BLACK_LEVEL:
		if ((ctrl->value = sn9c102_i2c_read(cam, 0x01)) < 0)
			return -EIO;
		ctrl->value = (ctrl->value & 0x08) ? 1 : 0;
		return 0;
	default:
		return -EINVAL;
	}
}


static int hv7131r_set_ctrl(struct sn9c102_device* cam,
			    const struct v4l2_control* ctrl)
{
	int err = 0;

	switch (ctrl->id) {
	case V4L2_CID_GAIN:
		err += sn9c102_i2c_write(cam, 0x30, ctrl->value);
		break;
	case V4L2_CID_RED_BALANCE:
		err += sn9c102_i2c_write(cam, 0x31, ctrl->value);
		break;
	case V4L2_CID_BLUE_BALANCE:
		err += sn9c102_i2c_write(cam, 0x33, ctrl->value);
		break;
	case SN9C102_V4L2_CID_GREEN_BALANCE:
		err += sn9c102_i2c_write(cam, 0x32, ctrl->value);
		break;
	case V4L2_CID_BLACK_LEVEL:
		{
			int r = sn9c102_i2c_read(cam, 0x01);
			if (r < 0)
				return -EIO;
			err += sn9c102_i2c_write(cam, 0x01,
						 (ctrl->value<<3) | (r&0xf7));
		}
		break;
	default:
		return -EINVAL;
	}

	return err ? -EIO : 0;
}


static int hv7131r_set_crop(struct sn9c102_device* cam,
			    const struct v4l2_rect* rect)
{
	struct sn9c102_sensor* s = sn9c102_get_sensor(cam);
	int err = 0;
	u8 h_start = (u8)(rect->left - s->cropcap.bounds.left) + 1,
	   v_start = (u8)(rect->top - s->cropcap.bounds.top) + 1;

	err += sn9c102_write_reg(cam, h_start, 0x12);
	err += sn9c102_write_reg(cam, v_start, 0x13);

	return err;
}


static int hv7131r_set_pix_format(struct sn9c102_device* cam,
				  const struct v4l2_pix_format* pix)
{
	int err = 0;

	switch (sn9c102_get_bridge(cam)) {
	case BRIDGE_SN9C103:
		if (pix->pixelformat == V4L2_PIX_FMT_SBGGR8) {
			err += sn9c102_write_reg(cam, 0xa0, 0x19);
			err += sn9c102_i2c_write(cam, 0x01, 0x04);
		} else {
			err += sn9c102_write_reg(cam, 0x30, 0x19);
			err += sn9c102_i2c_write(cam, 0x01, 0x04);
		}
		break;
	case BRIDGE_SN9C105:
	case BRIDGE_SN9C120:
		if (pix->pixelformat == V4L2_PIX_FMT_SBGGR8) {
			err += sn9c102_write_reg(cam, 0xa5, 0x17);
			err += sn9c102_i2c_write(cam, 0x01, 0x24);
		} else {
			err += sn9c102_write_reg(cam, 0xa3, 0x17);
			err += sn9c102_i2c_write(cam, 0x01, 0x04);
		}
		break;
	default:
		break;
	}

	return err;
}


static const struct sn9c102_sensor hv7131r = {
	.name = "HV7131R",
	.maintainer = "Luca Risolia <luca.risolia@studio.unibo.it>",
	.supported_bridge = BRIDGE_SN9C103 | BRIDGE_SN9C105 | BRIDGE_SN9C120,
	.sysfs_ops = SN9C102_I2C_READ | SN9C102_I2C_WRITE,
	.frequency = SN9C102_I2C_100KHZ,
	.interface = SN9C102_I2C_2WIRES,
	.i2c_slave_id = 0x11,
	.init = &hv7131r_init,
	.qctrl = {
		{
			.id = V4L2_CID_GAIN,
			.type = V4L2_CTRL_TYPE_INTEGER,
			.name = "global gain",
			.minimum = 0x00,
			.maximum = 0xff,
			.step = 0x01,
			.default_value = 0x40,
			.flags = 0,
		},
		{
			.id = V4L2_CID_RED_BALANCE,
			.type = V4L2_CTRL_TYPE_INTEGER,
			.name = "red balance",
			.minimum = 0x00,
			.maximum = 0x3f,
			.step = 0x01,
			.default_value = 0x08,
			.flags = 0,
		},
		{
			.id = V4L2_CID_BLUE_BALANCE,
			.type = V4L2_CTRL_TYPE_INTEGER,
			.name = "blue balance",
			.minimum = 0x00,
			.maximum = 0x3f,
			.step = 0x01,
			.default_value = 0x1a,
			.flags = 0,
		},
		{
			.id = SN9C102_V4L2_CID_GREEN_BALANCE,
			.type = V4L2_CTRL_TYPE_INTEGER,
			.name = "green balance",
			.minimum = 0x00,
			.maximum = 0x3f,
			.step = 0x01,
			.default_value = 0x2f,
			.flags = 0,
		},
		{
			.id = V4L2_CID_BLACK_LEVEL,
			.type = V4L2_CTRL_TYPE_BOOLEAN,
			.name = "auto black level compensation",
			.minimum = 0x00,
			.maximum = 0x01,
			.step = 0x01,
			.default_value = 0x00,
			.flags = 0,
		},
	},
	.get_ctrl = &hv7131r_get_ctrl,
	.set_ctrl = &hv7131r_set_ctrl,
	.cropcap = {
		.bounds = {
			.left = 0,
			.top = 0,
			.width = 640,
			.height = 480,
		},
		.defrect = {
			.left = 0,
			.top = 0,
			.width = 640,
			.height = 480,
		},
	},
	.set_crop = &hv7131r_set_crop,
	.pix_format = {
		.width = 640,
		.height = 480,
		.pixelformat = V4L2_PIX_FMT_SBGGR8,
		.priv = 8,
	},
	.set_pix_format = &hv7131r_set_pix_format
};


int sn9c102_probe_hv7131r(struct sn9c102_device* cam)
{
	int devid, err;

	err = sn9c102_write_const_regs(cam, {0x09, 0x01}, {0x44, 0x02},
				       {0x34, 0x01}, {0x20, 0x17},
				       {0x34, 0x01}, {0x46, 0x01});

	devid = sn9c102_i2c_try_read(cam, &hv7131r, 0x00);
	if (err || devid < 0)
		return -EIO;

	if (devid != 0x02)
		return -ENODEV;

	sn9c102_attach_sensor(cam, &hv7131r);

	return 0;
}