#include "stm32c0xx_hal.h"
#include <string.h>
#include <stdio.h>
#include "lcd.h"
#include "keypad.h"

// LCD 引脚定义
#define RS_Pin GPIO_PIN_0
#define E_Pin  GPIO_PIN_1
#define D4_Pin GPIO_PIN_2
#define D5_Pin GPIO_PIN_3
#define D6_Pin GPIO_PIN_4
#define D7_Pin GPIO_PIN_5
#define LCD_Port GPIOB

// 全局变量
int goods[4] = {0, 0, 0, 0};   // A,B,C,D 数量
int selected = -1;               // 当前选择货物编号 (0~3)
int blink_state = 1;            // 闪烁状态，1=显示，0=隐藏
uint32_t last_blink = 0;        // 上次闪烁时间（ms）

// 函数声明
void SystemClock_Config(void);
static void MX_GPIO_Init(void);
void update_LCD(void);

int main(void) {
    HAL_Init();
    SystemClock_Config();
    MX_GPIO_Init();

    LCD_Init();
    update_LCD();

    while (1) {
        uint8_t key = keypad_get_key();

        switch (key) {
            case KEY_A_INC: 
                goods[0]++; 
                update_LCD(); 
                break;
            case KEY_B_INC: 
                goods[1]++; 
                update_LCD(); 
                break;
            case KEY_C_INC: 
                goods[2]++; 
                update_LCD(); 
                break;
            case KEY_D_INC: 
                goods[3]++; 
                update_LCD(); 
                break;

            case KEY_SELECT: 
                selected = (selected + 1) % 5; 
                update_LCD();
                break;

            case KEY_INC:
                goods[selected]++; 
                update_LCD();
                break;

            case KEY_DEC:
                if (goods[selected] > 0) goods[selected]--;
                update_LCD();
                break;

            case KEY_RESET:
                goods[selected] = 0;
                update_LCD();
                break;

            default:
                break;
        }

        //只有选中有效时闪烁数字
        if (selected >= 0 && selected <= 3) {
            if (HAL_GetTick() - last_blink >= 500) {
                blink_state = !blink_state;
                last_blink = HAL_GetTick();
                update_LCD();
            }
        } else {
            // 未选中时数字常显，若隐藏则恢复显示
            if (!blink_state) {
                blink_state = 1;
                update_LCD();
            }
        }

        HAL_Delay(50); // 扫描间隔
    }
}

// === LCD 更新显示函数 ===
void update_LCD(void) {
    char buf[17];

    // 第一行显示 A 和 B
    LCD_SetCursor(0, 0);
    LCD_WriteString("A:");
    if (selected == 0) {
        if (blink_state) {
            sprintf(buf, "%d", goods[0]);
            LCD_WriteString(buf);
        } else {
            LCD_WriteString(" "); // 数字闪烁隐藏，假设最多两位数
        }
    } else {
        sprintf(buf, "%d", goods[0]);
        LCD_WriteString(buf);
    }

    LCD_WriteString(" B:");
    if (selected == 1) {
        if (blink_state) {
            sprintf(buf, "%d", goods[1]);
            LCD_WriteString(buf);
        } else {
            LCD_WriteString(" ");
        }
    } else {
        sprintf(buf, "%d", goods[1]);
        LCD_WriteString(buf);
    }

    // 第二行显示 C 和 D
    LCD_SetCursor(1, 0);
    LCD_WriteString("C:");
    if (selected == 2) {
        if (blink_state) {
            sprintf(buf, "%d", goods[2]);
            LCD_WriteString(buf);
        } else {
            LCD_WriteString(" ");
        }
    } else {
        sprintf(buf, "%d", goods[2]);
        LCD_WriteString(buf);
    }

    LCD_WriteString(" D:");
    if (selected == 3) {
        if (blink_state) {
            sprintf(buf, "%d", goods[3]);
            LCD_WriteString(buf);
        } else {
            LCD_WriteString(" ");
        }
    } else {
        sprintf(buf, "%d", goods[3]);
        LCD_WriteString(buf);
    }

    // 右下角显示当前选中货物字母，不闪烁
    LCD_SetCursor(1, 12);
    if (selected >= 0 && selected <= 3) {
        LCD_WriteChar('A' + selected);
    } else {
        LCD_WriteChar(' ');
    }
}

// 时钟配置，使用内部HSI时钟
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;
    HAL_RCC_OscConfig(&RCC_OscInitStruct);

    RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK | RCC_CLOCKTYPE_SYSCLK;
    RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_HSI;
    RCC_ClkInitStruct.AHBCLKDivider = RCC_SYSCLK_DIV1;
    RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV1;
    HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_0);
}

// GPIO 初始化，LCD 和 按键引脚
static void MX_GPIO_Init(void) {
    __HAL_RCC_GPIOB_CLK_ENABLE();

    GPIO_InitTypeDef GPIO_InitStruct = {0};

    // LCD 引脚初始化（推挽输出）
    GPIO_InitStruct.Pin = RS_Pin|E_Pin|D4_Pin|D5_Pin|D6_Pin|D7_Pin;
    GPIO_InitStruct.Mode = GPIO_MODE_OUTPUT_PP;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_LOW;
    HAL_GPIO_Init(LCD_Port, &GPIO_InitStruct);

    // 按键引脚初始化（输入上拉）
    keypad_init();
}

// 错误处理函数（死循环）
void Error_Handler(void) {
    while (1) {}
}