/*
 * 模块名称：RS-485 半双工通信驱动
 * 模块功能：RS-485 半双工通信驱动实现，核心要点：
 *          1. 发送前拉高 DE（切换到发送态）
 *          2. HAL_UART_Transmit 内部已等�?TC（发送完成）标志
 *          3. 发送完毕后拉低 DE（切回接收态）
 *          4. 使用 HAL_UARTEx_ReceiveToIdle_IT 实现变长帧接�? *          参考：ST AN3070 应用笔记、controllerstech.com RS485 教程
 * 适用平台：STM32F4 系列
 * 作者：王建�? * 创建日期�?026-07-18
 * 修改记录�? *     2026-07-18  王建�? 创建初始版本
 */

#include "rs485.h"

/*
 * 内部辅助�?- 方向控制
 */
#define RS485_DIR_TX(h)  HAL_GPIO_WritePin((h)->dir_port, (h)->dir_pin, GPIO_PIN_SET)
#define RS485_DIR_RX(h)  HAL_GPIO_WritePin((h)->dir_port, (h)->dir_pin, GPIO_PIN_RESET)

/*
 * 公共函数实现
 */

/*
 * 函数功能：初始化 RS-485 句柄
 * 入口参数：handle   - RS-485 句柄指针
 *           huart    - UART 外设句柄
 *           dir_port - DE 引脚 GPIO 端口
 *           dir_pin  - DE 引脚 GPIO 编号
 * 返回值：�? * 限定条件：UART �?GPIO 须先�?CubeMX 完成初始�? * 函数说明：初始化后立即将 DE 拉低，使 RS-485 收发器处于接收状�? */
void rs485_init(rs485_handle_t *p_handle,
                UART_HandleTypeDef *p_huart,
                GPIO_TypeDef *p_dir_port,
                uint16_t dir_pin) {
    p_handle->huart    = p_huart;
    p_handle->dir_port = p_dir_port;
    p_handle->dir_pin  = dir_pin;
    p_handle->rx_size  = 0;

    /* 默认进入接收状�?*/
    RS485_DIR_RX(p_handle);
}

/*
 * 函数功能：阻塞方式发送数�? * 入口参数：handle  - RS-485 句柄指针
 *           data    - 待发送数据缓冲区
 *           len     - 待发送字节数
 *           timeout - 发送超时（ms），�?0 使用默认�? * 返回值：HAL_OK / HAL_TIMEOUT / HAL_ERROR
 * 限定条件：须在主循环�?RTOS 任务中调用，不可�?UART 中断中调�? * 函数说明�? *   时序：DE=HIGH �?UART 发�?N 字节 �?�?TC �?DE=LOW
 *   HAL_UART_Transmit 内部会等�?TXE（发送寄存器空）逐字节搬数据�? *   最后还会等�?TC（发送完成）标志，确保最后一字节的停止位已移出�? *   因此返回后再拉低 DE 是安全的
 */
HAL_StatusTypeDef rs485_transmit(rs485_handle_t *p_handle,
                                  const uint8_t *p_data,
                                  uint16_t len,
                                  uint32_t timeout) {
    HAL_StatusTypeDef status = HAL_OK;

    if (timeout == 0U) {
        timeout = RS485_TIMEOUT_DEFAULT;
    }

    /* 1. 切换到发送状�?*/
    RS485_DIR_TX(p_handle);

    /* 2. 阻塞发送（内部等待 TC�?*/
    status = HAL_UART_Transmit(p_handle->huart,
                                (uint8_t *)p_data,
                                len,
                                timeout);

    /* 3. 发送完毕或超时，切回接收状�?*/
    RS485_DIR_RX(p_handle);

    return status;
}

/*
 * 函数功能：启动中断方式接收（IDLE 空闲帧检测）
 * 入口参数：handle    - RS-485 句柄指针
 *           buf       - 接收缓冲�? *           buf_size  - 缓冲区最大容�? * 返回值：HAL_OK / HAL_ERROR
 * 限定条件：UART 须已开启全局中断（NVIC 使能�? * 函数说明�? *   使用 HAL_UARTEx_ReceiveToIdle_IT 实现变长帧接收�? *   �?UART 总线空闲超过 1 个字符时间后，硬件触�?IDLE 中断�? *   HAL 自动调用 HAL_UARTEx_RxEventCallback 并返回已接收字节数�? *   用户须在该回调中重新调用本函数重新开启接�? */
HAL_StatusTypeDef rs485_receive_start(rs485_handle_t *p_handle,
                                       uint8_t *p_buf,
                                       uint16_t buf_size) {
    p_handle->rx_size = 0;
    return HAL_UARTEx_ReceiveToIdle_IT(p_handle->huart, p_buf, buf_size);
}

/*
 * 函数功能：缓存最近一次接收的字节�? * 入口参数：handle - RS-485 句柄指针
 *           size   - 本次接收到的字节�? * 返回值：�? * 限定条件：须�?HAL_UARTEx_RxEventCallback 中调�? */
void rs485_rx_set_size(rs485_handle_t *p_handle, uint16_t size) {
    p_handle->rx_size = size;
}

/*
 * 函数功能：获取最近一次接收的字节�? * 入口参数：handle - RS-485 句柄指针
 * 返回值：字节�? */
uint16_t rs485_rx_get_size(const rs485_handle_t *p_handle) {
    return p_handle->rx_size;
}