// SPDX-License-Identifier: GPL-2.0
/*
 * AD5446 SPI I2C driver
 *
 * Copyright 2025 Analog Devices Inc.
 */
#include <linux/err.h>
#include <linux/module.h>
#include <linux/mod_devicetable.h>
#include <linux/i2c.h>

#include <asm/byteorder.h>

#include "ad5446.h"

static int ad5622_write(struct ad5446_state *st, unsigned int val)
{
        struct i2c_client *client = to_i2c_client(st->dev);
        int ret;

        st->d16 = cpu_to_be16(val);

        ret = i2c_master_send_dmasafe(client, (char *)&st->d16, sizeof(st->d16));
        if (ret < 0)
                return ret;
        if (ret != sizeof(st->d16))
                return -EIO;

        return 0;
}

static int ad5446_i2c_probe(struct i2c_client *i2c)
{
        const struct i2c_device_id *id = i2c_client_get_device_id(i2c);
        const struct ad5446_chip_info *chip_info;

        chip_info = i2c_get_match_data(i2c);
        if (!chip_info)
                return -ENODEV;

        return ad5446_probe(&i2c->dev, id->name, chip_info);
}

/*
 * ad5446_supported_i2c_device_ids:
 * The AD5620/40/60 parts are available in different fixed internal reference
 * voltage options. The actual part numbers may look differently
 * (and a bit cryptic), however this style is used to make clear which
 * parts are supported here.
 */

static const struct ad5446_chip_info ad5602_chip_info = {
        .channel = AD5446_CHANNEL_POWERDOWN(8, 16, 4),
        .write = ad5622_write,
};

static const struct ad5446_chip_info ad5612_chip_info = {
        .channel = AD5446_CHANNEL_POWERDOWN(10, 16, 2),
        .write = ad5622_write,
};

static const struct ad5446_chip_info ad5622_chip_info = {
        .channel = AD5446_CHANNEL_POWERDOWN(12, 16, 0),
        .write = ad5622_write,
};

static const struct i2c_device_id ad5446_i2c_ids[] = {
        {"ad5301", (kernel_ulong_t)&ad5602_chip_info},
        {"ad5311", (kernel_ulong_t)&ad5612_chip_info},
        {"ad5321", (kernel_ulong_t)&ad5622_chip_info},
        {"ad5602", (kernel_ulong_t)&ad5602_chip_info},
        {"ad5612", (kernel_ulong_t)&ad5612_chip_info},
        {"ad5622", (kernel_ulong_t)&ad5622_chip_info},
        { }
};
MODULE_DEVICE_TABLE(i2c, ad5446_i2c_ids);

static const struct of_device_id ad5446_i2c_of_ids[] = {
        { .compatible = "adi,ad5301", .data = &ad5602_chip_info },
        { .compatible = "adi,ad5311", .data = &ad5612_chip_info },
        { .compatible = "adi,ad5321", .data = &ad5622_chip_info },
        { .compatible = "adi,ad5602", .data = &ad5602_chip_info },
        { .compatible = "adi,ad5612", .data = &ad5612_chip_info },
        { .compatible = "adi,ad5622", .data = &ad5622_chip_info },
        { }
};
MODULE_DEVICE_TABLE(OF, ad5446_i2c_of_ids);

static struct i2c_driver ad5446_i2c_driver = {
        .driver = {
                .name   = "ad5446",
                .of_match_table = ad5446_i2c_of_ids,
        },
        .probe = ad5446_i2c_probe,
        .id_table = ad5446_i2c_ids,
};
module_i2c_driver(ad5446_i2c_driver);

MODULE_AUTHOR("Nuno Sá <nuno.sa@analog.com>");
MODULE_DESCRIPTION("Analog Devices AD5622 and similar I2C DACs");
MODULE_LICENSE("GPL");
MODULE_IMPORT_NS("IIO_AD5446");