xref: /openbmc/linux/drivers/iio/adc/ina2xx-adc.c (revision efe4a1ac)
1 /*
2  * INA2XX Current and Power Monitors
3  *
4  * Copyright 2015 Baylibre SAS.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 as
8  * published by the Free Software Foundation.
9  *
10  * Based on linux/drivers/iio/adc/ad7291.c
11  * Copyright 2010-2011 Analog Devices Inc.
12  *
13  * Based on linux/drivers/hwmon/ina2xx.c
14  * Copyright 2012 Lothar Felten <l-felten@ti.com>
15  *
16  * Licensed under the GPL-2 or later.
17  *
18  * IIO driver for INA219-220-226-230-231
19  *
20  * Configurable 7-bit I2C slave address from 0x40 to 0x4F
21  */
22 
23 #include <linux/delay.h>
24 #include <linux/i2c.h>
25 #include <linux/iio/iio.h>
26 #include <linux/iio/buffer.h>
27 #include <linux/iio/kfifo_buf.h>
28 #include <linux/iio/sysfs.h>
29 #include <linux/kthread.h>
30 #include <linux/module.h>
31 #include <linux/of_device.h>
32 #include <linux/regmap.h>
33 #include <linux/util_macros.h>
34 
35 #include <linux/platform_data/ina2xx.h>
36 
37 /* INA2XX registers definition */
38 #define INA2XX_CONFIG                   0x00
39 #define INA2XX_SHUNT_VOLTAGE            0x01	/* readonly */
40 #define INA2XX_BUS_VOLTAGE              0x02	/* readonly */
41 #define INA2XX_POWER                    0x03	/* readonly */
42 #define INA2XX_CURRENT                  0x04	/* readonly */
43 #define INA2XX_CALIBRATION              0x05
44 
45 #define INA226_ALERT_MASK		GENMASK(2, 1)
46 #define INA266_CVRF			BIT(3)
47 
48 #define INA2XX_MAX_REGISTERS            8
49 
50 /* settings - depend on use case */
51 #define INA219_CONFIG_DEFAULT           0x399F	/* PGA=8 */
52 #define INA226_CONFIG_DEFAULT           0x4327
53 #define INA226_DEFAULT_AVG              4
54 #define INA226_DEFAULT_IT		1110
55 
56 #define INA2XX_RSHUNT_DEFAULT           10000
57 
58 /*
59  * bit mask for reading the averaging setting in the configuration register
60  * FIXME: use regmap_fields.
61  */
62 #define INA2XX_MODE_MASK	GENMASK(3, 0)
63 
64 #define INA226_AVG_MASK		GENMASK(11, 9)
65 #define INA226_SHIFT_AVG(val)	((val) << 9)
66 
67 /* Integration time for VBus */
68 #define INA226_ITB_MASK		GENMASK(8, 6)
69 #define INA226_SHIFT_ITB(val)	((val) << 6)
70 
71 /* Integration time for VShunt */
72 #define INA226_ITS_MASK		GENMASK(5, 3)
73 #define INA226_SHIFT_ITS(val)	((val) << 3)
74 
75 /* Cosmetic macro giving the sampling period for a full P=UxI cycle */
76 #define SAMPLING_PERIOD(c)	((c->int_time_vbus + c->int_time_vshunt) \
77 				 * c->avg)
78 
79 static bool ina2xx_is_writeable_reg(struct device *dev, unsigned int reg)
80 {
81 	return (reg == INA2XX_CONFIG) || (reg > INA2XX_CURRENT);
82 }
83 
84 static bool ina2xx_is_volatile_reg(struct device *dev, unsigned int reg)
85 {
86 	return (reg != INA2XX_CONFIG);
87 }
88 
89 static inline bool is_signed_reg(unsigned int reg)
90 {
91 	return (reg == INA2XX_SHUNT_VOLTAGE) || (reg == INA2XX_CURRENT);
92 }
93 
94 static const struct regmap_config ina2xx_regmap_config = {
95 	.reg_bits = 8,
96 	.val_bits = 16,
97 	.max_register = INA2XX_MAX_REGISTERS,
98 	.writeable_reg = ina2xx_is_writeable_reg,
99 	.volatile_reg = ina2xx_is_volatile_reg,
100 };
101 
102 enum ina2xx_ids { ina219, ina226 };
103 
104 struct ina2xx_config {
105 	u16 config_default;
106 	int calibration_factor;
107 	int shunt_div;
108 	int bus_voltage_shift;
109 	int bus_voltage_lsb;	/* uV */
110 	int power_lsb;		/* uW */
111 };
112 
113 struct ina2xx_chip_info {
114 	struct regmap *regmap;
115 	struct task_struct *task;
116 	const struct ina2xx_config *config;
117 	struct mutex state_lock;
118 	unsigned int shunt_resistor;
119 	int avg;
120 	int int_time_vbus; /* Bus voltage integration time uS */
121 	int int_time_vshunt; /* Shunt voltage integration time uS */
122 	bool allow_async_readout;
123 };
124 
125 static const struct ina2xx_config ina2xx_config[] = {
126 	[ina219] = {
127 		.config_default = INA219_CONFIG_DEFAULT,
128 		.calibration_factor = 40960000,
129 		.shunt_div = 100,
130 		.bus_voltage_shift = 3,
131 		.bus_voltage_lsb = 4000,
132 		.power_lsb = 20000,
133 	},
134 	[ina226] = {
135 		.config_default = INA226_CONFIG_DEFAULT,
136 		.calibration_factor = 5120000,
137 		.shunt_div = 400,
138 		.bus_voltage_shift = 0,
139 		.bus_voltage_lsb = 1250,
140 		.power_lsb = 25000,
141 	},
142 };
143 
144 static int ina2xx_read_raw(struct iio_dev *indio_dev,
145 			   struct iio_chan_spec const *chan,
146 			   int *val, int *val2, long mask)
147 {
148 	int ret;
149 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
150 	unsigned int regval;
151 
152 	switch (mask) {
153 	case IIO_CHAN_INFO_RAW:
154 		ret = regmap_read(chip->regmap, chan->address, &regval);
155 		if (ret)
156 			return ret;
157 
158 		if (is_signed_reg(chan->address))
159 			*val = (s16) regval;
160 		else
161 			*val  = regval;
162 
163 		return IIO_VAL_INT;
164 
165 	case IIO_CHAN_INFO_OVERSAMPLING_RATIO:
166 		*val = chip->avg;
167 		return IIO_VAL_INT;
168 
169 	case IIO_CHAN_INFO_INT_TIME:
170 		*val = 0;
171 		if (chan->address == INA2XX_SHUNT_VOLTAGE)
172 			*val2 = chip->int_time_vshunt;
173 		else
174 			*val2 = chip->int_time_vbus;
175 
176 		return IIO_VAL_INT_PLUS_MICRO;
177 
178 	case IIO_CHAN_INFO_SAMP_FREQ:
179 		/*
180 		 * Sample freq is read only, it is a consequence of
181 		 * 1/AVG*(CT_bus+CT_shunt).
182 		 */
183 		*val = DIV_ROUND_CLOSEST(1000000, SAMPLING_PERIOD(chip));
184 
185 		return IIO_VAL_INT;
186 
187 	case IIO_CHAN_INFO_SCALE:
188 		switch (chan->address) {
189 		case INA2XX_SHUNT_VOLTAGE:
190 			/* processed (mV) = raw/shunt_div */
191 			*val2 = chip->config->shunt_div;
192 			*val = 1;
193 			return IIO_VAL_FRACTIONAL;
194 
195 		case INA2XX_BUS_VOLTAGE:
196 			/* processed (mV) = raw*lsb (uV) / (1000 << shift) */
197 			*val = chip->config->bus_voltage_lsb;
198 			*val2 = 1000 << chip->config->bus_voltage_shift;
199 			return IIO_VAL_FRACTIONAL;
200 
201 		case INA2XX_POWER:
202 			/* processed (mW) = raw*lsb (uW) / 1000 */
203 			*val = chip->config->power_lsb;
204 			*val2 = 1000;
205 			return IIO_VAL_FRACTIONAL;
206 
207 		case INA2XX_CURRENT:
208 			/* processed (mA) = raw (mA) */
209 			*val = 1;
210 			return IIO_VAL_INT;
211 		}
212 	}
213 
214 	return -EINVAL;
215 }
216 
217 /*
218  * Available averaging rates for ina226. The indices correspond with
219  * the bit values expected by the chip (according to the ina226 datasheet,
220  * table 3 AVG bit settings, found at
221  * http://www.ti.com/lit/ds/symlink/ina226.pdf.
222  */
223 static const int ina226_avg_tab[] = { 1, 4, 16, 64, 128, 256, 512, 1024 };
224 
225 static int ina226_set_average(struct ina2xx_chip_info *chip, unsigned int val,
226 			      unsigned int *config)
227 {
228 	int bits;
229 
230 	if (val > 1024 || val < 1)
231 		return -EINVAL;
232 
233 	bits = find_closest(val, ina226_avg_tab,
234 			    ARRAY_SIZE(ina226_avg_tab));
235 
236 	chip->avg = ina226_avg_tab[bits];
237 
238 	*config &= ~INA226_AVG_MASK;
239 	*config |= INA226_SHIFT_AVG(bits) & INA226_AVG_MASK;
240 
241 	return 0;
242 }
243 
244 /* Conversion times in uS */
245 static const int ina226_conv_time_tab[] = { 140, 204, 332, 588, 1100,
246 					    2116, 4156, 8244 };
247 
248 static int ina226_set_int_time_vbus(struct ina2xx_chip_info *chip,
249 				    unsigned int val_us, unsigned int *config)
250 {
251 	int bits;
252 
253 	if (val_us > 8244 || val_us < 140)
254 		return -EINVAL;
255 
256 	bits = find_closest(val_us, ina226_conv_time_tab,
257 			    ARRAY_SIZE(ina226_conv_time_tab));
258 
259 	chip->int_time_vbus = ina226_conv_time_tab[bits];
260 
261 	*config &= ~INA226_ITB_MASK;
262 	*config |= INA226_SHIFT_ITB(bits) & INA226_ITB_MASK;
263 
264 	return 0;
265 }
266 
267 static int ina226_set_int_time_vshunt(struct ina2xx_chip_info *chip,
268 				      unsigned int val_us, unsigned int *config)
269 {
270 	int bits;
271 
272 	if (val_us > 8244 || val_us < 140)
273 		return -EINVAL;
274 
275 	bits = find_closest(val_us, ina226_conv_time_tab,
276 			    ARRAY_SIZE(ina226_conv_time_tab));
277 
278 	chip->int_time_vshunt = ina226_conv_time_tab[bits];
279 
280 	*config &= ~INA226_ITS_MASK;
281 	*config |= INA226_SHIFT_ITS(bits) & INA226_ITS_MASK;
282 
283 	return 0;
284 }
285 
286 static int ina2xx_write_raw(struct iio_dev *indio_dev,
287 			    struct iio_chan_spec const *chan,
288 			    int val, int val2, long mask)
289 {
290 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
291 	unsigned int config, tmp;
292 	int ret;
293 
294 	if (iio_buffer_enabled(indio_dev))
295 		return -EBUSY;
296 
297 	mutex_lock(&chip->state_lock);
298 
299 	ret = regmap_read(chip->regmap, INA2XX_CONFIG, &config);
300 	if (ret)
301 		goto err;
302 
303 	tmp = config;
304 
305 	switch (mask) {
306 	case IIO_CHAN_INFO_OVERSAMPLING_RATIO:
307 		ret = ina226_set_average(chip, val, &tmp);
308 		break;
309 
310 	case IIO_CHAN_INFO_INT_TIME:
311 		if (chan->address == INA2XX_SHUNT_VOLTAGE)
312 			ret = ina226_set_int_time_vshunt(chip, val2, &tmp);
313 		else
314 			ret = ina226_set_int_time_vbus(chip, val2, &tmp);
315 		break;
316 
317 	default:
318 		ret = -EINVAL;
319 	}
320 
321 	if (!ret && (tmp != config))
322 		ret = regmap_write(chip->regmap, INA2XX_CONFIG, tmp);
323 err:
324 	mutex_unlock(&chip->state_lock);
325 
326 	return ret;
327 }
328 
329 static ssize_t ina2xx_allow_async_readout_show(struct device *dev,
330 					   struct device_attribute *attr,
331 					   char *buf)
332 {
333 	struct ina2xx_chip_info *chip = iio_priv(dev_to_iio_dev(dev));
334 
335 	return sprintf(buf, "%d\n", chip->allow_async_readout);
336 }
337 
338 static ssize_t ina2xx_allow_async_readout_store(struct device *dev,
339 				struct device_attribute *attr,
340 				const char *buf, size_t len)
341 {
342 	struct ina2xx_chip_info *chip = iio_priv(dev_to_iio_dev(dev));
343 	bool val;
344 	int ret;
345 
346 	ret = strtobool((const char *) buf, &val);
347 	if (ret)
348 		return ret;
349 
350 	chip->allow_async_readout = val;
351 
352 	return len;
353 }
354 
355 /*
356  * Set current LSB to 1mA, shunt is in uOhms
357  * (equation 13 in datasheet). We hardcode a Current_LSB
358  * of 1.0 x10-6. The only remaining parameter is RShunt.
359  * There is no need to expose the CALIBRATION register
360  * to the user for now. But we need to reset this register
361  * if the user updates RShunt after driver init, e.g upon
362  * reading an EEPROM/Probe-type value.
363  */
364 static int ina2xx_set_calibration(struct ina2xx_chip_info *chip)
365 {
366 	u16 regval = DIV_ROUND_CLOSEST(chip->config->calibration_factor,
367 				   chip->shunt_resistor);
368 
369 	return regmap_write(chip->regmap, INA2XX_CALIBRATION, regval);
370 }
371 
372 static int set_shunt_resistor(struct ina2xx_chip_info *chip, unsigned int val)
373 {
374 	if (val <= 0 || val > chip->config->calibration_factor)
375 		return -EINVAL;
376 
377 	chip->shunt_resistor = val;
378 
379 	return 0;
380 }
381 
382 static ssize_t ina2xx_shunt_resistor_show(struct device *dev,
383 					  struct device_attribute *attr,
384 					  char *buf)
385 {
386 	struct ina2xx_chip_info *chip = iio_priv(dev_to_iio_dev(dev));
387 
388 	return sprintf(buf, "%d\n", chip->shunt_resistor);
389 }
390 
391 static ssize_t ina2xx_shunt_resistor_store(struct device *dev,
392 					   struct device_attribute *attr,
393 					   const char *buf, size_t len)
394 {
395 	struct ina2xx_chip_info *chip = iio_priv(dev_to_iio_dev(dev));
396 	unsigned long val;
397 	int ret;
398 
399 	ret = kstrtoul((const char *) buf, 10, &val);
400 	if (ret)
401 		return ret;
402 
403 	ret = set_shunt_resistor(chip, val);
404 	if (ret)
405 		return ret;
406 
407 	/* Update the Calibration register */
408 	ret = ina2xx_set_calibration(chip);
409 	if (ret)
410 		return ret;
411 
412 	return len;
413 }
414 
415 #define INA2XX_CHAN(_type, _index, _address) { \
416 	.type = (_type), \
417 	.address = (_address), \
418 	.indexed = 1, \
419 	.channel = (_index), \
420 	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) \
421 	| BIT(IIO_CHAN_INFO_SCALE), \
422 	.info_mask_shared_by_dir = BIT(IIO_CHAN_INFO_SAMP_FREQ) | \
423 				   BIT(IIO_CHAN_INFO_OVERSAMPLING_RATIO), \
424 	.scan_index = (_index), \
425 	.scan_type = { \
426 		.sign = 'u', \
427 		.realbits = 16, \
428 		.storagebits = 16, \
429 		.endianness = IIO_CPU, \
430 	} \
431 }
432 
433 /*
434  * Sampling Freq is a consequence of the integration times of
435  * the Voltage channels.
436  */
437 #define INA2XX_CHAN_VOLTAGE(_index, _address) { \
438 	.type = IIO_VOLTAGE, \
439 	.address = (_address), \
440 	.indexed = 1, \
441 	.channel = (_index), \
442 	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) | \
443 			      BIT(IIO_CHAN_INFO_SCALE) | \
444 			      BIT(IIO_CHAN_INFO_INT_TIME), \
445 	.scan_index = (_index), \
446 	.scan_type = { \
447 		.sign = 'u', \
448 		.realbits = 16, \
449 		.storagebits = 16, \
450 		.endianness = IIO_LE, \
451 	} \
452 }
453 
454 static const struct iio_chan_spec ina2xx_channels[] = {
455 	INA2XX_CHAN_VOLTAGE(0, INA2XX_SHUNT_VOLTAGE),
456 	INA2XX_CHAN_VOLTAGE(1, INA2XX_BUS_VOLTAGE),
457 	INA2XX_CHAN(IIO_POWER, 2, INA2XX_POWER),
458 	INA2XX_CHAN(IIO_CURRENT, 3, INA2XX_CURRENT),
459 	IIO_CHAN_SOFT_TIMESTAMP(4),
460 };
461 
462 static int ina2xx_work_buffer(struct iio_dev *indio_dev)
463 {
464 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
465 	unsigned short data[8];
466 	int bit, ret, i = 0;
467 	s64 time_a, time_b;
468 	unsigned int alert;
469 
470 	time_a = iio_get_time_ns(indio_dev);
471 
472 	/*
473 	 * Because the timer thread and the chip conversion clock
474 	 * are asynchronous, the period difference will eventually
475 	 * result in reading V[k-1] again, or skip V[k] at time Tk.
476 	 * In order to resync the timer with the conversion process
477 	 * we check the ConVersionReadyFlag.
478 	 * On hardware that supports using the ALERT pin to toggle a
479 	 * GPIO a triggered buffer could be used instead.
480 	 * For now, we pay for that extra read of the ALERT register
481 	 */
482 	if (!chip->allow_async_readout)
483 		do {
484 			ret = regmap_read(chip->regmap, INA226_ALERT_MASK,
485 					  &alert);
486 			if (ret < 0)
487 				return ret;
488 
489 			alert &= INA266_CVRF;
490 		} while (!alert);
491 
492 	/*
493 	 * Single register reads: bulk_read will not work with ina226
494 	 * as there is no auto-increment of the address register for
495 	 * data length longer than 16bits.
496 	 */
497 	for_each_set_bit(bit, indio_dev->active_scan_mask,
498 			 indio_dev->masklength) {
499 		unsigned int val;
500 
501 		ret = regmap_read(chip->regmap,
502 				  INA2XX_SHUNT_VOLTAGE + bit, &val);
503 		if (ret < 0)
504 			return ret;
505 
506 		data[i++] = val;
507 	}
508 
509 	time_b = iio_get_time_ns(indio_dev);
510 
511 	iio_push_to_buffers_with_timestamp(indio_dev,
512 					   (unsigned int *)data, time_a);
513 
514 	return (unsigned long)(time_b - time_a) / 1000;
515 };
516 
517 static int ina2xx_capture_thread(void *data)
518 {
519 	struct iio_dev *indio_dev = data;
520 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
521 	unsigned int sampling_us = SAMPLING_PERIOD(chip);
522 	int buffer_us;
523 
524 	/*
525 	 * Poll a bit faster than the chip internal Fs, in case
526 	 * we wish to sync with the conversion ready flag.
527 	 */
528 	if (!chip->allow_async_readout)
529 		sampling_us -= 200;
530 
531 	do {
532 		buffer_us = ina2xx_work_buffer(indio_dev);
533 		if (buffer_us < 0)
534 			return buffer_us;
535 
536 		if (sampling_us > buffer_us)
537 			udelay(sampling_us - buffer_us);
538 
539 	} while (!kthread_should_stop());
540 
541 	return 0;
542 }
543 
544 static int ina2xx_buffer_enable(struct iio_dev *indio_dev)
545 {
546 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
547 	unsigned int sampling_us = SAMPLING_PERIOD(chip);
548 
549 	dev_dbg(&indio_dev->dev, "Enabling buffer w/ scan_mask %02x, freq = %d, avg =%u\n",
550 		(unsigned int)(*indio_dev->active_scan_mask),
551 		1000000 / sampling_us, chip->avg);
552 
553 	dev_dbg(&indio_dev->dev, "Expected work period: %u us\n", sampling_us);
554 	dev_dbg(&indio_dev->dev, "Async readout mode: %d\n",
555 		chip->allow_async_readout);
556 
557 	chip->task = kthread_run(ina2xx_capture_thread, (void *)indio_dev,
558 				 "%s:%d-%uus", indio_dev->name, indio_dev->id,
559 				 sampling_us);
560 
561 	return PTR_ERR_OR_ZERO(chip->task);
562 }
563 
564 static int ina2xx_buffer_disable(struct iio_dev *indio_dev)
565 {
566 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
567 
568 	if (chip->task) {
569 		kthread_stop(chip->task);
570 		chip->task = NULL;
571 	}
572 
573 	return 0;
574 }
575 
576 static const struct iio_buffer_setup_ops ina2xx_setup_ops = {
577 	.postenable = &ina2xx_buffer_enable,
578 	.predisable = &ina2xx_buffer_disable,
579 };
580 
581 static int ina2xx_debug_reg(struct iio_dev *indio_dev,
582 			    unsigned reg, unsigned writeval, unsigned *readval)
583 {
584 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
585 
586 	if (!readval)
587 		return regmap_write(chip->regmap, reg, writeval);
588 
589 	return regmap_read(chip->regmap, reg, readval);
590 }
591 
592 /* Possible integration times for vshunt and vbus */
593 static IIO_CONST_ATTR_INT_TIME_AVAIL("0.000140 0.000204 0.000332 0.000588 0.001100 0.002116 0.004156 0.008244");
594 
595 static IIO_DEVICE_ATTR(in_allow_async_readout, S_IRUGO | S_IWUSR,
596 		       ina2xx_allow_async_readout_show,
597 		       ina2xx_allow_async_readout_store, 0);
598 
599 static IIO_DEVICE_ATTR(in_shunt_resistor, S_IRUGO | S_IWUSR,
600 		       ina2xx_shunt_resistor_show,
601 		       ina2xx_shunt_resistor_store, 0);
602 
603 static struct attribute *ina2xx_attributes[] = {
604 	&iio_dev_attr_in_allow_async_readout.dev_attr.attr,
605 	&iio_const_attr_integration_time_available.dev_attr.attr,
606 	&iio_dev_attr_in_shunt_resistor.dev_attr.attr,
607 	NULL,
608 };
609 
610 static const struct attribute_group ina2xx_attribute_group = {
611 	.attrs = ina2xx_attributes,
612 };
613 
614 static const struct iio_info ina2xx_info = {
615 	.driver_module = THIS_MODULE,
616 	.attrs = &ina2xx_attribute_group,
617 	.read_raw = ina2xx_read_raw,
618 	.write_raw = ina2xx_write_raw,
619 	.debugfs_reg_access = ina2xx_debug_reg,
620 };
621 
622 /* Initialize the configuration and calibration registers. */
623 static int ina2xx_init(struct ina2xx_chip_info *chip, unsigned int config)
624 {
625 	int ret = regmap_write(chip->regmap, INA2XX_CONFIG, config);
626 	if (ret)
627 		return ret;
628 
629 	return ina2xx_set_calibration(chip);
630 }
631 
632 static int ina2xx_probe(struct i2c_client *client,
633 			const struct i2c_device_id *id)
634 {
635 	struct ina2xx_chip_info *chip;
636 	struct iio_dev *indio_dev;
637 	struct iio_buffer *buffer;
638 	unsigned int val;
639 	enum ina2xx_ids type;
640 	int ret;
641 
642 	indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*chip));
643 	if (!indio_dev)
644 		return -ENOMEM;
645 
646 	chip = iio_priv(indio_dev);
647 
648 	/* This is only used for device removal purposes. */
649 	i2c_set_clientdata(client, indio_dev);
650 
651 	chip->regmap = devm_regmap_init_i2c(client, &ina2xx_regmap_config);
652 	if (IS_ERR(chip->regmap)) {
653 		dev_err(&client->dev, "failed to allocate register map\n");
654 		return PTR_ERR(chip->regmap);
655 	}
656 
657 	if (client->dev.of_node)
658 		type = (enum ina2xx_ids)of_device_get_match_data(&client->dev);
659 	else
660 		type = id->driver_data;
661 	chip->config = &ina2xx_config[type];
662 
663 	mutex_init(&chip->state_lock);
664 
665 	if (of_property_read_u32(client->dev.of_node,
666 				 "shunt-resistor", &val) < 0) {
667 		struct ina2xx_platform_data *pdata =
668 		    dev_get_platdata(&client->dev);
669 
670 		if (pdata)
671 			val = pdata->shunt_uohms;
672 		else
673 			val = INA2XX_RSHUNT_DEFAULT;
674 	}
675 
676 	ret = set_shunt_resistor(chip, val);
677 	if (ret)
678 		return ret;
679 
680 	/* Patch the current config register with default. */
681 	val = chip->config->config_default;
682 
683 	if (id->driver_data == ina226) {
684 		ina226_set_average(chip, INA226_DEFAULT_AVG, &val);
685 		ina226_set_int_time_vbus(chip, INA226_DEFAULT_IT, &val);
686 		ina226_set_int_time_vshunt(chip, INA226_DEFAULT_IT, &val);
687 	}
688 
689 	ret = ina2xx_init(chip, val);
690 	if (ret) {
691 		dev_err(&client->dev, "error configuring the device\n");
692 		return ret;
693 	}
694 
695 	indio_dev->modes = INDIO_DIRECT_MODE | INDIO_BUFFER_SOFTWARE;
696 	indio_dev->dev.parent = &client->dev;
697 	indio_dev->dev.of_node = client->dev.of_node;
698 	indio_dev->channels = ina2xx_channels;
699 	indio_dev->num_channels = ARRAY_SIZE(ina2xx_channels);
700 	indio_dev->name = id->name;
701 	indio_dev->info = &ina2xx_info;
702 	indio_dev->setup_ops = &ina2xx_setup_ops;
703 
704 	buffer = devm_iio_kfifo_allocate(&indio_dev->dev);
705 	if (!buffer)
706 		return -ENOMEM;
707 
708 	iio_device_attach_buffer(indio_dev, buffer);
709 
710 	return iio_device_register(indio_dev);
711 }
712 
713 static int ina2xx_remove(struct i2c_client *client)
714 {
715 	struct iio_dev *indio_dev = i2c_get_clientdata(client);
716 	struct ina2xx_chip_info *chip = iio_priv(indio_dev);
717 
718 	iio_device_unregister(indio_dev);
719 
720 	/* Powerdown */
721 	return regmap_update_bits(chip->regmap, INA2XX_CONFIG,
722 				  INA2XX_MODE_MASK, 0);
723 }
724 
725 static const struct i2c_device_id ina2xx_id[] = {
726 	{"ina219", ina219},
727 	{"ina220", ina219},
728 	{"ina226", ina226},
729 	{"ina230", ina226},
730 	{"ina231", ina226},
731 	{}
732 };
733 MODULE_DEVICE_TABLE(i2c, ina2xx_id);
734 
735 static const struct of_device_id ina2xx_of_match[] = {
736 	{
737 		.compatible = "ti,ina219",
738 		.data = (void *)ina219
739 	},
740 	{
741 		.compatible = "ti,ina220",
742 		.data = (void *)ina219
743 	},
744 	{
745 		.compatible = "ti,ina226",
746 		.data = (void *)ina226
747 	},
748 	{
749 		.compatible = "ti,ina230",
750 		.data = (void *)ina226
751 	},
752 	{
753 		.compatible = "ti,ina231",
754 		.data = (void *)ina226
755 	},
756 	{},
757 };
758 MODULE_DEVICE_TABLE(of, ina2xx_of_match);
759 
760 static struct i2c_driver ina2xx_driver = {
761 	.driver = {
762 		   .name = KBUILD_MODNAME,
763 		   .of_match_table = ina2xx_of_match,
764 	},
765 	.probe = ina2xx_probe,
766 	.remove = ina2xx_remove,
767 	.id_table = ina2xx_id,
768 };
769 module_i2c_driver(ina2xx_driver);
770 
771 MODULE_AUTHOR("Marc Titinger <marc.titinger@baylibre.com>");
772 MODULE_DESCRIPTION("Texas Instruments INA2XX ADC driver");
773 MODULE_LICENSE("GPL v2");
774