1// SPDX-License-Identifier: GPL-2.0
2/*
3 * ADC driver for the Ingenic JZ47xx SoCs
4 * Copyright (c) 2019 Artur Rojek <contact@artur-rojek.eu>
5 *
6 * based on drivers/mfd/jz4740-adc.c
7 */
8
9#include <dt-bindings/iio/adc/ingenic,adc.h>
10#include <linux/clk.h>
11#include <linux/iio/buffer.h>
12#include <linux/iio/iio.h>
13#include <linux/interrupt.h>
14#include <linux/io.h>
15#include <linux/iopoll.h>
16#include <linux/kernel.h>
17#include <linux/module.h>
18#include <linux/mod_devicetable.h>
19#include <linux/mutex.h>
20#include <linux/platform_device.h>
21
22#define JZ_ADC_REG_ENABLE		0x00
23#define JZ_ADC_REG_CFG			0x04
24#define JZ_ADC_REG_CTRL			0x08
25#define JZ_ADC_REG_STATUS		0x0c
26#define JZ_ADC_REG_ADSAME		0x10
27#define JZ_ADC_REG_ADWAIT		0x14
28#define JZ_ADC_REG_ADTCH		0x18
29#define JZ_ADC_REG_ADBDAT		0x1c
30#define JZ_ADC_REG_ADSDAT		0x20
31#define JZ_ADC_REG_ADCMD		0x24
32#define JZ_ADC_REG_ADCLK		0x28
33
34#define JZ_ADC_REG_ENABLE_PD		BIT(7)
35#define JZ_ADC_REG_CFG_AUX_MD		(BIT(0) | BIT(1))
36#define JZ_ADC_REG_CFG_BAT_MD		BIT(4)
37#define JZ_ADC_REG_CFG_SAMPLE_NUM(n)	((n) << 10)
38#define JZ_ADC_REG_CFG_PULL_UP(n)	((n) << 16)
39#define JZ_ADC_REG_CFG_CMD_SEL		BIT(22)
40#define JZ_ADC_REG_CFG_TOUCH_OPS_MASK	(BIT(31) | GENMASK(23, 10))
41#define JZ_ADC_REG_ADCLK_CLKDIV_LSB	0
42#define JZ4725B_ADC_REG_ADCLK_CLKDIV10US_LSB	16
43#define JZ4770_ADC_REG_ADCLK_CLKDIV10US_LSB	8
44#define JZ4770_ADC_REG_ADCLK_CLKDIVMS_LSB	16
45
46#define JZ_ADC_REG_ADCMD_YNADC		BIT(7)
47#define JZ_ADC_REG_ADCMD_YPADC		BIT(8)
48#define JZ_ADC_REG_ADCMD_XNADC		BIT(9)
49#define JZ_ADC_REG_ADCMD_XPADC		BIT(10)
50#define JZ_ADC_REG_ADCMD_VREFPYP	BIT(11)
51#define JZ_ADC_REG_ADCMD_VREFPXP	BIT(12)
52#define JZ_ADC_REG_ADCMD_VREFPXN	BIT(13)
53#define JZ_ADC_REG_ADCMD_VREFPAUX	BIT(14)
54#define JZ_ADC_REG_ADCMD_VREFPVDD33	BIT(15)
55#define JZ_ADC_REG_ADCMD_VREFNYN	BIT(16)
56#define JZ_ADC_REG_ADCMD_VREFNXP	BIT(17)
57#define JZ_ADC_REG_ADCMD_VREFNXN	BIT(18)
58#define JZ_ADC_REG_ADCMD_VREFAUX	BIT(19)
59#define JZ_ADC_REG_ADCMD_YNGRU		BIT(20)
60#define JZ_ADC_REG_ADCMD_XNGRU		BIT(21)
61#define JZ_ADC_REG_ADCMD_XPGRU		BIT(22)
62#define JZ_ADC_REG_ADCMD_YPSUP		BIT(23)
63#define JZ_ADC_REG_ADCMD_XNSUP		BIT(24)
64#define JZ_ADC_REG_ADCMD_XPSUP		BIT(25)
65
66#define JZ_ADC_AUX_VREF				3300
67#define JZ_ADC_AUX_VREF_BITS			12
68#define JZ_ADC_BATTERY_LOW_VREF			2500
69#define JZ_ADC_BATTERY_LOW_VREF_BITS		12
70#define JZ4725B_ADC_BATTERY_HIGH_VREF		7500
71#define JZ4725B_ADC_BATTERY_HIGH_VREF_BITS	10
72#define JZ4740_ADC_BATTERY_HIGH_VREF		(7500 * 0.986)
73#define JZ4740_ADC_BATTERY_HIGH_VREF_BITS	12
74#define JZ4770_ADC_BATTERY_VREF			1200
75#define JZ4770_ADC_BATTERY_VREF_BITS		12
76
77#define JZ_ADC_IRQ_AUX			BIT(0)
78#define JZ_ADC_IRQ_BATTERY		BIT(1)
79#define JZ_ADC_IRQ_TOUCH		BIT(2)
80#define JZ_ADC_IRQ_PEN_DOWN		BIT(3)
81#define JZ_ADC_IRQ_PEN_UP		BIT(4)
82#define JZ_ADC_IRQ_PEN_DOWN_SLEEP	BIT(5)
83#define JZ_ADC_IRQ_SLEEP		BIT(7)
84
85struct ingenic_adc;
86
87struct ingenic_adc_soc_data {
88	unsigned int battery_high_vref;
89	unsigned int battery_high_vref_bits;
90	const int *battery_raw_avail;
91	size_t battery_raw_avail_size;
92	const int *battery_scale_avail;
93	size_t battery_scale_avail_size;
94	unsigned int battery_vref_mode: 1;
95	unsigned int has_aux2: 1;
96	const struct iio_chan_spec *channels;
97	unsigned int num_channels;
98	int (*init_clk_div)(struct device *dev, struct ingenic_adc *adc);
99};
100
101struct ingenic_adc {
102	void __iomem *base;
103	struct clk *clk;
104	struct mutex lock;
105	struct mutex aux_lock;
106	const struct ingenic_adc_soc_data *soc_data;
107	bool low_vref_mode;
108};
109
110static void ingenic_adc_set_adcmd(struct iio_dev *iio_dev, unsigned long mask)
111{
112	struct ingenic_adc *adc = iio_priv(iio_dev);
113
114	mutex_lock(&adc->lock);
115
116	/* Init ADCMD */
117	readl(adc->base + JZ_ADC_REG_ADCMD);
118
119	if (mask & 0x3) {
120		/* Second channel (INGENIC_ADC_TOUCH_YP): sample YP vs. GND */
121		writel(JZ_ADC_REG_ADCMD_XNGRU
122		       | JZ_ADC_REG_ADCMD_VREFNXN | JZ_ADC_REG_ADCMD_VREFPVDD33
123		       | JZ_ADC_REG_ADCMD_YPADC,
124		       adc->base + JZ_ADC_REG_ADCMD);
125
126		/* First channel (INGENIC_ADC_TOUCH_XP): sample XP vs. GND */
127		writel(JZ_ADC_REG_ADCMD_YNGRU
128		       | JZ_ADC_REG_ADCMD_VREFNYN | JZ_ADC_REG_ADCMD_VREFPVDD33
129		       | JZ_ADC_REG_ADCMD_XPADC,
130		       adc->base + JZ_ADC_REG_ADCMD);
131	}
132
133	if (mask & 0xc) {
134		/* Fourth channel (INGENIC_ADC_TOUCH_YN): sample YN vs. GND */
135		writel(JZ_ADC_REG_ADCMD_XNGRU
136		       | JZ_ADC_REG_ADCMD_VREFNXN | JZ_ADC_REG_ADCMD_VREFPVDD33
137		       | JZ_ADC_REG_ADCMD_YNADC,
138		       adc->base + JZ_ADC_REG_ADCMD);
139
140		/* Third channel (INGENIC_ADC_TOUCH_XN): sample XN vs. GND */
141		writel(JZ_ADC_REG_ADCMD_YNGRU
142		       | JZ_ADC_REG_ADCMD_VREFNYN | JZ_ADC_REG_ADCMD_VREFPVDD33
143		       | JZ_ADC_REG_ADCMD_XNADC,
144		       adc->base + JZ_ADC_REG_ADCMD);
145	}
146
147	if (mask & 0x30) {
148		/* Sixth channel (INGENIC_ADC_TOUCH_YD): sample YP vs. YN */
149		writel(JZ_ADC_REG_ADCMD_VREFNYN | JZ_ADC_REG_ADCMD_VREFPVDD33
150		       | JZ_ADC_REG_ADCMD_YPADC,
151		       adc->base + JZ_ADC_REG_ADCMD);
152
153		/* Fifth channel (INGENIC_ADC_TOUCH_XD): sample XP vs. XN */
154		writel(JZ_ADC_REG_ADCMD_VREFNXN | JZ_ADC_REG_ADCMD_VREFPVDD33
155		       | JZ_ADC_REG_ADCMD_XPADC,
156		       adc->base + JZ_ADC_REG_ADCMD);
157	}
158
159	/* We're done */
160	writel(0, adc->base + JZ_ADC_REG_ADCMD);
161
162	mutex_unlock(&adc->lock);
163}
164
165static void ingenic_adc_set_config(struct ingenic_adc *adc,
166				   uint32_t mask,
167				   uint32_t val)
168{
169	uint32_t cfg;
170
171	mutex_lock(&adc->lock);
172
173	cfg = readl(adc->base + JZ_ADC_REG_CFG) & ~mask;
174	cfg |= val;
175	writel(cfg, adc->base + JZ_ADC_REG_CFG);
176
177	mutex_unlock(&adc->lock);
178}
179
180static void ingenic_adc_enable_unlocked(struct ingenic_adc *adc,
181					int engine,
182					bool enabled)
183{
184	u8 val;
185
186	val = readb(adc->base + JZ_ADC_REG_ENABLE);
187
188	if (enabled)
189		val |= BIT(engine);
190	else
191		val &= ~BIT(engine);
192
193	writeb(val, adc->base + JZ_ADC_REG_ENABLE);
194}
195
196static void ingenic_adc_enable(struct ingenic_adc *adc,
197			       int engine,
198			       bool enabled)
199{
200	mutex_lock(&adc->lock);
201	ingenic_adc_enable_unlocked(adc, engine, enabled);
202	mutex_unlock(&adc->lock);
203}
204
205static int ingenic_adc_capture(struct ingenic_adc *adc,
206			       int engine)
207{
208	u32 cfg;
209	u8 val;
210	int ret;
211
212	/*
213	 * Disable CMD_SEL temporarily, because it causes wrong VBAT readings,
214	 * probably due to the switch of VREF. We must keep the lock here to
215	 * avoid races with the buffer enable/disable functions.
216	 */
217	mutex_lock(&adc->lock);
218	cfg = readl(adc->base + JZ_ADC_REG_CFG);
219	writel(cfg & ~JZ_ADC_REG_CFG_CMD_SEL, adc->base + JZ_ADC_REG_CFG);
220
221	ingenic_adc_enable_unlocked(adc, engine, true);
222	ret = readb_poll_timeout(adc->base + JZ_ADC_REG_ENABLE, val,
223				 !(val & BIT(engine)), 250, 1000);
224	if (ret)
225		ingenic_adc_enable_unlocked(adc, engine, false);
226
227	writel(cfg, adc->base + JZ_ADC_REG_CFG);
228	mutex_unlock(&adc->lock);
229
230	return ret;
231}
232
233static int ingenic_adc_write_raw(struct iio_dev *iio_dev,
234				 struct iio_chan_spec const *chan,
235				 int val,
236				 int val2,
237				 long m)
238{
239	struct ingenic_adc *adc = iio_priv(iio_dev);
240	struct device *dev = iio_dev->dev.parent;
241	int ret;
242
243	switch (m) {
244	case IIO_CHAN_INFO_SCALE:
245		switch (chan->channel) {
246		case INGENIC_ADC_BATTERY:
247			if (!adc->soc_data->battery_vref_mode)
248				return -EINVAL;
249
250			ret = clk_enable(adc->clk);
251			if (ret) {
252				dev_err(dev, "Failed to enable clock: %d\n",
253					ret);
254				return ret;
255			}
256
257			if (val > JZ_ADC_BATTERY_LOW_VREF) {
258				ingenic_adc_set_config(adc,
259						       JZ_ADC_REG_CFG_BAT_MD,
260						       0);
261				adc->low_vref_mode = false;
262			} else {
263				ingenic_adc_set_config(adc,
264						       JZ_ADC_REG_CFG_BAT_MD,
265						       JZ_ADC_REG_CFG_BAT_MD);
266				adc->low_vref_mode = true;
267			}
268
269			clk_disable(adc->clk);
270
271			return 0;
272		default:
273			return -EINVAL;
274		}
275	default:
276		return -EINVAL;
277	}
278}
279
280static const int jz4725b_adc_battery_raw_avail[] = {
281	0, 1, (1 << JZ_ADC_BATTERY_LOW_VREF_BITS) - 1,
282};
283
284static const int jz4725b_adc_battery_scale_avail[] = {
285	JZ4725B_ADC_BATTERY_HIGH_VREF, JZ4725B_ADC_BATTERY_HIGH_VREF_BITS,
286	JZ_ADC_BATTERY_LOW_VREF, JZ_ADC_BATTERY_LOW_VREF_BITS,
287};
288
289static const int jz4740_adc_battery_raw_avail[] = {
290	0, 1, (1 << JZ_ADC_BATTERY_LOW_VREF_BITS) - 1,
291};
292
293static const int jz4740_adc_battery_scale_avail[] = {
294	JZ4740_ADC_BATTERY_HIGH_VREF, JZ4740_ADC_BATTERY_HIGH_VREF_BITS,
295	JZ_ADC_BATTERY_LOW_VREF, JZ_ADC_BATTERY_LOW_VREF_BITS,
296};
297
298static const int jz4770_adc_battery_raw_avail[] = {
299	0, 1, (1 << JZ4770_ADC_BATTERY_VREF_BITS) - 1,
300};
301
302static const int jz4770_adc_battery_scale_avail[] = {
303	JZ4770_ADC_BATTERY_VREF, JZ4770_ADC_BATTERY_VREF_BITS,
304};
305
306static int jz4725b_adc_init_clk_div(struct device *dev, struct ingenic_adc *adc)
307{
308	struct clk *parent_clk;
309	unsigned long parent_rate, rate;
310	unsigned int div_main, div_10us;
311
312	parent_clk = clk_get_parent(adc->clk);
313	if (!parent_clk) {
314		dev_err(dev, "ADC clock has no parent\n");
315		return -ENODEV;
316	}
317	parent_rate = clk_get_rate(parent_clk);
318
319	/*
320	 * The JZ4725B ADC works at 500 kHz to 8 MHz.
321	 * We pick the highest rate possible.
322	 * In practice we typically get 6 MHz, half of the 12 MHz EXT clock.
323	 */
324	div_main = DIV_ROUND_UP(parent_rate, 8000000);
325	div_main = clamp(div_main, 1u, 64u);
326	rate = parent_rate / div_main;
327	if (rate < 500000 || rate > 8000000) {
328		dev_err(dev, "No valid divider for ADC main clock\n");
329		return -EINVAL;
330	}
331
332	/* We also need a divider that produces a 10us clock. */
333	div_10us = DIV_ROUND_UP(rate, 100000);
334
335	writel(((div_10us - 1) << JZ4725B_ADC_REG_ADCLK_CLKDIV10US_LSB) |
336	       (div_main - 1) << JZ_ADC_REG_ADCLK_CLKDIV_LSB,
337	       adc->base + JZ_ADC_REG_ADCLK);
338
339	return 0;
340}
341
342static int jz4770_adc_init_clk_div(struct device *dev, struct ingenic_adc *adc)
343{
344	struct clk *parent_clk;
345	unsigned long parent_rate, rate;
346	unsigned int div_main, div_ms, div_10us;
347
348	parent_clk = clk_get_parent(adc->clk);
349	if (!parent_clk) {
350		dev_err(dev, "ADC clock has no parent\n");
351		return -ENODEV;
352	}
353	parent_rate = clk_get_rate(parent_clk);
354
355	/*
356	 * The JZ4770 ADC works at 20 kHz to 200 kHz.
357	 * We pick the highest rate possible.
358	 */
359	div_main = DIV_ROUND_UP(parent_rate, 200000);
360	div_main = clamp(div_main, 1u, 256u);
361	rate = parent_rate / div_main;
362	if (rate < 20000 || rate > 200000) {
363		dev_err(dev, "No valid divider for ADC main clock\n");
364		return -EINVAL;
365	}
366
367	/* We also need a divider that produces a 10us clock. */
368	div_10us = DIV_ROUND_UP(rate, 10000);
369	/* And another, which produces a 1ms clock. */
370	div_ms = DIV_ROUND_UP(rate, 1000);
371
372	writel(((div_ms - 1) << JZ4770_ADC_REG_ADCLK_CLKDIVMS_LSB) |
373	       ((div_10us - 1) << JZ4770_ADC_REG_ADCLK_CLKDIV10US_LSB) |
374	       (div_main - 1) << JZ_ADC_REG_ADCLK_CLKDIV_LSB,
375	       adc->base + JZ_ADC_REG_ADCLK);
376
377	return 0;
378}
379
380static const struct iio_chan_spec jz4740_channels[] = {
381	{
382		.extend_name = "aux",
383		.type = IIO_VOLTAGE,
384		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
385				      BIT(IIO_CHAN_INFO_SCALE),
386		.indexed = 1,
387		.channel = INGENIC_ADC_AUX,
388		.scan_index = -1,
389	},
390	{
391		.extend_name = "battery",
392		.type = IIO_VOLTAGE,
393		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
394				      BIT(IIO_CHAN_INFO_SCALE),
395		.info_mask_separate_available = BIT(IIO_CHAN_INFO_RAW) |
396						BIT(IIO_CHAN_INFO_SCALE),
397		.indexed = 1,
398		.channel = INGENIC_ADC_BATTERY,
399		.scan_index = -1,
400	},
401};
402
403static const struct iio_chan_spec jz4770_channels[] = {
404	{
405		.type = IIO_VOLTAGE,
406		.indexed = 1,
407		.channel = INGENIC_ADC_TOUCH_XP,
408		.scan_index = 0,
409		.scan_type = {
410			.sign = 'u',
411			.realbits = 12,
412			.storagebits = 16,
413		},
414	},
415	{
416		.type = IIO_VOLTAGE,
417		.indexed = 1,
418		.channel = INGENIC_ADC_TOUCH_YP,
419		.scan_index = 1,
420		.scan_type = {
421			.sign = 'u',
422			.realbits = 12,
423			.storagebits = 16,
424		},
425	},
426	{
427		.type = IIO_VOLTAGE,
428		.indexed = 1,
429		.channel = INGENIC_ADC_TOUCH_XN,
430		.scan_index = 2,
431		.scan_type = {
432			.sign = 'u',
433			.realbits = 12,
434			.storagebits = 16,
435		},
436	},
437	{
438		.type = IIO_VOLTAGE,
439		.indexed = 1,
440		.channel = INGENIC_ADC_TOUCH_YN,
441		.scan_index = 3,
442		.scan_type = {
443			.sign = 'u',
444			.realbits = 12,
445			.storagebits = 16,
446		},
447	},
448	{
449		.type = IIO_VOLTAGE,
450		.indexed = 1,
451		.channel = INGENIC_ADC_TOUCH_XD,
452		.scan_index = 4,
453		.scan_type = {
454			.sign = 'u',
455			.realbits = 12,
456			.storagebits = 16,
457		},
458	},
459	{
460		.type = IIO_VOLTAGE,
461		.indexed = 1,
462		.channel = INGENIC_ADC_TOUCH_YD,
463		.scan_index = 5,
464		.scan_type = {
465			.sign = 'u',
466			.realbits = 12,
467			.storagebits = 16,
468		},
469	},
470	{
471		.extend_name = "aux",
472		.type = IIO_VOLTAGE,
473		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
474				      BIT(IIO_CHAN_INFO_SCALE),
475		.indexed = 1,
476		.channel = INGENIC_ADC_AUX,
477		.scan_index = -1,
478	},
479	{
480		.extend_name = "battery",
481		.type = IIO_VOLTAGE,
482		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
483				      BIT(IIO_CHAN_INFO_SCALE),
484		.info_mask_separate_available = BIT(IIO_CHAN_INFO_RAW) |
485						BIT(IIO_CHAN_INFO_SCALE),
486		.indexed = 1,
487		.channel = INGENIC_ADC_BATTERY,
488		.scan_index = -1,
489	},
490	{
491		.extend_name = "aux2",
492		.type = IIO_VOLTAGE,
493		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
494				      BIT(IIO_CHAN_INFO_SCALE),
495		.indexed = 1,
496		.channel = INGENIC_ADC_AUX2,
497		.scan_index = -1,
498	},
499};
500
501static const struct ingenic_adc_soc_data jz4725b_adc_soc_data = {
502	.battery_high_vref = JZ4725B_ADC_BATTERY_HIGH_VREF,
503	.battery_high_vref_bits = JZ4725B_ADC_BATTERY_HIGH_VREF_BITS,
504	.battery_raw_avail = jz4725b_adc_battery_raw_avail,
505	.battery_raw_avail_size = ARRAY_SIZE(jz4725b_adc_battery_raw_avail),
506	.battery_scale_avail = jz4725b_adc_battery_scale_avail,
507	.battery_scale_avail_size = ARRAY_SIZE(jz4725b_adc_battery_scale_avail),
508	.battery_vref_mode = true,
509	.has_aux2 = false,
510	.channels = jz4740_channels,
511	.num_channels = ARRAY_SIZE(jz4740_channels),
512	.init_clk_div = jz4725b_adc_init_clk_div,
513};
514
515static const struct ingenic_adc_soc_data jz4740_adc_soc_data = {
516	.battery_high_vref = JZ4740_ADC_BATTERY_HIGH_VREF,
517	.battery_high_vref_bits = JZ4740_ADC_BATTERY_HIGH_VREF_BITS,
518	.battery_raw_avail = jz4740_adc_battery_raw_avail,
519	.battery_raw_avail_size = ARRAY_SIZE(jz4740_adc_battery_raw_avail),
520	.battery_scale_avail = jz4740_adc_battery_scale_avail,
521	.battery_scale_avail_size = ARRAY_SIZE(jz4740_adc_battery_scale_avail),
522	.battery_vref_mode = true,
523	.has_aux2 = false,
524	.channels = jz4740_channels,
525	.num_channels = ARRAY_SIZE(jz4740_channels),
526	.init_clk_div = NULL, /* no ADCLK register on JZ4740 */
527};
528
529static const struct ingenic_adc_soc_data jz4770_adc_soc_data = {
530	.battery_high_vref = JZ4770_ADC_BATTERY_VREF,
531	.battery_high_vref_bits = JZ4770_ADC_BATTERY_VREF_BITS,
532	.battery_raw_avail = jz4770_adc_battery_raw_avail,
533	.battery_raw_avail_size = ARRAY_SIZE(jz4770_adc_battery_raw_avail),
534	.battery_scale_avail = jz4770_adc_battery_scale_avail,
535	.battery_scale_avail_size = ARRAY_SIZE(jz4770_adc_battery_scale_avail),
536	.battery_vref_mode = false,
537	.has_aux2 = true,
538	.channels = jz4770_channels,
539	.num_channels = ARRAY_SIZE(jz4770_channels),
540	.init_clk_div = jz4770_adc_init_clk_div,
541};
542
543static int ingenic_adc_read_avail(struct iio_dev *iio_dev,
544				  struct iio_chan_spec const *chan,
545				  const int **vals,
546				  int *type,
547				  int *length,
548				  long m)
549{
550	struct ingenic_adc *adc = iio_priv(iio_dev);
551
552	switch (m) {
553	case IIO_CHAN_INFO_RAW:
554		*type = IIO_VAL_INT;
555		*length = adc->soc_data->battery_raw_avail_size;
556		*vals = adc->soc_data->battery_raw_avail;
557		return IIO_AVAIL_RANGE;
558	case IIO_CHAN_INFO_SCALE:
559		*type = IIO_VAL_FRACTIONAL_LOG2;
560		*length = adc->soc_data->battery_scale_avail_size;
561		*vals = adc->soc_data->battery_scale_avail;
562		return IIO_AVAIL_LIST;
563	default:
564		return -EINVAL;
565	};
566}
567
568static int ingenic_adc_read_chan_info_raw(struct iio_dev *iio_dev,
569					  struct iio_chan_spec const *chan,
570					  int *val)
571{
572	int bit, ret, engine = (chan->channel == INGENIC_ADC_BATTERY);
573	struct ingenic_adc *adc = iio_priv(iio_dev);
574
575	ret = clk_enable(adc->clk);
576	if (ret) {
577		dev_err(iio_dev->dev.parent, "Failed to enable clock: %d\n",
578			ret);
579		return ret;
580	}
581
582	/* We cannot sample AUX/AUX2 in parallel. */
583	mutex_lock(&adc->aux_lock);
584	if (adc->soc_data->has_aux2 && engine == 0) {
585		bit = BIT(chan->channel == INGENIC_ADC_AUX2);
586		ingenic_adc_set_config(adc, JZ_ADC_REG_CFG_AUX_MD, bit);
587	}
588
589	ret = ingenic_adc_capture(adc, engine);
590	if (ret)
591		goto out;
592
593	switch (chan->channel) {
594	case INGENIC_ADC_AUX:
595	case INGENIC_ADC_AUX2:
596		*val = readw(adc->base + JZ_ADC_REG_ADSDAT);
597		break;
598	case INGENIC_ADC_BATTERY:
599		*val = readw(adc->base + JZ_ADC_REG_ADBDAT);
600		break;
601	}
602
603	ret = IIO_VAL_INT;
604out:
605	mutex_unlock(&adc->aux_lock);
606	clk_disable(adc->clk);
607
608	return ret;
609}
610
611static int ingenic_adc_read_raw(struct iio_dev *iio_dev,
612				struct iio_chan_spec const *chan,
613				int *val,
614				int *val2,
615				long m)
616{
617	struct ingenic_adc *adc = iio_priv(iio_dev);
618
619	switch (m) {
620	case IIO_CHAN_INFO_RAW:
621		return ingenic_adc_read_chan_info_raw(iio_dev, chan, val);
622	case IIO_CHAN_INFO_SCALE:
623		switch (chan->channel) {
624		case INGENIC_ADC_AUX:
625		case INGENIC_ADC_AUX2:
626			*val = JZ_ADC_AUX_VREF;
627			*val2 = JZ_ADC_AUX_VREF_BITS;
628			break;
629		case INGENIC_ADC_BATTERY:
630			if (adc->low_vref_mode) {
631				*val = JZ_ADC_BATTERY_LOW_VREF;
632				*val2 = JZ_ADC_BATTERY_LOW_VREF_BITS;
633			} else {
634				*val = adc->soc_data->battery_high_vref;
635				*val2 = adc->soc_data->battery_high_vref_bits;
636			}
637			break;
638		}
639
640		return IIO_VAL_FRACTIONAL_LOG2;
641	default:
642		return -EINVAL;
643	}
644}
645
646static int ingenic_adc_of_xlate(struct iio_dev *iio_dev,
647				const struct of_phandle_args *iiospec)
648{
649	int i;
650
651	if (!iiospec->args_count)
652		return -EINVAL;
653
654	for (i = 0; i < iio_dev->num_channels; ++i)
655		if (iio_dev->channels[i].channel == iiospec->args[0])
656			return i;
657
658	return -EINVAL;
659}
660
661static void ingenic_adc_clk_cleanup(void *data)
662{
663	clk_unprepare(data);
664}
665
666static const struct iio_info ingenic_adc_info = {
667	.write_raw = ingenic_adc_write_raw,
668	.read_raw = ingenic_adc_read_raw,
669	.read_avail = ingenic_adc_read_avail,
670	.of_xlate = ingenic_adc_of_xlate,
671};
672
673static int ingenic_adc_buffer_enable(struct iio_dev *iio_dev)
674{
675	struct ingenic_adc *adc = iio_priv(iio_dev);
676	int ret;
677
678	ret = clk_enable(adc->clk);
679	if (ret) {
680		dev_err(iio_dev->dev.parent, "Failed to enable clock: %d\n",
681			ret);
682		return ret;
683	}
684
685	/* It takes significant time for the touchscreen hw to stabilize. */
686	msleep(50);
687	ingenic_adc_set_config(adc, JZ_ADC_REG_CFG_TOUCH_OPS_MASK,
688			       JZ_ADC_REG_CFG_SAMPLE_NUM(4) |
689			       JZ_ADC_REG_CFG_PULL_UP(4));
690
691	writew(80, adc->base + JZ_ADC_REG_ADWAIT);
692	writew(2, adc->base + JZ_ADC_REG_ADSAME);
693	writeb((u8)~JZ_ADC_IRQ_TOUCH, adc->base + JZ_ADC_REG_CTRL);
694	writel(0, adc->base + JZ_ADC_REG_ADTCH);
695
696	ingenic_adc_set_config(adc, JZ_ADC_REG_CFG_CMD_SEL,
697			       JZ_ADC_REG_CFG_CMD_SEL);
698	ingenic_adc_set_adcmd(iio_dev, iio_dev->active_scan_mask[0]);
699
700	ingenic_adc_enable(adc, 2, true);
701
702	return 0;
703}
704
705static int ingenic_adc_buffer_disable(struct iio_dev *iio_dev)
706{
707	struct ingenic_adc *adc = iio_priv(iio_dev);
708
709	ingenic_adc_enable(adc, 2, false);
710
711	ingenic_adc_set_config(adc, JZ_ADC_REG_CFG_CMD_SEL, 0);
712
713	writeb(0xff, adc->base + JZ_ADC_REG_CTRL);
714	writeb(0xff, adc->base + JZ_ADC_REG_STATUS);
715	ingenic_adc_set_config(adc, JZ_ADC_REG_CFG_TOUCH_OPS_MASK, 0);
716	writew(0, adc->base + JZ_ADC_REG_ADSAME);
717	writew(0, adc->base + JZ_ADC_REG_ADWAIT);
718	clk_disable(adc->clk);
719
720	return 0;
721}
722
723static const struct iio_buffer_setup_ops ingenic_buffer_setup_ops = {
724	.postenable = &ingenic_adc_buffer_enable,
725	.predisable = &ingenic_adc_buffer_disable
726};
727
728static irqreturn_t ingenic_adc_irq(int irq, void *data)
729{
730	struct iio_dev *iio_dev = data;
731	struct ingenic_adc *adc = iio_priv(iio_dev);
732	unsigned long mask = iio_dev->active_scan_mask[0];
733	unsigned int i;
734	u32 tdat[3];
735
736	for (i = 0; i < ARRAY_SIZE(tdat); mask >>= 2, i++) {
737		if (mask & 0x3)
738			tdat[i] = readl(adc->base + JZ_ADC_REG_ADTCH);
739		else
740			tdat[i] = 0;
741	}
742
743	iio_push_to_buffers(iio_dev, tdat);
744	writeb(JZ_ADC_IRQ_TOUCH, adc->base + JZ_ADC_REG_STATUS);
745
746	return IRQ_HANDLED;
747}
748
749static int ingenic_adc_probe(struct platform_device *pdev)
750{
751	struct device *dev = &pdev->dev;
752	struct iio_dev *iio_dev;
753	struct ingenic_adc *adc;
754	const struct ingenic_adc_soc_data *soc_data;
755	int irq, ret;
756
757	soc_data = device_get_match_data(dev);
758	if (!soc_data)
759		return -EINVAL;
760
761	iio_dev = devm_iio_device_alloc(dev, sizeof(*adc));
762	if (!iio_dev)
763		return -ENOMEM;
764
765	adc = iio_priv(iio_dev);
766	mutex_init(&adc->lock);
767	mutex_init(&adc->aux_lock);
768	adc->soc_data = soc_data;
769
770	irq = platform_get_irq(pdev, 0);
771	if (irq < 0)
772		return irq;
773
774	ret = devm_request_irq(dev, irq, ingenic_adc_irq, 0,
775			       dev_name(dev), iio_dev);
776	if (ret < 0) {
777		dev_err(dev, "Failed to request irq: %d\n", ret);
778		return ret;
779	}
780
781	adc->base = devm_platform_ioremap_resource(pdev, 0);
782	if (IS_ERR(adc->base))
783		return PTR_ERR(adc->base);
784
785	adc->clk = devm_clk_get(dev, "adc");
786	if (IS_ERR(adc->clk)) {
787		dev_err(dev, "Unable to get clock\n");
788		return PTR_ERR(adc->clk);
789	}
790
791	ret = clk_prepare_enable(adc->clk);
792	if (ret) {
793		dev_err(dev, "Failed to enable clock\n");
794		return ret;
795	}
796
797	/* Set clock dividers. */
798	if (soc_data->init_clk_div) {
799		ret = soc_data->init_clk_div(dev, adc);
800		if (ret) {
801			clk_disable_unprepare(adc->clk);
802			return ret;
803		}
804	}
805
806	/* Put hardware in a known passive state. */
807	writeb(0x00, adc->base + JZ_ADC_REG_ENABLE);
808	writeb(0xff, adc->base + JZ_ADC_REG_CTRL);
809	usleep_range(2000, 3000); /* Must wait at least 2ms. */
810	clk_disable(adc->clk);
811
812	ret = devm_add_action_or_reset(dev, ingenic_adc_clk_cleanup, adc->clk);
813	if (ret) {
814		dev_err(dev, "Unable to add action\n");
815		return ret;
816	}
817
818	iio_dev->name = "jz-adc";
819	iio_dev->modes = INDIO_DIRECT_MODE | INDIO_BUFFER_SOFTWARE;
820	iio_dev->setup_ops = &ingenic_buffer_setup_ops;
821	iio_dev->channels = soc_data->channels;
822	iio_dev->num_channels = soc_data->num_channels;
823	iio_dev->info = &ingenic_adc_info;
824
825	ret = devm_iio_device_register(dev, iio_dev);
826	if (ret)
827		dev_err(dev, "Unable to register IIO device\n");
828
829	return ret;
830}
831
832static const struct of_device_id ingenic_adc_of_match[] = {
833	{ .compatible = "ingenic,jz4725b-adc", .data = &jz4725b_adc_soc_data, },
834	{ .compatible = "ingenic,jz4740-adc", .data = &jz4740_adc_soc_data, },
835	{ .compatible = "ingenic,jz4770-adc", .data = &jz4770_adc_soc_data, },
836	{ },
837};
838MODULE_DEVICE_TABLE(of, ingenic_adc_of_match);
839
840static struct platform_driver ingenic_adc_driver = {
841	.driver = {
842		.name = "ingenic-adc",
843		.of_match_table = ingenic_adc_of_match,
844	},
845	.probe = ingenic_adc_probe,
846};
847module_platform_driver(ingenic_adc_driver);
848MODULE_LICENSE("GPL v2");
849