1 // SPDX-License-Identifier: GPL-2.0
2 /* ADC driver for sunxi platforms' (A10, A13 and A31) GPADC
3  *
4  * Copyright (c) 2016 Quentin Schulz <quentin.schulz@free-electrons.com>
5  *
6  * The Allwinner SoCs all have an ADC that can also act as a touchscreen
7  * controller and a thermal sensor.
8  * The thermal sensor works only when the ADC acts as a touchscreen controller
9  * and is configured to throw an interrupt every fixed periods of time (let say
10  * every X seconds).
11  * One would be tempted to disable the IP on the hardware side rather than
12  * disabling interrupts to save some power but that resets the internal clock of
13  * the IP, resulting in having to wait X seconds every time we want to read the
14  * value of the thermal sensor.
15  * This is also the reason of using autosuspend in pm_runtime. If there was no
16  * autosuspend, the thermal sensor would need X seconds after every
17  * pm_runtime_get_sync to get a value from the ADC. The autosuspend allows the
18  * thermal sensor to be requested again in a certain time span before it gets
19  * shutdown for not being used.
20  */
21 
22 #include <linux/completion.h>
23 #include <linux/interrupt.h>
24 #include <linux/io.h>
25 #include <linux/module.h>
26 #include <linux/of.h>
27 #include <linux/of_device.h>
28 #include <linux/platform_device.h>
29 #include <linux/pm_runtime.h>
30 #include <linux/regmap.h>
31 #include <linux/thermal.h>
32 #include <linux/delay.h>
33 
34 #include <linux/iio/iio.h>
35 #include <linux/iio/driver.h>
36 #include <linux/iio/machine.h>
37 #include <linux/mfd/sun4i-gpadc.h>
38 
39 static unsigned int sun4i_gpadc_chan_select(unsigned int chan)
40 {
41 	return SUN4I_GPADC_CTRL1_ADC_CHAN_SELECT(chan);
42 }
43 
44 static unsigned int sun6i_gpadc_chan_select(unsigned int chan)
45 {
46 	return SUN6I_GPADC_CTRL1_ADC_CHAN_SELECT(chan);
47 }
48 
49 struct gpadc_data {
50 	int		temp_offset;
51 	int		temp_scale;
52 	unsigned int	tp_mode_en;
53 	unsigned int	tp_adc_select;
54 	unsigned int	(*adc_chan_select)(unsigned int chan);
55 	unsigned int	adc_chan_mask;
56 };
57 
58 static const struct gpadc_data sun4i_gpadc_data = {
59 	.temp_offset = -1932,
60 	.temp_scale = 133,
61 	.tp_mode_en = SUN4I_GPADC_CTRL1_TP_MODE_EN,
62 	.tp_adc_select = SUN4I_GPADC_CTRL1_TP_ADC_SELECT,
63 	.adc_chan_select = &sun4i_gpadc_chan_select,
64 	.adc_chan_mask = SUN4I_GPADC_CTRL1_ADC_CHAN_MASK,
65 };
66 
67 static const struct gpadc_data sun5i_gpadc_data = {
68 	.temp_offset = -1447,
69 	.temp_scale = 100,
70 	.tp_mode_en = SUN4I_GPADC_CTRL1_TP_MODE_EN,
71 	.tp_adc_select = SUN4I_GPADC_CTRL1_TP_ADC_SELECT,
72 	.adc_chan_select = &sun4i_gpadc_chan_select,
73 	.adc_chan_mask = SUN4I_GPADC_CTRL1_ADC_CHAN_MASK,
74 };
75 
76 static const struct gpadc_data sun6i_gpadc_data = {
77 	.temp_offset = -1623,
78 	.temp_scale = 167,
79 	.tp_mode_en = SUN6I_GPADC_CTRL1_TP_MODE_EN,
80 	.tp_adc_select = SUN6I_GPADC_CTRL1_TP_ADC_SELECT,
81 	.adc_chan_select = &sun6i_gpadc_chan_select,
82 	.adc_chan_mask = SUN6I_GPADC_CTRL1_ADC_CHAN_MASK,
83 };
84 
85 static const struct gpadc_data sun8i_a33_gpadc_data = {
86 	.temp_offset = -1662,
87 	.temp_scale = 162,
88 	.tp_mode_en = SUN8I_GPADC_CTRL1_CHOP_TEMP_EN,
89 };
90 
91 struct sun4i_gpadc_iio {
92 	struct iio_dev			*indio_dev;
93 	struct completion		completion;
94 	int				temp_data;
95 	u32				adc_data;
96 	struct regmap			*regmap;
97 	unsigned int			fifo_data_irq;
98 	atomic_t			ignore_fifo_data_irq;
99 	unsigned int			temp_data_irq;
100 	atomic_t			ignore_temp_data_irq;
101 	const struct gpadc_data		*data;
102 	bool				no_irq;
103 	/* prevents concurrent reads of temperature and ADC */
104 	struct mutex			mutex;
105 	struct thermal_zone_device	*tzd;
106 	struct device			*sensor_device;
107 };
108 
109 #define SUN4I_GPADC_ADC_CHANNEL(_channel, _name) {		\
110 	.type = IIO_VOLTAGE,					\
111 	.indexed = 1,						\
112 	.channel = _channel,					\
113 	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),		\
114 	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),	\
115 	.datasheet_name = _name,				\
116 }
117 
118 static struct iio_map sun4i_gpadc_hwmon_maps[] = {
119 	{
120 		.adc_channel_label = "temp_adc",
121 		.consumer_dev_name = "iio_hwmon.0",
122 	},
123 	{ /* sentinel */ },
124 };
125 
126 static const struct iio_chan_spec sun4i_gpadc_channels[] = {
127 	SUN4I_GPADC_ADC_CHANNEL(0, "adc_chan0"),
128 	SUN4I_GPADC_ADC_CHANNEL(1, "adc_chan1"),
129 	SUN4I_GPADC_ADC_CHANNEL(2, "adc_chan2"),
130 	SUN4I_GPADC_ADC_CHANNEL(3, "adc_chan3"),
131 	{
132 		.type = IIO_TEMP,
133 		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
134 				      BIT(IIO_CHAN_INFO_SCALE) |
135 				      BIT(IIO_CHAN_INFO_OFFSET),
136 		.datasheet_name = "temp_adc",
137 	},
138 };
139 
140 static const struct iio_chan_spec sun4i_gpadc_channels_no_temp[] = {
141 	SUN4I_GPADC_ADC_CHANNEL(0, "adc_chan0"),
142 	SUN4I_GPADC_ADC_CHANNEL(1, "adc_chan1"),
143 	SUN4I_GPADC_ADC_CHANNEL(2, "adc_chan2"),
144 	SUN4I_GPADC_ADC_CHANNEL(3, "adc_chan3"),
145 };
146 
147 static const struct iio_chan_spec sun8i_a33_gpadc_channels[] = {
148 	{
149 		.type = IIO_TEMP,
150 		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
151 				      BIT(IIO_CHAN_INFO_SCALE) |
152 				      BIT(IIO_CHAN_INFO_OFFSET),
153 		.datasheet_name = "temp_adc",
154 	},
155 };
156 
157 static const struct regmap_config sun4i_gpadc_regmap_config = {
158 	.reg_bits = 32,
159 	.val_bits = 32,
160 	.reg_stride = 4,
161 	.fast_io = true,
162 };
163 
164 static int sun4i_prepare_for_irq(struct iio_dev *indio_dev, int channel,
165 				 unsigned int irq)
166 {
167 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
168 	int ret;
169 	u32 reg;
170 
171 	pm_runtime_get_sync(indio_dev->dev.parent);
172 
173 	reinit_completion(&info->completion);
174 
175 	ret = regmap_write(info->regmap, SUN4I_GPADC_INT_FIFOC,
176 			   SUN4I_GPADC_INT_FIFOC_TP_FIFO_TRIG_LEVEL(1) |
177 			   SUN4I_GPADC_INT_FIFOC_TP_FIFO_FLUSH);
178 	if (ret)
179 		return ret;
180 
181 	ret = regmap_read(info->regmap, SUN4I_GPADC_CTRL1, &reg);
182 	if (ret)
183 		return ret;
184 
185 	if (irq == info->fifo_data_irq) {
186 		ret = regmap_write(info->regmap, SUN4I_GPADC_CTRL1,
187 				   info->data->tp_mode_en |
188 				   info->data->tp_adc_select |
189 				   info->data->adc_chan_select(channel));
190 		/*
191 		 * When the IP changes channel, it needs a bit of time to get
192 		 * correct values.
193 		 */
194 		if ((reg & info->data->adc_chan_mask) !=
195 			 info->data->adc_chan_select(channel))
196 			mdelay(10);
197 
198 	} else {
199 		/*
200 		 * The temperature sensor returns valid data only when the ADC
201 		 * operates in touchscreen mode.
202 		 */
203 		ret = regmap_write(info->regmap, SUN4I_GPADC_CTRL1,
204 				   info->data->tp_mode_en);
205 	}
206 
207 	if (ret)
208 		return ret;
209 
210 	/*
211 	 * When the IP changes mode between ADC or touchscreen, it
212 	 * needs a bit of time to get correct values.
213 	 */
214 	if ((reg & info->data->tp_adc_select) != info->data->tp_adc_select)
215 		mdelay(100);
216 
217 	return 0;
218 }
219 
220 static int sun4i_gpadc_read(struct iio_dev *indio_dev, int channel, int *val,
221 			    unsigned int irq)
222 {
223 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
224 	int ret;
225 
226 	mutex_lock(&info->mutex);
227 
228 	ret = sun4i_prepare_for_irq(indio_dev, channel, irq);
229 	if (ret)
230 		goto err;
231 
232 	enable_irq(irq);
233 
234 	/*
235 	 * The temperature sensor throws an interruption periodically (currently
236 	 * set at periods of ~0.6s in sun4i_gpadc_runtime_resume). A 1s delay
237 	 * makes sure an interruption occurs in normal conditions. If it doesn't
238 	 * occur, then there is a timeout.
239 	 */
240 	if (!wait_for_completion_timeout(&info->completion,
241 					 msecs_to_jiffies(1000))) {
242 		ret = -ETIMEDOUT;
243 		goto err;
244 	}
245 
246 	if (irq == info->fifo_data_irq)
247 		*val = info->adc_data;
248 	else
249 		*val = info->temp_data;
250 
251 	ret = 0;
252 	pm_runtime_mark_last_busy(indio_dev->dev.parent);
253 
254 err:
255 	pm_runtime_put_autosuspend(indio_dev->dev.parent);
256 	disable_irq(irq);
257 	mutex_unlock(&info->mutex);
258 
259 	return ret;
260 }
261 
262 static int sun4i_gpadc_adc_read(struct iio_dev *indio_dev, int channel,
263 				int *val)
264 {
265 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
266 
267 	return sun4i_gpadc_read(indio_dev, channel, val, info->fifo_data_irq);
268 }
269 
270 static int sun4i_gpadc_temp_read(struct iio_dev *indio_dev, int *val)
271 {
272 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
273 
274 	if (info->no_irq) {
275 		pm_runtime_get_sync(indio_dev->dev.parent);
276 
277 		regmap_read(info->regmap, SUN4I_GPADC_TEMP_DATA, val);
278 
279 		pm_runtime_mark_last_busy(indio_dev->dev.parent);
280 		pm_runtime_put_autosuspend(indio_dev->dev.parent);
281 
282 		return 0;
283 	}
284 
285 	return sun4i_gpadc_read(indio_dev, 0, val, info->temp_data_irq);
286 }
287 
288 static int sun4i_gpadc_temp_offset(struct iio_dev *indio_dev, int *val)
289 {
290 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
291 
292 	*val = info->data->temp_offset;
293 
294 	return 0;
295 }
296 
297 static int sun4i_gpadc_temp_scale(struct iio_dev *indio_dev, int *val)
298 {
299 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
300 
301 	*val = info->data->temp_scale;
302 
303 	return 0;
304 }
305 
306 static int sun4i_gpadc_read_raw(struct iio_dev *indio_dev,
307 				struct iio_chan_spec const *chan, int *val,
308 				int *val2, long mask)
309 {
310 	int ret;
311 
312 	switch (mask) {
313 	case IIO_CHAN_INFO_OFFSET:
314 		ret = sun4i_gpadc_temp_offset(indio_dev, val);
315 		if (ret)
316 			return ret;
317 
318 		return IIO_VAL_INT;
319 	case IIO_CHAN_INFO_RAW:
320 		if (chan->type == IIO_VOLTAGE)
321 			ret = sun4i_gpadc_adc_read(indio_dev, chan->channel,
322 						   val);
323 		else
324 			ret = sun4i_gpadc_temp_read(indio_dev, val);
325 
326 		if (ret)
327 			return ret;
328 
329 		return IIO_VAL_INT;
330 	case IIO_CHAN_INFO_SCALE:
331 		if (chan->type == IIO_VOLTAGE) {
332 			/* 3000mV / 4096 * raw */
333 			*val = 0;
334 			*val2 = 732421875;
335 			return IIO_VAL_INT_PLUS_NANO;
336 		}
337 
338 		ret = sun4i_gpadc_temp_scale(indio_dev, val);
339 		if (ret)
340 			return ret;
341 
342 		return IIO_VAL_INT;
343 	default:
344 		return -EINVAL;
345 	}
346 
347 	return -EINVAL;
348 }
349 
350 static const struct iio_info sun4i_gpadc_iio_info = {
351 	.read_raw = sun4i_gpadc_read_raw,
352 };
353 
354 static irqreturn_t sun4i_gpadc_temp_data_irq_handler(int irq, void *dev_id)
355 {
356 	struct sun4i_gpadc_iio *info = dev_id;
357 
358 	if (atomic_read(&info->ignore_temp_data_irq))
359 		goto out;
360 
361 	if (!regmap_read(info->regmap, SUN4I_GPADC_TEMP_DATA, &info->temp_data))
362 		complete(&info->completion);
363 
364 out:
365 	return IRQ_HANDLED;
366 }
367 
368 static irqreturn_t sun4i_gpadc_fifo_data_irq_handler(int irq, void *dev_id)
369 {
370 	struct sun4i_gpadc_iio *info = dev_id;
371 
372 	if (atomic_read(&info->ignore_fifo_data_irq))
373 		goto out;
374 
375 	if (!regmap_read(info->regmap, SUN4I_GPADC_DATA, &info->adc_data))
376 		complete(&info->completion);
377 
378 out:
379 	return IRQ_HANDLED;
380 }
381 
382 static int sun4i_gpadc_runtime_suspend(struct device *dev)
383 {
384 	struct sun4i_gpadc_iio *info = iio_priv(dev_get_drvdata(dev));
385 
386 	/* Disable the ADC on IP */
387 	regmap_write(info->regmap, SUN4I_GPADC_CTRL1, 0);
388 	/* Disable temperature sensor on IP */
389 	regmap_write(info->regmap, SUN4I_GPADC_TPR, 0);
390 
391 	return 0;
392 }
393 
394 static int sun4i_gpadc_runtime_resume(struct device *dev)
395 {
396 	struct sun4i_gpadc_iio *info = iio_priv(dev_get_drvdata(dev));
397 
398 	/* clkin = 6MHz */
399 	regmap_write(info->regmap, SUN4I_GPADC_CTRL0,
400 		     SUN4I_GPADC_CTRL0_ADC_CLK_DIVIDER(2) |
401 		     SUN4I_GPADC_CTRL0_FS_DIV(7) |
402 		     SUN4I_GPADC_CTRL0_T_ACQ(63));
403 	regmap_write(info->regmap, SUN4I_GPADC_CTRL1, info->data->tp_mode_en);
404 	regmap_write(info->regmap, SUN4I_GPADC_CTRL3,
405 		     SUN4I_GPADC_CTRL3_FILTER_EN |
406 		     SUN4I_GPADC_CTRL3_FILTER_TYPE(1));
407 	/* period = SUN4I_GPADC_TPR_TEMP_PERIOD * 256 * 16 / clkin; ~0.6s */
408 	regmap_write(info->regmap, SUN4I_GPADC_TPR,
409 		     SUN4I_GPADC_TPR_TEMP_ENABLE |
410 		     SUN4I_GPADC_TPR_TEMP_PERIOD(800));
411 
412 	return 0;
413 }
414 
415 static int sun4i_gpadc_get_temp(void *data, int *temp)
416 {
417 	struct sun4i_gpadc_iio *info = data;
418 	int val, scale, offset;
419 
420 	if (sun4i_gpadc_temp_read(info->indio_dev, &val))
421 		return -ETIMEDOUT;
422 
423 	sun4i_gpadc_temp_scale(info->indio_dev, &scale);
424 	sun4i_gpadc_temp_offset(info->indio_dev, &offset);
425 
426 	*temp = (val + offset) * scale;
427 
428 	return 0;
429 }
430 
431 static const struct thermal_zone_of_device_ops sun4i_ts_tz_ops = {
432 	.get_temp = &sun4i_gpadc_get_temp,
433 };
434 
435 static const struct dev_pm_ops sun4i_gpadc_pm_ops = {
436 	.runtime_suspend = &sun4i_gpadc_runtime_suspend,
437 	.runtime_resume = &sun4i_gpadc_runtime_resume,
438 };
439 
440 static int sun4i_irq_init(struct platform_device *pdev, const char *name,
441 			  irq_handler_t handler, const char *devname,
442 			  unsigned int *irq, atomic_t *atomic)
443 {
444 	int ret;
445 	struct sun4i_gpadc_dev *mfd_dev = dev_get_drvdata(pdev->dev.parent);
446 	struct sun4i_gpadc_iio *info = iio_priv(dev_get_drvdata(&pdev->dev));
447 
448 	/*
449 	 * Once the interrupt is activated, the IP continuously performs
450 	 * conversions thus throws interrupts. The interrupt is activated right
451 	 * after being requested but we want to control when these interrupts
452 	 * occur thus we disable it right after being requested. However, an
453 	 * interrupt might occur between these two instructions and we have to
454 	 * make sure that does not happen, by using atomic flags. We set the
455 	 * flag before requesting the interrupt and unset it right after
456 	 * disabling the interrupt. When an interrupt occurs between these two
457 	 * instructions, reading the atomic flag will tell us to ignore the
458 	 * interrupt.
459 	 */
460 	atomic_set(atomic, 1);
461 
462 	ret = platform_get_irq_byname(pdev, name);
463 	if (ret < 0)
464 		return ret;
465 
466 	ret = regmap_irq_get_virq(mfd_dev->regmap_irqc, ret);
467 	if (ret < 0) {
468 		dev_err(&pdev->dev, "failed to get virq for irq %s\n", name);
469 		return ret;
470 	}
471 
472 	*irq = ret;
473 	ret = devm_request_any_context_irq(&pdev->dev, *irq, handler, 0,
474 					   devname, info);
475 	if (ret < 0) {
476 		dev_err(&pdev->dev, "could not request %s interrupt: %d\n",
477 			name, ret);
478 		return ret;
479 	}
480 
481 	disable_irq(*irq);
482 	atomic_set(atomic, 0);
483 
484 	return 0;
485 }
486 
487 static const struct of_device_id sun4i_gpadc_of_id[] = {
488 	{
489 		.compatible = "allwinner,sun8i-a33-ths",
490 		.data = &sun8i_a33_gpadc_data,
491 	},
492 	{ /* sentinel */ }
493 };
494 
495 static int sun4i_gpadc_probe_dt(struct platform_device *pdev,
496 				struct iio_dev *indio_dev)
497 {
498 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
499 	void __iomem *base;
500 	int ret;
501 
502 	info->data = of_device_get_match_data(&pdev->dev);
503 	if (!info->data)
504 		return -ENODEV;
505 
506 	info->no_irq = true;
507 	indio_dev->num_channels = ARRAY_SIZE(sun8i_a33_gpadc_channels);
508 	indio_dev->channels = sun8i_a33_gpadc_channels;
509 
510 	base = devm_platform_ioremap_resource(pdev, 0);
511 	if (IS_ERR(base))
512 		return PTR_ERR(base);
513 
514 	info->regmap = devm_regmap_init_mmio(&pdev->dev, base,
515 					     &sun4i_gpadc_regmap_config);
516 	if (IS_ERR(info->regmap)) {
517 		ret = PTR_ERR(info->regmap);
518 		dev_err(&pdev->dev, "failed to init regmap: %d\n", ret);
519 		return ret;
520 	}
521 
522 	if (IS_ENABLED(CONFIG_THERMAL_OF))
523 		info->sensor_device = &pdev->dev;
524 
525 	return 0;
526 }
527 
528 static int sun4i_gpadc_probe_mfd(struct platform_device *pdev,
529 				 struct iio_dev *indio_dev)
530 {
531 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
532 	struct sun4i_gpadc_dev *sun4i_gpadc_dev =
533 		dev_get_drvdata(pdev->dev.parent);
534 	int ret;
535 
536 	info->no_irq = false;
537 	info->regmap = sun4i_gpadc_dev->regmap;
538 
539 	indio_dev->num_channels = ARRAY_SIZE(sun4i_gpadc_channels);
540 	indio_dev->channels = sun4i_gpadc_channels;
541 
542 	info->data = (struct gpadc_data *)platform_get_device_id(pdev)->driver_data;
543 
544 	/*
545 	 * Since the controller needs to be in touchscreen mode for its thermal
546 	 * sensor to operate properly, and that switching between the two modes
547 	 * needs a delay, always registering in the thermal framework will
548 	 * significantly slow down the conversion rate of the ADCs.
549 	 *
550 	 * Therefore, instead of depending on THERMAL_OF in Kconfig, we only
551 	 * register the sensor if that option is enabled, eventually leaving
552 	 * that choice to the user.
553 	 */
554 
555 	if (IS_ENABLED(CONFIG_THERMAL_OF)) {
556 		/*
557 		 * This driver is a child of an MFD which has a node in the DT
558 		 * but not its children, because of DT backward compatibility
559 		 * for A10, A13 and A31 SoCs. Therefore, the resulting devices
560 		 * of this driver do not have an of_node variable.
561 		 * However, its parent (the MFD driver) has an of_node variable
562 		 * and since devm_thermal_zone_of_sensor_register uses its first
563 		 * argument to match the phandle defined in the node of the
564 		 * thermal driver with the of_node of the device passed as first
565 		 * argument and the third argument to call ops from
566 		 * thermal_zone_of_device_ops, the solution is to use the parent
567 		 * device as first argument to match the phandle with its
568 		 * of_node, and the device from this driver as third argument to
569 		 * return the temperature.
570 		 */
571 		info->sensor_device = pdev->dev.parent;
572 	} else {
573 		indio_dev->num_channels =
574 			ARRAY_SIZE(sun4i_gpadc_channels_no_temp);
575 		indio_dev->channels = sun4i_gpadc_channels_no_temp;
576 	}
577 
578 	if (IS_ENABLED(CONFIG_THERMAL_OF)) {
579 		ret = sun4i_irq_init(pdev, "TEMP_DATA_PENDING",
580 				     sun4i_gpadc_temp_data_irq_handler,
581 				     "temp_data", &info->temp_data_irq,
582 				     &info->ignore_temp_data_irq);
583 		if (ret < 0)
584 			return ret;
585 	}
586 
587 	ret = sun4i_irq_init(pdev, "FIFO_DATA_PENDING",
588 			     sun4i_gpadc_fifo_data_irq_handler, "fifo_data",
589 			     &info->fifo_data_irq, &info->ignore_fifo_data_irq);
590 	if (ret < 0)
591 		return ret;
592 
593 	if (IS_ENABLED(CONFIG_THERMAL_OF)) {
594 		ret = iio_map_array_register(indio_dev, sun4i_gpadc_hwmon_maps);
595 		if (ret < 0) {
596 			dev_err(&pdev->dev,
597 				"failed to register iio map array\n");
598 			return ret;
599 		}
600 	}
601 
602 	return 0;
603 }
604 
605 static int sun4i_gpadc_probe(struct platform_device *pdev)
606 {
607 	struct sun4i_gpadc_iio *info;
608 	struct iio_dev *indio_dev;
609 	int ret;
610 
611 	indio_dev = devm_iio_device_alloc(&pdev->dev, sizeof(*info));
612 	if (!indio_dev)
613 		return -ENOMEM;
614 
615 	info = iio_priv(indio_dev);
616 	platform_set_drvdata(pdev, indio_dev);
617 
618 	mutex_init(&info->mutex);
619 	info->indio_dev = indio_dev;
620 	init_completion(&info->completion);
621 	indio_dev->name = dev_name(&pdev->dev);
622 	indio_dev->info = &sun4i_gpadc_iio_info;
623 	indio_dev->modes = INDIO_DIRECT_MODE;
624 
625 	if (pdev->dev.of_node)
626 		ret = sun4i_gpadc_probe_dt(pdev, indio_dev);
627 	else
628 		ret = sun4i_gpadc_probe_mfd(pdev, indio_dev);
629 
630 	if (ret)
631 		return ret;
632 
633 	pm_runtime_set_autosuspend_delay(&pdev->dev,
634 					 SUN4I_GPADC_AUTOSUSPEND_DELAY);
635 	pm_runtime_use_autosuspend(&pdev->dev);
636 	pm_runtime_set_suspended(&pdev->dev);
637 	pm_runtime_enable(&pdev->dev);
638 
639 	if (IS_ENABLED(CONFIG_THERMAL_OF)) {
640 		info->tzd = thermal_zone_of_sensor_register(info->sensor_device,
641 							    0, info,
642 							    &sun4i_ts_tz_ops);
643 		/*
644 		 * Do not fail driver probing when failing to register in
645 		 * thermal because no thermal DT node is found.
646 		 */
647 		if (IS_ERR(info->tzd) && PTR_ERR(info->tzd) != -ENODEV) {
648 			dev_err(&pdev->dev,
649 				"could not register thermal sensor: %ld\n",
650 				PTR_ERR(info->tzd));
651 			return PTR_ERR(info->tzd);
652 		}
653 	}
654 
655 	ret = devm_iio_device_register(&pdev->dev, indio_dev);
656 	if (ret < 0) {
657 		dev_err(&pdev->dev, "could not register the device\n");
658 		goto err_map;
659 	}
660 
661 	return 0;
662 
663 err_map:
664 	if (!info->no_irq && IS_ENABLED(CONFIG_THERMAL_OF))
665 		iio_map_array_unregister(indio_dev);
666 
667 	pm_runtime_put(&pdev->dev);
668 	pm_runtime_disable(&pdev->dev);
669 
670 	return ret;
671 }
672 
673 static int sun4i_gpadc_remove(struct platform_device *pdev)
674 {
675 	struct iio_dev *indio_dev = platform_get_drvdata(pdev);
676 	struct sun4i_gpadc_iio *info = iio_priv(indio_dev);
677 
678 	pm_runtime_put(&pdev->dev);
679 	pm_runtime_disable(&pdev->dev);
680 
681 	if (!IS_ENABLED(CONFIG_THERMAL_OF))
682 		return 0;
683 
684 	thermal_zone_of_sensor_unregister(info->sensor_device, info->tzd);
685 
686 	if (!info->no_irq)
687 		iio_map_array_unregister(indio_dev);
688 
689 	return 0;
690 }
691 
692 static const struct platform_device_id sun4i_gpadc_id[] = {
693 	{ "sun4i-a10-gpadc-iio", (kernel_ulong_t)&sun4i_gpadc_data },
694 	{ "sun5i-a13-gpadc-iio", (kernel_ulong_t)&sun5i_gpadc_data },
695 	{ "sun6i-a31-gpadc-iio", (kernel_ulong_t)&sun6i_gpadc_data },
696 	{ /* sentinel */ },
697 };
698 MODULE_DEVICE_TABLE(platform, sun4i_gpadc_id);
699 
700 static struct platform_driver sun4i_gpadc_driver = {
701 	.driver = {
702 		.name = "sun4i-gpadc-iio",
703 		.of_match_table = sun4i_gpadc_of_id,
704 		.pm = &sun4i_gpadc_pm_ops,
705 	},
706 	.id_table = sun4i_gpadc_id,
707 	.probe = sun4i_gpadc_probe,
708 	.remove = sun4i_gpadc_remove,
709 };
710 MODULE_DEVICE_TABLE(of, sun4i_gpadc_of_id);
711 
712 module_platform_driver(sun4i_gpadc_driver);
713 
714 MODULE_DESCRIPTION("ADC driver for sunxi platforms");
715 MODULE_AUTHOR("Quentin Schulz <quentin.schulz@free-electrons.com>");
716 MODULE_LICENSE("GPL v2");
717