// SPDX-License-Identifier: GPL-2.0-only
/*
 * SPI IIO driver for Bosch BMA400 triaxial acceleration sensor.
 *
 * Copyright 2020 Dan Robertson <dan@dlrobertson.com>
 *
 */
#include <linux/bits.h>
#include <linux/init.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/regmap.h>
#include <linux/spi/spi.h>

#include "bma400.h"

#define BMA400_MAX_SPI_READ 2
#define BMA400_SPI_READ_BUFFER_SIZE (BMA400_MAX_SPI_READ + 1)

static int bma400_regmap_spi_read(void *context,
      const void *reg, size_t reg_size,
      void *val, size_t val_size)
{
 struct device *dev = context;
 struct spi_device *spi = to_spi_device(dev);
 u8 result[BMA400_SPI_READ_BUFFER_SIZE];
 ssize_t status;

 if (val_size > BMA400_MAX_SPI_READ)
  return -EINVAL;

 status = spi_write_then_read(spi, reg, 1, result, val_size + 1);
 if (status)
  return status;

 /*
  * From the BMA400 datasheet:
  *
  * > For a basic read operation two bytes have to be read and the first
  * > has to be dropped and the second byte must be interpreted.
 */
 memcpy(val, result + 1, val_size);

 return 0;
}

static int bma400_regmap_spi_write(void *context, const void *data,
       size_t count)
{
 struct device *dev = context;
 struct spi_device *spi = to_spi_device(dev);

 return spi_write(spi, data, count);
}

static const struct regmap_bus bma400_regmap_bus = {
 .read = bma400_regmap_spi_read,
 .write = bma400_regmap_spi_write,
 .read_flag_mask = BIT(7),
 .max_raw_read = BMA400_MAX_SPI_READ,
};

static int bma400_spi_probe(struct spi_device *spi)
{
 const struct spi_device_id *id = spi_get_device_id(spi);
 struct regmap *regmap;
 unsigned int val;
 int ret;

 regmap = devm_regmap_init(&spi->dev, &bma400_regmap_bus,
      &spi->dev, &bma400_regmap_config);
 if (IS_ERR(regmap)) {
  dev_err(&spi->dev, "failed to create regmap\n");
  return PTR_ERR(regmap);
 }

 /*
  * Per the bma400 datasheet, the first SPI read may
  * return garbage. As the datasheet recommends, the
  * chip ID register will be read here and checked
  * again in the following probe.
 */
 ret = regmap_read(regmap, BMA400_CHIP_ID_REG, &val);
 if (ret)
  dev_err(&spi->dev, "Failed to read chip id register\n");

 return bma400_probe(&spi->dev, regmap, spi->irq, id->name);
}

static const struct spi_device_id bma400_spi_ids[] = {
 { "bma400", 0 },
 { }
};
MODULE_DEVICE_TABLE(spi, bma400_spi_ids);

static const struct of_device_id bma400_of_spi_match[] = {
 { .compatible = "bosch,bma400" },
 { }
};
MODULE_DEVICE_TABLE(of, bma400_of_spi_match);

static struct spi_driver bma400_spi_driver = {
 .driver = {
  .name = "bma400",
  .of_match_table = bma400_of_spi_match,
 },
 .probe    = bma400_spi_probe,
 .id_table = bma400_spi_ids,
};

module_spi_driver(bma400_spi_driver);
MODULE_AUTHOR("Dan Robertson <dan@dlrobertson.com>");
MODULE_DESCRIPTION("Bosch BMA400 triaxial acceleration sensor (SPI)");
MODULE_LICENSE("GPL");
MODULE_IMPORT_NS("IIO_BMA400");