1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Fuel gauge driver for CellWise 2013 / 2015
4 *
5 * Copyright (C) 2012, RockChip
6 * Copyright (C) 2020, Tobias Schramm
7 *
8 * Authors: xuhuicong <xhc@rock-chips.com>
9 * Authors: Tobias Schramm <t.schramm@manjaro.org>
10 */
11
12 #include <linux/bits.h>
13 #include <linux/delay.h>
14 #include <linux/i2c.h>
15 #include <linux/gfp.h>
16 #include <linux/gpio/consumer.h>
17 #include <linux/kernel.h>
18 #include <linux/module.h>
19 #include <linux/power_supply.h>
20 #include <linux/property.h>
21 #include <linux/regmap.h>
22 #include <linux/time.h>
23 #include <linux/workqueue.h>
24 #include <linux/devm-helpers.h>
25
26 #define CW2015_SIZE_BATINFO 64
27
28 #define CW2015_RESET_TRIES 5
29
30 #define CW2015_REG_VERSION 0x00
31 #define CW2015_REG_VCELL 0x02
32 #define CW2015_REG_SOC 0x04
33 #define CW2015_REG_RRT_ALERT 0x06
34 #define CW2015_REG_CONFIG 0x08
35 #define CW2015_REG_MODE 0x0A
36 #define CW2015_REG_BATINFO 0x10
37
38 #define CW2015_MODE_SLEEP_MASK GENMASK(7, 6)
39 #define CW2015_MODE_SLEEP (0x03 << 6)
40 #define CW2015_MODE_NORMAL (0x00 << 6)
41 #define CW2015_MODE_QUICK_START (0x03 << 4)
42 #define CW2015_MODE_RESTART (0x0f << 0)
43
44 #define CW2015_CONFIG_UPDATE_FLG (0x01 << 1)
45 #define CW2015_ATHD(x) ((x) << 3)
46 #define CW2015_MASK_ATHD GENMASK(7, 3)
47 #define CW2015_MASK_SOC GENMASK(12, 0)
48
49 /* reset gauge of no valid state of charge could be polled for 40s */
50 #define CW2015_BAT_SOC_ERROR_MS (40 * MSEC_PER_SEC)
51 /* reset gauge if state of charge stuck for half an hour during charging */
52 #define CW2015_BAT_CHARGING_STUCK_MS (1800 * MSEC_PER_SEC)
53
54 /* poll interval from CellWise GPL Android driver example */
55 #define CW2015_DEFAULT_POLL_INTERVAL_MS 8000
56
57 #define CW2015_AVERAGING_SAMPLES 3
58
59 struct cw_battery {
60 struct device *dev;
61 struct workqueue_struct *battery_workqueue;
62 struct delayed_work battery_delay_work;
63 struct regmap *regmap;
64 struct power_supply *rk_bat;
65 struct power_supply_battery_info *battery;
66 u8 *bat_profile;
67
68 bool charger_attached;
69 bool battery_changed;
70
71 int soc;
72 int voltage_mv;
73 int status;
74 int time_to_empty;
75 int charge_count;
76
77 u32 poll_interval_ms;
78 u8 alert_level;
79
80 unsigned int read_errors;
81 unsigned int charge_stuck_cnt;
82 };
83
cw_read_word(struct cw_battery * cw_bat,u8 reg,u16 * val)84 static int cw_read_word(struct cw_battery *cw_bat, u8 reg, u16 *val)
85 {
86 __be16 value;
87 int ret;
88
89 ret = regmap_bulk_read(cw_bat->regmap, reg, &value, sizeof(value));
90 if (ret)
91 return ret;
92
93 *val = be16_to_cpu(value);
94 return 0;
95 }
96
cw_update_profile(struct cw_battery * cw_bat)97 static int cw_update_profile(struct cw_battery *cw_bat)
98 {
99 int ret;
100 unsigned int reg_val;
101 u8 reset_val;
102
103 /* make sure gauge is not in sleep mode */
104 ret = regmap_read(cw_bat->regmap, CW2015_REG_MODE, ®_val);
105 if (ret)
106 return ret;
107
108 reset_val = reg_val;
109 if ((reg_val & CW2015_MODE_SLEEP_MASK) == CW2015_MODE_SLEEP) {
110 dev_err(cw_bat->dev,
111 "Gauge is in sleep mode, can't update battery info\n");
112 return -EINVAL;
113 }
114
115 /* write new battery info */
116 ret = regmap_raw_write(cw_bat->regmap, CW2015_REG_BATINFO,
117 cw_bat->bat_profile,
118 CW2015_SIZE_BATINFO);
119 if (ret)
120 return ret;
121
122 /* set config update flag */
123 reg_val |= CW2015_CONFIG_UPDATE_FLG;
124 reg_val &= ~CW2015_MASK_ATHD;
125 reg_val |= CW2015_ATHD(cw_bat->alert_level);
126 ret = regmap_write(cw_bat->regmap, CW2015_REG_CONFIG, reg_val);
127 if (ret)
128 return ret;
129
130 /* reset gauge to apply new battery profile */
131 reset_val &= ~CW2015_MODE_RESTART;
132 reg_val = reset_val | CW2015_MODE_RESTART;
133 ret = regmap_write(cw_bat->regmap, CW2015_REG_MODE, reg_val);
134 if (ret)
135 return ret;
136
137 /* wait for gauge to reset */
138 msleep(20);
139
140 /* clear reset flag */
141 ret = regmap_write(cw_bat->regmap, CW2015_REG_MODE, reset_val);
142 if (ret)
143 return ret;
144
145 /* wait for gauge to become ready */
146 ret = regmap_read_poll_timeout(cw_bat->regmap, CW2015_REG_SOC,
147 reg_val, reg_val <= 100,
148 10 * USEC_PER_MSEC, 10 * USEC_PER_SEC);
149 if (ret)
150 dev_err(cw_bat->dev,
151 "Gauge did not become ready after profile upload\n");
152 else
153 dev_dbg(cw_bat->dev, "Battery profile updated\n");
154
155 return ret;
156 }
157
cw_init(struct cw_battery * cw_bat)158 static int cw_init(struct cw_battery *cw_bat)
159 {
160 int ret;
161 unsigned int reg_val = CW2015_MODE_SLEEP;
162
163 if ((reg_val & CW2015_MODE_SLEEP_MASK) == CW2015_MODE_SLEEP) {
164 reg_val = CW2015_MODE_NORMAL;
165 ret = regmap_write(cw_bat->regmap, CW2015_REG_MODE, reg_val);
166 if (ret)
167 return ret;
168 }
169
170 ret = regmap_read(cw_bat->regmap, CW2015_REG_CONFIG, ®_val);
171 if (ret)
172 return ret;
173
174 if ((reg_val & CW2015_MASK_ATHD) != CW2015_ATHD(cw_bat->alert_level)) {
175 dev_dbg(cw_bat->dev, "Setting new alert level\n");
176 reg_val &= ~CW2015_MASK_ATHD;
177 reg_val |= ~CW2015_ATHD(cw_bat->alert_level);
178 ret = regmap_write(cw_bat->regmap, CW2015_REG_CONFIG, reg_val);
179 if (ret)
180 return ret;
181 }
182
183 ret = regmap_read(cw_bat->regmap, CW2015_REG_CONFIG, ®_val);
184 if (ret)
185 return ret;
186
187 if (!(reg_val & CW2015_CONFIG_UPDATE_FLG)) {
188 dev_dbg(cw_bat->dev,
189 "Battery profile not present, uploading battery profile\n");
190 if (cw_bat->bat_profile) {
191 ret = cw_update_profile(cw_bat);
192 if (ret) {
193 dev_err(cw_bat->dev,
194 "Failed to upload battery profile\n");
195 return ret;
196 }
197 } else {
198 dev_warn(cw_bat->dev,
199 "No profile specified, continuing without profile\n");
200 }
201 } else if (cw_bat->bat_profile) {
202 u8 bat_info[CW2015_SIZE_BATINFO];
203
204 ret = regmap_raw_read(cw_bat->regmap, CW2015_REG_BATINFO,
205 bat_info, CW2015_SIZE_BATINFO);
206 if (ret) {
207 dev_err(cw_bat->dev,
208 "Failed to read stored battery profile\n");
209 return ret;
210 }
211
212 if (memcmp(bat_info, cw_bat->bat_profile, CW2015_SIZE_BATINFO)) {
213 dev_warn(cw_bat->dev, "Replacing stored battery profile\n");
214 ret = cw_update_profile(cw_bat);
215 if (ret)
216 return ret;
217 }
218 } else {
219 dev_warn(cw_bat->dev,
220 "Can't check current battery profile, no profile provided\n");
221 }
222
223 dev_dbg(cw_bat->dev, "Battery profile configured\n");
224 return 0;
225 }
226
cw_power_on_reset(struct cw_battery * cw_bat)227 static int cw_power_on_reset(struct cw_battery *cw_bat)
228 {
229 int ret;
230 unsigned char reset_val;
231
232 reset_val = CW2015_MODE_SLEEP;
233 ret = regmap_write(cw_bat->regmap, CW2015_REG_MODE, reset_val);
234 if (ret)
235 return ret;
236
237 /* wait for gauge to enter sleep */
238 msleep(20);
239
240 reset_val = CW2015_MODE_NORMAL;
241 ret = regmap_write(cw_bat->regmap, CW2015_REG_MODE, reset_val);
242 if (ret)
243 return ret;
244
245 ret = cw_init(cw_bat);
246 if (ret)
247 return ret;
248 return 0;
249 }
250
251 #define HYSTERESIS(current, previous, up, down) \
252 (((current) < (previous) + (up)) && ((current) > (previous) - (down)))
253
cw_get_soc(struct cw_battery * cw_bat)254 static int cw_get_soc(struct cw_battery *cw_bat)
255 {
256 unsigned int soc;
257 int ret;
258
259 ret = regmap_read(cw_bat->regmap, CW2015_REG_SOC, &soc);
260 if (ret)
261 return ret;
262
263 if (soc > 100) {
264 int max_error_cycles =
265 CW2015_BAT_SOC_ERROR_MS / cw_bat->poll_interval_ms;
266
267 dev_err(cw_bat->dev, "Invalid SoC %d%%\n", soc);
268 cw_bat->read_errors++;
269 if (cw_bat->read_errors > max_error_cycles) {
270 dev_warn(cw_bat->dev,
271 "Too many invalid SoC reports, resetting gauge\n");
272 cw_power_on_reset(cw_bat);
273 cw_bat->read_errors = 0;
274 }
275 return cw_bat->soc;
276 }
277 cw_bat->read_errors = 0;
278
279 /* Reset gauge if stuck while charging */
280 if (cw_bat->status == POWER_SUPPLY_STATUS_CHARGING && soc == cw_bat->soc) {
281 int max_stuck_cycles =
282 CW2015_BAT_CHARGING_STUCK_MS / cw_bat->poll_interval_ms;
283
284 cw_bat->charge_stuck_cnt++;
285 if (cw_bat->charge_stuck_cnt > max_stuck_cycles) {
286 dev_warn(cw_bat->dev,
287 "SoC stuck @%u%%, resetting gauge\n", soc);
288 cw_power_on_reset(cw_bat);
289 cw_bat->charge_stuck_cnt = 0;
290 }
291 } else {
292 cw_bat->charge_stuck_cnt = 0;
293 }
294
295 /* Ignore voltage dips during charge */
296 if (cw_bat->charger_attached && HYSTERESIS(soc, cw_bat->soc, 0, 3))
297 soc = cw_bat->soc;
298
299 /* Ignore voltage spikes during discharge */
300 if (!cw_bat->charger_attached && HYSTERESIS(soc, cw_bat->soc, 3, 0))
301 soc = cw_bat->soc;
302
303 return soc;
304 }
305
cw_get_voltage(struct cw_battery * cw_bat)306 static int cw_get_voltage(struct cw_battery *cw_bat)
307 {
308 int ret, i, voltage_mv;
309 u16 reg_val;
310 u32 avg = 0;
311
312 for (i = 0; i < CW2015_AVERAGING_SAMPLES; i++) {
313 ret = cw_read_word(cw_bat, CW2015_REG_VCELL, ®_val);
314 if (ret)
315 return ret;
316
317 avg += reg_val;
318 }
319 avg /= CW2015_AVERAGING_SAMPLES;
320
321 /*
322 * 305 uV per ADC step
323 * Use 312 / 1024 as efficient approximation of 305 / 1000
324 * Negligible error of 0.1%
325 */
326 voltage_mv = avg * 312 / 1024;
327
328 dev_dbg(cw_bat->dev, "Read voltage: %d mV, raw=0x%04x\n",
329 voltage_mv, reg_val);
330 return voltage_mv;
331 }
332
cw_get_time_to_empty(struct cw_battery * cw_bat)333 static int cw_get_time_to_empty(struct cw_battery *cw_bat)
334 {
335 int ret;
336 u16 value16;
337
338 ret = cw_read_word(cw_bat, CW2015_REG_RRT_ALERT, &value16);
339 if (ret)
340 return ret;
341
342 return value16 & CW2015_MASK_SOC;
343 }
344
cw_update_charge_status(struct cw_battery * cw_bat)345 static void cw_update_charge_status(struct cw_battery *cw_bat)
346 {
347 int ret;
348
349 ret = power_supply_am_i_supplied(cw_bat->rk_bat);
350 if (ret < 0) {
351 dev_warn(cw_bat->dev, "Failed to get supply state: %d\n", ret);
352 } else {
353 bool charger_attached;
354
355 charger_attached = !!ret;
356 if (cw_bat->charger_attached != charger_attached) {
357 cw_bat->battery_changed = true;
358 if (charger_attached)
359 cw_bat->charge_count++;
360 }
361 cw_bat->charger_attached = charger_attached;
362 }
363 }
364
cw_update_soc(struct cw_battery * cw_bat)365 static void cw_update_soc(struct cw_battery *cw_bat)
366 {
367 int soc;
368
369 soc = cw_get_soc(cw_bat);
370 if (soc < 0)
371 dev_err(cw_bat->dev, "Failed to get SoC from gauge: %d\n", soc);
372 else if (cw_bat->soc != soc) {
373 cw_bat->soc = soc;
374 cw_bat->battery_changed = true;
375 }
376 }
377
cw_update_voltage(struct cw_battery * cw_bat)378 static void cw_update_voltage(struct cw_battery *cw_bat)
379 {
380 int voltage_mv;
381
382 voltage_mv = cw_get_voltage(cw_bat);
383 if (voltage_mv < 0)
384 dev_err(cw_bat->dev, "Failed to get voltage from gauge: %d\n",
385 voltage_mv);
386 else
387 cw_bat->voltage_mv = voltage_mv;
388 }
389
cw_update_status(struct cw_battery * cw_bat)390 static void cw_update_status(struct cw_battery *cw_bat)
391 {
392 int status = POWER_SUPPLY_STATUS_DISCHARGING;
393
394 if (cw_bat->charger_attached) {
395 if (cw_bat->soc >= 100)
396 status = POWER_SUPPLY_STATUS_FULL;
397 else
398 status = POWER_SUPPLY_STATUS_CHARGING;
399 }
400
401 if (cw_bat->status != status)
402 cw_bat->battery_changed = true;
403 cw_bat->status = status;
404 }
405
cw_update_time_to_empty(struct cw_battery * cw_bat)406 static void cw_update_time_to_empty(struct cw_battery *cw_bat)
407 {
408 int time_to_empty;
409
410 time_to_empty = cw_get_time_to_empty(cw_bat);
411 if (time_to_empty < 0)
412 dev_err(cw_bat->dev, "Failed to get time to empty from gauge: %d\n",
413 time_to_empty);
414 else if (cw_bat->time_to_empty != time_to_empty) {
415 cw_bat->time_to_empty = time_to_empty;
416 cw_bat->battery_changed = true;
417 }
418 }
419
cw_bat_work(struct work_struct * work)420 static void cw_bat_work(struct work_struct *work)
421 {
422 struct delayed_work *delay_work;
423 struct cw_battery *cw_bat;
424 int ret;
425 unsigned int reg_val;
426
427 delay_work = to_delayed_work(work);
428 cw_bat = container_of(delay_work, struct cw_battery, battery_delay_work);
429 ret = regmap_read(cw_bat->regmap, CW2015_REG_MODE, ®_val);
430 if (ret) {
431 dev_err(cw_bat->dev, "Failed to read mode from gauge: %d\n", ret);
432 } else {
433 if ((reg_val & CW2015_MODE_SLEEP_MASK) == CW2015_MODE_SLEEP) {
434 int i;
435
436 for (i = 0; i < CW2015_RESET_TRIES; i++) {
437 if (!cw_power_on_reset(cw_bat))
438 break;
439 }
440 }
441 cw_update_soc(cw_bat);
442 cw_update_voltage(cw_bat);
443 cw_update_charge_status(cw_bat);
444 cw_update_status(cw_bat);
445 cw_update_time_to_empty(cw_bat);
446 }
447 dev_dbg(cw_bat->dev, "charger_attached = %d\n", cw_bat->charger_attached);
448 dev_dbg(cw_bat->dev, "status = %d\n", cw_bat->status);
449 dev_dbg(cw_bat->dev, "soc = %d%%\n", cw_bat->soc);
450 dev_dbg(cw_bat->dev, "voltage = %dmV\n", cw_bat->voltage_mv);
451
452 if (cw_bat->battery_changed)
453 power_supply_changed(cw_bat->rk_bat);
454 cw_bat->battery_changed = false;
455
456 queue_delayed_work(cw_bat->battery_workqueue,
457 &cw_bat->battery_delay_work,
458 msecs_to_jiffies(cw_bat->poll_interval_ms));
459 }
460
cw_battery_valid_time_to_empty(struct cw_battery * cw_bat)461 static bool cw_battery_valid_time_to_empty(struct cw_battery *cw_bat)
462 {
463 return cw_bat->time_to_empty > 0 &&
464 cw_bat->time_to_empty < CW2015_MASK_SOC &&
465 cw_bat->status == POWER_SUPPLY_STATUS_DISCHARGING;
466 }
467
cw_battery_get_property(struct power_supply * psy,enum power_supply_property psp,union power_supply_propval * val)468 static int cw_battery_get_property(struct power_supply *psy,
469 enum power_supply_property psp,
470 union power_supply_propval *val)
471 {
472 struct cw_battery *cw_bat;
473
474 cw_bat = power_supply_get_drvdata(psy);
475 switch (psp) {
476 case POWER_SUPPLY_PROP_CAPACITY:
477 val->intval = cw_bat->soc;
478 break;
479
480 case POWER_SUPPLY_PROP_STATUS:
481 val->intval = cw_bat->status;
482 break;
483
484 case POWER_SUPPLY_PROP_PRESENT:
485 val->intval = !!cw_bat->voltage_mv;
486 break;
487
488 case POWER_SUPPLY_PROP_VOLTAGE_NOW:
489 val->intval = cw_bat->voltage_mv * 1000;
490 break;
491
492 case POWER_SUPPLY_PROP_TIME_TO_EMPTY_NOW:
493 if (cw_battery_valid_time_to_empty(cw_bat))
494 val->intval = cw_bat->time_to_empty * 60;
495 else
496 val->intval = 0;
497 break;
498
499 case POWER_SUPPLY_PROP_TECHNOLOGY:
500 val->intval = POWER_SUPPLY_TECHNOLOGY_LION;
501 break;
502
503 case POWER_SUPPLY_PROP_CHARGE_COUNTER:
504 val->intval = cw_bat->charge_count;
505 break;
506
507 case POWER_SUPPLY_PROP_CHARGE_FULL:
508 case POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN:
509 if (cw_bat->battery->charge_full_design_uah > 0)
510 val->intval = cw_bat->battery->charge_full_design_uah;
511 else
512 val->intval = 0;
513 break;
514
515 case POWER_SUPPLY_PROP_CHARGE_NOW:
516 val->intval = cw_bat->battery->charge_full_design_uah;
517 val->intval = val->intval * cw_bat->soc / 100;
518 break;
519
520 case POWER_SUPPLY_PROP_CURRENT_NOW:
521 if (cw_battery_valid_time_to_empty(cw_bat) &&
522 cw_bat->battery->charge_full_design_uah > 0) {
523 /* calculate remaining capacity */
524 val->intval = cw_bat->battery->charge_full_design_uah;
525 val->intval = val->intval * cw_bat->soc / 100;
526
527 /* estimate current based on time to empty */
528 val->intval = 60 * val->intval / cw_bat->time_to_empty;
529 } else {
530 val->intval = 0;
531 }
532
533 break;
534
535 default:
536 break;
537 }
538 return 0;
539 }
540
541 static enum power_supply_property cw_battery_properties[] = {
542 POWER_SUPPLY_PROP_CAPACITY,
543 POWER_SUPPLY_PROP_STATUS,
544 POWER_SUPPLY_PROP_PRESENT,
545 POWER_SUPPLY_PROP_VOLTAGE_NOW,
546 POWER_SUPPLY_PROP_TIME_TO_EMPTY_NOW,
547 POWER_SUPPLY_PROP_TECHNOLOGY,
548 POWER_SUPPLY_PROP_CHARGE_COUNTER,
549 POWER_SUPPLY_PROP_CHARGE_FULL,
550 POWER_SUPPLY_PROP_CHARGE_FULL_DESIGN,
551 POWER_SUPPLY_PROP_CHARGE_NOW,
552 POWER_SUPPLY_PROP_CURRENT_NOW,
553 };
554
555 static const struct power_supply_desc cw2015_bat_desc = {
556 .name = "cw2015-battery",
557 .type = POWER_SUPPLY_TYPE_BATTERY,
558 .properties = cw_battery_properties,
559 .num_properties = ARRAY_SIZE(cw_battery_properties),
560 .get_property = cw_battery_get_property,
561 };
562
cw2015_parse_properties(struct cw_battery * cw_bat)563 static int cw2015_parse_properties(struct cw_battery *cw_bat)
564 {
565 struct device *dev = cw_bat->dev;
566 int length;
567 int ret;
568
569 length = device_property_count_u8(dev, "cellwise,battery-profile");
570 if (length < 0) {
571 dev_warn(cw_bat->dev,
572 "No battery-profile found, using current flash contents\n");
573 } else if (length != CW2015_SIZE_BATINFO) {
574 dev_err(cw_bat->dev, "battery-profile must be %d bytes\n",
575 CW2015_SIZE_BATINFO);
576 return -EINVAL;
577 } else {
578 cw_bat->bat_profile = devm_kzalloc(dev, length, GFP_KERNEL);
579 if (!cw_bat->bat_profile)
580 return -ENOMEM;
581
582 ret = device_property_read_u8_array(dev,
583 "cellwise,battery-profile",
584 cw_bat->bat_profile,
585 length);
586 if (ret)
587 return ret;
588 }
589
590 ret = device_property_read_u32(dev, "cellwise,monitor-interval-ms",
591 &cw_bat->poll_interval_ms);
592 if (ret) {
593 dev_dbg(cw_bat->dev, "Using default poll interval\n");
594 cw_bat->poll_interval_ms = CW2015_DEFAULT_POLL_INTERVAL_MS;
595 }
596
597 return 0;
598 }
599
600 static const struct regmap_range regmap_ranges_rd_yes[] = {
601 regmap_reg_range(CW2015_REG_VERSION, CW2015_REG_VERSION),
602 regmap_reg_range(CW2015_REG_VCELL, CW2015_REG_CONFIG),
603 regmap_reg_range(CW2015_REG_MODE, CW2015_REG_MODE),
604 regmap_reg_range(CW2015_REG_BATINFO,
605 CW2015_REG_BATINFO + CW2015_SIZE_BATINFO - 1),
606 };
607
608 static const struct regmap_access_table regmap_rd_table = {
609 .yes_ranges = regmap_ranges_rd_yes,
610 .n_yes_ranges = 4,
611 };
612
613 static const struct regmap_range regmap_ranges_wr_yes[] = {
614 regmap_reg_range(CW2015_REG_RRT_ALERT, CW2015_REG_CONFIG),
615 regmap_reg_range(CW2015_REG_MODE, CW2015_REG_MODE),
616 regmap_reg_range(CW2015_REG_BATINFO,
617 CW2015_REG_BATINFO + CW2015_SIZE_BATINFO - 1),
618 };
619
620 static const struct regmap_access_table regmap_wr_table = {
621 .yes_ranges = regmap_ranges_wr_yes,
622 .n_yes_ranges = 3,
623 };
624
625 static const struct regmap_range regmap_ranges_vol_yes[] = {
626 regmap_reg_range(CW2015_REG_VCELL, CW2015_REG_SOC + 1),
627 };
628
629 static const struct regmap_access_table regmap_vol_table = {
630 .yes_ranges = regmap_ranges_vol_yes,
631 .n_yes_ranges = 1,
632 };
633
634 static const struct regmap_config cw2015_regmap_config = {
635 .reg_bits = 8,
636 .val_bits = 8,
637 .rd_table = ®map_rd_table,
638 .wr_table = ®map_wr_table,
639 .volatile_table = ®map_vol_table,
640 .max_register = CW2015_REG_BATINFO + CW2015_SIZE_BATINFO - 1,
641 };
642
cw_bat_probe(struct i2c_client * client)643 static int cw_bat_probe(struct i2c_client *client)
644 {
645 int ret;
646 struct cw_battery *cw_bat;
647 struct power_supply_config psy_cfg = { 0 };
648
649 cw_bat = devm_kzalloc(&client->dev, sizeof(*cw_bat), GFP_KERNEL);
650 if (!cw_bat)
651 return -ENOMEM;
652
653 i2c_set_clientdata(client, cw_bat);
654 cw_bat->dev = &client->dev;
655 cw_bat->soc = 1;
656
657 ret = cw2015_parse_properties(cw_bat);
658 if (ret) {
659 dev_err(cw_bat->dev, "Failed to parse cw2015 properties\n");
660 return ret;
661 }
662
663 cw_bat->regmap = devm_regmap_init_i2c(client, &cw2015_regmap_config);
664 if (IS_ERR(cw_bat->regmap)) {
665 dev_err(cw_bat->dev, "Failed to allocate regmap: %ld\n",
666 PTR_ERR(cw_bat->regmap));
667 return PTR_ERR(cw_bat->regmap);
668 }
669
670 ret = cw_init(cw_bat);
671 if (ret) {
672 dev_err(cw_bat->dev, "Init failed: %d\n", ret);
673 return ret;
674 }
675
676 psy_cfg.drv_data = cw_bat;
677 psy_cfg.fwnode = dev_fwnode(cw_bat->dev);
678
679 cw_bat->rk_bat = devm_power_supply_register(&client->dev,
680 &cw2015_bat_desc,
681 &psy_cfg);
682 if (IS_ERR(cw_bat->rk_bat)) {
683 /* try again if this happens */
684 dev_err_probe(&client->dev, PTR_ERR(cw_bat->rk_bat),
685 "Failed to register power supply\n");
686 return PTR_ERR(cw_bat->rk_bat);
687 }
688
689 ret = power_supply_get_battery_info(cw_bat->rk_bat, &cw_bat->battery);
690 if (ret) {
691 /* Allocate an empty battery */
692 cw_bat->battery = devm_kzalloc(&client->dev,
693 sizeof(*cw_bat->battery),
694 GFP_KERNEL);
695 if (!cw_bat->battery)
696 return -ENOMEM;
697 dev_warn(cw_bat->dev,
698 "No monitored battery, some properties will be missing\n");
699 }
700
701 cw_bat->battery_workqueue = create_singlethread_workqueue("rk_battery");
702 if (!cw_bat->battery_workqueue)
703 return -ENOMEM;
704
705 devm_delayed_work_autocancel(&client->dev,
706 &cw_bat->battery_delay_work, cw_bat_work);
707 queue_delayed_work(cw_bat->battery_workqueue,
708 &cw_bat->battery_delay_work, msecs_to_jiffies(10));
709 return 0;
710 }
711
cw_bat_suspend(struct device * dev)712 static int __maybe_unused cw_bat_suspend(struct device *dev)
713 {
714 struct i2c_client *client = to_i2c_client(dev);
715 struct cw_battery *cw_bat = i2c_get_clientdata(client);
716
717 cancel_delayed_work_sync(&cw_bat->battery_delay_work);
718 return 0;
719 }
720
cw_bat_resume(struct device * dev)721 static int __maybe_unused cw_bat_resume(struct device *dev)
722 {
723 struct i2c_client *client = to_i2c_client(dev);
724 struct cw_battery *cw_bat = i2c_get_clientdata(client);
725
726 queue_delayed_work(cw_bat->battery_workqueue,
727 &cw_bat->battery_delay_work, 0);
728 return 0;
729 }
730
731 static SIMPLE_DEV_PM_OPS(cw_bat_pm_ops, cw_bat_suspend, cw_bat_resume);
732
733 static const struct i2c_device_id cw_bat_id_table[] = {
734 { "cw2015" },
735 { }
736 };
737
738 static const struct of_device_id cw2015_of_match[] = {
739 { .compatible = "cellwise,cw2015" },
740 { }
741 };
742 MODULE_DEVICE_TABLE(of, cw2015_of_match);
743
744 static struct i2c_driver cw_bat_driver = {
745 .driver = {
746 .name = "cw2015",
747 .of_match_table = cw2015_of_match,
748 .pm = &cw_bat_pm_ops,
749 },
750 .probe = cw_bat_probe,
751 .id_table = cw_bat_id_table,
752 };
753
754 module_i2c_driver(cw_bat_driver);
755
756 MODULE_AUTHOR("xhc<xhc@rock-chips.com>");
757 MODULE_AUTHOR("Tobias Schramm <t.schramm@manjaro.org>");
758 MODULE_DESCRIPTION("cw2015/cw2013 battery driver");
759 MODULE_LICENSE("GPL");
760