drivers/mfd/rk808.c - kernel/msm-5.4 - Gitiles

 // 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>
 #include <linux/syscore_ops.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[] = {
 	{
 		.start  = RK808_IRQ_RTC_ALARM,
 		.end    = RK808_IRQ_RTC_ALARM,
 		.flags  = IORESOURCE_IRQ,
 	}
 };

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

 static struct resource rk805_key_resources[] = {
 	{
 		.start  = RK805_IRQ_PWRON_FALL,
 		.end    = RK805_IRQ_PWRON_FALL,
 		.flags  = IORESOURCE_IRQ,
 	},
 	{
 		.start  = RK805_IRQ_PWRON_RISE,
 		.end    = RK805_IRQ_PWRON_RISE,
 		.flags  = IORESOURCE_IRQ,
 	}
 };

 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 = "rk8xx-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_GPIO_IO_POL_REG, SLP_SD_MSK, SLEEP_FUN},
 	{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_H},
 	{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 rk805_device_shutdown(void)
 {
 	int ret;
 	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

 	if (!rk808)
 		return;

 	ret = regmap_update_bits(rk808->regmap,
 				 RK805_DEV_CTRL_REG,
 				 DEV_OFF, DEV_OFF);
 	if (ret)
 		dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
 }

 static void rk805_device_shutdown_prepare(void)
 {
 	int ret;
 	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

 	if (!rk808)
 		return;

 	ret = regmap_update_bits(rk808->regmap,
 				 RK805_GPIO_IO_POL_REG,
 				 SLP_SD_MSK, SHUTDOWN_FUN);
 	if (ret)
 		dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
 }

 static void rk808_device_shutdown(void)
 {
 	int ret;
 	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

 	if (!rk808)
 		return;

 	ret = regmap_update_bits(rk808->regmap,
 				 RK808_DEVCTRL_REG,
 				 DEV_OFF_RST, DEV_OFF_RST);
 	if (ret)
 		dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
 }

 static void rk818_device_shutdown(void)
 {
 	int ret;
 	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

 	if (!rk808)
 		return;

 	ret = regmap_update_bits(rk808->regmap,
 				 RK818_DEVCTRL_REG,
 				 DEV_OFF, DEV_OFF);
 	if (ret)
 		dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
 }

 static void rk8xx_syscore_shutdown(void)
 {
 	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 	int ret;

 	if (system_state == SYSTEM_POWER_OFF &&
 	    (rk808->variant == RK809_ID || rk808->variant == RK817_ID)) {
 		ret = regmap_update_bits(rk808->regmap,
 					 RK817_SYS_CFG(3),
 					 RK817_SLPPIN_FUNC_MSK,
 					 SLPPIN_DN_FUN);
 		if (ret) {
 			dev_warn(&rk808_i2c_client->dev,
 				 "Cannot switch to power down function\n");
 		}
 	}
 }

 static struct syscore_ops rk808_syscore_ops = {
 	.shutdown = rk8xx_syscore_shutdown,
 };

 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 pm_off = 0, 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\n",
 			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\n",
 			RK808_ID_LSB);
 		return lsb;
 	}

 	rk808->variant = ((msb << 8) | lsb) & RK8XX_ID_MSK;
 	dev_info(&client->dev, "chip id: 0x%x\n", (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);
 		rk808->pm_pwroff_fn = rk805_device_shutdown;
 		rk808->pm_pwroff_prep_fn = rk805_device_shutdown_prepare;
 		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);
 		rk808->pm_pwroff_fn = rk808_device_shutdown;
 		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);
 		rk808->pm_pwroff_fn = rk818_device_shutdown;
 		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);
 		register_syscore_ops(&rk808_syscore_ops);
 		break;
 	default:
 		dev_err(&client->dev, "Unsupported RK8XX ID %lu\n",
 			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\n");
 		return PTR_ERR(rk808->regmap);
 	}

 	if (!client->irq) {
 		dev_err(&client->dev, "No interrupt support, no core IRQ\n");
 		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\n", 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\n",
 				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\n", ret);
 		goto err_irq;
 	}

 	pm_off = of_property_read_bool(np,
 				"rockchip,system-power-controller");
 	if (pm_off && !pm_power_off) {
 		rk808_i2c_client = client;
 		pm_power_off = rk808->pm_pwroff_fn;
 	}

 	if (pm_off && !pm_power_off_prepare) {
 		if (!rk808_i2c_client)
 			rk808_i2c_client = client;
 		pm_power_off_prepare = rk808->pm_pwroff_prep_fn;
 	}

 	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 (rk808->pm_pwroff_fn && pm_power_off == rk808->pm_pwroff_fn)
 		pm_power_off = NULL;

 	/**
 	 * As above, check if the pointer is set by us before overwrite.
 	 */
 	if (rk808->pm_pwroff_prep_fn &&
 	    pm_power_off_prepare == rk808->pm_pwroff_prep_fn)
 		pm_power_off_prepare = NULL;

 	return 0;
 }

 static int rk8xx_suspend(struct device *dev)
 {
 	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 	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_SLP_FUN);
 		break;
 	default:
 		break;
 	}

 	return ret;
 }

 static int rk8xx_resume(struct device *dev)
 {
 	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 	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;
 }
 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,
 };

 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");
	// 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>
	#include <linux/syscore_ops.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[] = {
	{
	.start = RK808_IRQ_RTC_ALARM,
	.end = RK808_IRQ_RTC_ALARM,
	.flags = IORESOURCE_IRQ,
	}
	};

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

	static struct resource rk805_key_resources[] = {
	{
	.start = RK805_IRQ_PWRON_FALL,
	.end = RK805_IRQ_PWRON_FALL,
	.flags = IORESOURCE_IRQ,
	},
	{
	.start = RK805_IRQ_PWRON_RISE,
	.end = RK805_IRQ_PWRON_RISE,
	.flags = IORESOURCE_IRQ,
	}
	};

	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 = "rk8xx-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_GPIO_IO_POL_REG, SLP_SD_MSK, SLEEP_FUN},
	{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_H},
	{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 rk805_device_shutdown(void)
	{
	int ret;
	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

	if (!rk808)
	return;

	ret = regmap_update_bits(rk808->regmap,
	RK805_DEV_CTRL_REG,
	DEV_OFF, DEV_OFF);
	if (ret)
	dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
	}

	static void rk805_device_shutdown_prepare(void)
	{
	int ret;
	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

	if (!rk808)
	return;

	ret = regmap_update_bits(rk808->regmap,
	RK805_GPIO_IO_POL_REG,
	SLP_SD_MSK, SHUTDOWN_FUN);
	if (ret)
	dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
	}

	static void rk808_device_shutdown(void)
	{
	int ret;
	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

	if (!rk808)
	return;

	ret = regmap_update_bits(rk808->regmap,
	RK808_DEVCTRL_REG,
	DEV_OFF_RST, DEV_OFF_RST);
	if (ret)
	dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
	}

	static void rk818_device_shutdown(void)
	{
	int ret;
	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);

	if (!rk808)
	return;

	ret = regmap_update_bits(rk808->regmap,
	RK818_DEVCTRL_REG,
	DEV_OFF, DEV_OFF);
	if (ret)
	dev_err(&rk808_i2c_client->dev, "Failed to shutdown device!\n");
	}

	static void rk8xx_syscore_shutdown(void)
	{
	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
	int ret;

	if (system_state == SYSTEM_POWER_OFF &&
	(rk808->variant == RK809_ID \|\| rk808->variant == RK817_ID)) {
	ret = regmap_update_bits(rk808->regmap,
	RK817_SYS_CFG(3),
	RK817_SLPPIN_FUNC_MSK,
	SLPPIN_DN_FUN);
	if (ret) {
	dev_warn(&rk808_i2c_client->dev,
	"Cannot switch to power down function\n");
	}
	}
	}

	static struct syscore_ops rk808_syscore_ops = {
	.shutdown = rk8xx_syscore_shutdown,
	};

	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 pm_off = 0, 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\n",
	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\n",
	RK808_ID_LSB);
	return lsb;
	}

	rk808->variant = ((msb << 8) \| lsb) & RK8XX_ID_MSK;
	dev_info(&client->dev, "chip id: 0x%x\n", (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);
	rk808->pm_pwroff_fn = rk805_device_shutdown;
	rk808->pm_pwroff_prep_fn = rk805_device_shutdown_prepare;
	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);
	rk808->pm_pwroff_fn = rk808_device_shutdown;
	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);
	rk808->pm_pwroff_fn = rk818_device_shutdown;
	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);
	register_syscore_ops(&rk808_syscore_ops);
	break;
	default:
	dev_err(&client->dev, "Unsupported RK8XX ID %lu\n",
	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\n");
	return PTR_ERR(rk808->regmap);
	}

	if (!client->irq) {
	dev_err(&client->dev, "No interrupt support, no core IRQ\n");
	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\n", 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\n",
	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\n", ret);
	goto err_irq;
	}

	pm_off = of_property_read_bool(np,
	"rockchip,system-power-controller");
	if (pm_off && !pm_power_off) {
	rk808_i2c_client = client;
	pm_power_off = rk808->pm_pwroff_fn;
	}

	if (pm_off && !pm_power_off_prepare) {
	if (!rk808_i2c_client)
	rk808_i2c_client = client;
	pm_power_off_prepare = rk808->pm_pwroff_prep_fn;
	}

	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 (rk808->pm_pwroff_fn && pm_power_off == rk808->pm_pwroff_fn)
	pm_power_off = NULL;

	/**
	* As above, check if the pointer is set by us before overwrite.
	*/
	if (rk808->pm_pwroff_prep_fn &&
	pm_power_off_prepare == rk808->pm_pwroff_prep_fn)
	pm_power_off_prepare = NULL;

	return 0;
	}

	static int rk8xx_suspend(struct device *dev)
	{
	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
	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_SLP_FUN);
	break;
	default:
	break;
	}

	return ret;
	}

	static int rk8xx_resume(struct device *dev)
	{
	struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
	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;
	}
	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,
	};

	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");