// SPDX-License-Identifier: GPL-2.0
/*
 * ov4689 driver
 *
 * Copyright (C) 2017 Fuzhou Rockchip Electronics Co., Ltd.
 * Copyright (C) 2022, 2024 Mikhail Rudenko
 */

#include <linux/clk.h>
#include <linux/delay.h>
#include <linux/gpio/consumer.h>
#include <linux/i2c.h>
#include <linux/module.h>
#include <linux/pm_runtime.h>
#include <linux/regulator/consumer.h>
#include <media/media-entity.h>
#include <media/v4l2-async.h>
#include <media/v4l2-cci.h>
#include <media/v4l2-ctrls.h>
#include <media/v4l2-subdev.h>
#include <media/v4l2-fwnode.h>

#define OV4689_REG_CTRL_MODE		CCI_REG8(0x0100)
#define OV4689_MODE_SW_STANDBY		0x0
#define OV4689_MODE_STREAMING		BIT(0)

#define OV4689_REG_CHIP_ID		CCI_REG16(0x300a)
#define CHIP_ID				0x004688

#define OV4689_REG_EXPOSURE		CCI_REG24(0x3500)
#define OV4689_EXPOSURE_MIN		4
#define OV4689_EXPOSURE_STEP		1

#define OV4689_REG_GAIN			CCI_REG16(0x3508)
#define OV4689_GAIN_STEP		1
#define OV4689_GAIN_DEFAULT		0x80

#define OV4689_REG_DIG_GAIN		CCI_REG16(0x352a)
#define OV4689_DIG_GAIN_MIN		1
#define OV4689_DIG_GAIN_MAX		0x7fff
#define OV4689_DIG_GAIN_STEP		1
#define OV4689_DIG_GAIN_DEFAULT		0x800

#define OV4689_REG_H_CROP_START		CCI_REG16(0x3800)
#define OV4689_REG_V_CROP_START		CCI_REG16(0x3802)
#define OV4689_REG_H_CROP_END		CCI_REG16(0x3804)
#define OV4689_REG_V_CROP_END		CCI_REG16(0x3806)
#define OV4689_REG_H_OUTPUT_SIZE	CCI_REG16(0x3808)
#define OV4689_REG_V_OUTPUT_SIZE	CCI_REG16(0x380a)

#define OV4689_REG_HTS			CCI_REG16(0x380c)
#define OV4689_HTS_DIVIDER		4
#define OV4689_HTS_MAX			0x7fff

#define OV4689_REG_VTS			CCI_REG16(0x380e)
#define OV4689_VTS_MAX			0x7fff

#define OV4689_REG_H_WIN_OFF		CCI_REG16(0x3810)
#define OV4689_REG_V_WIN_OFF		CCI_REG16(0x3812)

#define OV4689_REG_TIMING_FORMAT1	CCI_REG8(0x3820) /* Vertical */
#define OV4689_REG_TIMING_FORMAT2	CCI_REG8(0x3821) /* Horizontal */
#define OV4689_TIMING_FLIP_MASK		GENMASK(2, 1)
#define OV4689_TIMING_FLIP_ARRAY	BIT(1)
#define OV4689_TIMING_FLIP_DIGITAL	BIT(2)
#define OV4689_TIMING_FLIP_BOTH		(OV4689_TIMING_FLIP_ARRAY |\
					 OV4689_TIMING_FLIP_DIGITAL)

#define OV4689_REG_ANCHOR_LEFT_START	CCI_REG16(0x4020)
#define OV4689_ANCHOR_LEFT_START_DEF	576
#define OV4689_REG_ANCHOR_LEFT_END	CCI_REG16(0x4022)
#define OV4689_ANCHOR_LEFT_END_DEF	831
#define OV4689_REG_ANCHOR_RIGHT_START	CCI_REG16(0x4024)
#define OV4689_ANCHOR_RIGHT_START_DEF	1984
#define OV4689_REG_ANCHOR_RIGHT_END	CCI_REG16(0x4026)
#define OV4689_ANCHOR_RIGHT_END_DEF	2239

#define OV4689_REG_VFIFO_CTRL_01	CCI_REG8(0x4601)

#define OV4689_REG_WB_GAIN_RED		CCI_REG16(0x500c)
#define OV4689_REG_WB_GAIN_BLUE		CCI_REG16(0x5010)
#define OV4689_WB_GAIN_MIN		1
#define OV4689_WB_GAIN_MAX		0xfff
#define OV4689_WB_GAIN_STEP		1
#define OV4689_WB_GAIN_DEFAULT		0x400

#define OV4689_REG_TEST_PATTERN		CCI_REG8(0x5040)
#define OV4689_TEST_PATTERN_ENABLE	0x80
#define OV4689_TEST_PATTERN_DISABLE	0x0

#define OV4689_LANES			4
#define OV4689_XVCLK_FREQ		24000000

#define OV4689_PIXEL_ARRAY_WIDTH	2720
#define OV4689_PIXEL_ARRAY_HEIGHT	1536
#define OV4689_DUMMY_ROWS		8	/* 8 dummy rows on each side */
#define OV4689_DUMMY_COLUMNS		16	/* 16 dummy columns on each side */

static const char *const ov4689_supply_names[] = {
	"avdd", /* Analog power */
	"dovdd", /* Digital I/O power */
	"dvdd", /* Digital core power */
};

enum ov4689_mode_id {
	OV4689_MODE_2688_1520 = 0,
	OV4689_NUM_MODES,
};

struct ov4689_mode {
	enum ov4689_mode_id id;
	u32 width;
	u32 height;
	u32 hts_def;
	u32 hts_min;
	u32 vts_def;
	u32 exp_def;
	u32 pixel_rate;
	const struct cci_reg_sequence *reg_list;
	unsigned int num_regs;
};

struct ov4689 {
	struct device *dev;
	struct regmap *regmap;
	struct clk *xvclk;
	struct gpio_desc *reset_gpio;
	struct gpio_desc *pwdn_gpio;
	struct regulator_bulk_data supplies[ARRAY_SIZE(ov4689_supply_names)];

	struct v4l2_subdev subdev;
	struct media_pad pad;

	u32 clock_rate;

	struct v4l2_ctrl_handler ctrl_handler;
	struct v4l2_ctrl *exposure;

	const struct ov4689_mode *cur_mode;
};

#define to_ov4689(sd) container_of(sd, struct ov4689, subdev)

struct ov4689_gain_range {
	u32 logical_min;
	u32 logical_max;
	u32 offset;
	u32 divider;
	u32 physical_min;
	u32 physical_max;
};

/*
 * Xclk 24Mhz
 * max_framerate 90fps
 * mipi_datarate per lane 1008Mbps
 */
static const struct cci_reg_sequence ov4689_2688x1520_regs[] = {
	/* System control*/
	{ CCI_REG8(0x0103), 0x01 }, /* SC_CTRL0103 software_reset = 1 */
	{ CCI_REG8(0x3000), 0x20 }, /* SC_CMMN_PAD_OEN0 FSIN_output_enable = 1 */
	{ CCI_REG8(0x3021), 0x03 }, /*
				     * SC_CMMN_MISC_CTRL fst_stby_ctr = 0,
				     * sleep_no_latch_enable = 0
				     */

	/* AEC PK */
	{ CCI_REG8(0x3503), 0x04 }, /* AEC_MANUAL gain_input_as_sensor_gain_format = 1 */

	/* ADC and analog control*/
	{ CCI_REG8(0x3603), 0x40 },
	{ CCI_REG8(0x3604), 0x02 },
	{ CCI_REG8(0x3609), 0x12 },
	{ CCI_REG8(0x360c), 0x08 },
	{ CCI_REG8(0x360f), 0xe5 },
	{ CCI_REG8(0x3608), 0x8f },
	{ CCI_REG8(0x3611), 0x00 },
	{ CCI_REG8(0x3613), 0xf7 },
	{ CCI_REG8(0x3616), 0x58 },
	{ CCI_REG8(0x3619), 0x99 },
	{ CCI_REG8(0x361b), 0x60 },
	{ CCI_REG8(0x361e), 0x79 },
	{ CCI_REG8(0x3634), 0x10 },
	{ CCI_REG8(0x3635), 0x10 },
	{ CCI_REG8(0x3636), 0x15 },
	{ CCI_REG8(0x3646), 0x86 },
	{ CCI_REG8(0x364a), 0x0b },

	/* Sensor control */
	{ CCI_REG8(0x3700), 0x17 },
	{ CCI_REG8(0x3701), 0x22 },
	{ CCI_REG8(0x3703), 0x10 },
	{ CCI_REG8(0x370a), 0x37 },
	{ CCI_REG8(0x3706), 0x63 },
	{ CCI_REG8(0x3709), 0x3c },
	{ CCI_REG8(0x370c), 0x30 },
	{ CCI_REG8(0x3710), 0x24 },
	{ CCI_REG8(0x3720), 0x28 },
	{ CCI_REG8(0x3729), 0x7b },
	{ CCI_REG8(0x372b), 0xbd },
	{ CCI_REG8(0x372c), 0xbc },
	{ CCI_REG8(0x372e), 0x52 },
	{ CCI_REG8(0x373c), 0x0e },
	{ CCI_REG8(0x373e), 0x33 },
	{ CCI_REG8(0x3743), 0x10 },
	{ CCI_REG8(0x3744), 0x88 },
	{ CCI_REG8(0x3745), 0xc0 },
	{ CCI_REG8(0x374c), 0x00 },
	{ CCI_REG8(0x374e), 0x23 },
	{ CCI_REG8(0x3751), 0x7b },
	{ CCI_REG8(0x3753), 0xbd },
	{ CCI_REG8(0x3754), 0xbc },
	{ CCI_REG8(0x3756), 0x52 },
	{ CCI_REG8(0x376b), 0x20 },
	{ CCI_REG8(0x3774), 0x51 },
	{ CCI_REG8(0x3776), 0xbd },
	{ CCI_REG8(0x3777), 0xbd },
	{ CCI_REG8(0x3781), 0x18 },
	{ CCI_REG8(0x3783), 0x25 },
	{ CCI_REG8(0x3798), 0x1b },

	/* Timing control */
	{ CCI_REG8(0x3819), 0x01 }, /* VSYNC_END_L vsync_end_point[7:0] = 0x01 */

	/* OTP control */
	{ CCI_REG8(0x3d85), 0x36 }, /* OTP_REG85 OTP_power_up_load_setting_enable = 1,
				     * OTP_power_up_load_data_enable = 1,
				     * OTP_bist_select = 1 (compare with zero)
				     */
	{ CCI_REG8(0x3d8c), 0x71 }, /* OTP_SETTING_STT_ADDRESS_H */
	{ CCI_REG8(0x3d8d), 0xcb }, /* OTP_SETTING_STT_ADDRESS_L */

	/* BLC registers*/
	{ CCI_REG8(0x4001), 0x40 }, /* DEBUG_MODE */
	{ CCI_REG8(0x401b), 0x00 }, /* DEBUG_MODE */
	{ CCI_REG8(0x401d), 0x00 }, /* DEBUG_MODE */
	{ CCI_REG8(0x401f), 0x00 }, /* DEBUG_MODE */

	/* ADC sync control */
	{ CCI_REG8(0x4500), 0x6c }, /* ADC_SYNC_CTRL */
	{ CCI_REG8(0x4503), 0x01 }, /* ADC_SYNC_CTRL */

	/* Temperature monitor */
	{ CCI_REG8(0x4d00), 0x04 }, /* TPM_CTRL_00 tmp_slope[15:8] = 0x04 */
	{ CCI_REG8(0x4d01), 0x42 }, /* TPM_CTRL_01 tmp_slope[7:0] = 0x42 */
	{ CCI_REG8(0x4d02), 0xd1 }, /* TPM_CTRL_02 tpm_offset[31:24] = 0xd1 */
	{ CCI_REG8(0x4d03), 0x93 }, /* TPM_CTRL_03 tpm_offset[23:16] = 0x93 */
	{ CCI_REG8(0x4d04), 0xf5 }, /* TPM_CTRL_04 tpm_offset[15:8]  = 0xf5 */
	{ CCI_REG8(0x4d05), 0xc1 }, /* TPM_CTRL_05 tpm_offset[7:0]   = 0xc1 */

	/* pre-ISP control */
	{ CCI_REG8(0x5050), 0x0c }, /* DEBUG_MODE */

	/* OTP-DPC control */
	{ CCI_REG8(0x5501), 0x10 }, /* OTP_DPC_START_L otp_start_address[7:0] = 0x10 */
	{ CCI_REG8(0x5503), 0x0f }, /* OTP_DPC_END_L otp_end_address[7:0] = 0x0f */
};

static const struct ov4689_mode supported_modes[] = {
	{
		.id = OV4689_MODE_2688_1520,
		.width = 2688,
		.height = 1520,
		.exp_def = 1536,
		.hts_def = 10296,
		.hts_min = 3432,
		.vts_def = 1554,
		.pixel_rate = 480000000,
		.reg_list = ov4689_2688x1520_regs,
		.num_regs = ARRAY_SIZE(ov4689_2688x1520_regs),
	},
};

static const u64 link_freq_menu_items[] = { 504000000 };

static const char *const ov4689_test_pattern_menu[] = {
	"Disabled",
	"Vertical Color Bar Type 1",
	"Vertical Color Bar Type 2",
	"Vertical Color Bar Type 3",
	"Vertical Color Bar Type 4"
};

/*
 * These coefficients are based on those used in Rockchip's camera
 * engine, with minor tweaks for continuity.
 */
static const struct ov4689_gain_range ov4689_gain_ranges[] = {
	{
		.logical_min = 0,
		.logical_max = 255,
		.offset = 0,
		.divider = 1,
		.physical_min = 0,
		.physical_max = 255,
	},
	{
		.logical_min = 256,
		.logical_max = 511,
		.offset = 252,
		.divider = 2,
		.physical_min = 376,
		.physical_max = 504,
	},
	{
		.logical_min = 512,
		.logical_max = 1023,
		.offset = 758,
		.divider = 4,
		.physical_min = 884,
		.physical_max = 1012,
	},
	{
		.logical_min = 1024,
		.logical_max = 2047,
		.offset = 1788,
		.divider = 8,
		.physical_min = 1912,
		.physical_max = 2047,
	},
};

static void ov4689_fill_fmt(const struct ov4689_mode *mode,
			    struct v4l2_mbus_framefmt *fmt)
{
	fmt->code = MEDIA_BUS_FMT_SBGGR10_1X10;
	fmt->width = mode->width;
	fmt->height = mode->height;
	fmt->field = V4L2_FIELD_NONE;
}

static int ov4689_set_fmt(struct v4l2_subdev *sd,
			  struct v4l2_subdev_state *sd_state,
			  struct v4l2_subdev_format *fmt)
{
	struct v4l2_mbus_framefmt *mbus_fmt = &fmt->format;
	struct ov4689 *ov4689 = to_ov4689(sd);

	/* only one mode supported for now */
	ov4689_fill_fmt(ov4689->cur_mode, mbus_fmt);

	return 0;
}

static int ov4689_enum_mbus_code(struct v4l2_subdev *sd,
				 struct v4l2_subdev_state *sd_state,
				 struct v4l2_subdev_mbus_code_enum *code)
{
	if (code->index != 0)
		return -EINVAL;
	code->code = MEDIA_BUS_FMT_SBGGR10_1X10;

	return 0;
}

static int ov4689_enum_frame_sizes(struct v4l2_subdev *sd,
				   struct v4l2_subdev_state *sd_state,
				   struct v4l2_subdev_frame_size_enum *fse)
{
	if (fse->index >= ARRAY_SIZE(supported_modes))
		return -EINVAL;

	if (fse->code != MEDIA_BUS_FMT_SBGGR10_1X10)
		return -EINVAL;

	fse->min_width = supported_modes[fse->index].width;
	fse->max_width = supported_modes[fse->index].width;
	fse->max_height = supported_modes[fse->index].height;
	fse->min_height = supported_modes[fse->index].height;

	return 0;
}

static int ov4689_enable_test_pattern(struct ov4689 *ov4689, u32 pattern)
{
	u32 val;

	if (pattern)
		val = (pattern - 1) | OV4689_TEST_PATTERN_ENABLE;
	else
		val = OV4689_TEST_PATTERN_DISABLE;

	return cci_write(ov4689->regmap, OV4689_REG_TEST_PATTERN,
			 val, NULL);
}

static int ov4689_get_selection(struct v4l2_subdev *sd,
				struct v4l2_subdev_state *state,
				struct v4l2_subdev_selection *sel)
{
	if (sel->which != V4L2_SUBDEV_FORMAT_ACTIVE)
		return -EINVAL;

	switch (sel->target) {
	case V4L2_SEL_TGT_CROP_BOUNDS:
		sel->r.top = 0;
		sel->r.left = 0;
		sel->r.width = OV4689_PIXEL_ARRAY_WIDTH;
		sel->r.height = OV4689_PIXEL_ARRAY_HEIGHT;
		return 0;
	case V4L2_SEL_TGT_CROP:
	case V4L2_SEL_TGT_CROP_DEFAULT:
		sel->r.top = OV4689_DUMMY_ROWS;
		sel->r.left = OV4689_DUMMY_COLUMNS;
		sel->r.width =
			OV4689_PIXEL_ARRAY_WIDTH - 2 * OV4689_DUMMY_COLUMNS;
		sel->r.height =
			OV4689_PIXEL_ARRAY_HEIGHT - 2 * OV4689_DUMMY_ROWS;
		return 0;
	}

	return -EINVAL;
}

static int ov4689_setup_timings(struct ov4689 *ov4689)
{
	const struct ov4689_mode *mode = ov4689->cur_mode;
	struct regmap *rm = ov4689->regmap;
	int ret = 0;

	cci_write(rm, OV4689_REG_H_CROP_START, 8, &ret);
	cci_write(rm, OV4689_REG_V_CROP_START, 8, &ret);
	cci_write(rm, OV4689_REG_H_CROP_END, 2711, &ret);
	cci_write(rm, OV4689_REG_V_CROP_END, 1531, &ret);

	cci_write(rm, OV4689_REG_H_OUTPUT_SIZE, mode->width, &ret);
	cci_write(rm, OV4689_REG_V_OUTPUT_SIZE, mode->height, &ret);

	cci_write(rm, OV4689_REG_H_WIN_OFF, 8, &ret);
	cci_write(rm, OV4689_REG_V_WIN_OFF, 4, &ret);

	cci_write(rm, OV4689_REG_VFIFO_CTRL_01, 167, &ret);

	return ret;
}

static int ov4689_setup_blc_anchors(struct ov4689 *ov4689)
{
	struct regmap *rm = ov4689->regmap;
	int ret = 0;

	cci_write(rm, OV4689_REG_ANCHOR_LEFT_START, 16, &ret);
	cci_write(rm, OV4689_REG_ANCHOR_LEFT_END, 1999, &ret);
	cci_write(rm, OV4689_REG_ANCHOR_RIGHT_START, 2400, &ret);
	cci_write(rm, OV4689_REG_ANCHOR_RIGHT_END, 2415, &ret);

	return ret;
}

static int ov4689_s_stream(struct v4l2_subdev *sd, int on)
{
	struct ov4689 *ov4689 = to_ov4689(sd);
	struct v4l2_subdev_state *sd_state;
	struct device *dev = ov4689->dev;
	int ret = 0;

	sd_state = v4l2_subdev_lock_and_get_active_state(&ov4689->subdev);

	if (on) {
		ret = pm_runtime_resume_and_get(dev);
		if (ret < 0)
			goto unlock_and_return;

		ret = cci_multi_reg_write(ov4689->regmap,
					  ov4689->cur_mode->reg_list,
					  ov4689->cur_mode->num_regs,
					  NULL);
		if (ret) {
			pm_runtime_put(dev);
			goto unlock_and_return;
		}

		ret = ov4689_setup_timings(ov4689);
		if (ret) {
			pm_runtime_put(dev);
			goto unlock_and_return;
		}

		ret = ov4689_setup_blc_anchors(ov4689);
		if (ret) {
			pm_runtime_put(dev);
			goto unlock_and_return;
		}

		ret = __v4l2_ctrl_handler_setup(&ov4689->ctrl_handler);
		if (ret) {
			pm_runtime_put(dev);
			goto unlock_and_return;
		}

		ret = cci_write(ov4689->regmap, OV4689_REG_CTRL_MODE,
				OV4689_MODE_STREAMING, NULL);
		if (ret) {
			pm_runtime_put(dev);
			goto unlock_and_return;
		}
	} else {
		cci_write(ov4689->regmap, OV4689_REG_CTRL_MODE,
			  OV4689_MODE_SW_STANDBY, NULL);
		pm_runtime_put_autosuspend(dev);
	}

unlock_and_return:
	v4l2_subdev_unlock_state(sd_state);

	return ret;
}

/* Calculate the delay in us by clock rate and clock cycles */
static inline u32 ov4689_cal_delay(struct ov4689 *ov4689, u32 cycles)
{
	return DIV_ROUND_UP(cycles * 1000,
			    DIV_ROUND_UP(ov4689->clock_rate, 1000));
}

static int __maybe_unused ov4689_power_on(struct device *dev)
{
	struct v4l2_subdev *sd = dev_get_drvdata(dev);
	struct ov4689 *ov4689 = to_ov4689(sd);
	u32 delay_us;
	int ret;

	ret = clk_prepare_enable(ov4689->xvclk);
	if (ret < 0) {
		dev_err(dev, "Failed to enable xvclk\n");
		return ret;
	}

	gpiod_set_value_cansleep(ov4689->reset_gpio, 1);

	ret = regulator_bulk_enable(ARRAY_SIZE(ov4689_supply_names),
				    ov4689->supplies);
	if (ret < 0) {
		dev_err(dev, "Failed to enable regulators\n");
		goto disable_clk;
	}

	gpiod_set_value_cansleep(ov4689->reset_gpio, 0);
	usleep_range(500, 1000);
	gpiod_set_value_cansleep(ov4689->pwdn_gpio, 0);

	/* 8192 cycles prior to first SCCB transaction */
	delay_us = ov4689_cal_delay(ov4689, 8192);
	usleep_range(delay_us, delay_us * 2);

	return 0;

disable_clk:
	clk_disable_unprepare(ov4689->xvclk);

	return ret;
}

static int __maybe_unused ov4689_power_off(struct device *dev)
{
	struct v4l2_subdev *sd = dev_get_drvdata(dev);
	struct ov4689 *ov4689 = to_ov4689(sd);

	gpiod_set_value_cansleep(ov4689->pwdn_gpio, 1);
	clk_disable_unprepare(ov4689->xvclk);
	gpiod_set_value_cansleep(ov4689->reset_gpio, 1);
	regulator_bulk_disable(ARRAY_SIZE(ov4689_supply_names),
			       ov4689->supplies);
	return 0;
}

static int ov4689_init_state(struct v4l2_subdev *sd,
			     struct v4l2_subdev_state *sd_state)
{
	struct v4l2_mbus_framefmt *fmt =
		v4l2_subdev_state_get_format(sd_state, 0);

	ov4689_fill_fmt(&supported_modes[OV4689_MODE_2688_1520], fmt);

	return 0;
}

static const struct dev_pm_ops ov4689_pm_ops = {
	SET_RUNTIME_PM_OPS(ov4689_power_off, ov4689_power_on, NULL)
};

static const struct v4l2_subdev_video_ops ov4689_video_ops = {
	.s_stream = ov4689_s_stream,
};

static const struct v4l2_subdev_pad_ops ov4689_pad_ops = {
	.enum_mbus_code = ov4689_enum_mbus_code,
	.enum_frame_size = ov4689_enum_frame_sizes,
	.get_fmt = v4l2_subdev_get_fmt,
	.set_fmt = ov4689_set_fmt,
	.get_selection = ov4689_get_selection,
};

static const struct v4l2_subdev_internal_ops ov4689_internal_ops = {
	.init_state = ov4689_init_state,
};

static const struct v4l2_subdev_ops ov4689_subdev_ops = {
	.video = &ov4689_video_ops,
	.pad = &ov4689_pad_ops,
};

/*
 * Map userspace (logical) gain to sensor (physical) gain using
 * ov4689_gain_ranges table.
 */
static int ov4689_map_gain(struct ov4689 *ov4689, int logical_gain, int *result)
{
	const struct ov4689_gain_range *range;
	unsigned int n;

	for (n = 0; n < ARRAY_SIZE(ov4689_gain_ranges); n++) {
		if (logical_gain >= ov4689_gain_ranges[n].logical_min &&
		    logical_gain <= ov4689_gain_ranges[n].logical_max)
			break;
	}

	if (n == ARRAY_SIZE(ov4689_gain_ranges)) {
		dev_warn_ratelimited(ov4689->dev,
				     "no mapping found for gain %d\n",
				     logical_gain);
		return -EINVAL;
	}

	range = &ov4689_gain_ranges[n];

	*result = clamp(range->offset + (logical_gain) / range->divider,
			range->physical_min, range->physical_max);
	return 0;
}

static int ov4689_set_ctrl(struct v4l2_ctrl *ctrl)
{
	struct ov4689 *ov4689 =
		container_of(ctrl->handler, struct ov4689, ctrl_handler);
	struct regmap *regmap = ov4689->regmap;
	struct device *dev = ov4689->dev;
	int sensor_gain = 0;
	s64 max_expo;
	int ret = 0;

	/* Propagate change of current control to all related controls */
	switch (ctrl->id) {
	case V4L2_CID_VBLANK:
		/* Update max exposure while meeting expected vblanking */
		max_expo = ov4689->cur_mode->height + ctrl->val - 4;
		__v4l2_ctrl_modify_range(ov4689->exposure,
					 ov4689->exposure->minimum, max_expo,
					 ov4689->exposure->step,
					 ov4689->exposure->default_value);
		break;
	}

	if (!pm_runtime_get_if_in_use(dev))
		return 0;

	switch (ctrl->id) {
	case V4L2_CID_EXPOSURE:
		/* 4 least significant bits of exposure are fractional part */
		cci_write(regmap, OV4689_REG_EXPOSURE, ctrl->val << 4, &ret);
		break;
	case V4L2_CID_ANALOGUE_GAIN:
		ret = ov4689_map_gain(ov4689, ctrl->val, &sensor_gain);
		cci_write(regmap, OV4689_REG_GAIN, sensor_gain, &ret);
		break;
	case V4L2_CID_VBLANK:
		cci_write(regmap, OV4689_REG_VTS,
			  ctrl->val + ov4689->cur_mode->height, &ret);
		break;
	case V4L2_CID_TEST_PATTERN:
		ret = ov4689_enable_test_pattern(ov4689, ctrl->val);
		break;
	case V4L2_CID_HBLANK:
		cci_write(regmap, OV4689_REG_HTS,
			  (ctrl->val + ov4689->cur_mode->width) /
			  OV4689_HTS_DIVIDER, &ret);
		break;
	case V4L2_CID_VFLIP:
		cci_update_bits(regmap, OV4689_REG_TIMING_FORMAT1,
				OV4689_TIMING_FLIP_MASK,
				ctrl->val ? OV4689_TIMING_FLIP_BOTH : 0, &ret);
		break;
	case V4L2_CID_HFLIP:
		cci_update_bits(regmap, OV4689_REG_TIMING_FORMAT2,
				OV4689_TIMING_FLIP_MASK,
				ctrl->val ? 0 : OV4689_TIMING_FLIP_BOTH, &ret);
		break;
	case V4L2_CID_DIGITAL_GAIN:
		cci_write(regmap, OV4689_REG_DIG_GAIN, ctrl->val, &ret);
		break;
	case V4L2_CID_RED_BALANCE:
		cci_write(regmap, OV4689_REG_WB_GAIN_RED, ctrl->val, &ret);
		break;
	case V4L2_CID_BLUE_BALANCE:
		cci_write(regmap, OV4689_REG_WB_GAIN_BLUE, ctrl->val, &ret);
		break;
	default:
		dev_warn(dev, "%s Unhandled id:0x%x, val:0x%x\n",
			 __func__, ctrl->id, ctrl->val);
		ret = -EINVAL;
		break;
	}

	pm_runtime_put_autosuspend(dev);

	return ret;
}

static const struct v4l2_ctrl_ops ov4689_ctrl_ops = {
	.s_ctrl = ov4689_set_ctrl,
};

static int ov4689_initialize_controls(struct ov4689 *ov4689)
{
	struct i2c_client *client = v4l2_get_subdevdata(&ov4689->subdev);
	struct v4l2_fwnode_device_properties props;
	struct v4l2_ctrl_handler *handler;
	const struct ov4689_mode *mode;
	s64 exposure_max, vblank_def;
	s64 hblank_def, hblank_min;
	struct v4l2_ctrl *ctrl;
	int ret;

	handler = &ov4689->ctrl_handler;
	mode = ov4689->cur_mode;
	ret = v4l2_ctrl_handler_init(handler, 15);
	if (ret)
		return ret;

	ctrl = v4l2_ctrl_new_int_menu(handler, NULL, V4L2_CID_LINK_FREQ, 0, 0,
				      link_freq_menu_items);
	if (ctrl)
		ctrl->flags |= V4L2_CTRL_FLAG_READ_ONLY;

	v4l2_ctrl_new_std(handler, NULL, V4L2_CID_PIXEL_RATE, 0,
			  mode->pixel_rate, 1, mode->pixel_rate);

	hblank_def = mode->hts_def - mode->width;
	hblank_min = mode->hts_min - mode->width;
	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_HBLANK,
			  hblank_min, OV4689_HTS_MAX - mode->width,
			  OV4689_HTS_DIVIDER, hblank_def);

	vblank_def = mode->vts_def - mode->height;
	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_VBLANK,
			  vblank_def, OV4689_VTS_MAX - mode->height, 1,
			  vblank_def);

	exposure_max = mode->vts_def - 4;
	ov4689->exposure =
		v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_EXPOSURE,
				  OV4689_EXPOSURE_MIN, exposure_max,
				  OV4689_EXPOSURE_STEP, mode->exp_def);

	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_ANALOGUE_GAIN,
			  ov4689_gain_ranges[0].logical_min,
			  ov4689_gain_ranges[ARRAY_SIZE(ov4689_gain_ranges) - 1]
				  .logical_max,
			  OV4689_GAIN_STEP, OV4689_GAIN_DEFAULT);

	v4l2_ctrl_new_std_menu_items(handler, &ov4689_ctrl_ops,
				     V4L2_CID_TEST_PATTERN,
				     ARRAY_SIZE(ov4689_test_pattern_menu) - 1,
				     0, 0, ov4689_test_pattern_menu);

	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_VFLIP, 0, 1, 1, 0);
	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_HFLIP, 0, 1, 1, 0);

	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_DIGITAL_GAIN,
			  OV4689_DIG_GAIN_MIN, OV4689_DIG_GAIN_MAX,
			  OV4689_DIG_GAIN_STEP, OV4689_DIG_GAIN_DEFAULT);

	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_RED_BALANCE,
			  OV4689_WB_GAIN_MIN, OV4689_WB_GAIN_MAX,
			  OV4689_WB_GAIN_STEP, OV4689_WB_GAIN_DEFAULT);

	v4l2_ctrl_new_std(handler, &ov4689_ctrl_ops, V4L2_CID_BLUE_BALANCE,
			  OV4689_WB_GAIN_MIN, OV4689_WB_GAIN_MAX,
			  OV4689_WB_GAIN_STEP, OV4689_WB_GAIN_DEFAULT);

	if (handler->error) {
		ret = handler->error;
		dev_err(ov4689->dev, "Failed to init controls(%d)\n", ret);
		goto err_free_handler;
	}

	ret = v4l2_fwnode_device_parse(&client->dev, &props);
	if (ret)
		goto err_free_handler;

	ret = v4l2_ctrl_new_fwnode_properties(handler, &ov4689_ctrl_ops,
					      &props);
	if (ret)
		goto err_free_handler;

	ov4689->subdev.ctrl_handler = handler;

	return 0;

err_free_handler:
	v4l2_ctrl_handler_free(handler);

	return ret;
}

static int ov4689_check_sensor_id(struct ov4689 *ov4689,
				  struct i2c_client *client)
{
	struct device *dev = ov4689->dev;
	u64 id = 0;
	int ret;

	ret = cci_read(ov4689->regmap, OV4689_REG_CHIP_ID, &id, NULL);
	if (ret) {
		dev_err(dev, "Cannot read sensor ID\n");
		return ret;
	}

	if (id != CHIP_ID) {
		dev_err(dev, "Unexpected sensor ID %06llx, expected %06x\n",
			id, CHIP_ID);
		return -ENODEV;
	}

	dev_info(dev, "Detected OV%06x sensor\n", CHIP_ID);

	return 0;
}

static int ov4689_configure_regulators(struct ov4689 *ov4689)
{
	unsigned int i;

	for (i = 0; i < ARRAY_SIZE(ov4689_supply_names); i++)
		ov4689->supplies[i].supply = ov4689_supply_names[i];

	return devm_regulator_bulk_get(ov4689->dev,
				       ARRAY_SIZE(ov4689_supply_names),
				       ov4689->supplies);
}

static u64 ov4689_check_link_frequency(struct v4l2_fwnode_endpoint *ep)
{
	const u64 *freqs = link_freq_menu_items;
	unsigned int i, j;

	for (i = 0; i < ARRAY_SIZE(link_freq_menu_items); i++) {
		for (j = 0; j < ep->nr_of_link_frequencies; j++)
			if (freqs[i] == ep->link_frequencies[j])
				return freqs[i];
	}

	return 0;
}

static int ov4689_check_hwcfg(struct device *dev)
{
	struct fwnode_handle *fwnode = dev_fwnode(dev);
	struct v4l2_fwnode_endpoint bus_cfg = {
		.bus_type = V4L2_MBUS_CSI2_DPHY,
	};
	struct fwnode_handle *endpoint;
	int ret;

	endpoint = fwnode_graph_get_next_endpoint(fwnode, NULL);
	if (!endpoint)
		return -EINVAL;

	ret = v4l2_fwnode_endpoint_alloc_parse(endpoint, &bus_cfg);
	fwnode_handle_put(endpoint);
	if (ret)
		return ret;

	if (bus_cfg.bus.mipi_csi2.num_data_lanes != OV4689_LANES) {
		dev_err(dev, "Only a 4-lane CSI2 config is supported");
		ret = -EINVAL;
		goto out_free_bus_cfg;
	}

	if (!ov4689_check_link_frequency(&bus_cfg)) {
		dev_err(dev, "No supported link frequency found\n");
		ret = -EINVAL;
	}

out_free_bus_cfg:
	v4l2_fwnode_endpoint_free(&bus_cfg);

	return ret;
}

static int ov4689_probe(struct i2c_client *client)
{
	struct device *dev = &client->dev;
	struct v4l2_subdev *sd;
	struct ov4689 *ov4689;
	int ret;

	ret = ov4689_check_hwcfg(dev);
	if (ret)
		return ret;

	ov4689 = devm_kzalloc(dev, sizeof(*ov4689), GFP_KERNEL);
	if (!ov4689)
		return -ENOMEM;

	ov4689->dev = dev;

	ov4689->cur_mode = &supported_modes[OV4689_MODE_2688_1520];

	ov4689->xvclk = devm_v4l2_sensor_clk_get(dev, NULL);
	if (IS_ERR(ov4689->xvclk))
		return dev_err_probe(dev, PTR_ERR(ov4689->xvclk),
				     "Failed to get external clock\n");

	ov4689->clock_rate = clk_get_rate(ov4689->xvclk);
	if (ov4689->clock_rate != OV4689_XVCLK_FREQ) {
		dev_err(dev,
			"External clock rate mismatch: got %d Hz, expected %d Hz\n",
			ov4689->clock_rate, OV4689_XVCLK_FREQ);
		return -EINVAL;
	}

	ov4689->regmap = devm_cci_regmap_init_i2c(client, 16);
	if (IS_ERR(ov4689->regmap)) {
		ret = PTR_ERR(ov4689->regmap);
		dev_err(dev, "failed to initialize CCI: %d\n", ret);
		return ret;
	}

	ov4689->reset_gpio = devm_gpiod_get_optional(dev, "reset",
						     GPIOD_OUT_LOW);
	if (IS_ERR(ov4689->reset_gpio)) {
		dev_err(dev, "Failed to get reset-gpios\n");
		return PTR_ERR(ov4689->reset_gpio);
	}

	ov4689->pwdn_gpio = devm_gpiod_get_optional(dev, "pwdn", GPIOD_OUT_LOW);
	if (IS_ERR(ov4689->pwdn_gpio)) {
		dev_err(dev, "Failed to get pwdn-gpios\n");
		return PTR_ERR(ov4689->pwdn_gpio);
	}

	ret = ov4689_configure_regulators(ov4689);
	if (ret)
		return dev_err_probe(dev, ret,
				     "Failed to get power regulators\n");

	sd = &ov4689->subdev;
	v4l2_i2c_subdev_init(sd, client, &ov4689_subdev_ops);
	sd->internal_ops = &ov4689_internal_ops;
	sd->flags |= V4L2_SUBDEV_FL_HAS_DEVNODE;
	ret = ov4689_initialize_controls(ov4689);
	if (ret) {
		dev_err(dev, "Failed to initialize controls\n");
		return ret;
	}

	ret = ov4689_power_on(dev);
	if (ret)
		goto err_free_handler;

	ret = ov4689_check_sensor_id(ov4689, client);
	if (ret)
		goto err_power_off;


	sd->entity.function = MEDIA_ENT_F_CAM_SENSOR;
	ov4689->pad.flags = MEDIA_PAD_FL_SOURCE;
	ret = media_entity_pads_init(&sd->entity, 1, &ov4689->pad);
	if (ret < 0)
		goto err_power_off;

	sd->state_lock = ov4689->ctrl_handler.lock;
	ret = v4l2_subdev_init_finalize(sd);
	if (ret) {
		dev_err(dev, "Could not register v4l2 device\n");
		goto err_clean_entity;
	}

	pm_runtime_set_active(dev);
	pm_runtime_get_noresume(dev);
	pm_runtime_enable(dev);
	pm_runtime_set_autosuspend_delay(dev, 1000);
	pm_runtime_use_autosuspend(dev);

	ret = v4l2_async_register_subdev_sensor(sd);
	if (ret) {
		dev_err(dev, "v4l2 async register subdev failed\n");
		goto err_clean_subdev_pm;
	}

	pm_runtime_put_autosuspend(dev);

	return 0;

err_clean_subdev_pm:
	pm_runtime_disable(dev);
	pm_runtime_put_noidle(dev);
	v4l2_subdev_cleanup(sd);
err_clean_entity:
	media_entity_cleanup(&sd->entity);
err_power_off:
	ov4689_power_off(dev);
err_free_handler:
	v4l2_ctrl_handler_free(&ov4689->ctrl_handler);

	return ret;
}

static void ov4689_remove(struct i2c_client *client)
{
	struct v4l2_subdev *sd = i2c_get_clientdata(client);
	struct ov4689 *ov4689 = to_ov4689(sd);

	v4l2_async_unregister_subdev(sd);
	media_entity_cleanup(&sd->entity);
	v4l2_subdev_cleanup(sd);
	v4l2_ctrl_handler_free(&ov4689->ctrl_handler);

	pm_runtime_disable(&client->dev);
	if (!pm_runtime_status_suspended(&client->dev))
		ov4689_power_off(&client->dev);
	pm_runtime_set_suspended(&client->dev);
}

static const struct of_device_id ov4689_of_match[] = {
	{ .compatible = "ovti,ov4689" },
	{},
};
MODULE_DEVICE_TABLE(of, ov4689_of_match);

static struct i2c_driver ov4689_i2c_driver = {
	.driver = {
		.name = "ov4689",
		.pm = &ov4689_pm_ops,
		.of_match_table = ov4689_of_match,
	},
	.probe = ov4689_probe,
	.remove	= ov4689_remove,
};

module_i2c_driver(ov4689_i2c_driver);

MODULE_DESCRIPTION("OmniVision ov4689 sensor driver");
MODULE_LICENSE("GPL");