ARM64: DTS: Fix Firefly board audio driver
[firefly-linux-kernel-4.4.55.git] / drivers / mfd / rk808.c
index 4b1e4399754bf2950c5662d763e3763653137b9d..2eadd2c10e6994f8366af01221675f355b1361de 100644 (file)
 
 #include <linux/i2c.h>
 #include <linux/interrupt.h>
+#include <linux/kernel.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/of_gpio.h>
+#include <linux/gpio.h>
+#include <linux/delay.h>
 
 struct rk808_reg_data {
        int addr;
@@ -57,6 +62,102 @@ static bool rk808_is_volatile_reg(struct device *dev, unsigned int reg)
        return false;
 }
 
+static int rk808_shutdown(struct regmap *regmap)
+{
+       int ret;
+
+       ret = regmap_update_bits(regmap,
+                                RK808_DEVCTRL_REG,
+                                DEV_OFF, DEV_OFF);
+       if (ret)
+               printk("DEV_OFF error!\n");
+
+       mdelay(2);
+       ret = regmap_update_bits(regmap,
+                                RK808_DCDC_EN_REG,
+                                BUCK1_EN_MASK | BUCK2_EN_MASK | BUCK3_EN_MASK | BUCK4_EN_MASK, 0xf);
+       if (ret)
+               printk("RK808_DCDC_EN_REG error!\n");
+       ret = regmap_update_bits(regmap,
+                                RK808_LDO_EN_REG,
+                                0xff, 0xff);
+       if (ret)
+               printk("RK808_LDO_EN_REG error!\n");
+
+       return ret;
+}
+
+static int rk816_shutdown(struct regmap *regmap)
+{
+       int ret;
+
+       ret = regmap_update_bits(regmap,
+                                RK816_DEV_CTRL_REG,
+                                DEV_OFF, DEV_OFF);
+       return ret;
+}
+
+static int rk818_shutdown(struct regmap *regmap)
+{
+       int ret;
+
+       ret = regmap_update_bits(regmap,
+                                RK818_DEVCTRL_REG,
+                                DEV_OFF, DEV_OFF);
+       return ret;
+}
+
+static int rk805_shutdown_prepare(struct regmap *regmap)
+{
+       int ret;
+
+       /* close rtc int when power off */
+       regmap_update_bits(regmap,
+                          RK808_INT_STS_MSK_REG1,
+                          (0x3 << 5), (0x3 << 5));
+       regmap_update_bits(regmap,
+                          RK808_RTC_INT_REG,
+                          (0x3 << 2), (0x0 << 2));
+
+       /* pmic sleep shutdown function */
+       ret = regmap_update_bits(regmap,
+                                RK805_GPIO_IO_POL_REG,
+                                SLP_SD_MSK, SHUTDOWN_FUN);
+       return ret;
+}
+
+static bool rk818_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_EN_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:
+       case RK808_INT_STS_MSK_REG1:
+       case RK808_INT_STS_MSK_REG2:
+       case RK818_SUP_STS_REG ... RK818_SAVE_DATA19:
+               return true;
+       }
+
+       return false;
+}
+
 static const struct regmap_config rk808_regmap_config = {
        .reg_bits = 8,
        .val_bits = 8,
@@ -65,6 +166,30 @@ static const struct regmap_config rk808_regmap_config = {
        .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 rk816_regmap_config = {
+       .reg_bits = 8,
+       .val_bits = 8,
+       .max_register = RK816_DATA18_REG,
+       .cache_type = REGCACHE_RBTREE,
+       .volatile_reg = rk818_is_volatile_reg,
+};
+
+static const struct regmap_config rk818_regmap_config = {
+       .reg_bits = 8,
+       .val_bits = 8,
+       .max_register = RK818_SAVE_DATA19,
+       .cache_type = REGCACHE_RBTREE,
+       .volatile_reg = rk818_is_volatile_reg,
+};
+
 static struct resource rtc_resources[] = {
        {
                .start  = RK808_IRQ_RTC_ALARM,
@@ -73,6 +198,32 @@ static struct resource rtc_resources[] = {
        }
 };
 
+static struct resource pwrkey_resources[] = {
+       {
+               .start  = RK805_IRQ_PWRON_RISE,
+               .end    = RK805_IRQ_PWRON_RISE,
+               .flags  = IORESOURCE_IRQ,
+       },
+       {
+               .start  = RK805_IRQ_PWRON_FALL,
+               .end    = RK805_IRQ_PWRON_FALL,
+               .flags  = IORESOURCE_IRQ,
+       },
+};
+
+static struct resource rk816_pwrkey_resources[] = {
+       {
+               .start  = RK816_IRQ_PWRON_RISE,
+               .end    = RK816_IRQ_PWRON_RISE,
+               .flags  = IORESOURCE_IRQ,
+       },
+       {
+               .start  = RK816_IRQ_PWRON_FALL,
+               .end    = RK816_IRQ_PWRON_FALL,
+               .flags  = IORESOURCE_IRQ,
+       },
+};
+
 static const struct mfd_cell rk808s[] = {
        { .name = "rk808-clkout", },
        { .name = "rk808-regulator", },
@@ -83,13 +234,14 @@ static const struct mfd_cell rk808s[] = {
        },
 };
 
-static const struct rk808_reg_data pre_init_reg[] = {
+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_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
        { RK808_VB_MON_REG,       MASK_ALL,         VB_LO_ACT |
                                                    VB_LO_SEL_3500MV },
 };
@@ -148,7 +300,302 @@ static struct regmap_irq_chip rk808_irq_chip = {
        .init_ack_masked = true,
 };
 
+static const struct mfd_cell rk816s[] = {
+       { .name = "rk808-clkout", },
+       { .name = "rk808-regulator", },
+       { .name = "rk8xx-gpio", },
+       {
+               .name = "rk8xx-pwrkey",
+               .num_resources = ARRAY_SIZE(rk816_pwrkey_resources),
+               .resources = &rk816_pwrkey_resources[0],
+       },
+       {
+               .name = "rk808-rtc",
+               .num_resources = ARRAY_SIZE(rtc_resources),
+               .resources = &rtc_resources[0],
+       },
+};
+
+static const struct rk808_reg_data rk816_pre_init_reg[] = {
+       /* buck4 Max ILMIT*/
+       {RK816_BUCK4_CONFIG_REG, BUCK4_MAX_ILIMIT, REG_WRITE_MSK},
+       /* hotdie temperature: 105c*/
+       {RK816_THERMAL_REG, TEMP105C, REG_WRITE_MSK},
+       /* set buck 12.5mv/us */
+       {RK816_BUCK1_CONFIG_REG, BUCK_RATE_12_5MV_US, BUCK_RATE_MSK},
+       {RK816_BUCK2_CONFIG_REG, BUCK_RATE_12_5MV_US, BUCK_RATE_MSK},
+       /* enable RTC_PERIOD & RTC_ALARM int */
+       {RK816_INT_STS_MSK_REG2, RTC_PERIOD_ALARM_INT_EN, REG_WRITE_MSK},
+       /* set bat 3.0 low and act shutdown */
+       {RK816_VB_MON_REG, RK816_VBAT_LOW_3V0 | EN_VABT_LOW_SHUT_DOWN,
+       VBAT_LOW_VOL_MASK | VBAT_LOW_ACT_MASK},
+       /* enable PWRON rising/faling int */
+       {RK816_INT_STS_MSK_REG1, RK816_PWRON_FALL_RISE_INT_EN, REG_WRITE_MSK},
+       /* enable PLUG IN/OUT int */
+       {RK816_INT_STS_MSK_REG3, PLUGIN_OUT_INT_EN, REG_WRITE_MSK},
+       /* clear int flags */
+       {RK816_INT_STS_REG1, ALL_INT_FLAGS_ST, REG_WRITE_MSK},
+       {RK816_INT_STS_REG2, ALL_INT_FLAGS_ST, REG_WRITE_MSK},
+       {RK816_INT_STS_REG3, ALL_INT_FLAGS_ST, REG_WRITE_MSK},
+       {RK816_DCDC_EN_REG2, BOOST_DISABLE, BOOST_EN_MASK},
+};
+
+static const struct regmap_irq rk816_irqs[] = {
+       /* INT_STS */
+       [RK816_IRQ_PWRON_FALL] = {
+               .mask = RK816_IRQ_PWRON_FALL_MSK,
+               .reg_offset = 0,
+       },
+       [RK816_IRQ_PWRON_RISE] = {
+               .mask = RK816_IRQ_PWRON_RISE_MSK,
+               .reg_offset = 0,
+       },
+       [RK816_IRQ_VB_LOW] = {
+               .mask = RK816_IRQ_VB_LOW_MSK,
+               .reg_offset = 1,
+       },
+       [RK816_IRQ_PWRON] = {
+               .mask = RK816_IRQ_PWRON_MSK,
+               .reg_offset = 1,
+       },
+       [RK816_IRQ_PWRON_LP] = {
+               .mask = RK816_IRQ_PWRON_LP_MSK,
+               .reg_offset = 1,
+       },
+       [RK816_IRQ_HOTDIE] = {
+               .mask = RK816_IRQ_HOTDIE_MSK,
+               .reg_offset = 1,
+       },
+       [RK816_IRQ_RTC_ALARM] = {
+               .mask = RK816_IRQ_RTC_ALARM_MSK,
+               .reg_offset = 1,
+       },
+       [RK816_IRQ_RTC_PERIOD] = {
+               .mask = RK816_IRQ_RTC_PERIOD_MSK,
+               .reg_offset = 1,
+       },
+       [RK816_IRQ_USB_OV] = {
+               .mask = RK816_IRQ_USB_OV_MSK,
+               .reg_offset = 1,
+       },
+};
+
+static struct regmap_irq_chip rk816_irq_chip = {
+       .name = "rk816",
+       .irqs = rk816_irqs,
+       .num_irqs = ARRAY_SIZE(rk816_irqs),
+       .num_regs = 2,
+       .irq_reg_stride = 3,
+       .status_base = RK816_INT_STS_REG1,
+       .mask_base = RK816_INT_STS_MSK_REG1,
+       .ack_base = RK816_INT_STS_REG1,
+       .init_ack_masked = true,
+};
+
+static const struct mfd_cell rk818s[] = {
+       { .name = "rk808-clkout", },
+       { .name = "rk808-regulator", },
+       { .name = "rk818-battery", .of_compatible = "rk818-battery", },
+       { .name = "rk818-charger", },
+       {
+               .name = "rk808-rtc",
+               .num_resources = ARRAY_SIZE(rtc_resources),
+               .resources = &rtc_resources[0],
+       },
+};
+
+static const struct rk808_reg_data rk818_pre_init_reg[] = {
+       { RK818_H5V_EN_REG, REF_RDY_CTRL_ENABLE | H5V_EN_MASK,
+                                       REF_RDY_CTRL_ENABLE | H5V_EN_ENABLE },
+       { RK818_DCDC_EN_REG, BOOST_EN_MASK | SWITCH_EN_MASK,
+                                       BOOST_EN_ENABLE | SWITCH_EN_ENABLE },
+       { RK818_SLEEP_SET_OFF_REG1, OTG_SLP_SET_MASK, OTG_SLP_SET_OFF },
+       { RK818_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_250MA },
+       { RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
+};
+
+static const struct regmap_irq rk818_irqs[] = {
+       /* INT_STS */
+       [RK818_IRQ_VOUT_LO] = {
+               .mask = VOUT_LO_MASK,
+               .reg_offset = 0,
+       },
+       [RK818_IRQ_VB_LO] = {
+               .mask = VB_LO_MASK,
+               .reg_offset = 0,
+       },
+       [RK818_IRQ_PWRON] = {
+               .mask = PWRON_MASK,
+               .reg_offset = 0,
+       },
+       [RK818_IRQ_PWRON_LP] = {
+               .mask = PWRON_LP_MASK,
+               .reg_offset = 0,
+       },
+       [RK818_IRQ_HOTDIE] = {
+               .mask = HOTDIE_MASK,
+               .reg_offset = 0,
+       },
+       [RK818_IRQ_RTC_ALARM] = {
+               .mask = RTC_ALARM_MASK,
+               .reg_offset = 0,
+       },
+       [RK818_IRQ_RTC_PERIOD] = {
+               .mask = RTC_PERIOD_MASK,
+               .reg_offset = 0,
+       },
+       [RK818_IRQ_USB_OV] = {
+               .mask = USB_OV_MASK,
+               .reg_offset = 0,
+       },
+       /* INT_STS2 */
+       [RK818_IRQ_PLUG_IN] = {
+               .mask = PLUG_IN_MASK,
+               .reg_offset = 1,
+       },
+       [RK818_IRQ_PLUG_OUT] = {
+               .mask = PLUG_OUT_MASK,
+               .reg_offset = 1,
+       },
+       [RK818_IRQ_CHG_OK] = {
+               .mask = CHGOK_MASK,
+               .reg_offset = 1,
+       },
+       [RK818_IRQ_CHG_TE] = {
+               .mask = CHGTE_MASK,
+               .reg_offset = 1,
+       },
+       [RK818_IRQ_CHG_TS1] = {
+               .mask = CHGTS1_MASK,
+               .reg_offset = 1,
+       },
+       [RK818_IRQ_TS2] = {
+               .mask = TS2_MASK,
+               .reg_offset = 1,
+       },
+       [RK818_IRQ_CHG_CVTLIM] = {
+               .mask = CHG_CVTLIM_MASK,
+               .reg_offset = 1,
+       },
+       [RK818_IRQ_DISCHG_ILIM] = {
+               .mask = DISCHG_ILIM_MASK,
+               .reg_offset = 1,
+       },
+};
+
+static 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 = RK808_INT_STS_REG1,
+       .mask_base = RK808_INT_STS_MSK_REG1,
+       .ack_base = RK808_INT_STS_REG1,
+       .init_ack_masked = true,
+};
+
+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 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 mfd_cell rk805s[] = {
+       { .name = "rk808-clkout", },
+       { .name = "rk818-regulator", },
+       { .name = "rk8xx-gpio", },
+       {
+               .name = "rk8xx-pwrkey",
+               .num_resources = ARRAY_SIZE(pwrkey_resources),
+               .resources = &pwrkey_resources[0],
+       },
+       {
+               .name = "rk808-rtc",
+               .num_resources = ARRAY_SIZE(rtc_resources),
+               .resources = &rtc_resources[0],
+       },
+};
+
+static const struct rk808_reg_data rk805_pre_init_reg[] = {
+       {RK805_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_400MA},
+       {RK805_GPIO_IO_POL_REG, SLP_SD_MSK, SLEEP_FUN},
+       {RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
+       {RK805_THERMAL_REG, TEMP_HOTDIE_MSK, TEMP115C},
+};
+
+static int (*pm_shutdown)(struct regmap *regmap);
+static int (*pm_shutdown_prepare)(struct regmap *regmap);
 static struct i2c_client *rk808_i2c_client;
+
+static void rk808_device_shutdown_prepare(void)
+{
+       int ret;
+       struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
+
+    if (gpio_is_valid(rk808->stby_gpio)) {
+        printk("rk808->stby_gpio(%d)=low\n", rk808->stby_gpio);
+        gpio_direction_output(rk808->stby_gpio, 0);
+    }
+    if (gpio_is_valid(rk808->hold_gpio)) {
+        printk("rk808->hold_gpio(%d)=low\n", rk808->hold_gpio);
+        gpio_direction_output(rk808->hold_gpio, 0);
+        mdelay(200);
+    }
+       if (!rk808) {
+               dev_warn(&rk808_i2c_client->dev,
+                        "have no rk808, so do nothing here\n");
+               return;
+       }
+
+       if (pm_shutdown_prepare) {
+               ret = pm_shutdown_prepare(rk808->regmap);
+               if (ret)
+                       dev_err(&rk808_i2c_client->dev,
+                               "power off prepare error!\n");
+       }
+}
+
 static void rk808_device_shutdown(void)
 {
        int ret;
@@ -160,21 +607,106 @@ static void rk808_device_shutdown(void)
                return;
        }
 
-       ret = regmap_update_bits(rk808->regmap,
-                                RK808_DEVCTRL_REG,
-                                DEV_OFF_RST, DEV_OFF_RST);
-       if (ret)
-               dev_err(&rk808_i2c_client->dev, "power off error!\n");
+       /* close rtc int when power off */
+#if 0
+       regmap_update_bits(rk808->regmap,
+                          RK808_INT_STS_MSK_REG1,
+                          (0x3 << 5), (0x3 << 5));
+       regmap_update_bits(rk808->regmap,
+                          RK808_RTC_INT_REG,
+                          (0x3 << 2), (0x0 << 2));
+#endif
+
+       if (gpio_is_valid(rk808->stby_gpio)) {
+           printk("rk808->stby_gpio(%d)=low\n", rk808->stby_gpio);
+           gpio_direction_output(rk808->stby_gpio, 0);
+       }
+       if (gpio_is_valid(rk808->hold_gpio)) {
+           printk("rk808->hold_gpio(%d)=low\n", rk808->hold_gpio);
+           gpio_direction_output(rk808->hold_gpio, 0);
+           mdelay(200);
+       }
+       if (pm_shutdown) {
+               ret = pm_shutdown(rk808->regmap);
+               if (ret)
+                       dev_err(&rk808_i2c_client->dev, "power off error!\n");
+       }
+    printk("%s-%d: pm_shutdown\n", __FUNCTION__, __LINE__);
 }
 
+static ssize_t rk8xx_dbg_store(struct device *dev,
+                              struct device_attribute *attr,
+                              const char *buf, size_t count)
+{
+       int ret;
+       char cmd;
+       u32 input[2], addr, data;
+       struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
+
+       ret = sscanf(buf, "%c ", &cmd);
+       switch (cmd) {
+       case 'w':
+               ret = sscanf(buf, "%c %x %x ", &cmd, &input[0], &input[1]);
+               if (ret != 3) {
+                       pr_err("erro! cmd format: echo w [addr] [value]\n");
+                       goto out;
+               };
+               addr = input[0] & 0xff;
+               data = input[1] & 0xff;
+               pr_info("cmd : %c %x %x\n\n", cmd, input[0], input[1]);
+               regmap_write(rk808->regmap, addr, data);
+               regmap_read(rk808->regmap, addr, &data);
+               pr_info("new: %x %x\n", addr, data);
+               break;
+       case 'r':
+               ret = sscanf(buf, "%c %x ", &cmd, &input[0]);
+               if (ret != 2) {
+                       pr_err("erro! cmd format: echo r [addr]\n");
+                       goto out;
+               };
+               pr_info("cmd : %c %x\n\n", cmd, input[0]);
+               addr = input[0] & 0xff;
+               regmap_read(rk808->regmap, addr, &data);
+               pr_info("%x %x\n", input[0], data);
+               break;
+       default:
+               pr_err("Unknown command\n");
+               break;
+       }
+
+out:
+       return count;
+}
+
+static struct kobject *rk8xx_kobj;
+static struct device_attribute rk8xx_attrs =
+               __ATTR(rk8xx_dbg, 0200, NULL, rk8xx_dbg_store);
+
+static const struct of_device_id rk808_of_match[] = {
+       { .compatible = "rockchip,rk805" },
+       { .compatible = "rockchip,rk808" },
+       { .compatible = "rockchip,rk816" },
+       { .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;
-       int pm_off = 0;
-       int ret;
-       int i;
+       int (*pm_shutdown_fn)(struct regmap *regmap) = NULL;
+       int (*pm_shutdown_prepare_fn)(struct regmap *regmap) = NULL;
+       const struct rk808_reg_data *pre_init_reg;
+       const struct regmap_config *regmap_config;
+       const struct regmap_irq_chip *irq_chip;
+       const struct mfd_cell *cell;
+       u8 on_source = 0, off_source = 0;
+       int msb, lsb, reg_num, cell_num;
+       int ret, i, pm_off = 0;
+       unsigned int on, off;
 
        if (!client->irq) {
                dev_err(&client->dev, "No interrupt support, no core IRQ\n");
@@ -185,26 +717,114 @@ static int rk808_probe(struct i2c_client *client,
        if (!rk808)
                return -ENOMEM;
 
-       rk808->regmap = devm_regmap_init_i2c(client, &rk808_regmap_config);
+       /* read Chip variant */
+       msb = i2c_smbus_read_byte_data(client, RK808_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, RK808_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, "Pmic Chip id: 0x%lx\n", rk808->variant);
+
+       /* set Chip platform init data*/
+       switch (rk808->variant) {
+       case RK818_ID:
+               cell = rk818s;
+               cell_num = ARRAY_SIZE(rk818s);
+               pre_init_reg = rk818_pre_init_reg;
+               reg_num = ARRAY_SIZE(rk818_pre_init_reg);
+               regmap_config = &rk818_regmap_config;
+               irq_chip = &rk818_irq_chip;
+               pm_shutdown_fn = rk818_shutdown;
+               on_source = RK818_ON_SOURCE_REG;
+               off_source = RK818_OFF_SOURCE_REG;
+               break;
+       case RK816_ID:
+               cell = rk816s;
+               cell_num = ARRAY_SIZE(rk816s);
+               pre_init_reg = rk816_pre_init_reg;
+               reg_num = ARRAY_SIZE(rk816_pre_init_reg);
+               regmap_config = &rk816_regmap_config;
+               irq_chip = &rk816_irq_chip;
+               pm_shutdown_fn = rk816_shutdown;
+               on_source = RK816_ON_SOURCE_REG;
+               off_source = RK816_OFF_SOURCE_REG;
+               break;
+       case RK808_ID:
+               cell = rk808s;
+               cell_num = ARRAY_SIZE(rk808s);
+               pre_init_reg = rk808_pre_init_reg;
+               reg_num = ARRAY_SIZE(rk808_pre_init_reg);
+               regmap_config = &rk808_regmap_config;
+               irq_chip = &rk808_irq_chip;
+               pm_shutdown_fn = rk808_shutdown;
+               break;
+       case RK805_ID:
+               cell = rk805s;
+               cell_num = ARRAY_SIZE(rk805s);
+               pre_init_reg = rk805_pre_init_reg;
+               reg_num = ARRAY_SIZE(rk805_pre_init_reg);
+               regmap_config = &rk805_regmap_config;
+               irq_chip = &rk805_irq_chip;
+               pm_shutdown_prepare_fn = rk805_shutdown_prepare;
+               on_source = RK805_ON_SOURCE_REG;
+               off_source = RK805_OFF_SOURCE_REG;
+               break;
+       default:
+               dev_err(&client->dev, "unsupported RK8XX ID 0x%lx\n",
+                       rk808->variant);
+               return -EINVAL;
+       }
+
+       rk808->regmap = devm_regmap_init_i2c(client, regmap_config);
        if (IS_ERR(rk808->regmap)) {
                dev_err(&client->dev, "regmap initialization failed\n");
                return PTR_ERR(rk808->regmap);
        }
 
-       for (i = 0; i < ARRAY_SIZE(pre_init_reg); i++) {
-               ret = regmap_update_bits(rk808->regmap, pre_init_reg[i].addr,
+       /* on & off source */
+       if (on_source && off_source) {
+               ret = regmap_read(rk808->regmap, on_source, &on);
+               if (ret) {
+                       dev_err(&client->dev, "read 0x%x failed\n", on_source);
+                       return ret;
+               }
+
+               ret = regmap_read(rk808->regmap, off_source, &off);
+               if (ret) {
+                       dev_err(&client->dev, "read 0x%x failed\n", off_source);
+                       return ret;
+               }
+
+               dev_info(&client->dev, "source: on=0x%02x, off=0x%02x\n",
+                        on, off);
+       }
+
+       for (i = 0; i < reg_num; 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);
+                               "0x%x write err\n",
+                               pre_init_reg[i].addr);
                        return ret;
                }
        }
 
        ret = regmap_add_irq_chip(rk808->regmap, client->irq,
                                  IRQF_ONESHOT, -1,
-                                 &rk808_irq_chip, &rk808->irq_data);
+                                 irq_chip, &rk808->irq_data);
        if (ret) {
                dev_err(&client->dev, "Failed to add irq_chip %d\n", ret);
                return ret;
@@ -214,7 +834,7 @@ static int rk808_probe(struct i2c_client *client,
        i2c_set_clientdata(client, rk808);
 
        ret = mfd_add_devices(&client->dev, -1,
-                             rk808s, ARRAY_SIZE(rk808s),
+                             cell, cell_num,
                              NULL, 0, regmap_irq_get_domain(rk808->irq_data));
        if (ret) {
                dev_err(&client->dev, "failed to add MFD devices %d\n", ret);
@@ -223,9 +843,53 @@ static int rk808_probe(struct i2c_client *client,
 
        pm_off = of_property_read_bool(np,
                                "rockchip,system-power-controller");
-       if (pm_off && !pm_power_off) {
+       if (pm_off) {
                rk808_i2c_client = client;
-               pm_power_off = rk808_device_shutdown;
+               if (pm_shutdown_prepare_fn) {
+                       pm_shutdown_prepare = pm_shutdown_prepare_fn;
+                       pm_power_off_prepare = rk808_device_shutdown_prepare;
+               }
+               if (pm_shutdown_fn) {
+                       pm_shutdown = pm_shutdown_fn;
+                       pm_power_off = rk808_device_shutdown;
+               }
+       }
+
+       rk8xx_kobj = kobject_create_and_add("rk8xx", NULL);
+       if (rk8xx_kobj) {
+               ret = sysfs_create_file(rk8xx_kobj, &rk8xx_attrs.attr);
+               if (ret)
+                       dev_err(&client->dev, "create rk8xx sysfs error\n");
+       }
+       //pmic,hold-gpio
+       rk808->hold_gpio = of_get_named_gpio(np, "pmic,hold-gpio", 0);
+       if (gpio_is_valid(rk808->hold_gpio)) {
+               ret = devm_gpio_request(&client->dev, rk808->hold_gpio, "pmic-hold-gpio");
+               if(ret < 0){
+                       dev_err(&client->dev, "pmic-hold-gpio request ERROR\n");
+                       goto err_irq;
+               }
+        dev_dbg(&client->dev, "hold_gpio(%d)=high\n", rk808->hold_gpio);
+        gpio_direction_output(rk808->hold_gpio, 1);
+       }
+    else {
+               dev_err(&client->dev, "Can not read property pmic,hold-gpio\n");
+               //goto err_irq;
+    }
+
+       //pmic,stby-gpio
+       rk808->stby_gpio = of_get_named_gpio(np, "pmic,stby-gpio", 0);
+       if (gpio_is_valid(rk808->stby_gpio)) {
+               ret = devm_gpio_request(&client->dev, rk808->stby_gpio, "pmic-stby-gpio");
+               if(ret < 0){
+                       dev_err(&client->dev, "devm_gpio_request pmic-stby-gpio request ERROR\n");
+                       goto err_irq;
+               }
+        dev_dbg(&client->dev, "stby_gpio(%d)=low\n", rk808->stby_gpio);
+        gpio_direction_output(rk808->stby_gpio, 1);
+       } else {
+               dev_err(&client->dev, "Can not read property pmic,stby-gpio\n");
+               //goto err_irq;
        }
 
        return 0;
@@ -241,19 +905,19 @@ static int rk808_remove(struct i2c_client *client)
 
        regmap_del_irq_chip(client->irq, rk808->irq_data);
        mfd_remove_devices(&client->dev);
-       pm_power_off = NULL;
+       if (pm_power_off == rk808_device_shutdown)
+               pm_power_off = NULL;
+       if (pm_power_off_prepare == rk808_device_shutdown_prepare)
+               pm_power_off_prepare = NULL;
 
        return 0;
 }
 
-static const struct of_device_id rk808_of_match[] = {
-       { .compatible = "rockchip,rk808" },
-       { },
-};
-MODULE_DEVICE_TABLE(of, rk808_of_match);
-
 static const struct i2c_device_id rk808_ids[] = {
+       { "rk805" },
        { "rk808" },
+       { "rk816" },
+       { "rk818" },
        { },
 };
 MODULE_DEVICE_TABLE(i2c, rk808_ids);
@@ -273,4 +937,5 @@ 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("Chen jianhong <chenjh@rock-chips.com>");
 MODULE_DESCRIPTION("RK808 PMIC driver");