#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

// LED 引脚定义
#define LED_A_Pin GPIO_PIN_0
#define LED_B_Pin GPIO_PIN_1
#define LED_C_Pin GPIO_PIN_4
#define LED_D_Pin GPIO_PIN_11
#define LED_Port GPIOA

// 全局变量
int goods[4] = {0, 0, 0, 0}; // A,B,C,D 数量

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

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();
            blink_LED(LED_A_Pin);
            break;

        case KEY_B_INC:
            goods[1]++;
            update_LCD();
            blink_LED(LED_B_Pin);
            break;

        case KEY_C_INC:
            goods[2]++;
            update_LCD();
            blink_LED(LED_C_Pin);
            break;

        case KEY_D_INC:
            goods[3]++;
            update_LCD();
            blink_LED(LED_D_Pin);
            break;

        default:
            break;
        }

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

// === LED 闪烁函数 ===
void blink_LED(uint16_t pin)
{
    HAL_GPIO_WritePin(LED_Port, pin, GPIO_PIN_SET);
    HAL_Delay(100); // 点亮 100ms
    HAL_GPIO_WritePin(LED_Port, pin, GPIO_PIN_RESET);
}

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

    // 第一行显示 A 和 B
    LCD_SetCursor(0, 0);
    sprintf(buf, "A:%d B:%d", goods[0], goods[1]);
    LCD_WriteString(buf);

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

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

    // LED 引脚初始化（推挽输出）
    GPIO_InitStruct.Pin = LED_A_Pin|LED_B_Pin|LED_C_Pin|LED_D_Pin;
    GPIO_InitStruct.Mode = GPIO_MODE_OUTPUT_PP;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_LOW;
    HAL_GPIO_Init(LED_Port, &GPIO_InitStruct);

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

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