// SPDX-License-Identifier: GPL-2.0
/*
 * BME680 - SPI Driver
 *
 * Copyright (C) 2018 Himanshu Jha <himanshujha199640@gmail.com>
 */
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/regmap.h>
#include <linux/spi/spi.h>

#include "bme680.h"

struct bme680_spi_bus_context {
        struct spi_device *spi;
        u8 current_page;
};

/*
 * In SPI mode there are only 7 address bits, a "page" register determines
 * which part of the 8-bit range is active. This function looks at the address
 * and writes the page selection bit if needed
 */
static int bme680_regmap_spi_select_page(
        struct bme680_spi_bus_context *ctx, u8 reg)
{
        struct spi_device *spi = ctx->spi;
        int ret;
        u8 buf[2];
        u8 page = (reg & 0x80) ? 0 : 1; /* Page "1" is low range */

        if (page == ctx->current_page)
                return 0;

        /*
         * Data sheet claims we're only allowed to change bit 4, so we must do
         * a read-modify-write on each and every page select
         */
        buf[0] = BME680_REG_STATUS;
        ret = spi_write_then_read(spi, buf, 1, buf + 1, 1);
        if (ret < 0) {
                dev_err(&spi->dev, "failed to set page %u\n", page);
                return ret;
        }

        buf[0] = BME680_REG_STATUS;
        if (page)
                buf[1] |= BME680_SPI_MEM_PAGE_BIT;
        else
                buf[1] &= ~BME680_SPI_MEM_PAGE_BIT;

        ret = spi_write(spi, buf, 2);
        if (ret < 0) {
                dev_err(&spi->dev, "failed to set page %u\n", page);
                return ret;
        }

        ctx->current_page = page;

        return 0;
}

static int bme680_regmap_spi_write(void *context, const void *data,
                                   size_t count)
{
        struct bme680_spi_bus_context *ctx = context;
        struct spi_device *spi = ctx->spi;
        int ret;
        u8 buf[2];

        memcpy(buf, data, 2);

        ret = bme680_regmap_spi_select_page(ctx, buf[0]);
        if (ret)
                return ret;

        /*
         * The SPI register address (= full register address without bit 7)
         * and the write command (bit7 = RW = '0')
         */
        buf[0] &= ~0x80;

        return spi_write(spi, buf, 2);
}

static int bme680_regmap_spi_read(void *context, const void *reg,
                                  size_t reg_size, void *val, size_t val_size)
{
        struct bme680_spi_bus_context *ctx = context;
        struct spi_device *spi = ctx->spi;
        int ret;
        u8 addr = *(const u8 *)reg;

        ret = bme680_regmap_spi_select_page(ctx, addr);
        if (ret)
                return ret;

        addr |= 0x80; /* bit7 = RW = '1' */

        return spi_write_then_read(spi, &addr, 1, val, val_size);
}

static const struct regmap_bus bme680_regmap_bus = {
        .write = bme680_regmap_spi_write,
        .read = bme680_regmap_spi_read,
        .reg_format_endian_default = REGMAP_ENDIAN_BIG,
        .val_format_endian_default = REGMAP_ENDIAN_BIG,
};

static int bme680_spi_probe(struct spi_device *spi)
{
        const struct spi_device_id *id = spi_get_device_id(spi);
        struct bme680_spi_bus_context *bus_context;
        struct regmap *regmap;

        bus_context = devm_kzalloc(&spi->dev, sizeof(*bus_context), GFP_KERNEL);
        if (!bus_context)
                return -ENOMEM;

        bus_context->spi = spi;
        bus_context->current_page = 0xff; /* Undefined on warm boot */

        regmap = devm_regmap_init(&spi->dev, &bme680_regmap_bus,
                                  bus_context, &bme680_regmap_config);
        if (IS_ERR(regmap)) {
                dev_err(&spi->dev, "Failed to register spi regmap %ld\n", PTR_ERR(regmap));
                return PTR_ERR(regmap);
        }

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

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

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

static struct spi_driver bme680_spi_driver = {
        .driver = {
                .name                   = "bme680_spi",
                .of_match_table         = bme680_of_spi_match,
                .pm = pm_ptr(&bme680_dev_pm_ops),
        },
        .probe = bme680_spi_probe,
        .id_table = bme680_spi_id,
};
module_spi_driver(bme680_spi_driver);

MODULE_AUTHOR("Himanshu Jha <himanshujha199640@gmail.com>");
MODULE_DESCRIPTION("Bosch BME680 SPI driver");
MODULE_LICENSE("GPL v2");
MODULE_IMPORT_NS("IIO_BME680");