// SPDX-License-Identifier: GPL-2.0
/*
 * AD5672R, AD5674R, AD5676, AD5676R, AD5679R,
 * AD5681R, AD5682R, AD5683, AD5683R, AD5684,
 * AD5684R, AD5685R, AD5686, AD5686R
 * Digital to analog converters driver
 *
 * Copyright 2018 Analog Devices Inc.
 */

#include <linux/array_size.h>
#include <linux/errno.h>
#include <linux/module.h>
#include <linux/spi/spi.h>

#include <asm/byteorder.h>

#include "ad5686.h"

static int ad5686_spi_write(struct ad5686_state *st,
                            u8 cmd, u8 addr, u16 val)
{
        struct spi_device *spi = to_spi_device(st->dev);
        u8 tx_len, *buf;

        switch (st->chip_info->regmap_type) {
        case AD5310_REGMAP:
                st->data[0].d16 = cpu_to_be16(AD5310_CMD(cmd) |
                                              val);
                buf = &st->data[0].d8[0];
                tx_len = 2;
                break;
        case AD5683_REGMAP:
                st->data[0].d32 = cpu_to_be32(AD5686_CMD(cmd) |
                                              AD5683_DATA(val));
                buf = &st->data[0].d8[1];
                tx_len = 3;
                break;
        case AD5686_REGMAP:
                st->data[0].d32 = cpu_to_be32(AD5686_CMD(cmd) |
                                              AD5686_ADDR(addr) |
                                              val);
                buf = &st->data[0].d8[1];
                tx_len = 3;
                break;
        default:
                return -EINVAL;
        }

        return spi_write(spi, buf, tx_len);
}

static int ad5686_spi_read(struct ad5686_state *st, u8 addr)
{
        struct spi_transfer t[] = {
                {
                        .tx_buf = &st->data[0].d8[1],
                        .len = 3,
                        .cs_change = 1,
                }, {
                        .tx_buf = &st->data[1].d8[1],
                        .rx_buf = &st->data[2].d8[1],
                        .len = 3,
                },
        };
        struct spi_device *spi = to_spi_device(st->dev);
        u8 cmd = 0;
        int ret;

        switch (st->chip_info->regmap_type) {
        case AD5310_REGMAP:
                return -ENOTSUPP;
        case AD5683_REGMAP:
                cmd = AD5686_CMD_READBACK_ENABLE_V2;
                break;
        case AD5686_REGMAP:
                cmd = AD5686_CMD_READBACK_ENABLE;
                break;
        default:
                return -EINVAL;
        }

        st->data[0].d32 = cpu_to_be32(AD5686_CMD(cmd) |
                                      AD5686_ADDR(addr));
        st->data[1].d32 = cpu_to_be32(AD5686_CMD(AD5686_CMD_NOOP));

        ret = spi_sync_transfer(spi, t, ARRAY_SIZE(t));
        if (ret < 0)
                return ret;

        return be32_to_cpu(st->data[2].d32);
}

static const struct ad5686_bus_ops ad5686_spi_ops = {
        .write = ad5686_spi_write,
        .read = ad5686_spi_read,
};

static int ad5686_spi_probe(struct spi_device *spi)
{
        return ad5686_probe(&spi->dev, spi_get_device_match_data(spi),
                            spi->modalias, &ad5686_spi_ops);
}

static const struct spi_device_id ad5686_spi_id[] = {
        { .name = "ad5310r", .driver_data = (kernel_ulong_t)&ad5310r_chip_info },
        { .name = "ad5672r", .driver_data = (kernel_ulong_t)&ad5672r_chip_info },
        { .name = "ad5674r", .driver_data = (kernel_ulong_t)&ad5674r_chip_info },
        { .name = "ad5676",  .driver_data = (kernel_ulong_t)&ad5676_chip_info },
        { .name = "ad5676r", .driver_data = (kernel_ulong_t)&ad5676r_chip_info },
        { .name = "ad5679r", .driver_data = (kernel_ulong_t)&ad5679r_chip_info },
        { .name = "ad5681r", .driver_data = (kernel_ulong_t)&ad5681r_chip_info },
        { .name = "ad5682r", .driver_data = (kernel_ulong_t)&ad5682r_chip_info },
        { .name = "ad5683",  .driver_data = (kernel_ulong_t)&ad5683_chip_info },
        { .name = "ad5683r", .driver_data = (kernel_ulong_t)&ad5683r_chip_info },
        { .name = "ad5684",  .driver_data = (kernel_ulong_t)&ad5684_chip_info },
        { .name = "ad5684r", .driver_data = (kernel_ulong_t)&ad5684r_chip_info },
        { .name = "ad5685",  .driver_data = (kernel_ulong_t)&ad5685r_chip_info }, /* nonexistent */
        { .name = "ad5685r", .driver_data = (kernel_ulong_t)&ad5685r_chip_info },
        { .name = "ad5686",  .driver_data = (kernel_ulong_t)&ad5686_chip_info },
        { .name = "ad5686r", .driver_data = (kernel_ulong_t)&ad5686r_chip_info },
        { }
};
MODULE_DEVICE_TABLE(spi, ad5686_spi_id);

static const struct of_device_id ad5686_of_match[] = {
        { .compatible = "adi,ad5310r", .data = &ad5310r_chip_info },
        { .compatible = "adi,ad5672r", .data = &ad5672r_chip_info },
        { .compatible = "adi,ad5674r", .data = &ad5674r_chip_info },
        { .compatible = "adi,ad5676",  .data = &ad5676_chip_info },
        { .compatible = "adi,ad5676r", .data = &ad5676r_chip_info },
        { .compatible = "adi,ad5679r", .data = &ad5679r_chip_info },
        { .compatible = "adi,ad5681r", .data = &ad5681r_chip_info },
        { .compatible = "adi,ad5682r", .data = &ad5682r_chip_info },
        { .compatible = "adi,ad5683",  .data = &ad5683_chip_info },
        { .compatible = "adi,ad5683r", .data = &ad5683r_chip_info },
        { .compatible = "adi,ad5684",  .data = &ad5684_chip_info },
        { .compatible = "adi,ad5684r", .data = &ad5684r_chip_info },
        { .compatible = "adi,ad5685r", .data = &ad5685r_chip_info },
        { .compatible = "adi,ad5686",  .data = &ad5686_chip_info },
        { .compatible = "adi,ad5686r", .data = &ad5686r_chip_info },
        { }
};
MODULE_DEVICE_TABLE(of, ad5686_of_match);

static struct spi_driver ad5686_spi_driver = {
        .driver = {
                .name = "ad5686",
                .of_match_table = ad5686_of_match,
        },
        .probe = ad5686_spi_probe,
        .id_table = ad5686_spi_id,
};

module_spi_driver(ad5686_spi_driver);

MODULE_AUTHOR("Stefan Popa <stefan.popa@analog.com>");
MODULE_DESCRIPTION("Analog Devices AD5686 and similar multi-channel DACs");
MODULE_LICENSE("GPL v2");
MODULE_IMPORT_NS("IIO_AD5686");