// SPDX-License-Identifier: GPL-2.0-only

  /*

   * MFD core driver for Rockchip RK808/RK818

   *
   * Copyright (c) 2014, Fuzhou Rockchip Electronics Co., Ltd
   *
   * Author: Chris Zhong <zyw@rock-chips.com>
   * Author: Zhang Qing <zhangqing@rock-chips.com>
   *

   * Copyright (C) 2016 PHYTEC Messtechnik GmbH
   *
   * Author: Wadim Egorov <w.egorov@phytec.de>

   */
  
  #include <linux/i2c.h>
  #include <linux/interrupt.h>
  #include <linux/mfd/rk808.h>
  #include <linux/mfd/core.h>
  #include <linux/module.h>

  #include <linux/of_device.h>

  #include <linux/regmap.h>
  
  struct rk808_reg_data {
  	int addr;
  	int mask;
  	int value;
  };

  static bool rk808_is_volatile_reg(struct device *dev, unsigned int reg)
  {
  	/*
  	 * Notes:
  	 * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
  	 *   we don't use that feature.  It's better to cache.
  	 * - It's unlikely we care that RK808_DEVCTRL_REG is volatile since
  	 *   bits are cleared in case when we shutoff anyway, but better safe.
  	 */
  
  	switch (reg) {
  	case RK808_SECONDS_REG ... RK808_WEEKS_REG:
  	case RK808_RTC_STATUS_REG:
  	case RK808_VB_MON_REG:
  	case RK808_THERMAL_REG:
  	case RK808_DCDC_UV_STS_REG:
  	case RK808_LDO_UV_STS_REG:
  	case RK808_DCDC_PG_REG:
  	case RK808_LDO_PG_REG:
  	case RK808_DEVCTRL_REG:
  	case RK808_INT_STS_REG1:
  	case RK808_INT_STS_REG2:
  		return true;
  	}
  
  	return false;
  }

  static bool rk817_is_volatile_reg(struct device *dev, unsigned int reg)
  {
  	/*
  	 * Notes:
  	 * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
  	 *   we don't use that feature.  It's better to cache.
  	 */
  
  	switch (reg) {
  	case RK817_SECONDS_REG ... RK817_WEEKS_REG:
  	case RK817_RTC_STATUS_REG:
  	case RK817_INT_STS_REG0:
  	case RK817_INT_STS_REG1:
  	case RK817_INT_STS_REG2:
  	case RK817_SYS_STS:
  		return true;
  	}
  
  	return true;
  }

  static const struct regmap_config rk818_regmap_config = {
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = RK818_USB_CTRL_REG,
  	.cache_type = REGCACHE_RBTREE,
  	.volatile_reg = rk808_is_volatile_reg,
  };

  static const struct regmap_config rk805_regmap_config = {
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = RK805_OFF_SOURCE_REG,
  	.cache_type = REGCACHE_RBTREE,
  	.volatile_reg = rk808_is_volatile_reg,
  };

  static const struct regmap_config rk808_regmap_config = {
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = RK808_IO_POL_REG,

  	.cache_type = REGCACHE_RBTREE,
  	.volatile_reg = rk808_is_volatile_reg,

  };

  static const struct regmap_config rk817_regmap_config = {
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = RK817_GPIO_INT_CFG,
  	.cache_type = REGCACHE_NONE,
  	.volatile_reg = rk817_is_volatile_reg,
  };

  static struct resource rtc_resources[] = {

  	DEFINE_RES_IRQ(RK808_IRQ_RTC_ALARM),

  };

  static struct resource rk817_rtc_resources[] = {
  	DEFINE_RES_IRQ(RK817_IRQ_RTC_ALARM),
  };

  static struct resource rk805_key_resources[] = {

  	DEFINE_RES_IRQ(RK805_IRQ_PWRON_RISE),
  	DEFINE_RES_IRQ(RK805_IRQ_PWRON_FALL),

  };

  static struct resource rk817_pwrkey_resources[] = {
  	DEFINE_RES_IRQ(RK817_IRQ_PWRON_RISE),
  	DEFINE_RES_IRQ(RK817_IRQ_PWRON_FALL),
  };

  static const struct mfd_cell rk805s[] = {
  	{ .name = "rk808-clkout", },
  	{ .name = "rk808-regulator", },

  	{ .name = "rk805-pinctrl", },

  	{
  		.name = "rk808-rtc",
  		.num_resources = ARRAY_SIZE(rtc_resources),
  		.resources = &rtc_resources[0],
  	},

  	{	.name = "rk805-pwrkey",
  		.num_resources = ARRAY_SIZE(rk805_key_resources),
  		.resources = &rk805_key_resources[0],
  	},

  };

  static const struct mfd_cell rk808s[] = {
  	{ .name = "rk808-clkout", },
  	{ .name = "rk808-regulator", },
  	{
  		.name = "rk808-rtc",
  		.num_resources = ARRAY_SIZE(rtc_resources),

  		.resources = rtc_resources,

  	},
  };

  static const struct mfd_cell rk817s[] = {
  	{ .name = "rk808-clkout",},
  	{ .name = "rk808-regulator",},
  	{

  		.name = "rk805-pwrkey",

  		.num_resources = ARRAY_SIZE(rk817_pwrkey_resources),
  		.resources = &rk817_pwrkey_resources[0],
  	},
  	{
  		.name = "rk808-rtc",
  		.num_resources = ARRAY_SIZE(rk817_rtc_resources),
  		.resources = &rk817_rtc_resources[0],
  	},
  };

  static const struct mfd_cell rk818s[] = {
  	{ .name = "rk808-clkout", },
  	{ .name = "rk808-regulator", },
  	{
  		.name = "rk808-rtc",
  		.num_resources = ARRAY_SIZE(rtc_resources),
  		.resources = rtc_resources,
  	},
  };

  static const struct rk808_reg_data rk805_pre_init_reg[] = {
  	{RK805_BUCK1_CONFIG_REG, RK805_BUCK1_2_ILMAX_MASK,
  				 RK805_BUCK1_2_ILMAX_4000MA},
  	{RK805_BUCK2_CONFIG_REG, RK805_BUCK1_2_ILMAX_MASK,
  				 RK805_BUCK1_2_ILMAX_4000MA},
  	{RK805_BUCK3_CONFIG_REG, RK805_BUCK3_4_ILMAX_MASK,
  				 RK805_BUCK3_ILMAX_3000MA},
  	{RK805_BUCK4_CONFIG_REG, RK805_BUCK3_4_ILMAX_MASK,
  				 RK805_BUCK4_ILMAX_3500MA},
  	{RK805_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_400MA},

  	{RK805_THERMAL_REG, TEMP_HOTDIE_MSK, TEMP115C},
  };

  static const struct rk808_reg_data rk808_pre_init_reg[] = {

  	{ RK808_BUCK3_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_150MA },
  	{ RK808_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_200MA },
  	{ RK808_BOOST_CONFIG_REG, BOOST_ILMIN_MASK, BOOST_ILMIN_100MA },
  	{ RK808_BUCK1_CONFIG_REG, BUCK1_RATE_MASK,  BUCK_ILMIN_200MA },
  	{ RK808_BUCK2_CONFIG_REG, BUCK2_RATE_MASK,  BUCK_ILMIN_200MA },

  	{ RK808_DCDC_UV_ACT_REG,  BUCK_UV_ACT_MASK, BUCK_UV_ACT_DISABLE},

  	{ RK808_VB_MON_REG,       MASK_ALL,         VB_LO_ACT |
  						    VB_LO_SEL_3500MV },
  };

  static const struct rk808_reg_data rk817_pre_init_reg[] = {
  	{RK817_RTC_CTRL_REG, RTC_STOP, RTC_STOP},

  	{RK817_GPIO_INT_CFG, RK817_INT_POL_MSK, RK817_INT_POL_L},

  	{RK817_SYS_CFG(1), RK817_HOTDIE_TEMP_MSK | RK817_TSD_TEMP_MSK,
  					   RK817_HOTDIE_105 | RK817_TSD_140},
  };

  static const struct rk808_reg_data rk818_pre_init_reg[] = {
  	/* improve efficiency */
  	{ RK818_BUCK2_CONFIG_REG, BUCK2_RATE_MASK,  BUCK_ILMIN_250MA },
  	{ RK818_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_250MA },
  	{ RK818_BOOST_CONFIG_REG, BOOST_ILMIN_MASK, BOOST_ILMIN_100MA },
  	{ RK818_USB_CTRL_REG,	  RK818_USB_ILIM_SEL_MASK,
  						    RK818_USB_ILMIN_2000MA },
  	/* close charger when usb lower then 3.4V */
  	{ RK818_USB_CTRL_REG,	  RK818_USB_CHG_SD_VSEL_MASK,
  						    (0x7 << 4) },
  	/* no action when vref */
  	{ RK818_H5V_EN_REG,	  BIT(1),	    RK818_REF_RDY_CTRL },
  	/* enable HDMI 5V */
  	{ RK818_H5V_EN_REG,	  BIT(0),	    RK818_H5V_EN },
  	{ RK808_VB_MON_REG,	  MASK_ALL,	    VB_LO_ACT |
  						    VB_LO_SEL_3500MV },
  };

  static const struct regmap_irq rk805_irqs[] = {
  	[RK805_IRQ_PWRON_RISE] = {
  		.mask = RK805_IRQ_PWRON_RISE_MSK,
  		.reg_offset = 0,
  	},
  	[RK805_IRQ_VB_LOW] = {
  		.mask = RK805_IRQ_VB_LOW_MSK,
  		.reg_offset = 0,
  	},
  	[RK805_IRQ_PWRON] = {
  		.mask = RK805_IRQ_PWRON_MSK,
  		.reg_offset = 0,
  	},
  	[RK805_IRQ_PWRON_LP] = {
  		.mask = RK805_IRQ_PWRON_LP_MSK,
  		.reg_offset = 0,
  	},
  	[RK805_IRQ_HOTDIE] = {
  		.mask = RK805_IRQ_HOTDIE_MSK,
  		.reg_offset = 0,
  	},
  	[RK805_IRQ_RTC_ALARM] = {
  		.mask = RK805_IRQ_RTC_ALARM_MSK,
  		.reg_offset = 0,
  	},
  	[RK805_IRQ_RTC_PERIOD] = {
  		.mask = RK805_IRQ_RTC_PERIOD_MSK,
  		.reg_offset = 0,
  	},
  	[RK805_IRQ_PWRON_FALL] = {
  		.mask = RK805_IRQ_PWRON_FALL_MSK,
  		.reg_offset = 0,
  	},
  };

  static const struct regmap_irq rk808_irqs[] = {
  	/* INT_STS */
  	[RK808_IRQ_VOUT_LO] = {
  		.mask = RK808_IRQ_VOUT_LO_MSK,
  		.reg_offset = 0,
  	},
  	[RK808_IRQ_VB_LO] = {
  		.mask = RK808_IRQ_VB_LO_MSK,
  		.reg_offset = 0,
  	},
  	[RK808_IRQ_PWRON] = {
  		.mask = RK808_IRQ_PWRON_MSK,
  		.reg_offset = 0,
  	},
  	[RK808_IRQ_PWRON_LP] = {
  		.mask = RK808_IRQ_PWRON_LP_MSK,
  		.reg_offset = 0,
  	},
  	[RK808_IRQ_HOTDIE] = {
  		.mask = RK808_IRQ_HOTDIE_MSK,
  		.reg_offset = 0,
  	},
  	[RK808_IRQ_RTC_ALARM] = {
  		.mask = RK808_IRQ_RTC_ALARM_MSK,
  		.reg_offset = 0,
  	},
  	[RK808_IRQ_RTC_PERIOD] = {
  		.mask = RK808_IRQ_RTC_PERIOD_MSK,
  		.reg_offset = 0,
  	},
  
  	/* INT_STS2 */
  	[RK808_IRQ_PLUG_IN_INT] = {
  		.mask = RK808_IRQ_PLUG_IN_INT_MSK,
  		.reg_offset = 1,
  	},
  	[RK808_IRQ_PLUG_OUT_INT] = {
  		.mask = RK808_IRQ_PLUG_OUT_INT_MSK,
  		.reg_offset = 1,
  	},
  };

  static const struct regmap_irq rk818_irqs[] = {
  	/* INT_STS */
  	[RK818_IRQ_VOUT_LO] = {
  		.mask = RK818_IRQ_VOUT_LO_MSK,
  		.reg_offset = 0,
  	},
  	[RK818_IRQ_VB_LO] = {
  		.mask = RK818_IRQ_VB_LO_MSK,
  		.reg_offset = 0,
  	},
  	[RK818_IRQ_PWRON] = {
  		.mask = RK818_IRQ_PWRON_MSK,
  		.reg_offset = 0,
  	},
  	[RK818_IRQ_PWRON_LP] = {
  		.mask = RK818_IRQ_PWRON_LP_MSK,
  		.reg_offset = 0,
  	},
  	[RK818_IRQ_HOTDIE] = {
  		.mask = RK818_IRQ_HOTDIE_MSK,
  		.reg_offset = 0,
  	},
  	[RK818_IRQ_RTC_ALARM] = {
  		.mask = RK818_IRQ_RTC_ALARM_MSK,
  		.reg_offset = 0,
  	},
  	[RK818_IRQ_RTC_PERIOD] = {
  		.mask = RK818_IRQ_RTC_PERIOD_MSK,
  		.reg_offset = 0,
  	},
  	[RK818_IRQ_USB_OV] = {
  		.mask = RK818_IRQ_USB_OV_MSK,
  		.reg_offset = 0,
  	},
  
  	/* INT_STS2 */
  	[RK818_IRQ_PLUG_IN] = {
  		.mask = RK818_IRQ_PLUG_IN_MSK,
  		.reg_offset = 1,
  	},
  	[RK818_IRQ_PLUG_OUT] = {
  		.mask = RK818_IRQ_PLUG_OUT_MSK,
  		.reg_offset = 1,
  	},
  	[RK818_IRQ_CHG_OK] = {
  		.mask = RK818_IRQ_CHG_OK_MSK,
  		.reg_offset = 1,
  	},
  	[RK818_IRQ_CHG_TE] = {
  		.mask = RK818_IRQ_CHG_TE_MSK,
  		.reg_offset = 1,
  	},
  	[RK818_IRQ_CHG_TS1] = {
  		.mask = RK818_IRQ_CHG_TS1_MSK,
  		.reg_offset = 1,
  	},
  	[RK818_IRQ_TS2] = {
  		.mask = RK818_IRQ_TS2_MSK,
  		.reg_offset = 1,
  	},
  	[RK818_IRQ_CHG_CVTLIM] = {
  		.mask = RK818_IRQ_CHG_CVTLIM_MSK,
  		.reg_offset = 1,
  	},
  	[RK818_IRQ_DISCHG_ILIM] = {
  		.mask = RK818_IRQ_DISCHG_ILIM_MSK,
  		.reg_offset = 1,
  	},
  };

  static const struct regmap_irq rk817_irqs[RK817_IRQ_END] = {
  	REGMAP_IRQ_REG_LINE(0, 8),
  	REGMAP_IRQ_REG_LINE(1, 8),
  	REGMAP_IRQ_REG_LINE(2, 8),
  	REGMAP_IRQ_REG_LINE(3, 8),
  	REGMAP_IRQ_REG_LINE(4, 8),
  	REGMAP_IRQ_REG_LINE(5, 8),
  	REGMAP_IRQ_REG_LINE(6, 8),
  	REGMAP_IRQ_REG_LINE(7, 8),
  	REGMAP_IRQ_REG_LINE(8, 8),
  	REGMAP_IRQ_REG_LINE(9, 8),
  	REGMAP_IRQ_REG_LINE(10, 8),
  	REGMAP_IRQ_REG_LINE(11, 8),
  	REGMAP_IRQ_REG_LINE(12, 8),
  	REGMAP_IRQ_REG_LINE(13, 8),
  	REGMAP_IRQ_REG_LINE(14, 8),
  	REGMAP_IRQ_REG_LINE(15, 8),
  	REGMAP_IRQ_REG_LINE(16, 8),
  	REGMAP_IRQ_REG_LINE(17, 8),
  	REGMAP_IRQ_REG_LINE(18, 8),
  	REGMAP_IRQ_REG_LINE(19, 8),
  	REGMAP_IRQ_REG_LINE(20, 8),
  	REGMAP_IRQ_REG_LINE(21, 8),
  	REGMAP_IRQ_REG_LINE(22, 8),
  	REGMAP_IRQ_REG_LINE(23, 8)
  };

  static struct regmap_irq_chip rk805_irq_chip = {
  	.name = "rk805",
  	.irqs = rk805_irqs,
  	.num_irqs = ARRAY_SIZE(rk805_irqs),
  	.num_regs = 1,
  	.status_base = RK805_INT_STS_REG,
  	.mask_base = RK805_INT_STS_MSK_REG,
  	.ack_base = RK805_INT_STS_REG,
  	.init_ack_masked = true,
  };

  static const struct regmap_irq_chip rk808_irq_chip = {

  	.name = "rk808",
  	.irqs = rk808_irqs,
  	.num_irqs = ARRAY_SIZE(rk808_irqs),
  	.num_regs = 2,
  	.irq_reg_stride = 2,
  	.status_base = RK808_INT_STS_REG1,
  	.mask_base = RK808_INT_STS_MSK_REG1,
  	.ack_base = RK808_INT_STS_REG1,
  	.init_ack_masked = true,
  };

  static struct regmap_irq_chip rk817_irq_chip = {
  	.name = "rk817",
  	.irqs = rk817_irqs,
  	.num_irqs = ARRAY_SIZE(rk817_irqs),
  	.num_regs = 3,
  	.irq_reg_stride = 2,
  	.status_base = RK817_INT_STS_REG0,
  	.mask_base = RK817_INT_STS_MSK_REG0,
  	.ack_base = RK817_INT_STS_REG0,
  	.init_ack_masked = true,
  };

  static const struct regmap_irq_chip rk818_irq_chip = {

  	.name = "rk818",
  	.irqs = rk818_irqs,
  	.num_irqs = ARRAY_SIZE(rk818_irqs),
  	.num_regs = 2,
  	.irq_reg_stride = 2,
  	.status_base = RK818_INT_STS_REG1,
  	.mask_base = RK818_INT_STS_MSK_REG1,
  	.ack_base = RK818_INT_STS_REG1,
  	.init_ack_masked = true,
  };

  static struct i2c_client *rk808_i2c_client;

  static void rk808_pm_power_off(void)

  {
  	int ret;

  	unsigned int reg, bit;

  	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

  	switch (rk808->variant) {
  	case RK805_ID:
  		reg = RK805_DEV_CTRL_REG;
  		bit = DEV_OFF;
  		break;
  	case RK808_ID:
  		reg = RK808_DEVCTRL_REG,
  		bit = DEV_OFF_RST;
  		break;
  	case RK818_ID:
  		reg = RK818_DEVCTRL_REG;
  		bit = DEV_OFF;
  		break;
  	default:

  		return;

  	}
  	ret = regmap_update_bits(rk808->regmap, reg, bit, bit);

  	if (ret)

  		dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!
  ");

  }

  static void rk8xx_shutdown(struct i2c_client *client)

  {

  	struct rk808 *rk808 = i2c_get_clientdata(client);

  	int ret;

  	switch (rk808->variant) {

  	case RK805_ID:
  		ret = regmap_update_bits(rk808->regmap,
  					 RK805_GPIO_IO_POL_REG,
  					 SLP_SD_MSK,
  					 SHUTDOWN_FUN);
  		break;

  	case RK809_ID:
  	case RK817_ID:

  		ret = regmap_update_bits(rk808->regmap,
  					 RK817_SYS_CFG(3),
  					 RK817_SLPPIN_FUNC_MSK,
  					 SLPPIN_DN_FUN);

  		break;
  	default:
  		return;

  	}

  	if (ret)
  		dev_warn(&client->dev,
  			 "Cannot switch to power down function
  ");

  }

  static const struct of_device_id rk808_of_match[] = {

  	{ .compatible = "rockchip,rk805" },

  	{ .compatible = "rockchip,rk808" },

  	{ .compatible = "rockchip,rk809" },
  	{ .compatible = "rockchip,rk817" },

  	{ .compatible = "rockchip,rk818" },
  	{ },
  };
  MODULE_DEVICE_TABLE(of, rk808_of_match);

  static int rk808_probe(struct i2c_client *client,
  		       const struct i2c_device_id *id)
  {
  	struct device_node *np = client->dev.of_node;
  	struct rk808 *rk808;

  	const struct rk808_reg_data *pre_init_reg;
  	const struct mfd_cell *cells;
  	int nr_pre_init_regs;
  	int nr_cells;

  	int msb, lsb;

  	unsigned char pmic_id_msb, pmic_id_lsb;

  	int ret;
  	int i;

  	rk808 = devm_kzalloc(&client->dev, sizeof(*rk808), GFP_KERNEL);
  	if (!rk808)
  		return -ENOMEM;

  	if (of_device_is_compatible(np, "rockchip,rk817") ||
  	    of_device_is_compatible(np, "rockchip,rk809")) {
  		pmic_id_msb = RK817_ID_MSB;
  		pmic_id_lsb = RK817_ID_LSB;
  	} else {
  		pmic_id_msb = RK808_ID_MSB;
  		pmic_id_lsb = RK808_ID_LSB;
  	}

  	/* Read chip variant */

  	msb = i2c_smbus_read_byte_data(client, pmic_id_msb);

  	if (msb < 0) {
  		dev_err(&client->dev, "failed to read the chip id at 0x%x
  ",

  			RK808_ID_MSB);

  		return msb;

  	}

  	lsb = i2c_smbus_read_byte_data(client, pmic_id_lsb);

  	if (lsb < 0) {
  		dev_err(&client->dev, "failed to read the chip id at 0x%x
  ",
  			RK808_ID_LSB);
  		return lsb;
  	}
  
  	rk808->variant = ((msb << 8) | lsb) & RK8XX_ID_MSK;
  	dev_info(&client->dev, "chip id: 0x%x
  ", (unsigned int)rk808->variant);

  
  	switch (rk808->variant) {

  	case RK805_ID:
  		rk808->regmap_cfg = &rk805_regmap_config;
  		rk808->regmap_irq_chip = &rk805_irq_chip;
  		pre_init_reg = rk805_pre_init_reg;
  		nr_pre_init_regs = ARRAY_SIZE(rk805_pre_init_reg);
  		cells = rk805s;
  		nr_cells = ARRAY_SIZE(rk805s);

  		break;

  	case RK808_ID:
  		rk808->regmap_cfg = &rk808_regmap_config;
  		rk808->regmap_irq_chip = &rk808_irq_chip;
  		pre_init_reg = rk808_pre_init_reg;
  		nr_pre_init_regs = ARRAY_SIZE(rk808_pre_init_reg);
  		cells = rk808s;
  		nr_cells = ARRAY_SIZE(rk808s);
  		break;
  	case RK818_ID:
  		rk808->regmap_cfg = &rk818_regmap_config;
  		rk808->regmap_irq_chip = &rk818_irq_chip;
  		pre_init_reg = rk818_pre_init_reg;
  		nr_pre_init_regs = ARRAY_SIZE(rk818_pre_init_reg);
  		cells = rk818s;
  		nr_cells = ARRAY_SIZE(rk818s);
  		break;

  	case RK809_ID:
  	case RK817_ID:
  		rk808->regmap_cfg = &rk817_regmap_config;
  		rk808->regmap_irq_chip = &rk817_irq_chip;
  		pre_init_reg = rk817_pre_init_reg;
  		nr_pre_init_regs = ARRAY_SIZE(rk817_pre_init_reg);
  		cells = rk817s;
  		nr_cells = ARRAY_SIZE(rk817s);

  		break;
  	default:
  		dev_err(&client->dev, "Unsupported RK8XX ID %lu
  ",
  			rk808->variant);
  		return -EINVAL;
  	}
  
  	rk808->i2c = client;
  	i2c_set_clientdata(client, rk808);
  
  	rk808->regmap = devm_regmap_init_i2c(client, rk808->regmap_cfg);

  	if (IS_ERR(rk808->regmap)) {
  		dev_err(&client->dev, "regmap initialization failed
  ");
  		return PTR_ERR(rk808->regmap);
  	}

  	if (!client->irq) {
  		dev_err(&client->dev, "No interrupt support, no core IRQ
  ");
  		return -EINVAL;

  	}
  
  	ret = regmap_add_irq_chip(rk808->regmap, client->irq,
  				  IRQF_ONESHOT, -1,

  				  rk808->regmap_irq_chip, &rk808->irq_data);

  	if (ret) {
  		dev_err(&client->dev, "Failed to add irq_chip %d
  ", ret);
  		return ret;
  	}

  	for (i = 0; i < nr_pre_init_regs; i++) {
  		ret = regmap_update_bits(rk808->regmap,
  					pre_init_reg[i].addr,
  					pre_init_reg[i].mask,
  					pre_init_reg[i].value);
  		if (ret) {
  			dev_err(&client->dev,
  				"0x%x write err
  ",
  				pre_init_reg[i].addr);
  			return ret;
  		}
  	}

  	ret = devm_mfd_add_devices(&client->dev, PLATFORM_DEVID_NONE,
  			      cells, nr_cells, NULL, 0,
  			      regmap_irq_get_domain(rk808->irq_data));

  	if (ret) {
  		dev_err(&client->dev, "failed to add MFD devices %d
  ", ret);
  		goto err_irq;
  	}

  	if (of_property_read_bool(np, "rockchip,system-power-controller")) {

  		rk808_i2c_client = client;

  		pm_power_off = rk808_pm_power_off;

  	}
  
  	return 0;
  
  err_irq:
  	regmap_del_irq_chip(client->irq, rk808->irq_data);
  	return ret;
  }
  
  static int rk808_remove(struct i2c_client *client)
  {
  	struct rk808 *rk808 = i2c_get_clientdata(client);
  
  	regmap_del_irq_chip(client->irq, rk808->irq_data);

  
  	/**
  	 * pm_power_off may points to a function from another module.
  	 * Check if the pointer is set by us and only then overwrite it.
  	 */

  	if (pm_power_off == rk808_pm_power_off)

  		pm_power_off = NULL;

  
  	return 0;
  }

  static int __maybe_unused rk8xx_suspend(struct device *dev)

  {

  	struct rk808 *rk808 = i2c_get_clientdata(to_i2c_client(dev));

  	int ret = 0;
  
  	switch (rk808->variant) {

  	case RK805_ID:
  		ret = regmap_update_bits(rk808->regmap,
  					 RK805_GPIO_IO_POL_REG,
  					 SLP_SD_MSK,
  					 SLEEP_FUN);
  		break;

  	case RK809_ID:
  	case RK817_ID:
  		ret = regmap_update_bits(rk808->regmap,
  					 RK817_SYS_CFG(3),
  					 RK817_SLPPIN_FUNC_MSK,
  					 SLPPIN_SLP_FUN);
  		break;
  	default:
  		break;
  	}
  
  	return ret;
  }

  static int __maybe_unused rk8xx_resume(struct device *dev)

  {

  	struct rk808 *rk808 = i2c_get_clientdata(to_i2c_client(dev));

  	int ret = 0;
  
  	switch (rk808->variant) {
  	case RK809_ID:
  	case RK817_ID:
  		ret = regmap_update_bits(rk808->regmap,
  					 RK817_SYS_CFG(3),
  					 RK817_SLPPIN_FUNC_MSK,
  					 SLPPIN_NULL_FUN);
  		break;
  	default:
  		break;
  	}
  
  	return ret;
  }

  static SIMPLE_DEV_PM_OPS(rk8xx_pm_ops, rk8xx_suspend, rk8xx_resume);

  static struct i2c_driver rk808_i2c_driver = {
  	.driver = {
  		.name = "rk808",
  		.of_match_table = rk808_of_match,

  		.pm = &rk8xx_pm_ops,

  	},
  	.probe    = rk808_probe,
  	.remove   = rk808_remove,

  	.shutdown = rk8xx_shutdown,

  };
  
  module_i2c_driver(rk808_i2c_driver);
  
  MODULE_LICENSE("GPL");
  MODULE_AUTHOR("Chris Zhong <zyw@rock-chips.com>");
  MODULE_AUTHOR("Zhang Qing <zhangqing@rock-chips.com>");

  MODULE_AUTHOR("Wadim Egorov <w.egorov@phytec.de>");
  MODULE_DESCRIPTION("RK808/RK818 PMIC driver");