#include "stm32c0xx_hal.h"
#include "spi_ili9341.h"
#include <stdlib.h>
#include <stdbool.h>

#define User_Button_Pin GPIO_PIN_13
#define User_Button_GPIO_Port GPIOC
#define User_Button_EXTI_IRQn EXTI4_15_IRQn
#define Led_Pin GPIO_PIN_5
#define Led_GPIO_Port GPIOA

I2C_HandleTypeDef hi2c1;
SPI_HandleTypeDef hspi1;
UART_HandleTypeDef huart2;

void SystemClock_Config(void);
static void MX_GPIO_Init(void);
static void MX_I2C1_Init(void);
static void MX_USART2_UART_Init(void);
static void MX_SPI1_Init(void);

#define NUM_RELAYS 32
#define DEVICE_PADDING 5
#define DEVICE_WIDTH 70
#define DEVICE_HEIGHT 50
#define DEVICE_PER_ROW 4
#define COLOR_RED 0xF800 // Example color value for red
#define COLOR_GREEN 0x07E0 // Example color value for green

// Define GPIO pins for relay monitoring
int RELAY_PINS[] = {GPIO_PIN_0, GPIO_PIN_1 /* Add pins for all 32 relays */};
GPIO_TypeDef* GPIO_PORT[NUM_RELAYS] = {GPIOA, GPIOA, /* Add GPIO ports for all 32 relays */};

bool relayStatus[NUM_RELAYS] = {0}; // Array to store relay status (false: low, true: high)
char* machineNames[NUM_RELAYS] = {
  "Mach1", "Mach2", "Mach3", "Mach4", "Mach5", "Mach6", "Mach7", "Mach8",
  "Mach9", "Mach10", "Mach11", "Mach12", "Mach13", "Mach14", "Mach15", "Mach16",
  "Mach17", "Mach18", "Mach19", "Mach20", "Mach21", "Mach22", "Mach23", "Mach24",
  "Mach25", "Mach26", "Mach27", "Mach28", "Mach29", "Mach30", "Mach31", "Mach32"
};

void ControlPin_Init() {
    GPIO_InitTypeDef GPIO_InitStruct = {0};
    HAL_GPIO_WritePin(GPIOA, GPIO_PIN_9 | GPIO_PIN_6 | GPIO_PIN_15, GPIO_PIN_RESET);
    GPIO_InitStruct.Pin = GPIO_PIN_9 | GPIO_PIN_6 | GPIO_PIN_15;
    GPIO_InitStruct.Mode = GPIO_MODE_OUTPUT_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_VERY_HIGH;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);
}

void HAL_GPIO_EXTI_Callback(uint16_t GPIO_Pin) {
    // Handle GPIO interrupt for relay state change
    // Update relayStatus array based on the GPIO_Pin
}

// Function prototypes
void updateHMI(void);
void drawDevice(int x, int y, bool isHigh, char* name);

void drawDevice(int x, int y, bool isHigh, char* name) {
    // Draw device box
    uint16_t color = (isHigh) ? COLOR_GREEN : COLOR_RED;
    TFT9341_FillRect(x, y, DEVICE_WIDTH, DEVICE_HEIGHT, color);
    TFT9341_DrawRect(x, y, DEVICE_WIDTH, DEVICE_HEIGHT, TFT9341_WHITE);

    // Draw status text
    TFT9341_SetTextColor(TFT9341_WHITE);
    TFT9341_SetFont(&Font24);
    TFT9341_String(x + 5, y + DEVICE_HEIGHT / 2 - 4, name);
}

void updateHMI() {
    // Update the HMI display based on relayStatus array
    TFT9341_FillScreen(TFT9341_BLACK); // Clear the screen

    // Draw the header
    TFT9341_SetTextColor(TFT9341_WHITE);
    TFT9341_SetFont(&Font20);
    TFT9341_String(10, 10, "Relay Status");

    // Update the HMI display to show the state of 32 devices
    int x = DEVICE_PADDING;
    int y = 50;
    for (int i = 0; i < NUM_RELAYS; i++) {
        drawDevice(x, y, relayStatus[i], machineNames[i]);
        x += DEVICE_WIDTH + DEVICE_PADDING;
        if ((i + 1) % DEVICE_PER_ROW == 0) {
            x = DEVICE_PADDING;
            y += DEVICE_HEIGHT + DEVICE_PADDING;
        }
    }
}

int main(void) {
    HAL_Init();
    SystemClock_Config();
    MX_GPIO_Init();
    MX_USART2_UART_Init();
    MX_SPI1_Init();
    ControlPin_Init();

    TFT9341_ini(240, 320);
    HAL_Delay(100);
    TFT9341_SetRotation(1);
    TFT9341_SetFont(&Font20);

    while (1) {
        // Main loop
        // Monitor relay states and update relayStatus array
        updateHMI(); // Update the HMI display
        HAL_Delay(1000); // Adjust delay as needed
    }
}

// Function to initialize GPIO pins for relay monitoring
void MX_GPIO_Init(void) {
    // Initialize GPIO pins for relay monitoring
    GPIO_InitTypeDef GPIO_InitStruct = {0};
    for (int i = 0; i < NUM_RELAYS; i++) {
        GPIO_InitStruct.Pin = RELAY_PINS[i];
        GPIO_InitStruct.Mode = GPIO_MODE_IT_RISING_FALLING;
        GPIO_InitStruct.Pull = GPIO_NOPULL;
        HAL_GPIO_Init(GPIO_PORT[i], &GPIO_InitStruct);
    }
}

void SystemClock_Config(void) {
    RCC_OscInitTypeDef RCC_OscInitStruct = {0};
    RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};

    RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSI;
    RCC_OscInitStruct.HSIState = RCC_HSI_ON;
    RCC_OscInitStruct.HSIDiv = RCC_HSI_DIV1;
    RCC_OscInitStruct.HSICalibrationValue = RCC_HSICALIBRATION_DEFAULT;
    if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK) {
        Error_Handler();
    }

    RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK | RCC_CLOCKTYPE_SYSCLK | RCC_CLOCKTYPE_PCLK1;
    RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_HSI;
    RCC_ClkInitStruct.SYSCLKDivider = RCC_SYSCLK_DIV1;
    RCC_ClkInitStruct.AHBCLKDivider = RCC_HCLK_DIV1;
    RCC_ClkInitStruct.APB1CLKDivider = RCC_APB1_DIV1;

    if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_1) != HAL_OK) {
        Error_Handler();
    }
}

static void MX_I2C1_Init(void) {
    hi2c1.Instance = I2C1;
    hi2c1.Init.Timing = 0x20303E5D;
    hi2c1.Init.OwnAddress1 = 0;
    hi2c1.Init.AddressingMode = I2C_ADDRESSINGMODE_7BIT;
    hi2c1.Init.DualAddressMode = I2C_DUALADDRESS_DISABLE;
    hi2c1.Init.OwnAddress2 = 0;
    hi2c1.Init.OwnAddress2Masks = I2C_OA2_NOMASK;
    hi2c1.Init.GeneralCallMode = I2C_GENERALCALL_DISABLE;
    hi2c1.Init.NoStretchMode = I2C_NOSTRETCH_DISABLE;
    if (HAL_I2C_Init(&hi2c1) != HAL_OK) {
        Error_Handler();
    }

    if (HAL_I2CEx_ConfigAnalogFilter(&hi2c1, I2C_ANALOGFILTER_ENABLE) != HAL_OK) {
        Error_Handler();
    }

    if (HAL_I2CEx_ConfigDigitalFilter(&hi2c1, 0) != HAL_OK) {
        Error_Handler();
    }
}

static void MX_SPI1_Init(void) {
    GPIO_InitTypeDef GPIO_InitStruct = {0};
    RCC_PeriphCLKInitTypeDef PeriphClkInit = {0};

    PeriphClkInit.PeriphClockSelection = RCC_PERIPHCLK_I2S1;
    PeriphClkInit.I2s1ClockSelection = RCC_I2S1CLKSOURCE_SYSCLK;
    if (HAL_RCCEx_PeriphCLKConfig(&PeriphClkInit) != HAL_OK) {
        Error_Handler();
    }

    __HAL_RCC_SPI1_CLK_ENABLE();
    __HAL_RCC_GPIOA_CLK_ENABLE();
    GPIO_InitStruct.Pin = GPIO_PIN_1 | GPIO_PIN_7;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_LOW;
    GPIO_InitStruct.Alternate = GPIO_AF0_SPI1;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    hspi1.Instance = SPI1;
    hspi1.Init.Mode = SPI_MODE_MASTER;
    hspi1.Init.Direction = SPI_DIRECTION_2LINES;
    hspi1.Init.DataSize = SPI_DATASIZE_8BIT;
    hspi1.Init.CLKPolarity = SPI_POLARITY_LOW;
    hspi1.Init.CLKPhase = SPI_PHASE_1EDGE;
    hspi1.Init.NSS = SPI_NSS_SOFT;
    hspi1.Init.BaudRatePrescaler = SPI_BAUDRATEPRESCALER_4;
    hspi1.Init.FirstBit = SPI_FIRSTBIT_MSB;
    hspi1.Init.TIMode = SPI_TIMODE_DISABLE;
    hspi1.Init.CRCCalculation = SPI_CRCCALCULATION_DISABLE;
    hspi1.Init.CRCPolynomial = 10;
    hspi1.Init.CRCLength = SPI_CRC_LENGTH_DATASIZE;
    hspi1.Init.NSSPMode = SPI_NSS_PULSE_ENABLE;
    if (HAL_SPI_Init(&hspi1) != HAL_OK) {
        Error_Handler();
    }
}

static void MX_USART2_UART_Init(void) {
    huart2.Instance = USART2;
    huart2.Init.BaudRate = 115200;
    huart2.Init.WordLength = UART_WORDLENGTH_8B;
    huart2.Init.StopBits = UART_STOPBITS_1;
    huart2.Init.Parity = UART_PARITY_NONE;
    huart2.Init.Mode = UART_MODE_TX_RX;
    huart2.Init.HwFlowCtl = UART_HWCONTROL_NONE;
    huart2.Init.OverSampling = UART_OVERSAMPLING_16;
    huart2.Init.OneBitSampling = UART_ONE_BIT_SAMPLE_DISABLE;
    huart2.Init.ClockPrescaler = UART_PRESCALER_DIV1;
    huart2.AdvancedInit.AdvFeatureInit = UART_ADVFEATURE_NO_INIT;
    if (HAL_UART_Init(&huart2) != HAL_OK) {
        Error_Handler();
    }
}

void Error_Handler(void) {
    __disable_irq();
    while (1) {
    }
}

#ifdef USE_FULL_ASSERT
void assert_failed(uint8_t *file, uint32_t line) {
    /* User can add his own implementation to report the HAL error return state */
}
#endif