Merge tag 'timers-urgent-2021-02-22' of git://git.kernel.org/pub/scm/linux/kernel...
[linux-2.6-microblaze.git] / drivers / iio / adc / ingenic-adc.c
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
85 struct ingenic_adc;
86
87 struct 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
101 struct 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
110 static 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
165 static 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
180 static 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
196 static 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
205 static 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
233 static 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
280 static const int jz4725b_adc_battery_raw_avail[] = {
281         0, 1, (1 << JZ_ADC_BATTERY_LOW_VREF_BITS) - 1,
282 };
283
284 static 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
289 static const int jz4740_adc_battery_raw_avail[] = {
290         0, 1, (1 << JZ_ADC_BATTERY_LOW_VREF_BITS) - 1,
291 };
292
293 static 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
298 static const int jz4770_adc_battery_raw_avail[] = {
299         0, 1, (1 << JZ4770_ADC_BATTERY_VREF_BITS) - 1,
300 };
301
302 static const int jz4770_adc_battery_scale_avail[] = {
303         JZ4770_ADC_BATTERY_VREF, JZ4770_ADC_BATTERY_VREF_BITS,
304 };
305
306 static 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
342 static 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
380 static 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
403 static 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
501 static 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
515 static 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
529 static 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
543 static 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
568 static 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;
604 out:
605         mutex_unlock(&adc->aux_lock);
606         clk_disable(adc->clk);
607
608         return ret;
609 }
610
611 static 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
646 static 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
661 static void ingenic_adc_clk_cleanup(void *data)
662 {
663         clk_unprepare(data);
664 }
665
666 static 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
673 static 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
705 static 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
723 static const struct iio_buffer_setup_ops ingenic_buffer_setup_ops = {
724         .postenable = &ingenic_adc_buffer_enable,
725         .predisable = &ingenic_adc_buffer_disable
726 };
727
728 static 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
749 static 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
832 static 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 };
838 MODULE_DEVICE_TABLE(of, ingenic_adc_of_match);
839
840 static 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 };
847 module_platform_driver(ingenic_adc_driver);
848 MODULE_LICENSE("GPL v2");