// SPDX-License-Identifier: GPL-2.0-only
/*
 * Driver for Sensirion sdp500 and sdp510 pressure sensors
 *
 * Datasheet: https://sensirion.com/resource/datasheet/sdp600
 */

#include <linux/i2c.h>
#include <linux/crc8.h>
#include <linux/iio/iio.h>
#include <linux/mod_devicetable.h>
#include <linux/regulator/consumer.h>
#include <linux/unaligned.h>

#define SDP500_CRC8_POLYNOMIAL  0x31   /* x8+x5+x4+1 (normalized to 0x31) */
#define SDP500_READ_SIZE        3

#define SDP500_I2C_START_MEAS 0xF1

struct sdp500_data {
	struct device *dev;
};

DECLARE_CRC8_TABLE(sdp500_crc8_table);

static int sdp500_start_measurement(struct sdp500_data *data)
{
	struct i2c_client *client = to_i2c_client(data->dev);

	return i2c_smbus_write_byte(client, SDP500_I2C_START_MEAS);
}

static const struct iio_chan_spec sdp500_channels[] = {
	{
		.type = IIO_PRESSURE,
		.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
			BIT(IIO_CHAN_INFO_SCALE),
	},
};

static int sdp500_read_raw(struct iio_dev *indio_dev,
			  struct iio_chan_spec const *chan,
			  int *val, int *val2, long mask)
{
	int ret;
	u8 rxbuf[SDP500_READ_SIZE];
	u8 received_crc, calculated_crc;
	struct sdp500_data *data = iio_priv(indio_dev);
	struct i2c_client *client = to_i2c_client(data->dev);

	switch (mask) {
	case IIO_CHAN_INFO_RAW:
		ret = i2c_master_recv(client, rxbuf, SDP500_READ_SIZE);
		if (ret < 0) {
			dev_err(data->dev, "Failed to receive data");
			return ret;
		}
		if (ret != SDP500_READ_SIZE) {
			dev_err(data->dev, "Data is received wrongly");
			return -EIO;
		}

		received_crc = rxbuf[2];
		calculated_crc = crc8(sdp500_crc8_table, rxbuf,
				      sizeof(rxbuf) - 1, 0x00);
		if (received_crc != calculated_crc) {
			dev_err(data->dev,
				"calculated crc = 0x%.2X, received 0x%.2X",
				calculated_crc, received_crc);
			return -EIO;
		}

		*val = get_unaligned_be16(rxbuf);
		return IIO_VAL_INT;
	case IIO_CHAN_INFO_SCALE:
		*val = 1;
		*val2 = 60;

		return IIO_VAL_FRACTIONAL;
	default:
		return -EINVAL;
	}
}

static const struct iio_info sdp500_info = {
	.read_raw = &sdp500_read_raw,
};

static int sdp500_probe(struct i2c_client *client)
{
	struct iio_dev *indio_dev;
	struct sdp500_data *data;
	struct device *dev = &client->dev;
	int ret;
	u8 rxbuf[SDP500_READ_SIZE];

	ret = devm_regulator_get_enable(dev, "vdd");
	if (ret)
		return dev_err_probe(dev, ret,
			"Failed to get and enable regulator\n");

	indio_dev = devm_iio_device_alloc(dev, sizeof(*data));
	if (!indio_dev)
		return -ENOMEM;

	/* has to be done before the first i2c communication */
	crc8_populate_msb(sdp500_crc8_table, SDP500_CRC8_POLYNOMIAL);

	data = iio_priv(indio_dev);
	data->dev = dev;

	indio_dev->name = "sdp500";
	indio_dev->channels = sdp500_channels;
	indio_dev->info = &sdp500_info;
	indio_dev->modes = INDIO_DIRECT_MODE;
	indio_dev->num_channels = ARRAY_SIZE(sdp500_channels);

	ret = sdp500_start_measurement(data);
	if (ret)
		return dev_err_probe(dev, ret, "Failed to start measurement");

	/* First measurement is not correct, read it out to get rid of it */
	i2c_master_recv(client, rxbuf, SDP500_READ_SIZE);

	ret = devm_iio_device_register(dev, indio_dev);
	if (ret)
		return dev_err_probe(dev, ret, "Failed to register indio_dev");

	return 0;
}

static const struct i2c_device_id sdp500_id[] = {
	{ "sdp500" },
	{ }
};
MODULE_DEVICE_TABLE(i2c, sdp500_id);

static const struct of_device_id sdp500_of_match[] = {
	{ .compatible = "sensirion,sdp500" },
	{ }
};
MODULE_DEVICE_TABLE(of, sdp500_of_match);

static struct i2c_driver sdp500_driver = {
	.driver = {
		.name = "sensirion,sdp500",
		.of_match_table = sdp500_of_match,
	},
	.probe = sdp500_probe,
	.id_table = sdp500_id,
};
module_i2c_driver(sdp500_driver);

MODULE_AUTHOR("Thomas Sioutas <thomas.sioutas@prodrive-technologies.com>");
MODULE_DESCRIPTION("Driver for Sensirion SDP500 differential pressure sensor");
MODULE_LICENSE("GPL");