162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * IIO driver for the Apex Embedded Systems STX104
462306a36Sopenharmony_ci * Copyright (C) 2016 William Breathitt Gray
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci#include <linux/bitfield.h>
762306a36Sopenharmony_ci#include <linux/bits.h>
862306a36Sopenharmony_ci#include <linux/device.h>
962306a36Sopenharmony_ci#include <linux/err.h>
1062306a36Sopenharmony_ci#include <linux/gpio/regmap.h>
1162306a36Sopenharmony_ci#include <linux/iio/iio.h>
1262306a36Sopenharmony_ci#include <linux/iio/types.h>
1362306a36Sopenharmony_ci#include <linux/isa.h>
1462306a36Sopenharmony_ci#include <linux/kernel.h>
1562306a36Sopenharmony_ci#include <linux/limits.h>
1662306a36Sopenharmony_ci#include <linux/module.h>
1762306a36Sopenharmony_ci#include <linux/moduleparam.h>
1862306a36Sopenharmony_ci#include <linux/mutex.h>
1962306a36Sopenharmony_ci#include <linux/regmap.h>
2062306a36Sopenharmony_ci#include <linux/types.h>
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_ci#define STX104_OUT_CHAN(chan) {				\
2362306a36Sopenharmony_ci	.type = IIO_VOLTAGE,				\
2462306a36Sopenharmony_ci	.channel = chan,				\
2562306a36Sopenharmony_ci	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),	\
2662306a36Sopenharmony_ci	.indexed = 1,					\
2762306a36Sopenharmony_ci	.output = 1					\
2862306a36Sopenharmony_ci}
2962306a36Sopenharmony_ci#define STX104_IN_CHAN(chan, diff) {					\
3062306a36Sopenharmony_ci	.type = IIO_VOLTAGE,						\
3162306a36Sopenharmony_ci	.channel = chan,						\
3262306a36Sopenharmony_ci	.channel2 = chan,						\
3362306a36Sopenharmony_ci	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_HARDWAREGAIN) |	\
3462306a36Sopenharmony_ci		BIT(IIO_CHAN_INFO_OFFSET) | BIT(IIO_CHAN_INFO_SCALE),	\
3562306a36Sopenharmony_ci	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),			\
3662306a36Sopenharmony_ci	.indexed = 1,							\
3762306a36Sopenharmony_ci	.differential = diff						\
3862306a36Sopenharmony_ci}
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci#define STX104_NUM_OUT_CHAN 2
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_ci#define STX104_EXTENT 16
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_cistatic unsigned int base[max_num_isa_dev(STX104_EXTENT)];
4562306a36Sopenharmony_cistatic unsigned int num_stx104;
4662306a36Sopenharmony_cimodule_param_hw_array(base, uint, ioport, &num_stx104, 0);
4762306a36Sopenharmony_ciMODULE_PARM_DESC(base, "Apex Embedded Systems STX104 base addresses");
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_ci#define STX104_AIO_BASE 0x0
5062306a36Sopenharmony_ci#define STX104_SOFTWARE_STROBE STX104_AIO_BASE
5162306a36Sopenharmony_ci#define STX104_ADC_DATA STX104_AIO_BASE
5262306a36Sopenharmony_ci#define STX104_ADC_CHANNEL (STX104_AIO_BASE + 0x2)
5362306a36Sopenharmony_ci#define STX104_DIO_REG (STX104_AIO_BASE + 0x3)
5462306a36Sopenharmony_ci#define STX104_DAC_BASE (STX104_AIO_BASE + 0x4)
5562306a36Sopenharmony_ci#define STX104_ADC_STATUS (STX104_AIO_BASE + 0x8)
5662306a36Sopenharmony_ci#define STX104_ADC_CONTROL (STX104_AIO_BASE + 0x9)
5762306a36Sopenharmony_ci#define STX104_ADC_CONFIGURATION (STX104_AIO_BASE + 0x11)
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci#define STX104_AIO_DATA_STRIDE 2
6062306a36Sopenharmony_ci#define STX104_DAC_OFFSET(_channel) (STX104_DAC_BASE + STX104_AIO_DATA_STRIDE * (_channel))
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_ci/* ADC Channel */
6362306a36Sopenharmony_ci#define STX104_FC GENMASK(3, 0)
6462306a36Sopenharmony_ci#define STX104_LC GENMASK(7, 4)
6562306a36Sopenharmony_ci#define STX104_SINGLE_CHANNEL(_channel) \
6662306a36Sopenharmony_ci	(u8_encode_bits(_channel, STX104_FC) | u8_encode_bits(_channel, STX104_LC))
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_ci/* ADC Status */
6962306a36Sopenharmony_ci#define STX104_SD BIT(5)
7062306a36Sopenharmony_ci#define STX104_CNV BIT(7)
7162306a36Sopenharmony_ci#define STX104_DIFFERENTIAL 1
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci/* ADC Control */
7462306a36Sopenharmony_ci#define STX104_ALSS GENMASK(1, 0)
7562306a36Sopenharmony_ci#define STX104_SOFTWARE_TRIGGER u8_encode_bits(0x0, STX104_ALSS)
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_ci/* ADC Configuration */
7862306a36Sopenharmony_ci#define STX104_GAIN GENMASK(1, 0)
7962306a36Sopenharmony_ci#define STX104_ADBU BIT(2)
8062306a36Sopenharmony_ci#define STX104_BIPOLAR 0
8162306a36Sopenharmony_ci#define STX104_GAIN_X1 0
8262306a36Sopenharmony_ci#define STX104_GAIN_X2 1
8362306a36Sopenharmony_ci#define STX104_GAIN_X4 2
8462306a36Sopenharmony_ci#define STX104_GAIN_X8 3
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci/**
8762306a36Sopenharmony_ci * struct stx104_iio - IIO device private data structure
8862306a36Sopenharmony_ci * @lock: synchronization lock to prevent I/O race conditions
8962306a36Sopenharmony_ci * @aio_data_map: Regmap for analog I/O data
9062306a36Sopenharmony_ci * @aio_ctl_map: Regmap for analog I/O control
9162306a36Sopenharmony_ci */
9262306a36Sopenharmony_cistruct stx104_iio {
9362306a36Sopenharmony_ci	struct mutex lock;
9462306a36Sopenharmony_ci	struct regmap *aio_data_map;
9562306a36Sopenharmony_ci	struct regmap *aio_ctl_map;
9662306a36Sopenharmony_ci};
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_cistatic const struct regmap_range aio_ctl_wr_ranges[] = {
9962306a36Sopenharmony_ci	regmap_reg_range(0x0, 0x0), regmap_reg_range(0x2, 0x2), regmap_reg_range(0x9, 0x9),
10062306a36Sopenharmony_ci	regmap_reg_range(0x11, 0x11),
10162306a36Sopenharmony_ci};
10262306a36Sopenharmony_cistatic const struct regmap_range aio_ctl_rd_ranges[] = {
10362306a36Sopenharmony_ci	regmap_reg_range(0x2, 0x2), regmap_reg_range(0x8, 0x9), regmap_reg_range(0x11, 0x11),
10462306a36Sopenharmony_ci};
10562306a36Sopenharmony_cistatic const struct regmap_range aio_ctl_volatile_ranges[] = {
10662306a36Sopenharmony_ci	regmap_reg_range(0x8, 0x8),
10762306a36Sopenharmony_ci};
10862306a36Sopenharmony_cistatic const struct regmap_access_table aio_ctl_wr_table = {
10962306a36Sopenharmony_ci	.yes_ranges = aio_ctl_wr_ranges,
11062306a36Sopenharmony_ci	.n_yes_ranges = ARRAY_SIZE(aio_ctl_wr_ranges),
11162306a36Sopenharmony_ci};
11262306a36Sopenharmony_cistatic const struct regmap_access_table aio_ctl_rd_table = {
11362306a36Sopenharmony_ci	.yes_ranges = aio_ctl_rd_ranges,
11462306a36Sopenharmony_ci	.n_yes_ranges = ARRAY_SIZE(aio_ctl_rd_ranges),
11562306a36Sopenharmony_ci};
11662306a36Sopenharmony_cistatic const struct regmap_access_table aio_ctl_volatile_table = {
11762306a36Sopenharmony_ci	.yes_ranges = aio_ctl_volatile_ranges,
11862306a36Sopenharmony_ci	.n_yes_ranges = ARRAY_SIZE(aio_ctl_volatile_ranges),
11962306a36Sopenharmony_ci};
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_cistatic const struct regmap_config aio_ctl_regmap_config = {
12262306a36Sopenharmony_ci	.name = "aio_ctl",
12362306a36Sopenharmony_ci	.reg_bits = 8,
12462306a36Sopenharmony_ci	.reg_stride = 1,
12562306a36Sopenharmony_ci	.reg_base = STX104_AIO_BASE,
12662306a36Sopenharmony_ci	.val_bits = 8,
12762306a36Sopenharmony_ci	.io_port = true,
12862306a36Sopenharmony_ci	.wr_table = &aio_ctl_wr_table,
12962306a36Sopenharmony_ci	.rd_table = &aio_ctl_rd_table,
13062306a36Sopenharmony_ci	.volatile_table = &aio_ctl_volatile_table,
13162306a36Sopenharmony_ci	.cache_type = REGCACHE_FLAT,
13262306a36Sopenharmony_ci};
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_cistatic const struct regmap_range aio_data_wr_ranges[] = {
13562306a36Sopenharmony_ci	regmap_reg_range(0x4, 0x6),
13662306a36Sopenharmony_ci};
13762306a36Sopenharmony_cistatic const struct regmap_range aio_data_rd_ranges[] = {
13862306a36Sopenharmony_ci	regmap_reg_range(0x0, 0x0),
13962306a36Sopenharmony_ci};
14062306a36Sopenharmony_cistatic const struct regmap_access_table aio_data_wr_table = {
14162306a36Sopenharmony_ci	.yes_ranges = aio_data_wr_ranges,
14262306a36Sopenharmony_ci	.n_yes_ranges = ARRAY_SIZE(aio_data_wr_ranges),
14362306a36Sopenharmony_ci};
14462306a36Sopenharmony_cistatic const struct regmap_access_table aio_data_rd_table = {
14562306a36Sopenharmony_ci	.yes_ranges = aio_data_rd_ranges,
14662306a36Sopenharmony_ci	.n_yes_ranges = ARRAY_SIZE(aio_data_rd_ranges),
14762306a36Sopenharmony_ci};
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_cistatic const struct regmap_config aio_data_regmap_config = {
15062306a36Sopenharmony_ci	.name = "aio_data",
15162306a36Sopenharmony_ci	.reg_bits = 16,
15262306a36Sopenharmony_ci	.reg_stride = STX104_AIO_DATA_STRIDE,
15362306a36Sopenharmony_ci	.reg_base = STX104_AIO_BASE,
15462306a36Sopenharmony_ci	.val_bits = 16,
15562306a36Sopenharmony_ci	.io_port = true,
15662306a36Sopenharmony_ci	.wr_table = &aio_data_wr_table,
15762306a36Sopenharmony_ci	.rd_table = &aio_data_rd_table,
15862306a36Sopenharmony_ci	.volatile_table = &aio_data_rd_table,
15962306a36Sopenharmony_ci	.cache_type = REGCACHE_FLAT,
16062306a36Sopenharmony_ci};
16162306a36Sopenharmony_ci
16262306a36Sopenharmony_cistatic const struct regmap_config dio_regmap_config = {
16362306a36Sopenharmony_ci	.name = "dio",
16462306a36Sopenharmony_ci	.reg_bits = 8,
16562306a36Sopenharmony_ci	.reg_stride = 1,
16662306a36Sopenharmony_ci	.reg_base = STX104_DIO_REG,
16762306a36Sopenharmony_ci	.val_bits = 8,
16862306a36Sopenharmony_ci	.io_port = true,
16962306a36Sopenharmony_ci};
17062306a36Sopenharmony_ci
17162306a36Sopenharmony_cistatic int stx104_read_raw(struct iio_dev *indio_dev,
17262306a36Sopenharmony_ci	struct iio_chan_spec const *chan, int *val, int *val2, long mask)
17362306a36Sopenharmony_ci{
17462306a36Sopenharmony_ci	struct stx104_iio *const priv = iio_priv(indio_dev);
17562306a36Sopenharmony_ci	int err;
17662306a36Sopenharmony_ci	unsigned int adc_config;
17762306a36Sopenharmony_ci	unsigned int value;
17862306a36Sopenharmony_ci	unsigned int adc_status;
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_ci	switch (mask) {
18162306a36Sopenharmony_ci	case IIO_CHAN_INFO_HARDWAREGAIN:
18262306a36Sopenharmony_ci		err = regmap_read(priv->aio_ctl_map, STX104_ADC_CONFIGURATION, &adc_config);
18362306a36Sopenharmony_ci		if (err)
18462306a36Sopenharmony_ci			return err;
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_ci		*val = BIT(u8_get_bits(adc_config, STX104_GAIN));
18762306a36Sopenharmony_ci		return IIO_VAL_INT;
18862306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
18962306a36Sopenharmony_ci		if (chan->output) {
19062306a36Sopenharmony_ci			err = regmap_read(priv->aio_data_map, STX104_DAC_OFFSET(chan->channel),
19162306a36Sopenharmony_ci					  &value);
19262306a36Sopenharmony_ci			if (err)
19362306a36Sopenharmony_ci				return err;
19462306a36Sopenharmony_ci			*val = value;
19562306a36Sopenharmony_ci			return IIO_VAL_INT;
19662306a36Sopenharmony_ci		}
19762306a36Sopenharmony_ci
19862306a36Sopenharmony_ci		mutex_lock(&priv->lock);
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_ci		/* select ADC channel */
20162306a36Sopenharmony_ci		err = regmap_write(priv->aio_ctl_map, STX104_ADC_CHANNEL,
20262306a36Sopenharmony_ci				   STX104_SINGLE_CHANNEL(chan->channel));
20362306a36Sopenharmony_ci		if (err) {
20462306a36Sopenharmony_ci			mutex_unlock(&priv->lock);
20562306a36Sopenharmony_ci			return err;
20662306a36Sopenharmony_ci		}
20762306a36Sopenharmony_ci
20862306a36Sopenharmony_ci		/*
20962306a36Sopenharmony_ci		 * Trigger ADC sample capture by writing to the 8-bit Software Strobe Register and
21062306a36Sopenharmony_ci		 * wait for completion; the conversion time range is 5 microseconds to 53.68 seconds
21162306a36Sopenharmony_ci		 * in steps of 25 nanoseconds. The actual Analog Input Frame Timer time interval is
21262306a36Sopenharmony_ci		 * calculated as:
21362306a36Sopenharmony_ci		 * ai_time_frame_ns = ( AIFT + 1 ) * ( 25 nanoseconds ).
21462306a36Sopenharmony_ci		 * Where 0 <= AIFT <= 2147483648.
21562306a36Sopenharmony_ci		 */
21662306a36Sopenharmony_ci		err = regmap_write(priv->aio_ctl_map, STX104_SOFTWARE_STROBE, 0);
21762306a36Sopenharmony_ci		if (err) {
21862306a36Sopenharmony_ci			mutex_unlock(&priv->lock);
21962306a36Sopenharmony_ci			return err;
22062306a36Sopenharmony_ci		}
22162306a36Sopenharmony_ci		err = regmap_read_poll_timeout(priv->aio_ctl_map, STX104_ADC_STATUS, adc_status,
22262306a36Sopenharmony_ci					       !u8_get_bits(adc_status, STX104_CNV), 0, 53687092);
22362306a36Sopenharmony_ci		if (err) {
22462306a36Sopenharmony_ci			mutex_unlock(&priv->lock);
22562306a36Sopenharmony_ci			return err;
22662306a36Sopenharmony_ci		}
22762306a36Sopenharmony_ci
22862306a36Sopenharmony_ci		err = regmap_read(priv->aio_data_map, STX104_ADC_DATA, &value);
22962306a36Sopenharmony_ci		if (err) {
23062306a36Sopenharmony_ci			mutex_unlock(&priv->lock);
23162306a36Sopenharmony_ci			return err;
23262306a36Sopenharmony_ci		}
23362306a36Sopenharmony_ci		*val = value;
23462306a36Sopenharmony_ci
23562306a36Sopenharmony_ci		mutex_unlock(&priv->lock);
23662306a36Sopenharmony_ci		return IIO_VAL_INT;
23762306a36Sopenharmony_ci	case IIO_CHAN_INFO_OFFSET:
23862306a36Sopenharmony_ci		/* get ADC bipolar/unipolar configuration */
23962306a36Sopenharmony_ci		err = regmap_read(priv->aio_ctl_map, STX104_ADC_CONFIGURATION, &adc_config);
24062306a36Sopenharmony_ci		if (err)
24162306a36Sopenharmony_ci			return err;
24262306a36Sopenharmony_ci
24362306a36Sopenharmony_ci		*val = (u8_get_bits(adc_config, STX104_ADBU) == STX104_BIPOLAR) ? -32768 : 0;
24462306a36Sopenharmony_ci		return IIO_VAL_INT;
24562306a36Sopenharmony_ci	case IIO_CHAN_INFO_SCALE:
24662306a36Sopenharmony_ci		/* get ADC bipolar/unipolar and gain configuration */
24762306a36Sopenharmony_ci		err = regmap_read(priv->aio_ctl_map, STX104_ADC_CONFIGURATION, &adc_config);
24862306a36Sopenharmony_ci		if (err)
24962306a36Sopenharmony_ci			return err;
25062306a36Sopenharmony_ci
25162306a36Sopenharmony_ci		*val = 5;
25262306a36Sopenharmony_ci		*val2 = (u8_get_bits(adc_config, STX104_ADBU) == STX104_BIPOLAR) ? 14 : 15;
25362306a36Sopenharmony_ci		*val2 += u8_get_bits(adc_config, STX104_GAIN);
25462306a36Sopenharmony_ci		return IIO_VAL_FRACTIONAL_LOG2;
25562306a36Sopenharmony_ci	}
25662306a36Sopenharmony_ci
25762306a36Sopenharmony_ci	return -EINVAL;
25862306a36Sopenharmony_ci}
25962306a36Sopenharmony_ci
26062306a36Sopenharmony_cistatic int stx104_write_raw(struct iio_dev *indio_dev,
26162306a36Sopenharmony_ci	struct iio_chan_spec const *chan, int val, int val2, long mask)
26262306a36Sopenharmony_ci{
26362306a36Sopenharmony_ci	struct stx104_iio *const priv = iio_priv(indio_dev);
26462306a36Sopenharmony_ci	u8 gain;
26562306a36Sopenharmony_ci
26662306a36Sopenharmony_ci	switch (mask) {
26762306a36Sopenharmony_ci	case IIO_CHAN_INFO_HARDWAREGAIN:
26862306a36Sopenharmony_ci		/* Only four gain states (x1, x2, x4, x8) */
26962306a36Sopenharmony_ci		switch (val) {
27062306a36Sopenharmony_ci		case 1:
27162306a36Sopenharmony_ci			gain = STX104_GAIN_X1;
27262306a36Sopenharmony_ci			break;
27362306a36Sopenharmony_ci		case 2:
27462306a36Sopenharmony_ci			gain = STX104_GAIN_X2;
27562306a36Sopenharmony_ci			break;
27662306a36Sopenharmony_ci		case 4:
27762306a36Sopenharmony_ci			gain = STX104_GAIN_X4;
27862306a36Sopenharmony_ci			break;
27962306a36Sopenharmony_ci		case 8:
28062306a36Sopenharmony_ci			gain = STX104_GAIN_X8;
28162306a36Sopenharmony_ci			break;
28262306a36Sopenharmony_ci		default:
28362306a36Sopenharmony_ci			return -EINVAL;
28462306a36Sopenharmony_ci		}
28562306a36Sopenharmony_ci
28662306a36Sopenharmony_ci		return regmap_write(priv->aio_ctl_map, STX104_ADC_CONFIGURATION, gain);
28762306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
28862306a36Sopenharmony_ci		if (!chan->output)
28962306a36Sopenharmony_ci			return -EINVAL;
29062306a36Sopenharmony_ci
29162306a36Sopenharmony_ci		if (val < 0 || val > U16_MAX)
29262306a36Sopenharmony_ci			return -EINVAL;
29362306a36Sopenharmony_ci
29462306a36Sopenharmony_ci		return regmap_write(priv->aio_data_map, STX104_DAC_OFFSET(chan->channel), val);
29562306a36Sopenharmony_ci	}
29662306a36Sopenharmony_ci
29762306a36Sopenharmony_ci	return -EINVAL;
29862306a36Sopenharmony_ci}
29962306a36Sopenharmony_ci
30062306a36Sopenharmony_cistatic const struct iio_info stx104_info = {
30162306a36Sopenharmony_ci	.read_raw = stx104_read_raw,
30262306a36Sopenharmony_ci	.write_raw = stx104_write_raw
30362306a36Sopenharmony_ci};
30462306a36Sopenharmony_ci
30562306a36Sopenharmony_ci/* single-ended input channels configuration */
30662306a36Sopenharmony_cistatic const struct iio_chan_spec stx104_channels_sing[] = {
30762306a36Sopenharmony_ci	STX104_OUT_CHAN(0), STX104_OUT_CHAN(1),
30862306a36Sopenharmony_ci	STX104_IN_CHAN(0, 0), STX104_IN_CHAN(1, 0), STX104_IN_CHAN(2, 0),
30962306a36Sopenharmony_ci	STX104_IN_CHAN(3, 0), STX104_IN_CHAN(4, 0), STX104_IN_CHAN(5, 0),
31062306a36Sopenharmony_ci	STX104_IN_CHAN(6, 0), STX104_IN_CHAN(7, 0), STX104_IN_CHAN(8, 0),
31162306a36Sopenharmony_ci	STX104_IN_CHAN(9, 0), STX104_IN_CHAN(10, 0), STX104_IN_CHAN(11, 0),
31262306a36Sopenharmony_ci	STX104_IN_CHAN(12, 0), STX104_IN_CHAN(13, 0), STX104_IN_CHAN(14, 0),
31362306a36Sopenharmony_ci	STX104_IN_CHAN(15, 0)
31462306a36Sopenharmony_ci};
31562306a36Sopenharmony_ci/* differential input channels configuration */
31662306a36Sopenharmony_cistatic const struct iio_chan_spec stx104_channels_diff[] = {
31762306a36Sopenharmony_ci	STX104_OUT_CHAN(0), STX104_OUT_CHAN(1),
31862306a36Sopenharmony_ci	STX104_IN_CHAN(0, 1), STX104_IN_CHAN(1, 1), STX104_IN_CHAN(2, 1),
31962306a36Sopenharmony_ci	STX104_IN_CHAN(3, 1), STX104_IN_CHAN(4, 1), STX104_IN_CHAN(5, 1),
32062306a36Sopenharmony_ci	STX104_IN_CHAN(6, 1), STX104_IN_CHAN(7, 1)
32162306a36Sopenharmony_ci};
32262306a36Sopenharmony_ci
32362306a36Sopenharmony_cistatic int stx104_reg_mask_xlate(struct gpio_regmap *const gpio, const unsigned int base,
32462306a36Sopenharmony_ci				 unsigned int offset, unsigned int *const reg,
32562306a36Sopenharmony_ci				 unsigned int *const mask)
32662306a36Sopenharmony_ci{
32762306a36Sopenharmony_ci	/* Output lines are located at same register bit offsets as input lines */
32862306a36Sopenharmony_ci	if (offset >= 4)
32962306a36Sopenharmony_ci		offset -= 4;
33062306a36Sopenharmony_ci
33162306a36Sopenharmony_ci	*reg = base;
33262306a36Sopenharmony_ci	*mask = BIT(offset);
33362306a36Sopenharmony_ci
33462306a36Sopenharmony_ci	return 0;
33562306a36Sopenharmony_ci}
33662306a36Sopenharmony_ci
33762306a36Sopenharmony_ci#define STX104_NGPIO 8
33862306a36Sopenharmony_cistatic const char *stx104_names[STX104_NGPIO] = {
33962306a36Sopenharmony_ci	"DIN0", "DIN1", "DIN2", "DIN3", "DOUT0", "DOUT1", "DOUT2", "DOUT3"
34062306a36Sopenharmony_ci};
34162306a36Sopenharmony_ci
34262306a36Sopenharmony_cistatic int stx104_init_hw(struct stx104_iio *const priv)
34362306a36Sopenharmony_ci{
34462306a36Sopenharmony_ci	int err;
34562306a36Sopenharmony_ci
34662306a36Sopenharmony_ci	/* configure device for software trigger operation */
34762306a36Sopenharmony_ci	err = regmap_write(priv->aio_ctl_map, STX104_ADC_CONTROL, STX104_SOFTWARE_TRIGGER);
34862306a36Sopenharmony_ci	if (err)
34962306a36Sopenharmony_ci		return err;
35062306a36Sopenharmony_ci
35162306a36Sopenharmony_ci	/* initialize gain setting to x1 */
35262306a36Sopenharmony_ci	err = regmap_write(priv->aio_ctl_map, STX104_ADC_CONFIGURATION, STX104_GAIN_X1);
35362306a36Sopenharmony_ci	if (err)
35462306a36Sopenharmony_ci		return err;
35562306a36Sopenharmony_ci
35662306a36Sopenharmony_ci	/* initialize DAC outputs to 0V */
35762306a36Sopenharmony_ci	err = regmap_write(priv->aio_data_map, STX104_DAC_BASE, 0);
35862306a36Sopenharmony_ci	if (err)
35962306a36Sopenharmony_ci		return err;
36062306a36Sopenharmony_ci	err = regmap_write(priv->aio_data_map, STX104_DAC_BASE + STX104_AIO_DATA_STRIDE, 0);
36162306a36Sopenharmony_ci	if (err)
36262306a36Sopenharmony_ci		return err;
36362306a36Sopenharmony_ci
36462306a36Sopenharmony_ci	return 0;
36562306a36Sopenharmony_ci}
36662306a36Sopenharmony_ci
36762306a36Sopenharmony_cistatic int stx104_probe(struct device *dev, unsigned int id)
36862306a36Sopenharmony_ci{
36962306a36Sopenharmony_ci	struct iio_dev *indio_dev;
37062306a36Sopenharmony_ci	struct stx104_iio *priv;
37162306a36Sopenharmony_ci	struct gpio_regmap_config gpio_config;
37262306a36Sopenharmony_ci	void __iomem *stx104_base;
37362306a36Sopenharmony_ci	struct regmap *aio_ctl_map;
37462306a36Sopenharmony_ci	struct regmap *aio_data_map;
37562306a36Sopenharmony_ci	struct regmap *dio_map;
37662306a36Sopenharmony_ci	int err;
37762306a36Sopenharmony_ci	unsigned int adc_status;
37862306a36Sopenharmony_ci
37962306a36Sopenharmony_ci	indio_dev = devm_iio_device_alloc(dev, sizeof(*priv));
38062306a36Sopenharmony_ci	if (!indio_dev)
38162306a36Sopenharmony_ci		return -ENOMEM;
38262306a36Sopenharmony_ci
38362306a36Sopenharmony_ci	if (!devm_request_region(dev, base[id], STX104_EXTENT,
38462306a36Sopenharmony_ci		dev_name(dev))) {
38562306a36Sopenharmony_ci		dev_err(dev, "Unable to lock port addresses (0x%X-0x%X)\n",
38662306a36Sopenharmony_ci			base[id], base[id] + STX104_EXTENT);
38762306a36Sopenharmony_ci		return -EBUSY;
38862306a36Sopenharmony_ci	}
38962306a36Sopenharmony_ci
39062306a36Sopenharmony_ci	stx104_base = devm_ioport_map(dev, base[id], STX104_EXTENT);
39162306a36Sopenharmony_ci	if (!stx104_base)
39262306a36Sopenharmony_ci		return -ENOMEM;
39362306a36Sopenharmony_ci
39462306a36Sopenharmony_ci	aio_ctl_map = devm_regmap_init_mmio(dev, stx104_base, &aio_ctl_regmap_config);
39562306a36Sopenharmony_ci	if (IS_ERR(aio_ctl_map))
39662306a36Sopenharmony_ci		return dev_err_probe(dev, PTR_ERR(aio_ctl_map),
39762306a36Sopenharmony_ci				     "Unable to initialize aio_ctl register map\n");
39862306a36Sopenharmony_ci
39962306a36Sopenharmony_ci	aio_data_map = devm_regmap_init_mmio(dev, stx104_base, &aio_data_regmap_config);
40062306a36Sopenharmony_ci	if (IS_ERR(aio_data_map))
40162306a36Sopenharmony_ci		return dev_err_probe(dev, PTR_ERR(aio_data_map),
40262306a36Sopenharmony_ci				     "Unable to initialize aio_data register map\n");
40362306a36Sopenharmony_ci
40462306a36Sopenharmony_ci	dio_map = devm_regmap_init_mmio(dev, stx104_base, &dio_regmap_config);
40562306a36Sopenharmony_ci	if (IS_ERR(dio_map))
40662306a36Sopenharmony_ci		return dev_err_probe(dev, PTR_ERR(dio_map),
40762306a36Sopenharmony_ci				     "Unable to initialize dio register map\n");
40862306a36Sopenharmony_ci
40962306a36Sopenharmony_ci	priv = iio_priv(indio_dev);
41062306a36Sopenharmony_ci	priv->aio_ctl_map = aio_ctl_map;
41162306a36Sopenharmony_ci	priv->aio_data_map = aio_data_map;
41262306a36Sopenharmony_ci
41362306a36Sopenharmony_ci	indio_dev->info = &stx104_info;
41462306a36Sopenharmony_ci	indio_dev->modes = INDIO_DIRECT_MODE;
41562306a36Sopenharmony_ci
41662306a36Sopenharmony_ci	err = regmap_read(aio_ctl_map, STX104_ADC_STATUS, &adc_status);
41762306a36Sopenharmony_ci	if (err)
41862306a36Sopenharmony_ci		return err;
41962306a36Sopenharmony_ci
42062306a36Sopenharmony_ci	if (u8_get_bits(adc_status, STX104_SD) == STX104_DIFFERENTIAL) {
42162306a36Sopenharmony_ci		indio_dev->num_channels = ARRAY_SIZE(stx104_channels_diff);
42262306a36Sopenharmony_ci		indio_dev->channels = stx104_channels_diff;
42362306a36Sopenharmony_ci	} else {
42462306a36Sopenharmony_ci		indio_dev->num_channels = ARRAY_SIZE(stx104_channels_sing);
42562306a36Sopenharmony_ci		indio_dev->channels = stx104_channels_sing;
42662306a36Sopenharmony_ci	}
42762306a36Sopenharmony_ci
42862306a36Sopenharmony_ci	indio_dev->name = dev_name(dev);
42962306a36Sopenharmony_ci
43062306a36Sopenharmony_ci	mutex_init(&priv->lock);
43162306a36Sopenharmony_ci
43262306a36Sopenharmony_ci	err = stx104_init_hw(priv);
43362306a36Sopenharmony_ci	if (err)
43462306a36Sopenharmony_ci		return err;
43562306a36Sopenharmony_ci
43662306a36Sopenharmony_ci	err = devm_iio_device_register(dev, indio_dev);
43762306a36Sopenharmony_ci	if (err)
43862306a36Sopenharmony_ci		return err;
43962306a36Sopenharmony_ci
44062306a36Sopenharmony_ci	gpio_config = (struct gpio_regmap_config) {
44162306a36Sopenharmony_ci		.parent = dev,
44262306a36Sopenharmony_ci		.regmap = dio_map,
44362306a36Sopenharmony_ci		.ngpio = STX104_NGPIO,
44462306a36Sopenharmony_ci		.names = stx104_names,
44562306a36Sopenharmony_ci		.reg_dat_base = GPIO_REGMAP_ADDR(STX104_DIO_REG),
44662306a36Sopenharmony_ci		.reg_set_base = GPIO_REGMAP_ADDR(STX104_DIO_REG),
44762306a36Sopenharmony_ci		.ngpio_per_reg = STX104_NGPIO,
44862306a36Sopenharmony_ci		.reg_mask_xlate = stx104_reg_mask_xlate,
44962306a36Sopenharmony_ci		.drvdata = dio_map,
45062306a36Sopenharmony_ci	};
45162306a36Sopenharmony_ci
45262306a36Sopenharmony_ci	return PTR_ERR_OR_ZERO(devm_gpio_regmap_register(dev, &gpio_config));
45362306a36Sopenharmony_ci}
45462306a36Sopenharmony_ci
45562306a36Sopenharmony_cistatic struct isa_driver stx104_driver = {
45662306a36Sopenharmony_ci	.probe = stx104_probe,
45762306a36Sopenharmony_ci	.driver = {
45862306a36Sopenharmony_ci		.name = "stx104"
45962306a36Sopenharmony_ci	},
46062306a36Sopenharmony_ci};
46162306a36Sopenharmony_ci
46262306a36Sopenharmony_cimodule_isa_driver(stx104_driver, num_stx104);
46362306a36Sopenharmony_ci
46462306a36Sopenharmony_ciMODULE_AUTHOR("William Breathitt Gray <vilhelm.gray@gmail.com>");
46562306a36Sopenharmony_ciMODULE_DESCRIPTION("Apex Embedded Systems STX104 IIO driver");
46662306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
467