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
59struct 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
84static 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
97static 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, &reg_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
158static 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, &reg_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, &reg_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
227static 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
254static 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
306static 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, &reg_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
333static 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
345static 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
365static 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
378static 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
390static 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
406static 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
420static 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, &reg_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
461static 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
468static 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
541static 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
555static 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
563static 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
600static 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
608static const struct regmap_access_table regmap_rd_table = {
609	.yes_ranges = regmap_ranges_rd_yes,
610	.n_yes_ranges = 4,
611};
612
613static 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
620static const struct regmap_access_table regmap_wr_table = {
621	.yes_ranges = regmap_ranges_wr_yes,
622	.n_yes_ranges = 3,
623};
624
625static const struct regmap_range regmap_ranges_vol_yes[] = {
626	regmap_reg_range(CW2015_REG_VCELL, CW2015_REG_SOC + 1),
627};
628
629static const struct regmap_access_table regmap_vol_table = {
630	.yes_ranges = regmap_ranges_vol_yes,
631	.n_yes_ranges = 1,
632};
633
634static const struct regmap_config cw2015_regmap_config = {
635	.reg_bits = 8,
636	.val_bits = 8,
637	.rd_table = &regmap_rd_table,
638	.wr_table = &regmap_wr_table,
639	.volatile_table = &regmap_vol_table,
640	.max_register = CW2015_REG_BATINFO + CW2015_SIZE_BATINFO - 1,
641};
642
643static 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
712static 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
721static 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
731static SIMPLE_DEV_PM_OPS(cw_bat_pm_ops, cw_bat_suspend, cw_bat_resume);
732
733static const struct i2c_device_id cw_bat_id_table[] = {
734	{ "cw2015", 0 },
735	{ }
736};
737
738static const struct of_device_id cw2015_of_match[] = {
739	{ .compatible = "cellwise,cw2015" },
740	{ }
741};
742MODULE_DEVICE_TABLE(of, cw2015_of_match);
743
744static 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
754module_i2c_driver(cw_bat_driver);
755
756MODULE_AUTHOR("xhc<xhc@rock-chips.com>");
757MODULE_AUTHOR("Tobias Schramm <t.schramm@manjaro.org>");
758MODULE_DESCRIPTION("cw2015/cw2013 battery driver");
759MODULE_LICENSE("GPL");
760