/******************************************************************************
 * @file     uart.c
 * @brief    UART firmware functions.
 * @version  V4.0.1
 * @date     30-July-2026
 *
 * @note
 * Copyright (C) 2022 Spintrol Electronic Technology (Shanghai) Co., Ltd.. All rights reserved.
 *
 * @attention
 * THIS SOFTWARE JUST PROVIDES CUSTOMERS WITH CODING INFORMATION REGARDING
 * THEIR PRODUCTS, WHICH AIMS AT SAVING TIME FOR THEM. SPINTROL SHALL NOT BE
 * LIABLE FOR THE USE OF THE SOFTWARE. SPINTROL DOES NOT GUARANTEE THE
 * CORRECTNESS OF THIS SOFTWARE AND RESERVES THE RIGHT TO MODIFY THE SOFTWARE
 * WITHOUT NOTIFICATION.
 *
 ******************************************************************************/

#include "uart.h"


#if defined(__CC_ARM)
    #pragma push
    #pragma O1
#endif




/******************************************************************************
 * @brief      Initializes the UART
 *
 * @param[in]  UARTx      :  Select the UART module
 * @param[in]  u32Baudrate:  Baudrate
 *
 * @return     none
 *
 ******************************************************************************/
#if defined(__ICCARM__)
#pragma optimize=low
#endif
void UART_Init(UART_REGS *UARTx, uint32_t u32Baudrate)
{
    uint32_t baudRateDivisor;

    /* Enable UART Clock */
    CLOCK_EnableModule(UART_MODULE);

    /************Configure the Baud Rate*********************************/
    /* Calculate UART clock divisor */
    baudRateDivisor = CLOCK_GetModuleClock(UART_MODULE) / (16 * u32Baudrate);

    /* Enable DLAB to set baud rate */
    UARTx->UARTLCR.bit.DLAB = UARTLCR_BIT_DLAB_DIVISOR_ACCESS;

    /* Config the DLL and DLH registers */
    UARTx->UARTRBR_UARTTHR_UARTDLL.UARTDLL.bit.VAL = baudRateDivisor & 0xFF;
    UARTx->UARTDLH_UARTIER.UARTDLH.bit.VAL = (baudRateDivisor >> 0x08) & 0xFF;

    /* Disable DLAB */
    UARTx->UARTLCR.bit.DLAB = UARTLCR_BIT_DLAB_DIVISOR_LOCK;


    /***********Configure Line control register (LCR)********************/
    /* Set parity */
    UARTx->UARTLCR.bit.PEN = UARTLCR_BIT_PEN_NO_PARITY;

    /* Set UART Data Length and stop bit */
    UARTx->UARTLCR.bit.WLS = UARTLCR_BIT_WLS_8_DATA_BIT;  /* 8 data bits */
    UARTx->UARTLCR.bit.STB = UARTLCR_BIT_STB_1_STOP_BIT;  /* 1 stop bit  */

    /* Uart unit enable */
    UARTx->UARTDLH_UARTIER.UARTIER.bit.UUE = UARTIER_BIT_UUE_UART_ENABLE;
}




/******************************************************************************
 * @brief      Set FIFO interrupt trigger threshold
 *
 * @param[in]  UARTx     :  Select the UART module
 * @param[in]  eRxTrigLvl:  Receive FIFO trigger threshold
 * @param[in]  eTxTrigLvl:  Transmit FIFO trigger threshold
 *
 * @return     none
 *
 ******************************************************************************/
#if defined(__ICCARM__)
#pragma optimize=low
#endif
void UART_SetFIFOTriggerThreshold(UART_REGS *UARTx, UART_RxThresholdEnum eRxTrigLvl, UART_TxThresholdEnum eTxTrigLvl)
{
    /* Set RX and TX threshold */
    UARTx->UARTIIR_UARTFCR.UARTFCR.all = (((eRxTrigLvl) << UARTFCR_ALL_RXTH_Pos) & UARTFCR_ALL_RXTH_Msk) |
                                         (((eTxTrigLvl) << UARTFCR_ALL_TXTH_Pos) & UARTFCR_ALL_TXTH_Msk) |
                                         UARTFCR_ALL_TRFIFOE_FIFO_ENABLE;
}




/******************************************************************************
 * @brief      Writes the specified string to the UART peripheral in safe way
 *
 * @param[in]  UARTx      :  Select the UART module
 * @param[in]  pu8Text    :  String message to be sent
 * @param[in]  u16TextLen :  Length of string message to be sent
 * @return     none
 *
 ******************************************************************************/
#if defined(__ICCARM__)
#pragma optimize=low
#endif
void UART_WriteTextSafe(UART_REGS *UARTx, const uint8_t *pu8Text, uint16_t u16TextLen)
{
    uint16_t u16Tmp = 0 ;
    for (u16Tmp = 0 ; u16Tmp < u16TextLen ; u16Tmp++)
    {
        UART_WriteByte(UARTx, *(pu8Text+u16Tmp));
        /* Wait until transmiter empty */
        while ((UARTx->UARTLSR.all & UARTLSR_ALL_TXDONE_Msk) == UARTLSR_ALL_TXDONE_TRANSMIT_NOT_DONE);
    }
    if (u16Tmp == u16TextLen)
    {
        UART_WriteByte(UARTx, '\0');
        /* Wait until transmiter empty */
        while ((UARTx->UARTLSR.all & UARTLSR_ALL_TXDONE_Msk) == UARTLSR_ALL_TXDONE_TRANSMIT_NOT_DONE);
    }
}




/******************************************************************************
 * @brief      Writes the specified string to the UART peripheral
 *
 * @param[in]  UARTx  :  Select the UART module
 * @param[in]  pu8Text:  String message to be sent
 *
 * @return     none
 *
 ******************************************************************************/
#if defined(__ICCARM__)
#pragma optimize=low
#endif
void UART_WriteText(UART_REGS *UARTx, const uint8_t *pu8Text)
{
    while (*pu8Text)
    {
        UART_WriteByte(UARTx, *(pu8Text++));

        /* Wait until transmiter empty */
        while ((UARTx->UARTLSR.all & UARTLSR_ALL_TXDONE_Msk) == UARTLSR_ALL_TXDONE_TRANSMIT_NOT_DONE);
    }
}




/******************************************************************************
 * @brief      Writes a specified number of bytes to the UART peripheral
 *
 * @param[in]  UARTx    :  Select the UART module
 * @param[in]  pu8Buffer:  Byte array that contains the data to send
 * @param[in]  u32Offset:  Zero-based byte offset in the buffer parameter
                           at which to begin copying bytes to UART peripheral
 * @param[in]  u32Count :  Number of byte data to be sent
 *
 * @return     none
 *
 ******************************************************************************/
#if defined(__ICCARM__)
#pragma optimize=low
#endif
void UART_Write(UART_REGS *UARTx, uint8_t *pu8Buffer, uint32_t u32Offset, uint32_t u32Count)
{
    uint32_t i = 0 ;
    uint32_t j = 0 ;
    uint16_t u16RestByteCount = 0;
    uint16_t u16FifoSendCount = 0 ;
    uint16_t u16FifoByteCount = 0 ;
    
    u16FifoByteCount = UART_GetFIFOByteCount(UARTx);
    u16FifoSendCount = u32Count / u16FifoByteCount;
    u16RestByteCount = u32Count % u16FifoByteCount;
    
    if ( UART_IS_FIFO_MODEL == UART_GetFIFOModule(UARTx))
    {
    /* Uart in FIFO Module*/
        for (i = 0; i < u16FifoSendCount; i++)
        {
            for ( j = 0 ; j < u16FifoByteCount ; j++)
            {
                UART_WriteByte(UARTx, (uint8_t)pu8Buffer[u32Offset + i*u16FifoByteCount + j]);
            }
            while ((UARTx->UARTLSR.all & UARTLSR_ALL_TXDONE_Msk) == UARTLSR_ALL_TXDONE_TRANSMIT_NOT_DONE)
            {
                ;
            }
        }
        if (u16RestByteCount > 0)
        {
            for ( j = 0 ; j < u16RestByteCount ; j++)
            {
                UART_WriteByte(UARTx, (uint8_t)pu8Buffer[u32Offset + i*u16FifoByteCount + j]);
            }
            /* Wait until transmiter empty */
            while ((UARTx->UARTLSR.all & UARTLSR_ALL_TXDONE_Msk) == UARTLSR_ALL_TXDONE_TRANSMIT_NOT_DONE)
            {
                ;
            }
        }
    }
    else
    {
    /* Uart in no FIFO Module*/    
        for (i = 0; i < u32Count; i++)
        {
            UART_WriteByte(UARTx, (uint8_t)pu8Buffer[u32Offset + i]);

            /* Wait until transmiter empty */
            while ((UARTx->UARTLSR.all & UARTLSR_ALL_TXDONE_Msk) == UARTLSR_ALL_TXDONE_TRANSMIT_NOT_DONE)
            {
                ;
            }
        }
    }
}




/******************************************************************************
 * @brief      Reads a specified number of bytes from the UART peripheral
 *
 * @param[in]  UARTx    :  Select the UART module
 * @param[in]  pu8Buffer:  Byte array that contains the data to be read
 * @param[in]  u32Offset:  Offset in the buffer array to begin reading from
 * @param[in]  u32Count :  Number of byte data to be read
 *
 * @return     none
 *
 ******************************************************************************/
#if defined(__ICCARM__)
#pragma optimize=low
#endif
void UART_Read(UART_REGS *UARTx, uint8_t *pu8Buffer, uint32_t u32Offset, uint32_t u32Count)
{
    uint32_t i;

    for (i = 0; i < u32Count; i++)
    {
        /* Wait until data is available in RBR or the FIFO */
        while ((UARTx->UARTLSR.all & UARTLSR_ALL_DR_Msk) == UARTLSR_ALL_DR_NO_DATA_RECEVIED);

        /* Read one byte data from UART peripheral */
        pu8Buffer[u32Offset + i] = UART_ReadByte(UARTx);
    }
}


#if defined(__CC_ARM)
    #pragma pop
#endif


/******************* Copyright (C) 2022 Spintrol Electronic Technology (Shanghai) Co., Ltd. ***** END OF FILE ****/
