/*
 * @copyright (c) 2023-2024, MR Development Team
 *
 * @license SPDX-License-Identifier: Apache-2.0
 *
 * @date 2023-11-06    MacRsh       First version
 */

#include "include/device/mr_adc.h"

#ifdef MR_USING_ADC

static int adc_channel_set_configure(struct mr_adc *adc, int channel, struct mr_adc_config config)
{
    struct mr_adc_ops *ops = (struct mr_adc_ops *)adc->dev.drv->ops;

    /* Check channel is valid */
    if (channel < 0 || channel >= 32)
    {
        return MR_EINVAL;
    }

    /* Configure the channel */
    int ret = ops->channel_configure(adc, channel, config.state);
    if (ret < 0)
    {
        return ret;
    }

    /* Enable or disable the channel */
    if (config.state == MR_ENABLE)
    {
        MR_BIT_SET(adc->channel, (1 << channel));
    } else
    {
        MR_BIT_CLR(adc->channel, (1 << channel));
    }
    return MR_EOK;
}

static int adc_channel_get_configure(struct mr_adc *adc, int channel, struct mr_adc_config *config)
{
    /* Check channel is valid */
    if (channel < 0 || channel >= 32)
    {
        return MR_EINVAL;
    }

    /* Get configure */
    config->state = MR_BIT_IS_SET(adc->channel, (1 << channel));
    return MR_EOK;
}

static int mr_adc_open(struct mr_dev *dev)
{
    struct mr_adc *adc = (struct mr_adc *)dev;
    struct mr_adc_ops *ops = (struct mr_adc_ops *)dev->drv->ops;

    return ops->configure(adc, MR_ENABLE);
}

static int mr_adc_close(struct mr_dev *dev)
{
    struct mr_adc *adc = (struct mr_adc *)dev;
    struct mr_adc_ops *ops = (struct mr_adc_ops *)dev->drv->ops;

#ifdef MR_USING_ADC_AUTO_DISABLE
    /* Disable all channels */
    for (size_t i = 0; i < 32; i++)
    {
        if (MR_BIT_IS_SET(adc->channel, (1 << i)) == MR_ENABLE)
        {
            ops->channel_configure(adc, (int)i, MR_DISABLE);
            MR_BIT_CLR(adc->channel, (1 << i));
        }
    }
#endif /* MR_USING_ADC_AUTO_DISABLE */

    return ops->configure(adc, MR_DISABLE);
}

static ssize_t mr_adc_read(struct mr_dev *dev, int off, void *buf, size_t size, int async)
{
    struct mr_adc *adc = (struct mr_adc *)dev;
    struct mr_adc_ops *ops = (struct mr_adc_ops *)dev->drv->ops;
    uint32_t *rd_buf = (uint32_t *)buf;
    ssize_t rd_size;

#ifdef MR_USING_ADC_CHANNEL_CHECK
    /* Check if the channel is enabled */
    if (MR_BIT_IS_SET(adc->channel, (1 << off)) == MR_DISABLE)
    {
        return MR_EINVAL;
    }
#endif /* MR_USING_ADC_CHANNEL_CHECK */

    MR_BIT_CLR(size, sizeof(*rd_buf) - 1);
    for (rd_size = 0; rd_size < size; rd_size += sizeof(*rd_buf))
    {
        *rd_buf = ops->read(adc, off);
        rd_buf++;
    }
    return rd_size;
}

static ssize_t mr_adc_ioctl(struct mr_dev *dev, int off, int cmd, void *args)
{
    struct mr_adc *adc = (struct mr_adc *)dev;

    switch (cmd)
    {
        case MR_CTL_ADC_SET_CHANNEL_CONFIG:
        {
            if (args != MR_NULL)
            {
                struct mr_adc_config config = *((struct mr_adc_config *)args);

                int ret = adc_channel_set_configure(adc, off, config);
                if (ret < 0)
                {
                    return ret;
                }
                return sizeof(config);
            }
            return MR_EINVAL;
        }
        case MR_CTL_ADC_GET_CHANNEL_CONFIG:
        {
            if (args != MR_NULL)
            {
                struct mr_adc_config *config = (struct mr_adc_config *)args;

                int ret = adc_channel_get_configure(adc, off, config);
                if (ret < 0)
                {
                    return ret;
                }
                return sizeof(*config);
            }
        }
        default:
        {
            return MR_ENOTSUP;
        }
    }
}

/**
 * @brief This function registers an adc.
 *
 * @param adc The adc.
 * @param name The name of the adc.
 * @param drv The driver of the adc.
 *
 * @return MR_EOK on success, otherwise an error code.
 */
int mr_adc_register(struct mr_adc *adc, const char *name, struct mr_drv *drv)
{
    static struct mr_dev_ops ops =
        {
            mr_adc_open,
            mr_adc_close,
            mr_adc_read,
            MR_NULL,
            mr_adc_ioctl,
            MR_NULL
        };

    MR_ASSERT(adc != MR_NULL);
    MR_ASSERT(name != MR_NULL);
    MR_ASSERT(drv != MR_NULL);
    MR_ASSERT(drv->ops != MR_NULL);

    /* Initialize the fields */
    adc->channel = 0;

    /* Register the adc */
    return mr_dev_register(&adc->dev, name, Mr_Dev_Type_ADC, MR_SFLAG_RDONLY, &ops, drv);
}

#endif /* MR_USING_ADC */