162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Maxim Integrated DS1803 and similar digital potentiometer driver
462306a36Sopenharmony_ci * Copyright (c) 2016 Slawomir Stepien
562306a36Sopenharmony_ci * Copyright (c) 2022 Jagath Jog J
662306a36Sopenharmony_ci *
762306a36Sopenharmony_ci * Datasheet: https://datasheets.maximintegrated.com/en/ds/DS1803.pdf
862306a36Sopenharmony_ci * Datasheet: https://datasheets.maximintegrated.com/en/ds/DS3502.pdf
962306a36Sopenharmony_ci *
1062306a36Sopenharmony_ci * DEVID	#Wipers	#Positions	Resistor Opts (kOhm)	i2c address
1162306a36Sopenharmony_ci * ds1803	2	256		10, 50, 100		0101xxx
1262306a36Sopenharmony_ci * ds3502	1	128		10			01010xx
1362306a36Sopenharmony_ci */
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci#include <linux/err.h>
1662306a36Sopenharmony_ci#include <linux/export.h>
1762306a36Sopenharmony_ci#include <linux/i2c.h>
1862306a36Sopenharmony_ci#include <linux/iio/iio.h>
1962306a36Sopenharmony_ci#include <linux/module.h>
2062306a36Sopenharmony_ci#include <linux/mod_devicetable.h>
2162306a36Sopenharmony_ci#include <linux/property.h>
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci#define DS1803_WIPER_0         0xA9
2462306a36Sopenharmony_ci#define DS1803_WIPER_1         0xAA
2562306a36Sopenharmony_ci#define DS3502_WR_IVR          0x00
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_cienum ds1803_type {
2862306a36Sopenharmony_ci	DS1803_010,
2962306a36Sopenharmony_ci	DS1803_050,
3062306a36Sopenharmony_ci	DS1803_100,
3162306a36Sopenharmony_ci	DS3502,
3262306a36Sopenharmony_ci};
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_cistruct ds1803_cfg {
3562306a36Sopenharmony_ci	int wipers;
3662306a36Sopenharmony_ci	int avail[3];
3762306a36Sopenharmony_ci	int kohms;
3862306a36Sopenharmony_ci	const struct iio_chan_spec *channels;
3962306a36Sopenharmony_ci	u8 num_channels;
4062306a36Sopenharmony_ci	int (*read)(struct iio_dev *indio_dev,
4162306a36Sopenharmony_ci		    struct iio_chan_spec const *chan, int *val);
4262306a36Sopenharmony_ci};
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_cistruct ds1803_data {
4562306a36Sopenharmony_ci	struct i2c_client *client;
4662306a36Sopenharmony_ci	const struct ds1803_cfg *cfg;
4762306a36Sopenharmony_ci};
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_ci#define DS1803_CHANNEL(ch, addr) {					\
5062306a36Sopenharmony_ci	.type = IIO_RESISTANCE,						\
5162306a36Sopenharmony_ci	.indexed = 1,							\
5262306a36Sopenharmony_ci	.output = 1,							\
5362306a36Sopenharmony_ci	.channel = (ch),						\
5462306a36Sopenharmony_ci	.address = (addr),						\
5562306a36Sopenharmony_ci	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),			\
5662306a36Sopenharmony_ci	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),		\
5762306a36Sopenharmony_ci	.info_mask_shared_by_type_available = BIT(IIO_CHAN_INFO_RAW),   \
5862306a36Sopenharmony_ci}
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_cistatic const struct iio_chan_spec ds1803_channels[] = {
6162306a36Sopenharmony_ci	DS1803_CHANNEL(0, DS1803_WIPER_0),
6262306a36Sopenharmony_ci	DS1803_CHANNEL(1, DS1803_WIPER_1),
6362306a36Sopenharmony_ci};
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_cistatic const struct iio_chan_spec ds3502_channels[] = {
6662306a36Sopenharmony_ci	DS1803_CHANNEL(0, DS3502_WR_IVR),
6762306a36Sopenharmony_ci};
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_cistatic int ds1803_read(struct iio_dev *indio_dev,
7062306a36Sopenharmony_ci		       struct iio_chan_spec const *chan,
7162306a36Sopenharmony_ci		       int *val)
7262306a36Sopenharmony_ci{
7362306a36Sopenharmony_ci	struct ds1803_data *data = iio_priv(indio_dev);
7462306a36Sopenharmony_ci	int ret;
7562306a36Sopenharmony_ci	u8 result[ARRAY_SIZE(ds1803_channels)];
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_ci	ret = i2c_master_recv(data->client, result, indio_dev->num_channels);
7862306a36Sopenharmony_ci	if (ret < 0)
7962306a36Sopenharmony_ci		return ret;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	*val = result[chan->channel];
8262306a36Sopenharmony_ci	return ret;
8362306a36Sopenharmony_ci}
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_cistatic int ds3502_read(struct iio_dev *indio_dev,
8662306a36Sopenharmony_ci		       struct iio_chan_spec const *chan,
8762306a36Sopenharmony_ci		       int *val)
8862306a36Sopenharmony_ci{
8962306a36Sopenharmony_ci	struct ds1803_data *data = iio_priv(indio_dev);
9062306a36Sopenharmony_ci	int ret;
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_ci	ret = i2c_smbus_read_byte_data(data->client, chan->address);
9362306a36Sopenharmony_ci	if (ret < 0)
9462306a36Sopenharmony_ci		return ret;
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	*val = ret;
9762306a36Sopenharmony_ci	return ret;
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_cistatic const struct ds1803_cfg ds1803_cfg[] = {
10162306a36Sopenharmony_ci	[DS1803_010] = {
10262306a36Sopenharmony_ci		.wipers = 2,
10362306a36Sopenharmony_ci		.avail = { 0, 1, 255 },
10462306a36Sopenharmony_ci		.kohms =  10,
10562306a36Sopenharmony_ci		.channels = ds1803_channels,
10662306a36Sopenharmony_ci		.num_channels = ARRAY_SIZE(ds1803_channels),
10762306a36Sopenharmony_ci		.read = ds1803_read,
10862306a36Sopenharmony_ci	},
10962306a36Sopenharmony_ci	[DS1803_050] = {
11062306a36Sopenharmony_ci		.wipers = 2,
11162306a36Sopenharmony_ci		.avail = { 0, 1, 255 },
11262306a36Sopenharmony_ci		.kohms =  50,
11362306a36Sopenharmony_ci		.channels = ds1803_channels,
11462306a36Sopenharmony_ci		.num_channels = ARRAY_SIZE(ds1803_channels),
11562306a36Sopenharmony_ci		.read = ds1803_read,
11662306a36Sopenharmony_ci	},
11762306a36Sopenharmony_ci	[DS1803_100] = {
11862306a36Sopenharmony_ci		.wipers = 2,
11962306a36Sopenharmony_ci		.avail = { 0, 1, 255 },
12062306a36Sopenharmony_ci		.kohms = 100,
12162306a36Sopenharmony_ci		.channels = ds1803_channels,
12262306a36Sopenharmony_ci		.num_channels = ARRAY_SIZE(ds1803_channels),
12362306a36Sopenharmony_ci		.read = ds1803_read,
12462306a36Sopenharmony_ci	},
12562306a36Sopenharmony_ci	[DS3502] = {
12662306a36Sopenharmony_ci	  .wipers = 1,
12762306a36Sopenharmony_ci	  .avail = { 0, 1, 127 },
12862306a36Sopenharmony_ci	  .kohms =  10,
12962306a36Sopenharmony_ci	  .channels = ds3502_channels,
13062306a36Sopenharmony_ci	  .num_channels = ARRAY_SIZE(ds3502_channels),
13162306a36Sopenharmony_ci	  .read = ds3502_read,
13262306a36Sopenharmony_ci	},
13362306a36Sopenharmony_ci};
13462306a36Sopenharmony_ci
13562306a36Sopenharmony_cistatic int ds1803_read_raw(struct iio_dev *indio_dev,
13662306a36Sopenharmony_ci			   struct iio_chan_spec const *chan,
13762306a36Sopenharmony_ci			   int *val, int *val2, long mask)
13862306a36Sopenharmony_ci{
13962306a36Sopenharmony_ci	struct ds1803_data *data = iio_priv(indio_dev);
14062306a36Sopenharmony_ci	int ret;
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_ci	switch (mask) {
14362306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
14462306a36Sopenharmony_ci		ret = data->cfg->read(indio_dev, chan, val);
14562306a36Sopenharmony_ci		if (ret < 0)
14662306a36Sopenharmony_ci			return ret;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci		return IIO_VAL_INT;
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ci	case IIO_CHAN_INFO_SCALE:
15162306a36Sopenharmony_ci		*val = 1000 * data->cfg->kohms;
15262306a36Sopenharmony_ci		*val2 = data->cfg->avail[2]; /* Max wiper position */
15362306a36Sopenharmony_ci		return IIO_VAL_FRACTIONAL;
15462306a36Sopenharmony_ci	}
15562306a36Sopenharmony_ci
15662306a36Sopenharmony_ci	return -EINVAL;
15762306a36Sopenharmony_ci}
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_cistatic int ds1803_write_raw(struct iio_dev *indio_dev,
16062306a36Sopenharmony_ci			    struct iio_chan_spec const *chan,
16162306a36Sopenharmony_ci			    int val, int val2, long mask)
16262306a36Sopenharmony_ci{
16362306a36Sopenharmony_ci	struct ds1803_data *data = iio_priv(indio_dev);
16462306a36Sopenharmony_ci	u8 addr = chan->address;
16562306a36Sopenharmony_ci	int max_pos = data->cfg->avail[2];
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_ci	if (val2 != 0)
16862306a36Sopenharmony_ci		return -EINVAL;
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	switch (mask) {
17162306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
17262306a36Sopenharmony_ci		if (val > max_pos || val < 0)
17362306a36Sopenharmony_ci			return -EINVAL;
17462306a36Sopenharmony_ci		break;
17562306a36Sopenharmony_ci	default:
17662306a36Sopenharmony_ci		return -EINVAL;
17762306a36Sopenharmony_ci	}
17862306a36Sopenharmony_ci
17962306a36Sopenharmony_ci	return i2c_smbus_write_byte_data(data->client, addr, val);
18062306a36Sopenharmony_ci}
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_cistatic int ds1803_read_avail(struct iio_dev *indio_dev,
18362306a36Sopenharmony_ci			     struct iio_chan_spec const *chan,
18462306a36Sopenharmony_ci			     const int **vals, int *type,
18562306a36Sopenharmony_ci			     int *length, long mask)
18662306a36Sopenharmony_ci{
18762306a36Sopenharmony_ci	struct ds1803_data *data = iio_priv(indio_dev);
18862306a36Sopenharmony_ci
18962306a36Sopenharmony_ci	switch (mask) {
19062306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
19162306a36Sopenharmony_ci		*vals = data->cfg->avail;
19262306a36Sopenharmony_ci		*length = ARRAY_SIZE(data->cfg->avail);
19362306a36Sopenharmony_ci		*type = IIO_VAL_INT;
19462306a36Sopenharmony_ci		return IIO_AVAIL_RANGE;
19562306a36Sopenharmony_ci	}
19662306a36Sopenharmony_ci	return -EINVAL;
19762306a36Sopenharmony_ci}
19862306a36Sopenharmony_ci
19962306a36Sopenharmony_cistatic const struct iio_info ds1803_info = {
20062306a36Sopenharmony_ci	.read_raw = ds1803_read_raw,
20162306a36Sopenharmony_ci	.write_raw = ds1803_write_raw,
20262306a36Sopenharmony_ci	.read_avail = ds1803_read_avail,
20362306a36Sopenharmony_ci};
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_cistatic int ds1803_probe(struct i2c_client *client)
20662306a36Sopenharmony_ci{
20762306a36Sopenharmony_ci	const struct i2c_device_id *id = i2c_client_get_device_id(client);
20862306a36Sopenharmony_ci	struct device *dev = &client->dev;
20962306a36Sopenharmony_ci	struct ds1803_data *data;
21062306a36Sopenharmony_ci	struct iio_dev *indio_dev;
21162306a36Sopenharmony_ci
21262306a36Sopenharmony_ci	indio_dev = devm_iio_device_alloc(dev, sizeof(*data));
21362306a36Sopenharmony_ci	if (!indio_dev)
21462306a36Sopenharmony_ci		return -ENOMEM;
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_ci	i2c_set_clientdata(client, indio_dev);
21762306a36Sopenharmony_ci
21862306a36Sopenharmony_ci	data = iio_priv(indio_dev);
21962306a36Sopenharmony_ci	data->client = client;
22062306a36Sopenharmony_ci	data->cfg = device_get_match_data(dev);
22162306a36Sopenharmony_ci	if (!data->cfg)
22262306a36Sopenharmony_ci		data->cfg = &ds1803_cfg[id->driver_data];
22362306a36Sopenharmony_ci
22462306a36Sopenharmony_ci	indio_dev->info = &ds1803_info;
22562306a36Sopenharmony_ci	indio_dev->channels = data->cfg->channels;
22662306a36Sopenharmony_ci	indio_dev->num_channels = data->cfg->num_channels;
22762306a36Sopenharmony_ci	indio_dev->name = client->name;
22862306a36Sopenharmony_ci
22962306a36Sopenharmony_ci	return devm_iio_device_register(dev, indio_dev);
23062306a36Sopenharmony_ci}
23162306a36Sopenharmony_ci
23262306a36Sopenharmony_cistatic const struct of_device_id ds1803_dt_ids[] = {
23362306a36Sopenharmony_ci	{ .compatible = "maxim,ds1803-010", .data = &ds1803_cfg[DS1803_010] },
23462306a36Sopenharmony_ci	{ .compatible = "maxim,ds1803-050", .data = &ds1803_cfg[DS1803_050] },
23562306a36Sopenharmony_ci	{ .compatible = "maxim,ds1803-100", .data = &ds1803_cfg[DS1803_100] },
23662306a36Sopenharmony_ci	{ .compatible = "maxim,ds3502", .data = &ds1803_cfg[DS3502] },
23762306a36Sopenharmony_ci	{}
23862306a36Sopenharmony_ci};
23962306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, ds1803_dt_ids);
24062306a36Sopenharmony_ci
24162306a36Sopenharmony_cistatic const struct i2c_device_id ds1803_id[] = {
24262306a36Sopenharmony_ci	{ "ds1803-010", DS1803_010 },
24362306a36Sopenharmony_ci	{ "ds1803-050", DS1803_050 },
24462306a36Sopenharmony_ci	{ "ds1803-100", DS1803_100 },
24562306a36Sopenharmony_ci	{ "ds3502", DS3502 },
24662306a36Sopenharmony_ci	{}
24762306a36Sopenharmony_ci};
24862306a36Sopenharmony_ciMODULE_DEVICE_TABLE(i2c, ds1803_id);
24962306a36Sopenharmony_ci
25062306a36Sopenharmony_cistatic struct i2c_driver ds1803_driver = {
25162306a36Sopenharmony_ci	.driver = {
25262306a36Sopenharmony_ci		.name	= "ds1803",
25362306a36Sopenharmony_ci		.of_match_table = ds1803_dt_ids,
25462306a36Sopenharmony_ci	},
25562306a36Sopenharmony_ci	.probe		= ds1803_probe,
25662306a36Sopenharmony_ci	.id_table	= ds1803_id,
25762306a36Sopenharmony_ci};
25862306a36Sopenharmony_ci
25962306a36Sopenharmony_cimodule_i2c_driver(ds1803_driver);
26062306a36Sopenharmony_ci
26162306a36Sopenharmony_ciMODULE_AUTHOR("Slawomir Stepien <sst@poczta.fm>");
26262306a36Sopenharmony_ciMODULE_AUTHOR("Jagath Jog J <jagathjog1996@gmail.com>");
26362306a36Sopenharmony_ciMODULE_DESCRIPTION("DS1803 digital potentiometer");
26462306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
265