#include "bflb_acomp.h"
#include "bflb_core.h"
#include "bflb_gpio.h"
#include "hardware/acomp_reg.h"

#if defined(BL702) || defined(BL602) || defined(BL702L)
#define ACOMP_BASE ((uint32_t)0x4000f000)
#elif defined(BL616) || defined(BL606P) || defined(BL808) || defined(BL628)
#define ACOMP_BASE ((uint32_t)0x2000f000)
#endif

void bflb_acomp_init(uint8_t acomp_id, const struct bflb_acomp_config_s *config)
{
    uint32_t regval;
    uint32_t reg_base;

    if(AON_ACOMP0_ID == acomp_id){
        reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP0_CTRL_OFFSET;
    } else {
        reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP1_CTRL_OFFSET;
    }

    /* disable ACOMPx */
    regval = getreg32(reg_base);
    regval &=~ AON_ACOMP_EN;
    putreg32(regval, reg_base);

    /* set ACOMPx config */
    if(config->mux_en) {
        regval |= AON_ACOMP_MUX_EN;
    } else {
        regval &=~ AON_ACOMP_MUX_EN;
    }
    regval &=~ AON_ACOMP_POS_SEL_MASK;
    regval |= (config->pos_chan_sel << AON_ACOMP_POS_SEL_SHIFT);
    regval &=~ AON_ACOMP_NEG_SEL_MASK;
    regval |= (config->neg_chan_sel << AON_ACOMP_NEG_SEL_SHIFT);
    regval &=~ AON_ACOMP_LEVEL_SEL_MASK;
    regval |= (config->scaling_factor << AON_ACOMP_LEVEL_SEL_SHIFT);
    regval &=~ AON_ACOMP_BIAS_PROG_MASK;
    regval |= (config->bias_prog << AON_ACOMP_BIAS_PROG_SHIFT);
    regval &=~ AON_ACOMP_HYST_SELP_MASK;
    regval |= (config->hysteresis_pos_volt << AON_ACOMP_HYST_SELP_SHIFT);
    regval &=~ AON_ACOMP_HYST_SELN_MASK;
    regval |= (config->hysteresis_neg_volt << AON_ACOMP_HYST_SELN_SHIFT);
    putreg32(regval, reg_base);

    reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP_CFG_OFFSET;
    regval = getreg32(reg_base);
    regval &=~ AON_ACOMP_VREF_SEL_MASK;
    regval |= (config->vio_sel << AON_ACOMP_POS_SEL_SHIFT);
    putreg32(regval, reg_base);
}

void bflb_acomp_enable(uint8_t acomp_id)
{
    uint32_t regval;
    uint32_t reg_base;

    if(AON_ACOMP0_ID == acomp_id){
        reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP0_CTRL_OFFSET;
    } else {
        reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP1_CTRL_OFFSET;
    }

    regval = getreg32(reg_base);
    regval |= AON_ACOMP_EN;
    putreg32(regval, reg_base);
}

void bflb_acomp_disable(uint8_t acomp_id)
{
    uint32_t regval;
    uint32_t reg_base;

    if(AON_ACOMP0_ID == acomp_id){
        reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP0_CTRL_OFFSET;
    } else {
        reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP1_CTRL_OFFSET;
    }

    /* disable ACOMPx */
    regval = getreg32(reg_base);
    regval &=~ AON_ACOMP_EN;
    putreg32(regval, reg_base);
}

uint32_t bflb_acomp_get_result(uint8_t acomp_id)
{
    uint32_t regval;
    uint32_t reg_base;

    reg_base = ACOMP_BASE + AON_ACOMP_REG_ACOMP_CFG_OFFSET;
    regval = getreg32(reg_base);

    if(AON_ACOMP0_ID == acomp_id){
        return (regval & AON_ACOMP0_OUT_RAW_DATA_MASK) >> AON_ACOMP0_OUT_RAW_DATA_SHIFT;
    } else {
        return (regval & AON_ACOMP1_OUT_RAW_DATA_MASK) >> AON_ACOMP1_OUT_RAW_DATA_SHIFT;
    }
}

uint32_t bflb_acomp_gpio_2_chanid(uint32_t pin, uint32_t* channel)
{
#if defined(BL602)
    if ( pin == GPIO_PIN_12 ) {
        *channel = AON_ACOMP_CHAN_ADC0;
    } else if ( pin == GPIO_PIN_4 ) {
        *channel = AON_ACOMP_CHAN_ADC1;
    } else if ( pin == GPIO_PIN_14 ) {
        *channel = AON_ACOMP_CHAN_ADC2;
    } else if ( pin == GPIO_PIN_13 ) {
        *channel = AON_ACOMP_CHAN_ADC3;
    } else if ( pin == GPIO_PIN_5 ) {
        *channel = AON_ACOMP_CHAN_ADC4;
    } else if ( pin == GPIO_PIN_6 ) {
        *channel = AON_ACOMP_CHAN_ADC5;
    } else if ( pin == GPIO_PIN_9 ) {
        *channel = AON_ACOMP_CHAN_ADC6;
    // } else if ( pin == GPIO_PIN_9 ) {
    //     *channel = AON_ACOMP_CHAN_ADC7;
    } else {
        return -1;
    }
#elif defined(BL702)
    if ( pin == GPIO_PIN_8 ) {
        *channel = AON_ACOMP_CHAN_ADC0;
    } else if ( pin == GPIO_PIN_15 ) {
        *channel = AON_ACOMP_CHAN_ADC1;
    } else if ( pin == GPIO_PIN_17 ) {
        *channel = AON_ACOMP_CHAN_ADC2;
    } else if ( pin == GPIO_PIN_11 ) {
        *channel = AON_ACOMP_CHAN_ADC3;
    } else if ( pin == GPIO_PIN_12 ) {
        *channel = AON_ACOMP_CHAN_ADC4;
    } else if ( pin == GPIO_PIN_14 ) {
        *channel = AON_ACOMP_CHAN_ADC5;
    } else if ( pin == GPIO_PIN_7 ) {
        *channel = AON_ACOMP_CHAN_ADC6;
    } else if ( pin == GPIO_PIN_9 ) {
        *channel = AON_ACOMP_CHAN_ADC7;
    } else {
        return -1;
    }
#elif defined(BL702L)
    if ( pin == GPIO_PIN_14 ) {
        *channel = AON_ACOMP_CHAN_ADC0;
    } else if ( pin == GPIO_PIN_15 ) {
        *channel = AON_ACOMP_CHAN_ADC1;
    } else if ( pin == GPIO_PIN_17 ) {
        *channel = AON_ACOMP_CHAN_ADC2;
    } else if ( pin == GPIO_PIN_18 ) {
        *channel = AON_ACOMP_CHAN_ADC3;
    } else if ( pin == GPIO_PIN_19 ) {
        *channel = AON_ACOMP_CHAN_ADC4;
    } else if ( pin == GPIO_PIN_20 ) {
        *channel = AON_ACOMP_CHAN_ADC5;
    } else if ( pin == GPIO_PIN_21 ) {
        *channel = AON_ACOMP_CHAN_ADC6;
    } else if ( pin == GPIO_PIN_7 ) {
        *channel = AON_ACOMP_CHAN_ADC7;
    } else {
        return -1;
    }
#elif defined(BL616)
    if (pin == GPIO_PIN_20) {
        *channel = AON_ACOMP_CHAN_ADC0;
    } else if (pin == GPIO_PIN_19) {
        *channel = AON_ACOMP_CHAN_ADC1;
    } else if (pin == GPIO_PIN_2) {
        *channel = AON_ACOMP_CHAN_ADC2;
    } else if (pin == GPIO_PIN_3) {
        *channel = AON_ACOMP_CHAN_ADC3;
    } else if (pin == GPIO_PIN_14) {
        *channel = AON_ACOMP_CHAN_ADC4;
    } else if (pin == GPIO_PIN_13) {
        *channel = AON_ACOMP_CHAN_ADC5;
    } else if (pin == GPIO_PIN_12) {
        *channel = AON_ACOMP_CHAN_ADC6;
    } else if (pin == GPIO_PIN_10) {
        *channel = AON_ACOMP_CHAN_ADC7;
    } else {
        return -1;
    }
#elif defined(BL606P) || defined(BL808)
    if (pin == GPIO_PIN_17) {
        *channel = AON_ACOMP_CHAN_ADC0;
    } else if (pin == GPIO_PIN_5) {
        *channel = AON_ACOMP_CHAN_ADC1;
    } else if (pin == GPIO_PIN_4) {
        *channel = AON_ACOMP_CHAN_ADC2;
    } else if (pin == GPIO_PIN_11) {
        *channel = AON_ACOMP_CHAN_ADC3;
    } else if (pin == GPIO_PIN_6) {
        *channel = AON_ACOMP_CHAN_ADC4;
    } else if (pin == GPIO_PIN_40) {
        *channel = AON_ACOMP_CHAN_ADC5;
    } else if (pin == GPIO_PIN_12) {
        *channel = AON_ACOMP_CHAN_ADC6;
    } else if (pin == GPIO_PIN_13) {
        *channel = AON_ACOMP_CHAN_ADC7;
    } else {
        return -1;
    }
#else
    return -1;
#endif
    return 0;
}

uint32_t bflb_acomp_chanid_2_gpio(uint32_t channel, uint32_t* pin)
{
#if defined(BL602)
    if ( channel == AON_ACOMP_CHAN_ADC0 ) {
        *pin = GPIO_PIN_12;
    } else if ( channel == AON_ACOMP_CHAN_ADC1 ) {
        *pin = GPIO_PIN_4;
    } else if ( channel == AON_ACOMP_CHAN_ADC2 ) {
        *pin = GPIO_PIN_14;
    } else if ( channel == AON_ACOMP_CHAN_ADC3 ) {
        *pin = GPIO_PIN_13;
    } else if ( channel == AON_ACOMP_CHAN_ADC4 ) {
        *pin = GPIO_PIN_5;
    } else if ( channel == AON_ACOMP_CHAN_ADC5 ) {
        *pin = GPIO_PIN_6;
    } else if ( channel == AON_ACOMP_CHAN_ADC6 ) {
        *pin = GPIO_PIN_9;
    } else if ( channel == AON_ACOMP_CHAN_ADC7 ) {
        *pin = GPIO_PIN_9;
    } else {
        return -1;
    }
#elif defined(BL702)
    if ( channel == AON_ACOMP_CHAN_ADC0 ) {
        *pin = GPIO_PIN_8;
    } else if ( channel == AON_ACOMP_CHAN_ADC1 ) {
        *pin = GPIO_PIN_15;
    } else if ( channel == AON_ACOMP_CHAN_ADC2 ) {
        *pin = GPIO_PIN_17;
    } else if ( channel == AON_ACOMP_CHAN_ADC3 ) {
        *pin = GPIO_PIN_11;
    } else if ( channel == AON_ACOMP_CHAN_ADC4 ) {
        *pin = GPIO_PIN_12;
    } else if ( channel == AON_ACOMP_CHAN_ADC5 ) {
        *pin = GPIO_PIN_14;
    } else if ( channel == AON_ACOMP_CHAN_ADC6 ) {
        *pin = GPIO_PIN_7;
    } else if ( channel == AON_ACOMP_CHAN_ADC7 ) {
        *pin = GPIO_PIN_9;
    } else {
        return -1;
    }
#elif defined(BL702L)
    if ( channel == AON_ACOMP_CHAN_ADC0 ) {
        *pin = GPIO_PIN_14;
    } else if ( channel == AON_ACOMP_CHAN_ADC1 ) {
        *pin = GPIO_PIN_15;
    } else if ( channel == AON_ACOMP_CHAN_ADC2 ) {
        *pin = GPIO_PIN_17;
    } else if ( channel == AON_ACOMP_CHAN_ADC3 ) {
        *pin = GPIO_PIN_18;
    } else if ( channel == AON_ACOMP_CHAN_ADC4 ) {
        *pin = GPIO_PIN_19;
    } else if ( channel == AON_ACOMP_CHAN_ADC5 ) {
        *pin = GPIO_PIN_20;
    } else if ( channel == AON_ACOMP_CHAN_ADC6 ) {
        *pin = GPIO_PIN_21;
    } else if ( channel == AON_ACOMP_CHAN_ADC7 ) {
        *pin = GPIO_PIN_7;
    } else {
        return -1;
    }
#elif defined(BL616)
    if (channel == AON_ACOMP_CHAN_ADC0) {
        *pin = GPIO_PIN_20;
    } else if (channel == AON_ACOMP_CHAN_ADC1) {
        *pin = GPIO_PIN_19;
    } else if (channel == AON_ACOMP_CHAN_ADC2) {
        *pin = GPIO_PIN_2;
    } else if (channel == AON_ACOMP_CHAN_ADC3) {
        *pin = GPIO_PIN_3;
    } else if (channel == AON_ACOMP_CHAN_ADC4) {
        *pin = GPIO_PIN_14;
    } else if (channel == AON_ACOMP_CHAN_ADC5) {
        *pin = GPIO_PIN_13;
    } else if (channel == AON_ACOMP_CHAN_ADC6) {
        *pin = GPIO_PIN_12;
    } else if (channel == AON_ACOMP_CHAN_ADC7) {
        *pin = GPIO_PIN_10;
    } else {
        return -1;
    }
#elif defined(BL606P) || defined(BL808)
    if (channel == AON_ACOMP_CHAN_ADC0) {
        *pin = GPIO_PIN_17;
    } else if (channel == AON_ACOMP_CHAN_ADC1) {
        *pin = GPIO_PIN_5;
    } else if (channel == AON_ACOMP_CHAN_ADC2) {
        *pin = GPIO_PIN_4;
    } else if (channel == AON_ACOMP_CHAN_ADC3) {
        *pin = GPIO_PIN_11;
    } else if (channel == AON_ACOMP_CHAN_ADC4) {
        *pin = GPIO_PIN_6;
    } else if (channel == AON_ACOMP_CHAN_ADC5) {
        *pin = GPIO_PIN_40;
    } else if (channel == AON_ACOMP_CHAN_ADC6) {
        *pin = GPIO_PIN_12;
    } else if (channel == AON_ACOMP_CHAN_ADC7) {
        *pin = GPIO_PIN_13;
    } else {
        return -1;
    }
#else
    return -1;
#endif
    return 0;
}