Momentum-Apps/firmware/targets/f7/furi-hal/furi-hal-uart.c

#include <furi-hal-uart.h>
#include <stdbool.h>
#include <stm32wbxx_ll_lpuart.h>
#include <stm32wbxx_ll_usart.h>
#include <furi-hal-resources.h>

#include <furi.h>

static void (*irq_cb[2])(uint8_t ev, uint8_t data);

static void furi_hal_usart_init(uint32_t baud) {
    hal_gpio_init_ex(
        &gpio_usart_tx,
        GpioModeAltFunctionPushPull,
        GpioPullUp,
        GpioSpeedVeryHigh,
        GpioAltFn7USART1);
    hal_gpio_init_ex(
        &gpio_usart_rx,
        GpioModeAltFunctionPushPull,
        GpioPullUp,
        GpioSpeedVeryHigh,
        GpioAltFn7USART1);

    LL_USART_InitTypeDef USART_InitStruct = {0};
    USART_InitStruct.PrescalerValue = LL_USART_PRESCALER_DIV1;
    USART_InitStruct.BaudRate = baud;
    USART_InitStruct.DataWidth = LL_USART_DATAWIDTH_8B;
    USART_InitStruct.StopBits = LL_USART_STOPBITS_1;
    USART_InitStruct.Parity = LL_USART_PARITY_NONE;
    USART_InitStruct.TransferDirection = LL_USART_DIRECTION_TX_RX;
    USART_InitStruct.HardwareFlowControl = LL_USART_HWCONTROL_NONE;
    USART_InitStruct.OverSampling = LL_USART_OVERSAMPLING_16;
    LL_USART_Init(USART1, &USART_InitStruct);
    LL_USART_SetTXFIFOThreshold(USART1, LL_USART_FIFOTHRESHOLD_1_2);
    LL_USART_EnableFIFO(USART1);
    LL_USART_ConfigAsyncMode(USART1);

    LL_USART_Enable(USART1);

    while(!LL_USART_IsActiveFlag_TEACK(USART1));

    LL_USART_EnableIT_RXNE_RXFNE(USART1);
    LL_USART_EnableIT_IDLE(USART1);
    HAL_NVIC_SetPriority(USART1_IRQn, 5, 0);
}

static void furi_hal_lpuart_init(uint32_t baud) {
    hal_gpio_init_ex(
        &gpio_ext_pc0,
        GpioModeAltFunctionPushPull,
        GpioPullUp,
        GpioSpeedVeryHigh,
        GpioAltFn8LPUART1);
    hal_gpio_init_ex(
        &gpio_ext_pc1,
        GpioModeAltFunctionPushPull,
        GpioPullUp,
        GpioSpeedVeryHigh,
        GpioAltFn8LPUART1);

    LL_LPUART_InitTypeDef LPUART_InitStruct = {0};
    LPUART_InitStruct.PrescalerValue = LL_LPUART_PRESCALER_DIV1;
    LPUART_InitStruct.BaudRate = 115200;
    LPUART_InitStruct.DataWidth = LL_LPUART_DATAWIDTH_8B;
    LPUART_InitStruct.StopBits = LL_LPUART_STOPBITS_1;
    LPUART_InitStruct.Parity = LL_LPUART_PARITY_NONE;
    LPUART_InitStruct.TransferDirection = LL_LPUART_DIRECTION_TX_RX;
    LPUART_InitStruct.HardwareFlowControl = LL_LPUART_HWCONTROL_NONE;
    LL_LPUART_Init(LPUART1, &LPUART_InitStruct);
    LL_LPUART_SetTXFIFOThreshold(LPUART1, LL_LPUART_FIFOTHRESHOLD_1_8);
    LL_LPUART_SetRXFIFOThreshold(LPUART1, LL_LPUART_FIFOTHRESHOLD_1_8);
    LL_LPUART_EnableFIFO(LPUART1);

    LL_LPUART_Enable(LPUART1);

    while((!(LL_LPUART_IsActiveFlag_TEACK(LPUART1))) || (!(LL_LPUART_IsActiveFlag_REACK(LPUART1))));

    furi_hal_uart_set_br(FuriHalUartIdLPUART1, baud);

    LL_LPUART_EnableIT_RXNE_RXFNE(LPUART1);
    LL_LPUART_EnableIT_IDLE(LPUART1);
    HAL_NVIC_SetPriority(LPUART1_IRQn, 5, 0);
}

void furi_hal_uart_init(FuriHalUartId ch, uint32_t baud) {
    if (ch == FuriHalUartIdLPUART1)
        furi_hal_lpuart_init(baud);
    else if (ch == FuriHalUartIdUSART1)
        furi_hal_usart_init(baud);
}

void furi_hal_uart_set_br(FuriHalUartId ch, uint32_t baud) {
    if (ch == FuriHalUartIdUSART1) {
        if (LL_USART_IsEnabled(USART1)) {
            // Wait for transfer complete flag
            while (!LL_USART_IsActiveFlag_TC(USART1));
            LL_USART_Disable(USART1);
            uint32_t uartclk = LL_RCC_GetUSARTClockFreq(LL_RCC_USART1_CLKSOURCE);
            LL_USART_SetBaudRate(USART1, uartclk, LL_USART_PRESCALER_DIV1, LL_USART_OVERSAMPLING_16, baud);
            LL_USART_Enable(USART1);
        }
    } else if (ch == FuriHalUartIdLPUART1) {
        if (LL_LPUART_IsEnabled(LPUART1)) {
            // Wait for transfer complete flag
            while (!LL_LPUART_IsActiveFlag_TC(LPUART1));
            LL_LPUART_Disable(LPUART1);
            uint32_t uartclk = LL_RCC_GetLPUARTClockFreq(LL_RCC_GetLPUARTClockSource(LL_RCC_LPUART1_CLKSOURCE_PCLK1));
            if (uartclk/baud > 4095) {
                LL_LPUART_SetPrescaler(LPUART1, LL_LPUART_PRESCALER_DIV32);
                LL_LPUART_SetBaudRate(LPUART1, uartclk, LL_LPUART_PRESCALER_DIV32, baud);
            } else {
                LL_LPUART_SetPrescaler(LPUART1, LL_LPUART_PRESCALER_DIV1);
                LL_LPUART_SetBaudRate(LPUART1, uartclk, LL_LPUART_PRESCALER_DIV1, baud);
            }
            LL_LPUART_Enable(LPUART1);
        }
    }
}

void furi_hal_uart_deinit(FuriHalUartId ch) {
    furi_hal_uart_set_irq_cb(ch, NULL);
    if (ch == FuriHalUartIdUSART1) {
        LL_USART_Disable(USART1);
        hal_gpio_init(&gpio_usart_tx, GpioModeAnalog, GpioPullNo, GpioSpeedLow);
        hal_gpio_init(&gpio_usart_rx, GpioModeAnalog, GpioPullNo, GpioSpeedLow);
    } else if (ch == FuriHalUartIdLPUART1) {
        LL_LPUART_Disable(LPUART1);
        hal_gpio_init(&gpio_ext_pc0, GpioModeAnalog, GpioPullNo, GpioSpeedLow);
        hal_gpio_init(&gpio_ext_pc1, GpioModeAnalog, GpioPullNo, GpioSpeedLow);
    }
}

void furi_hal_uart_tx(FuriHalUartId ch, uint8_t* buffer, size_t buffer_size) {
    if (ch == FuriHalUartIdUSART1) {
        if (LL_USART_IsEnabled(USART1) == 0)
            return;

        while(buffer_size > 0) {
            while (!LL_USART_IsActiveFlag_TXE(USART1));

            LL_USART_TransmitData8(USART1, *buffer);
            buffer++;
            buffer_size--;
        }

    } else if (ch == FuriHalUartIdLPUART1) {
        if (LL_LPUART_IsEnabled(LPUART1) == 0)
            return;

        while(buffer_size > 0) {
            while (!LL_LPUART_IsActiveFlag_TXE(LPUART1));

            LL_LPUART_TransmitData8(LPUART1, *buffer);
    
            buffer++;
            buffer_size--;
        }
    }
}

void furi_hal_uart_set_irq_cb(FuriHalUartId ch, void (*cb)(UartIrqEvent ev, uint8_t data)) {
    if (cb == NULL) {
        if (ch == FuriHalUartIdUSART1)
            NVIC_DisableIRQ(USART1_IRQn);
        else if (ch == FuriHalUartIdLPUART1)
            NVIC_DisableIRQ(LPUART1_IRQn);
        irq_cb[ch] = cb;
    } else {
        irq_cb[ch] = cb;
        if (ch == FuriHalUartIdUSART1)
            NVIC_EnableIRQ(USART1_IRQn);
        else if (ch == FuriHalUartIdLPUART1)
            NVIC_EnableIRQ(LPUART1_IRQn);
    }
}

void LPUART1_IRQHandler(void) {
    if (LL_LPUART_IsActiveFlag_RXNE_RXFNE(LPUART1)) {
        uint8_t data = LL_LPUART_ReceiveData8(LPUART1);
        irq_cb[FuriHalUartIdLPUART1](UartIrqEventRXNE, data);
    } else if (LL_LPUART_IsActiveFlag_IDLE(LPUART1)) {
        irq_cb[FuriHalUartIdLPUART1](UartIrqEventIDLE, 0);
        LL_LPUART_ClearFlag_IDLE(LPUART1);
    }
    //TODO: more events
}

void USART1_IRQHandler(void) {
    if (LL_USART_IsActiveFlag_RXNE_RXFNE(USART1)) {
        uint8_t data = LL_USART_ReceiveData8(USART1);
        irq_cb[FuriHalUartIdUSART1](UartIrqEventRXNE, data);
    } else if (LL_USART_IsActiveFlag_IDLE(USART1)) {
        irq_cb[FuriHalUartIdUSART1](UartIrqEventIDLE, 0);
        LL_USART_ClearFlag_IDLE(USART1);
    }
}