162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2017 Axis Communications AB
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Driver for Texas Instruments' ADC084S021 ADC chip.
662306a36Sopenharmony_ci * Datasheets can be found here:
762306a36Sopenharmony_ci * https://www.ti.com/lit/ds/symlink/adc084s021.pdf
862306a36Sopenharmony_ci */
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include <linux/err.h>
1162306a36Sopenharmony_ci#include <linux/spi/spi.h>
1262306a36Sopenharmony_ci#include <linux/module.h>
1362306a36Sopenharmony_ci#include <linux/mod_devicetable.h>
1462306a36Sopenharmony_ci#include <linux/interrupt.h>
1562306a36Sopenharmony_ci#include <linux/iio/iio.h>
1662306a36Sopenharmony_ci#include <linux/iio/buffer.h>
1762306a36Sopenharmony_ci#include <linux/iio/triggered_buffer.h>
1862306a36Sopenharmony_ci#include <linux/iio/trigger_consumer.h>
1962306a36Sopenharmony_ci#include <linux/regulator/consumer.h>
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci#define ADC084S021_DRIVER_NAME "adc084s021"
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_cistruct adc084s021 {
2462306a36Sopenharmony_ci	struct spi_device *spi;
2562306a36Sopenharmony_ci	struct spi_message message;
2662306a36Sopenharmony_ci	struct spi_transfer spi_trans;
2762306a36Sopenharmony_ci	struct regulator *reg;
2862306a36Sopenharmony_ci	struct mutex lock;
2962306a36Sopenharmony_ci	/* Buffer used to align data */
3062306a36Sopenharmony_ci	struct {
3162306a36Sopenharmony_ci		__be16 channels[4];
3262306a36Sopenharmony_ci		s64 ts __aligned(8);
3362306a36Sopenharmony_ci	} scan;
3462306a36Sopenharmony_ci	/*
3562306a36Sopenharmony_ci	 * DMA (thus cache coherency maintenance) may require the
3662306a36Sopenharmony_ci	 * transfer buffers to live in their own cache line.
3762306a36Sopenharmony_ci	 */
3862306a36Sopenharmony_ci	u16 tx_buf[4] __aligned(IIO_DMA_MINALIGN);
3962306a36Sopenharmony_ci	__be16 rx_buf[5]; /* First 16-bits are trash */
4062306a36Sopenharmony_ci};
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_ci#define ADC084S021_VOLTAGE_CHANNEL(num)                  \
4362306a36Sopenharmony_ci	{                                                      \
4462306a36Sopenharmony_ci		.type = IIO_VOLTAGE,                                 \
4562306a36Sopenharmony_ci		.channel = (num),                                    \
4662306a36Sopenharmony_ci		.indexed = 1,                                        \
4762306a36Sopenharmony_ci		.scan_index = (num),                                 \
4862306a36Sopenharmony_ci		.scan_type = {                                       \
4962306a36Sopenharmony_ci			.sign = 'u',                                       \
5062306a36Sopenharmony_ci			.realbits = 8,                                     \
5162306a36Sopenharmony_ci			.storagebits = 16,                                 \
5262306a36Sopenharmony_ci			.shift = 4,                                        \
5362306a36Sopenharmony_ci			.endianness = IIO_BE,                              \
5462306a36Sopenharmony_ci		},                                                   \
5562306a36Sopenharmony_ci		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),        \
5662306a36Sopenharmony_ci		.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),\
5762306a36Sopenharmony_ci	}
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_cistatic const struct iio_chan_spec adc084s021_channels[] = {
6062306a36Sopenharmony_ci	ADC084S021_VOLTAGE_CHANNEL(0),
6162306a36Sopenharmony_ci	ADC084S021_VOLTAGE_CHANNEL(1),
6262306a36Sopenharmony_ci	ADC084S021_VOLTAGE_CHANNEL(2),
6362306a36Sopenharmony_ci	ADC084S021_VOLTAGE_CHANNEL(3),
6462306a36Sopenharmony_ci	IIO_CHAN_SOFT_TIMESTAMP(4),
6562306a36Sopenharmony_ci};
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci/**
6862306a36Sopenharmony_ci * adc084s021_adc_conversion() - Read an ADC channel and return its value.
6962306a36Sopenharmony_ci *
7062306a36Sopenharmony_ci * @adc: The ADC SPI data.
7162306a36Sopenharmony_ci * @data: Buffer for converted data.
7262306a36Sopenharmony_ci */
7362306a36Sopenharmony_cistatic int adc084s021_adc_conversion(struct adc084s021 *adc, __be16 *data)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	int n_words = (adc->spi_trans.len >> 1) - 1; /* Discard first word */
7662306a36Sopenharmony_ci	int ret, i = 0;
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	/* Do the transfer */
7962306a36Sopenharmony_ci	ret = spi_sync(adc->spi, &adc->message);
8062306a36Sopenharmony_ci	if (ret < 0)
8162306a36Sopenharmony_ci		return ret;
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci	for (; i < n_words; i++)
8462306a36Sopenharmony_ci		*(data + i) = adc->rx_buf[i + 1];
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	return ret;
8762306a36Sopenharmony_ci}
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_cistatic int adc084s021_read_raw(struct iio_dev *indio_dev,
9062306a36Sopenharmony_ci			   struct iio_chan_spec const *channel, int *val,
9162306a36Sopenharmony_ci			   int *val2, long mask)
9262306a36Sopenharmony_ci{
9362306a36Sopenharmony_ci	struct adc084s021 *adc = iio_priv(indio_dev);
9462306a36Sopenharmony_ci	int ret;
9562306a36Sopenharmony_ci	__be16 be_val;
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	switch (mask) {
9862306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
9962306a36Sopenharmony_ci		ret = iio_device_claim_direct_mode(indio_dev);
10062306a36Sopenharmony_ci		if (ret < 0)
10162306a36Sopenharmony_ci			return ret;
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci		ret = regulator_enable(adc->reg);
10462306a36Sopenharmony_ci		if (ret) {
10562306a36Sopenharmony_ci			iio_device_release_direct_mode(indio_dev);
10662306a36Sopenharmony_ci			return ret;
10762306a36Sopenharmony_ci		}
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci		adc->tx_buf[0] = channel->channel << 3;
11062306a36Sopenharmony_ci		ret = adc084s021_adc_conversion(adc, &be_val);
11162306a36Sopenharmony_ci		iio_device_release_direct_mode(indio_dev);
11262306a36Sopenharmony_ci		regulator_disable(adc->reg);
11362306a36Sopenharmony_ci		if (ret < 0)
11462306a36Sopenharmony_ci			return ret;
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_ci		*val = be16_to_cpu(be_val);
11762306a36Sopenharmony_ci		*val = (*val >> channel->scan_type.shift) & 0xff;
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_ci		return IIO_VAL_INT;
12062306a36Sopenharmony_ci	case IIO_CHAN_INFO_SCALE:
12162306a36Sopenharmony_ci		ret = regulator_enable(adc->reg);
12262306a36Sopenharmony_ci		if (ret)
12362306a36Sopenharmony_ci			return ret;
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci		ret = regulator_get_voltage(adc->reg);
12662306a36Sopenharmony_ci		regulator_disable(adc->reg);
12762306a36Sopenharmony_ci		if (ret < 0)
12862306a36Sopenharmony_ci			return ret;
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci		*val = ret / 1000;
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_ci		return IIO_VAL_INT;
13362306a36Sopenharmony_ci	default:
13462306a36Sopenharmony_ci		return -EINVAL;
13562306a36Sopenharmony_ci	}
13662306a36Sopenharmony_ci}
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ci/**
13962306a36Sopenharmony_ci * adc084s021_buffer_trigger_handler() - Read ADC channels and push to buffer.
14062306a36Sopenharmony_ci *
14162306a36Sopenharmony_ci * @irq: The interrupt number (not used).
14262306a36Sopenharmony_ci * @pollfunc: Pointer to the poll func.
14362306a36Sopenharmony_ci */
14462306a36Sopenharmony_cistatic irqreturn_t adc084s021_buffer_trigger_handler(int irq, void *pollfunc)
14562306a36Sopenharmony_ci{
14662306a36Sopenharmony_ci	struct iio_poll_func *pf = pollfunc;
14762306a36Sopenharmony_ci	struct iio_dev *indio_dev = pf->indio_dev;
14862306a36Sopenharmony_ci	struct adc084s021 *adc = iio_priv(indio_dev);
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ci	mutex_lock(&adc->lock);
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_ci	if (adc084s021_adc_conversion(adc, adc->scan.channels) < 0)
15362306a36Sopenharmony_ci		dev_err(&adc->spi->dev, "Failed to read data\n");
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	iio_push_to_buffers_with_timestamp(indio_dev, &adc->scan,
15662306a36Sopenharmony_ci					   iio_get_time_ns(indio_dev));
15762306a36Sopenharmony_ci	mutex_unlock(&adc->lock);
15862306a36Sopenharmony_ci	iio_trigger_notify_done(indio_dev->trig);
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ci	return IRQ_HANDLED;
16162306a36Sopenharmony_ci}
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_cistatic int adc084s021_buffer_preenable(struct iio_dev *indio_dev)
16462306a36Sopenharmony_ci{
16562306a36Sopenharmony_ci	struct adc084s021 *adc = iio_priv(indio_dev);
16662306a36Sopenharmony_ci	int scan_index;
16762306a36Sopenharmony_ci	int i = 0;
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_ci	for_each_set_bit(scan_index, indio_dev->active_scan_mask,
17062306a36Sopenharmony_ci			 indio_dev->masklength) {
17162306a36Sopenharmony_ci		const struct iio_chan_spec *channel =
17262306a36Sopenharmony_ci			&indio_dev->channels[scan_index];
17362306a36Sopenharmony_ci		adc->tx_buf[i++] = channel->channel << 3;
17462306a36Sopenharmony_ci	}
17562306a36Sopenharmony_ci	adc->spi_trans.len = 2 + (i * sizeof(__be16)); /* Trash + channels */
17662306a36Sopenharmony_ci
17762306a36Sopenharmony_ci	return regulator_enable(adc->reg);
17862306a36Sopenharmony_ci}
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_cistatic int adc084s021_buffer_postdisable(struct iio_dev *indio_dev)
18162306a36Sopenharmony_ci{
18262306a36Sopenharmony_ci	struct adc084s021 *adc = iio_priv(indio_dev);
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_ci	adc->spi_trans.len = 4; /* Trash + single channel */
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_ci	return regulator_disable(adc->reg);
18762306a36Sopenharmony_ci}
18862306a36Sopenharmony_ci
18962306a36Sopenharmony_cistatic const struct iio_info adc084s021_info = {
19062306a36Sopenharmony_ci	.read_raw = adc084s021_read_raw,
19162306a36Sopenharmony_ci};
19262306a36Sopenharmony_ci
19362306a36Sopenharmony_cistatic const struct iio_buffer_setup_ops adc084s021_buffer_setup_ops = {
19462306a36Sopenharmony_ci	.preenable = adc084s021_buffer_preenable,
19562306a36Sopenharmony_ci	.postdisable = adc084s021_buffer_postdisable,
19662306a36Sopenharmony_ci};
19762306a36Sopenharmony_ci
19862306a36Sopenharmony_cistatic int adc084s021_probe(struct spi_device *spi)
19962306a36Sopenharmony_ci{
20062306a36Sopenharmony_ci	struct iio_dev *indio_dev;
20162306a36Sopenharmony_ci	struct adc084s021 *adc;
20262306a36Sopenharmony_ci	int ret;
20362306a36Sopenharmony_ci
20462306a36Sopenharmony_ci	indio_dev = devm_iio_device_alloc(&spi->dev, sizeof(*adc));
20562306a36Sopenharmony_ci	if (!indio_dev) {
20662306a36Sopenharmony_ci		dev_err(&spi->dev, "Failed to allocate IIO device\n");
20762306a36Sopenharmony_ci		return -ENOMEM;
20862306a36Sopenharmony_ci	}
20962306a36Sopenharmony_ci
21062306a36Sopenharmony_ci	adc = iio_priv(indio_dev);
21162306a36Sopenharmony_ci	adc->spi = spi;
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci	/* Initiate the Industrial I/O device */
21462306a36Sopenharmony_ci	indio_dev->name = spi_get_device_id(spi)->name;
21562306a36Sopenharmony_ci	indio_dev->modes = INDIO_DIRECT_MODE;
21662306a36Sopenharmony_ci	indio_dev->info = &adc084s021_info;
21762306a36Sopenharmony_ci	indio_dev->channels = adc084s021_channels;
21862306a36Sopenharmony_ci	indio_dev->num_channels = ARRAY_SIZE(adc084s021_channels);
21962306a36Sopenharmony_ci
22062306a36Sopenharmony_ci	/* Create SPI transfer for channel reads */
22162306a36Sopenharmony_ci	adc->spi_trans.tx_buf = adc->tx_buf;
22262306a36Sopenharmony_ci	adc->spi_trans.rx_buf = adc->rx_buf;
22362306a36Sopenharmony_ci	adc->spi_trans.len = 4; /* Trash + single channel */
22462306a36Sopenharmony_ci	spi_message_init_with_transfers(&adc->message, &adc->spi_trans, 1);
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	adc->reg = devm_regulator_get(&spi->dev, "vref");
22762306a36Sopenharmony_ci	if (IS_ERR(adc->reg))
22862306a36Sopenharmony_ci		return PTR_ERR(adc->reg);
22962306a36Sopenharmony_ci
23062306a36Sopenharmony_ci	mutex_init(&adc->lock);
23162306a36Sopenharmony_ci
23262306a36Sopenharmony_ci	/* Setup triggered buffer with pollfunction */
23362306a36Sopenharmony_ci	ret = devm_iio_triggered_buffer_setup(&spi->dev, indio_dev, NULL,
23462306a36Sopenharmony_ci					    adc084s021_buffer_trigger_handler,
23562306a36Sopenharmony_ci					    &adc084s021_buffer_setup_ops);
23662306a36Sopenharmony_ci	if (ret) {
23762306a36Sopenharmony_ci		dev_err(&spi->dev, "Failed to setup triggered buffer\n");
23862306a36Sopenharmony_ci		return ret;
23962306a36Sopenharmony_ci	}
24062306a36Sopenharmony_ci
24162306a36Sopenharmony_ci	return devm_iio_device_register(&spi->dev, indio_dev);
24262306a36Sopenharmony_ci}
24362306a36Sopenharmony_ci
24462306a36Sopenharmony_cistatic const struct of_device_id adc084s021_of_match[] = {
24562306a36Sopenharmony_ci	{ .compatible = "ti,adc084s021", },
24662306a36Sopenharmony_ci	{},
24762306a36Sopenharmony_ci};
24862306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, adc084s021_of_match);
24962306a36Sopenharmony_ci
25062306a36Sopenharmony_cistatic const struct spi_device_id adc084s021_id[] = {
25162306a36Sopenharmony_ci	{ ADC084S021_DRIVER_NAME, 0 },
25262306a36Sopenharmony_ci	{}
25362306a36Sopenharmony_ci};
25462306a36Sopenharmony_ciMODULE_DEVICE_TABLE(spi, adc084s021_id);
25562306a36Sopenharmony_ci
25662306a36Sopenharmony_cistatic struct spi_driver adc084s021_driver = {
25762306a36Sopenharmony_ci	.driver = {
25862306a36Sopenharmony_ci		.name = ADC084S021_DRIVER_NAME,
25962306a36Sopenharmony_ci		.of_match_table = adc084s021_of_match,
26062306a36Sopenharmony_ci	},
26162306a36Sopenharmony_ci	.probe = adc084s021_probe,
26262306a36Sopenharmony_ci	.id_table = adc084s021_id,
26362306a36Sopenharmony_ci};
26462306a36Sopenharmony_cimodule_spi_driver(adc084s021_driver);
26562306a36Sopenharmony_ci
26662306a36Sopenharmony_ciMODULE_AUTHOR("Mårten Lindahl <martenli@axis.com>");
26762306a36Sopenharmony_ciMODULE_DESCRIPTION("Texas Instruments ADC084S021");
26862306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
26962306a36Sopenharmony_ciMODULE_VERSION("1.0");
270