162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0+
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Analog Devices AD5110 digital potentiometer driver
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2021 Mugilraj Dhavachelvan <dmugil2000@gmail.com>
662306a36Sopenharmony_ci *
762306a36Sopenharmony_ci * Datasheet: https://www.analog.com/media/en/technical-documentation/data-sheets/AD5110_5112_5114.pdf
862306a36Sopenharmony_ci */
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include <linux/bitfield.h>
1162306a36Sopenharmony_ci#include <linux/delay.h>
1262306a36Sopenharmony_ci#include <linux/device.h>
1362306a36Sopenharmony_ci#include <linux/i2c.h>
1462306a36Sopenharmony_ci#include <linux/module.h>
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci#include <linux/iio/iio.h>
1762306a36Sopenharmony_ci#include <linux/iio/sysfs.h>
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci/* AD5110 commands */
2062306a36Sopenharmony_ci#define AD5110_EEPROM_WR	1
2162306a36Sopenharmony_ci#define AD5110_RDAC_WR		2
2262306a36Sopenharmony_ci#define AD5110_SHUTDOWN	3
2362306a36Sopenharmony_ci#define AD5110_RESET		4
2462306a36Sopenharmony_ci#define AD5110_RDAC_RD		5
2562306a36Sopenharmony_ci#define AD5110_EEPROM_RD	6
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci/* AD5110_EEPROM_RD data */
2862306a36Sopenharmony_ci#define AD5110_WIPER_POS	0
2962306a36Sopenharmony_ci#define AD5110_RESISTOR_TOL	1
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci#define AD5110_WIPER_RESISTANCE	70
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_cistruct ad5110_cfg {
3462306a36Sopenharmony_ci	int max_pos;
3562306a36Sopenharmony_ci	int kohms;
3662306a36Sopenharmony_ci	int shift;
3762306a36Sopenharmony_ci};
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_cienum ad5110_type {
4062306a36Sopenharmony_ci	AD5110_10,
4162306a36Sopenharmony_ci	AD5110_80,
4262306a36Sopenharmony_ci	AD5112_05,
4362306a36Sopenharmony_ci	AD5112_10,
4462306a36Sopenharmony_ci	AD5112_80,
4562306a36Sopenharmony_ci	AD5114_10,
4662306a36Sopenharmony_ci	AD5114_80,
4762306a36Sopenharmony_ci};
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_cistatic const struct ad5110_cfg ad5110_cfg[] = {
5062306a36Sopenharmony_ci	[AD5110_10] = { .max_pos = 128, .kohms = 10 },
5162306a36Sopenharmony_ci	[AD5110_80] = { .max_pos = 128, .kohms = 80 },
5262306a36Sopenharmony_ci	[AD5112_05] = { .max_pos = 64, .kohms = 5, .shift = 1 },
5362306a36Sopenharmony_ci	[AD5112_10] = { .max_pos = 64, .kohms = 10, .shift = 1 },
5462306a36Sopenharmony_ci	[AD5112_80] = { .max_pos = 64, .kohms = 80, .shift = 1 },
5562306a36Sopenharmony_ci	[AD5114_10] = { .max_pos = 32, .kohms = 10, .shift = 2 },
5662306a36Sopenharmony_ci	[AD5114_80] = { .max_pos = 32, .kohms = 80, .shift = 2 },
5762306a36Sopenharmony_ci};
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_cistruct ad5110_data {
6062306a36Sopenharmony_ci	struct i2c_client       *client;
6162306a36Sopenharmony_ci	s16			tol;		/* resistor tolerance */
6262306a36Sopenharmony_ci	bool			enable;
6362306a36Sopenharmony_ci	struct mutex            lock;
6462306a36Sopenharmony_ci	const struct ad5110_cfg	*cfg;
6562306a36Sopenharmony_ci	/*
6662306a36Sopenharmony_ci	 * DMA (thus cache coherency maintenance) may require the
6762306a36Sopenharmony_ci	 * transfer buffers to live in their own cache lines.
6862306a36Sopenharmony_ci	 */
6962306a36Sopenharmony_ci	u8			buf[2] __aligned(IIO_DMA_MINALIGN);
7062306a36Sopenharmony_ci};
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_cistatic const struct iio_chan_spec ad5110_channels[] = {
7362306a36Sopenharmony_ci	{
7462306a36Sopenharmony_ci		.type = IIO_RESISTANCE,
7562306a36Sopenharmony_ci		.output = 1,
7662306a36Sopenharmony_ci		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) | BIT(IIO_CHAN_INFO_OFFSET) |
7762306a36Sopenharmony_ci					BIT(IIO_CHAN_INFO_SCALE) | BIT(IIO_CHAN_INFO_ENABLE),
7862306a36Sopenharmony_ci	},
7962306a36Sopenharmony_ci};
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_cistatic int ad5110_read(struct ad5110_data *data, u8 cmd, int *val)
8262306a36Sopenharmony_ci{
8362306a36Sopenharmony_ci	int ret;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	mutex_lock(&data->lock);
8662306a36Sopenharmony_ci	data->buf[0] = cmd;
8762306a36Sopenharmony_ci	data->buf[1] = *val;
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci	ret = i2c_master_send_dmasafe(data->client, data->buf, sizeof(data->buf));
9062306a36Sopenharmony_ci	if (ret < 0) {
9162306a36Sopenharmony_ci		goto error;
9262306a36Sopenharmony_ci	} else if (ret != sizeof(data->buf)) {
9362306a36Sopenharmony_ci		ret = -EIO;
9462306a36Sopenharmony_ci		goto error;
9562306a36Sopenharmony_ci	}
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	ret = i2c_master_recv_dmasafe(data->client, data->buf, 1);
9862306a36Sopenharmony_ci	if (ret < 0) {
9962306a36Sopenharmony_ci		goto error;
10062306a36Sopenharmony_ci	} else if (ret != 1) {
10162306a36Sopenharmony_ci		ret = -EIO;
10262306a36Sopenharmony_ci		goto error;
10362306a36Sopenharmony_ci	}
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	*val = data->buf[0];
10662306a36Sopenharmony_ci	ret = 0;
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_cierror:
10962306a36Sopenharmony_ci	mutex_unlock(&data->lock);
11062306a36Sopenharmony_ci	return ret;
11162306a36Sopenharmony_ci}
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_cistatic int ad5110_write(struct ad5110_data *data, u8 cmd, u8 val)
11462306a36Sopenharmony_ci{
11562306a36Sopenharmony_ci	int ret;
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ci	mutex_lock(&data->lock);
11862306a36Sopenharmony_ci	data->buf[0] = cmd;
11962306a36Sopenharmony_ci	data->buf[1] = val;
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ci	ret = i2c_master_send_dmasafe(data->client, data->buf, sizeof(data->buf));
12262306a36Sopenharmony_ci	if (ret < 0) {
12362306a36Sopenharmony_ci		goto error;
12462306a36Sopenharmony_ci	} else if (ret != sizeof(data->buf)) {
12562306a36Sopenharmony_ci		ret = -EIO;
12662306a36Sopenharmony_ci		goto error;
12762306a36Sopenharmony_ci	}
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_ci	ret = 0;
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_cierror:
13262306a36Sopenharmony_ci	mutex_unlock(&data->lock);
13362306a36Sopenharmony_ci	return ret;
13462306a36Sopenharmony_ci}
13562306a36Sopenharmony_ci
13662306a36Sopenharmony_cistatic int ad5110_resistor_tol(struct ad5110_data *data, u8 cmd, int val)
13762306a36Sopenharmony_ci{
13862306a36Sopenharmony_ci	int ret;
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci	ret = ad5110_read(data, cmd, &val);
14162306a36Sopenharmony_ci	if (ret)
14262306a36Sopenharmony_ci		return ret;
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_ci	data->tol = data->cfg->kohms * (val & GENMASK(6, 0)) * 10 / 8;
14562306a36Sopenharmony_ci	if (!(val & BIT(7)))
14662306a36Sopenharmony_ci		data->tol *= -1;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	return 0;
14962306a36Sopenharmony_ci}
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_cistatic ssize_t store_eeprom_show(struct device *dev,
15262306a36Sopenharmony_ci				  struct device_attribute *attr,
15362306a36Sopenharmony_ci				  char *buf)
15462306a36Sopenharmony_ci{
15562306a36Sopenharmony_ci	struct iio_dev *indio_dev = dev_to_iio_dev(dev);
15662306a36Sopenharmony_ci	struct ad5110_data *data = iio_priv(indio_dev);
15762306a36Sopenharmony_ci	int val = AD5110_WIPER_POS;
15862306a36Sopenharmony_ci	int ret;
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ci	ret = ad5110_read(data, AD5110_EEPROM_RD, &val);
16162306a36Sopenharmony_ci	if (ret)
16262306a36Sopenharmony_ci		return ret;
16362306a36Sopenharmony_ci
16462306a36Sopenharmony_ci	val = val >> data->cfg->shift;
16562306a36Sopenharmony_ci	return iio_format_value(buf, IIO_VAL_INT, 1, &val);
16662306a36Sopenharmony_ci}
16762306a36Sopenharmony_ci
16862306a36Sopenharmony_cistatic ssize_t store_eeprom_store(struct device *dev,
16962306a36Sopenharmony_ci				   struct device_attribute *attr,
17062306a36Sopenharmony_ci				   const char *buf, size_t len)
17162306a36Sopenharmony_ci{
17262306a36Sopenharmony_ci	struct iio_dev *indio_dev = dev_to_iio_dev(dev);
17362306a36Sopenharmony_ci	struct ad5110_data *data = iio_priv(indio_dev);
17462306a36Sopenharmony_ci	int ret;
17562306a36Sopenharmony_ci
17662306a36Sopenharmony_ci	ret = ad5110_write(data, AD5110_EEPROM_WR, 0);
17762306a36Sopenharmony_ci	if (ret) {
17862306a36Sopenharmony_ci		dev_err(&data->client->dev, "RDAC to EEPROM write failed\n");
17962306a36Sopenharmony_ci		return ret;
18062306a36Sopenharmony_ci	}
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_ci	/* The storing of EEPROM data takes approximately 18 ms. */
18362306a36Sopenharmony_ci	msleep(20);
18462306a36Sopenharmony_ci
18562306a36Sopenharmony_ci	return len;
18662306a36Sopenharmony_ci}
18762306a36Sopenharmony_ci
18862306a36Sopenharmony_cistatic IIO_DEVICE_ATTR_RW(store_eeprom, 0);
18962306a36Sopenharmony_ci
19062306a36Sopenharmony_cistatic struct attribute *ad5110_attributes[] = {
19162306a36Sopenharmony_ci	&iio_dev_attr_store_eeprom.dev_attr.attr,
19262306a36Sopenharmony_ci	NULL
19362306a36Sopenharmony_ci};
19462306a36Sopenharmony_ci
19562306a36Sopenharmony_cistatic const struct attribute_group ad5110_attribute_group = {
19662306a36Sopenharmony_ci	.attrs = ad5110_attributes,
19762306a36Sopenharmony_ci};
19862306a36Sopenharmony_ci
19962306a36Sopenharmony_cistatic int ad5110_read_raw(struct iio_dev *indio_dev,
20062306a36Sopenharmony_ci			   struct iio_chan_spec const *chan,
20162306a36Sopenharmony_ci			   int *val, int *val2, long mask)
20262306a36Sopenharmony_ci{
20362306a36Sopenharmony_ci	struct ad5110_data *data = iio_priv(indio_dev);
20462306a36Sopenharmony_ci	int ret;
20562306a36Sopenharmony_ci
20662306a36Sopenharmony_ci	switch (mask) {
20762306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
20862306a36Sopenharmony_ci		ret = ad5110_read(data, AD5110_RDAC_RD, val);
20962306a36Sopenharmony_ci		if (ret)
21062306a36Sopenharmony_ci			return ret;
21162306a36Sopenharmony_ci
21262306a36Sopenharmony_ci		*val = *val >> data->cfg->shift;
21362306a36Sopenharmony_ci		return IIO_VAL_INT;
21462306a36Sopenharmony_ci	case IIO_CHAN_INFO_OFFSET:
21562306a36Sopenharmony_ci		*val = AD5110_WIPER_RESISTANCE * data->cfg->max_pos;
21662306a36Sopenharmony_ci		*val2 = 1000 * data->cfg->kohms + data->tol;
21762306a36Sopenharmony_ci		return IIO_VAL_FRACTIONAL;
21862306a36Sopenharmony_ci	case IIO_CHAN_INFO_SCALE:
21962306a36Sopenharmony_ci		*val = 1000 * data->cfg->kohms + data->tol;
22062306a36Sopenharmony_ci		*val2 = data->cfg->max_pos;
22162306a36Sopenharmony_ci		return IIO_VAL_FRACTIONAL;
22262306a36Sopenharmony_ci	case IIO_CHAN_INFO_ENABLE:
22362306a36Sopenharmony_ci		*val = data->enable;
22462306a36Sopenharmony_ci		return IIO_VAL_INT;
22562306a36Sopenharmony_ci	default:
22662306a36Sopenharmony_ci		return -EINVAL;
22762306a36Sopenharmony_ci	}
22862306a36Sopenharmony_ci}
22962306a36Sopenharmony_ci
23062306a36Sopenharmony_cistatic int ad5110_write_raw(struct iio_dev *indio_dev,
23162306a36Sopenharmony_ci			    struct iio_chan_spec const *chan,
23262306a36Sopenharmony_ci			    int val, int val2, long mask)
23362306a36Sopenharmony_ci{
23462306a36Sopenharmony_ci	struct ad5110_data *data = iio_priv(indio_dev);
23562306a36Sopenharmony_ci	int ret;
23662306a36Sopenharmony_ci
23762306a36Sopenharmony_ci	switch (mask) {
23862306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
23962306a36Sopenharmony_ci		if (val > data->cfg->max_pos || val < 0)
24062306a36Sopenharmony_ci			return -EINVAL;
24162306a36Sopenharmony_ci
24262306a36Sopenharmony_ci		return ad5110_write(data, AD5110_RDAC_WR, val << data->cfg->shift);
24362306a36Sopenharmony_ci	case IIO_CHAN_INFO_ENABLE:
24462306a36Sopenharmony_ci		if (val < 0 || val > 1)
24562306a36Sopenharmony_ci			return -EINVAL;
24662306a36Sopenharmony_ci		if (data->enable == val)
24762306a36Sopenharmony_ci			return 0;
24862306a36Sopenharmony_ci		ret = ad5110_write(data, AD5110_SHUTDOWN, val ? 0 : 1);
24962306a36Sopenharmony_ci		if (ret)
25062306a36Sopenharmony_ci			return ret;
25162306a36Sopenharmony_ci		data->enable = val;
25262306a36Sopenharmony_ci		return 0;
25362306a36Sopenharmony_ci	default:
25462306a36Sopenharmony_ci		return -EINVAL;
25562306a36Sopenharmony_ci	}
25662306a36Sopenharmony_ci}
25762306a36Sopenharmony_ci
25862306a36Sopenharmony_cistatic const struct iio_info ad5110_info = {
25962306a36Sopenharmony_ci	.read_raw = ad5110_read_raw,
26062306a36Sopenharmony_ci	.write_raw = ad5110_write_raw,
26162306a36Sopenharmony_ci	.attrs = &ad5110_attribute_group,
26262306a36Sopenharmony_ci};
26362306a36Sopenharmony_ci
26462306a36Sopenharmony_ci#define AD5110_COMPATIBLE(of_compatible, cfg) {	\
26562306a36Sopenharmony_ci			.compatible = of_compatible,	\
26662306a36Sopenharmony_ci			.data = &ad5110_cfg[cfg],	\
26762306a36Sopenharmony_ci}
26862306a36Sopenharmony_ci
26962306a36Sopenharmony_cistatic const struct of_device_id ad5110_of_match[] = {
27062306a36Sopenharmony_ci	AD5110_COMPATIBLE("adi,ad5110-10", AD5110_10),
27162306a36Sopenharmony_ci	AD5110_COMPATIBLE("adi,ad5110-80", AD5110_80),
27262306a36Sopenharmony_ci	AD5110_COMPATIBLE("adi,ad5112-05", AD5112_05),
27362306a36Sopenharmony_ci	AD5110_COMPATIBLE("adi,ad5112-10", AD5112_10),
27462306a36Sopenharmony_ci	AD5110_COMPATIBLE("adi,ad5112-80", AD5112_80),
27562306a36Sopenharmony_ci	AD5110_COMPATIBLE("adi,ad5114-10", AD5114_10),
27662306a36Sopenharmony_ci	AD5110_COMPATIBLE("adi,ad5114-80", AD5114_80),
27762306a36Sopenharmony_ci	{ }
27862306a36Sopenharmony_ci};
27962306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, ad5110_of_match);
28062306a36Sopenharmony_ci
28162306a36Sopenharmony_cistatic const struct i2c_device_id ad5110_id[] = {
28262306a36Sopenharmony_ci	{ "ad5110-10", AD5110_10 },
28362306a36Sopenharmony_ci	{ "ad5110-80", AD5110_80 },
28462306a36Sopenharmony_ci	{ "ad5112-05", AD5112_05 },
28562306a36Sopenharmony_ci	{ "ad5112-10", AD5112_10 },
28662306a36Sopenharmony_ci	{ "ad5112-80", AD5112_80 },
28762306a36Sopenharmony_ci	{ "ad5114-10", AD5114_10 },
28862306a36Sopenharmony_ci	{ "ad5114-80", AD5114_80 },
28962306a36Sopenharmony_ci	{ }
29062306a36Sopenharmony_ci};
29162306a36Sopenharmony_ciMODULE_DEVICE_TABLE(i2c, ad5110_id);
29262306a36Sopenharmony_ci
29362306a36Sopenharmony_cistatic int ad5110_probe(struct i2c_client *client)
29462306a36Sopenharmony_ci{
29562306a36Sopenharmony_ci	struct device *dev = &client->dev;
29662306a36Sopenharmony_ci	struct iio_dev *indio_dev;
29762306a36Sopenharmony_ci	struct ad5110_data *data;
29862306a36Sopenharmony_ci	int ret;
29962306a36Sopenharmony_ci
30062306a36Sopenharmony_ci	indio_dev = devm_iio_device_alloc(dev, sizeof(*data));
30162306a36Sopenharmony_ci	if (!indio_dev)
30262306a36Sopenharmony_ci		return -ENOMEM;
30362306a36Sopenharmony_ci
30462306a36Sopenharmony_ci	data = iio_priv(indio_dev);
30562306a36Sopenharmony_ci	data->client = client;
30662306a36Sopenharmony_ci	mutex_init(&data->lock);
30762306a36Sopenharmony_ci	data->enable = 1;
30862306a36Sopenharmony_ci	data->cfg = device_get_match_data(dev);
30962306a36Sopenharmony_ci
31062306a36Sopenharmony_ci	/* refresh RDAC register with EEPROM */
31162306a36Sopenharmony_ci	ret = ad5110_write(data, AD5110_RESET, 0);
31262306a36Sopenharmony_ci	if (ret) {
31362306a36Sopenharmony_ci		dev_err(dev, "Refresh RDAC with EEPROM failed\n");
31462306a36Sopenharmony_ci		return ret;
31562306a36Sopenharmony_ci	}
31662306a36Sopenharmony_ci
31762306a36Sopenharmony_ci	ret = ad5110_resistor_tol(data, AD5110_EEPROM_RD, AD5110_RESISTOR_TOL);
31862306a36Sopenharmony_ci	if (ret) {
31962306a36Sopenharmony_ci		dev_err(dev, "Read resistor tolerance failed\n");
32062306a36Sopenharmony_ci		return ret;
32162306a36Sopenharmony_ci	}
32262306a36Sopenharmony_ci
32362306a36Sopenharmony_ci	indio_dev->modes = INDIO_DIRECT_MODE;
32462306a36Sopenharmony_ci	indio_dev->info = &ad5110_info;
32562306a36Sopenharmony_ci	indio_dev->channels = ad5110_channels;
32662306a36Sopenharmony_ci	indio_dev->num_channels = ARRAY_SIZE(ad5110_channels);
32762306a36Sopenharmony_ci	indio_dev->name = client->name;
32862306a36Sopenharmony_ci
32962306a36Sopenharmony_ci	return devm_iio_device_register(dev, indio_dev);
33062306a36Sopenharmony_ci}
33162306a36Sopenharmony_ci
33262306a36Sopenharmony_cistatic struct i2c_driver ad5110_driver = {
33362306a36Sopenharmony_ci	.driver = {
33462306a36Sopenharmony_ci		.name	= "ad5110",
33562306a36Sopenharmony_ci		.of_match_table = ad5110_of_match,
33662306a36Sopenharmony_ci	},
33762306a36Sopenharmony_ci	.probe		= ad5110_probe,
33862306a36Sopenharmony_ci	.id_table	= ad5110_id,
33962306a36Sopenharmony_ci};
34062306a36Sopenharmony_cimodule_i2c_driver(ad5110_driver);
34162306a36Sopenharmony_ci
34262306a36Sopenharmony_ciMODULE_AUTHOR("Mugilraj Dhavachelvan <dmugil2000@gmail.com>");
34362306a36Sopenharmony_ciMODULE_DESCRIPTION("AD5110 digital potentiometer");
34462306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
345