/*
 * drivers/display/lcd/lcd_bl_ldim/iw7027.c
 *
 * 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 named License,
 * or 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.
 *
 */

#include <common.h>
#include <malloc.h>
#include <spi.h>
#include <asm/arch/gpio.h>
#ifdef CONFIG_OF_LIBFDT
#include <libfdt.h>
#endif
#include <amlogic/aml_lcd.h>
#include <amlogic/aml_ldim.h>
#include "../aml_lcd_reg.h"
#include "../aml_lcd_common.h"
#include "ldim_drv.h"
#include "ldim_dev_drv.h"

#define NORMAL_MSG            (0<<7)
#define BROADCAST_MSG         (1<<7)
#define BLOCK_DATA            (0<<6)
#define SINGLE_DATA           (1<<6)
#define IW7027_DEV_ADDR        1

static int iw7027_on_flag;

struct iw7027 {
	int cs_hold_delay;
	int cs_clk_delay;
	unsigned char cmd_size;
	unsigned char *init_data;
	unsigned int init_data_cnt;
	unsigned int chip_cnt;
	struct spi_slave *spi;
};
static struct iw7027 *bl_iw7027;

static unsigned char *val_brightness;

#if 0
static unsigned char iw7027_init_data[LDIM_INIT_ON_SIZE] = {
	0xc0, 0x23, 0x03,
	0xc0, 0x24, 0xff,
	0xc0, 0x25, 0x00,
	0xc0, 0x26, 0x00,
	0xc0, 0x27, 0x60,
	0xc0, 0x29, 0x00,
	0xc0, 0x2a, 0x00,
	0xc0, 0x2b, 0x00,
	0xc0, 0x2c, 0x73,
	0xc0, 0x2d, 0x37,
	0xc0, 0x31, 0x93,
	0xc0, 0x32, 0x0f,
	0xc0, 0x33, 0xff,
	0xc0, 0x34, 0xc8,
	0xc0, 0x35, 0xbf,
	0xff, 0x00, 0x00,
};
#endif

//iw7027 register write
static int iw7027_wreg(struct spi_slave *spi, unsigned char addr, unsigned char val)
{
	unsigned char tbuf[3];
	int ret;

	if (lcd_debug_print_flag)
		LDIMPR("%s: 0x%02x = 0x%02x\n", __func__, addr, val);

	tbuf[0] = NORMAL_MSG | SINGLE_DATA | IW7027_DEV_ADDR;
	tbuf[1] = addr & 0x7f;
	tbuf[2] = val;
	ret = ldim_spi_write(spi, tbuf, 3);

	return ret;
}

//iw7027 register read 1 bytes
static int iw7027_rreg(struct spi_slave *spi, unsigned char addr,
		unsigned char *val, int rlen)
{
	unsigned char tbuf[20], rbuf[20];
	int ret, n, xlen;
	unsigned char chip_id = 0x01;

	/*set read flag*/
	if (addr > 0x80)
		iw7027_wreg(spi, 0x78, 0x80);

	n = bl_iw7027->chip_cnt + rlen;
	if (rlen == 1)
		xlen = 2 + n;
	else
		xlen = 3 + n;

	if (rlen == 1) {
		tbuf[0] = NORMAL_MSG | SINGLE_DATA | (chip_id & 0x3f);
		tbuf[1] = (addr & 0x7f) | 0x80;
		memset(&tbuf[2], 0, n);
	} else {
		tbuf[0] = NORMAL_MSG | BLOCK_DATA | (chip_id & 0x3f);
		tbuf[1] = rlen;
		tbuf[2] = (addr & 0x7f) | 0x80;
		memset(&tbuf[3], 0, n);
	}

	ret = ldim_spi_read(spi, tbuf, xlen, rbuf, rlen);
	*val = rbuf[3];

	/*set read flag*/
	if (addr > 0x80)
		iw7027_wreg(spi, 0x78, 0x00);

	return ret;
}

//iw7027 block write
static int iw7027_wregs(struct spi_slave *spi, unsigned char addr,
		unsigned char *val, int len)
{
	unsigned char tbuf[35];
	int ret, i;
	int n, xlen;

	if (lcd_debug_print_flag) {
		LDIMPR("%s: ", __func__);
		for (i = 0; i < len; i++)
			printf("0x%02x ", val[i]);
		printf("\n");
	}
/*
	len = 0x20;

	//tbuf[0] = NORMAL_MSG | BLOCK_DATA | IW7027_DEV_ADDR;
	tbuf[0] = BROADCAST_MSG | BLOCK_DATA | 0x00;
	tbuf[1] = len;
	tbuf[2] = addr & 0x7f;
	memcpy(&tbuf[3], val, len);
	ret = ldim_spi_write(spi, tbuf, (len + 3));
*/

	if (len == 0) {
		LDIMERR("%s: len is 0\n", __func__);
		return -1;
	}
	n = bl_iw7027->chip_cnt - 1;
	if (len == 1)
		xlen = 2 + 1 + n;
	else
		xlen = 3 + len + n;

	if (len == 1) {
		tbuf[0] = BROADCAST_MSG | SINGLE_DATA | 0x00;
		tbuf[1] = addr & 0x7f;
		tbuf[2] = val[0];
		memset(&tbuf[3], 0, n);
	} else {
		tbuf[0] = BROADCAST_MSG | BLOCK_DATA | 0x00;
		tbuf[1] = len;
		tbuf[2] = addr & 0x7f;
		memcpy(&tbuf[3], val, len);
		memset(&tbuf[3 + len], 0, n);
	}
	ret = ldim_spi_write(spi, tbuf, xlen);

	return ret;
}

static int ldim_power_cmd_dynamic_size(void)
{
	unsigned char *table;
	int i = 0, j, step = 0, max_len = 0;
	unsigned char type, cmd_size;
	int delay_ms, ret = 0;

	table = bl_iw7027->init_data;
	max_len = bl_iw7027->init_data_cnt;

	while ((i + 1) < max_len) {
		type = table[i];
		if (type == LCD_EXT_CMD_TYPE_END)
			break;
		if (lcd_debug_print_flag) {
			LDIMPR("%s: step %d: type=0x%02x, cmd_size=%d\n",
				__func__, step, type, table[i+1]);
		}
		cmd_size = table[i + 1];
		if (cmd_size == 0)
			goto power_cmd_dynamic_next;
		if ((i + 2 + cmd_size) > max_len)
			break;

		if (type == LCD_EXT_CMD_TYPE_NONE) {
			/* do nothing */
		} else if (type == LCD_EXT_CMD_TYPE_DELAY) {
			delay_ms = 0;
			for (j = 0; j < cmd_size; j++)
				delay_ms += table[i + 2 + j];
			if (delay_ms > 0)
				mdelay(delay_ms);
		} else if (type == LCD_EXT_CMD_TYPE_CMD) {
			ret = iw7027_wregs(bl_iw7027->spi,
				table[i + 2], &table[i + 3], (cmd_size - 1));
			udelay(1);
		} else if (type == LCD_EXT_CMD_TYPE_CMD_DELAY) {
			ret = iw7027_wregs(bl_iw7027->spi,
				table[i + 2], &table[i + 3], (cmd_size - 2));
			udelay(1);
			if (table[i + 4] > 0)
				mdelay(table[i + 4]);
		} else {
			LDIMERR("%s: type 0x%02x invalid\n", __func__, type);
		}
power_cmd_dynamic_next:
		i += (cmd_size + 2);
		step++;
	}

	return ret;
}

static int ldim_power_cmd_fixed_size(void)
{
	unsigned char *table;
	int i = 0, j, step = 0, max_len = 0;
	unsigned char type, cmd_size;
	int delay_ms, ret = 0;

	cmd_size = bl_iw7027->cmd_size;
	if (cmd_size < 2) {
		LDIMERR("%s: invalid cmd_size %d\n", __func__, cmd_size);
		return -1;
	}

	table = bl_iw7027->init_data;
	max_len = bl_iw7027->init_data_cnt;

	while ((i + cmd_size) <= max_len) {
		type = table[i];
		if (type == LCD_EXT_CMD_TYPE_END)
			break;
		if (lcd_debug_print_flag) {
			LDIMPR("%s: step %d: type=0x%02x, cmd_size=%d\n",
				__func__, step, type, cmd_size);
		}
		if (type == LCD_EXT_CMD_TYPE_NONE) {
			/* do nothing */
		} else if (type == LCD_EXT_CMD_TYPE_DELAY) {
			delay_ms = 0;
			for (j = 0; j < (cmd_size - 1); j++)
				delay_ms += table[i+1+j];
			if (delay_ms > 0)
				mdelay(delay_ms);
		} else if (type == LCD_EXT_CMD_TYPE_CMD) {
			ret = iw7027_wregs(bl_iw7027->spi,
				table[i + 1], &table[i + 2], 1);
			udelay(1);
		} else if (type == LCD_EXT_CMD_TYPE_CMD_DELAY) {
			ret = iw7027_wregs(bl_iw7027->spi,
				table[i + 1], &table[i + 2], 1);
			udelay(1);
			if (table[i+3] > 0)
				mdelay(table[i+3]);
		} else {
			LDIMERR("%s: type 0x%02x invalid\n", __func__, type);
		}
		i += cmd_size;
		step++;
	}

	return ret;
}

static int iw7027_power_on_init(void)
{
	unsigned char cmd_size;
	int ret = 0;

	cmd_size = bl_iw7027->cmd_size;
	if (cmd_size < 2) {
		LDIMERR("%s: invalid cmd_size %d\n", __func__, cmd_size);
		return -1;
	}
	if (cmd_size == LCD_EXT_CMD_SIZE_DYNAMIC)
		ret = ldim_power_cmd_dynamic_size();
	else
		ret = ldim_power_cmd_fixed_size();

	return ret;
}

static int iw7027_hw_init_on(void)
{
	int i;
	unsigned char  reg_duty_chk = 0 , reg_chk = 0;
	unsigned char reg, temp[2];
	struct aml_ldim_driver_s *ldim_drv = aml_ldim_get_driver();

	/* step 1: system power_on */
	LDIMPR("%s: iw7027 system power_on\n", __func__);
	ldim_set_gpio(ldim_drv->ldev_conf->en_gpio, ldim_drv->ldev_conf->en_gpio_on);
	ldim_set_duty_pwm(&(ldim_drv->ldev_conf->ldim_pwm_config));
	ldim_set_duty_pwm(&(ldim_drv->ldev_conf->analog_pwm_config));

	/* step 2: delay for internal logic stable */
	mdelay(10);

	/* step 3: SPI communication check */
	LDIMPR("%s: SPI Communication Check\n", __func__);
	for (i = 0; i <= 10; i++) {
		reg = 0x00;
		temp[0] = 0x06;
		iw7027_wregs(bl_iw7027->spi, reg, temp, 1);
		iw7027_rreg(bl_iw7027->spi, 0x00, &reg_chk, 1);
		if (reg_chk == 0x06)
			break;
		if (i == 10) {
			LDIMERR("%s: SPI communication check error\n",
				__func__);
		}
	}

	/* step 4: configure initial registers */
	LDIMPR("%s: Write initial control registers\n", __func__);
	iw7027_power_on_init();

	/* step 5: supply stable vsync */
	LDIMPR("%s: open Vsync\n", __func__);
	ldim_drv->pinmux_ctrl(1);

	/* step 6: delay for system clock and light bar PSU stable */
	mdelay(550);

	/* step 7: start calibration */
	LDIMPR("%s: start calibration\n", __func__);
	reg = 0x00;
	temp[0] = 0x07;
	iw7027_wregs(bl_iw7027->spi, reg, temp, 1);
	mdelay(200);

	/* step 8: calibration done or not */
	i = 0;
	while (i++ < 1000) {
		iw7027_rreg(bl_iw7027->spi, 0xb3, &reg_duty_chk, 1);
		/*VDAC statue reg :FB1=[0x5] FB2=[0x50]*/
		/*The current platform using FB1*/
		if ((reg_duty_chk & 0xf) == 0x05)
			break;
		mdelay(1);
	}
	LDIMPR("%s: calibration done: [%d] = %x\n", __func__, i, reg_duty_chk);

	return 0;
}

static int iw7027_hw_init_off(void)
{
	struct aml_ldim_driver_s *ldim_drv = aml_ldim_get_driver();

	ldim_drv->pinmux_ctrl(0);
	ldim_set_gpio(ldim_drv->ldev_conf->en_gpio, ldim_drv->ldev_conf->en_gpio_off);

	return 0;
}

static unsigned int dim_max, dim_min;
static unsigned int iw7027_get_value(unsigned int level)
{
	unsigned int val;

	val = dim_min + ((level * (dim_max - dim_min)) / LD_DATA_MAX);

	return val;
}

static int iw7027_smr(unsigned short *buf, unsigned char len)
{
	unsigned int i, temp;
	unsigned short num;
	struct aml_ldim_driver_s *ldim_drv = aml_ldim_get_driver();

	if (iw7027_on_flag == 0) {
		LDIMPR("%s: on_flag=%d\n", __func__, iw7027_on_flag);
		return 0;
	}
	num = ldim_drv->ldev_conf->bl_regnum;
	if (len != num) {
		LDIMERR("%s: data len %d invalid\n", __func__, len);
		return -1;
	}
	if (val_brightness == NULL) {
		LDIMERR("%s: val_brightness is null\n", __func__);
		return -1;
	}

	dim_max = ldim_drv->ldev_conf->dim_max;
	dim_min = ldim_drv->ldev_conf->dim_min;

	for (i = 0; i < num; i++) {
		temp = iw7027_get_value(buf[i]);
		val_brightness[2*i] = (temp >> 8) & 0xf;
		val_brightness[2*i+1] = temp & 0xff;
	}

	iw7027_wregs(bl_iw7027->spi, 0x40, val_brightness, 0x20);

	return 0;
}

static int iw7027_power_on(void)
{
	iw7027_hw_init_on();
	iw7027_on_flag = 1;

	LDIMPR("%s: ok\n", __func__);
	return 0;
}

static int iw7027_power_off(void)
{
	iw7027_on_flag = 0;
	iw7027_hw_init_off();

	LDIMPR("%s: ok\n", __func__);
	return 0;
}

static int iw7027_ldim_driver_update(struct aml_ldim_driver_s *ldim_drv)
{
	ldim_drv->device_power_on = iw7027_power_on;
	ldim_drv->device_power_off = iw7027_power_off;
	ldim_drv->device_bri_update = iw7027_smr;
	return 0;
}

int ldim_dev_iw7027_probe(struct aml_ldim_driver_s *ldim_drv)
{
	if (ldim_drv->spi_info->spi == NULL) {
		LDIMERR("%s: spi is null\n", __func__);
		return -1;
	}

	bl_iw7027 = (struct iw7027 *)malloc(sizeof(struct iw7027));
	if (bl_iw7027 == NULL) {
		LDIMERR("iw7027 malloc error\n");
		return -1;
	}
	memset(bl_iw7027, 0, sizeof(struct iw7027));

	iw7027_on_flag = 0;

	bl_iw7027->spi = ldim_drv->spi_info->spi;
	bl_iw7027->cs_hold_delay = ldim_drv->ldev_conf->cs_hold_delay;
	bl_iw7027->cs_clk_delay = ldim_drv->ldev_conf->cs_clk_delay;
	bl_iw7027->cmd_size = ldim_drv->ldev_conf->cmd_size;
	bl_iw7027->init_data = ldim_drv->ldev_conf->init_on;
	bl_iw7027->init_data_cnt = ldim_drv->ldev_conf->init_on_cnt;
	bl_iw7027->chip_cnt = ldim_drv->ldev_conf->device_count;
	LDIMPR("chip_cnt = %d\n", bl_iw7027->chip_cnt);

	val_brightness = (unsigned char *)malloc(
		ldim_drv->ldev_conf->bl_regnum * 2 * sizeof(unsigned char));
	if (val_brightness == NULL) {
		LDIMERR("malloc val_brightness failed\n");
		free(bl_iw7027);
		return -1;
	}

	iw7027_ldim_driver_update(ldim_drv);

	printf("%s: ok\n", __func__);

	return 0;
}

int ldim_dev_iw7027_remove(struct aml_ldim_driver_s *ldim_drv)
{
	if (val_brightness) {
		free(val_brightness);
		val_brightness = NULL;
	}

	if (bl_iw7027) {
		free(bl_iw7027);
		bl_iw7027 = NULL;
	}
	return 0;
}