1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * ADC0831/ADC0832/ADC0834/ADC0838 8-bit ADC driver
4  *
5  * Copyright (c) 2016 Akinobu Mita <akinobu.mita@gmail.com>
6  *
7  * Datasheet: https://www.ti.com/lit/ds/symlink/adc0832-n.pdf
8  */
9 
10 #include <linux/module.h>
11 #include <linux/mod_devicetable.h>
12 #include <linux/spi/spi.h>
13 #include <linux/iio/iio.h>
14 #include <linux/regulator/consumer.h>
15 #include <linux/iio/buffer.h>
16 #include <linux/iio/trigger.h>
17 #include <linux/iio/triggered_buffer.h>
18 #include <linux/iio/trigger_consumer.h>
19 
20 enum {
21 	adc0831,
22 	adc0832,
23 	adc0834,
24 	adc0838,
25 };
26 
27 struct adc0832 {
28 	struct spi_device *spi;
29 	struct regulator *reg;
30 	struct mutex lock;
31 	u8 mux_bits;
32 	/*
33 	 * Max size needed: 16x 1 byte ADC data + 8 bytes timestamp
34 	 * May be shorter if not all channels are enabled subject
35 	 * to the timestamp remaining 8 byte aligned.
36 	 */
37 	u8 data[24] __aligned(8);
38 
39 	u8 tx_buf[2] __aligned(IIO_DMA_MINALIGN);
40 	u8 rx_buf[2];
41 };
42 
43 #define ADC0832_VOLTAGE_CHANNEL(chan)					\
44 	{								\
45 		.type = IIO_VOLTAGE,					\
46 		.indexed = 1,						\
47 		.channel = chan,					\
48 		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),		\
49 		.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),	\
50 		.scan_index = chan,					\
51 		.scan_type = {						\
52 			.sign = 'u',					\
53 			.realbits = 8,					\
54 			.storagebits = 8,				\
55 		},							\
56 	}
57 
58 #define ADC0832_VOLTAGE_CHANNEL_DIFF(chan1, chan2, si)			\
59 	{								\
60 		.type = IIO_VOLTAGE,					\
61 		.indexed = 1,						\
62 		.channel = (chan1),					\
63 		.channel2 = (chan2),					\
64 		.differential = 1,					\
65 		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),		\
66 		.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),	\
67 		.scan_index = si,					\
68 		.scan_type = {						\
69 			.sign = 'u',					\
70 			.realbits = 8,					\
71 			.storagebits = 8,				\
72 		},							\
73 	}
74 
75 static const struct iio_chan_spec adc0831_channels[] = {
76 	ADC0832_VOLTAGE_CHANNEL_DIFF(0, 1, 0),
77 	IIO_CHAN_SOFT_TIMESTAMP(1),
78 };
79 
80 static const struct iio_chan_spec adc0832_channels[] = {
81 	ADC0832_VOLTAGE_CHANNEL(0),
82 	ADC0832_VOLTAGE_CHANNEL(1),
83 	ADC0832_VOLTAGE_CHANNEL_DIFF(0, 1, 2),
84 	ADC0832_VOLTAGE_CHANNEL_DIFF(1, 0, 3),
85 	IIO_CHAN_SOFT_TIMESTAMP(4),
86 };
87 
88 static const struct iio_chan_spec adc0834_channels[] = {
89 	ADC0832_VOLTAGE_CHANNEL(0),
90 	ADC0832_VOLTAGE_CHANNEL(1),
91 	ADC0832_VOLTAGE_CHANNEL(2),
92 	ADC0832_VOLTAGE_CHANNEL(3),
93 	ADC0832_VOLTAGE_CHANNEL_DIFF(0, 1, 4),
94 	ADC0832_VOLTAGE_CHANNEL_DIFF(1, 0, 5),
95 	ADC0832_VOLTAGE_CHANNEL_DIFF(2, 3, 6),
96 	ADC0832_VOLTAGE_CHANNEL_DIFF(3, 2, 7),
97 	IIO_CHAN_SOFT_TIMESTAMP(8),
98 };
99 
100 static const struct iio_chan_spec adc0838_channels[] = {
101 	ADC0832_VOLTAGE_CHANNEL(0),
102 	ADC0832_VOLTAGE_CHANNEL(1),
103 	ADC0832_VOLTAGE_CHANNEL(2),
104 	ADC0832_VOLTAGE_CHANNEL(3),
105 	ADC0832_VOLTAGE_CHANNEL(4),
106 	ADC0832_VOLTAGE_CHANNEL(5),
107 	ADC0832_VOLTAGE_CHANNEL(6),
108 	ADC0832_VOLTAGE_CHANNEL(7),
109 	ADC0832_VOLTAGE_CHANNEL_DIFF(0, 1, 8),
110 	ADC0832_VOLTAGE_CHANNEL_DIFF(1, 0, 9),
111 	ADC0832_VOLTAGE_CHANNEL_DIFF(2, 3, 10),
112 	ADC0832_VOLTAGE_CHANNEL_DIFF(3, 2, 11),
113 	ADC0832_VOLTAGE_CHANNEL_DIFF(4, 5, 12),
114 	ADC0832_VOLTAGE_CHANNEL_DIFF(5, 4, 13),
115 	ADC0832_VOLTAGE_CHANNEL_DIFF(6, 7, 14),
116 	ADC0832_VOLTAGE_CHANNEL_DIFF(7, 6, 15),
117 	IIO_CHAN_SOFT_TIMESTAMP(16),
118 };
119 
adc0831_adc_conversion(struct adc0832 * adc)120 static int adc0831_adc_conversion(struct adc0832 *adc)
121 {
122 	struct spi_device *spi = adc->spi;
123 	int ret;
124 
125 	ret = spi_read(spi, &adc->rx_buf, 2);
126 	if (ret)
127 		return ret;
128 
129 	/*
130 	 * Skip TRI-STATE and a leading zero
131 	 */
132 	return (adc->rx_buf[0] << 2 & 0xff) | (adc->rx_buf[1] >> 6);
133 }
134 
adc0832_adc_conversion(struct adc0832 * adc,int channel,bool differential)135 static int adc0832_adc_conversion(struct adc0832 *adc, int channel,
136 				bool differential)
137 {
138 	struct spi_device *spi = adc->spi;
139 	struct spi_transfer xfer = {
140 		.tx_buf = adc->tx_buf,
141 		.rx_buf = adc->rx_buf,
142 		.len = 2,
143 	};
144 	int ret;
145 
146 	if (!adc->mux_bits)
147 		return adc0831_adc_conversion(adc);
148 
149 	/* start bit */
150 	adc->tx_buf[0] = 1 << (adc->mux_bits + 1);
151 	/* single-ended or differential */
152 	adc->tx_buf[0] |= differential ? 0 : (1 << adc->mux_bits);
153 	/* odd / sign */
154 	adc->tx_buf[0] |= (channel % 2) << (adc->mux_bits - 1);
155 	/* select */
156 	if (adc->mux_bits > 1)
157 		adc->tx_buf[0] |= channel / 2;
158 
159 	/* align Data output BIT7 (MSB) to 8-bit boundary */
160 	adc->tx_buf[0] <<= 1;
161 
162 	ret = spi_sync_transfer(spi, &xfer, 1);
163 	if (ret)
164 		return ret;
165 
166 	return adc->rx_buf[1];
167 }
168 
adc0832_read_raw(struct iio_dev * iio,struct iio_chan_spec const * channel,int * value,int * shift,long mask)169 static int adc0832_read_raw(struct iio_dev *iio,
170 			struct iio_chan_spec const *channel, int *value,
171 			int *shift, long mask)
172 {
173 	struct adc0832 *adc = iio_priv(iio);
174 
175 	switch (mask) {
176 	case IIO_CHAN_INFO_RAW:
177 		mutex_lock(&adc->lock);
178 		*value = adc0832_adc_conversion(adc, channel->channel,
179 						channel->differential);
180 		mutex_unlock(&adc->lock);
181 		if (*value < 0)
182 			return *value;
183 
184 		return IIO_VAL_INT;
185 	case IIO_CHAN_INFO_SCALE:
186 		*value = regulator_get_voltage(adc->reg);
187 		if (*value < 0)
188 			return *value;
189 
190 		/* convert regulator output voltage to mV */
191 		*value /= 1000;
192 		*shift = 8;
193 
194 		return IIO_VAL_FRACTIONAL_LOG2;
195 	}
196 
197 	return -EINVAL;
198 }
199 
200 static const struct iio_info adc0832_info = {
201 	.read_raw = adc0832_read_raw,
202 };
203 
adc0832_trigger_handler(int irq,void * p)204 static irqreturn_t adc0832_trigger_handler(int irq, void *p)
205 {
206 	struct iio_poll_func *pf = p;
207 	struct iio_dev *indio_dev = pf->indio_dev;
208 	struct adc0832 *adc = iio_priv(indio_dev);
209 	int scan_index;
210 	int i = 0;
211 
212 	mutex_lock(&adc->lock);
213 
214 	for_each_set_bit(scan_index, indio_dev->active_scan_mask,
215 			 indio_dev->masklength) {
216 		const struct iio_chan_spec *scan_chan =
217 				&indio_dev->channels[scan_index];
218 		int ret = adc0832_adc_conversion(adc, scan_chan->channel,
219 						 scan_chan->differential);
220 		if (ret < 0) {
221 			dev_warn(&adc->spi->dev,
222 				 "failed to get conversion data\n");
223 			goto out;
224 		}
225 
226 		adc->data[i] = ret;
227 		i++;
228 	}
229 	iio_push_to_buffers_with_timestamp(indio_dev, adc->data,
230 					   iio_get_time_ns(indio_dev));
231 out:
232 	mutex_unlock(&adc->lock);
233 
234 	iio_trigger_notify_done(indio_dev->trig);
235 
236 	return IRQ_HANDLED;
237 }
238 
adc0832_reg_disable(void * reg)239 static void adc0832_reg_disable(void *reg)
240 {
241 	regulator_disable(reg);
242 }
243 
adc0832_probe(struct spi_device * spi)244 static int adc0832_probe(struct spi_device *spi)
245 {
246 	struct iio_dev *indio_dev;
247 	struct adc0832 *adc;
248 	int ret;
249 
250 	indio_dev = devm_iio_device_alloc(&spi->dev, sizeof(*adc));
251 	if (!indio_dev)
252 		return -ENOMEM;
253 
254 	adc = iio_priv(indio_dev);
255 	adc->spi = spi;
256 	mutex_init(&adc->lock);
257 
258 	indio_dev->name = spi_get_device_id(spi)->name;
259 	indio_dev->info = &adc0832_info;
260 	indio_dev->modes = INDIO_DIRECT_MODE;
261 
262 	switch (spi_get_device_id(spi)->driver_data) {
263 	case adc0831:
264 		adc->mux_bits = 0;
265 		indio_dev->channels = adc0831_channels;
266 		indio_dev->num_channels = ARRAY_SIZE(adc0831_channels);
267 		break;
268 	case adc0832:
269 		adc->mux_bits = 1;
270 		indio_dev->channels = adc0832_channels;
271 		indio_dev->num_channels = ARRAY_SIZE(adc0832_channels);
272 		break;
273 	case adc0834:
274 		adc->mux_bits = 2;
275 		indio_dev->channels = adc0834_channels;
276 		indio_dev->num_channels = ARRAY_SIZE(adc0834_channels);
277 		break;
278 	case adc0838:
279 		adc->mux_bits = 3;
280 		indio_dev->channels = adc0838_channels;
281 		indio_dev->num_channels = ARRAY_SIZE(adc0838_channels);
282 		break;
283 	default:
284 		return -EINVAL;
285 	}
286 
287 	adc->reg = devm_regulator_get(&spi->dev, "vref");
288 	if (IS_ERR(adc->reg))
289 		return PTR_ERR(adc->reg);
290 
291 	ret = regulator_enable(adc->reg);
292 	if (ret)
293 		return ret;
294 
295 	ret = devm_add_action_or_reset(&spi->dev, adc0832_reg_disable,
296 				       adc->reg);
297 	if (ret)
298 		return ret;
299 
300 	ret = devm_iio_triggered_buffer_setup(&spi->dev, indio_dev, NULL,
301 					      adc0832_trigger_handler, NULL);
302 	if (ret)
303 		return ret;
304 
305 	return devm_iio_device_register(&spi->dev, indio_dev);
306 }
307 
308 static const struct of_device_id adc0832_dt_ids[] = {
309 	{ .compatible = "ti,adc0831", },
310 	{ .compatible = "ti,adc0832", },
311 	{ .compatible = "ti,adc0834", },
312 	{ .compatible = "ti,adc0838", },
313 	{}
314 };
315 MODULE_DEVICE_TABLE(of, adc0832_dt_ids);
316 
317 static const struct spi_device_id adc0832_id[] = {
318 	{ "adc0831", adc0831 },
319 	{ "adc0832", adc0832 },
320 	{ "adc0834", adc0834 },
321 	{ "adc0838", adc0838 },
322 	{}
323 };
324 MODULE_DEVICE_TABLE(spi, adc0832_id);
325 
326 static struct spi_driver adc0832_driver = {
327 	.driver = {
328 		.name = "adc0832",
329 		.of_match_table = adc0832_dt_ids,
330 	},
331 	.probe = adc0832_probe,
332 	.id_table = adc0832_id,
333 };
334 module_spi_driver(adc0832_driver);
335 
336 MODULE_AUTHOR("Akinobu Mita <akinobu.mita@gmail.com>");
337 MODULE_DESCRIPTION("ADC0831/ADC0832/ADC0834/ADC0838 driver");
338 MODULE_LICENSE("GPL v2");
339