arm64: dts: qcom: sm8550: add TRNG node
[linux-modified.git] / drivers / power / supply / bq2515x_charger.c
1 // SPDX-License-Identifier: GPL-2.0
2 // BQ2515X Battery Charger Driver
3 // Copyright (C) 2020 Texas Instruments Incorporated - https://www.ti.com/
4
5 #include <linux/err.h>
6 #include <linux/i2c.h>
7 #include <linux/init.h>
8 #include <linux/kernel.h>
9 #include <linux/module.h>
10 #include <linux/gpio/consumer.h>
11 #include <linux/power_supply.h>
12 #include <linux/regmap.h>
13 #include <linux/types.h>
14
15 #define BQ2515X_MANUFACTURER "Texas Instruments"
16
17 #define BQ2515X_STAT0           0x00
18 #define BQ2515X_STAT1           0x01
19 #define BQ2515X_STAT2           0x02
20 #define BQ2515X_FLAG0           0x03
21 #define BQ2515X_FLAG1           0x04
22 #define BQ2515X_FLAG2           0x05
23 #define BQ2515X_FLAG3           0x06
24 #define BQ2515X_MASK0           0x07
25 #define BQ2515X_MASK1           0x08
26 #define BQ2515X_MASK2           0x09
27 #define BQ2515X_MASK3           0x0a
28 #define BQ2515X_VBAT_CTRL       0x12
29 #define BQ2515X_ICHG_CTRL       0x13
30 #define BQ2515X_PCHRGCTRL       0x14
31 #define BQ2515X_TERMCTRL        0x15
32 #define BQ2515X_BUVLO           0x16
33 #define BQ2515X_CHARGERCTRL0    0x17
34 #define BQ2515X_CHARGERCTRL1    0x18
35 #define BQ2515X_ILIMCTRL        0x19
36 #define BQ2515X_LDOCTRL         0x1d
37 #define BQ2515X_MRCTRL          0x30
38 #define BQ2515X_ICCTRL0         0x35
39 #define BQ2515X_ICCTRL1         0x36
40 #define BQ2515X_ICCTRL2         0x37
41 #define BQ2515X_ADCCTRL0        0x40
42 #define BQ2515X_ADCCTRL1        0x41
43 #define BQ2515X_ADC_VBAT_M      0x42
44 #define BQ2515X_ADC_VBAT_L      0x43
45 #define BQ2515X_ADC_TS_M        0x44
46 #define BQ2515X_ADC_TS_L        0x45
47 #define BQ2515X_ADC_ICHG_M      0x46
48 #define BQ2515X_ADC_ICHG_L      0x47
49 #define BQ2515X_ADC_ADCIN_M     0x48
50 #define BQ2515X_ADC_ADCIN_L     0x49
51 #define BQ2515X_ADC_VIN_M       0x4a
52 #define BQ2515X_ADC_VIN_L       0x4b
53 #define BQ2515X_ADC_PMID_M      0x4c
54 #define BQ2515X_ADC_PMID_L      0x4d
55 #define BQ2515X_ADC_IIN_M       0x4e
56 #define BQ2515X_ADC_IIN_L       0x4f
57 #define BQ2515X_ADC_COMP1_M     0x52
58 #define BQ2515X_ADC_COMP1_L     0X53
59 #define BQ2515X_ADC_COMP2_M     0X54
60 #define BQ2515X_ADC_COMP2_L     0x55
61 #define BQ2515X_ADC_COMP3_M     0x56
62 #define BQ2515X_ADC_COMP3_L     0x57
63 #define BQ2515X_ADC_READ_EN     0x58
64 #define BQ2515X_TS_FASTCHGCTRL  0x61
65 #define BQ2515X_TS_COLD         0x62
66 #define BQ2515X_TS_COOL         0x63
67 #define BQ2515X_TS_WARM         0x64
68 #define BQ2515X_TS_HOT          0x65
69 #define BQ2515X_DEVICE_ID       0x6f
70
71 #define BQ2515X_DEFAULT_ICHG_UA         10000
72 #define BQ25150_DEFAULT_ILIM_UA         100000
73 #define BQ25155_DEFAULT_ILIM_UA         500000
74 #define BQ2515X_DEFAULT_VBAT_REG_UV     4200000
75 #define BQ2515X_DEFAULT_IPRECHARGE_UA   2500
76
77 #define BQ2515X_DIVISOR                         65536
78 #define BQ2515X_VBAT_BASE_VOLT                  3600000
79 #define BQ2515X_VBAT_REG_MAX                    4600000
80 #define BQ2515X_VBAT_REG_MIN                    3600000
81 #define BQ2515X_VBAT_STEP_UV                    10000
82 #define BQ2515X_UV_FACTOR                       1000000
83 #define BQ2515X_VBAT_MULTIPLIER                 6
84 #define BQ2515X_ICHG_DIVISOR                    52429
85 #define BQ2515X_ICHG_CURR_STEP_THRESH_UA        318750
86 #define BQ2515X_ICHG_MIN_UA                     0
87 #define BQ2515X_ICHG_MAX_UA                     500000
88 #define BQ2515X_ICHG_RNG_1B0_UA                 1250
89 #define BQ2515X_ICHG_RNG_1B1_UA                 2500
90 #define BQ2515X_VLOWV_SEL_1B0_UV                3000000
91 #define BQ2515X_VLOWV_SEL_1B1_UV                2800000
92 #define BQ2515X_PRECHRG_ICHRG_RNGE_1875_UA      18750
93 #define BQ2515X_PRECHRG_ICHRG_RNGE_3750_UA      37500
94 #define BQ2515X_TWAKE2_MIN_US                   1700000
95 #define BQ2515X_TWAKE2_MAX_US                   2300000
96
97 #define BQ2515X_ILIM_150MA      0x2
98 #define BQ2515X_ILIM_MASK       0x7
99 #define BQ2515X_ILIM_MIN        50000
100 #define BQ2515X_ILIM_MAX        600000
101 #define BQ2515X_HEALTH_MASK     0xf
102 #define BQ2515X_ICHGRNG_MASK    0x80
103 #define BQ2515X_STAT0_MASK      0x0f
104 #define BQ2515X_STAT1_MASK      0x1f
105 #define BQ2515X_PRECHARGE_MASK  0x1f
106
107 #define BQ2515X_TS_HOT_STAT             BIT(0)
108 #define BQ2515X_TS_WARM_STAT            BIT(1)
109 #define BQ2515X_TS_COOL_STAT            BIT(2)
110 #define BQ2515X_TS_COLD_STAT            BIT(3)
111 #define BQ2515X_SAFETY_TIMER_EXP        BIT(5)
112
113 #define BQ2515X_EN_VBAT_READ            BIT(3)
114 #define BQ2515X_EN_ICHG_READ            BIT(5)
115
116 #define BQ2515X_VIN_GOOD                BIT(0)
117 #define BQ2515X_CHRG_DONE               BIT(5)
118 #define BQ2515X_CV_CHRG_MODE            BIT(6)
119
120 #define BQ2515X_VIN_OVP_FAULT_STAT      BIT(7)
121
122 #define BQ2515X_WATCHDOG_DISABLE        BIT(4)
123
124 #define BQ2515X_ICHARGE_RANGE           BIT(7)
125
126 #define BQ2515X_VLOWV_SEL               BIT(5)
127
128 #define BQ2515X_CHARGER_DISABLE         BIT(0)
129
130 #define BQ2515X_HWRESET_14S_WD          BIT(1)
131
132 static const int bq2515x_ilim_lvl_values[] = {
133         50000, 100000, 150000, 200000, 300000, 400000, 500000, 600000
134 };
135
136 /**
137  * struct bq2515x_init_data -
138  * @ilim: input current limit
139  * @ichg: fast charge current
140  * @vbatreg: battery regulation voltage
141  * @iprechg: precharge current
142  */
143 struct bq2515x_init_data {
144         int ilim;
145         int ichg;
146         int vbatreg;
147         int iprechg;
148 };
149
150 /**
151  * struct bq2515x_info -
152  * @regmap_config: register map config
153  * @ilim: input current limit
154  */
155 struct bq2515x_info {
156         const struct regmap_config *regmap_config;
157         int ilim;
158 };
159
160 /**
161  * struct bq2515x_device -
162  * @mains: mains properties
163  * @battery: battery properties
164  * @regmap: register map structure
165  * @dev: device structure
166  *
167  * @reset_gpio: manual reset (MR) pin
168  * @powerdown_gpio: low power mode pin
169  * @ac_detect_gpio: power good (PG) pin
170  * @ce_gpio: charge enable (CE) pin
171  *
172  * @info: device info
173  * @model_name: string value describing device model
174  * @mains_online: boolean value indicating power supply online
175  *
176  * @init_data: charger initialization data structure
177  */
178 struct bq2515x_device {
179         struct power_supply *mains;
180         struct power_supply *battery;
181         struct regmap *regmap;
182         struct device *dev;
183
184         struct gpio_desc *reset_gpio;
185         struct gpio_desc *powerdown_gpio;
186         struct gpio_desc *ac_detect_gpio;
187         struct gpio_desc *ce_gpio;
188
189         const struct bq2515x_info *info;
190         char model_name[I2C_NAME_SIZE];
191         bool mains_online;
192
193         struct bq2515x_init_data init_data;
194 };
195
196 static const struct reg_default bq25150_reg_defaults[] = {
197         {BQ2515X_FLAG0, 0x0},
198         {BQ2515X_FLAG1, 0x0},
199         {BQ2515X_FLAG2, 0x0},
200         {BQ2515X_FLAG3, 0x0},
201         {BQ2515X_MASK0, 0x0},
202         {BQ2515X_MASK1, 0x0},
203         {BQ2515X_MASK2, 0x71},
204         {BQ2515X_MASK3, 0x0},
205         {BQ2515X_VBAT_CTRL, 0x3C},
206         {BQ2515X_ICHG_CTRL, 0x8},
207         {BQ2515X_PCHRGCTRL, 0x2},
208         {BQ2515X_TERMCTRL, 0x14},
209         {BQ2515X_BUVLO, 0x0},
210         {BQ2515X_CHARGERCTRL0, 0x82},
211         {BQ2515X_CHARGERCTRL1, 0x42},
212         {BQ2515X_ILIMCTRL, 0x1},
213         {BQ2515X_LDOCTRL, 0xB0},
214         {BQ2515X_MRCTRL, 0x2A},
215         {BQ2515X_ICCTRL0, 0x10},
216         {BQ2515X_ICCTRL1, 0x0},
217         {BQ2515X_ICCTRL2, 0x0},
218         {BQ2515X_ADCCTRL0, 0x2},
219         {BQ2515X_ADCCTRL1, 0x40},
220         {BQ2515X_ADC_COMP1_M, 0x23},
221         {BQ2515X_ADC_COMP1_L, 0x20},
222         {BQ2515X_ADC_COMP2_M, 0x38},
223         {BQ2515X_ADC_COMP2_L, 0x90},
224         {BQ2515X_ADC_COMP3_M, 0x0},
225         {BQ2515X_ADC_COMP3_L, 0x0},
226         {BQ2515X_ADC_READ_EN, 0x0},
227         {BQ2515X_TS_FASTCHGCTRL, 0x34},
228         {BQ2515X_TS_COLD, 0x7C},
229         {BQ2515X_TS_COOL, 0x6D},
230         {BQ2515X_TS_WARM, 0x38},
231         {BQ2515X_TS_HOT, 0x27},
232         {BQ2515X_DEVICE_ID, 0x20},
233 };
234
235 static const struct reg_default bq25155_reg_defaults[] = {
236         {BQ2515X_FLAG0, 0x0},
237         {BQ2515X_FLAG1, 0x0},
238         {BQ2515X_FLAG2, 0x0},
239         {BQ2515X_FLAG3, 0x0},
240         {BQ2515X_MASK0, 0x0},
241         {BQ2515X_MASK1, 0x0},
242         {BQ2515X_MASK2, 0x71},
243         {BQ2515X_MASK3, 0x0},
244         {BQ2515X_VBAT_CTRL, 0x3C},
245         {BQ2515X_ICHG_CTRL, 0x8},
246         {BQ2515X_PCHRGCTRL, 0x2},
247         {BQ2515X_TERMCTRL, 0x14},
248         {BQ2515X_BUVLO, 0x0},
249         {BQ2515X_CHARGERCTRL0, 0x82},
250         {BQ2515X_CHARGERCTRL1, 0xC2},
251         {BQ2515X_ILIMCTRL, 0x6},
252         {BQ2515X_LDOCTRL, 0xB0},
253         {BQ2515X_MRCTRL, 0x2A},
254         {BQ2515X_ICCTRL0, 0x10},
255         {BQ2515X_ICCTRL1, 0x0},
256         {BQ2515X_ICCTRL2, 0x40},
257         {BQ2515X_ADCCTRL0, 0x2},
258         {BQ2515X_ADCCTRL1, 0x40},
259         {BQ2515X_ADC_COMP1_M, 0x23},
260         {BQ2515X_ADC_COMP1_L, 0x20},
261         {BQ2515X_ADC_COMP2_M, 0x38},
262         {BQ2515X_ADC_COMP2_L, 0x90},
263         {BQ2515X_ADC_COMP3_M, 0x0},
264         {BQ2515X_ADC_COMP3_L, 0x0},
265         {BQ2515X_ADC_READ_EN, 0x0},
266         {BQ2515X_TS_FASTCHGCTRL, 0x34},
267         {BQ2515X_TS_COLD, 0x7C},
268         {BQ2515X_TS_COOL, 0x6D},
269         {BQ2515X_TS_WARM, 0x38},
270         {BQ2515X_TS_HOT, 0x27},
271         {BQ2515X_DEVICE_ID, 0x35},
272 };
273
274 static int bq2515x_wake_up(struct bq2515x_device *bq2515x)
275 {
276         int ret;
277         int val;
278
279         /* Read the STAT register if we can read it then the device is out
280          * of ship mode.  If the register cannot be read then attempt to wake
281          * it up and enable the ADC.
282          */
283         ret = regmap_read(bq2515x->regmap, BQ2515X_STAT0, &val);
284         if (ret)
285                 return ret;
286
287         /* Need to toggle LP and bring device out of ship mode. The device
288          * will exit the ship mode when the MR pin is held low for at least
289          * t_WAKE2 as shown in section 8.3.7.1 of the datasheet.
290          */
291         gpiod_set_value_cansleep(bq2515x->powerdown_gpio, 0);
292
293         gpiod_set_value_cansleep(bq2515x->reset_gpio, 0);
294         usleep_range(BQ2515X_TWAKE2_MIN_US, BQ2515X_TWAKE2_MAX_US);
295         gpiod_set_value_cansleep(bq2515x->reset_gpio, 1);
296
297         return regmap_write(bq2515x->regmap, BQ2515X_ADC_READ_EN,
298                                 (BQ2515X_EN_VBAT_READ | BQ2515X_EN_ICHG_READ));
299 }
300
301 static int bq2515x_update_ps_status(struct bq2515x_device *bq2515x)
302 {
303         bool dc = false;
304         unsigned int val;
305         int ret;
306
307         if (bq2515x->ac_detect_gpio)
308                 val = gpiod_get_value_cansleep(bq2515x->ac_detect_gpio);
309         else {
310                 ret = regmap_read(bq2515x->regmap, BQ2515X_STAT0, &val);
311                 if (ret)
312                         return ret;
313         }
314
315         dc = val & BQ2515X_VIN_GOOD;
316
317         ret = bq2515x->mains_online != dc;
318
319         bq2515x->mains_online = dc;
320
321         return ret;
322 }
323
324 static int bq2515x_disable_watchdog_timers(struct bq2515x_device *bq2515x)
325 {
326         int ret;
327
328         ret = regmap_update_bits(bq2515x->regmap, BQ2515X_CHARGERCTRL0,
329                         BQ2515X_WATCHDOG_DISABLE, BQ2515X_WATCHDOG_DISABLE);
330         if (ret)
331                 return ret;
332
333         return regmap_update_bits(bq2515x->regmap, BQ2515X_ICCTRL2,
334                                                 BQ2515X_HWRESET_14S_WD, 0);
335 }
336
337 static int bq2515x_get_battery_voltage_now(struct bq2515x_device *bq2515x)
338 {
339         int ret;
340         int vbat_msb;
341         int vbat_lsb;
342         uint32_t vbat_measurement;
343
344         if (!bq2515x->mains_online)
345                 bq2515x_wake_up(bq2515x);
346
347         ret = regmap_read(bq2515x->regmap, BQ2515X_ADC_VBAT_M, &vbat_msb);
348         if (ret)
349                 return ret;
350
351         ret = regmap_read(bq2515x->regmap, BQ2515X_ADC_VBAT_L, &vbat_lsb);
352         if (ret)
353                 return ret;
354
355         vbat_measurement = (vbat_msb << 8) | vbat_lsb;
356
357         return vbat_measurement * (BQ2515X_UV_FACTOR / BQ2515X_DIVISOR) *
358                                                 BQ2515X_VBAT_MULTIPLIER;
359 }
360
361 static int bq2515x_get_battery_current_now(struct bq2515x_device *bq2515x)
362 {
363         int ret;
364         int ichg_msb;
365         int ichg_lsb;
366         uint32_t ichg_measurement;
367         u16 ichg_multiplier = BQ2515X_ICHG_RNG_1B0_UA;
368         unsigned int ichg_reg_code, reg_code;
369         unsigned int icharge_range = 0, pchrgctrl;
370         unsigned int buvlo, vlowv_sel, vlowv = BQ2515X_VLOWV_SEL_1B0_UV;
371
372         if (!bq2515x->mains_online)
373                 return -ENODATA;
374
375         ret = regmap_read(bq2515x->regmap, BQ2515X_ADC_ICHG_M, &ichg_msb);
376         if (ret)
377                 return ret;
378
379         ret = regmap_read(bq2515x->regmap, BQ2515X_ADC_ICHG_L, &ichg_lsb);
380         if (ret)
381                 return ret;
382
383         ichg_measurement = (ichg_msb << 8) | ichg_lsb;
384
385         ret = regmap_read(bq2515x->regmap, BQ2515X_BUVLO, &buvlo);
386         if (ret)
387                 return ret;
388
389         vlowv_sel = buvlo & BQ2515X_VLOWV_SEL;
390
391         if (vlowv_sel)
392                 vlowv = BQ2515X_VLOWV_SEL_1B1_UV;
393
394         if (bq2515x_get_battery_voltage_now(bq2515x) < vlowv) {
395                 ret = regmap_read(bq2515x->regmap, BQ2515X_PCHRGCTRL,
396                                                                 &pchrgctrl);
397                 if (ret)
398                         return ret;
399
400                 reg_code = pchrgctrl & BQ2515X_PRECHARGE_MASK;
401         } else {
402                 ret = regmap_read(bq2515x->regmap, BQ2515X_ICHG_CTRL,
403                                                         &ichg_reg_code);
404                 if (ret)
405                         return ret;
406
407                 reg_code = ichg_reg_code;
408         }
409
410         ret = regmap_read(bq2515x->regmap, BQ2515X_PCHRGCTRL, &pchrgctrl);
411         if (ret)
412                 return ret;
413
414         icharge_range = pchrgctrl & BQ2515X_ICHARGE_RANGE;
415
416         if (icharge_range)
417                 ichg_multiplier = BQ2515X_ICHG_RNG_1B1_UA;
418
419         return reg_code * (ichg_multiplier * ichg_measurement /
420                                                         BQ2515X_ICHG_DIVISOR);
421 }
422
423 static bool bq2515x_get_charge_disable(struct bq2515x_device *bq2515x)
424 {
425         int ret;
426         int ce_pin;
427         int icctrl2;
428         int charger_disable;
429
430         ce_pin = gpiod_get_value_cansleep(bq2515x->ce_gpio);
431
432         ret = regmap_read(bq2515x->regmap, BQ2515X_ICCTRL2, &icctrl2);
433         if (ret)
434                 return ret;
435
436         charger_disable = icctrl2 & BQ2515X_CHARGER_DISABLE;
437
438         if (charger_disable || ce_pin)
439                 return true;
440
441         return false;
442 }
443
444 static int bq2515x_set_charge_disable(struct bq2515x_device *bq2515x, int val)
445 {
446         gpiod_set_value_cansleep(bq2515x->ce_gpio, val);
447
448         return regmap_update_bits(bq2515x->regmap, BQ2515X_ICCTRL2,
449                                         BQ2515X_CHARGER_DISABLE, val);
450 }
451
452 static int bq2515x_get_const_charge_current(struct bq2515x_device *bq2515x)
453 {
454         int ret;
455         u16 ichg_multiplier = BQ2515X_ICHG_RNG_1B0_UA;
456         unsigned int ichg_reg_code;
457         unsigned int pchrgctrl;
458         unsigned int icharge_range;
459
460         ret = regmap_read(bq2515x->regmap, BQ2515X_ICHG_CTRL, &ichg_reg_code);
461         if (ret)
462                 return ret;
463
464         ret = regmap_read(bq2515x->regmap, BQ2515X_PCHRGCTRL, &pchrgctrl);
465         if (ret)
466                 return ret;
467
468         icharge_range = pchrgctrl & BQ2515X_ICHARGE_RANGE;
469
470         if (icharge_range)
471                 ichg_multiplier = BQ2515X_ICHG_RNG_1B1_UA;
472
473         return ichg_reg_code * ichg_multiplier;
474 }
475
476 static int bq2515x_set_const_charge_current(struct bq2515x_device *bq2515x,
477                                                                 int val)
478 {
479         int ret;
480         unsigned int ichg_reg_code;
481         u16 ichg_multiplier = BQ2515X_ICHG_RNG_1B0_UA;
482         unsigned int icharge_range = 0;
483
484         if (val > BQ2515X_ICHG_MAX_UA || val < BQ2515X_ICHG_MIN_UA)
485                 return -EINVAL;
486
487         if (val > BQ2515X_ICHG_CURR_STEP_THRESH_UA) {
488                 ichg_multiplier = BQ2515X_ICHG_RNG_1B1_UA;
489                 icharge_range = BQ2515X_ICHARGE_RANGE;
490         }
491
492         bq2515x_set_charge_disable(bq2515x, 1);
493
494         ret = regmap_update_bits(bq2515x->regmap, BQ2515X_PCHRGCTRL,
495                                         BQ2515X_ICHARGE_RANGE, icharge_range);
496         if (ret)
497                 return ret;
498
499         ichg_reg_code = val / ichg_multiplier;
500
501         ret = regmap_write(bq2515x->regmap, BQ2515X_ICHG_CTRL, ichg_reg_code);
502         if (ret)
503                 return ret;
504
505         return bq2515x_set_charge_disable(bq2515x, 0);
506 }
507
508 static int bq2515x_get_precharge_current(struct bq2515x_device *bq2515x)
509 {
510         int ret;
511         unsigned int pchrgctrl;
512         unsigned int icharge_range;
513         u16 precharge_multiplier = BQ2515X_ICHG_RNG_1B0_UA;
514         unsigned int precharge_reg_code;
515
516         ret = regmap_read(bq2515x->regmap, BQ2515X_PCHRGCTRL, &pchrgctrl);
517         if (ret)
518                 return ret;
519
520         icharge_range = pchrgctrl & BQ2515X_ICHARGE_RANGE;
521
522         if (icharge_range)
523                 precharge_multiplier = BQ2515X_ICHG_RNG_1B1_UA;
524
525         precharge_reg_code = pchrgctrl & BQ2515X_PRECHARGE_MASK;
526
527         return precharge_reg_code * precharge_multiplier;
528 }
529
530 static int bq2515x_set_precharge_current(struct bq2515x_device *bq2515x,
531                                         int val)
532 {
533         int ret;
534         unsigned int pchrgctrl;
535         unsigned int icharge_range;
536         unsigned int precharge_reg_code;
537         unsigned int precharge_multiplier = BQ2515X_ICHG_RNG_1B0_UA;
538         unsigned int precharge_max_ua = BQ2515X_PRECHRG_ICHRG_RNGE_1875_UA;
539
540         ret = regmap_read(bq2515x->regmap, BQ2515X_PCHRGCTRL, &pchrgctrl);
541         if (ret)
542                 return ret;
543
544         icharge_range = pchrgctrl & BQ2515X_ICHARGE_RANGE;
545
546         if (icharge_range) {
547                 precharge_max_ua = BQ2515X_PRECHRG_ICHRG_RNGE_3750_UA;
548                 precharge_multiplier = BQ2515X_ICHG_RNG_1B1_UA;
549         } else {
550                 precharge_max_ua = BQ2515X_PRECHRG_ICHRG_RNGE_1875_UA;
551                 precharge_multiplier = BQ2515X_ICHG_RNG_1B0_UA;
552         }
553         if (val > precharge_max_ua || val < BQ2515X_ICHG_MIN_UA)
554                 return -EINVAL;
555
556         precharge_reg_code = val / precharge_multiplier;
557
558         ret = bq2515x_set_charge_disable(bq2515x, 1);
559         if (ret)
560                 return ret;
561
562         ret = regmap_update_bits(bq2515x->regmap, BQ2515X_PCHRGCTRL,
563                                 BQ2515X_PRECHARGE_MASK, precharge_reg_code);
564         if (ret)
565                 return ret;
566
567         return bq2515x_set_charge_disable(bq2515x, 0);
568 }
569
570 static int bq2515x_charging_status(struct bq2515x_device *bq2515x,
571                                    union power_supply_propval *val)
572 {
573         bool status0_no_fault;
574         bool status1_no_fault;
575         bool ce_status;
576         bool charge_done;
577         unsigned int status;
578         int ret;
579
580         if (!bq2515x->mains_online) {
581                 val->intval = POWER_SUPPLY_STATUS_DISCHARGING;
582                 return 0;
583         }
584
585         ret = regmap_read(bq2515x->regmap, BQ2515X_STAT0, &status);
586         if (ret)
587                 return ret;
588
589         /*
590          * The code block below is used to determine if any faults from the
591          * STAT0 register are disbaling charging or if the charge has completed
592          * according to the CHARGE_DONE_STAT bit.
593          */
594         if (((status & BQ2515X_STAT0_MASK) == true) &
595                         ((status & BQ2515X_CHRG_DONE) == false)) {
596                 status0_no_fault = true;
597                 charge_done = false;
598         } else if (status & BQ2515X_CHRG_DONE) {
599                 charge_done = true;
600                 status0_no_fault = false;
601         } else {
602                 status0_no_fault = false;
603                 charge_done = false;
604         }
605
606         ret = regmap_read(bq2515x->regmap, BQ2515X_STAT1, &status);
607         if (ret)
608                 return ret;
609         /*
610          * The code block below is used to determine if any faults from the
611          * STAT1 register are disbaling charging
612          */
613         if ((status & BQ2515X_STAT1_MASK) == false)
614                 status1_no_fault = true;
615         else
616                 status1_no_fault = false;
617
618         ce_status = (!bq2515x_get_charge_disable(bq2515x));
619
620         /*
621          * If there are no faults and charging is enabled, then status is
622          * charging. Otherwise, if charging is complete, then status is full.
623          * Otherwise, if a fault exists or charging is disabled, then status is
624          * not charging
625          */
626         if (status0_no_fault & status1_no_fault & ce_status)
627                 val->intval = POWER_SUPPLY_STATUS_CHARGING;
628         else if (charge_done)
629                 val->intval = POWER_SUPPLY_STATUS_FULL;
630         else if (!(status0_no_fault & status1_no_fault & ce_status))
631                 val->intval = POWER_SUPPLY_STATUS_NOT_CHARGING;
632
633         return 0;
634 }
635
636 static int bq2515x_get_batt_reg(struct bq2515x_device *bq2515x)
637 {
638         int vbat_reg_code;
639         int ret;
640
641         ret = regmap_read(bq2515x->regmap, BQ2515X_VBAT_CTRL, &vbat_reg_code);
642         if (ret)
643                 return ret;
644
645         return BQ2515X_VBAT_BASE_VOLT + vbat_reg_code * BQ2515X_VBAT_STEP_UV;
646 }
647
648 static int bq2515x_set_batt_reg(struct bq2515x_device *bq2515x, int val)
649 {
650         int vbat_reg_code;
651
652         if (val > BQ2515X_VBAT_REG_MAX || val < BQ2515X_VBAT_REG_MIN)
653                 return -EINVAL;
654
655         vbat_reg_code = (val - BQ2515X_VBAT_BASE_VOLT) / BQ2515X_VBAT_STEP_UV;
656
657         return regmap_write(bq2515x->regmap, BQ2515X_VBAT_CTRL, vbat_reg_code);
658 }
659
660 static int bq2515x_get_ilim_lvl(struct bq2515x_device *bq2515x)
661 {
662         int ret;
663         int ilimctrl;
664
665         ret = regmap_read(bq2515x->regmap, BQ2515X_ILIMCTRL, &ilimctrl);
666         if (ret)
667                 return ret;
668
669         return bq2515x_ilim_lvl_values[ilimctrl & BQ2515X_ILIM_MASK];
670 }
671
672 static int bq2515x_set_ilim_lvl(struct bq2515x_device *bq2515x, int val)
673 {
674         int i = 0;
675         unsigned int array_size = ARRAY_SIZE(bq2515x_ilim_lvl_values);
676
677         for (i = array_size - 1; i > 0; i--) {
678                 if (val >= bq2515x_ilim_lvl_values[i])
679                         break;
680         }
681         return regmap_write(bq2515x->regmap, BQ2515X_ILIMCTRL, i);
682 }
683
684 static int bq2515x_power_supply_property_is_writeable(struct power_supply *psy,
685                                         enum power_supply_property prop)
686 {
687         switch (prop) {
688         case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
689         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE:
690         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT:
691         case POWER_SUPPLY_PROP_PRECHARGE_CURRENT:
692                 return true;
693         default:
694                 return false;
695         }
696 }
697
698 static int bq2515x_charger_get_health(struct bq2515x_device *bq2515x,
699                                       union power_supply_propval *val)
700 {
701         int health = POWER_SUPPLY_HEALTH_GOOD;
702         int ret;
703         unsigned int stat1;
704         unsigned int flag3;
705
706         if (!bq2515x->mains_online)
707                 bq2515x_wake_up(bq2515x);
708
709         ret = regmap_read(bq2515x->regmap, BQ2515X_FLAG3, &flag3);
710         if (ret)
711                 return ret;
712
713         ret = regmap_read(bq2515x->regmap, BQ2515X_STAT1, &stat1);
714         if (ret)
715                 return ret;
716
717         if (stat1 & BQ2515X_HEALTH_MASK) {
718                 switch (stat1 & BQ2515X_HEALTH_MASK) {
719                 case BQ2515X_TS_HOT_STAT:
720                         health = POWER_SUPPLY_HEALTH_HOT;
721                         break;
722                 case BQ2515X_TS_WARM_STAT:
723                         health = POWER_SUPPLY_HEALTH_WARM;
724                         break;
725                 case BQ2515X_TS_COOL_STAT:
726                         health = POWER_SUPPLY_HEALTH_COOL;
727                         break;
728                 case BQ2515X_TS_COLD_STAT:
729                         health = POWER_SUPPLY_HEALTH_COLD;
730                         break;
731                 default:
732                         health = POWER_SUPPLY_HEALTH_UNKNOWN;
733                         break;
734                 }
735         }
736
737         if (stat1 & BQ2515X_VIN_OVP_FAULT_STAT)
738                 health = POWER_SUPPLY_HEALTH_OVERVOLTAGE;
739
740         if (flag3 & BQ2515X_SAFETY_TIMER_EXP)
741                 health = POWER_SUPPLY_HEALTH_SAFETY_TIMER_EXPIRE;
742
743         val->intval = health;
744         return 0;
745 }
746
747 static int bq2515x_mains_set_property(struct power_supply *psy,
748                 enum power_supply_property prop,
749                 const union power_supply_propval *val)
750 {
751         struct bq2515x_device *bq2515x = power_supply_get_drvdata(psy);
752         int ret;
753
754         switch (prop) {
755         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE:
756                 ret = bq2515x_set_batt_reg(bq2515x, val->intval);
757                 break;
758
759         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT:
760                 ret = bq2515x_set_const_charge_current(bq2515x, val->intval);
761                 break;
762
763         case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
764                 ret = bq2515x_set_ilim_lvl(bq2515x, val->intval);
765                 break;
766
767         case POWER_SUPPLY_PROP_PRECHARGE_CURRENT:
768                 ret = bq2515x_set_precharge_current(bq2515x, val->intval);
769                 break;
770
771         default:
772                 return -EINVAL;
773         }
774
775         return ret;
776 }
777
778 static int bq2515x_mains_get_property(struct power_supply *psy,
779                                      enum power_supply_property prop,
780                                      union power_supply_propval *val)
781 {
782         struct bq2515x_device *bq2515x = power_supply_get_drvdata(psy);
783         int ret = 0;
784
785         switch (prop) {
786
787         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT:
788                 ret = bq2515x_get_const_charge_current(bq2515x);
789                 if (ret < 0)
790                         return ret;
791
792                 val->intval = ret;
793                 break;
794
795         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE:
796                 ret = bq2515x_get_batt_reg(bq2515x);
797                 if (ret < 0)
798                         return ret;
799                 val->intval = ret;
800                 break;
801
802         case POWER_SUPPLY_PROP_PRECHARGE_CURRENT:
803                 ret = bq2515x_get_precharge_current(bq2515x);
804                 if (ret < 0)
805                         return ret;
806                 val->intval = ret;
807                 break;
808
809         case POWER_SUPPLY_PROP_ONLINE:
810                 val->intval = bq2515x->mains_online;
811                 break;
812
813         case POWER_SUPPLY_PROP_HEALTH:
814                 ret = bq2515x_charger_get_health(bq2515x, val);
815                 if (ret)
816                         val->intval = POWER_SUPPLY_HEALTH_UNKNOWN;
817                 break;
818
819         case POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT:
820                 ret = bq2515x_get_ilim_lvl(bq2515x);
821                 if (ret < 0)
822                         return ret;
823                 val->intval = ret;
824                 break;
825
826         case POWER_SUPPLY_PROP_MODEL_NAME:
827                 val->strval = bq2515x->model_name;
828                 break;
829
830         case POWER_SUPPLY_PROP_MANUFACTURER:
831                 val->strval = BQ2515X_MANUFACTURER;
832                 break;
833
834         case POWER_SUPPLY_PROP_STATUS:
835                 ret = bq2515x_charging_status(bq2515x, val);
836                 if (ret)
837                         return ret;
838                 break;
839
840         default:
841                 return -EINVAL;
842         }
843
844         return ret;
845 }
846
847 static int bq2515x_battery_get_property(struct power_supply *psy,
848                                        enum power_supply_property prop,
849                                        union power_supply_propval *val)
850 {
851         struct bq2515x_device *bq2515x = power_supply_get_drvdata(psy);
852         int ret;
853
854         ret = bq2515x_update_ps_status(bq2515x);
855         if (ret)
856                 return ret;
857
858         switch (prop) {
859
860         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE_MAX:
861                 ret = bq2515x->init_data.vbatreg;
862                 if (ret < 0)
863                         return ret;
864                 val->intval = ret;
865                 break;
866
867         case POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT_MAX:
868                 ret = bq2515x->init_data.ichg;
869                 if (ret < 0)
870                         return ret;
871                 val->intval = ret;
872                 break;
873
874         case POWER_SUPPLY_PROP_VOLTAGE_NOW:
875                 ret = bq2515x_get_battery_voltage_now(bq2515x);
876                 if (ret < 0)
877                         return ret;
878                 val->intval = ret;
879                 break;
880
881         case POWER_SUPPLY_PROP_CURRENT_NOW:
882                 ret = bq2515x_get_battery_current_now(bq2515x);
883                 if (ret < 0)
884                         return ret;
885                 val->intval = ret;
886                 break;
887
888         default:
889                 return -EINVAL;
890         }
891         return 0;
892 }
893
894 static const enum power_supply_property bq2515x_battery_properties[] = {
895         POWER_SUPPLY_PROP_VOLTAGE_NOW,
896         POWER_SUPPLY_PROP_CURRENT_NOW,
897         POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT_MAX,
898         POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE_MAX,
899 };
900
901 static const enum power_supply_property bq2515x_mains_properties[] = {
902         POWER_SUPPLY_PROP_ONLINE,
903         POWER_SUPPLY_PROP_STATUS,
904         POWER_SUPPLY_PROP_HEALTH,
905         POWER_SUPPLY_PROP_INPUT_CURRENT_LIMIT,
906         POWER_SUPPLY_PROP_MODEL_NAME,
907         POWER_SUPPLY_PROP_MANUFACTURER,
908         POWER_SUPPLY_PROP_CONSTANT_CHARGE_VOLTAGE,
909         POWER_SUPPLY_PROP_CONSTANT_CHARGE_CURRENT,
910         POWER_SUPPLY_PROP_PRECHARGE_CURRENT,
911 };
912
913 static const struct power_supply_desc bq2515x_mains_desc = {
914         .name                   = "bq2515x-mains",
915         .type                   = POWER_SUPPLY_TYPE_MAINS,
916         .get_property           = bq2515x_mains_get_property,
917         .set_property           = bq2515x_mains_set_property,
918         .properties             = bq2515x_mains_properties,
919         .num_properties         = ARRAY_SIZE(bq2515x_mains_properties),
920         .property_is_writeable  = bq2515x_power_supply_property_is_writeable,
921 };
922
923 static const struct power_supply_desc bq2515x_battery_desc = {
924         .name                   = "bq2515x-battery",
925         .type                   = POWER_SUPPLY_TYPE_BATTERY,
926         .get_property           = bq2515x_battery_get_property,
927         .properties             = bq2515x_battery_properties,
928         .num_properties         = ARRAY_SIZE(bq2515x_battery_properties),
929         .property_is_writeable  = bq2515x_power_supply_property_is_writeable,
930 };
931
932 static int bq2515x_power_supply_register(struct bq2515x_device *bq2515x,
933                 struct device *dev, struct power_supply_config psy_cfg)
934 {
935         bq2515x->mains = devm_power_supply_register(bq2515x->dev,
936                                                     &bq2515x_mains_desc,
937                                                     &psy_cfg);
938         if (IS_ERR(bq2515x->mains))
939                 return -EINVAL;
940
941         bq2515x->battery = devm_power_supply_register(bq2515x->dev,
942                                                       &bq2515x_battery_desc,
943                                                       &psy_cfg);
944         if (IS_ERR(bq2515x->battery))
945                 return -EINVAL;
946
947         return 0;
948 }
949
950 static int bq2515x_hw_init(struct bq2515x_device *bq2515x)
951 {
952         int ret;
953         struct power_supply_battery_info *bat_info;
954
955         ret = bq2515x_disable_watchdog_timers(bq2515x);
956         if (ret)
957                 return ret;
958
959         if (bq2515x->init_data.ilim) {
960                 ret = bq2515x_set_ilim_lvl(bq2515x, bq2515x->init_data.ilim);
961                 if (ret)
962                         return ret;
963         }
964
965         ret = power_supply_get_battery_info(bq2515x->mains, &bat_info);
966         if (ret) {
967                 dev_warn(bq2515x->dev, "battery info missing, default values will be applied\n");
968
969                 bq2515x->init_data.ichg = BQ2515X_DEFAULT_ICHG_UA;
970
971                 bq2515x->init_data.vbatreg = BQ2515X_DEFAULT_VBAT_REG_UV;
972
973                 bq2515x->init_data.iprechg = BQ2515X_DEFAULT_IPRECHARGE_UA;
974
975         } else {
976                 bq2515x->init_data.ichg =
977                                 bat_info->constant_charge_current_max_ua;
978
979                 bq2515x->init_data.vbatreg =
980                                 bat_info->constant_charge_voltage_max_uv;
981
982                 bq2515x->init_data.iprechg =
983                                 bat_info->precharge_current_ua;
984         }
985
986         ret = bq2515x_set_const_charge_current(bq2515x,
987                                                 bq2515x->init_data.ichg);
988         if (ret)
989                 return ret;
990
991         ret = bq2515x_set_batt_reg(bq2515x, bq2515x->init_data.vbatreg);
992         if (ret)
993                 return ret;
994
995         return bq2515x_set_precharge_current(bq2515x,
996                                                 bq2515x->init_data.iprechg);
997 }
998
999 static int bq2515x_read_properties(struct bq2515x_device *bq2515x)
1000 {
1001         int ret;
1002
1003         ret = device_property_read_u32(bq2515x->dev,
1004                                       "input-current-limit-microamp",
1005                                       &bq2515x->init_data.ilim);
1006         if (ret)
1007                 bq2515x->init_data.ilim = bq2515x->info->ilim;
1008
1009         bq2515x->ac_detect_gpio = devm_gpiod_get_optional(bq2515x->dev,
1010                                                    "ac-detect", GPIOD_IN);
1011         if (IS_ERR(bq2515x->ac_detect_gpio)) {
1012                 ret = PTR_ERR(bq2515x->ac_detect_gpio);
1013                 dev_err(bq2515x->dev, "Failed to get ac detect");
1014                 return ret;
1015         }
1016
1017         bq2515x->reset_gpio = devm_gpiod_get_optional(bq2515x->dev,
1018                                                    "reset", GPIOD_OUT_LOW);
1019         if (IS_ERR(bq2515x->reset_gpio)) {
1020                 ret = PTR_ERR(bq2515x->reset_gpio);
1021                 dev_err(bq2515x->dev, "Failed to get reset");
1022                 return ret;
1023         }
1024
1025         bq2515x->powerdown_gpio = devm_gpiod_get_optional(bq2515x->dev,
1026                                                 "powerdown", GPIOD_OUT_LOW);
1027         if (IS_ERR(bq2515x->powerdown_gpio)) {
1028                 ret = PTR_ERR(bq2515x->powerdown_gpio);
1029                 dev_err(bq2515x->dev, "Failed to get powerdown");
1030                 return ret;
1031         }
1032
1033         bq2515x->ce_gpio = devm_gpiod_get_optional(bq2515x->dev,
1034                                                    "charge-enable",
1035                                                    GPIOD_OUT_LOW);
1036         if (IS_ERR(bq2515x->ce_gpio)) {
1037                 ret = PTR_ERR(bq2515x->ce_gpio);
1038                 dev_err(bq2515x->dev, "Failed to get ce");
1039                 return ret;
1040         }
1041
1042         return 0;
1043 }
1044
1045 static bool bq2515x_volatile_register(struct device *dev, unsigned int reg)
1046 {
1047         switch (reg) {
1048         case BQ2515X_STAT0 ... BQ2515X_FLAG3:
1049         case BQ2515X_ADC_VBAT_M ... BQ2515X_ADC_IIN_L:
1050                 return true;
1051         default:
1052                 return false;
1053         }
1054 }
1055
1056 static const struct regmap_config bq25150_regmap_config = {
1057         .reg_bits = 8,
1058         .val_bits = 8,
1059
1060         .max_register           = BQ2515X_DEVICE_ID,
1061         .reg_defaults           = bq25150_reg_defaults,
1062         .num_reg_defaults       = ARRAY_SIZE(bq25150_reg_defaults),
1063         .cache_type             = REGCACHE_RBTREE,
1064         .volatile_reg           = bq2515x_volatile_register,
1065 };
1066
1067 static const struct regmap_config bq25155_regmap_config = {
1068         .reg_bits = 8,
1069         .val_bits = 8,
1070
1071         .max_register           = BQ2515X_DEVICE_ID,
1072         .reg_defaults           = bq25155_reg_defaults,
1073         .num_reg_defaults       = ARRAY_SIZE(bq25155_reg_defaults),
1074         .cache_type             = REGCACHE_RBTREE,
1075         .volatile_reg           = bq2515x_volatile_register,
1076 };
1077
1078 static int bq2515x_probe(struct i2c_client *client)
1079 {
1080         const struct i2c_device_id *id = i2c_client_get_device_id(client);
1081         struct device *dev = &client->dev;
1082         struct bq2515x_device *bq2515x;
1083         struct power_supply_config charger_cfg = {};
1084         int ret;
1085
1086         bq2515x = devm_kzalloc(dev, sizeof(*bq2515x), GFP_KERNEL);
1087         if (!bq2515x)
1088                 return -ENOMEM;
1089
1090         bq2515x->dev = dev;
1091
1092         strscpy(bq2515x->model_name, id->name, sizeof(bq2515x->model_name));
1093
1094         bq2515x->info = i2c_get_match_data(client);
1095         bq2515x->regmap = devm_regmap_init_i2c(client,
1096                                                bq2515x->info->regmap_config);
1097         if (IS_ERR(bq2515x->regmap)) {
1098                 dev_err(dev, "failed to allocate register map\n");
1099                 return PTR_ERR(bq2515x->regmap);
1100         }
1101
1102         i2c_set_clientdata(client, bq2515x);
1103
1104         charger_cfg.drv_data = bq2515x;
1105         charger_cfg.of_node = dev->of_node;
1106
1107         ret = bq2515x_read_properties(bq2515x);
1108         if (ret) {
1109                 dev_err(dev, "Failed to read device tree properties %d\n",
1110                                                                         ret);
1111                 return ret;
1112         }
1113
1114         ret = bq2515x_power_supply_register(bq2515x, dev, charger_cfg);
1115         if (ret) {
1116                 dev_err(dev, "failed to register power supply\n");
1117                 return ret;
1118         }
1119
1120         ret = bq2515x_hw_init(bq2515x);
1121         if (ret) {
1122                 dev_err(dev, "Cannot initialize the chip\n");
1123                 return ret;
1124         }
1125
1126         return 0;
1127 }
1128
1129 static const struct bq2515x_info bq25150 = {
1130         .regmap_config = &bq25150_regmap_config,
1131         .ilim = BQ25150_DEFAULT_ILIM_UA,
1132 };
1133
1134 static const struct bq2515x_info bq25155 = {
1135         .regmap_config = &bq25155_regmap_config,
1136         .ilim = BQ25155_DEFAULT_ILIM_UA,
1137 };
1138
1139 static const struct i2c_device_id bq2515x_i2c_ids[] = {
1140         { "bq25150", (kernel_ulong_t)&bq25150 },
1141         { "bq25155", (kernel_ulong_t)&bq25155 },
1142         {}
1143 };
1144 MODULE_DEVICE_TABLE(i2c, bq2515x_i2c_ids);
1145
1146 static const struct of_device_id bq2515x_of_match[] = {
1147         { .compatible = "ti,bq25150", .data = &bq25150 },
1148         { .compatible = "ti,bq25155", .data = &bq25155 },
1149         {}
1150 };
1151 MODULE_DEVICE_TABLE(of, bq2515x_of_match);
1152
1153 static struct i2c_driver bq2515x_driver = {
1154         .driver = {
1155                 .name = "bq2515x-charger",
1156                 .of_match_table = bq2515x_of_match,
1157         },
1158         .probe = bq2515x_probe,
1159         .id_table = bq2515x_i2c_ids,
1160 };
1161 module_i2c_driver(bq2515x_driver);
1162
1163 MODULE_AUTHOR("Dan Murphy <dmurphy@ti.com>");
1164 MODULE_AUTHOR("Ricardo Rivera-Matos <r-rivera-matos@ti.com>");
1165 MODULE_DESCRIPTION("BQ2515X charger driver");
1166 MODULE_LICENSE("GPL v2");