#include "system.h"
#include "rgb.h"
#include "ui.h"
#include "clock.h"
#include "../Drivers/lcd.h"
#include "../Drivers/rf.h"
#include "../Drivers/adc.h"
#include "../Drivers/pcf8563.h"
#include "intrins.h"

// 引用 main.c �的全�唤醒与运行状态变�
extern volatile u8 p0_wakeup_flag;
extern volatile u8 p2_wakeup_flag;
extern volatile u8 p3_wakeup_flag;
extern volatile u8 rf_wakeup_flag;

/**
 * @brief 系统 GPIO �口模式初始化
 * @details 配置各引脚的推挽输出、高阻输入及准双向模式，建立系统初�电气环境�
 */
void GPIO_Init(void)
{
    // 使能访问扩展特殊功能寄存� XSFR
    EAXFR = 1;

    // 1. 配置 PWR_HOLD (P0.0) 为推挽输出模� (P0M1 bit0=0, P0M0 bit0=1)
    P0M1 &= ~(1 << 0);
    P0M0 |= (1 << 0);
    PWR_HOLD = 1; // 上电��时间拉高，锁存电源保持供�

    // 2. 配置 KEY_UP (P0.1), KEY_CONFIRM (P0.2) 为高阻输�/带上拉准双向
    P0M1 &= ~((1 << 1) | (1 << 2));
    P0M0 &= ~((1 << 1) | (1 << 2));
    KEY_UP = 1;
    KEY_CONFIRM = 1;

    // 3. 初�化 STC32G �� ADC 模块 (其中�动配� P0.3 � P1.3 为高阻输入模�)
    ADC_Init();

    // 4. 配置 PCF8563 I2C 管脚 P3.2 (SCL) � P3.3 (SDA) 为准双向口模式，并开上拉
    P3M1 &= ~((1 << 2) | (1 << 3));
    P3M0 &= ~((1 << 2) | (1 << 3));
    P3PU |= (1 << 2) | (1 << 3); // �� P3.2 � P3.3 的内� 4.7K 上拉电阻
    RTC_SCL = 1;
    RTC_SDA = 1;

    // 5. 配置推挽输出引脚:
    // MOTOR (P2.5), RGB_DIN (P2.3) -> P2M1 bit5,3=0; P2M0 bit5,3=1
    P2M1 &= ~((1 << 5) | (1 << 3));
    P2M0 |= ((1 << 5) | (1 << 3));
    MOTOR = 0;
    RGB_DIN = 0;    // P2.3 初�拉低，防�上电期间 WS2812B �触发

    // 配置 KEY_SOS (P2.6) 为带上拉准双向口模式以确保输入稳�
    P2M1 &= ~(1 << 6);
    P2M0 &= ~(1 << 6);
    KEY_SOS = 1;
    (*(unsigned char volatile xdata *)0xFE12) |= (1 << 6); // 使能 P2.6 的强内部上拉电阻 (P2PU = 0xFE12)

    // SGM_CTRL (P1.7), LCD 控制管脚 (P1.0, P1.1, P1.4, P1.5, P1.6) �为推挽输�
    P1M1 &= ~((1 << 7) | (1 << 0) | (1 << 1) | (1 << 4) | (1 << 5) | (1 << 6));
    P1M0 |= ((1 << 7) | (1 << 0) | (1 << 1) | (1 << 4) | (1 << 5) | (1 << 6));

    // SHUT (P3.5) 射�/外�开�使能脚置为推挽输�
    P3M1 &= ~(1 << 5);
    P3M0 |= (1 << 5);
    SHUT = 0; // 低电平开�工作

    // 7. 配置 CHRG_DET (P1.2) 为高阻输入模式，适配 DET 模拟 ADC 通道 2
    //    由于�精密模拟电平�测，切勿��引脚内部上拉电阻
    P1M1 |= (1 << 2);
    P1M0 &= ~(1 << 2);
    // P1PU 在扩� XSFR 空间�没有�定义的话�以用 P1 �口寄存器或�是关闭上拉�
    // STC32G ��口上拉寄存器� P1PU (FE11H)，因为是 XSFR 我们� config.h 里有 extern 或直接操�
    // 扩展 XSFR P1PU 地址� 0xFE11
    (*(unsigned char volatile xdata *)0xFE11) &= ~(1 << 2); 
}

/**
 * @brief 初�化 UART1 串口通信
 * @details 设定波特� 115200 (使用 Timer2 作为发生��24.0MHz 晶振)
 */
void Uart1_Init(void)
{
    SCON = 0x50;  // 8 位数�, �变波特率
    AUXR |= 0x01; // 串口 1 选择 Timer2 做波特率发生�
    AUXR |= 0x04; // Timer2 时钟� 1T 模式
    T2L = 0xCC;   // 24MHz / 115200 = 208 (0xD0), 对应 reload � 65536 - 52 = 0xFFCC
    T2H = 0xFF;
    AUXR |= 0x10; // �� Timer2 运�
}

/**
 * @brief 串口 1 发�单字节数据
 * @param dat 待发送的字节
 */
void Uart_SendByte(u8 dat)
{
    SBUF = dat;
    while (!TI);
    TI = 0;
}

/**
 * @brief 串口 1 发�字符串
 * @param s 字�串指针
 */
void Uart_SendString(char *s)
{
    while (*s) {
        Uart_SendByte(*s++);
    }
}

/**
 * @brief 串口 1 非阻塞接收字�
 * @return char 接收到的字�，若无接收数据则返� '\0'
 */
char Uart_RxChar(void)
{
    if (RI) {
        RI = 0;
        return SBUF;
    }
    return '\0';
}

/**
 * @brief 阻��秒延�
 * @param ms �秒数
 */
void Delay_ms(u16 ms)
{
    u16 i, j;
    for (i = 0; i < ms; i++) {
        for (j = 0; j < 2000; j++) {
            _nop_();
        }
    }
}

/**
 * @brief 阻�微秒延�
 * @param us �秒数
 */
void Delay_us(u16 us)
{
    u16 i;
    for (i = 0; i < us; i++) {
        _nop_(); _nop_(); _nop_(); _nop_();
    }
}

/**
 * @brief 系统�机上电硬件自�函数
 * @details �� PWR_HOLD 锁存状��ADC 采样基准与串口��，输出完整自� report
 */
void Self_Test(void)
{
    u16 key_down_adc;
    u16 vbat_mv;
    u16 batt_adc_raw;
    u8 percent;

    Uart_SendString("\r\n========================================\r\n");
    Uart_SendString("[SELF-TEST] System Power-On Self Test...\r\n");

    // 执� I2C 总线器件�描，查找 RTC �否物理响�
    PCF8563_ScanDiagnostic();

    // 0. I2C �件脚电平诊断测试 (验证引脚�控�与��)
    Uart_SendString("[I2C-DIAG] Pin Level Test Starting...\r\n");
    // � 1 测试 (准双�+上拉)
    P3M1 &= ~((1 << 2) | (1 << 3));
    P3M0 &= ~((1 << 2) | (1 << 3));
    P3PU |= (1 << 2) | (1 << 3);
    RTC_SCL = 1;
    RTC_SDA = 1;
    Delay_ms(10);
    Uart_SendString("[I2C-DIAG] Write SCL=1, SDA=1 -> Real Level: SCL=");
    Uart_SendByte((u8)(RTC_SCL ? '1' : '0'));
    Uart_SendString(", SDA=");
    Uart_SendByte((u8)(RTC_SDA ? '1' : '0'));
    Uart_SendString("\r\n");

    // � 0 测试
    RTC_SCL = 0;
    RTC_SDA = 0;
    Delay_ms(10);
    Uart_SendString("[I2C-DIAG] Write SCL=0, SDA=0 -> Real Level: SCL=");
    Uart_SendByte((u8)(RTC_SCL ? '1' : '0'));
    Uart_SendString(", SDA=");
    Uart_SendByte((u8)(RTC_SDA ? '1' : '0'));
    Uart_SendString("\r\n");

    // 0.2 执� RTC 底层单独试�测�
    PCF8563_TryReadDiagnostic();

    // 1. 验证 PWR_HOLD 锁存状�
    if (PWR_HOLD == 1) {
        Uart_SendString("[SELF-TEST] PWR_HOLD Pin State: HIGH [OK]\r\n");
    } else {
        Uart_SendString("[SELF-TEST] PWR_HOLD Pin State: LOW [FAIL]\r\n");
    }

    // 2. �� P0.3 ADC 按键基准常�电�
    key_down_adc = ADC_ReadFiltered(ADC_CHANNEL_KEY_DOWN);
    Uart_SendString("[SELF-TEST] KEY_DOWN (P0.3) ADC Raw Value: ");
    Uart_SendByte((u8)('0' + (key_down_adc / 1000) % 10));
    Uart_SendByte((u8)('0' + (key_down_adc / 100) % 10));
    Uart_SendByte((u8)('0' + (key_down_adc / 10) % 10));
    Uart_SendByte((u8)('0' + (key_down_adc % 10)));
    Uart_SendString(" [OK]\r\n");

    // 2.2 �� KEY_SOS (P2.6) 物理电平常�
    Uart_SendString("[SELF-TEST] KEY_SOS (P2.6) Pin Level: ");
    Uart_SendByte((u8)(KEY_SOS ? '1' : '0'));
    Uart_SendString("\r\n");

    // 3. �� P1.3 电池电压�测与百分�
    vbat_mv = ADC_GetBatteryVoltage_mV();
    percent = ADC_GetBatteryPercent();
    batt_adc_raw = ADC_ReadFiltered(ADC_CHANNEL_BATTERY);

    Uart_SendString("[SELF-TEST] Battery Raw ADC: ");
    Uart_SendByte((u8)('0' + (batt_adc_raw / 1000) % 10));
    Uart_SendByte((u8)('0' + (batt_adc_raw / 100) % 10));
    Uart_SendByte((u8)('0' + (batt_adc_raw / 10) % 10));
    Uart_SendByte((u8)('0' + (batt_adc_raw % 10)));

    Uart_SendString(" -> Voltage: ");
    Uart_SendByte((u8)('0' + (vbat_mv / 1000) % 10));
    Uart_SendByte('.');
    Uart_SendByte((u8)('0' + (vbat_mv / 100) % 10));
    Uart_SendByte((u8)('0' + (vbat_mv / 10) % 10));

    Uart_SendString(" V, Capacity: ");
    Uart_SendByte((u8)('0' + (percent / 100) % 10));
    Uart_SendByte((u8)('0' + (percent / 10) % 10));
    Uart_SendByte((u8)('0' + (percent % 10)));
    Uart_SendString("% [OK]\r\n");

    // 4. �查充电状�
    {
        u16 det_adc = ADC_ReadFiltered(ADC_CHANNEL_CHRG_DET);
        u16 det_mv = (det_adc * 3300UL) / 4095UL;
        Uart_SendString("[SELF-TEST] CHRG_DET (P1.2) ADC: ");
        Uart_SendByte((u8)('0' + (det_adc / 1000) % 10));
        Uart_SendByte((u8)('0' + (det_adc / 100) % 10));
        Uart_SendByte((u8)('0' + (det_adc / 10) % 10));
        Uart_SendByte((u8)('0' + (det_adc % 10)));
        Uart_SendString(" -> Voltage: ");
        Uart_SendByte((u8)('0' + (det_mv / 1000) % 10));
        Uart_SendByte('.');
        Uart_SendByte((u8)('0' + (det_mv / 100) % 10));
        Uart_SendByte((u8)('0' + (det_mv / 10) % 10));
        Uart_SendString(" V -> ");
        if (det_mv >= 800 && det_mv <= 1800) {
            Uart_SendString("CHARGING [OK]\r\n");
        } else if (det_mv >= 2200 && det_mv <= 2800) {
            Uart_SendString("FULLY CHARGED [OK]\r\n");
        } else {
            Uart_SendString("NOT CHARGING [OK]\r\n");
        }
    }

    Uart_SendString("[SELF-TEST] Self Test Completed Successfully!\r\n");
    Uart_SendString("========================================\r\n\r\n");
}

/**
 * @brief 执�关机断电流�
 * @details 关断外��震动提示并拉低 PWR_HOLD 引脚切断总电�
 */
void Power_Off(void)
{
    Uart_SendString("[SYS] Executing Power Off sequence...\r\n");

    // �达震动反� 100ms
    MOTOR = 1;
    Delay_ms(100);
    MOTOR = 0;

    // 关断屏幕背光� RGB ��
    SGM_CTRL = 1;   // 高电平断电，关闭背光��
    RGB_Send(0, 0, 0, 0, 0, 0);

    // 拉低 PWR_HOLD 切断电源锁存使�电源关�
    PWR_HOLD = 0;

    // 进入死循�等待电源物理切断
    while (1) {
        PWR_HOLD = 0;
    }
}

/**
 * @brief 系统切入低功耗停机休眠模�
 */
void Enter_Low_Power_Sleep(void)
{
    Uart_SendString("[SYS] Entering low power sleep...\r\n");

    // 1. 关闭/熄灭 RGB �
    RGB_Send(0, 0, 0, 0, 0, 0);

    // 2. � LCD �片写� Display OFF 指令 (0x28) 关显�
    WriteComm(0x28);
    Delay_ms(20);

    // 3. � LCD �片写� Sleep In 指令 (0x10) 关内部电荷泵
    WriteComm(0x10);
    Delay_ms(20);

    // 4. 关闭背光（高电平�电，关闭背光�屏）
    SGM_CTRL = 1;

    // 5. 屏幕接口防漏电�理
    LCD_CS = 1;
    LCD_RST = 1;
    LCD_DCX = 0;
    LCD_SCL = 0;
    LCD_SDI = 0;

    // 6. 射��置为关�接收模式（低功�，SHUT=0, RF_SetMode(0)�
    RF_SetMode(0);

    // 使能访问扩展特殊功能寄存�
    P_SW2 |= 0x80;

    // 7. 配置 Port0 ��唤醒� (KEY_UP=P0.1, KEY_CONFIRM=P0.2, KEY_DOWN=P0.3)
    P0IM1 |= 0x0E;
    P0IM0 &= ~0x0E; // 下降沿触�
    P0INTF = 0x00;  // 清除挂起的中�标志
    P0INTE |= 0x0E; // �� P0.1, P0.2, P0.3 ��允�
    P0WKUE |= 0x0E; // 使能 P0.1, P0.2, P0.3 掉电唤醒

    // 8. 配置 Port2 ��唤醒� (SOS=P2.6)
    P2IM1 |= 0x40;
    P2IM0 &= ~0x40; // 下降沿触�
    P2INTF = 0x00;  // 清除挂起标志
    P2INTE |= 0x40; // �� P2.6 ��允�
    P2WKUE |= 0x40; // 使能 P2.6 掉电唤醒

    // 9. 配置 Port3 ��唤醒� (RF_RX_DATA=P3.6)
    P3IM1 |= 0x40;
    P3IM0 |= 0x40;  // 双边沿触�
    P3INTF = 0x00;  // 清除挂起标志
    P3INTE |= 0x40; // �� P3.6 ��允�
    P3WKUE |= 0x40; // 使能 P3.6 掉电唤醒

    // �保引脚上拉和输入使能，防止因�空产生的功�泄�
    P2PU |= 0x40;
    P2IE |= 0x40;
    P3PU &= ~0x40;
    P3IE |= 0x40;

    // 清零各个唤醒�测标志位
    P0INTF = 0x00;
    P2INTF = 0x00;
    P3INTF = 0x00;

    p0_wakeup_flag = 0;
    p2_wakeup_flag = 0;
    p3_wakeup_flag = 0;
    rf_wakeup_flag = 0;

    // 10. 关闭系统滴答定时� Timer1 ��，防�休眠时�定时器�唤醒
    ET1 = 0;
    EA = 1; // ��全局��

    // 配置 PCON 挂起单片机，切入停机 (Power Down) 模式
    PCON |= 0x02; // PD = 1
    _nop_(); _nop_(); _nop_(); _nop_();

    // ====== �外部��唤醒后，在��继�运� ======
    Wakeup_Restore();
}

/**
 * @brief 休眠唤醒恢�
 */
void Wakeup_Restore(void)
{
    // ���ò����㰴������״̬�����������ϵ�/����˲̬����������³����ػ�
    extern volatile u16 key_up_hold;
    extern volatile u16 key_down_hold;
    extern volatile u16 key_confirm_hold;
    extern volatile u16 key_sos_hold;
    extern volatile u16 comb_hold;

    key_up_hold = 0;
    key_down_hold = 0;
    key_confirm_hold = 0;
    key_sos_hold = 0;
    comb_hold = 0;

    EAXFR = 1;
    // 1. 立即关闭�有掉电�部��允�与唤醒允�，防�重�
    P0INTE = 0x00;
    P2INTE = 0x00;
    P3INTE = 0x00;
    P0WKUE = 0x00;
    P2WKUE = 0x00;
    P3WKUE = 0x00;
    P0INTF = 0x00;
    P2INTF = 0x00;
    P3INTF = 0x00;

    // 2. 清零掉电唤醒定时�
    WKTCH = 0x00;
    WKTCL = 0x00;

    // 3. ��全局��并重�系统滴答定时�
    // 3. �жϿ���״̬�Ƴ�����Ļ��ʼ��֮����

    // 4. 恢�射频芯片工作模�
    RF_SetMode(1);

    // 打印唤醒调试日志
    if ((p0_wakeup_flag & 0x0E) || (p2_wakeup_flag & 0x40)) {
        Uart_SendString("[WR] Woken by KEY (P0=");
        Uart_SendHex8(p0_wakeup_flag);
        Uart_SendString(", P2=");
        Uart_SendHex8(p2_wakeup_flag);
        Uart_SendString(")\r\n");
    } else if ((p3_wakeup_flag & 0x40) || rf_wakeup_flag) {
        rf_wakeup_flag = 1;
        Uart_SendString("[WR] Woken by RF (P3.6)!\r\n");
    } else {
        Uart_SendString("[WR] Woken by TIMER/Other\r\n");
    }

    // 5. 恢�屏幕供电与背光（低电平上电，恢复背光亮屏）
    SGM_CTRL = 0;

    // 6. 延时 200ms 等待液晶稳压板上电稳定后，重新� LCD 寄存器初始化
    Delay_ms(200);
    LCD_Init();

    // ���¿���ȫ���ж��� Timer1 �δ�ʱ���жϣ���ȷ���ڵ�ѹ�ȶ���ſ�ʼ��������
    EA = 1;
    ET1 = 1;
    // 7. 充零闲置计时�
    inactivity_timer = 0;
}

/**
 * @brief 安全获取 Timer0 计数器� (防高字节滚动跳变)
 */
u16 GetTimer0_Safe(void)
{
    u8 high1, high2, low;
    do {
        high1 = TH0;
        low   = TL0;
        high2 = TH0;
    } while (high1 != high2);
    return (((u16)high1 << 8) | low);
}

/**
 * @brief 串口 1 发� 4 位十�进制字� (0x0~0xF)
 */
void Uart_SendHex4(u8 val)
{
    val &= 0x0F;
    if (val < 10) {
        Uart_SendByte('0' + val);
    } else {
        Uart_SendByte('A' + (val - 10));
    }
}

/**
 * @brief 串口 1 发� 8 位十�进制 (0x00~0xFF)
 */
void Uart_SendHex8(u8 val)
{
    Uart_SendHex4(val >> 4);
    Uart_SendHex4(val);
}

/**
 * @brief 串口 1 发� 16 位十�进制 (0x0000~0xFFFF)
 */
void Uart_SendHex16(u16 val)
{
    Uart_SendHex8((u8)(val >> 8));
    Uart_SendHex8((u8)val);
}

/**
 * @brief 串口 1 发� 20 位十�进制 (用于射� 5 � ID 打印)
 */
void Uart_SendHex20(u32 val)
{
    Uart_SendHex4((u8)(val >> 16));
    Uart_SendHex16((u16)val);
}

/**
 * @brief 串口 1 发� 32 位十�进制
 */
void Uart_SendHex32(u32 val)
{
    Uart_SendHex16((u16)(val >> 16));
    Uart_SendHex16((u16)val);
}