#include "cpg10_dev_mode.h"
#include "tw1508.h"

#include "nrf_gpio.h"
#include "nrf_log.h"

#if (DEF_ELITE_MODEL == DEF_CURRENT_PULSE_GANERATOR_10)

// The GPIO corresponding to the pin
const uint32_t pin_to_gpio_table[] = {
    [6]  = NRF_GPIO_PIN_MAP(0, 22),
    [8]  = NRF_GPIO_PIN_MAP(0, 25),
    [9]  = NRF_GPIO_PIN_MAP(0, 19),
    [10] = NRF_GPIO_PIN_MAP(0, 21),
    [11] = NRF_GPIO_PIN_MAP(1, 00),
    [12] = NRF_GPIO_PIN_MAP(0, 18),
    [13] = NRF_GPIO_PIN_MAP(0, 17),
    [14] = NRF_GPIO_PIN_MAP(0, 20),
    [16] = NRF_GPIO_PIN_MAP(0, 14),
    [17] = NRF_GPIO_PIN_MAP(0, 13),
    [18] = NRF_GPIO_PIN_MAP(0, 11),
    [20] = NRF_GPIO_PIN_MAP(0, 15),
    [25] = NRF_GPIO_PIN_MAP(1, 8),
    [26] = NRF_GPIO_PIN_MAP(0, 12),
    [27] = NRF_GPIO_PIN_MAP(0, 7),
    [28] = NRF_GPIO_PIN_MAP(1, 9),
    [29] = NRF_GPIO_PIN_MAP(0, 8),
    [30] = NRF_GPIO_PIN_MAP(0, 6),
    [31] = NRF_GPIO_PIN_MAP(0, 5),
    [32] = NRF_GPIO_PIN_MAP(0, 27),
    [33] = NRF_GPIO_PIN_MAP(0, 26),
    [34] = NRF_GPIO_PIN_MAP(0, 4),
    [36] = NRF_GPIO_PIN_MAP(0, 1),
    [37] = NRF_GPIO_PIN_MAP(0, 29),
    [38] = NRF_GPIO_PIN_MAP(0, 0),
    [39] = NRF_GPIO_PIN_MAP(0, 31),
    [40] = NRF_GPIO_PIN_MAP(1, 15),
    [41] = NRF_GPIO_PIN_MAP(0, 2),
    [42] = NRF_GPIO_PIN_MAP(0, 30),
    [43] = NRF_GPIO_PIN_MAP(0, 28),
    [44] = NRF_GPIO_PIN_MAP(1, 12),
    [45] = NRF_GPIO_PIN_MAP(1, 14),
    [46] = NRF_GPIO_PIN_MAP(0, 3),
    [47] = NRF_GPIO_PIN_MAP(1, 13),
    [48] = NRF_GPIO_PIN_MAP(1, 3),
    [49] = NRF_GPIO_PIN_MAP(1, 10),
    [50] = NRF_GPIO_PIN_MAP(1, 6),
    [51] = NRF_GPIO_PIN_MAP(1, 11),
    [52] = NRF_GPIO_PIN_MAP(0, 10),
    [53] = NRF_GPIO_PIN_MAP(0, 9),
    [59] = NRF_GPIO_PIN_MAP(1, 2),
    [60] = NRF_GPIO_PIN_MAP(0, 24),
    [61] = NRF_GPIO_PIN_MAP(0, 23),
    [62] = NRF_GPIO_PIN_MAP(0, 16),
};

static uint32_t bmd380pins_convert_to_gpio(uint32_t pin)
{
    uint32_t gpio;

    switch (pin)
    {
        case 6:
        case 8:
        case 9:
        case 10:
        case 11:
        case 12:
        case 13:
        case 14:
        case 16:
        case 17:
        case 18:
        case 20:
        case 25:
        case 26:
        case 27:
        case 28:
        case 29:
        case 30:
        case 31:
        case 32:
        case 33:
        case 34:
        case 36:
        case 37:
        case 38:
        case 39:
        case 40:
        case 41:
        case 42:
        case 43:
        case 44:
        case 45:
        case 46:
        case 47:
        case 48:
        case 49:
        case 50:
        case 51:
        case 52:
        case 53:
        case 59:
        case 60:
        case 61:
        case 62:
            gpio = pin_to_gpio_table[pin];
            break;
        default:
            gpio = UNDEF_GPIO;
            NRF_LOG_INFO("UNDEF_GPIO: pin %d can't convert to gpio number", pin);
            break;
    }

    return gpio;
}

static void set_bmd380_pin_signal(uint32_t pin, uint32_t high_low)
{
    uint32_t gpio = bmd380pins_convert_to_gpio(pin);

    if (gpio != UNDEF_GPIO)
    {
        nrf_gpio_pin_write(gpio, high_low);
        NRF_LOG_INFO("set pin %02d (gpio %02d) = %d", pin, gpio, high_low);
    }
}

#define ELECTRODES_A1HB1L 1
#define ELECTRODES_A1LB1H 2
#define ELECTRODES_A2HB2L 3
#define ELECTRODES_A2LB2H 4
#define ELECTRODES_A3HB3L 5
#define ELECTRODES_A3LB3H 6
#define ELECTRODES_A4HB4L 7
#define ELECTRODES_A4LB4H 8
static void cpg10_electrodes(uint32_t electrodes_mode)
{
    switch (electrodes_mode)
    {
        case ELECTRODES_A1HB1L:
            NRF_LOG_INFO("ELECTRODES_A1HB1L()");
            nrf_gpio_pin_write(VA1H_PIN, 1);
            nrf_gpio_pin_write(VB1L_PIN, 1);
            nrf_gpio_pin_write(VB1H_PIN, 0);
            nrf_gpio_pin_write(VA1L_PIN, 0);
            break;

        case ELECTRODES_A1LB1H:
            NRF_LOG_INFO("ELECTRODES_A1LB1H()");
            nrf_gpio_pin_write(VB1H_PIN, 1);
            nrf_gpio_pin_write(VA1L_PIN, 1);
            nrf_gpio_pin_write(VA1H_PIN, 0);
            nrf_gpio_pin_write(VB1L_PIN, 0);
            break;

        case ELECTRODES_A2HB2L:
            NRF_LOG_INFO("ELECTRODES_A2HB2L()");
            nrf_gpio_pin_write(VA2H_PIN, 1);
            nrf_gpio_pin_write(VB2L_PIN, 1);
            nrf_gpio_pin_write(VB2H_PIN, 0);
            nrf_gpio_pin_write(VA2L_PIN, 0);
            break;

        case ELECTRODES_A2LB2H:
            NRF_LOG_INFO("ELECTRODES_A2LB2H()");
            nrf_gpio_pin_write(VB2H_PIN, 1);
            nrf_gpio_pin_write(VA2L_PIN, 1);
            nrf_gpio_pin_write(VA2H_PIN, 0);
            nrf_gpio_pin_write(VB2L_PIN, 0);
            break;

        case ELECTRODES_A3HB3L:
            NRF_LOG_INFO("ELECTRODES_A3HB3L()");
            nrf_gpio_pin_write(VA3H_PIN, 1);
            nrf_gpio_pin_write(VB3L_PIN, 1);
            nrf_gpio_pin_write(VB3H_PIN, 0);
            nrf_gpio_pin_write(VA3L_PIN, 0);
            break;

        case ELECTRODES_A3LB3H:
            NRF_LOG_INFO("ELECTRODES_A3LB3H()");
            nrf_gpio_pin_write(VB3H_PIN, 1);
            nrf_gpio_pin_write(VA3L_PIN, 1);
            nrf_gpio_pin_write(VA3H_PIN, 0);
            nrf_gpio_pin_write(VB3L_PIN, 0);
            break;

        case ELECTRODES_A4HB4L:
            NRF_LOG_INFO("ELECTRODES_A4HB4L()");
            nrf_gpio_pin_write(VA4H_PIN, 1);
            nrf_gpio_pin_write(VB4L_PIN, 1);
            nrf_gpio_pin_write(VB4H_PIN, 0);
            nrf_gpio_pin_write(VA4L_PIN, 0);
            break;

        case ELECTRODES_A4LB4H:
            NRF_LOG_INFO("ELECTRODES_A4LB4H()");
            nrf_gpio_pin_write(VB4H_PIN, 1);
            nrf_gpio_pin_write(VA4L_PIN, 1);
            nrf_gpio_pin_write(VA4H_PIN, 0);
            nrf_gpio_pin_write(VB4L_PIN, 0);
            break;
    }
}

/*
    dev_mode_set_cpg10_electrodes
    (1)0x3000FF0001
            -func: ELECTRODES_A1HB1L

    (2)0x3000FF0002
            -func: ELECTRODES_A1LB1H

    (3)0x3000FF0003
            -func: ELECTRODES_A2HB2L

    (4)0x3000FF0004
            -func: ELECTRODES_A2LB2H

    (5)0x3000FF0005
            -func: ELECTRODES_A3HB3L

    (6)0x3000FF0006
            -func: ELECTRODES_A3LB3H

    (7)0x3000FF0007
            -func: ELECTRODES_A4HB4L

    (8)0x3000FF0008
            -func: ELECTRODES_A4LB4H
*/
void dev_mode_set_cpg10_electrodes(uint8_t *ins)
{
    struct __PACKED
    {
        uint8_t id     : 4;
        uint8_t        : 4;
        uint16_t magic : 16;
        uint8_t dev_opcode;
        uint8_t electrodes_opcode;
    } *p_ins = (void *)ins;

    cpg10_electrodes(p_ins->electrodes_opcode);
}

/*
    dev_mode_set_cpg10_tw1508
    (1)0x3000FF0100
            -func: tw1805_init()

    (2)0x3000FF0101aaaabbbb
            -func: tw1508_set()
            -aaaa: out_0 value     0~1023[LSB]
            -bbbb: out_1 value     0~1023[LSB]
*/
void dev_mode_set_cpg10_tw1508(uint8_t *ins)
{
    struct __PACKED
    {
        uint8_t id     : 4;
        uint8_t        : 4;
        uint16_t magic : 16;
        uint8_t dev_opcode;
        uint8_t tw1508_opcode;
        uint8_t param[];
    } *p_ins = (void *)ins;

    switch (p_ins->tw1508_opcode)
    {
        case 0x00: {
            tw1805_init();
            NRF_LOG_INFO("tw1805_init()");
            break;
        }

        case 0x01: {
            uint16_t out_0;
            uint16_t out_1;
            memcpy(&out_0, &p_ins->param[0], sizeof(out_0));
            memcpy(&out_1, &p_ins->param[2], sizeof(out_1));
            NRF_LOG_INFO("tw1508_set(%d, %d)", out_0, out_1);
            tw1508_set(out_0, out_1);
            break;
        }
    }
}

/*
    dev_mode_gpio_function
    (1)0x3000FFA000ppss
            -func: set_bmd380_pin_signal
            -pp: pin number     06h-3Fh
                06h: P0.22_GPIO
                08h: P0.25_GPIO
                ......
                3Eh: P0.16_GPIO
            -ss: signal         00h-01h
                00h: low
                01h: high

    (2)0x3000FFA001ss
        -func: set BMD380 all pin to high/low, except for HV_EN, SPI, and INPUT pins
            -ss: signal         00h-01h
                00h: low
                01h: high
*/
void dev_mode_gpio_function(uint8_t *ins)
{
    struct __PACKED
    {
        uint8_t id     : 4;
        uint8_t        : 4;
        uint16_t magic : 16;
        uint8_t dev_opcode;
        uint8_t gpio_function_opcode;
        uint8_t param[];
    } *p_ins = (void *)ins;

    switch (p_ins->gpio_function_opcode)
    {
        case 0x00: {
            uint32_t pin      = p_ins->param[0];
            uint32_t high_low = p_ins->param[1];
            set_bmd380_pin_signal(pin, high_low);
            break;
        }

        case 0x01: {
            uint32_t high_low = p_ins->param[0];
            uint32_t set_pin[] = {6, 8, 9, 10, 13, 14, 16, 17, 26, 27, 
                                    28, 30, 32, 33, 34, 36, 37, 38, 39, 40,
                                    41, 44, 45, 46, 47, 48, 49, 50, 51, 52, 
                                    53, 59, 60, 61, 62};

            for (int i=0; i<sizeof(set_pin) / sizeof(set_pin[0]); i++)
            {
                set_bmd380_pin_signal(set_pin[i], high_low);
            }
            break;
        }
    }
}

#endif