Merge tag 'dmaengine-5.2-rc1' of git://git.infradead.org/users/vkoul/slave-dma
[linux-2.6-microblaze.git] / drivers / iio / adc / at91-sama5d2_adc.c
1 /*
2  * Atmel ADC driver for SAMA5D2 devices and compatible.
3  *
4  * Copyright (C) 2015 Atmel,
5  *               2015 Ludovic Desroches <ludovic.desroches@atmel.com>
6  *
7  * This software is licensed under the terms of the GNU General Public
8  * License version 2, as published by the Free Software Foundation, and
9  * may be copied, distributed, and modified under those terms.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14  * GNU General Public License for more details.
15  */
16
17 #include <linux/bitops.h>
18 #include <linux/clk.h>
19 #include <linux/dma-mapping.h>
20 #include <linux/dmaengine.h>
21 #include <linux/interrupt.h>
22 #include <linux/io.h>
23 #include <linux/module.h>
24 #include <linux/of_device.h>
25 #include <linux/platform_device.h>
26 #include <linux/sched.h>
27 #include <linux/wait.h>
28 #include <linux/iio/iio.h>
29 #include <linux/iio/sysfs.h>
30 #include <linux/iio/buffer.h>
31 #include <linux/iio/trigger.h>
32 #include <linux/iio/trigger_consumer.h>
33 #include <linux/iio/triggered_buffer.h>
34 #include <linux/pinctrl/consumer.h>
35 #include <linux/regulator/consumer.h>
36
37 /* Control Register */
38 #define AT91_SAMA5D2_CR         0x00
39 /* Software Reset */
40 #define AT91_SAMA5D2_CR_SWRST           BIT(0)
41 /* Start Conversion */
42 #define AT91_SAMA5D2_CR_START           BIT(1)
43 /* Touchscreen Calibration */
44 #define AT91_SAMA5D2_CR_TSCALIB         BIT(2)
45 /* Comparison Restart */
46 #define AT91_SAMA5D2_CR_CMPRST          BIT(4)
47
48 /* Mode Register */
49 #define AT91_SAMA5D2_MR         0x04
50 /* Trigger Selection */
51 #define AT91_SAMA5D2_MR_TRGSEL(v)       ((v) << 1)
52 /* ADTRG */
53 #define AT91_SAMA5D2_MR_TRGSEL_TRIG0    0
54 /* TIOA0 */
55 #define AT91_SAMA5D2_MR_TRGSEL_TRIG1    1
56 /* TIOA1 */
57 #define AT91_SAMA5D2_MR_TRGSEL_TRIG2    2
58 /* TIOA2 */
59 #define AT91_SAMA5D2_MR_TRGSEL_TRIG3    3
60 /* PWM event line 0 */
61 #define AT91_SAMA5D2_MR_TRGSEL_TRIG4    4
62 /* PWM event line 1 */
63 #define AT91_SAMA5D2_MR_TRGSEL_TRIG5    5
64 /* TIOA3 */
65 #define AT91_SAMA5D2_MR_TRGSEL_TRIG6    6
66 /* RTCOUT0 */
67 #define AT91_SAMA5D2_MR_TRGSEL_TRIG7    7
68 /* Sleep Mode */
69 #define AT91_SAMA5D2_MR_SLEEP           BIT(5)
70 /* Fast Wake Up */
71 #define AT91_SAMA5D2_MR_FWUP            BIT(6)
72 /* Prescaler Rate Selection */
73 #define AT91_SAMA5D2_MR_PRESCAL(v)      ((v) << AT91_SAMA5D2_MR_PRESCAL_OFFSET)
74 #define AT91_SAMA5D2_MR_PRESCAL_OFFSET  8
75 #define AT91_SAMA5D2_MR_PRESCAL_MAX     0xff
76 #define AT91_SAMA5D2_MR_PRESCAL_MASK    GENMASK(15, 8)
77 /* Startup Time */
78 #define AT91_SAMA5D2_MR_STARTUP(v)      ((v) << 16)
79 #define AT91_SAMA5D2_MR_STARTUP_MASK    GENMASK(19, 16)
80 /* Analog Change */
81 #define AT91_SAMA5D2_MR_ANACH           BIT(23)
82 /* Tracking Time */
83 #define AT91_SAMA5D2_MR_TRACKTIM(v)     ((v) << 24)
84 #define AT91_SAMA5D2_MR_TRACKTIM_MAX    0xff
85 /* Transfer Time */
86 #define AT91_SAMA5D2_MR_TRANSFER(v)     ((v) << 28)
87 #define AT91_SAMA5D2_MR_TRANSFER_MAX    0x3
88 /* Use Sequence Enable */
89 #define AT91_SAMA5D2_MR_USEQ            BIT(31)
90
91 /* Channel Sequence Register 1 */
92 #define AT91_SAMA5D2_SEQR1      0x08
93 /* Channel Sequence Register 2 */
94 #define AT91_SAMA5D2_SEQR2      0x0c
95 /* Channel Enable Register */
96 #define AT91_SAMA5D2_CHER       0x10
97 /* Channel Disable Register */
98 #define AT91_SAMA5D2_CHDR       0x14
99 /* Channel Status Register */
100 #define AT91_SAMA5D2_CHSR       0x18
101 /* Last Converted Data Register */
102 #define AT91_SAMA5D2_LCDR       0x20
103 /* Interrupt Enable Register */
104 #define AT91_SAMA5D2_IER        0x24
105 /* Interrupt Enable Register - TS X measurement ready */
106 #define AT91_SAMA5D2_IER_XRDY   BIT(20)
107 /* Interrupt Enable Register - TS Y measurement ready */
108 #define AT91_SAMA5D2_IER_YRDY   BIT(21)
109 /* Interrupt Enable Register - TS pressure measurement ready */
110 #define AT91_SAMA5D2_IER_PRDY   BIT(22)
111 /* Interrupt Enable Register - general overrun error */
112 #define AT91_SAMA5D2_IER_GOVRE BIT(25)
113 /* Interrupt Enable Register - Pen detect */
114 #define AT91_SAMA5D2_IER_PEN    BIT(29)
115 /* Interrupt Enable Register - No pen detect */
116 #define AT91_SAMA5D2_IER_NOPEN  BIT(30)
117 /* Interrupt Disable Register */
118 #define AT91_SAMA5D2_IDR        0x28
119 /* Interrupt Mask Register */
120 #define AT91_SAMA5D2_IMR        0x2c
121 /* Interrupt Status Register */
122 #define AT91_SAMA5D2_ISR        0x30
123 /* Interrupt Status Register - Pen touching sense status */
124 #define AT91_SAMA5D2_ISR_PENS   BIT(31)
125 /* Last Channel Trigger Mode Register */
126 #define AT91_SAMA5D2_LCTMR      0x34
127 /* Last Channel Compare Window Register */
128 #define AT91_SAMA5D2_LCCWR      0x38
129 /* Overrun Status Register */
130 #define AT91_SAMA5D2_OVER       0x3c
131 /* Extended Mode Register */
132 #define AT91_SAMA5D2_EMR        0x40
133 /* Extended Mode Register - Oversampling rate */
134 #define AT91_SAMA5D2_EMR_OSR(V)                 ((V) << 16)
135 #define AT91_SAMA5D2_EMR_OSR_MASK               GENMASK(17, 16)
136 #define AT91_SAMA5D2_EMR_OSR_1SAMPLES           0
137 #define AT91_SAMA5D2_EMR_OSR_4SAMPLES           1
138 #define AT91_SAMA5D2_EMR_OSR_16SAMPLES          2
139
140 /* Extended Mode Register - Averaging on single trigger event */
141 #define AT91_SAMA5D2_EMR_ASTE(V)                ((V) << 20)
142 /* Compare Window Register */
143 #define AT91_SAMA5D2_CWR        0x44
144 /* Channel Gain Register */
145 #define AT91_SAMA5D2_CGR        0x48
146
147 /* Channel Offset Register */
148 #define AT91_SAMA5D2_COR        0x4c
149 #define AT91_SAMA5D2_COR_DIFF_OFFSET    16
150
151 /* Channel Data Register 0 */
152 #define AT91_SAMA5D2_CDR0       0x50
153 /* Analog Control Register */
154 #define AT91_SAMA5D2_ACR        0x94
155 /* Analog Control Register - Pen detect sensitivity mask */
156 #define AT91_SAMA5D2_ACR_PENDETSENS_MASK        GENMASK(1, 0)
157
158 /* Touchscreen Mode Register */
159 #define AT91_SAMA5D2_TSMR       0xb0
160 /* Touchscreen Mode Register - No touch mode */
161 #define AT91_SAMA5D2_TSMR_TSMODE_NONE           0
162 /* Touchscreen Mode Register - 4 wire screen, no pressure measurement */
163 #define AT91_SAMA5D2_TSMR_TSMODE_4WIRE_NO_PRESS 1
164 /* Touchscreen Mode Register - 4 wire screen, pressure measurement */
165 #define AT91_SAMA5D2_TSMR_TSMODE_4WIRE_PRESS    2
166 /* Touchscreen Mode Register - 5 wire screen */
167 #define AT91_SAMA5D2_TSMR_TSMODE_5WIRE          3
168 /* Touchscreen Mode Register - Average samples mask */
169 #define AT91_SAMA5D2_TSMR_TSAV_MASK             GENMASK(5, 4)
170 /* Touchscreen Mode Register - Average samples */
171 #define AT91_SAMA5D2_TSMR_TSAV(x)               ((x) << 4)
172 /* Touchscreen Mode Register - Touch/trigger frequency ratio mask */
173 #define AT91_SAMA5D2_TSMR_TSFREQ_MASK           GENMASK(11, 8)
174 /* Touchscreen Mode Register - Touch/trigger frequency ratio */
175 #define AT91_SAMA5D2_TSMR_TSFREQ(x)             ((x) << 8)
176 /* Touchscreen Mode Register - Pen Debounce Time mask */
177 #define AT91_SAMA5D2_TSMR_PENDBC_MASK           GENMASK(31, 28)
178 /* Touchscreen Mode Register - Pen Debounce Time */
179 #define AT91_SAMA5D2_TSMR_PENDBC(x)            ((x) << 28)
180 /* Touchscreen Mode Register - No DMA for touch measurements */
181 #define AT91_SAMA5D2_TSMR_NOTSDMA               BIT(22)
182 /* Touchscreen Mode Register - Disable pen detection */
183 #define AT91_SAMA5D2_TSMR_PENDET_DIS            (0 << 24)
184 /* Touchscreen Mode Register - Enable pen detection */
185 #define AT91_SAMA5D2_TSMR_PENDET_ENA            BIT(24)
186
187 /* Touchscreen X Position Register */
188 #define AT91_SAMA5D2_XPOSR      0xb4
189 /* Touchscreen Y Position Register */
190 #define AT91_SAMA5D2_YPOSR      0xb8
191 /* Touchscreen Pressure Register */
192 #define AT91_SAMA5D2_PRESSR     0xbc
193 /* Trigger Register */
194 #define AT91_SAMA5D2_TRGR       0xc0
195 /* Mask for TRGMOD field of TRGR register */
196 #define AT91_SAMA5D2_TRGR_TRGMOD_MASK GENMASK(2, 0)
197 /* No trigger, only software trigger can start conversions */
198 #define AT91_SAMA5D2_TRGR_TRGMOD_NO_TRIGGER 0
199 /* Trigger Mode external trigger rising edge */
200 #define AT91_SAMA5D2_TRGR_TRGMOD_EXT_TRIG_RISE 1
201 /* Trigger Mode external trigger falling edge */
202 #define AT91_SAMA5D2_TRGR_TRGMOD_EXT_TRIG_FALL 2
203 /* Trigger Mode external trigger any edge */
204 #define AT91_SAMA5D2_TRGR_TRGMOD_EXT_TRIG_ANY 3
205 /* Trigger Mode internal periodic */
206 #define AT91_SAMA5D2_TRGR_TRGMOD_PERIODIC 5
207 /* Trigger Mode - trigger period mask */
208 #define AT91_SAMA5D2_TRGR_TRGPER_MASK           GENMASK(31, 16)
209 /* Trigger Mode - trigger period */
210 #define AT91_SAMA5D2_TRGR_TRGPER(x)             ((x) << 16)
211
212 /* Correction Select Register */
213 #define AT91_SAMA5D2_COSR       0xd0
214 /* Correction Value Register */
215 #define AT91_SAMA5D2_CVR        0xd4
216 /* Channel Error Correction Register */
217 #define AT91_SAMA5D2_CECR       0xd8
218 /* Write Protection Mode Register */
219 #define AT91_SAMA5D2_WPMR       0xe4
220 /* Write Protection Status Register */
221 #define AT91_SAMA5D2_WPSR       0xe8
222 /* Version Register */
223 #define AT91_SAMA5D2_VERSION    0xfc
224
225 #define AT91_SAMA5D2_HW_TRIG_CNT 3
226 #define AT91_SAMA5D2_SINGLE_CHAN_CNT 12
227 #define AT91_SAMA5D2_DIFF_CHAN_CNT 6
228
229 #define AT91_SAMA5D2_TIMESTAMP_CHAN_IDX (AT91_SAMA5D2_SINGLE_CHAN_CNT + \
230                                          AT91_SAMA5D2_DIFF_CHAN_CNT + 1)
231
232 #define AT91_SAMA5D2_TOUCH_X_CHAN_IDX (AT91_SAMA5D2_SINGLE_CHAN_CNT + \
233                                          AT91_SAMA5D2_DIFF_CHAN_CNT * 2)
234 #define AT91_SAMA5D2_TOUCH_Y_CHAN_IDX   (AT91_SAMA5D2_TOUCH_X_CHAN_IDX + 1)
235 #define AT91_SAMA5D2_TOUCH_P_CHAN_IDX   (AT91_SAMA5D2_TOUCH_Y_CHAN_IDX + 1)
236 #define AT91_SAMA5D2_MAX_CHAN_IDX       AT91_SAMA5D2_TOUCH_P_CHAN_IDX
237
238 #define AT91_SAMA5D2_TOUCH_SAMPLE_PERIOD_US          2000    /* 2ms */
239 #define AT91_SAMA5D2_TOUCH_PEN_DETECT_DEBOUNCE_US    200
240
241 #define AT91_SAMA5D2_XYZ_MASK           GENMASK(11, 0)
242
243 #define AT91_SAMA5D2_MAX_POS_BITS                       12
244
245 /*
246  * Maximum number of bytes to hold conversion from all channels
247  * without the timestamp.
248  */
249 #define AT91_BUFFER_MAX_CONVERSION_BYTES ((AT91_SAMA5D2_SINGLE_CHAN_CNT + \
250                                          AT91_SAMA5D2_DIFF_CHAN_CNT) * 2)
251
252 /* This total must also include the timestamp */
253 #define AT91_BUFFER_MAX_BYTES (AT91_BUFFER_MAX_CONVERSION_BYTES + 8)
254
255 #define AT91_BUFFER_MAX_HWORDS (AT91_BUFFER_MAX_BYTES / 2)
256
257 #define AT91_HWFIFO_MAX_SIZE_STR        "128"
258 #define AT91_HWFIFO_MAX_SIZE            128
259
260 /* Possible values for oversampling ratio */
261 #define AT91_OSR_1SAMPLES               1
262 #define AT91_OSR_4SAMPLES               4
263 #define AT91_OSR_16SAMPLES              16
264
265 #define AT91_SAMA5D2_CHAN_SINGLE(num, addr)                             \
266         {                                                               \
267                 .type = IIO_VOLTAGE,                                    \
268                 .channel = num,                                         \
269                 .address = addr,                                        \
270                 .scan_index = num,                                      \
271                 .scan_type = {                                          \
272                         .sign = 'u',                                    \
273                         .realbits = 14,                                 \
274                         .storagebits = 16,                              \
275                 },                                                      \
276                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW),           \
277                 .info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),   \
278                 .info_mask_shared_by_all = BIT(IIO_CHAN_INFO_SAMP_FREQ)|\
279                                 BIT(IIO_CHAN_INFO_OVERSAMPLING_RATIO),  \
280                 .datasheet_name = "CH"#num,                             \
281                 .indexed = 1,                                           \
282         }
283
284 #define AT91_SAMA5D2_CHAN_DIFF(num, num2, addr)                         \
285         {                                                               \
286                 .type = IIO_VOLTAGE,                                    \
287                 .differential = 1,                                      \
288                 .channel = num,                                         \
289                 .channel2 = num2,                                       \
290                 .address = addr,                                        \
291                 .scan_index = num + AT91_SAMA5D2_SINGLE_CHAN_CNT,       \
292                 .scan_type = {                                          \
293                         .sign = 's',                                    \
294                         .realbits = 14,                                 \
295                         .storagebits = 16,                              \
296                 },                                                      \
297                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW),           \
298                 .info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),   \
299                 .info_mask_shared_by_all = BIT(IIO_CHAN_INFO_SAMP_FREQ)|\
300                                 BIT(IIO_CHAN_INFO_OVERSAMPLING_RATIO),  \
301                 .datasheet_name = "CH"#num"-CH"#num2,                   \
302                 .indexed = 1,                                           \
303         }
304
305 #define AT91_SAMA5D2_CHAN_TOUCH(num, name, mod)                         \
306         {                                                               \
307                 .type = IIO_POSITIONRELATIVE,                           \
308                 .modified = 1,                                          \
309                 .channel = num,                                         \
310                 .channel2 = mod,                                        \
311                 .scan_index = num,                                      \
312                 .scan_type = {                                          \
313                         .sign = 'u',                                    \
314                         .realbits = 12,                                 \
315                         .storagebits = 16,                              \
316                 },                                                      \
317                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW),           \
318                 .info_mask_shared_by_all = BIT(IIO_CHAN_INFO_SAMP_FREQ)|\
319                                 BIT(IIO_CHAN_INFO_OVERSAMPLING_RATIO),  \
320                 .datasheet_name = name,                                 \
321         }
322 #define AT91_SAMA5D2_CHAN_PRESSURE(num, name)                           \
323         {                                                               \
324                 .type = IIO_PRESSURE,                                   \
325                 .channel = num,                                         \
326                 .scan_index = num,                                      \
327                 .scan_type = {                                          \
328                         .sign = 'u',                                    \
329                         .realbits = 12,                                 \
330                         .storagebits = 16,                              \
331                 },                                                      \
332                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW),           \
333                 .info_mask_shared_by_all = BIT(IIO_CHAN_INFO_SAMP_FREQ)|\
334                                 BIT(IIO_CHAN_INFO_OVERSAMPLING_RATIO),  \
335                 .datasheet_name = name,                                 \
336         }
337
338 #define at91_adc_readl(st, reg)         readl_relaxed(st->base + reg)
339 #define at91_adc_writel(st, reg, val)   writel_relaxed(val, st->base + reg)
340
341 struct at91_adc_soc_info {
342         unsigned                        startup_time;
343         unsigned                        min_sample_rate;
344         unsigned                        max_sample_rate;
345 };
346
347 struct at91_adc_trigger {
348         char                            *name;
349         unsigned int                    trgmod_value;
350         unsigned int                    edge_type;
351         bool                            hw_trig;
352 };
353
354 /**
355  * at91_adc_dma - at91-sama5d2 dma information struct
356  * @dma_chan:           the dma channel acquired
357  * @rx_buf:             dma coherent allocated area
358  * @rx_dma_buf:         dma handler for the buffer
359  * @phys_addr:          physical address of the ADC base register
360  * @buf_idx:            index inside the dma buffer where reading was last done
361  * @rx_buf_sz:          size of buffer used by DMA operation
362  * @watermark:          number of conversions to copy before DMA triggers irq
363  * @dma_ts:             hold the start timestamp of dma operation
364  */
365 struct at91_adc_dma {
366         struct dma_chan                 *dma_chan;
367         u8                              *rx_buf;
368         dma_addr_t                      rx_dma_buf;
369         phys_addr_t                     phys_addr;
370         int                             buf_idx;
371         int                             rx_buf_sz;
372         int                             watermark;
373         s64                             dma_ts;
374 };
375
376 /**
377  * at91_adc_touch - at91-sama5d2 touchscreen information struct
378  * @sample_period_val:          the value for periodic trigger interval
379  * @touching:                   is the pen touching the screen or not
380  * @x_pos:                      temporary placeholder for pressure computation
381  * @channels_bitmask:           bitmask with the touchscreen channels enabled
382  * @workq:                      workqueue for buffer data pushing
383  */
384 struct at91_adc_touch {
385         u16                             sample_period_val;
386         bool                            touching;
387         u16                             x_pos;
388         unsigned long                   channels_bitmask;
389         struct work_struct              workq;
390 };
391
392 struct at91_adc_state {
393         void __iomem                    *base;
394         int                             irq;
395         struct clk                      *per_clk;
396         struct regulator                *reg;
397         struct regulator                *vref;
398         int                             vref_uv;
399         unsigned int                    current_sample_rate;
400         struct iio_trigger              *trig;
401         const struct at91_adc_trigger   *selected_trig;
402         const struct iio_chan_spec      *chan;
403         bool                            conversion_done;
404         u32                             conversion_value;
405         unsigned int                    oversampling_ratio;
406         struct at91_adc_soc_info        soc_info;
407         wait_queue_head_t               wq_data_available;
408         struct at91_adc_dma             dma_st;
409         struct at91_adc_touch           touch_st;
410         u16                             buffer[AT91_BUFFER_MAX_HWORDS];
411         /*
412          * lock to prevent concurrent 'single conversion' requests through
413          * sysfs.
414          */
415         struct mutex                    lock;
416 };
417
418 static const struct at91_adc_trigger at91_adc_trigger_list[] = {
419         {
420                 .name = "external_rising",
421                 .trgmod_value = AT91_SAMA5D2_TRGR_TRGMOD_EXT_TRIG_RISE,
422                 .edge_type = IRQ_TYPE_EDGE_RISING,
423                 .hw_trig = true,
424         },
425         {
426                 .name = "external_falling",
427                 .trgmod_value = AT91_SAMA5D2_TRGR_TRGMOD_EXT_TRIG_FALL,
428                 .edge_type = IRQ_TYPE_EDGE_FALLING,
429                 .hw_trig = true,
430         },
431         {
432                 .name = "external_any",
433                 .trgmod_value = AT91_SAMA5D2_TRGR_TRGMOD_EXT_TRIG_ANY,
434                 .edge_type = IRQ_TYPE_EDGE_BOTH,
435                 .hw_trig = true,
436         },
437         {
438                 .name = "software",
439                 .trgmod_value = AT91_SAMA5D2_TRGR_TRGMOD_NO_TRIGGER,
440                 .edge_type = IRQ_TYPE_NONE,
441                 .hw_trig = false,
442         },
443 };
444
445 static const struct iio_chan_spec at91_adc_channels[] = {
446         AT91_SAMA5D2_CHAN_SINGLE(0, 0x50),
447         AT91_SAMA5D2_CHAN_SINGLE(1, 0x54),
448         AT91_SAMA5D2_CHAN_SINGLE(2, 0x58),
449         AT91_SAMA5D2_CHAN_SINGLE(3, 0x5c),
450         AT91_SAMA5D2_CHAN_SINGLE(4, 0x60),
451         AT91_SAMA5D2_CHAN_SINGLE(5, 0x64),
452         AT91_SAMA5D2_CHAN_SINGLE(6, 0x68),
453         AT91_SAMA5D2_CHAN_SINGLE(7, 0x6c),
454         AT91_SAMA5D2_CHAN_SINGLE(8, 0x70),
455         AT91_SAMA5D2_CHAN_SINGLE(9, 0x74),
456         AT91_SAMA5D2_CHAN_SINGLE(10, 0x78),
457         AT91_SAMA5D2_CHAN_SINGLE(11, 0x7c),
458         AT91_SAMA5D2_CHAN_DIFF(0, 1, 0x50),
459         AT91_SAMA5D2_CHAN_DIFF(2, 3, 0x58),
460         AT91_SAMA5D2_CHAN_DIFF(4, 5, 0x60),
461         AT91_SAMA5D2_CHAN_DIFF(6, 7, 0x68),
462         AT91_SAMA5D2_CHAN_DIFF(8, 9, 0x70),
463         AT91_SAMA5D2_CHAN_DIFF(10, 11, 0x78),
464         IIO_CHAN_SOFT_TIMESTAMP(AT91_SAMA5D2_TIMESTAMP_CHAN_IDX),
465         AT91_SAMA5D2_CHAN_TOUCH(AT91_SAMA5D2_TOUCH_X_CHAN_IDX, "x", IIO_MOD_X),
466         AT91_SAMA5D2_CHAN_TOUCH(AT91_SAMA5D2_TOUCH_Y_CHAN_IDX, "y", IIO_MOD_Y),
467         AT91_SAMA5D2_CHAN_PRESSURE(AT91_SAMA5D2_TOUCH_P_CHAN_IDX, "pressure"),
468 };
469
470 static int at91_adc_chan_xlate(struct iio_dev *indio_dev, int chan)
471 {
472         int i;
473
474         for (i = 0; i < indio_dev->num_channels; i++) {
475                 if (indio_dev->channels[i].scan_index == chan)
476                         return i;
477         }
478         return -EINVAL;
479 }
480
481 static inline struct iio_chan_spec const *
482 at91_adc_chan_get(struct iio_dev *indio_dev, int chan)
483 {
484         int index = at91_adc_chan_xlate(indio_dev, chan);
485
486         if (index < 0)
487                 return NULL;
488         return indio_dev->channels + index;
489 }
490
491 static inline int at91_adc_of_xlate(struct iio_dev *indio_dev,
492                                     const struct of_phandle_args *iiospec)
493 {
494         return at91_adc_chan_xlate(indio_dev, iiospec->args[0]);
495 }
496
497 static void at91_adc_config_emr(struct at91_adc_state *st)
498 {
499         /* configure the extended mode register */
500         unsigned int emr = at91_adc_readl(st, AT91_SAMA5D2_EMR);
501
502         /* select oversampling per single trigger event */
503         emr |= AT91_SAMA5D2_EMR_ASTE(1);
504
505         /* delete leftover content if it's the case */
506         emr &= ~AT91_SAMA5D2_EMR_OSR_MASK;
507
508         /* select oversampling ratio from configuration */
509         switch (st->oversampling_ratio) {
510         case AT91_OSR_1SAMPLES:
511                 emr |= AT91_SAMA5D2_EMR_OSR(AT91_SAMA5D2_EMR_OSR_1SAMPLES) &
512                        AT91_SAMA5D2_EMR_OSR_MASK;
513                 break;
514         case AT91_OSR_4SAMPLES:
515                 emr |= AT91_SAMA5D2_EMR_OSR(AT91_SAMA5D2_EMR_OSR_4SAMPLES) &
516                        AT91_SAMA5D2_EMR_OSR_MASK;
517                 break;
518         case AT91_OSR_16SAMPLES:
519                 emr |= AT91_SAMA5D2_EMR_OSR(AT91_SAMA5D2_EMR_OSR_16SAMPLES) &
520                        AT91_SAMA5D2_EMR_OSR_MASK;
521                 break;
522         }
523
524         at91_adc_writel(st, AT91_SAMA5D2_EMR, emr);
525 }
526
527 static int at91_adc_adjust_val_osr(struct at91_adc_state *st, int *val)
528 {
529         if (st->oversampling_ratio == AT91_OSR_1SAMPLES) {
530                 /*
531                  * in this case we only have 12 bits of real data, but channel
532                  * is registered as 14 bits, so shift left two bits
533                  */
534                 *val <<= 2;
535         } else if (st->oversampling_ratio == AT91_OSR_4SAMPLES) {
536                 /*
537                  * in this case we have 13 bits of real data, but channel
538                  * is registered as 14 bits, so left shift one bit
539                  */
540                 *val <<= 1;
541         }
542
543         return IIO_VAL_INT;
544 }
545
546 static void at91_adc_adjust_val_osr_array(struct at91_adc_state *st, void *buf,
547                                           int len)
548 {
549         int i = 0, val;
550         u16 *buf_u16 = (u16 *) buf;
551
552         /*
553          * We are converting each two bytes (each sample).
554          * First convert the byte based array to u16, and convert each sample
555          * separately.
556          * Each value is two bytes in an array of chars, so to not shift
557          * more than we need, save the value separately.
558          * len is in bytes, so divide by two to get number of samples.
559          */
560         while (i < len / 2) {
561                 val = buf_u16[i];
562                 at91_adc_adjust_val_osr(st, &val);
563                 buf_u16[i] = val;
564                 i++;
565         }
566 }
567
568 static int at91_adc_configure_touch(struct at91_adc_state *st, bool state)
569 {
570         u32 clk_khz = st->current_sample_rate / 1000;
571         int i = 0;
572         u16 pendbc;
573         u32 tsmr, acr;
574
575         if (!state) {
576                 /* disabling touch IRQs and setting mode to no touch enabled */
577                 at91_adc_writel(st, AT91_SAMA5D2_IDR,
578                                 AT91_SAMA5D2_IER_PEN | AT91_SAMA5D2_IER_NOPEN);
579                 at91_adc_writel(st, AT91_SAMA5D2_TSMR, 0);
580                 return 0;
581         }
582         /*
583          * debounce time is in microseconds, we need it in milliseconds to
584          * multiply with kilohertz, so, divide by 1000, but after the multiply.
585          * round up to make sure pendbc is at least 1
586          */
587         pendbc = round_up(AT91_SAMA5D2_TOUCH_PEN_DETECT_DEBOUNCE_US *
588                           clk_khz / 1000, 1);
589
590         /* get the required exponent */
591         while (pendbc >> i++)
592                 ;
593
594         pendbc = i;
595
596         tsmr = AT91_SAMA5D2_TSMR_TSMODE_4WIRE_PRESS;
597
598         tsmr |= AT91_SAMA5D2_TSMR_TSAV(2) & AT91_SAMA5D2_TSMR_TSAV_MASK;
599         tsmr |= AT91_SAMA5D2_TSMR_PENDBC(pendbc) &
600                 AT91_SAMA5D2_TSMR_PENDBC_MASK;
601         tsmr |= AT91_SAMA5D2_TSMR_NOTSDMA;
602         tsmr |= AT91_SAMA5D2_TSMR_PENDET_ENA;
603         tsmr |= AT91_SAMA5D2_TSMR_TSFREQ(2) & AT91_SAMA5D2_TSMR_TSFREQ_MASK;
604
605         at91_adc_writel(st, AT91_SAMA5D2_TSMR, tsmr);
606
607         acr =  at91_adc_readl(st, AT91_SAMA5D2_ACR);
608         acr &= ~AT91_SAMA5D2_ACR_PENDETSENS_MASK;
609         acr |= 0x02 & AT91_SAMA5D2_ACR_PENDETSENS_MASK;
610         at91_adc_writel(st, AT91_SAMA5D2_ACR, acr);
611
612         /* Sample Period Time = (TRGPER + 1) / ADCClock */
613         st->touch_st.sample_period_val =
614                                  round_up((AT91_SAMA5D2_TOUCH_SAMPLE_PERIOD_US *
615                                  clk_khz / 1000) - 1, 1);
616         /* enable pen detect IRQ */
617         at91_adc_writel(st, AT91_SAMA5D2_IER, AT91_SAMA5D2_IER_PEN);
618
619         return 0;
620 }
621
622 static u16 at91_adc_touch_pos(struct at91_adc_state *st, int reg)
623 {
624         u32 val;
625         u32 scale, result, pos;
626
627         /*
628          * to obtain the actual position we must divide by scale
629          * and multiply with max, where
630          * max = 2^AT91_SAMA5D2_MAX_POS_BITS - 1
631          */
632         /* first half of register is the x or y, second half is the scale */
633         val = at91_adc_readl(st, reg);
634         if (!val)
635                 dev_dbg(&iio_priv_to_dev(st)->dev, "pos is 0\n");
636
637         pos = val & AT91_SAMA5D2_XYZ_MASK;
638         result = (pos << AT91_SAMA5D2_MAX_POS_BITS) - pos;
639         scale = (val >> 16) & AT91_SAMA5D2_XYZ_MASK;
640         if (scale == 0) {
641                 dev_err(&iio_priv_to_dev(st)->dev, "scale is 0\n");
642                 return 0;
643         }
644         result /= scale;
645
646         return result;
647 }
648
649 static u16 at91_adc_touch_x_pos(struct at91_adc_state *st)
650 {
651         st->touch_st.x_pos = at91_adc_touch_pos(st, AT91_SAMA5D2_XPOSR);
652         return st->touch_st.x_pos;
653 }
654
655 static u16 at91_adc_touch_y_pos(struct at91_adc_state *st)
656 {
657         return at91_adc_touch_pos(st, AT91_SAMA5D2_YPOSR);
658 }
659
660 static u16 at91_adc_touch_pressure(struct at91_adc_state *st)
661 {
662         u32 val;
663         u32 z1, z2;
664         u32 pres;
665         u32 rxp = 1;
666         u32 factor = 1000;
667
668         /* calculate the pressure */
669         val = at91_adc_readl(st, AT91_SAMA5D2_PRESSR);
670         z1 = val & AT91_SAMA5D2_XYZ_MASK;
671         z2 = (val >> 16) & AT91_SAMA5D2_XYZ_MASK;
672
673         if (z1 != 0)
674                 pres = rxp * (st->touch_st.x_pos * factor / 1024) *
675                         (z2 * factor / z1 - factor) /
676                         factor;
677         else
678                 pres = 0xFFFF;       /* no pen contact */
679
680         /*
681          * The pressure from device grows down, minimum is 0xFFFF, maximum 0x0.
682          * We compute it this way, but let's return it in the expected way,
683          * growing from 0 to 0xFFFF.
684          */
685         return 0xFFFF - pres;
686 }
687
688 static int at91_adc_read_position(struct at91_adc_state *st, int chan, u16 *val)
689 {
690         *val = 0;
691         if (!st->touch_st.touching)
692                 return -ENODATA;
693         if (chan == AT91_SAMA5D2_TOUCH_X_CHAN_IDX)
694                 *val = at91_adc_touch_x_pos(st);
695         else if (chan == AT91_SAMA5D2_TOUCH_Y_CHAN_IDX)
696                 *val = at91_adc_touch_y_pos(st);
697         else
698                 return -ENODATA;
699
700         return IIO_VAL_INT;
701 }
702
703 static int at91_adc_read_pressure(struct at91_adc_state *st, int chan, u16 *val)
704 {
705         *val = 0;
706         if (!st->touch_st.touching)
707                 return -ENODATA;
708         if (chan == AT91_SAMA5D2_TOUCH_P_CHAN_IDX)
709                 *val = at91_adc_touch_pressure(st);
710         else
711                 return -ENODATA;
712
713         return IIO_VAL_INT;
714 }
715
716 static int at91_adc_configure_trigger(struct iio_trigger *trig, bool state)
717 {
718         struct iio_dev *indio = iio_trigger_get_drvdata(trig);
719         struct at91_adc_state *st = iio_priv(indio);
720         u32 status = at91_adc_readl(st, AT91_SAMA5D2_TRGR);
721         u8 bit;
722
723         /* clear TRGMOD */
724         status &= ~AT91_SAMA5D2_TRGR_TRGMOD_MASK;
725
726         if (state)
727                 status |= st->selected_trig->trgmod_value;
728
729         /* set/unset hw trigger */
730         at91_adc_writel(st, AT91_SAMA5D2_TRGR, status);
731
732         for_each_set_bit(bit, indio->active_scan_mask, indio->num_channels) {
733                 struct iio_chan_spec const *chan = at91_adc_chan_get(indio, bit);
734
735                 if (!chan)
736                         continue;
737                 /* these channel types cannot be handled by this trigger */
738                 if (chan->type == IIO_POSITIONRELATIVE ||
739                     chan->type == IIO_PRESSURE)
740                         continue;
741
742                 if (state) {
743                         at91_adc_writel(st, AT91_SAMA5D2_CHER,
744                                         BIT(chan->channel));
745                         /* enable irq only if not using DMA */
746                         if (!st->dma_st.dma_chan) {
747                                 at91_adc_writel(st, AT91_SAMA5D2_IER,
748                                                 BIT(chan->channel));
749                         }
750                 } else {
751                         /* disable irq only if not using DMA */
752                         if (!st->dma_st.dma_chan) {
753                                 at91_adc_writel(st, AT91_SAMA5D2_IDR,
754                                                 BIT(chan->channel));
755                         }
756                         at91_adc_writel(st, AT91_SAMA5D2_CHDR,
757                                         BIT(chan->channel));
758                 }
759         }
760
761         return 0;
762 }
763
764 static int at91_adc_reenable_trigger(struct iio_trigger *trig)
765 {
766         struct iio_dev *indio = iio_trigger_get_drvdata(trig);
767         struct at91_adc_state *st = iio_priv(indio);
768
769         /* if we are using DMA, we must not reenable irq after each trigger */
770         if (st->dma_st.dma_chan)
771                 return 0;
772
773         enable_irq(st->irq);
774
775         /* Needed to ACK the DRDY interruption */
776         at91_adc_readl(st, AT91_SAMA5D2_LCDR);
777         return 0;
778 }
779
780 static const struct iio_trigger_ops at91_adc_trigger_ops = {
781         .set_trigger_state = &at91_adc_configure_trigger,
782         .try_reenable = &at91_adc_reenable_trigger,
783         .validate_device = iio_trigger_validate_own_device,
784 };
785
786 static int at91_adc_dma_size_done(struct at91_adc_state *st)
787 {
788         struct dma_tx_state state;
789         enum dma_status status;
790         int i, size;
791
792         status = dmaengine_tx_status(st->dma_st.dma_chan,
793                                      st->dma_st.dma_chan->cookie,
794                                      &state);
795         if (status != DMA_IN_PROGRESS)
796                 return 0;
797
798         /* Transferred length is size in bytes from end of buffer */
799         i = st->dma_st.rx_buf_sz - state.residue;
800
801         /* Return available bytes */
802         if (i >= st->dma_st.buf_idx)
803                 size = i - st->dma_st.buf_idx;
804         else
805                 size = st->dma_st.rx_buf_sz + i - st->dma_st.buf_idx;
806         return size;
807 }
808
809 static void at91_dma_buffer_done(void *data)
810 {
811         struct iio_dev *indio_dev = data;
812
813         iio_trigger_poll_chained(indio_dev->trig);
814 }
815
816 static int at91_adc_dma_start(struct iio_dev *indio_dev)
817 {
818         struct at91_adc_state *st = iio_priv(indio_dev);
819         struct dma_async_tx_descriptor *desc;
820         dma_cookie_t cookie;
821         int ret;
822         u8 bit;
823
824         if (!st->dma_st.dma_chan)
825                 return 0;
826
827         /* we start a new DMA, so set buffer index to start */
828         st->dma_st.buf_idx = 0;
829
830         /*
831          * compute buffer size w.r.t. watermark and enabled channels.
832          * scan_bytes is aligned so we need an exact size for DMA
833          */
834         st->dma_st.rx_buf_sz = 0;
835
836         for_each_set_bit(bit, indio_dev->active_scan_mask,
837                          indio_dev->num_channels) {
838                 struct iio_chan_spec const *chan =
839                                          at91_adc_chan_get(indio_dev, bit);
840
841                 if (!chan)
842                         continue;
843
844                 st->dma_st.rx_buf_sz += chan->scan_type.storagebits / 8;
845         }
846         st->dma_st.rx_buf_sz *= st->dma_st.watermark;
847
848         /* Prepare a DMA cyclic transaction */
849         desc = dmaengine_prep_dma_cyclic(st->dma_st.dma_chan,
850                                          st->dma_st.rx_dma_buf,
851                                          st->dma_st.rx_buf_sz,
852                                          st->dma_st.rx_buf_sz / 2,
853                                          DMA_DEV_TO_MEM, DMA_PREP_INTERRUPT);
854
855         if (!desc) {
856                 dev_err(&indio_dev->dev, "cannot prepare DMA cyclic\n");
857                 return -EBUSY;
858         }
859
860         desc->callback = at91_dma_buffer_done;
861         desc->callback_param = indio_dev;
862
863         cookie = dmaengine_submit(desc);
864         ret = dma_submit_error(cookie);
865         if (ret) {
866                 dev_err(&indio_dev->dev, "cannot submit DMA cyclic\n");
867                 dmaengine_terminate_async(st->dma_st.dma_chan);
868                 return ret;
869         }
870
871         /* enable general overrun error signaling */
872         at91_adc_writel(st, AT91_SAMA5D2_IER, AT91_SAMA5D2_IER_GOVRE);
873         /* Issue pending DMA requests */
874         dma_async_issue_pending(st->dma_st.dma_chan);
875
876         /* consider current time as DMA start time for timestamps */
877         st->dma_st.dma_ts = iio_get_time_ns(indio_dev);
878
879         dev_dbg(&indio_dev->dev, "DMA cyclic started\n");
880
881         return 0;
882 }
883
884 static int at91_adc_buffer_postenable(struct iio_dev *indio_dev)
885 {
886         int ret;
887         struct at91_adc_state *st = iio_priv(indio_dev);
888
889         /* check if we are enabling triggered buffer or the touchscreen */
890         if (bitmap_subset(indio_dev->active_scan_mask,
891                           &st->touch_st.channels_bitmask,
892                           AT91_SAMA5D2_MAX_CHAN_IDX + 1)) {
893                 /* touchscreen enabling */
894                 return at91_adc_configure_touch(st, true);
895         }
896         /* if we are not in triggered mode, we cannot enable the buffer. */
897         if (!(indio_dev->currentmode & INDIO_ALL_TRIGGERED_MODES))
898                 return -EINVAL;
899
900         /* we continue with the triggered buffer */
901         ret = at91_adc_dma_start(indio_dev);
902         if (ret) {
903                 dev_err(&indio_dev->dev, "buffer postenable failed\n");
904                 return ret;
905         }
906
907         return iio_triggered_buffer_postenable(indio_dev);
908 }
909
910 static int at91_adc_buffer_predisable(struct iio_dev *indio_dev)
911 {
912         struct at91_adc_state *st = iio_priv(indio_dev);
913         int ret;
914         u8 bit;
915
916         /* check if we are disabling triggered buffer or the touchscreen */
917         if (bitmap_subset(indio_dev->active_scan_mask,
918                           &st->touch_st.channels_bitmask,
919                           AT91_SAMA5D2_MAX_CHAN_IDX + 1)) {
920                 /* touchscreen disable */
921                 return at91_adc_configure_touch(st, false);
922         }
923         /* if we are not in triggered mode, nothing to do here */
924         if (!(indio_dev->currentmode & INDIO_ALL_TRIGGERED_MODES))
925                 return -EINVAL;
926
927         /* continue with the triggered buffer */
928         ret = iio_triggered_buffer_predisable(indio_dev);
929         if (ret < 0)
930                 dev_err(&indio_dev->dev, "buffer predisable failed\n");
931
932         if (!st->dma_st.dma_chan)
933                 return ret;
934
935         /* if we are using DMA we must clear registers and end DMA */
936         dmaengine_terminate_sync(st->dma_st.dma_chan);
937
938         /*
939          * For each enabled channel we must read the last converted value
940          * to clear EOC status and not get a possible interrupt later.
941          * This value is being read by DMA from LCDR anyway
942          */
943         for_each_set_bit(bit, indio_dev->active_scan_mask,
944                          indio_dev->num_channels) {
945                 struct iio_chan_spec const *chan =
946                                         at91_adc_chan_get(indio_dev, bit);
947
948                 if (!chan)
949                         continue;
950                 /* these channel types are virtual, no need to do anything */
951                 if (chan->type == IIO_POSITIONRELATIVE ||
952                     chan->type == IIO_PRESSURE)
953                         continue;
954                 if (st->dma_st.dma_chan)
955                         at91_adc_readl(st, chan->address);
956         }
957
958         /* read overflow register to clear possible overflow status */
959         at91_adc_readl(st, AT91_SAMA5D2_OVER);
960         return ret;
961 }
962
963 static const struct iio_buffer_setup_ops at91_buffer_setup_ops = {
964         .postenable = &at91_adc_buffer_postenable,
965         .predisable = &at91_adc_buffer_predisable,
966 };
967
968 static struct iio_trigger *at91_adc_allocate_trigger(struct iio_dev *indio,
969                                                      char *trigger_name)
970 {
971         struct iio_trigger *trig;
972         int ret;
973
974         trig = devm_iio_trigger_alloc(&indio->dev, "%s-dev%d-%s", indio->name,
975                                       indio->id, trigger_name);
976         if (!trig)
977                 return NULL;
978
979         trig->dev.parent = indio->dev.parent;
980         iio_trigger_set_drvdata(trig, indio);
981         trig->ops = &at91_adc_trigger_ops;
982
983         ret = devm_iio_trigger_register(&indio->dev, trig);
984         if (ret)
985                 return ERR_PTR(ret);
986
987         return trig;
988 }
989
990 static int at91_adc_trigger_init(struct iio_dev *indio)
991 {
992         struct at91_adc_state *st = iio_priv(indio);
993
994         st->trig = at91_adc_allocate_trigger(indio, st->selected_trig->name);
995         if (IS_ERR(st->trig)) {
996                 dev_err(&indio->dev,
997                         "could not allocate trigger\n");
998                 return PTR_ERR(st->trig);
999         }
1000
1001         return 0;
1002 }
1003
1004 static void at91_adc_trigger_handler_nodma(struct iio_dev *indio_dev,
1005                                            struct iio_poll_func *pf)
1006 {
1007         struct at91_adc_state *st = iio_priv(indio_dev);
1008         int i = 0;
1009         int val;
1010         u8 bit;
1011
1012         for_each_set_bit(bit, indio_dev->active_scan_mask,
1013                          indio_dev->num_channels) {
1014                 struct iio_chan_spec const *chan =
1015                                         at91_adc_chan_get(indio_dev, bit);
1016
1017                 if (!chan)
1018                         continue;
1019                 /*
1020                  * Our external trigger only supports the voltage channels.
1021                  * In case someone requested a different type of channel
1022                  * just put zeroes to buffer.
1023                  * This should not happen because we check the scan mode
1024                  * and scan mask when we enable the buffer, and we don't allow
1025                  * the buffer to start with a mixed mask (voltage and something
1026                  * else).
1027                  * Thus, emit a warning.
1028                  */
1029                 if (chan->type == IIO_VOLTAGE) {
1030                         val = at91_adc_readl(st, chan->address);
1031                         at91_adc_adjust_val_osr(st, &val);
1032                         st->buffer[i] = val;
1033                 } else {
1034                         st->buffer[i] = 0;
1035                         WARN(true, "This trigger cannot handle this type of channel");
1036                 }
1037                 i++;
1038         }
1039         iio_push_to_buffers_with_timestamp(indio_dev, st->buffer,
1040                                            pf->timestamp);
1041 }
1042
1043 static void at91_adc_trigger_handler_dma(struct iio_dev *indio_dev)
1044 {
1045         struct at91_adc_state *st = iio_priv(indio_dev);
1046         int transferred_len = at91_adc_dma_size_done(st);
1047         s64 ns = iio_get_time_ns(indio_dev);
1048         s64 interval;
1049         int sample_index = 0, sample_count, sample_size;
1050
1051         u32 status = at91_adc_readl(st, AT91_SAMA5D2_ISR);
1052         /* if we reached this point, we cannot sample faster */
1053         if (status & AT91_SAMA5D2_IER_GOVRE)
1054                 pr_info_ratelimited("%s: conversion overrun detected\n",
1055                                     indio_dev->name);
1056
1057         sample_size = div_s64(st->dma_st.rx_buf_sz, st->dma_st.watermark);
1058
1059         sample_count = div_s64(transferred_len, sample_size);
1060
1061         /*
1062          * interval between samples is total time since last transfer handling
1063          * divided by the number of samples (total size divided by sample size)
1064          */
1065         interval = div_s64((ns - st->dma_st.dma_ts), sample_count);
1066
1067         while (transferred_len >= sample_size) {
1068                 /*
1069                  * for all the values in the current sample,
1070                  * adjust the values inside the buffer for oversampling
1071                  */
1072                 at91_adc_adjust_val_osr_array(st,
1073                                         &st->dma_st.rx_buf[st->dma_st.buf_idx],
1074                                         sample_size);
1075
1076                 iio_push_to_buffers_with_timestamp(indio_dev,
1077                                 (st->dma_st.rx_buf + st->dma_st.buf_idx),
1078                                 (st->dma_st.dma_ts + interval * sample_index));
1079                 /* adjust remaining length */
1080                 transferred_len -= sample_size;
1081                 /* adjust buffer index */
1082                 st->dma_st.buf_idx += sample_size;
1083                 /* in case of reaching end of buffer, reset index */
1084                 if (st->dma_st.buf_idx >= st->dma_st.rx_buf_sz)
1085                         st->dma_st.buf_idx = 0;
1086                 sample_index++;
1087         }
1088         /* adjust saved time for next transfer handling */
1089         st->dma_st.dma_ts = iio_get_time_ns(indio_dev);
1090 }
1091
1092 static irqreturn_t at91_adc_trigger_handler(int irq, void *p)
1093 {
1094         struct iio_poll_func *pf = p;
1095         struct iio_dev *indio_dev = pf->indio_dev;
1096         struct at91_adc_state *st = iio_priv(indio_dev);
1097
1098         if (st->dma_st.dma_chan)
1099                 at91_adc_trigger_handler_dma(indio_dev);
1100         else
1101                 at91_adc_trigger_handler_nodma(indio_dev, pf);
1102
1103         iio_trigger_notify_done(indio_dev->trig);
1104
1105         return IRQ_HANDLED;
1106 }
1107
1108 static int at91_adc_buffer_init(struct iio_dev *indio)
1109 {
1110         struct at91_adc_state *st = iio_priv(indio);
1111
1112         if (st->selected_trig->hw_trig) {
1113                 return devm_iio_triggered_buffer_setup(&indio->dev, indio,
1114                         &iio_pollfunc_store_time,
1115                         &at91_adc_trigger_handler, &at91_buffer_setup_ops);
1116         }
1117         /*
1118          * we need to prepare the buffer ops in case we will get
1119          * another buffer attached (like a callback buffer for the touchscreen)
1120          */
1121         indio->setup_ops = &at91_buffer_setup_ops;
1122
1123         return 0;
1124 }
1125
1126 static unsigned at91_adc_startup_time(unsigned startup_time_min,
1127                                       unsigned adc_clk_khz)
1128 {
1129         static const unsigned int startup_lookup[] = {
1130                   0,   8,  16,  24,
1131                  64,  80,  96, 112,
1132                 512, 576, 640, 704,
1133                 768, 832, 896, 960
1134                 };
1135         unsigned ticks_min, i;
1136
1137         /*
1138          * Since the adc frequency is checked before, there is no reason
1139          * to not meet the startup time constraint.
1140          */
1141
1142         ticks_min = startup_time_min * adc_clk_khz / 1000;
1143         for (i = 0; i < ARRAY_SIZE(startup_lookup); i++)
1144                 if (startup_lookup[i] > ticks_min)
1145                         break;
1146
1147         return i;
1148 }
1149
1150 static void at91_adc_setup_samp_freq(struct at91_adc_state *st, unsigned freq)
1151 {
1152         struct iio_dev *indio_dev = iio_priv_to_dev(st);
1153         unsigned f_per, prescal, startup, mr;
1154
1155         f_per = clk_get_rate(st->per_clk);
1156         prescal = (f_per / (2 * freq)) - 1;
1157
1158         startup = at91_adc_startup_time(st->soc_info.startup_time,
1159                                         freq / 1000);
1160
1161         mr = at91_adc_readl(st, AT91_SAMA5D2_MR);
1162         mr &= ~(AT91_SAMA5D2_MR_STARTUP_MASK | AT91_SAMA5D2_MR_PRESCAL_MASK);
1163         mr |= AT91_SAMA5D2_MR_STARTUP(startup);
1164         mr |= AT91_SAMA5D2_MR_PRESCAL(prescal);
1165         at91_adc_writel(st, AT91_SAMA5D2_MR, mr);
1166
1167         dev_dbg(&indio_dev->dev, "freq: %u, startup: %u, prescal: %u\n",
1168                 freq, startup, prescal);
1169         st->current_sample_rate = freq;
1170 }
1171
1172 static inline unsigned at91_adc_get_sample_freq(struct at91_adc_state *st)
1173 {
1174         return st->current_sample_rate;
1175 }
1176
1177 static void at91_adc_touch_data_handler(struct iio_dev *indio_dev)
1178 {
1179         struct at91_adc_state *st = iio_priv(indio_dev);
1180         u8 bit;
1181         u16 val;
1182         int i = 0;
1183
1184         for_each_set_bit(bit, indio_dev->active_scan_mask,
1185                          AT91_SAMA5D2_MAX_CHAN_IDX + 1) {
1186                 struct iio_chan_spec const *chan =
1187                                          at91_adc_chan_get(indio_dev, bit);
1188
1189                 if (chan->type == IIO_POSITIONRELATIVE)
1190                         at91_adc_read_position(st, chan->channel, &val);
1191                 else if (chan->type == IIO_PRESSURE)
1192                         at91_adc_read_pressure(st, chan->channel, &val);
1193                 else
1194                         continue;
1195                 st->buffer[i] = val;
1196                 i++;
1197         }
1198         /*
1199          * Schedule work to push to buffers.
1200          * This is intended to push to the callback buffer that another driver
1201          * registered. We are still in a handler from our IRQ. If we push
1202          * directly, it means the other driver has it's callback called
1203          * from our IRQ context. Which is something we better avoid.
1204          * Let's schedule it after our IRQ is completed.
1205          */
1206         schedule_work(&st->touch_st.workq);
1207 }
1208
1209 static void at91_adc_pen_detect_interrupt(struct at91_adc_state *st)
1210 {
1211         at91_adc_writel(st, AT91_SAMA5D2_IDR, AT91_SAMA5D2_IER_PEN);
1212         at91_adc_writel(st, AT91_SAMA5D2_IER, AT91_SAMA5D2_IER_NOPEN |
1213                         AT91_SAMA5D2_IER_XRDY | AT91_SAMA5D2_IER_YRDY |
1214                         AT91_SAMA5D2_IER_PRDY);
1215         at91_adc_writel(st, AT91_SAMA5D2_TRGR,
1216                         AT91_SAMA5D2_TRGR_TRGMOD_PERIODIC |
1217                         AT91_SAMA5D2_TRGR_TRGPER(st->touch_st.sample_period_val));
1218         st->touch_st.touching = true;
1219 }
1220
1221 static void at91_adc_no_pen_detect_interrupt(struct at91_adc_state *st)
1222 {
1223         struct iio_dev *indio_dev = iio_priv_to_dev(st);
1224
1225         at91_adc_writel(st, AT91_SAMA5D2_TRGR,
1226                         AT91_SAMA5D2_TRGR_TRGMOD_NO_TRIGGER);
1227         at91_adc_writel(st, AT91_SAMA5D2_IDR, AT91_SAMA5D2_IER_NOPEN |
1228                         AT91_SAMA5D2_IER_XRDY | AT91_SAMA5D2_IER_YRDY |
1229                         AT91_SAMA5D2_IER_PRDY);
1230         st->touch_st.touching = false;
1231
1232         at91_adc_touch_data_handler(indio_dev);
1233
1234         at91_adc_writel(st, AT91_SAMA5D2_IER, AT91_SAMA5D2_IER_PEN);
1235 }
1236
1237 static void at91_adc_workq_handler(struct work_struct *workq)
1238 {
1239         struct at91_adc_touch *touch_st = container_of(workq,
1240                                         struct at91_adc_touch, workq);
1241         struct at91_adc_state *st = container_of(touch_st,
1242                                         struct at91_adc_state, touch_st);
1243         struct iio_dev *indio_dev = iio_priv_to_dev(st);
1244
1245         iio_push_to_buffers(indio_dev, st->buffer);
1246 }
1247
1248 static irqreturn_t at91_adc_interrupt(int irq, void *private)
1249 {
1250         struct iio_dev *indio = private;
1251         struct at91_adc_state *st = iio_priv(indio);
1252         u32 status = at91_adc_readl(st, AT91_SAMA5D2_ISR);
1253         u32 imr = at91_adc_readl(st, AT91_SAMA5D2_IMR);
1254         u32 rdy_mask = AT91_SAMA5D2_IER_XRDY | AT91_SAMA5D2_IER_YRDY |
1255                         AT91_SAMA5D2_IER_PRDY;
1256
1257         if (!(status & imr))
1258                 return IRQ_NONE;
1259         if (status & AT91_SAMA5D2_IER_PEN) {
1260                 /* pen detected IRQ */
1261                 at91_adc_pen_detect_interrupt(st);
1262         } else if ((status & AT91_SAMA5D2_IER_NOPEN)) {
1263                 /* nopen detected IRQ */
1264                 at91_adc_no_pen_detect_interrupt(st);
1265         } else if ((status & AT91_SAMA5D2_ISR_PENS) &&
1266                    ((status & rdy_mask) == rdy_mask)) {
1267                 /* periodic trigger IRQ - during pen sense */
1268                 at91_adc_touch_data_handler(indio);
1269         } else if (status & AT91_SAMA5D2_ISR_PENS) {
1270                 /*
1271                  * touching, but the measurements are not ready yet.
1272                  * read and ignore.
1273                  */
1274                 status = at91_adc_readl(st, AT91_SAMA5D2_XPOSR);
1275                 status = at91_adc_readl(st, AT91_SAMA5D2_YPOSR);
1276                 status = at91_adc_readl(st, AT91_SAMA5D2_PRESSR);
1277         } else if (iio_buffer_enabled(indio) && !st->dma_st.dma_chan) {
1278                 /* triggered buffer without DMA */
1279                 disable_irq_nosync(irq);
1280                 iio_trigger_poll(indio->trig);
1281         } else if (iio_buffer_enabled(indio) && st->dma_st.dma_chan) {
1282                 /* triggered buffer with DMA - should not happen */
1283                 disable_irq_nosync(irq);
1284                 WARN(true, "Unexpected irq occurred\n");
1285         } else if (!iio_buffer_enabled(indio)) {
1286                 /* software requested conversion */
1287                 st->conversion_value = at91_adc_readl(st, st->chan->address);
1288                 st->conversion_done = true;
1289                 wake_up_interruptible(&st->wq_data_available);
1290         }
1291         return IRQ_HANDLED;
1292 }
1293
1294 static int at91_adc_read_info_raw(struct iio_dev *indio_dev,
1295                                   struct iio_chan_spec const *chan, int *val)
1296 {
1297         struct at91_adc_state *st = iio_priv(indio_dev);
1298         u32 cor = 0;
1299         u16 tmp_val;
1300         int ret;
1301
1302         /*
1303          * Keep in mind that we cannot use software trigger or touchscreen
1304          * if external trigger is enabled
1305          */
1306         if (chan->type == IIO_POSITIONRELATIVE) {
1307                 ret = iio_device_claim_direct_mode(indio_dev);
1308                 if (ret)
1309                         return ret;
1310                 mutex_lock(&st->lock);
1311
1312                 ret = at91_adc_read_position(st, chan->channel,
1313                                              &tmp_val);
1314                 *val = tmp_val;
1315                 mutex_unlock(&st->lock);
1316                 iio_device_release_direct_mode(indio_dev);
1317
1318                 return at91_adc_adjust_val_osr(st, val);
1319         }
1320         if (chan->type == IIO_PRESSURE) {
1321                 ret = iio_device_claim_direct_mode(indio_dev);
1322                 if (ret)
1323                         return ret;
1324                 mutex_lock(&st->lock);
1325
1326                 ret = at91_adc_read_pressure(st, chan->channel,
1327                                              &tmp_val);
1328                 *val = tmp_val;
1329                 mutex_unlock(&st->lock);
1330                 iio_device_release_direct_mode(indio_dev);
1331
1332                 return at91_adc_adjust_val_osr(st, val);
1333         }
1334
1335         /* in this case we have a voltage channel */
1336
1337         ret = iio_device_claim_direct_mode(indio_dev);
1338         if (ret)
1339                 return ret;
1340         mutex_lock(&st->lock);
1341
1342         st->chan = chan;
1343
1344         if (chan->differential)
1345                 cor = (BIT(chan->channel) | BIT(chan->channel2)) <<
1346                       AT91_SAMA5D2_COR_DIFF_OFFSET;
1347
1348         at91_adc_writel(st, AT91_SAMA5D2_COR, cor);
1349         at91_adc_writel(st, AT91_SAMA5D2_CHER, BIT(chan->channel));
1350         at91_adc_writel(st, AT91_SAMA5D2_IER, BIT(chan->channel));
1351         at91_adc_writel(st, AT91_SAMA5D2_CR, AT91_SAMA5D2_CR_START);
1352
1353         ret = wait_event_interruptible_timeout(st->wq_data_available,
1354                                                st->conversion_done,
1355                                                msecs_to_jiffies(1000));
1356         if (ret == 0)
1357                 ret = -ETIMEDOUT;
1358
1359         if (ret > 0) {
1360                 *val = st->conversion_value;
1361                 ret = at91_adc_adjust_val_osr(st, val);
1362                 if (chan->scan_type.sign == 's')
1363                         *val = sign_extend32(*val, 11);
1364                 st->conversion_done = false;
1365         }
1366
1367         at91_adc_writel(st, AT91_SAMA5D2_IDR, BIT(chan->channel));
1368         at91_adc_writel(st, AT91_SAMA5D2_CHDR, BIT(chan->channel));
1369
1370         /* Needed to ACK the DRDY interruption */
1371         at91_adc_readl(st, AT91_SAMA5D2_LCDR);
1372
1373         mutex_unlock(&st->lock);
1374
1375         iio_device_release_direct_mode(indio_dev);
1376         return ret;
1377 }
1378
1379 static int at91_adc_read_raw(struct iio_dev *indio_dev,
1380                              struct iio_chan_spec const *chan,
1381                              int *val, int *val2, long mask)
1382 {
1383         struct at91_adc_state *st = iio_priv(indio_dev);
1384
1385         switch (mask) {
1386         case IIO_CHAN_INFO_RAW:
1387                 return at91_adc_read_info_raw(indio_dev, chan, val);
1388         case IIO_CHAN_INFO_SCALE:
1389                 *val = st->vref_uv / 1000;
1390                 if (chan->differential)
1391                         *val *= 2;
1392                 *val2 = chan->scan_type.realbits;
1393                 return IIO_VAL_FRACTIONAL_LOG2;
1394
1395         case IIO_CHAN_INFO_SAMP_FREQ:
1396                 *val = at91_adc_get_sample_freq(st);
1397                 return IIO_VAL_INT;
1398
1399         case IIO_CHAN_INFO_OVERSAMPLING_RATIO:
1400                 *val = st->oversampling_ratio;
1401                 return IIO_VAL_INT;
1402
1403         default:
1404                 return -EINVAL;
1405         }
1406 }
1407
1408 static int at91_adc_write_raw(struct iio_dev *indio_dev,
1409                               struct iio_chan_spec const *chan,
1410                               int val, int val2, long mask)
1411 {
1412         struct at91_adc_state *st = iio_priv(indio_dev);
1413
1414         switch (mask) {
1415         case IIO_CHAN_INFO_OVERSAMPLING_RATIO:
1416                 if ((val != AT91_OSR_1SAMPLES) && (val != AT91_OSR_4SAMPLES) &&
1417                     (val != AT91_OSR_16SAMPLES))
1418                         return -EINVAL;
1419                 /* if no change, optimize out */
1420                 if (val == st->oversampling_ratio)
1421                         return 0;
1422                 st->oversampling_ratio = val;
1423                 /* update ratio */
1424                 at91_adc_config_emr(st);
1425                 return 0;
1426         case IIO_CHAN_INFO_SAMP_FREQ:
1427                 if (val < st->soc_info.min_sample_rate ||
1428                     val > st->soc_info.max_sample_rate)
1429                         return -EINVAL;
1430
1431                 at91_adc_setup_samp_freq(st, val);
1432                 return 0;
1433         default:
1434                 return -EINVAL;
1435         };
1436 }
1437
1438 static void at91_adc_dma_init(struct platform_device *pdev)
1439 {
1440         struct iio_dev *indio_dev = platform_get_drvdata(pdev);
1441         struct at91_adc_state *st = iio_priv(indio_dev);
1442         struct dma_slave_config config = {0};
1443         /*
1444          * We make the buffer double the size of the fifo,
1445          * such that DMA uses one half of the buffer (full fifo size)
1446          * and the software uses the other half to read/write.
1447          */
1448         unsigned int pages = DIV_ROUND_UP(AT91_HWFIFO_MAX_SIZE *
1449                                           AT91_BUFFER_MAX_CONVERSION_BYTES * 2,
1450                                           PAGE_SIZE);
1451
1452         if (st->dma_st.dma_chan)
1453                 return;
1454
1455         st->dma_st.dma_chan = dma_request_slave_channel(&pdev->dev, "rx");
1456
1457         if (!st->dma_st.dma_chan)  {
1458                 dev_info(&pdev->dev, "can't get DMA channel\n");
1459                 goto dma_exit;
1460         }
1461
1462         st->dma_st.rx_buf = dma_alloc_coherent(st->dma_st.dma_chan->device->dev,
1463                                                pages * PAGE_SIZE,
1464                                                &st->dma_st.rx_dma_buf,
1465                                                GFP_KERNEL);
1466         if (!st->dma_st.rx_buf) {
1467                 dev_info(&pdev->dev, "can't allocate coherent DMA area\n");
1468                 goto dma_chan_disable;
1469         }
1470
1471         /* Configure DMA channel to read data register */
1472         config.direction = DMA_DEV_TO_MEM;
1473         config.src_addr = (phys_addr_t)(st->dma_st.phys_addr
1474                           + AT91_SAMA5D2_LCDR);
1475         config.src_addr_width = DMA_SLAVE_BUSWIDTH_2_BYTES;
1476         config.src_maxburst = 1;
1477         config.dst_maxburst = 1;
1478
1479         if (dmaengine_slave_config(st->dma_st.dma_chan, &config)) {
1480                 dev_info(&pdev->dev, "can't configure DMA slave\n");
1481                 goto dma_free_area;
1482         }
1483
1484         dev_info(&pdev->dev, "using %s for rx DMA transfers\n",
1485                  dma_chan_name(st->dma_st.dma_chan));
1486
1487         return;
1488
1489 dma_free_area:
1490         dma_free_coherent(st->dma_st.dma_chan->device->dev, pages * PAGE_SIZE,
1491                           st->dma_st.rx_buf, st->dma_st.rx_dma_buf);
1492 dma_chan_disable:
1493         dma_release_channel(st->dma_st.dma_chan);
1494         st->dma_st.dma_chan = 0;
1495 dma_exit:
1496         dev_info(&pdev->dev, "continuing without DMA support\n");
1497 }
1498
1499 static void at91_adc_dma_disable(struct platform_device *pdev)
1500 {
1501         struct iio_dev *indio_dev = platform_get_drvdata(pdev);
1502         struct at91_adc_state *st = iio_priv(indio_dev);
1503         unsigned int pages = DIV_ROUND_UP(AT91_HWFIFO_MAX_SIZE *
1504                                           AT91_BUFFER_MAX_CONVERSION_BYTES * 2,
1505                                           PAGE_SIZE);
1506
1507         /* if we are not using DMA, just return */
1508         if (!st->dma_st.dma_chan)
1509                 return;
1510
1511         /* wait for all transactions to be terminated first*/
1512         dmaengine_terminate_sync(st->dma_st.dma_chan);
1513
1514         dma_free_coherent(st->dma_st.dma_chan->device->dev, pages * PAGE_SIZE,
1515                           st->dma_st.rx_buf, st->dma_st.rx_dma_buf);
1516         dma_release_channel(st->dma_st.dma_chan);
1517         st->dma_st.dma_chan = 0;
1518
1519         dev_info(&pdev->dev, "continuing without DMA support\n");
1520 }
1521
1522 static int at91_adc_set_watermark(struct iio_dev *indio_dev, unsigned int val)
1523 {
1524         struct at91_adc_state *st = iio_priv(indio_dev);
1525
1526         if (val > AT91_HWFIFO_MAX_SIZE)
1527                 return -EINVAL;
1528
1529         if (!st->selected_trig->hw_trig) {
1530                 dev_dbg(&indio_dev->dev, "we need hw trigger for DMA\n");
1531                 return 0;
1532         }
1533
1534         dev_dbg(&indio_dev->dev, "new watermark is %u\n", val);
1535         st->dma_st.watermark = val;
1536
1537         /*
1538          * The logic here is: if we have watermark 1, it means we do
1539          * each conversion with it's own IRQ, thus we don't need DMA.
1540          * If the watermark is higher, we do DMA to do all the transfers in bulk
1541          */
1542
1543         if (val == 1)
1544                 at91_adc_dma_disable(to_platform_device(&indio_dev->dev));
1545         else if (val > 1)
1546                 at91_adc_dma_init(to_platform_device(&indio_dev->dev));
1547
1548         return 0;
1549 }
1550
1551 static int at91_adc_update_scan_mode(struct iio_dev *indio_dev,
1552                                      const unsigned long *scan_mask)
1553 {
1554         struct at91_adc_state *st = iio_priv(indio_dev);
1555
1556         if (bitmap_subset(scan_mask, &st->touch_st.channels_bitmask,
1557                           AT91_SAMA5D2_MAX_CHAN_IDX + 1))
1558                 return 0;
1559         /*
1560          * if the new bitmap is a combination of touchscreen and regular
1561          * channels, then we are not fine
1562          */
1563         if (bitmap_intersects(&st->touch_st.channels_bitmask, scan_mask,
1564                               AT91_SAMA5D2_MAX_CHAN_IDX + 1))
1565                 return -EINVAL;
1566         return 0;
1567 }
1568
1569 static void at91_adc_hw_init(struct at91_adc_state *st)
1570 {
1571         at91_adc_writel(st, AT91_SAMA5D2_CR, AT91_SAMA5D2_CR_SWRST);
1572         at91_adc_writel(st, AT91_SAMA5D2_IDR, 0xffffffff);
1573         /*
1574          * Transfer field must be set to 2 according to the datasheet and
1575          * allows different analog settings for each channel.
1576          */
1577         at91_adc_writel(st, AT91_SAMA5D2_MR,
1578                         AT91_SAMA5D2_MR_TRANSFER(2) | AT91_SAMA5D2_MR_ANACH);
1579
1580         at91_adc_setup_samp_freq(st, st->soc_info.min_sample_rate);
1581
1582         /* configure extended mode register */
1583         at91_adc_config_emr(st);
1584 }
1585
1586 static ssize_t at91_adc_get_fifo_state(struct device *dev,
1587                                        struct device_attribute *attr, char *buf)
1588 {
1589         struct iio_dev *indio_dev =
1590                         platform_get_drvdata(to_platform_device(dev));
1591         struct at91_adc_state *st = iio_priv(indio_dev);
1592
1593         return scnprintf(buf, PAGE_SIZE, "%d\n", !!st->dma_st.dma_chan);
1594 }
1595
1596 static ssize_t at91_adc_get_watermark(struct device *dev,
1597                                       struct device_attribute *attr, char *buf)
1598 {
1599         struct iio_dev *indio_dev =
1600                         platform_get_drvdata(to_platform_device(dev));
1601         struct at91_adc_state *st = iio_priv(indio_dev);
1602
1603         return scnprintf(buf, PAGE_SIZE, "%d\n", st->dma_st.watermark);
1604 }
1605
1606 static IIO_DEVICE_ATTR(hwfifo_enabled, 0444,
1607                        at91_adc_get_fifo_state, NULL, 0);
1608 static IIO_DEVICE_ATTR(hwfifo_watermark, 0444,
1609                        at91_adc_get_watermark, NULL, 0);
1610
1611 static IIO_CONST_ATTR(hwfifo_watermark_min, "2");
1612 static IIO_CONST_ATTR(hwfifo_watermark_max, AT91_HWFIFO_MAX_SIZE_STR);
1613
1614 static IIO_CONST_ATTR(oversampling_ratio_available,
1615                       __stringify(AT91_OSR_1SAMPLES) " "
1616                       __stringify(AT91_OSR_4SAMPLES) " "
1617                       __stringify(AT91_OSR_16SAMPLES));
1618
1619 static struct attribute *at91_adc_attributes[] = {
1620         &iio_const_attr_oversampling_ratio_available.dev_attr.attr,
1621         NULL,
1622 };
1623
1624 static const struct attribute_group at91_adc_attribute_group = {
1625         .attrs = at91_adc_attributes,
1626 };
1627
1628 static const struct attribute *at91_adc_fifo_attributes[] = {
1629         &iio_const_attr_hwfifo_watermark_min.dev_attr.attr,
1630         &iio_const_attr_hwfifo_watermark_max.dev_attr.attr,
1631         &iio_dev_attr_hwfifo_watermark.dev_attr.attr,
1632         &iio_dev_attr_hwfifo_enabled.dev_attr.attr,
1633         NULL,
1634 };
1635
1636 static const struct iio_info at91_adc_info = {
1637         .attrs = &at91_adc_attribute_group,
1638         .read_raw = &at91_adc_read_raw,
1639         .write_raw = &at91_adc_write_raw,
1640         .update_scan_mode = &at91_adc_update_scan_mode,
1641         .of_xlate = &at91_adc_of_xlate,
1642         .hwfifo_set_watermark = &at91_adc_set_watermark,
1643 };
1644
1645 static int at91_adc_probe(struct platform_device *pdev)
1646 {
1647         struct iio_dev *indio_dev;
1648         struct at91_adc_state *st;
1649         struct resource *res;
1650         int ret, i;
1651         u32 edge_type = IRQ_TYPE_NONE;
1652
1653         indio_dev = devm_iio_device_alloc(&pdev->dev, sizeof(*st));
1654         if (!indio_dev)
1655                 return -ENOMEM;
1656
1657         indio_dev->dev.parent = &pdev->dev;
1658         indio_dev->name = dev_name(&pdev->dev);
1659         indio_dev->modes = INDIO_DIRECT_MODE | INDIO_BUFFER_SOFTWARE;
1660         indio_dev->info = &at91_adc_info;
1661         indio_dev->channels = at91_adc_channels;
1662         indio_dev->num_channels = ARRAY_SIZE(at91_adc_channels);
1663
1664         st = iio_priv(indio_dev);
1665
1666         bitmap_set(&st->touch_st.channels_bitmask,
1667                    AT91_SAMA5D2_TOUCH_X_CHAN_IDX, 1);
1668         bitmap_set(&st->touch_st.channels_bitmask,
1669                    AT91_SAMA5D2_TOUCH_Y_CHAN_IDX, 1);
1670         bitmap_set(&st->touch_st.channels_bitmask,
1671                    AT91_SAMA5D2_TOUCH_P_CHAN_IDX, 1);
1672
1673         st->oversampling_ratio = AT91_OSR_1SAMPLES;
1674
1675         ret = of_property_read_u32(pdev->dev.of_node,
1676                                    "atmel,min-sample-rate-hz",
1677                                    &st->soc_info.min_sample_rate);
1678         if (ret) {
1679                 dev_err(&pdev->dev,
1680                         "invalid or missing value for atmel,min-sample-rate-hz\n");
1681                 return ret;
1682         }
1683
1684         ret = of_property_read_u32(pdev->dev.of_node,
1685                                    "atmel,max-sample-rate-hz",
1686                                    &st->soc_info.max_sample_rate);
1687         if (ret) {
1688                 dev_err(&pdev->dev,
1689                         "invalid or missing value for atmel,max-sample-rate-hz\n");
1690                 return ret;
1691         }
1692
1693         ret = of_property_read_u32(pdev->dev.of_node, "atmel,startup-time-ms",
1694                                    &st->soc_info.startup_time);
1695         if (ret) {
1696                 dev_err(&pdev->dev,
1697                         "invalid or missing value for atmel,startup-time-ms\n");
1698                 return ret;
1699         }
1700
1701         ret = of_property_read_u32(pdev->dev.of_node,
1702                                    "atmel,trigger-edge-type", &edge_type);
1703         if (ret) {
1704                 dev_dbg(&pdev->dev,
1705                         "atmel,trigger-edge-type not specified, only software trigger available\n");
1706         }
1707
1708         st->selected_trig = NULL;
1709
1710         /* find the right trigger, or no trigger at all */
1711         for (i = 0; i < AT91_SAMA5D2_HW_TRIG_CNT + 1; i++)
1712                 if (at91_adc_trigger_list[i].edge_type == edge_type) {
1713                         st->selected_trig = &at91_adc_trigger_list[i];
1714                         break;
1715                 }
1716
1717         if (!st->selected_trig) {
1718                 dev_err(&pdev->dev, "invalid external trigger edge value\n");
1719                 return -EINVAL;
1720         }
1721
1722         init_waitqueue_head(&st->wq_data_available);
1723         mutex_init(&st->lock);
1724         INIT_WORK(&st->touch_st.workq, at91_adc_workq_handler);
1725
1726         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
1727         if (!res)
1728                 return -EINVAL;
1729
1730         /* if we plan to use DMA, we need the physical address of the regs */
1731         st->dma_st.phys_addr = res->start;
1732
1733         st->base = devm_ioremap_resource(&pdev->dev, res);
1734         if (IS_ERR(st->base))
1735                 return PTR_ERR(st->base);
1736
1737         st->irq = platform_get_irq(pdev, 0);
1738         if (st->irq <= 0) {
1739                 if (!st->irq)
1740                         st->irq = -ENXIO;
1741
1742                 return st->irq;
1743         }
1744
1745         st->per_clk = devm_clk_get(&pdev->dev, "adc_clk");
1746         if (IS_ERR(st->per_clk))
1747                 return PTR_ERR(st->per_clk);
1748
1749         st->reg = devm_regulator_get(&pdev->dev, "vddana");
1750         if (IS_ERR(st->reg))
1751                 return PTR_ERR(st->reg);
1752
1753         st->vref = devm_regulator_get(&pdev->dev, "vref");
1754         if (IS_ERR(st->vref))
1755                 return PTR_ERR(st->vref);
1756
1757         ret = devm_request_irq(&pdev->dev, st->irq, at91_adc_interrupt, 0,
1758                                pdev->dev.driver->name, indio_dev);
1759         if (ret)
1760                 return ret;
1761
1762         ret = regulator_enable(st->reg);
1763         if (ret)
1764                 return ret;
1765
1766         ret = regulator_enable(st->vref);
1767         if (ret)
1768                 goto reg_disable;
1769
1770         st->vref_uv = regulator_get_voltage(st->vref);
1771         if (st->vref_uv <= 0) {
1772                 ret = -EINVAL;
1773                 goto vref_disable;
1774         }
1775
1776         at91_adc_hw_init(st);
1777
1778         ret = clk_prepare_enable(st->per_clk);
1779         if (ret)
1780                 goto vref_disable;
1781
1782         platform_set_drvdata(pdev, indio_dev);
1783
1784         ret = at91_adc_buffer_init(indio_dev);
1785         if (ret < 0) {
1786                 dev_err(&pdev->dev, "couldn't initialize the buffer.\n");
1787                 goto per_clk_disable_unprepare;
1788         }
1789
1790         if (st->selected_trig->hw_trig) {
1791                 ret = at91_adc_trigger_init(indio_dev);
1792                 if (ret < 0) {
1793                         dev_err(&pdev->dev, "couldn't setup the triggers.\n");
1794                         goto per_clk_disable_unprepare;
1795                 }
1796                 /*
1797                  * Initially the iio buffer has a length of 2 and
1798                  * a watermark of 1
1799                  */
1800                 st->dma_st.watermark = 1;
1801
1802                 iio_buffer_set_attrs(indio_dev->buffer,
1803                                      at91_adc_fifo_attributes);
1804         }
1805
1806         if (dma_coerce_mask_and_coherent(&indio_dev->dev, DMA_BIT_MASK(32)))
1807                 dev_info(&pdev->dev, "cannot set DMA mask to 32-bit\n");
1808
1809         ret = iio_device_register(indio_dev);
1810         if (ret < 0)
1811                 goto dma_disable;
1812
1813         if (st->selected_trig->hw_trig)
1814                 dev_info(&pdev->dev, "setting up trigger as %s\n",
1815                          st->selected_trig->name);
1816
1817         dev_info(&pdev->dev, "version: %x\n",
1818                  readl_relaxed(st->base + AT91_SAMA5D2_VERSION));
1819
1820         return 0;
1821
1822 dma_disable:
1823         at91_adc_dma_disable(pdev);
1824 per_clk_disable_unprepare:
1825         clk_disable_unprepare(st->per_clk);
1826 vref_disable:
1827         regulator_disable(st->vref);
1828 reg_disable:
1829         regulator_disable(st->reg);
1830         return ret;
1831 }
1832
1833 static int at91_adc_remove(struct platform_device *pdev)
1834 {
1835         struct iio_dev *indio_dev = platform_get_drvdata(pdev);
1836         struct at91_adc_state *st = iio_priv(indio_dev);
1837
1838         iio_device_unregister(indio_dev);
1839
1840         at91_adc_dma_disable(pdev);
1841
1842         clk_disable_unprepare(st->per_clk);
1843
1844         regulator_disable(st->vref);
1845         regulator_disable(st->reg);
1846
1847         return 0;
1848 }
1849
1850 static __maybe_unused int at91_adc_suspend(struct device *dev)
1851 {
1852         struct iio_dev *indio_dev =
1853                         platform_get_drvdata(to_platform_device(dev));
1854         struct at91_adc_state *st = iio_priv(indio_dev);
1855
1856         /*
1857          * Do a sofware reset of the ADC before we go to suspend.
1858          * this will ensure that all pins are free from being muxed by the ADC
1859          * and can be used by for other devices.
1860          * Otherwise, ADC will hog them and we can't go to suspend mode.
1861          */
1862         at91_adc_writel(st, AT91_SAMA5D2_CR, AT91_SAMA5D2_CR_SWRST);
1863
1864         clk_disable_unprepare(st->per_clk);
1865         regulator_disable(st->vref);
1866         regulator_disable(st->reg);
1867
1868         return pinctrl_pm_select_sleep_state(dev);
1869 }
1870
1871 static __maybe_unused int at91_adc_resume(struct device *dev)
1872 {
1873         struct iio_dev *indio_dev =
1874                         platform_get_drvdata(to_platform_device(dev));
1875         struct at91_adc_state *st = iio_priv(indio_dev);
1876         int ret;
1877
1878         ret = pinctrl_pm_select_default_state(dev);
1879         if (ret)
1880                 goto resume_failed;
1881
1882         ret = regulator_enable(st->reg);
1883         if (ret)
1884                 goto resume_failed;
1885
1886         ret = regulator_enable(st->vref);
1887         if (ret)
1888                 goto reg_disable_resume;
1889
1890         ret = clk_prepare_enable(st->per_clk);
1891         if (ret)
1892                 goto vref_disable_resume;
1893
1894         at91_adc_hw_init(st);
1895
1896         /* reconfiguring trigger hardware state */
1897         if (!iio_buffer_enabled(indio_dev))
1898                 return 0;
1899
1900         /* check if we are enabling triggered buffer or the touchscreen */
1901         if (bitmap_subset(indio_dev->active_scan_mask,
1902                           &st->touch_st.channels_bitmask,
1903                           AT91_SAMA5D2_MAX_CHAN_IDX + 1)) {
1904                 /* touchscreen enabling */
1905                 return at91_adc_configure_touch(st, true);
1906         } else {
1907                 return at91_adc_configure_trigger(st->trig, true);
1908         }
1909
1910         /* not needed but more explicit */
1911         return 0;
1912
1913 vref_disable_resume:
1914         regulator_disable(st->vref);
1915 reg_disable_resume:
1916         regulator_disable(st->reg);
1917 resume_failed:
1918         dev_err(&indio_dev->dev, "failed to resume\n");
1919         return ret;
1920 }
1921
1922 static SIMPLE_DEV_PM_OPS(at91_adc_pm_ops, at91_adc_suspend, at91_adc_resume);
1923
1924 static const struct of_device_id at91_adc_dt_match[] = {
1925         {
1926                 .compatible = "atmel,sama5d2-adc",
1927         }, {
1928                 /* sentinel */
1929         }
1930 };
1931 MODULE_DEVICE_TABLE(of, at91_adc_dt_match);
1932
1933 static struct platform_driver at91_adc_driver = {
1934         .probe = at91_adc_probe,
1935         .remove = at91_adc_remove,
1936         .driver = {
1937                 .name = "at91-sama5d2_adc",
1938                 .of_match_table = at91_adc_dt_match,
1939                 .pm = &at91_adc_pm_ops,
1940         },
1941 };
1942 module_platform_driver(at91_adc_driver)
1943
1944 MODULE_AUTHOR("Ludovic Desroches <ludovic.desroches@atmel.com>");
1945 MODULE_DESCRIPTION("Atmel AT91 SAMA5D2 ADC");
1946 MODULE_LICENSE("GPL v2");