#include "FreeRTOS.h"
#include "task.h"
#include "avr/io.h"
#include "zh_avr_160x_i2c.h"
#include "zh_avr_encoder.h"
#include "zh_avr_ac_dimmer.h"
#include "ate0003.h"

#define DEBUG

const char *component_cmm[] = {component_1_cmm, component_2_cmm, component_3_cmm, component_4_cmm, component_5_cmm};
const char *component_line_1[] = {component_1_line_1, component_2_line_1, component_3_line_1, component_4_line_1, component_5_line_1};
const char *component_line_2[] = {component_1_line_2, component_2_line_2, component_3_line_2, component_4_line_2, component_5_line_2};
const uint8_t component_channel[] = {0x0F, 0x04, 0x02, 0x08, 0x0C};

#ifdef DEBUG
#define BAUD_RATE 9600
#define BAUD_PRESCALE (F_CPU / 16 / BAUD_RATE - 1)

int usart(char byte, FILE *stream)
{
    while ((UCSR0A & (1 << UDRE0)) == 0)
    {
    }
    UDR0 = byte;
    return 0;
}
FILE uart = FDEV_SETUP_STREAM(usart, NULL, _FDEV_SETUP_WRITE);
#endif

void system_setup_task(void *pvParameters);
void component_setup_function(uint8_t component);

#ifdef DEBUG
TaskHandle_t system_setup_task_handle = {0};
#endif

zh_avr_pcf8574_handle_t button_handle = {0};
zh_avr_pcf8574_handle_t led1_handle = {0};
zh_avr_pcf8574_handle_t led2_handle = {0};
zh_avr_pcf8574_handle_t relay_handle = {0};
zh_avr_pcf8574_handle_t lcd_handle = {0};

zh_avr_encoder_handle_t power_encoder_handle = {0};
zh_avr_encoder_handle_t component_encoder_handle = {0};

volatile static bool is_work = false; // Work status.
volatile static bool is_dmm = false;  // DMM using status.
volatile static bool is_fix = false;  // FIX button status.
static uint8_t used_channels = 0;     // Permitted channels status. Depends of the selected component.
static bool is_initialized = false;   // Normal loading status.

int main(void)
{
    //** Encoders init **/
    zh_avr_encoder_init_config_t encoder_init_config = ZH_AVR_ENCODER_INIT_CONFIG_DEFAULT();
    encoder_init_config.gpio_port = AVR_PORTC;
    encoder_init_config.a_gpio_number = PORTC0;
    encoder_init_config.b_gpio_number = PORTC1;
    encoder_init_config.encoder_min_value = 0;
    encoder_init_config.encoder_max_value = 100;
    encoder_init_config.encoder_step = 5;
    encoder_init_config.encoder_number = POWER_ENCODER;
    zh_avr_encoder_init(&encoder_init_config, &power_encoder_handle);
    zh_avr_encoder_set(&power_encoder_handle, 100);
    encoder_init_config.a_gpio_number = PORTC2;
    encoder_init_config.b_gpio_number = PORTC3;
    encoder_init_config.encoder_min_value = 0;
    encoder_init_config.encoder_max_value = (sizeof(component_cmm) / sizeof(component_cmm[0])) - 1;
    encoder_init_config.encoder_number = COMPONENT_ENCODER;
    zh_avr_encoder_init(&encoder_init_config, &component_encoder_handle);
    zh_avr_encoder_set(&component_encoder_handle, 0);
    //** AC dimmer init **/
    zh_avr_ac_dimmer_init_config_t ac_dimmer_init_config = ZH_AVR_AC_DIMMER_INIT_CONFIG_DEFAULT();
    ac_dimmer_init_config.ac_dimmer_frequency = ZH_60HZ;
    ac_dimmer_init_config.zero_cross_port = AVR_PORTD;
    ac_dimmer_init_config.zero_cross_gpio = PORTD3;
    ac_dimmer_init_config.triac_port = AVR_PORTD;
    ac_dimmer_init_config.triac_gpio = PORTD4;
    zh_avr_ac_dimmer_init(&ac_dimmer_init_config);
#ifdef DEBUG
    UBRR0H = (BAUD_PRESCALE >> 8);
    UBRR0L = BAUD_PRESCALE;
    UCSR0B = (1 << RXEN0) | (1 << TXEN0);
    UCSR0C = (1 << UCSZ01) | (1 << UCSZ00);
    stdout = &uart;
    xTaskCreate(system_setup_task, "system_setup", 150, NULL, tskIDLE_PRIORITY, &system_setup_task_handle);
#else
    xTaskCreate(system_setup_task, NULL, 150, NULL, tskIDLE_PRIORITY, NULL);
#endif
    vTaskStartScheduler();
    return 0;
}

void system_setup_task(void *pvParameters)
{
    zh_avr_i2c_master_init(false);
    zh_avr_pcf8574_init_config_t pcf8574_init_config = ZH_AVR_PCF8574_INIT_CONFIG_DEFAULT();
    pcf8574_init_config.stack_size = 255;
    //** LED 1 extender init. All LED off. DMM red on. FIX blue on. **/
    pcf8574_init_config.i2c_address = LED1_I2C_ADDRESS;
    zh_avr_pcf8574_init(&pcf8574_init_config, &led1_handle);
    zh_avr_pcf8574_write(&led1_handle, 0xFA);
    //** LED 2 extender init. All LED off. **/
    pcf8574_init_config.i2c_address = LED2_I2C_ADDRESS;
    zh_avr_pcf8574_init(&pcf8574_init_config, &led2_handle);
    zh_avr_pcf8574_write(&led2_handle, 0xFF);
    //** RELAY extender init. All relay off. DMM relay on. **/
    pcf8574_init_config.i2c_address = RELAY_I2C_ADDRESS;
    zh_avr_pcf8574_init(&pcf8574_init_config, &relay_handle);
    zh_avr_pcf8574_write(&relay_handle, 0x02);
    //** LCD init **/
    pcf8574_init_config.i2c_address = LCD_I2C_ADDRESS;
    zh_avr_pcf8574_init(&pcf8574_init_config, &lcd_handle);
    zh_avr_160x_init(&lcd_handle, ZH_LCD_16X4);
    //** BUTTON extender init **/
    pcf8574_init_config.p0_gpio_work_mode = true;
    pcf8574_init_config.p1_gpio_work_mode = true;
    pcf8574_init_config.p2_gpio_work_mode = true;
    pcf8574_init_config.p3_gpio_work_mode = true;
    pcf8574_init_config.p4_gpio_work_mode = true;
    pcf8574_init_config.p5_gpio_work_mode = true;
    pcf8574_init_config.interrupt_port = AVR_PORTD;
    pcf8574_init_config.interrupt_gpio = PORTD2;
    pcf8574_init_config.i2c_address = BUTTON_I2C_ADDRESS;
    zh_avr_pcf8574_init(&pcf8574_init_config, &button_handle);
    //** Loading. Just for fun. **/
    zh_avr_160x_set_cursor(&lcd_handle, 0, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(text_company));
    zh_avr_160x_set_cursor(&lcd_handle, 1, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(text_model));
    zh_avr_160x_set_cursor(&lcd_handle, 2, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(text_firmware));
    zh_avr_160x_set_cursor(&lcd_handle, 3, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(text_loading));
    for (uint8_t i = 0; i <= 100; ++i)
    {
        zh_avr_160x_set_cursor(&lcd_handle, 3, 10);
        zh_avr_160x_print_int(&lcd_handle, i);
        zh_avr_160x_print_char(&lcd_handle, "%");
        vTaskDelay(50 / portTICK_PERIOD_MS);
    }
    //** Initial LCD text init. **/
    zh_avr_160x_lcd_clear(&lcd_handle);
    zh_avr_160x_set_cursor(&lcd_handle, 0, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(text_ac_power));
    zh_avr_160x_set_cursor(&lcd_handle, 0, 10);
    zh_avr_160x_print_int(&lcd_handle, 100);
    zh_avr_160x_print_char(&lcd_handle, "%");
    zh_avr_160x_set_cursor(&lcd_handle, 1, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(text_component));
    //**"ANY" component setup after loading **/
    used_channels = component_channel[0];
    component_setup_function(0);
    is_initialized = true;
    vTaskDelete(NULL);
}

#ifdef DEBUG
void vApplicationStackOverflowHook(TaskHandle_t xTask, char *pcTaskName)
{
    printf("Task %s Remaining Stack Size %d.\n", pcTaskName, uxTaskGetStackHighWaterMark(xTask));
}
#endif

void zh_avr_pcf8574_event_handler(zh_avr_pcf8574_event_on_isr_t *event)
{
    switch (event->gpio_number)
    {
    case DMM_BUTTON:
        if (event->gpio_level == LOW && is_work == false)
        {
            if (is_dmm == true)
            {
                zh_avr_pcf8574_write_gpio(&led1_handle, DMM_LED_RED, LED_ON);
                zh_avr_pcf8574_write_gpio(&led1_handle, DMM_LED_GREEN, LED_OFF);
                zh_avr_pcf8574_write_gpio(&relay_handle, DMM_RELAY, RELAY_ON);
                is_dmm = false;
            }
            else
            {
                zh_avr_pcf8574_write_gpio(&led1_handle, DMM_LED_RED, LED_OFF);
                zh_avr_pcf8574_write_gpio(&led1_handle, DMM_LED_GREEN, LED_ON);
                zh_avr_pcf8574_write_gpio(&relay_handle, DMM_RELAY, RELAY_OFF);
                is_dmm = true;
            }
        }
        break;
    case FIX_BUTTON:
        if (event->gpio_level == LOW)
        {
            zh_avr_pcf8574_write_gpio(&led1_handle, FIX_LED_BLUE, LED_OFF);
            zh_avr_pcf8574_write_gpio(&led1_handle, FIX_LED_GREEN, LED_ON);
            is_fix = true;
        }
        else
        {
            zh_avr_pcf8574_write_gpio(&led1_handle, FIX_LED_BLUE, LED_ON);
            zh_avr_pcf8574_write_gpio(&led1_handle, FIX_LED_GREEN, LED_OFF);
            is_fix = false;
        }
        break;
    case NUM1_BUTTON:
        if ((used_channels & (1 << CHANNEL1)) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM1_LED_BLUE, LED_OFF);
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM1_LED_GREEN, LED_ON);
            zh_avr_pcf8574_write_gpio(&relay_handle, L1_RELAY, RELAY_ON);
        }
        else
        {
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM1_LED_BLUE, LED_ON);
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM1_LED_GREEN, LED_OFF);
            zh_avr_pcf8574_write_gpio(&relay_handle, L1_RELAY, RELAY_OFF);
        }
        break;
    case NUM2_BUTTON:
        if ((used_channels & (1 << CHANNEL2)) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM2_LED_BLUE, LED_OFF);
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM2_LED_GREEN, LED_ON);
            zh_avr_pcf8574_write_gpio(&relay_handle, L2_RELAY, RELAY_ON);
        }
        else
        {
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM2_LED_BLUE, LED_ON);
            zh_avr_pcf8574_write_gpio(&led1_handle, NUM2_LED_GREEN, LED_OFF);
            zh_avr_pcf8574_write_gpio(&relay_handle, L2_RELAY, RELAY_OFF);
        }
        break;
    case NUM3_BUTTON:
        if ((used_channels & (1 << CHANNEL3)) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM3_LED_BLUE, LED_OFF);
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM3_LED_GREEN, LED_ON);
            zh_avr_pcf8574_write_gpio(&relay_handle, L3_RELAY, RELAY_ON);
        }
        else
        {
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM3_LED_BLUE, LED_ON);
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM3_LED_GREEN, LED_OFF);
            zh_avr_pcf8574_write_gpio(&relay_handle, L3_RELAY, RELAY_OFF);
        }
        break;
    case NUM4_BUTTON:
        if ((used_channels & (1 << CHANNEL4)) == 0)
        {
            break;
        }
        if (event->gpio_level == LOW)
        {
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM4_LED_BLUE, LED_OFF);
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM4_LED_GREEN, LED_ON);
            zh_avr_pcf8574_write_gpio(&relay_handle, L4_RELAY, RELAY_ON);
        }
        else
        {
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM4_LED_BLUE, LED_ON);
            zh_avr_pcf8574_write_gpio(&led2_handle, NUM4_LED_GREEN, LED_OFF);
            zh_avr_pcf8574_write_gpio(&relay_handle, L4_RELAY, RELAY_OFF);
        }
        break;
    default:
        break;
    }
#ifdef DEBUG
    printf("Interrupt happened on device address 0x%02X on GPIO number %d at level %d.\n", event->i2c_address, event->gpio_number, event->gpio_level);
#endif
}

void zh_avr_encoder_event_handler(zh_avr_encoder_event_on_isr_t *event)
{
    switch (event->encoder_number)
    {
    case POWER_ENCODER:
        zh_avr_160x_set_cursor(&lcd_handle, 0, 10);
        zh_avr_160x_print_int(&lcd_handle, (uint8_t)event->encoder_position);
        zh_avr_160x_print_char(&lcd_handle, "% ");
        zh_avr_ac_dimmer_set(event->encoder_position);
        break;
    case COMPONENT_ENCODER:
        component_setup_function((uint8_t)event->encoder_position);
        break;
    default:
        break;
    }
#ifdef DEBUG
    printf("Encoder number %d position %0.2f.\n", event->encoder_number, event->encoder_position);
#endif
}

ISR(PCINT1_vect)
{
    if (is_initialized == false)
    {
        return;
    }
    BaseType_t xHigherPriorityTaskWoken = pdFALSE;
    if (zh_avr_encoder_isr_handler(&power_encoder_handle) == pdTRUE)
    {
        xHigherPriorityTaskWoken = pdTRUE;
    }
    if (is_work == false)
    {
        if (zh_avr_encoder_isr_handler(&component_encoder_handle) == pdTRUE)
        {
            xHigherPriorityTaskWoken = pdTRUE;
        }
    }
    if (xHigherPriorityTaskWoken == pdTRUE)
    {
        portYIELD();
    }
}

ISR(PCINT2_vect)
{
    if (is_initialized == false)
    {
        return;
    }
    BaseType_t xHigherPriorityTaskWoken = pdFALSE;
    zh_avr_ac_dimmer_isr_handler();
    if (zh_avr_pcf8574_isr_handler() == pdTRUE)
    {
        xHigherPriorityTaskWoken = pdTRUE;
    }
    if (xHigherPriorityTaskWoken == pdTRUE)
    {
        portYIELD();
    }
}

void component_setup_function(uint8_t component)
{
    zh_avr_160x_set_cursor(&lcd_handle, 1, 10);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(component_cmm[component]));
    zh_avr_160x_set_cursor(&lcd_handle, 2, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(component_line_1[component]));
    zh_avr_160x_set_cursor(&lcd_handle, 3, 0);
    zh_avr_160x_print_char(&lcd_handle, pgm_read_ptr(component_line_2[component]));
    zh_avr_pcf8574_write(&led1_handle, 0x0F);
    zh_avr_pcf8574_write(&led2_handle, 0x0F);
    if ((used_channels & (1 << CHANNEL1)) == CHANNEL1)
    {
        zh_avr_pcf8574_write_gpio(&led1_handle, NUM1_LED_BLUE, LED_ON);
    }
    if ((used_channels & (1 << CHANNEL2)) == CHANNEL2)
    {
        zh_avr_pcf8574_write_gpio(&led1_handle, NUM2_LED_BLUE, LED_ON);
    }
    if ((used_channels & (1 << CHANNEL3)) == CHANNEL3)
    {
        zh_avr_pcf8574_write_gpio(&led2_handle, NUM3_LED_BLUE, LED_ON);
    }
    if ((used_channels & (1 << CHANNEL4)) == CHANNEL4)
    {
        zh_avr_pcf8574_write_gpio(&led2_handle, NUM4_LED_BLUE, LED_ON);
    }
}