drivers/mfd/rk808.c

   1 /*
   2  * MFD core driver for Rockchip RK808
   3  *
   4  * Copyright (c) 2014, Fuzhou Rockchip Electronics Co., Ltd
   5  *
   6  * Author: Chris Zhong <zyw@rock-chips.com>
   7  * Author: Zhang Qing <zhangqing@rock-chips.com>
   8  *
   9  * This program is free software; you can redistribute it and/or modify it
  10  * under the terms and conditions of the GNU General Public License,
  11  * version 2, as published by the Free Software Foundation.
  12  *
  13  * This program is distributed in the hope it will be useful, but WITHOUT
  14  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  15  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  16  * more details.
  17  */
  18
  19 #include <linux/i2c.h>
  20 #include <linux/interrupt.h>
  21 #include <linux/kernel.h>
  22 #include <linux/mfd/rk808.h>
  23 #include <linux/mfd/core.h>
  24 #include <linux/module.h>
  25 #include <linux/of_device.h>
  26 #include <linux/regmap.h>
  27
  28 struct rk808_reg_data {
  29         int addr;
  30         int mask;
  31         int value;
  32 };
  33
  34 static bool rk808_is_volatile_reg(struct device *dev, unsigned int reg)
  35 {
  36         /*
  37          * Notes:
  38          * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
  39          *   we don't use that feature.  It's better to cache.
  40          * - It's unlikely we care that RK808_DEVCTRL_REG is volatile since
  41          *   bits are cleared in case when we shutoff anyway, but better safe.
  42          */
  43
  44         switch (reg) {
  45         case RK808_SECONDS_REG ... RK808_WEEKS_REG:
  46         case RK808_RTC_STATUS_REG:
  47         case RK808_VB_MON_REG:
  48         case RK808_THERMAL_REG:
  49         case RK808_DCDC_UV_STS_REG:
  50         case RK808_LDO_UV_STS_REG:
  51         case RK808_DCDC_PG_REG:
  52         case RK808_LDO_PG_REG:
  53         case RK808_DEVCTRL_REG:
  54         case RK808_INT_STS_REG1:
  55         case RK808_INT_STS_REG2:
  56                 return true;
  57         }
  58
  59         return false;
  60 }
  61
  62 static int rk808_shutdown(struct regmap *regmap)
  63 {
  64         int ret;
  65
  66         ret = regmap_update_bits(regmap,
  67                                  RK808_DEVCTRL_REG,
  68                                  DEV_OFF_RST, DEV_OFF_RST);
  69         return ret;
  70 }
  71
  72 static int rk818_shutdown(struct regmap *regmap)
  73 {
  74         int ret;
  75
  76         ret = regmap_update_bits(regmap,
  77                                  RK818_DEVCTRL_REG,
  78                                  DEV_OFF, DEV_OFF);
  79         return ret;
  80 }
  81
  82 static int rk805_shutdown_prepare(struct regmap *regmap)
  83 {
  84         int ret;
  85
  86         /* close rtc int when power off */
  87         regmap_update_bits(regmap,
  88                            RK808_INT_STS_MSK_REG1,
  89                            (0x3 << 5), (0x3 << 5));
  90         regmap_update_bits(regmap,
  91                            RK808_RTC_INT_REG,
  92                            (0x3 << 2), (0x0 << 2));
  93
  94         /* pmic sleep shutdown function */
  95         ret = regmap_update_bits(regmap,
  96                                  RK805_GPIO_IO_POL_REG,
  97                                  SLP_SD_MSK, SHUTDOWN_FUN);
  98         return ret;
  99 }
 100
 101 static bool rk818_is_volatile_reg(struct device *dev, unsigned int reg)
 102 {
 103         /*
 104          * Notes:
 105          * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
 106          *   we don't use that feature.  It's better to cache.
 107          * - It's unlikely we care that RK808_DEVCTRL_REG is volatile since
 108          *   bits are cleared in case when we shutoff anyway, but better safe.
 109          */
 110
 111         switch (reg) {
 112         case RK808_SECONDS_REG ... RK808_WEEKS_REG:
 113         case RK808_RTC_STATUS_REG:
 114         case RK808_VB_MON_REG:
 115         case RK808_THERMAL_REG:
 116         case RK808_DCDC_EN_REG:
 117         case RK808_DCDC_UV_STS_REG:
 118         case RK808_LDO_UV_STS_REG:
 119         case RK808_DCDC_PG_REG:
 120         case RK808_LDO_PG_REG:
 121         case RK808_DEVCTRL_REG:
 122         case RK808_INT_STS_REG1:
 123         case RK808_INT_STS_REG2:
 124         case RK808_INT_STS_MSK_REG1:
 125         case RK808_INT_STS_MSK_REG2:
 126         case RK818_SUP_STS_REG ... RK818_SAVE_DATA19:
 127                 return true;
 128         }
 129
 130         return false;
 131 }
 132
 133 static const struct regmap_config rk808_regmap_config = {
 134         .reg_bits = 8,
 135         .val_bits = 8,
 136         .max_register = RK808_IO_POL_REG,
 137         .cache_type = REGCACHE_RBTREE,
 138         .volatile_reg = rk808_is_volatile_reg,
 139 };
 140
 141 static const struct regmap_config rk805_regmap_config = {
 142         .reg_bits = 8,
 143         .val_bits = 8,
 144         .max_register = RK805_OFF_SOURCE_REG,
 145         .cache_type = REGCACHE_RBTREE,
 146         .volatile_reg = rk808_is_volatile_reg,
 147 };
 148
 149 static struct resource rtc_resources[] = {
 150         {
 151                 .start  = RK808_IRQ_RTC_ALARM,
 152                 .end    = RK808_IRQ_RTC_ALARM,
 153                 .flags  = IORESOURCE_IRQ,
 154         }
 155 };
 156
 157 static struct resource pwrkey_resources[] = {
 158         {
 159                 .start  = RK805_IRQ_PWRON_RISE,
 160                 .end    = RK805_IRQ_PWRON_RISE,
 161                 .flags  = IORESOURCE_IRQ,
 162         },
 163         {
 164                 .start  = RK805_IRQ_PWRON_FALL,
 165                 .end    = RK805_IRQ_PWRON_FALL,
 166                 .flags  = IORESOURCE_IRQ,
 167         },
 168 };
 169
 170 static const struct mfd_cell rk808s[] = {
 171         { .name = "rk808-clkout", },
 172         { .name = "rk808-regulator", },
 173         {
 174                 .name = "rk808-rtc",
 175                 .num_resources = ARRAY_SIZE(rtc_resources),
 176                 .resources = &rtc_resources[0],
 177         },
 178 };
 179
 180 static const struct rk808_reg_data rk808_pre_init_reg[] = {
 181         { RK808_BUCK3_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_150MA },
 182         { RK808_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_200MA },
 183         { RK808_BOOST_CONFIG_REG, BOOST_ILMIN_MASK, BOOST_ILMIN_100MA },
 184         { RK808_BUCK1_CONFIG_REG, BUCK1_RATE_MASK,  BUCK_ILMIN_200MA },
 185         { RK808_BUCK2_CONFIG_REG, BUCK2_RATE_MASK,  BUCK_ILMIN_200MA },
 186         { RK808_DCDC_UV_ACT_REG,  BUCK_UV_ACT_MASK, BUCK_UV_ACT_DISABLE},
 187         { RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
 188         { RK808_VB_MON_REG,       MASK_ALL,         VB_LO_ACT |
 189                                                     VB_LO_SEL_3500MV },
 190 };
 191
 192 static const struct regmap_irq rk808_irqs[] = {
 193         /* INT_STS */
 194         [RK808_IRQ_VOUT_LO] = {
 195                 .mask = RK808_IRQ_VOUT_LO_MSK,
 196                 .reg_offset = 0,
 197         },
 198         [RK808_IRQ_VB_LO] = {
 199                 .mask = RK808_IRQ_VB_LO_MSK,
 200                 .reg_offset = 0,
 201         },
 202         [RK808_IRQ_PWRON] = {
 203                 .mask = RK808_IRQ_PWRON_MSK,
 204                 .reg_offset = 0,
 205         },
 206         [RK808_IRQ_PWRON_LP] = {
 207                 .mask = RK808_IRQ_PWRON_LP_MSK,
 208                 .reg_offset = 0,
 209         },
 210         [RK808_IRQ_HOTDIE] = {
 211                 .mask = RK808_IRQ_HOTDIE_MSK,
 212                 .reg_offset = 0,
 213         },
 214         [RK808_IRQ_RTC_ALARM] = {
 215                 .mask = RK808_IRQ_RTC_ALARM_MSK,
 216                 .reg_offset = 0,
 217         },
 218         [RK808_IRQ_RTC_PERIOD] = {
 219                 .mask = RK808_IRQ_RTC_PERIOD_MSK,
 220                 .reg_offset = 0,
 221         },
 222
 223         /* INT_STS2 */
 224         [RK808_IRQ_PLUG_IN_INT] = {
 225                 .mask = RK808_IRQ_PLUG_IN_INT_MSK,
 226                 .reg_offset = 1,
 227         },
 228         [RK808_IRQ_PLUG_OUT_INT] = {
 229                 .mask = RK808_IRQ_PLUG_OUT_INT_MSK,
 230                 .reg_offset = 1,
 231         },
 232 };
 233
 234 static struct regmap_irq_chip rk808_irq_chip = {
 235         .name = "rk808",
 236         .irqs = rk808_irqs,
 237         .num_irqs = ARRAY_SIZE(rk808_irqs),
 238         .num_regs = 2,
 239         .irq_reg_stride = 2,
 240         .status_base = RK808_INT_STS_REG1,
 241         .mask_base = RK808_INT_STS_MSK_REG1,
 242         .ack_base = RK808_INT_STS_REG1,
 243         .init_ack_masked = true,
 244 };
 245
 246 static const struct regmap_config rk818_regmap_config = {
 247         .reg_bits = 8,
 248         .val_bits = 8,
 249         .max_register = RK818_SAVE_DATA19,
 250         .cache_type = REGCACHE_RBTREE,
 251         .volatile_reg = rk818_is_volatile_reg,
 252 };
 253
 254 static const struct mfd_cell rk818s[] = {
 255         { .name = "rk808-clkout", },
 256         { .name = "rk808-regulator", },
 257         { .name = "rk818-battery", .of_compatible = "rk818-battery", },
 258         { .name = "rk818-charger", },
 259         {
 260                 .name = "rk808-rtc",
 261                 .num_resources = ARRAY_SIZE(rtc_resources),
 262                 .resources = &rtc_resources[0],
 263         },
 264 };
 265
 266 static const struct rk808_reg_data rk818_pre_init_reg[] = {
 267         { RK818_H5V_EN_REG, REF_RDY_CTRL_ENABLE | H5V_EN_MASK,
 268                                         REF_RDY_CTRL_ENABLE | H5V_EN_ENABLE },
 269         { RK818_DCDC_EN_REG, BOOST_EN_MASK | SWITCH_EN_MASK,
 270                                         BOOST_EN_ENABLE | SWITCH_EN_ENABLE },
 271         { RK818_SLEEP_SET_OFF_REG1, OTG_SLP_SET_MASK, OTG_SLP_SET_OFF },
 272         { RK818_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_250MA },
 273         { RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
 274 };
 275
 276 static const struct regmap_irq rk818_irqs[] = {
 277         /* INT_STS */
 278         [RK818_IRQ_VOUT_LO] = {
 279                 .mask = VOUT_LO_MASK,
 280                 .reg_offset = 0,
 281         },
 282         [RK818_IRQ_VB_LO] = {
 283                 .mask = VB_LO_MASK,
 284                 .reg_offset = 0,
 285         },
 286         [RK818_IRQ_PWRON] = {
 287                 .mask = PWRON_MASK,
 288                 .reg_offset = 0,
 289         },
 290         [RK818_IRQ_PWRON_LP] = {
 291                 .mask = PWRON_LP_MASK,
 292                 .reg_offset = 0,
 293         },
 294         [RK818_IRQ_HOTDIE] = {
 295                 .mask = HOTDIE_MASK,
 296                 .reg_offset = 0,
 297         },
 298         [RK818_IRQ_RTC_ALARM] = {
 299                 .mask = RTC_ALARM_MASK,
 300                 .reg_offset = 0,
 301         },
 302         [RK818_IRQ_RTC_PERIOD] = {
 303                 .mask = RTC_PERIOD_MASK,
 304                 .reg_offset = 0,
 305         },
 306         [RK818_IRQ_USB_OV] = {
 307                 .mask = USB_OV_MASK,
 308                 .reg_offset = 0,
 309         },
 310         /* INT_STS2 */
 311         [RK818_IRQ_PLUG_IN] = {
 312                 .mask = PLUG_IN_MASK,
 313                 .reg_offset = 1,
 314         },
 315         [RK818_IRQ_PLUG_OUT] = {
 316                 .mask = PLUG_OUT_MASK,
 317                 .reg_offset = 1,
 318         },
 319         [RK818_IRQ_CHG_OK] = {
 320                 .mask = CHGOK_MASK,
 321                 .reg_offset = 1,
 322         },
 323         [RK818_IRQ_CHG_TE] = {
 324                 .mask = CHGTE_MASK,
 325                 .reg_offset = 1,
 326         },
 327         [RK818_IRQ_CHG_TS1] = {
 328                 .mask = CHGTS1_MASK,
 329                 .reg_offset = 1,
 330         },
 331         [RK818_IRQ_TS2] = {
 332                 .mask = TS2_MASK,
 333                 .reg_offset = 1,
 334         },
 335         [RK818_IRQ_CHG_CVTLIM] = {
 336                 .mask = CHG_CVTLIM_MASK,
 337                 .reg_offset = 1,
 338         },
 339         [RK818_IRQ_DISCHG_ILIM] = {
 340                 .mask = DISCHG_ILIM_MASK,
 341                 .reg_offset = 1,
 342         },
 343 };
 344
 345 static struct regmap_irq_chip rk818_irq_chip = {
 346         .name = "rk818",
 347         .irqs = rk818_irqs,
 348         .num_irqs = ARRAY_SIZE(rk818_irqs),
 349         .num_regs = 2,
 350         .irq_reg_stride = 2,
 351         .status_base = RK808_INT_STS_REG1,
 352         .mask_base = RK808_INT_STS_MSK_REG1,
 353         .ack_base = RK808_INT_STS_REG1,
 354         .init_ack_masked = true,
 355 };
 356
 357 static const struct regmap_irq rk805_irqs[] = {
 358         [RK805_IRQ_PWRON_RISE] = {
 359                 .mask = RK805_IRQ_PWRON_RISE_MSK,
 360                 .reg_offset = 0,
 361         },
 362         [RK805_IRQ_VB_LOW] = {
 363                 .mask = RK805_IRQ_VB_LOW_MSK,
 364                 .reg_offset = 0,
 365         },
 366         [RK805_IRQ_PWRON] = {
 367                 .mask = RK805_IRQ_PWRON_MSK,
 368                 .reg_offset = 0,
 369         },
 370         [RK805_IRQ_PWRON_LP] = {
 371                 .mask = RK805_IRQ_PWRON_LP_MSK,
 372                 .reg_offset = 0,
 373         },
 374         [RK805_IRQ_HOTDIE] = {
 375                 .mask = RK805_IRQ_HOTDIE_MSK,
 376                 .reg_offset = 0,
 377         },
 378         [RK805_IRQ_RTC_ALARM] = {
 379                 .mask = RK805_IRQ_RTC_ALARM_MSK,
 380                 .reg_offset = 0,
 381         },
 382         [RK805_IRQ_RTC_PERIOD] = {
 383                 .mask = RK805_IRQ_RTC_PERIOD_MSK,
 384                 .reg_offset = 0,
 385         },
 386         [RK805_IRQ_PWRON_FALL] = {
 387                 .mask = RK805_IRQ_PWRON_FALL_MSK,
 388                 .reg_offset = 0,
 389         },
 390 };
 391
 392 static struct regmap_irq_chip rk805_irq_chip = {
 393         .name = "rk805",
 394         .irqs = rk805_irqs,
 395         .num_irqs = ARRAY_SIZE(rk805_irqs),
 396         .num_regs = 1,
 397         .status_base = RK805_INT_STS_REG,
 398         .mask_base = RK805_INT_STS_MSK_REG,
 399         .ack_base = RK805_INT_STS_REG,
 400         .init_ack_masked = true,
 401 };
 402
 403 static const struct mfd_cell rk805s[] = {
 404         { .name = "rk808-clkout", },
 405         { .name = "rk818-regulator", },
 406         { .name = "rk8xx-gpio", },
 407         {
 408                 .name = "rk8xx-pwrkey",
 409                 .num_resources = ARRAY_SIZE(pwrkey_resources),
 410                 .resources = &pwrkey_resources[0],
 411         },
 412         {
 413                 .name = "rk808-rtc",
 414                 .num_resources = ARRAY_SIZE(rtc_resources),
 415                 .resources = &rtc_resources[0],
 416         },
 417 };
 418
 419 static const struct rk808_reg_data rk805_pre_init_reg[] = {
 420         {RK805_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_400MA},
 421         {RK805_GPIO_IO_POL_REG, SLP_SD_MSK, SLEEP_FUN},
 422         {RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
 423         {RK805_THERMAL_REG, TEMP_HOTDIE_MSK, TEMP115C},
 424 };
 425
 426 static int (*pm_shutdown)(struct regmap *regmap);
 427 static int (*pm_shutdown_prepare)(struct regmap *regmap);
 428 static struct i2c_client *rk808_i2c_client;
 429
 430 static void rk808_device_shutdown_prepare(void)
 431 {
 432         int ret;
 433         struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 434
 435         if (!rk808) {
 436                 dev_warn(&rk808_i2c_client->dev,
 437                          "have no rk808, so do nothing here\n");
 438                 return;
 439         }
 440
 441         if (pm_shutdown_prepare) {
 442                 ret = pm_shutdown_prepare(rk808->regmap);
 443                 if (ret)
 444                         dev_err(&rk808_i2c_client->dev,
 445                                 "power off prepare error!\n");
 446         }
 447 }
 448
 449 static void rk808_device_shutdown(void)
 450 {
 451         int ret;
 452         struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 453
 454         if (!rk808) {
 455                 dev_warn(&rk808_i2c_client->dev,
 456                          "have no rk808, so do nothing here\n");
 457                 return;
 458         }
 459
 460         /* close rtc int when power off */
 461         regmap_update_bits(rk808->regmap,
 462                            RK808_INT_STS_MSK_REG1,
 463                            (0x3 << 5), (0x3 << 5));
 464         regmap_update_bits(rk808->regmap,
 465                            RK808_RTC_INT_REG,
 466                            (0x3 << 2), (0x0 << 2));
 467         if (pm_shutdown) {
 468                 ret = pm_shutdown(rk808->regmap);
 469                 if (ret)
 470                         dev_err(&rk808_i2c_client->dev, "power off error!\n");
 471         }
 472 }
 473
 474 static ssize_t rk8xx_dbg_store(struct device *dev,
 475                                struct device_attribute *attr,
 476                                const char *buf, size_t count)
 477 {
 478         int ret;
 479         char cmd;
 480         u32 input[2], addr, data;
 481         struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 482
 483         ret = sscanf(buf, "%c ", &cmd);
 484         switch (cmd) {
 485         case 'w':
 486                 ret = sscanf(buf, "%c %x %x ", &cmd, &input[0], &input[1]);
 487                 if (ret != 3) {
 488                         pr_err("erro! cmd format: echo w [addr] [value]\n");
 489                         goto out;
 490                 };
 491                 addr = input[0] & 0xff;
 492                 data = input[1] & 0xff;
 493                 pr_info("cmd : %c %x %x\n\n", cmd, input[0], input[1]);
 494                 regmap_write(rk808->regmap, addr, data);
 495                 regmap_read(rk808->regmap, addr, &data);
 496                 pr_info("new: %x %x\n", addr, data);
 497                 break;
 498         case 'r':
 499                 ret = sscanf(buf, "%c %x ", &cmd, &input[0]);
 500                 if (ret != 2) {
 501                         pr_err("erro! cmd format: echo r [addr]\n");
 502                         goto out;
 503                 };
 504                 pr_info("cmd : %c %x\n\n", cmd, input[0]);
 505                 addr = input[0] & 0xff;
 506                 regmap_read(rk808->regmap, addr, &data);
 507                 pr_info("%x %x\n", input[0], data);
 508                 break;
 509         default:
 510                 pr_err("Unknown command\n");
 511                 break;
 512         }
 513
 514 out:
 515         return count;
 516 }
 517
 518 static struct kobject *rk8xx_kobj;
 519 static struct device_attribute rk8xx_attrs =
 520                 __ATTR(rk8xx_dbg, 0200, NULL, rk8xx_dbg_store);
 521
 522 static const struct of_device_id rk808_of_match[] = {
 523         { .compatible = "rockchip,rk805" },
 524         { .compatible = "rockchip,rk808" },
 525         { .compatible = "rockchip,rk818" },
 526         { },
 527 };
 528
 529 MODULE_DEVICE_TABLE(of, rk808_of_match);
 530
 531 static int rk808_probe(struct i2c_client *client,
 532                        const struct i2c_device_id *id)
 533 {
 534         struct device_node *np = client->dev.of_node;
 535         struct rk808 *rk808;
 536         int (*pm_shutdown_fn)(struct regmap *regmap) = NULL;
 537         int (*pm_shutdown_prepare_fn)(struct regmap *regmap) = NULL;
 538         const struct rk808_reg_data *pre_init_reg;
 539         const struct regmap_config *regmap_config;
 540         const struct regmap_irq_chip *irq_chip;
 541         const struct mfd_cell *cell;
 542         u8 on_source = 0, off_source = 0;
 543         int msb, lsb, reg_num, cell_num;
 544         int ret, i, pm_off = 0;
 545         unsigned int on, off;
 546
 547         if (!client->irq) {
 548                 dev_err(&client->dev, "No interrupt support, no core IRQ\n");
 549                 return -EINVAL;
 550         }
 551
 552         rk808 = devm_kzalloc(&client->dev, sizeof(*rk808), GFP_KERNEL);
 553         if (!rk808)
 554                 return -ENOMEM;
 555
 556         /* read Chip variant */
 557         msb = i2c_smbus_read_byte_data(client, RK808_ID_MSB);
 558         if (msb < 0) {
 559                 dev_err(&client->dev, "failed to read the chip id at 0x%x\n",
 560                         RK808_ID_MSB);
 561                 return msb;
 562         }
 563
 564         lsb = i2c_smbus_read_byte_data(client, RK808_ID_LSB);
 565         if (lsb < 0) {
 566                 dev_err(&client->dev, "failed to read the chip id at 0x%x\n",
 567                         RK808_ID_LSB);
 568                 return lsb;
 569         }
 570
 571         rk808->variant = ((msb << 8) | lsb) & RK8XX_ID_MSK;
 572         dev_info(&client->dev, "Pmic Chip id: 0x%lx\n", rk808->variant);
 573
 574         /* set Chip platform init data*/
 575         switch (rk808->variant) {
 576         case RK818_ID:
 577                 cell = rk818s;
 578                 cell_num = ARRAY_SIZE(rk818s);
 579                 pre_init_reg = rk818_pre_init_reg;
 580                 reg_num = ARRAY_SIZE(rk818_pre_init_reg);
 581                 regmap_config = &rk818_regmap_config;
 582                 irq_chip = &rk818_irq_chip;
 583                 pm_shutdown_fn = rk818_shutdown;
 584                 on_source = RK818_ON_SOURCE_REG;
 585                 off_source = RK818_OFF_SOURCE_REG;
 586                 break;
 587         case RK808_ID:
 588                 cell = rk808s;
 589                 cell_num = ARRAY_SIZE(rk808s);
 590                 pre_init_reg = rk808_pre_init_reg;
 591                 reg_num = ARRAY_SIZE(rk808_pre_init_reg);
 592                 regmap_config = &rk808_regmap_config;
 593                 irq_chip = &rk808_irq_chip;
 594                 pm_shutdown_fn = rk808_shutdown;
 595                 break;
 596         case RK805_ID:
 597                 cell = rk805s;
 598                 cell_num = ARRAY_SIZE(rk805s);
 599                 pre_init_reg = rk805_pre_init_reg;
 600                 reg_num = ARRAY_SIZE(rk805_pre_init_reg);
 601                 regmap_config = &rk805_regmap_config;
 602                 irq_chip = &rk805_irq_chip;
 603                 pm_shutdown_prepare_fn = rk805_shutdown_prepare;
 604                 on_source = RK805_ON_SOURCE_REG;
 605                 off_source = RK805_OFF_SOURCE_REG;
 606                 break;
 607         default:
 608                 dev_err(&client->dev, "unsupported RK8XX ID 0x%lx\n",
 609                         rk808->variant);
 610                 return -EINVAL;
 611         }
 612
 613         rk808->regmap = devm_regmap_init_i2c(client, regmap_config);
 614         if (IS_ERR(rk808->regmap)) {
 615                 dev_err(&client->dev, "regmap initialization failed\n");
 616                 return PTR_ERR(rk808->regmap);
 617         }
 618
 619         /* on & off source */
 620         if (on_source && off_source) {
 621                 ret = regmap_read(rk808->regmap, on_source, &on);
 622                 if (ret) {
 623                         dev_err(&client->dev, "read 0x%x failed\n", on_source);
 624                         return ret;
 625                 }
 626
 627                 ret = regmap_read(rk808->regmap, off_source, &off);
 628                 if (ret) {
 629                         dev_err(&client->dev, "read 0x%x failed\n", off_source);
 630                         return ret;
 631                 }
 632
 633                 dev_info(&client->dev, "source: on=0x%02x, off=0x%02x\n",
 634                          on, off);
 635         }
 636
 637         for (i = 0; i < reg_num; i++) {
 638                 ret = regmap_update_bits(rk808->regmap,
 639                                          pre_init_reg[i].addr,
 640                                          pre_init_reg[i].mask,
 641                                          pre_init_reg[i].value);
 642                 if (ret) {
 643                         dev_err(&client->dev,
 644                                 "0x%x write err\n",
 645                                 pre_init_reg[i].addr);
 646                         return ret;
 647                 }
 648         }
 649
 650         ret = regmap_add_irq_chip(rk808->regmap, client->irq,
 651                                   IRQF_ONESHOT, -1,
 652                                   irq_chip, &rk808->irq_data);
 653         if (ret) {
 654                 dev_err(&client->dev, "Failed to add irq_chip %d\n", ret);
 655                 return ret;
 656         }
 657
 658         rk808->i2c = client;
 659         i2c_set_clientdata(client, rk808);
 660
 661         ret = mfd_add_devices(&client->dev, -1,
 662                               cell, cell_num,
 663                               NULL, 0, regmap_irq_get_domain(rk808->irq_data));
 664         if (ret) {
 665                 dev_err(&client->dev, "failed to add MFD devices %d\n", ret);
 666                 goto err_irq;
 667         }
 668
 669         pm_off = of_property_read_bool(np,
 670                                 "rockchip,system-power-controller");
 671         if (pm_off) {
 672                 rk808_i2c_client = client;
 673                 if (pm_shutdown_prepare_fn) {
 674                         pm_shutdown_prepare = pm_shutdown_prepare_fn;
 675                         pm_power_off_prepare = rk808_device_shutdown_prepare;
 676                 }
 677                 if (pm_shutdown_fn) {
 678                         pm_shutdown = pm_shutdown_fn;
 679                         pm_power_off = rk808_device_shutdown;
 680                 }
 681         }
 682
 683         rk8xx_kobj = kobject_create_and_add("rk8xx", NULL);
 684         if (rk8xx_kobj) {
 685                 ret = sysfs_create_file(rk8xx_kobj, &rk8xx_attrs.attr);
 686                 if (ret)
 687                         dev_err(&client->dev, "create rk8xx sysfs error\n");
 688         }
 689
 690         return 0;
 691
 692 err_irq:
 693         regmap_del_irq_chip(client->irq, rk808->irq_data);
 694         return ret;
 695 }
 696
 697 static int rk808_remove(struct i2c_client *client)
 698 {
 699         struct rk808 *rk808 = i2c_get_clientdata(client);
 700
 701         regmap_del_irq_chip(client->irq, rk808->irq_data);
 702         mfd_remove_devices(&client->dev);
 703         if (pm_power_off == rk808_device_shutdown)
 704                 pm_power_off = NULL;
 705         if (pm_power_off_prepare == rk808_device_shutdown_prepare)
 706                 pm_power_off_prepare = NULL;
 707
 708         return 0;
 709 }
 710
 711 static const struct i2c_device_id rk808_ids[] = {
 712         { "rk805" },
 713         { "rk808" },
 714         { "rk818" },
 715         { },
 716 };
 717 MODULE_DEVICE_TABLE(i2c, rk808_ids);
 718
 719 static struct i2c_driver rk808_i2c_driver = {
 720         .driver = {
 721                 .name = "rk808",
 722                 .of_match_table = rk808_of_match,
 723         },
 724         .probe    = rk808_probe,
 725         .remove   = rk808_remove,
 726         .id_table = rk808_ids,
 727 };
 728
 729 module_i2c_driver(rk808_i2c_driver);
 730
 731 MODULE_LICENSE("GPL");
 732 MODULE_AUTHOR("Chris Zhong <zyw@rock-chips.com>");
 733 MODULE_AUTHOR("Zhang Qing <zhangqing@rock-chips.com>");
 734 MODULE_AUTHOR("Chen jianhong <chenjh@rock-chips.com>");
 735 MODULE_DESCRIPTION("RK808 PMIC driver");