xref: /linux/drivers/iio/adc/ti-ads1119.c (revision fab183d632628381b466a41479489541ac0e29a0)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Texas Instruments ADS1119 ADC driver.
4  *
5  * Copyright 2024 Toradex
6  */
7 
8 #include <linux/bits.h>
9 #include <linux/bitfield.h>
10 #include <linux/completion.h>
11 #include <linux/delay.h>
12 #include <linux/dev_printk.h>
13 #include <linux/err.h>
14 #include <linux/gpio/consumer.h>
15 #include <linux/interrupt.h>
16 #include <linux/iopoll.h>
17 #include <linux/i2c.h>
18 #include <linux/kernel.h>
19 #include <linux/math.h>
20 #include <linux/module.h>
21 #include <linux/pm_runtime.h>
22 #include <linux/regulator/consumer.h>
23 #include <linux/units.h>
24 
25 #include <linux/iio/iio.h>
26 #include <linux/iio/buffer.h>
27 #include <linux/iio/trigger.h>
28 #include <linux/iio/triggered_buffer.h>
29 #include <linux/iio/trigger_consumer.h>
30 
31 #define ADS1119_CMD_RESET		0x06
32 #define ADS1119_CMD_POWERDOWN		0x02
33 #define ADS1119_CMD_START_SYNC		0x08
34 #define ADS1119_CMD_RDATA		0x10
35 #define ADS1119_CMD_RREG_CONFIG		0x20
36 #define ADS1119_CMD_RREG_STATUS		0x24
37 #define ADS1119_CMD_WREG		0x40
38 
39 #define ADS1119_CMD_RREG(reg)		(0x20 | (reg) << 2)
40 
41 /* Config register */
42 #define ADS1119_REG_CONFIG	0x00
43 #define ADS1119_CONFIG_VREF_FIELD	BIT(0)
44 #define ADS1119_CONFIG_CM_FIELD		BIT(1)
45 #define ADS1119_CONFIG_DR_FIELD		GENMASK(3, 2)
46 #define ADS1119_CONFIG_GAIN_FIELD	BIT(4)
47 #define ADS1119_CONFIG_MUX_FIELD	GENMASK(7, 5)
48 
49 #define ADS1119_VREF_INTERNAL	0
50 #define ADS1119_VREF_EXTERNAL	1
51 #define ADS1119_VREF_INTERNAL_VAL 2048000
52 
53 #define ADS1119_CM_SINGLE	0
54 #define ADS1119_CM_CONTINUOUS	1
55 
56 #define ADS1119_DR_20_SPS	0
57 #define ADS1119_DR_90_SPS	1
58 #define ADS1119_DR_330_SPS	2
59 #define ADS1119_DR_1000_SPS	3
60 
61 #define ADS1119_GAIN_1	0
62 #define ADS1119_GAIN_4	1
63 
64 #define ADS1119_MUX_AIN0_AIN1	0
65 #define ADS1119_MUX_AIN2_AIN3	1
66 #define ADS1119_MUX_AIN1_AIN2	2
67 #define ADS1119_MUX_AIN0	3
68 #define ADS1119_MUX_AIN1	4
69 #define ADS1119_MUX_AIN2	5
70 #define ADS1119_MUX_AIN3	6
71 #define ADS1119_MUX_SHORTED	7
72 
73 /* Status register */
74 #define ADS1119_REG_STATUS	0x01
75 #define ADS1119_STATUS_DRDY_FIELD	BIT(7)
76 
77 #define ADS1119_DEFAULT_GAIN		1
78 #define ADS1119_DEFAULT_DATARATE	20
79 
80 #define ADS1119_SUSPEND_DELAY		2000
81 
82 /* Timeout based on the minimum sample rate of 20 SPS (50000us) */
83 #define ADS1119_MAX_DRDY_TIMEOUT	85000
84 
85 #define ADS1119_MAX_CHANNELS		7
86 #define ADS1119_MAX_SINGLE_CHANNELS	4
87 
88 struct ads1119_channel_config {
89 	int gain;
90 	int datarate;
91 	int mux;
92 };
93 
94 struct ads1119_state {
95 	struct completion completion;
96 	struct i2c_client *client;
97 	struct gpio_desc *reset_gpio;
98 	struct iio_trigger *trig;
99 	struct ads1119_channel_config *channels_cfg;
100 	unsigned int num_channels_cfg;
101 	unsigned int cached_config;
102 	int vref_uV;
103 };
104 
105 static const char * const ads1119_power_supplies[] = {
106 	"avdd", "dvdd"
107 };
108 
109 static const int ads1119_available_datarates[] = {
110 	20, 90, 330, 1000,
111 };
112 
113 static const int ads1119_available_gains[] = {
114 	1, 1,
115 	1, 4,
116 };
117 
ads1119_upd_cfg_reg(struct ads1119_state * st,unsigned int fields,unsigned int val)118 static int ads1119_upd_cfg_reg(struct ads1119_state *st, unsigned int fields,
119 			       unsigned int val)
120 {
121 	unsigned int config = st->cached_config;
122 	int ret;
123 
124 	config &= ~fields;
125 	config |= val;
126 
127 	ret = i2c_smbus_write_byte_data(st->client, ADS1119_CMD_WREG, config);
128 	if (ret)
129 		return ret;
130 
131 	st->cached_config = config;
132 
133 	return 0;
134 }
135 
ads1119_data_ready(struct ads1119_state * st)136 static bool ads1119_data_ready(struct ads1119_state *st)
137 {
138 	int status;
139 
140 	status = i2c_smbus_read_byte_data(st->client, ADS1119_CMD_RREG_STATUS);
141 	if (status < 0)
142 		return false;
143 
144 	return FIELD_GET(ADS1119_STATUS_DRDY_FIELD, status);
145 }
146 
ads1119_reset(struct ads1119_state * st)147 static int ads1119_reset(struct ads1119_state *st)
148 {
149 	st->cached_config = 0;
150 
151 	if (!st->reset_gpio)
152 		return i2c_smbus_write_byte(st->client, ADS1119_CMD_RESET);
153 
154 	gpiod_set_value_cansleep(st->reset_gpio, 1);
155 	udelay(1);
156 	gpiod_set_value_cansleep(st->reset_gpio, 0);
157 	udelay(1);
158 
159 	return 0;
160 }
161 
ads1119_set_conv_mode(struct ads1119_state * st,bool continuous)162 static int ads1119_set_conv_mode(struct ads1119_state *st, bool continuous)
163 {
164 	unsigned int mode;
165 
166 	if (continuous)
167 		mode = ADS1119_CM_CONTINUOUS;
168 	else
169 		mode = ADS1119_CM_SINGLE;
170 
171 	return ads1119_upd_cfg_reg(st, ADS1119_CONFIG_CM_FIELD,
172 				   FIELD_PREP(ADS1119_CONFIG_CM_FIELD, mode));
173 }
174 
ads1119_get_hw_gain(int gain)175 static int ads1119_get_hw_gain(int gain)
176 {
177 	if (gain == 4)
178 		return ADS1119_GAIN_4;
179 	else
180 		return ADS1119_GAIN_1;
181 }
182 
ads1119_get_hw_datarate(int datarate)183 static int ads1119_get_hw_datarate(int datarate)
184 {
185 	switch (datarate) {
186 	case 90:
187 		return ADS1119_DR_90_SPS;
188 	case 330:
189 		return ADS1119_DR_330_SPS;
190 	case 1000:
191 		return ADS1119_DR_1000_SPS;
192 	case 20:
193 	default:
194 		return ADS1119_DR_20_SPS;
195 	}
196 }
197 
ads1119_configure_channel(struct ads1119_state * st,int mux,int gain,int datarate)198 static int ads1119_configure_channel(struct ads1119_state *st, int mux,
199 				     int gain, int datarate)
200 {
201 	int ret;
202 
203 	ret = ads1119_upd_cfg_reg(st, ADS1119_CONFIG_MUX_FIELD,
204 				  FIELD_PREP(ADS1119_CONFIG_MUX_FIELD, mux));
205 	if (ret)
206 		return ret;
207 
208 	ret = ads1119_upd_cfg_reg(st, ADS1119_CONFIG_GAIN_FIELD,
209 				  FIELD_PREP(ADS1119_CONFIG_GAIN_FIELD,
210 					     ads1119_get_hw_gain(gain)));
211 	if (ret)
212 		return ret;
213 
214 	return ads1119_upd_cfg_reg(st, ADS1119_CONFIG_DR_FIELD,
215 				   FIELD_PREP(ADS1119_CONFIG_DR_FIELD,
216 					      ads1119_get_hw_datarate(datarate)));
217 }
218 
ads1119_poll_data_ready(struct ads1119_state * st,struct iio_chan_spec const * chan)219 static int ads1119_poll_data_ready(struct ads1119_state *st,
220 				   struct iio_chan_spec const *chan)
221 {
222 	unsigned int datarate = st->channels_cfg[chan->address].datarate;
223 	unsigned long wait_time;
224 	bool data_ready;
225 
226 	/* Poll 5 times more than the data rate */
227 	wait_time = DIV_ROUND_CLOSEST(MICRO, 5 * datarate);
228 
229 	return read_poll_timeout(ads1119_data_ready, data_ready,
230 				 data_ready, wait_time,
231 				 ADS1119_MAX_DRDY_TIMEOUT, false, st);
232 }
233 
ads1119_read_data(struct ads1119_state * st,struct iio_chan_spec const * chan,unsigned int * val)234 static int ads1119_read_data(struct ads1119_state *st,
235 			     struct iio_chan_spec const *chan,
236 			     unsigned int *val)
237 {
238 	unsigned int timeout;
239 	int ret = 0;
240 
241 	timeout = msecs_to_jiffies(ADS1119_MAX_DRDY_TIMEOUT);
242 
243 	if (!st->client->irq) {
244 		ret = ads1119_poll_data_ready(st, chan);
245 		if (ret)
246 			return ret;
247 	} else if (!wait_for_completion_timeout(&st->completion, timeout)) {
248 		return -ETIMEDOUT;
249 	}
250 
251 	ret = i2c_smbus_read_word_swapped(st->client, ADS1119_CMD_RDATA);
252 	if (ret < 0)
253 		return ret;
254 
255 	*val = ret;
256 
257 	return 0;
258 }
259 
ads1119_single_conversion(struct ads1119_state * st,struct iio_chan_spec const * chan,int * val,bool calib_offset)260 static int ads1119_single_conversion(struct ads1119_state *st,
261 				     struct iio_chan_spec const *chan,
262 				     int *val,
263 				     bool calib_offset)
264 {
265 	struct device *dev = &st->client->dev;
266 	int mux = st->channels_cfg[chan->address].mux;
267 	int gain = st->channels_cfg[chan->address].gain;
268 	int datarate = st->channels_cfg[chan->address].datarate;
269 	unsigned int sample;
270 	int ret;
271 
272 	if (calib_offset)
273 		mux = ADS1119_MUX_SHORTED;
274 
275 	ret = pm_runtime_resume_and_get(dev);
276 	if (ret)
277 		return ret;
278 
279 	ret = ads1119_configure_channel(st, mux, gain, datarate);
280 	if (ret)
281 		goto pdown;
282 
283 	if (st->client->irq)
284 		reinit_completion(&st->completion);
285 
286 	ret = i2c_smbus_write_byte(st->client, ADS1119_CMD_START_SYNC);
287 	if (ret)
288 		goto pdown;
289 
290 	ret = ads1119_read_data(st, chan, &sample);
291 	if (ret)
292 		goto pdown;
293 
294 	*val = sign_extend32(sample, chan->scan_type.realbits - 1);
295 	ret = IIO_VAL_INT;
296 pdown:
297 	pm_runtime_put_autosuspend(dev);
298 	return ret;
299 }
300 
ads1119_validate_datarate(struct ads1119_state * st,int datarate)301 static int ads1119_validate_datarate(struct ads1119_state *st, int datarate)
302 {
303 	switch (datarate) {
304 	case 20:
305 	case 90:
306 	case 330:
307 	case 1000:
308 		return datarate;
309 	default:
310 		return -EINVAL;
311 	}
312 }
313 
ads1119_read_avail(struct iio_dev * indio_dev,struct iio_chan_spec const * chan,const int ** vals,int * type,int * length,long mask)314 static int ads1119_read_avail(struct iio_dev *indio_dev,
315 			      struct iio_chan_spec const *chan,
316 			      const int **vals, int *type, int *length,
317 			      long mask)
318 {
319 	switch (mask) {
320 	case IIO_CHAN_INFO_SCALE:
321 		*type = IIO_VAL_FRACTIONAL;
322 		*vals = ads1119_available_gains;
323 		*length = ARRAY_SIZE(ads1119_available_gains);
324 		return IIO_AVAIL_LIST;
325 	case IIO_CHAN_INFO_SAMP_FREQ:
326 		*type = IIO_VAL_INT;
327 		*vals = ads1119_available_datarates;
328 		*length = ARRAY_SIZE(ads1119_available_datarates);
329 		return IIO_AVAIL_LIST;
330 	default:
331 		return -EINVAL;
332 	}
333 }
334 
ads1119_read_raw(struct iio_dev * indio_dev,struct iio_chan_spec const * chan,int * val,int * val2,long mask)335 static int ads1119_read_raw(struct iio_dev *indio_dev,
336 			    struct iio_chan_spec const *chan, int *val,
337 			    int *val2, long mask)
338 {
339 	struct ads1119_state *st = iio_priv(indio_dev);
340 	unsigned int index = chan->address;
341 	int ret;
342 
343 	if (index >= st->num_channels_cfg)
344 		return -EINVAL;
345 
346 	switch (mask) {
347 	case IIO_CHAN_INFO_RAW:
348 		if (!iio_device_claim_direct(indio_dev))
349 			return -EBUSY;
350 		ret = ads1119_single_conversion(st, chan, val, false);
351 		iio_device_release_direct(indio_dev);
352 		return ret;
353 	case IIO_CHAN_INFO_OFFSET:
354 		if (!iio_device_claim_direct(indio_dev))
355 			return -EBUSY;
356 		ret = ads1119_single_conversion(st, chan, val, true);
357 		iio_device_release_direct(indio_dev);
358 		return ret;
359 	case IIO_CHAN_INFO_SCALE:
360 		*val = st->vref_uV / 1000;
361 		*val /= st->channels_cfg[index].gain;
362 		*val2 = chan->scan_type.realbits - 1;
363 		return IIO_VAL_FRACTIONAL_LOG2;
364 	case IIO_CHAN_INFO_SAMP_FREQ:
365 		*val = st->channels_cfg[index].datarate;
366 		return IIO_VAL_INT;
367 	default:
368 		return -EINVAL;
369 	}
370 }
371 
ads1119_write_raw(struct iio_dev * indio_dev,struct iio_chan_spec const * chan,int val,int val2,long mask)372 static int ads1119_write_raw(struct iio_dev *indio_dev,
373 			     struct iio_chan_spec const *chan, int val,
374 			     int val2, long mask)
375 {
376 	struct ads1119_state *st = iio_priv(indio_dev);
377 	unsigned int index = chan->address;
378 	int ret;
379 
380 	if (index >= st->num_channels_cfg)
381 		return -EINVAL;
382 
383 	switch (mask) {
384 	case IIO_CHAN_INFO_SCALE:
385 		ret = MICRO / ((val * MICRO) + val2);
386 		if (ret != 1 && ret != 4)
387 			return -EINVAL;
388 
389 		st->channels_cfg[index].gain = ret;
390 		return 0;
391 	case IIO_CHAN_INFO_SAMP_FREQ:
392 		ret = ads1119_validate_datarate(st, val);
393 		if (ret < 0)
394 			return ret;
395 
396 		st->channels_cfg[index].datarate = ret;
397 		return 0;
398 	default:
399 		return -EINVAL;
400 	}
401 }
402 
ads1119_debugfs_reg_access(struct iio_dev * indio_dev,unsigned int reg,unsigned int writeval,unsigned int * readval)403 static int ads1119_debugfs_reg_access(struct iio_dev *indio_dev,
404 				      unsigned int reg, unsigned int writeval,
405 				      unsigned int *readval)
406 {
407 	struct ads1119_state *st = iio_priv(indio_dev);
408 	int ret;
409 
410 	if (reg > ADS1119_REG_STATUS)
411 		return -EINVAL;
412 
413 	if (readval) {
414 		ret = i2c_smbus_read_byte_data(st->client,
415 					       ADS1119_CMD_RREG(reg));
416 		if (ret < 0)
417 			return ret;
418 
419 		*readval = ret;
420 		return 0;
421 	}
422 
423 	if (reg > ADS1119_REG_CONFIG)
424 		return -EINVAL;
425 
426 	return i2c_smbus_write_byte_data(st->client, ADS1119_CMD_WREG,
427 					 writeval);
428 }
429 
430 static const struct iio_info ads1119_info = {
431 	.read_avail = ads1119_read_avail,
432 	.read_raw = ads1119_read_raw,
433 	.write_raw = ads1119_write_raw,
434 	.debugfs_reg_access = ads1119_debugfs_reg_access,
435 };
436 
ads1119_triggered_buffer_preenable(struct iio_dev * indio_dev)437 static int ads1119_triggered_buffer_preenable(struct iio_dev *indio_dev)
438 {
439 	struct ads1119_state *st = iio_priv(indio_dev);
440 	struct device *dev = &st->client->dev;
441 	unsigned int index;
442 	int ret;
443 
444 	index = find_first_bit(indio_dev->active_scan_mask,
445 			       iio_get_masklength(indio_dev));
446 
447 	ret = ads1119_set_conv_mode(st, true);
448 	if (ret)
449 		return ret;
450 
451 	ret = ads1119_configure_channel(st,
452 					st->channels_cfg[index].mux,
453 					st->channels_cfg[index].gain,
454 					st->channels_cfg[index].datarate);
455 	if (ret)
456 		return ret;
457 
458 	ret = pm_runtime_resume_and_get(dev);
459 	if (ret)
460 		return ret;
461 
462 	ret = i2c_smbus_write_byte(st->client, ADS1119_CMD_START_SYNC);
463 	if (ret)
464 		pm_runtime_put_autosuspend(dev);
465 
466 	return ret;
467 }
468 
ads1119_triggered_buffer_postdisable(struct iio_dev * indio_dev)469 static int ads1119_triggered_buffer_postdisable(struct iio_dev *indio_dev)
470 {
471 	struct ads1119_state *st = iio_priv(indio_dev);
472 	struct device *dev = &st->client->dev;
473 	int ret;
474 
475 	ret = ads1119_set_conv_mode(st, false);
476 	if (ret)
477 		return ret;
478 
479 	pm_runtime_put_autosuspend(dev);
480 
481 	return 0;
482 }
483 
484 static const struct iio_buffer_setup_ops ads1119_buffer_setup_ops = {
485 	.preenable = ads1119_triggered_buffer_preenable,
486 	.postdisable = ads1119_triggered_buffer_postdisable,
487 	.validate_scan_mask = &iio_validate_scan_mask_onehot,
488 };
489 
490 static const struct iio_trigger_ops ads1119_trigger_ops = {
491 	.validate_device = &iio_trigger_validate_own_device,
492 };
493 
ads1119_irq_handler(int irq,void * dev_id)494 static irqreturn_t ads1119_irq_handler(int irq, void *dev_id)
495 {
496 	struct iio_dev *indio_dev = dev_id;
497 	struct ads1119_state *st = iio_priv(indio_dev);
498 
499 	if (iio_buffer_enabled(indio_dev) && iio_trigger_using_own(indio_dev))
500 		iio_trigger_poll(indio_dev->trig);
501 	else
502 		complete(&st->completion);
503 
504 	return IRQ_HANDLED;
505 }
506 
ads1119_trigger_handler(int irq,void * private)507 static irqreturn_t ads1119_trigger_handler(int irq, void *private)
508 {
509 	struct iio_poll_func *pf = private;
510 	struct iio_dev *indio_dev = pf->indio_dev;
511 	struct ads1119_state *st = iio_priv(indio_dev);
512 	struct {
513 		s16 sample;
514 		aligned_s64 timestamp;
515 	} scan = { };
516 	unsigned int index;
517 	int ret;
518 
519 	if (!iio_trigger_using_own(indio_dev)) {
520 		index = find_first_bit(indio_dev->active_scan_mask,
521 				       iio_get_masklength(indio_dev));
522 
523 		ret = ads1119_poll_data_ready(st, &indio_dev->channels[index]);
524 		if (ret) {
525 			dev_err(&st->client->dev,
526 				"Failed to poll data on trigger (%d)\n", ret);
527 			goto done;
528 		}
529 	}
530 
531 	ret = i2c_smbus_read_word_swapped(st->client, ADS1119_CMD_RDATA);
532 	if (ret < 0) {
533 		dev_err(&st->client->dev,
534 			"Failed to read data on trigger (%d)\n", ret);
535 		goto done;
536 	}
537 
538 	scan.sample = ret;
539 
540 	iio_push_to_buffers_with_ts(indio_dev, &scan, sizeof(scan),
541 				    iio_get_time_ns(indio_dev));
542 done:
543 	iio_trigger_notify_done(indio_dev->trig);
544 	return IRQ_HANDLED;
545 }
546 
ads1119_init(struct ads1119_state * st,bool vref_external)547 static int ads1119_init(struct ads1119_state *st, bool vref_external)
548 {
549 	int ret;
550 
551 	ret = ads1119_reset(st);
552 	if (ret)
553 		return ret;
554 
555 	if (vref_external)
556 		return ads1119_upd_cfg_reg(st,
557 					   ADS1119_CONFIG_VREF_FIELD,
558 					   FIELD_PREP(ADS1119_CONFIG_VREF_FIELD,
559 						      ADS1119_VREF_EXTERNAL));
560 	return 0;
561 }
562 
ads1119_map_analog_inputs_mux(int ain_pos,int ain_neg,bool differential)563 static int ads1119_map_analog_inputs_mux(int ain_pos, int ain_neg,
564 					 bool differential)
565 {
566 	if (ain_pos >= ADS1119_MAX_SINGLE_CHANNELS)
567 		return -EINVAL;
568 
569 	if (!differential)
570 		return ADS1119_MUX_AIN0 + ain_pos;
571 
572 	if (ain_pos == 0 && ain_neg == 1)
573 		return ADS1119_MUX_AIN0_AIN1;
574 	else if (ain_pos == 1 && ain_neg == 2)
575 		return ADS1119_MUX_AIN1_AIN2;
576 	else if (ain_pos == 2 && ain_neg == 3)
577 		return ADS1119_MUX_AIN2_AIN3;
578 
579 	return -EINVAL;
580 }
581 
ads1119_alloc_and_config_channels(struct iio_dev * indio_dev)582 static int ads1119_alloc_and_config_channels(struct iio_dev *indio_dev)
583 {
584 	const struct iio_chan_spec ads1119_channel =
585 		(const struct iio_chan_spec) {
586 		.type = IIO_VOLTAGE,
587 		.indexed = 1,
588 		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
589 		BIT(IIO_CHAN_INFO_SCALE) |
590 		BIT(IIO_CHAN_INFO_OFFSET) |
591 		BIT(IIO_CHAN_INFO_SAMP_FREQ),
592 		.info_mask_shared_by_all_available =
593 		BIT(IIO_CHAN_INFO_SCALE) |
594 		BIT(IIO_CHAN_INFO_SAMP_FREQ),
595 		.scan_type = {
596 			.sign = 's',
597 			.realbits = 16,
598 			.storagebits = 16,
599 			.endianness = IIO_CPU,
600 		},
601 	};
602 	const struct iio_chan_spec ads1119_ts = IIO_CHAN_SOFT_TIMESTAMP(0);
603 	struct ads1119_state *st = iio_priv(indio_dev);
604 	struct iio_chan_spec *iio_channels, *chan;
605 	struct device *dev = &st->client->dev;
606 	unsigned int num_channels, i;
607 	bool differential;
608 	u32 ain[2];
609 	int ret;
610 
611 	st->num_channels_cfg = device_get_child_node_count(dev);
612 	if (st->num_channels_cfg > ADS1119_MAX_CHANNELS)
613 		return dev_err_probe(dev, -EINVAL,
614 				     "Too many channels %d, max is %d\n",
615 				     st->num_channels_cfg,
616 				     ADS1119_MAX_CHANNELS);
617 
618 	st->channels_cfg = devm_kcalloc(dev, st->num_channels_cfg,
619 					sizeof(*st->channels_cfg), GFP_KERNEL);
620 	if (!st->channels_cfg)
621 		return -ENOMEM;
622 
623 	/* Allocate one more iio channel for the timestamp */
624 	num_channels = st->num_channels_cfg + 1;
625 	iio_channels = devm_kcalloc(dev, num_channels, sizeof(*iio_channels),
626 				    GFP_KERNEL);
627 	if (!iio_channels)
628 		return -ENOMEM;
629 
630 	i = 0;
631 
632 	device_for_each_child_node_scoped(dev, child) {
633 		chan = &iio_channels[i];
634 
635 		differential = fwnode_property_present(child, "diff-channels");
636 		if (differential)
637 			ret = fwnode_property_read_u32_array(child,
638 							     "diff-channels",
639 							     ain, 2);
640 		else
641 			ret = fwnode_property_read_u32(child, "single-channel",
642 						       &ain[0]);
643 
644 		if (ret)
645 			return dev_err_probe(dev, ret,
646 					     "Failed to get channel property\n");
647 
648 		ret = ads1119_map_analog_inputs_mux(ain[0], ain[1],
649 						    differential);
650 		if (ret < 0)
651 			return dev_err_probe(dev, ret,
652 					     "Invalid channel value\n");
653 
654 		st->channels_cfg[i].mux = ret;
655 		st->channels_cfg[i].gain = ADS1119_DEFAULT_GAIN;
656 		st->channels_cfg[i].datarate = ADS1119_DEFAULT_DATARATE;
657 
658 		*chan = ads1119_channel;
659 		chan->channel = ain[0];
660 		chan->address = i;
661 		chan->scan_index = i;
662 
663 		if (differential) {
664 			chan->channel2 = ain[1];
665 			chan->differential = 1;
666 		}
667 
668 		dev_dbg(dev, "channel: index %d, mux %d\n", i,
669 			st->channels_cfg[i].mux);
670 
671 		i++;
672 	}
673 
674 	iio_channels[i] = ads1119_ts;
675 	iio_channels[i].address = i;
676 	iio_channels[i].scan_index = i;
677 
678 	indio_dev->channels = iio_channels;
679 	indio_dev->num_channels = num_channels;
680 
681 	return 0;
682 }
683 
ads1119_powerdown(void * data)684 static void ads1119_powerdown(void *data)
685 {
686 	struct ads1119_state *st = data;
687 
688 	i2c_smbus_write_byte(st->client, ADS1119_CMD_POWERDOWN);
689 }
690 
ads1119_probe(struct i2c_client * client)691 static int ads1119_probe(struct i2c_client *client)
692 {
693 	struct iio_dev *indio_dev;
694 	struct ads1119_state *st;
695 	struct device *dev = &client->dev;
696 	bool vref_external = true;
697 	int ret;
698 
699 	indio_dev = devm_iio_device_alloc(dev, sizeof(*st));
700 	if (!indio_dev)
701 		return -ENOMEM;
702 
703 	st = iio_priv(indio_dev);
704 	st->client = client;
705 
706 	indio_dev->name = "ads1119";
707 	indio_dev->info = &ads1119_info;
708 	indio_dev->modes = INDIO_DIRECT_MODE;
709 
710 	i2c_set_clientdata(client, indio_dev);
711 
712 	ret = devm_regulator_bulk_get_enable(dev,
713 					     ARRAY_SIZE(ads1119_power_supplies),
714 					     ads1119_power_supplies);
715 	if (ret)
716 		return dev_err_probe(dev, ret,
717 				     "Failed to get and enable supplies\n");
718 
719 	st->vref_uV = devm_regulator_get_enable_read_voltage(dev, "vref");
720 	if (st->vref_uV == -ENODEV) {
721 		vref_external = false;
722 		st->vref_uV = ADS1119_VREF_INTERNAL_VAL;
723 	} else if (st->vref_uV < 0) {
724 		return dev_err_probe(dev, st->vref_uV, "Failed to get vref\n");
725 	}
726 
727 	st->reset_gpio = devm_gpiod_get_optional(dev, "reset", GPIOD_OUT_LOW);
728 	if (IS_ERR(st->reset_gpio))
729 		return dev_err_probe(dev, PTR_ERR(st->reset_gpio),
730 				     "Failed to get reset gpio\n");
731 
732 	ret = ads1119_alloc_and_config_channels(indio_dev);
733 	if (ret)
734 		return ret;
735 
736 	init_completion(&st->completion);
737 
738 	ret = devm_iio_triggered_buffer_setup(dev, indio_dev, NULL,
739 					      ads1119_trigger_handler,
740 					      &ads1119_buffer_setup_ops);
741 	if (ret)
742 		return dev_err_probe(dev, ret, "Failed to setup IIO buffer\n");
743 
744 	if (client->irq > 0) {
745 		ret = devm_request_irq(dev, client->irq, ads1119_irq_handler,
746 				       IRQF_NO_THREAD, "ads1119", indio_dev);
747 		if (ret)
748 			return ret;
749 
750 		st->trig = devm_iio_trigger_alloc(dev, "%s-dev%d",
751 						  indio_dev->name,
752 						  iio_device_id(indio_dev));
753 		if (!st->trig)
754 			return -ENOMEM;
755 
756 		st->trig->ops = &ads1119_trigger_ops;
757 		iio_trigger_set_drvdata(st->trig, indio_dev);
758 
759 		ret = devm_iio_trigger_register(dev, st->trig);
760 		if (ret)
761 			return dev_err_probe(dev, ret,
762 					     "Failed to register IIO trigger\n");
763 	}
764 
765 	ret = ads1119_init(st, vref_external);
766 	if (ret)
767 		return dev_err_probe(dev, ret,
768 				     "Failed to initialize device\n");
769 
770 	pm_runtime_set_autosuspend_delay(dev, ADS1119_SUSPEND_DELAY);
771 	pm_runtime_use_autosuspend(dev);
772 	pm_runtime_mark_last_busy(dev);
773 	pm_runtime_set_active(dev);
774 
775 	ret = devm_pm_runtime_enable(dev);
776 	if (ret)
777 		return dev_err_probe(dev, ret, "Failed to enable pm runtime\n");
778 
779 	ret = devm_add_action_or_reset(dev, ads1119_powerdown, st);
780 	if (ret)
781 		return ret;
782 
783 	return devm_iio_device_register(dev, indio_dev);
784 }
785 
ads1119_runtime_suspend(struct device * dev)786 static int ads1119_runtime_suspend(struct device *dev)
787 {
788 	struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
789 	struct ads1119_state *st = iio_priv(indio_dev);
790 
791 	return i2c_smbus_write_byte(st->client, ADS1119_CMD_POWERDOWN);
792 }
793 
794 /*
795  * The ADS1119 does not require a resume function because it automatically
796  * powers on after a reset.
797  * After a power down command, the ADS1119 can still communicate but turns off
798  * its analog parts. To resume from power down, the device will power up again
799  * upon receiving a start/sync command.
800  */
801 static DEFINE_RUNTIME_DEV_PM_OPS(ads1119_pm_ops, ads1119_runtime_suspend,
802 				 NULL, NULL);
803 
804 static const struct of_device_id __maybe_unused ads1119_of_match[] = {
805 	{ .compatible = "ti,ads1119" },
806 	{ }
807 };
808 MODULE_DEVICE_TABLE(of, ads1119_of_match);
809 
810 static const struct i2c_device_id ads1119_id[] = {
811 	{ .name = "ads1119" },
812 	{ }
813 };
814 MODULE_DEVICE_TABLE(i2c, ads1119_id);
815 
816 static struct i2c_driver ads1119_driver = {
817 	.driver = {
818 		.name = "ads1119",
819 		.of_match_table = ads1119_of_match,
820 		.pm = pm_ptr(&ads1119_pm_ops),
821 	},
822 	.probe = ads1119_probe,
823 	.id_table = ads1119_id,
824 };
825 module_i2c_driver(ads1119_driver);
826 
827 MODULE_AUTHOR("João Paulo Gonçalves <joao.goncalves@toradex.com>");
828 MODULE_DESCRIPTION("Texas Instruments ADS1119 ADC Driver");
829 MODULE_LICENSE("GPL");
830