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