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 rk816_shutdown(struct regmap *regmap)
  73 {
  74         int ret;
  75
  76         ret = regmap_update_bits(regmap,
  77                                  RK816_DEV_CTRL_REG,
  78                                  DEV_OFF, DEV_OFF);
  79         return ret;
  80 }
  81
  82 static int rk818_shutdown(struct regmap *regmap)
  83 {
  84         int ret;
  85
  86         ret = regmap_update_bits(regmap,
  87                                  RK818_DEVCTRL_REG,
  88                                  DEV_OFF, DEV_OFF);
  89         return ret;
  90 }
  91
  92 static int rk805_shutdown_prepare(struct regmap *regmap)
  93 {
  94         int ret;
  95
  96         /* close rtc int when power off */
  97         regmap_update_bits(regmap,
  98                            RK808_INT_STS_MSK_REG1,
  99                            (0x3 << 5), (0x3 << 5));
 100         regmap_update_bits(regmap,
 101                            RK808_RTC_INT_REG,
 102                            (0x3 << 2), (0x0 << 2));
 103
 104         /* pmic sleep shutdown function */
 105         ret = regmap_update_bits(regmap,
 106                                  RK805_GPIO_IO_POL_REG,
 107                                  SLP_SD_MSK, SHUTDOWN_FUN);
 108         return ret;
 109 }
 110
 111 static bool rk818_is_volatile_reg(struct device *dev, unsigned int reg)
 112 {
 113         /*
 114          * Notes:
 115          * - Technically the ROUND_30s bit makes RTC_CTRL_REG volatile, but
 116          *   we don't use that feature.  It's better to cache.
 117          * - It's unlikely we care that RK808_DEVCTRL_REG is volatile since
 118          *   bits are cleared in case when we shutoff anyway, but better safe.
 119          */
 120
 121         switch (reg) {
 122         case RK808_SECONDS_REG ... RK808_WEEKS_REG:
 123         case RK808_RTC_STATUS_REG:
 124         case RK808_VB_MON_REG:
 125         case RK808_THERMAL_REG:
 126         case RK808_DCDC_EN_REG:
 127         case RK808_DCDC_UV_STS_REG:
 128         case RK808_LDO_UV_STS_REG:
 129         case RK808_DCDC_PG_REG:
 130         case RK808_LDO_PG_REG:
 131         case RK808_DEVCTRL_REG:
 132         case RK808_INT_STS_REG1:
 133         case RK808_INT_STS_REG2:
 134         case RK808_INT_STS_MSK_REG1:
 135         case RK808_INT_STS_MSK_REG2:
 136         case RK818_SUP_STS_REG ... RK818_SAVE_DATA19:
 137                 return true;
 138         }
 139
 140         return false;
 141 }
 142
 143 static const struct regmap_config rk808_regmap_config = {
 144         .reg_bits = 8,
 145         .val_bits = 8,
 146         .max_register = RK808_IO_POL_REG,
 147         .cache_type = REGCACHE_RBTREE,
 148         .volatile_reg = rk808_is_volatile_reg,
 149 };
 150
 151 static const struct regmap_config rk805_regmap_config = {
 152         .reg_bits = 8,
 153         .val_bits = 8,
 154         .max_register = RK805_OFF_SOURCE_REG,
 155         .cache_type = REGCACHE_RBTREE,
 156         .volatile_reg = rk808_is_volatile_reg,
 157 };
 158
 159 static const struct regmap_config rk816_regmap_config = {
 160         .reg_bits = 8,
 161         .val_bits = 8,
 162         .max_register = RK816_DATA18_REG,
 163         .cache_type = REGCACHE_RBTREE,
 164         .volatile_reg = rk818_is_volatile_reg,
 165 };
 166
 167 static const struct regmap_config rk818_regmap_config = {
 168         .reg_bits = 8,
 169         .val_bits = 8,
 170         .max_register = RK818_SAVE_DATA19,
 171         .cache_type = REGCACHE_RBTREE,
 172         .volatile_reg = rk818_is_volatile_reg,
 173 };
 174
 175 static struct resource rtc_resources[] = {
 176         {
 177                 .start  = RK808_IRQ_RTC_ALARM,
 178                 .end    = RK808_IRQ_RTC_ALARM,
 179                 .flags  = IORESOURCE_IRQ,
 180         }
 181 };
 182
 183 static struct resource pwrkey_resources[] = {
 184         {
 185                 .start  = RK805_IRQ_PWRON_RISE,
 186                 .end    = RK805_IRQ_PWRON_RISE,
 187                 .flags  = IORESOURCE_IRQ,
 188         },
 189         {
 190                 .start  = RK805_IRQ_PWRON_FALL,
 191                 .end    = RK805_IRQ_PWRON_FALL,
 192                 .flags  = IORESOURCE_IRQ,
 193         },
 194 };
 195
 196 static struct resource rk816_pwrkey_resources[] = {
 197         {
 198                 .start  = RK816_IRQ_PWRON_RISE,
 199                 .end    = RK816_IRQ_PWRON_RISE,
 200                 .flags  = IORESOURCE_IRQ,
 201         },
 202         {
 203                 .start  = RK816_IRQ_PWRON_FALL,
 204                 .end    = RK816_IRQ_PWRON_FALL,
 205                 .flags  = IORESOURCE_IRQ,
 206         },
 207 };
 208
 209 static const struct mfd_cell rk808s[] = {
 210         { .name = "rk808-clkout", },
 211         { .name = "rk808-regulator", },
 212         {
 213                 .name = "rk808-rtc",
 214                 .num_resources = ARRAY_SIZE(rtc_resources),
 215                 .resources = &rtc_resources[0],
 216         },
 217 };
 218
 219 static const struct rk808_reg_data rk808_pre_init_reg[] = {
 220         { RK808_BUCK3_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_150MA },
 221         { RK808_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_200MA },
 222         { RK808_BOOST_CONFIG_REG, BOOST_ILMIN_MASK, BOOST_ILMIN_100MA },
 223         { RK808_BUCK1_CONFIG_REG, BUCK1_RATE_MASK,  BUCK_ILMIN_200MA },
 224         { RK808_BUCK2_CONFIG_REG, BUCK2_RATE_MASK,  BUCK_ILMIN_200MA },
 225         { RK808_DCDC_UV_ACT_REG,  BUCK_UV_ACT_MASK, BUCK_UV_ACT_DISABLE},
 226         { RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
 227         { RK808_VB_MON_REG,       MASK_ALL,         VB_LO_ACT |
 228                                                     VB_LO_SEL_3500MV },
 229 };
 230
 231 static const struct regmap_irq rk808_irqs[] = {
 232         /* INT_STS */
 233         [RK808_IRQ_VOUT_LO] = {
 234                 .mask = RK808_IRQ_VOUT_LO_MSK,
 235                 .reg_offset = 0,
 236         },
 237         [RK808_IRQ_VB_LO] = {
 238                 .mask = RK808_IRQ_VB_LO_MSK,
 239                 .reg_offset = 0,
 240         },
 241         [RK808_IRQ_PWRON] = {
 242                 .mask = RK808_IRQ_PWRON_MSK,
 243                 .reg_offset = 0,
 244         },
 245         [RK808_IRQ_PWRON_LP] = {
 246                 .mask = RK808_IRQ_PWRON_LP_MSK,
 247                 .reg_offset = 0,
 248         },
 249         [RK808_IRQ_HOTDIE] = {
 250                 .mask = RK808_IRQ_HOTDIE_MSK,
 251                 .reg_offset = 0,
 252         },
 253         [RK808_IRQ_RTC_ALARM] = {
 254                 .mask = RK808_IRQ_RTC_ALARM_MSK,
 255                 .reg_offset = 0,
 256         },
 257         [RK808_IRQ_RTC_PERIOD] = {
 258                 .mask = RK808_IRQ_RTC_PERIOD_MSK,
 259                 .reg_offset = 0,
 260         },
 261
 262         /* INT_STS2 */
 263         [RK808_IRQ_PLUG_IN_INT] = {
 264                 .mask = RK808_IRQ_PLUG_IN_INT_MSK,
 265                 .reg_offset = 1,
 266         },
 267         [RK808_IRQ_PLUG_OUT_INT] = {
 268                 .mask = RK808_IRQ_PLUG_OUT_INT_MSK,
 269                 .reg_offset = 1,
 270         },
 271 };
 272
 273 static struct regmap_irq_chip rk808_irq_chip = {
 274         .name = "rk808",
 275         .irqs = rk808_irqs,
 276         .num_irqs = ARRAY_SIZE(rk808_irqs),
 277         .num_regs = 2,
 278         .irq_reg_stride = 2,
 279         .status_base = RK808_INT_STS_REG1,
 280         .mask_base = RK808_INT_STS_MSK_REG1,
 281         .ack_base = RK808_INT_STS_REG1,
 282         .init_ack_masked = true,
 283 };
 284
 285 static const struct mfd_cell rk816s[] = {
 286         { .name = "rk808-clkout", },
 287         { .name = "rk808-regulator", },
 288         { .name = "rk8xx-gpio", },
 289         {
 290                 .name = "rk8xx-pwrkey",
 291                 .num_resources = ARRAY_SIZE(rk816_pwrkey_resources),
 292                 .resources = &rk816_pwrkey_resources[0],
 293         },
 294         {
 295                 .name = "rk808-rtc",
 296                 .num_resources = ARRAY_SIZE(rtc_resources),
 297                 .resources = &rtc_resources[0],
 298         },
 299 };
 300
 301 static const struct rk808_reg_data rk816_pre_init_reg[] = {
 302         /* buck4 Max ILMIT*/
 303         {RK816_BUCK4_CONFIG_REG, BUCK4_MAX_ILIMIT, REG_WRITE_MSK},
 304         /* hotdie temperature: 105c*/
 305         {RK816_THERMAL_REG, TEMP105C, REG_WRITE_MSK},
 306         /* set buck 12.5mv/us */
 307         {RK816_BUCK1_CONFIG_REG, BUCK_RATE_12_5MV_US, BUCK_RATE_MSK},
 308         {RK816_BUCK2_CONFIG_REG, BUCK_RATE_12_5MV_US, BUCK_RATE_MSK},
 309         /* enable RTC_PERIOD & RTC_ALARM int */
 310         {RK816_INT_STS_MSK_REG2, RTC_PERIOD_ALARM_INT_EN, REG_WRITE_MSK},
 311         /* set bat 3.0 low and act shutdown */
 312         {RK816_VB_MON_REG, RK816_VBAT_LOW_3V0 | EN_VABT_LOW_SHUT_DOWN,
 313         VBAT_LOW_VOL_MASK | VBAT_LOW_ACT_MASK},
 314         /* enable PWRON rising/faling int */
 315         {RK816_INT_STS_MSK_REG1, RK816_PWRON_FALL_RISE_INT_EN, REG_WRITE_MSK},
 316         /* enable PLUG IN/OUT int */
 317         {RK816_INT_STS_MSK_REG3, PLUGIN_OUT_INT_EN, REG_WRITE_MSK},
 318         /* clear int flags */
 319         {RK816_INT_STS_REG1, ALL_INT_FLAGS_ST, REG_WRITE_MSK},
 320         {RK816_INT_STS_REG2, ALL_INT_FLAGS_ST, REG_WRITE_MSK},
 321         {RK816_INT_STS_REG3, ALL_INT_FLAGS_ST, REG_WRITE_MSK},
 322         {RK816_DCDC_EN_REG2, BOOST_DISABLE, BOOST_EN_MASK},
 323 };
 324
 325 static const struct regmap_irq rk816_irqs[] = {
 326         /* INT_STS */
 327         [RK816_IRQ_PWRON_FALL] = {
 328                 .mask = RK816_IRQ_PWRON_FALL_MSK,
 329                 .reg_offset = 0,
 330         },
 331         [RK816_IRQ_PWRON_RISE] = {
 332                 .mask = RK816_IRQ_PWRON_RISE_MSK,
 333                 .reg_offset = 0,
 334         },
 335         [RK816_IRQ_VB_LOW] = {
 336                 .mask = RK816_IRQ_VB_LOW_MSK,
 337                 .reg_offset = 1,
 338         },
 339         [RK816_IRQ_PWRON] = {
 340                 .mask = RK816_IRQ_PWRON_MSK,
 341                 .reg_offset = 1,
 342         },
 343         [RK816_IRQ_PWRON_LP] = {
 344                 .mask = RK816_IRQ_PWRON_LP_MSK,
 345                 .reg_offset = 1,
 346         },
 347         [RK816_IRQ_HOTDIE] = {
 348                 .mask = RK816_IRQ_HOTDIE_MSK,
 349                 .reg_offset = 1,
 350         },
 351         [RK816_IRQ_RTC_ALARM] = {
 352                 .mask = RK816_IRQ_RTC_ALARM_MSK,
 353                 .reg_offset = 1,
 354         },
 355         [RK816_IRQ_RTC_PERIOD] = {
 356                 .mask = RK816_IRQ_RTC_PERIOD_MSK,
 357                 .reg_offset = 1,
 358         },
 359         [RK816_IRQ_USB_OV] = {
 360                 .mask = RK816_IRQ_USB_OV_MSK,
 361                 .reg_offset = 1,
 362         },
 363 };
 364
 365 static struct regmap_irq_chip rk816_irq_chip = {
 366         .name = "rk816",
 367         .irqs = rk816_irqs,
 368         .num_irqs = ARRAY_SIZE(rk816_irqs),
 369         .num_regs = 2,
 370         .irq_reg_stride = 3,
 371         .status_base = RK816_INT_STS_REG1,
 372         .mask_base = RK816_INT_STS_MSK_REG1,
 373         .ack_base = RK816_INT_STS_REG1,
 374         .init_ack_masked = true,
 375 };
 376
 377 static const struct mfd_cell rk818s[] = {
 378         { .name = "rk808-clkout", },
 379         { .name = "rk808-regulator", },
 380         { .name = "rk818-battery", .of_compatible = "rk818-battery", },
 381         { .name = "rk818-charger", },
 382         {
 383                 .name = "rk808-rtc",
 384                 .num_resources = ARRAY_SIZE(rtc_resources),
 385                 .resources = &rtc_resources[0],
 386         },
 387 };
 388
 389 static const struct rk808_reg_data rk818_pre_init_reg[] = {
 390         { RK818_H5V_EN_REG, REF_RDY_CTRL_ENABLE | H5V_EN_MASK,
 391                                         REF_RDY_CTRL_ENABLE | H5V_EN_ENABLE },
 392         { RK818_DCDC_EN_REG, BOOST_EN_MASK | SWITCH_EN_MASK,
 393                                         BOOST_EN_ENABLE | SWITCH_EN_ENABLE },
 394         { RK818_SLEEP_SET_OFF_REG1, OTG_SLP_SET_MASK, OTG_SLP_SET_OFF },
 395         { RK818_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK,  BUCK_ILMIN_250MA },
 396         { RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
 397 };
 398
 399 static const struct regmap_irq rk818_irqs[] = {
 400         /* INT_STS */
 401         [RK818_IRQ_VOUT_LO] = {
 402                 .mask = VOUT_LO_MASK,
 403                 .reg_offset = 0,
 404         },
 405         [RK818_IRQ_VB_LO] = {
 406                 .mask = VB_LO_MASK,
 407                 .reg_offset = 0,
 408         },
 409         [RK818_IRQ_PWRON] = {
 410                 .mask = PWRON_MASK,
 411                 .reg_offset = 0,
 412         },
 413         [RK818_IRQ_PWRON_LP] = {
 414                 .mask = PWRON_LP_MASK,
 415                 .reg_offset = 0,
 416         },
 417         [RK818_IRQ_HOTDIE] = {
 418                 .mask = HOTDIE_MASK,
 419                 .reg_offset = 0,
 420         },
 421         [RK818_IRQ_RTC_ALARM] = {
 422                 .mask = RTC_ALARM_MASK,
 423                 .reg_offset = 0,
 424         },
 425         [RK818_IRQ_RTC_PERIOD] = {
 426                 .mask = RTC_PERIOD_MASK,
 427                 .reg_offset = 0,
 428         },
 429         [RK818_IRQ_USB_OV] = {
 430                 .mask = USB_OV_MASK,
 431                 .reg_offset = 0,
 432         },
 433         /* INT_STS2 */
 434         [RK818_IRQ_PLUG_IN] = {
 435                 .mask = PLUG_IN_MASK,
 436                 .reg_offset = 1,
 437         },
 438         [RK818_IRQ_PLUG_OUT] = {
 439                 .mask = PLUG_OUT_MASK,
 440                 .reg_offset = 1,
 441         },
 442         [RK818_IRQ_CHG_OK] = {
 443                 .mask = CHGOK_MASK,
 444                 .reg_offset = 1,
 445         },
 446         [RK818_IRQ_CHG_TE] = {
 447                 .mask = CHGTE_MASK,
 448                 .reg_offset = 1,
 449         },
 450         [RK818_IRQ_CHG_TS1] = {
 451                 .mask = CHGTS1_MASK,
 452                 .reg_offset = 1,
 453         },
 454         [RK818_IRQ_TS2] = {
 455                 .mask = TS2_MASK,
 456                 .reg_offset = 1,
 457         },
 458         [RK818_IRQ_CHG_CVTLIM] = {
 459                 .mask = CHG_CVTLIM_MASK,
 460                 .reg_offset = 1,
 461         },
 462         [RK818_IRQ_DISCHG_ILIM] = {
 463                 .mask = DISCHG_ILIM_MASK,
 464                 .reg_offset = 1,
 465         },
 466 };
 467
 468 static struct regmap_irq_chip rk818_irq_chip = {
 469         .name = "rk818",
 470         .irqs = rk818_irqs,
 471         .num_irqs = ARRAY_SIZE(rk818_irqs),
 472         .num_regs = 2,
 473         .irq_reg_stride = 2,
 474         .status_base = RK808_INT_STS_REG1,
 475         .mask_base = RK808_INT_STS_MSK_REG1,
 476         .ack_base = RK808_INT_STS_REG1,
 477         .init_ack_masked = true,
 478 };
 479
 480 static const struct regmap_irq rk805_irqs[] = {
 481         [RK805_IRQ_PWRON_RISE] = {
 482                 .mask = RK805_IRQ_PWRON_RISE_MSK,
 483                 .reg_offset = 0,
 484         },
 485         [RK805_IRQ_VB_LOW] = {
 486                 .mask = RK805_IRQ_VB_LOW_MSK,
 487                 .reg_offset = 0,
 488         },
 489         [RK805_IRQ_PWRON] = {
 490                 .mask = RK805_IRQ_PWRON_MSK,
 491                 .reg_offset = 0,
 492         },
 493         [RK805_IRQ_PWRON_LP] = {
 494                 .mask = RK805_IRQ_PWRON_LP_MSK,
 495                 .reg_offset = 0,
 496         },
 497         [RK805_IRQ_HOTDIE] = {
 498                 .mask = RK805_IRQ_HOTDIE_MSK,
 499                 .reg_offset = 0,
 500         },
 501         [RK805_IRQ_RTC_ALARM] = {
 502                 .mask = RK805_IRQ_RTC_ALARM_MSK,
 503                 .reg_offset = 0,
 504         },
 505         [RK805_IRQ_RTC_PERIOD] = {
 506                 .mask = RK805_IRQ_RTC_PERIOD_MSK,
 507                 .reg_offset = 0,
 508         },
 509         [RK805_IRQ_PWRON_FALL] = {
 510                 .mask = RK805_IRQ_PWRON_FALL_MSK,
 511                 .reg_offset = 0,
 512         },
 513 };
 514
 515 static struct regmap_irq_chip rk805_irq_chip = {
 516         .name = "rk805",
 517         .irqs = rk805_irqs,
 518         .num_irqs = ARRAY_SIZE(rk805_irqs),
 519         .num_regs = 1,
 520         .status_base = RK805_INT_STS_REG,
 521         .mask_base = RK805_INT_STS_MSK_REG,
 522         .ack_base = RK805_INT_STS_REG,
 523         .init_ack_masked = true,
 524 };
 525
 526 static const struct mfd_cell rk805s[] = {
 527         { .name = "rk808-clkout", },
 528         { .name = "rk818-regulator", },
 529         { .name = "rk8xx-gpio", },
 530         {
 531                 .name = "rk8xx-pwrkey",
 532                 .num_resources = ARRAY_SIZE(pwrkey_resources),
 533                 .resources = &pwrkey_resources[0],
 534         },
 535         {
 536                 .name = "rk808-rtc",
 537                 .num_resources = ARRAY_SIZE(rtc_resources),
 538                 .resources = &rtc_resources[0],
 539         },
 540 };
 541
 542 static const struct rk808_reg_data rk805_pre_init_reg[] = {
 543         {RK805_BUCK4_CONFIG_REG, BUCK_ILMIN_MASK, BUCK_ILMIN_400MA},
 544         {RK805_GPIO_IO_POL_REG, SLP_SD_MSK, SLEEP_FUN},
 545         {RK808_RTC_CTRL_REG, RTC_STOP, RTC_STOP},
 546         {RK805_THERMAL_REG, TEMP_HOTDIE_MSK, TEMP115C},
 547 };
 548
 549 static int (*pm_shutdown)(struct regmap *regmap);
 550 static int (*pm_shutdown_prepare)(struct regmap *regmap);
 551 static struct i2c_client *rk808_i2c_client;
 552
 553 static void rk808_device_shutdown_prepare(void)
 554 {
 555         int ret;
 556         struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 557
 558         if (!rk808) {
 559                 dev_warn(&rk808_i2c_client->dev,
 560                          "have no rk808, so do nothing here\n");
 561                 return;
 562         }
 563
 564         if (pm_shutdown_prepare) {
 565                 ret = pm_shutdown_prepare(rk808->regmap);
 566                 if (ret)
 567                         dev_err(&rk808_i2c_client->dev,
 568                                 "power off prepare error!\n");
 569         }
 570 }
 571
 572 static void rk808_device_shutdown(void)
 573 {
 574         int ret;
 575         struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 576
 577         if (!rk808) {
 578                 dev_warn(&rk808_i2c_client->dev,
 579                          "have no rk808, so do nothing here\n");
 580                 return;
 581         }
 582
 583         /* close rtc int when power off */
 584         regmap_update_bits(rk808->regmap,
 585                            RK808_INT_STS_MSK_REG1,
 586                            (0x3 << 5), (0x3 << 5));
 587         regmap_update_bits(rk808->regmap,
 588                            RK808_RTC_INT_REG,
 589                            (0x3 << 2), (0x0 << 2));
 590         if (pm_shutdown) {
 591                 ret = pm_shutdown(rk808->regmap);
 592                 if (ret)
 593                         dev_err(&rk808_i2c_client->dev, "power off error!\n");
 594         }
 595 }
 596
 597 static ssize_t rk8xx_dbg_store(struct device *dev,
 598                                struct device_attribute *attr,
 599                                const char *buf, size_t count)
 600 {
 601         int ret;
 602         char cmd;
 603         u32 input[2], addr, data;
 604         struct rk808 *rk808 = i2c_get_clientdata(rk808_i2c_client);
 605
 606         ret = sscanf(buf, "%c ", &cmd);
 607         switch (cmd) {
 608         case 'w':
 609                 ret = sscanf(buf, "%c %x %x ", &cmd, &input[0], &input[1]);
 610                 if (ret != 3) {
 611                         pr_err("erro! cmd format: echo w [addr] [value]\n");
 612                         goto out;
 613                 };
 614                 addr = input[0] & 0xff;
 615                 data = input[1] & 0xff;
 616                 pr_info("cmd : %c %x %x\n\n", cmd, input[0], input[1]);
 617                 regmap_write(rk808->regmap, addr, data);
 618                 regmap_read(rk808->regmap, addr, &data);
 619                 pr_info("new: %x %x\n", addr, data);
 620                 break;
 621         case 'r':
 622                 ret = sscanf(buf, "%c %x ", &cmd, &input[0]);
 623                 if (ret != 2) {
 624                         pr_err("erro! cmd format: echo r [addr]\n");
 625                         goto out;
 626                 };
 627                 pr_info("cmd : %c %x\n\n", cmd, input[0]);
 628                 addr = input[0] & 0xff;
 629                 regmap_read(rk808->regmap, addr, &data);
 630                 pr_info("%x %x\n", input[0], data);
 631                 break;
 632         default:
 633                 pr_err("Unknown command\n");
 634                 break;
 635         }
 636
 637 out:
 638         return count;
 639 }
 640
 641 static struct kobject *rk8xx_kobj;
 642 static struct device_attribute rk8xx_attrs =
 643                 __ATTR(rk8xx_dbg, 0200, NULL, rk8xx_dbg_store);
 644
 645 static const struct of_device_id rk808_of_match[] = {
 646         { .compatible = "rockchip,rk805" },
 647         { .compatible = "rockchip,rk808" },
 648         { .compatible = "rockchip,rk816" },
 649         { .compatible = "rockchip,rk818" },
 650         { },
 651 };
 652
 653 MODULE_DEVICE_TABLE(of, rk808_of_match);
 654
 655 static int rk808_probe(struct i2c_client *client,
 656                        const struct i2c_device_id *id)
 657 {
 658         struct device_node *np = client->dev.of_node;
 659         struct rk808 *rk808;
 660         int (*pm_shutdown_fn)(struct regmap *regmap) = NULL;
 661         int (*pm_shutdown_prepare_fn)(struct regmap *regmap) = NULL;
 662         const struct rk808_reg_data *pre_init_reg;
 663         const struct regmap_config *regmap_config;
 664         const struct regmap_irq_chip *irq_chip;
 665         const struct mfd_cell *cell;
 666         u8 on_source = 0, off_source = 0;
 667         int msb, lsb, reg_num, cell_num;
 668         int ret, i, pm_off = 0;
 669         unsigned int on, off;
 670
 671         if (!client->irq) {
 672                 dev_err(&client->dev, "No interrupt support, no core IRQ\n");
 673                 return -EINVAL;
 674         }
 675
 676         rk808 = devm_kzalloc(&client->dev, sizeof(*rk808), GFP_KERNEL);
 677         if (!rk808)
 678                 return -ENOMEM;
 679
 680         /* read Chip variant */
 681         msb = i2c_smbus_read_byte_data(client, RK808_ID_MSB);
 682         if (msb < 0) {
 683                 dev_err(&client->dev, "failed to read the chip id at 0x%x\n",
 684                         RK808_ID_MSB);
 685                 return msb;
 686         }
 687
 688         lsb = i2c_smbus_read_byte_data(client, RK808_ID_LSB);
 689         if (lsb < 0) {
 690                 dev_err(&client->dev, "failed to read the chip id at 0x%x\n",
 691                         RK808_ID_LSB);
 692                 return lsb;
 693         }
 694
 695         rk808->variant = ((msb << 8) | lsb) & RK8XX_ID_MSK;
 696         dev_info(&client->dev, "Pmic Chip id: 0x%lx\n", rk808->variant);
 697
 698         /* set Chip platform init data*/
 699         switch (rk808->variant) {
 700         case RK818_ID:
 701                 cell = rk818s;
 702                 cell_num = ARRAY_SIZE(rk818s);
 703                 pre_init_reg = rk818_pre_init_reg;
 704                 reg_num = ARRAY_SIZE(rk818_pre_init_reg);
 705                 regmap_config = &rk818_regmap_config;
 706                 irq_chip = &rk818_irq_chip;
 707                 pm_shutdown_fn = rk818_shutdown;
 708                 on_source = RK818_ON_SOURCE_REG;
 709                 off_source = RK818_OFF_SOURCE_REG;
 710                 break;
 711         case RK816_ID:
 712                 cell = rk816s;
 713                 cell_num = ARRAY_SIZE(rk816s);
 714                 pre_init_reg = rk816_pre_init_reg;
 715                 reg_num = ARRAY_SIZE(rk816_pre_init_reg);
 716                 regmap_config = &rk816_regmap_config;
 717                 irq_chip = &rk816_irq_chip;
 718                 pm_shutdown_fn = rk816_shutdown;
 719                 on_source = RK816_ON_SOURCE_REG;
 720                 off_source = RK816_OFF_SOURCE_REG;
 721                 break;
 722         case RK808_ID:
 723                 cell = rk808s;
 724                 cell_num = ARRAY_SIZE(rk808s);
 725                 pre_init_reg = rk808_pre_init_reg;
 726                 reg_num = ARRAY_SIZE(rk808_pre_init_reg);
 727                 regmap_config = &rk808_regmap_config;
 728                 irq_chip = &rk808_irq_chip;
 729                 pm_shutdown_fn = rk808_shutdown;
 730                 break;
 731         case RK805_ID:
 732                 cell = rk805s;
 733                 cell_num = ARRAY_SIZE(rk805s);
 734                 pre_init_reg = rk805_pre_init_reg;
 735                 reg_num = ARRAY_SIZE(rk805_pre_init_reg);
 736                 regmap_config = &rk805_regmap_config;
 737                 irq_chip = &rk805_irq_chip;
 738                 pm_shutdown_prepare_fn = rk805_shutdown_prepare;
 739                 on_source = RK805_ON_SOURCE_REG;
 740                 off_source = RK805_OFF_SOURCE_REG;
 741                 break;
 742         default:
 743                 dev_err(&client->dev, "unsupported RK8XX ID 0x%lx\n",
 744                         rk808->variant);
 745                 return -EINVAL;
 746         }
 747
 748         rk808->regmap = devm_regmap_init_i2c(client, regmap_config);
 749         if (IS_ERR(rk808->regmap)) {
 750                 dev_err(&client->dev, "regmap initialization failed\n");
 751                 return PTR_ERR(rk808->regmap);
 752         }
 753
 754         /* on & off source */
 755         if (on_source && off_source) {
 756                 ret = regmap_read(rk808->regmap, on_source, &on);
 757                 if (ret) {
 758                         dev_err(&client->dev, "read 0x%x failed\n", on_source);
 759                         return ret;
 760                 }
 761
 762                 ret = regmap_read(rk808->regmap, off_source, &off);
 763                 if (ret) {
 764                         dev_err(&client->dev, "read 0x%x failed\n", off_source);
 765                         return ret;
 766                 }
 767
 768                 dev_info(&client->dev, "source: on=0x%02x, off=0x%02x\n",
 769                          on, off);
 770         }
 771
 772         for (i = 0; i < reg_num; i++) {
 773                 ret = regmap_update_bits(rk808->regmap,
 774                                          pre_init_reg[i].addr,
 775                                          pre_init_reg[i].mask,
 776                                          pre_init_reg[i].value);
 777                 if (ret) {
 778                         dev_err(&client->dev,
 779                                 "0x%x write err\n",
 780                                 pre_init_reg[i].addr);
 781                         return ret;
 782                 }
 783         }
 784
 785         ret = regmap_add_irq_chip(rk808->regmap, client->irq,
 786                                   IRQF_ONESHOT, -1,
 787                                   irq_chip, &rk808->irq_data);
 788         if (ret) {
 789                 dev_err(&client->dev, "Failed to add irq_chip %d\n", ret);
 790                 return ret;
 791         }
 792
 793         rk808->i2c = client;
 794         i2c_set_clientdata(client, rk808);
 795
 796         ret = mfd_add_devices(&client->dev, -1,
 797                               cell, cell_num,
 798                               NULL, 0, regmap_irq_get_domain(rk808->irq_data));
 799         if (ret) {
 800                 dev_err(&client->dev, "failed to add MFD devices %d\n", ret);
 801                 goto err_irq;
 802         }
 803
 804         pm_off = of_property_read_bool(np,
 805                                 "rockchip,system-power-controller");
 806         if (pm_off) {
 807                 rk808_i2c_client = client;
 808                 if (pm_shutdown_prepare_fn) {
 809                         pm_shutdown_prepare = pm_shutdown_prepare_fn;
 810                         pm_power_off_prepare = rk808_device_shutdown_prepare;
 811                 }
 812                 if (pm_shutdown_fn) {
 813                         pm_shutdown = pm_shutdown_fn;
 814                         pm_power_off = rk808_device_shutdown;
 815                 }
 816         }
 817
 818         rk8xx_kobj = kobject_create_and_add("rk8xx", NULL);
 819         if (rk8xx_kobj) {
 820                 ret = sysfs_create_file(rk8xx_kobj, &rk8xx_attrs.attr);
 821                 if (ret)
 822                         dev_err(&client->dev, "create rk8xx sysfs error\n");
 823         }
 824
 825         return 0;
 826
 827 err_irq:
 828         regmap_del_irq_chip(client->irq, rk808->irq_data);
 829         return ret;
 830 }
 831
 832 static int rk808_remove(struct i2c_client *client)
 833 {
 834         struct rk808 *rk808 = i2c_get_clientdata(client);
 835
 836         regmap_del_irq_chip(client->irq, rk808->irq_data);
 837         mfd_remove_devices(&client->dev);
 838         if (pm_power_off == rk808_device_shutdown)
 839                 pm_power_off = NULL;
 840         if (pm_power_off_prepare == rk808_device_shutdown_prepare)
 841                 pm_power_off_prepare = NULL;
 842
 843         return 0;
 844 }
 845
 846 static const struct i2c_device_id rk808_ids[] = {
 847         { "rk805" },
 848         { "rk808" },
 849         { "rk816" },
 850         { "rk818" },
 851         { },
 852 };
 853 MODULE_DEVICE_TABLE(i2c, rk808_ids);
 854
 855 static struct i2c_driver rk808_i2c_driver = {
 856         .driver = {
 857                 .name = "rk808",
 858                 .of_match_table = rk808_of_match,
 859         },
 860         .probe    = rk808_probe,
 861         .remove   = rk808_remove,
 862         .id_table = rk808_ids,
 863 };
 864
 865 module_i2c_driver(rk808_i2c_driver);
 866
 867 MODULE_LICENSE("GPL");
 868 MODULE_AUTHOR("Chris Zhong <zyw@rock-chips.com>");
 869 MODULE_AUTHOR("Zhang Qing <zhangqing@rock-chips.com>");
 870 MODULE_AUTHOR("Chen jianhong <chenjh@rock-chips.com>");
 871 MODULE_DESCRIPTION("RK808 PMIC driver");