162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * IIO driver for Domintech DMARD06 accelerometer
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2016 Aleksei Mamlin <mamlinav@gmail.com>
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci
862306a36Sopenharmony_ci#include <linux/module.h>
962306a36Sopenharmony_ci#include <linux/mod_devicetable.h>
1062306a36Sopenharmony_ci#include <linux/i2c.h>
1162306a36Sopenharmony_ci#include <linux/iio/iio.h>
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#define DMARD06_DRV_NAME		"dmard06"
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci/* Device data registers */
1662306a36Sopenharmony_ci#define DMARD06_CHIP_ID_REG		0x0f
1762306a36Sopenharmony_ci#define DMARD06_TOUT_REG		0x40
1862306a36Sopenharmony_ci#define DMARD06_XOUT_REG		0x41
1962306a36Sopenharmony_ci#define DMARD06_YOUT_REG		0x42
2062306a36Sopenharmony_ci#define DMARD06_ZOUT_REG		0x43
2162306a36Sopenharmony_ci#define DMARD06_CTRL1_REG		0x44
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci/* Device ID value */
2462306a36Sopenharmony_ci#define DMARD05_CHIP_ID			0x05
2562306a36Sopenharmony_ci#define DMARD06_CHIP_ID			0x06
2662306a36Sopenharmony_ci#define DMARD07_CHIP_ID			0x07
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci/* Device values */
2962306a36Sopenharmony_ci#define DMARD05_AXIS_SCALE_VAL		15625
3062306a36Sopenharmony_ci#define DMARD06_AXIS_SCALE_VAL		31250
3162306a36Sopenharmony_ci#define DMARD06_TEMP_CENTER_VAL		25
3262306a36Sopenharmony_ci#define DMARD06_SIGN_BIT		7
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci/* Device power modes */
3562306a36Sopenharmony_ci#define DMARD06_MODE_NORMAL		0x27
3662306a36Sopenharmony_ci#define DMARD06_MODE_POWERDOWN		0x00
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_ci/* Device channels */
3962306a36Sopenharmony_ci#define DMARD06_ACCEL_CHANNEL(_axis, _reg) {			\
4062306a36Sopenharmony_ci	.type = IIO_ACCEL,					\
4162306a36Sopenharmony_ci	.address = _reg,					\
4262306a36Sopenharmony_ci	.channel2 = IIO_MOD_##_axis,				\
4362306a36Sopenharmony_ci	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW),		\
4462306a36Sopenharmony_ci	.info_mask_shared_by_type = BIT(IIO_CHAN_INFO_SCALE),	\
4562306a36Sopenharmony_ci	.modified = 1,						\
4662306a36Sopenharmony_ci}
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci#define DMARD06_TEMP_CHANNEL(_reg) {				\
4962306a36Sopenharmony_ci	.type = IIO_TEMP,					\
5062306a36Sopenharmony_ci	.address = _reg,					\
5162306a36Sopenharmony_ci	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |		\
5262306a36Sopenharmony_ci			      BIT(IIO_CHAN_INFO_OFFSET),	\
5362306a36Sopenharmony_ci}
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_cistruct dmard06_data {
5662306a36Sopenharmony_ci	struct i2c_client *client;
5762306a36Sopenharmony_ci	u8 chip_id;
5862306a36Sopenharmony_ci};
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_cistatic const struct iio_chan_spec dmard06_channels[] = {
6162306a36Sopenharmony_ci	DMARD06_ACCEL_CHANNEL(X, DMARD06_XOUT_REG),
6262306a36Sopenharmony_ci	DMARD06_ACCEL_CHANNEL(Y, DMARD06_YOUT_REG),
6362306a36Sopenharmony_ci	DMARD06_ACCEL_CHANNEL(Z, DMARD06_ZOUT_REG),
6462306a36Sopenharmony_ci	DMARD06_TEMP_CHANNEL(DMARD06_TOUT_REG),
6562306a36Sopenharmony_ci};
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_cistatic int dmard06_read_raw(struct iio_dev *indio_dev,
6862306a36Sopenharmony_ci			    struct iio_chan_spec const *chan,
6962306a36Sopenharmony_ci			    int *val, int *val2, long mask)
7062306a36Sopenharmony_ci{
7162306a36Sopenharmony_ci	struct dmard06_data *dmard06 = iio_priv(indio_dev);
7262306a36Sopenharmony_ci	int ret;
7362306a36Sopenharmony_ci
7462306a36Sopenharmony_ci	switch (mask) {
7562306a36Sopenharmony_ci	case IIO_CHAN_INFO_RAW:
7662306a36Sopenharmony_ci		ret = i2c_smbus_read_byte_data(dmard06->client,
7762306a36Sopenharmony_ci					       chan->address);
7862306a36Sopenharmony_ci		if (ret < 0) {
7962306a36Sopenharmony_ci			dev_err(&dmard06->client->dev,
8062306a36Sopenharmony_ci				"Error reading data: %d\n", ret);
8162306a36Sopenharmony_ci			return ret;
8262306a36Sopenharmony_ci		}
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci		*val = sign_extend32(ret, DMARD06_SIGN_BIT);
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci		if (dmard06->chip_id == DMARD06_CHIP_ID)
8762306a36Sopenharmony_ci			*val = *val >> 1;
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci		switch (chan->type) {
9062306a36Sopenharmony_ci		case IIO_ACCEL:
9162306a36Sopenharmony_ci			return IIO_VAL_INT;
9262306a36Sopenharmony_ci		case IIO_TEMP:
9362306a36Sopenharmony_ci			if (dmard06->chip_id != DMARD06_CHIP_ID)
9462306a36Sopenharmony_ci				*val = *val / 2;
9562306a36Sopenharmony_ci			return IIO_VAL_INT;
9662306a36Sopenharmony_ci		default:
9762306a36Sopenharmony_ci			return -EINVAL;
9862306a36Sopenharmony_ci		}
9962306a36Sopenharmony_ci	case IIO_CHAN_INFO_OFFSET:
10062306a36Sopenharmony_ci		switch (chan->type) {
10162306a36Sopenharmony_ci		case IIO_TEMP:
10262306a36Sopenharmony_ci			*val = DMARD06_TEMP_CENTER_VAL;
10362306a36Sopenharmony_ci			return IIO_VAL_INT;
10462306a36Sopenharmony_ci		default:
10562306a36Sopenharmony_ci			return -EINVAL;
10662306a36Sopenharmony_ci		}
10762306a36Sopenharmony_ci	case IIO_CHAN_INFO_SCALE:
10862306a36Sopenharmony_ci		switch (chan->type) {
10962306a36Sopenharmony_ci		case IIO_ACCEL:
11062306a36Sopenharmony_ci			*val = 0;
11162306a36Sopenharmony_ci			if (dmard06->chip_id == DMARD06_CHIP_ID)
11262306a36Sopenharmony_ci				*val2 = DMARD06_AXIS_SCALE_VAL;
11362306a36Sopenharmony_ci			else
11462306a36Sopenharmony_ci				*val2 = DMARD05_AXIS_SCALE_VAL;
11562306a36Sopenharmony_ci			return IIO_VAL_INT_PLUS_MICRO;
11662306a36Sopenharmony_ci		default:
11762306a36Sopenharmony_ci			return -EINVAL;
11862306a36Sopenharmony_ci		}
11962306a36Sopenharmony_ci	default:
12062306a36Sopenharmony_ci		return -EINVAL;
12162306a36Sopenharmony_ci	}
12262306a36Sopenharmony_ci}
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_cistatic const struct iio_info dmard06_info = {
12562306a36Sopenharmony_ci	.read_raw	= dmard06_read_raw,
12662306a36Sopenharmony_ci};
12762306a36Sopenharmony_ci
12862306a36Sopenharmony_cistatic int dmard06_probe(struct i2c_client *client)
12962306a36Sopenharmony_ci{
13062306a36Sopenharmony_ci	int ret;
13162306a36Sopenharmony_ci	struct iio_dev *indio_dev;
13262306a36Sopenharmony_ci	struct dmard06_data *dmard06;
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ci	if (!i2c_check_functionality(client->adapter, I2C_FUNC_I2C)) {
13562306a36Sopenharmony_ci		dev_err(&client->dev, "I2C check functionality failed\n");
13662306a36Sopenharmony_ci		return -ENXIO;
13762306a36Sopenharmony_ci	}
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*dmard06));
14062306a36Sopenharmony_ci	if (!indio_dev) {
14162306a36Sopenharmony_ci		dev_err(&client->dev, "Failed to allocate iio device\n");
14262306a36Sopenharmony_ci		return -ENOMEM;
14362306a36Sopenharmony_ci	}
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	dmard06 = iio_priv(indio_dev);
14662306a36Sopenharmony_ci	dmard06->client = client;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	ret = i2c_smbus_read_byte_data(dmard06->client, DMARD06_CHIP_ID_REG);
14962306a36Sopenharmony_ci	if (ret < 0) {
15062306a36Sopenharmony_ci		dev_err(&client->dev, "Error reading chip id: %d\n", ret);
15162306a36Sopenharmony_ci		return ret;
15262306a36Sopenharmony_ci	}
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	if (ret != DMARD05_CHIP_ID && ret != DMARD06_CHIP_ID &&
15562306a36Sopenharmony_ci	    ret != DMARD07_CHIP_ID) {
15662306a36Sopenharmony_ci		dev_err(&client->dev, "Invalid chip id: %02d\n", ret);
15762306a36Sopenharmony_ci		return -ENODEV;
15862306a36Sopenharmony_ci	}
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ci	dmard06->chip_id = ret;
16162306a36Sopenharmony_ci
16262306a36Sopenharmony_ci	i2c_set_clientdata(client, indio_dev);
16362306a36Sopenharmony_ci	indio_dev->name = DMARD06_DRV_NAME;
16462306a36Sopenharmony_ci	indio_dev->modes = INDIO_DIRECT_MODE;
16562306a36Sopenharmony_ci	indio_dev->channels = dmard06_channels;
16662306a36Sopenharmony_ci	indio_dev->num_channels = ARRAY_SIZE(dmard06_channels);
16762306a36Sopenharmony_ci	indio_dev->info = &dmard06_info;
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_ci	return devm_iio_device_register(&client->dev, indio_dev);
17062306a36Sopenharmony_ci}
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_cistatic int dmard06_suspend(struct device *dev)
17362306a36Sopenharmony_ci{
17462306a36Sopenharmony_ci	struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
17562306a36Sopenharmony_ci	struct dmard06_data *dmard06 = iio_priv(indio_dev);
17662306a36Sopenharmony_ci	int ret;
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci	ret = i2c_smbus_write_byte_data(dmard06->client, DMARD06_CTRL1_REG,
17962306a36Sopenharmony_ci					DMARD06_MODE_POWERDOWN);
18062306a36Sopenharmony_ci	if (ret < 0)
18162306a36Sopenharmony_ci		return ret;
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_ci	return 0;
18462306a36Sopenharmony_ci}
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_cistatic int dmard06_resume(struct device *dev)
18762306a36Sopenharmony_ci{
18862306a36Sopenharmony_ci	struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
18962306a36Sopenharmony_ci	struct dmard06_data *dmard06 = iio_priv(indio_dev);
19062306a36Sopenharmony_ci	int ret;
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	ret = i2c_smbus_write_byte_data(dmard06->client, DMARD06_CTRL1_REG,
19362306a36Sopenharmony_ci					DMARD06_MODE_NORMAL);
19462306a36Sopenharmony_ci	if (ret < 0)
19562306a36Sopenharmony_ci		return ret;
19662306a36Sopenharmony_ci
19762306a36Sopenharmony_ci	return 0;
19862306a36Sopenharmony_ci}
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_cistatic DEFINE_SIMPLE_DEV_PM_OPS(dmard06_pm_ops, dmard06_suspend,
20162306a36Sopenharmony_ci				dmard06_resume);
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_cistatic const struct i2c_device_id dmard06_id[] = {
20462306a36Sopenharmony_ci	{ "dmard05", 0 },
20562306a36Sopenharmony_ci	{ "dmard06", 0 },
20662306a36Sopenharmony_ci	{ "dmard07", 0 },
20762306a36Sopenharmony_ci	{ }
20862306a36Sopenharmony_ci};
20962306a36Sopenharmony_ciMODULE_DEVICE_TABLE(i2c, dmard06_id);
21062306a36Sopenharmony_ci
21162306a36Sopenharmony_cistatic const struct of_device_id dmard06_of_match[] = {
21262306a36Sopenharmony_ci	{ .compatible = "domintech,dmard05" },
21362306a36Sopenharmony_ci	{ .compatible = "domintech,dmard06" },
21462306a36Sopenharmony_ci	{ .compatible = "domintech,dmard07" },
21562306a36Sopenharmony_ci	{ }
21662306a36Sopenharmony_ci};
21762306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, dmard06_of_match);
21862306a36Sopenharmony_ci
21962306a36Sopenharmony_cistatic struct i2c_driver dmard06_driver = {
22062306a36Sopenharmony_ci	.probe = dmard06_probe,
22162306a36Sopenharmony_ci	.id_table = dmard06_id,
22262306a36Sopenharmony_ci	.driver = {
22362306a36Sopenharmony_ci		.name = DMARD06_DRV_NAME,
22462306a36Sopenharmony_ci		.of_match_table = dmard06_of_match,
22562306a36Sopenharmony_ci		.pm = pm_sleep_ptr(&dmard06_pm_ops),
22662306a36Sopenharmony_ci	},
22762306a36Sopenharmony_ci};
22862306a36Sopenharmony_cimodule_i2c_driver(dmard06_driver);
22962306a36Sopenharmony_ci
23062306a36Sopenharmony_ciMODULE_AUTHOR("Aleksei Mamlin <mamlinav@gmail.com>");
23162306a36Sopenharmony_ciMODULE_DESCRIPTION("Domintech DMARD06 accelerometer driver");
23262306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
233