#include "uart.h"
#include "gd32e23x_usart.h"
#include "gd32e23x_rcu.h"
#include "gd32e23x_gpio.h"
#include "board_config.h"
#include "uart_ring_buffer.h"

void rs485_init(void) {

    #ifndef RS485_MAX13487
    /* 使能 GPIOA 和 USART0 时钟 */
    rcu_periph_clock_enable(RS485_GPIO_RCU);
    rcu_periph_clock_enable(RS485_RCU);

    /* 配置 PA2 为 USART0_TX，PA3 为 USART0_RX */
    gpio_af_set(RS485_GPIO_PORT, GPIO_AF_1, RS485_TX_PIN | RS485_RX_PIN | RS485_EN_PIN);

    gpio_mode_set(RS485_GPIO_PORT, GPIO_MODE_AF, GPIO_PUPD_PULLUP, RS485_TX_PIN | RS485_RX_PIN);
    gpio_output_options_set(RS485_GPIO_PORT, GPIO_OTYPE_PP, GPIO_OSPEED_50MHZ, RS485_TX_PIN | RS485_RX_PIN);

    gpio_mode_set(RS485_GPIO_PORT, GPIO_MODE_AF, GPIO_PUPD_NONE, RS485_EN_PIN);
    gpio_output_options_set(RS485_GPIO_PORT, GPIO_OTYPE_PP, GPIO_OSPEED_50MHZ, RS485_EN_PIN);

    /* 配置波特率、数据位、停止位等 */
    usart_deinit(RS485_PHY);
    usart_word_length_set(RS485_PHY, USART_WL_8BIT);
    usart_stop_bit_set(RS485_PHY, USART_STB_1BIT);
    usart_parity_config(RS485_PHY, USART_PM_NONE);
    usart_baudrate_set(RS485_PHY, RS485_BAUDRATE);
    usart_receive_config(RS485_PHY, USART_RECEIVE_ENABLE);
    usart_transmit_config(RS485_PHY, USART_TRANSMIT_ENABLE);

    usart_driver_assertime_config(RS485_PHY, 0x01);
    usart_driver_deassertime_config(RS485_PHY, 0x10);

    usart_rs485_driver_enable(RS485_PHY);

    usart_enable(RS485_PHY);

    nvic_irq_enable(RS485_IRQ, 0);
    usart_interrupt_enable(RS485_PHY, USART_INT_RBNE);
    // usart_interrupt_enable(RS485_PHY, USART_INT_IDLE);

    #else
    rcu_periph_clock_enable(RS485_GPIO_RCU);
    rcu_periph_clock_enable(RS485_RCU);

    gpio_af_set(RS485_GPIO_PORT, GPIO_AF_1, GPIO_PIN_2 | GPIO_PIN_3);

    /* configure USART Tx&Rx as alternate function push-pull */
    gpio_mode_set(RS485_GPIO_PORT, GPIO_MODE_AF, GPIO_PUPD_PULLUP, RS485_TX_PIN | RS485_RX_PIN);
    gpio_output_options_set(RS485_GPIO_PORT, GPIO_OTYPE_PP, GPIO_OSPEED_10MHZ, RS485_TX_PIN | RS485_RX_PIN);

    /* configure RS485 EN Pin */
    gpio_mode_set(RS485_GPIO_PORT, GPIO_MODE_OUTPUT, GPIO_PUPD_NONE, RS485_EN_PIN);
    gpio_output_options_set(RS485_GPIO_PORT, GPIO_OTYPE_PP, GPIO_OSPEED_50MHZ, RS485_EN_PIN);
    gpio_bit_write(RS485_GPIO_PORT, RS485_EN_PIN, SET);

    /* USART configure */
    usart_deinit(RS485_PHY);
    usart_baudrate_set(RS485_PHY, RS485_BAUDRATE);
    usart_receive_config(RS485_PHY, USART_RECEIVE_ENABLE);
    usart_transmit_config(RS485_PHY, USART_TRANSMIT_ENABLE);

    usart_enable(RS485_PHY);

    nvic_irq_enable(USART0_IRQn, 0);
    usart_interrupt_enable(RS485_PHY, USART_INT_RBNE);
    usart_interrupt_enable(RS485_PHY, USART_INT_IDLE);

    #endif // RS485_MAX13487
    }

/******************************************************************************/
/* 具体的中断处理函数实现 */
/******************************************************************************/

void usart0_irq_handler(void) {
    // 处理USART0的接收中断
    if(usart_interrupt_flag_get(USART0, USART_INT_FLAG_RBNE)) {
        uint8_t data = usart_data_receive(USART0);
        // 使用原有的环形缓冲区处理逻辑
        (void)uart_ring_buffer_put(data); // 缓冲满时丢弃，返回值可用于统计
    }
    
    // 处理USART0的空闲中断
    if(usart_interrupt_flag_get(USART0, USART_INT_FLAG_IDLE)) {
        usart_interrupt_flag_clear(USART0, USART_INT_FLAG_IDLE);
        // 在这里添加空闲中断处理逻辑
    }
}

void usart1_irq_handler(void) {
    // 处理USART1的接收中断
    if(usart_interrupt_flag_get(USART1, USART_INT_FLAG_RBNE)) {
        uint8_t data = usart_data_receive(USART1);
        // 使用原有的环形缓冲区处理逻辑
        (void)uart_ring_buffer_put(data); // 缓冲满时丢弃，返回值可用于统计
    }
    
    // 处理USART1的空闲中断
    if(usart_interrupt_flag_get(USART1, USART_INT_FLAG_IDLE)) {
        usart_interrupt_flag_clear(USART1, USART_INT_FLAG_IDLE);
        // 在这里添加空闲中断处理逻辑
    }
}
