/*
 * Copyright (c) 2023 Juan Manuel Cruz <jcruz@fi.uba.ar> <jcruz@frba.utn.edu.ar>.
 * Copyright (c) 2026 Martin Javier Klöckner <mklockner@fi.uba.ar>
 *
 * See file `LICENSE` for full details
 */

// TODO: *_interface libraries should already include *_attribute headers

#include "main.h"

#include "logger.h"
#include "dwt.h"

#include "app.h"
#include "task_system.h"
#include "task_buttons.h"
#include "task_buzzer.h"
#include "task_leds.h"
#include "task_menu.h"
#include "task_temp_ctrl.h"
#include "task_temp_sensor.h"
#include "task_remote.h"

#define G_APP_CNT_INI           0ul
#define G_APP_TICK_CNT_INI      0ul
#define G_APP_WCET_INI          0ul
#define G_APP_WCET_MEAN_SAMPLES 1000 // 1 second update rate

#define TASK_X_WCET_INI            0ul
#define TASK_X_WCET_SAMPLE_TIME_MS 10000ul
#define TASK_X_TICK_INI            0ul
#define TASK_X_DELAY_MIN           0ul

typedef struct {
    void (*task_init)(void *);
    void (*task_update)(void *);
    void *parameters;
} task_cfg_t;

typedef struct {
    uint32_t WCET;          // Task worst-case execution time in microseconds
    uint32_t WCET_interval; // WCET in last `TASK_X_WCET_SAMPLE_TIME_MS` ms
} task_dta_t;

shared_data_type shared_data;
uint32_t g_app_runtime_us;
uint32_t g_app_runtime_us_mean;     // Mean of g_app_runtime_us in
                                    // G_APP_WCET_MEAN_SAMPLES
uint32_t g_app_runtime_us_mean_aux;
uint32_t g_app_runtime_us_mean_cnt;
uint32_t g_app_WCET_us;             // Worst case execution time historically
uint32_t g_app_WCET_sample_us;      // Worst case execution time in sample time
volatile uint32_t g_app_tick_cnt;

const task_cfg_t task_cfg_list[] = {
    {task_system_init,      task_system_update,      &shared_data},
    {task_buttons_init,     task_buttons_update,     NULL},
    {task_buzzer_init,      task_buzzer_update,      NULL},
    {task_leds_init,        task_leds_update,        NULL},
    {task_menu_init,        task_menu_update,        &shared_data},
    {task_temp_ctrl_init,   task_temp_ctrl_update,   &shared_data},
    {task_temp_sensor_init, task_temp_sensor_update, &shared_data},
    {task_remote_init,      task_remote_update,      &shared_data},
};

#define TASK_QTY (sizeof(task_cfg_list)/sizeof(task_cfg_t))

task_dta_t task_dta_list[TASK_QTY] = {0};

void app_init(void)
{
    LOGGER_INFO("Initializing `app`...");

    cycle_counter_init();

    shared_data_type *p_shared_data = &shared_data;

    p_shared_data->temp_set_point    = 0;
    p_shared_data->temp_sensor       = -20;
    p_shared_data->temp_ctrl_pwm_dc  = 0;
    p_shared_data->temp_ctrl_enabled = false;
    p_shared_data->timer             = NULL;

    for (uint32_t index = 0; TASK_QTY > index; index++)
    {
        (*task_cfg_list[index].task_init)(task_cfg_list[index].parameters);
        task_dta_list[index].WCET = TASK_X_WCET_INI;
    }

    __asm("CPSID i");
    g_app_tick_cnt              = G_APP_TICK_CNT_INI;
    g_task_system_tick_cnt      = G_APP_TICK_CNT_INI;
    g_task_buttons_tick_cnt     = G_APP_TICK_CNT_INI;
    g_task_leds_tick_cnt        = G_APP_TICK_CNT_INI;
    g_task_menu_tick_cnt        = G_APP_TICK_CNT_INI;
    g_task_temp_ctrl_tick_cnt   = G_APP_TICK_CNT_INI;
    g_task_temp_sensor_tick_cnt = G_APP_TICK_CNT_INI;
    g_task_remote_tick_cnt      = G_APP_TICK_CNT_INI;
    __asm("CPSIE i");

    LOGGER_INFO("Done initializing `app`");
    g_app_WCET_us = G_APP_WCET_INI;
    g_app_runtime_us_mean = g_app_runtime_us_mean_aux = G_APP_WCET_INI;
    g_app_runtime_us_mean_cnt = 0;
}

void app_update(void)
{
    uint32_t index;
    bool b_time_update_required = false;
    uint32_t cycle_counter_time_us;

    __asm("CPSID i");
    if (G_APP_TICK_CNT_INI < g_app_tick_cnt)
    {
        g_app_tick_cnt--;
        b_time_update_required = true;
    }
    __asm("CPSIE i");

    while (b_time_update_required)
    {
        g_app_runtime_us = 0;

        for (index = 0; TASK_QTY > index; index++)
        {
            cycle_counter_reset();

            if (0 == g_app_WCET_sample_us)
            {
                g_app_WCET_sample_us = TASK_X_WCET_SAMPLE_TIME_MS;
                task_dta_list[index].WCET_interval = 0;
            }
            else
            {
                g_app_WCET_sample_us--;
            }

            (*task_cfg_list[index].task_update)(task_cfg_list[index].parameters);

            cycle_counter_time_us = cycle_counter_get_time_us();
            g_app_runtime_us += cycle_counter_time_us;

            if (g_app_WCET_us < g_app_runtime_us) {
                g_app_WCET_us = g_app_runtime_us;
                LOGGER_INFO("App worst execution time (WCET): %lu us", g_app_WCET_us);
            }

            if (task_dta_list[index].WCET < cycle_counter_time_us)
            {
                task_dta_list[index].WCET = cycle_counter_time_us;
            }

            if (task_dta_list[index].WCET_interval< cycle_counter_time_us)
            {
                task_dta_list[index].WCET_interval = cycle_counter_time_us;
            }
        }

        if (G_APP_WCET_MEAN_SAMPLES == g_app_runtime_us_mean_cnt)
        {
            g_app_runtime_us_mean_cnt = 0;
            g_app_runtime_us_mean = g_app_runtime_us_mean_aux / G_APP_WCET_MEAN_SAMPLES;
            g_app_runtime_us_mean_aux = G_APP_WCET_INI;
        }
        else
        {
            g_app_runtime_us_mean_cnt++;
            g_app_runtime_us_mean_aux += g_app_runtime_us;
        }

        __asm("CPSID i");
        if (G_APP_TICK_CNT_INI < g_app_tick_cnt)
        {
            g_app_tick_cnt--;
            b_time_update_required = true;
        }
        else
        {
            b_time_update_required = false;
        }
        __asm("CPSIE i");
    }
}

// Callbacks definition

// This is called every ms
void HAL_SYSTICK_Callback(void)
{
    g_app_tick_cnt++;

    g_task_system_tick_cnt++;
    g_task_buttons_tick_cnt++;
    g_task_buzzer_tick_cnt++;
    g_task_leds_tick_cnt++;
    g_task_menu_tick_cnt++;
    g_task_temp_ctrl_tick_cnt++;
    g_task_temp_sensor_tick_cnt++;
    g_task_remote_tick_cnt++;
}

void HAL_UART_TxCpltCallback(UART_HandleTypeDef *huart)
{
    if (huart->Instance == USART1)
    {
        b_ds18b20_dma_tx_done = true;
    }
}

void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart)
{
    if (huart->Instance == USART1)
    {
        b_ds18b20_dma_rx_done = true;
    }
    if (huart->Instance == USART3)
    {
        task_remote_queue_enqueue(&task_remote_rx_queue, task_remote_uart_rx_byte);
        HAL_UART_Receive_IT(&huart3, &task_remote_uart_rx_byte, 1);
    }
}

void HAL_UARTEx_RxEventCallback(UART_HandleTypeDef *huart, uint16_t size)
{
    if (huart->Instance == USART1)
    {
        b_ds18b20_dma_rx_done = true;
    }
}

void HAL_UART_ErrorCallback(UART_HandleTypeDef *huart)
{
    if(huart->Instance == USART1)
    {
        ;
    }
}