stm32_uart.hpp

#pragma once
 
#include <atomic>
 
#include "main.h"
#include "stm32_uart_gpdma.hpp"
 
#ifdef HAL_UART_MODULE_ENABLED
 
#ifdef UART
#undef UART
#endif
 
#include "double_buffer.hpp"
#include "flag.hpp"
#include "libxr_def.hpp"
#include "libxr_rw.hpp"
#include "serialized_service.hpp"
#include "uart.hpp"
 
typedef enum : uint8_t
{
#ifdef USART1
  STM32_USART1,
#endif
#ifdef USART2
  STM32_USART2,
#endif
#ifdef USART3
  STM32_USART3,
#endif
#ifdef USART4
  STM32_USART4,
#endif
#ifdef USART5
  STM32_USART5,
#endif
#ifdef USART6
  STM32_USART6,
#endif
#ifdef USART7
  STM32_USART7,
#endif
#ifdef USART8
  STM32_USART8,
#endif
#ifdef USART9
  STM32_USART9,
#endif
#ifdef USART10
  STM32_USART10,
#endif
#ifdef USART11
  STM32_USART11,
#endif
#ifdef USART12
  STM32_USART12,
#endif
#ifdef USART13
  STM32_USART13,
#endif
#ifdef UART1
  STM32_UART1,
#endif
#ifdef UART2
  STM32_UART2,
#endif
#ifdef UART3
  STM32_UART3,
#endif
#ifdef UART4
  STM32_UART4,
#endif
#ifdef UART5
  STM32_UART5,
#endif
#ifdef UART6
  STM32_UART6,
#endif
#ifdef UART7
  STM32_UART7,
#endif
#ifdef UART8
  STM32_UART8,
#endif
#ifdef UART9
  STM32_UART9,
#endif
#ifdef UART10
  STM32_UART10,
#endif
#ifdef UART11
  STM32_UART11,
#endif
#ifdef UART12
  STM32_UART12,
#endif
#ifdef UART13
  STM32_UART13,
#endif
#ifdef LPUART1
  STM32_LPUART1,
#endif
#ifdef LPUART2
  STM32_LPUART2,
#endif
#ifdef LPUART3
  STM32_LPUART3,
#endif
  STM32_UART_NUMBER,
  STM32_UART_ID_ERROR
} stm32_uart_id_t;
 
stm32_uart_id_t stm32_uart_get_id(USART_TypeDef* addr);
 
namespace LibXR
{
class STM32UART : public UART
{
 public:
  static void WriteFun(WritePort& port, bool in_isr);
 
  STM32UART(UART_HandleTypeDef* uart_handle, RawData dma_buff_rx, RawData dma_buff_tx,
            uint32_t tx_queue_size = 5);
 
  ErrorCode SetConfig(UART::Configuration config, bool in_isr = false) override;
 
  void SetRxDMA(bool in_isr = false);
  void TxCompleteIRQHandler();
  void RxEventIRQHandler();
  void ErrorIRQHandler();
  void AbortCompleteIRQHandler();
 
  ReadPort _read_port;
  WritePort _write_port;
 
  RawData dma_buff_rx_;
  DoubleBuffer dma_buff_tx_;
 
  size_t last_rx_pos_ = 0;
 
  UART_HandleTypeDef* uart_handle_;
 
  Flag::Plain tx_busy_;
 
  stm32_uart_id_t id_ = STM32_UART_ID_ERROR;
 
  static STM32UART* map[STM32_UART_NUMBER];  // NOLINT
 
 private:
  enum class ConfigState : uint32_t
  {
    EMPTY = 0U,
    RESERVED = 1U,
    PUBLISHED = 2U,
  };
 
  static constexpr uint32_t TX_EVENT_WRITE = 1U << 0U;
  static constexpr uint32_t TX_EVENT_DONE = 1U << 1U;
  static constexpr uint32_t TX_EVENT_CONFIG = 1U << 2U;
  static constexpr uint32_t TX_EVENT_RX_WORK = 1U << 3U;
  static constexpr uint32_t TX_EVENT_ABORT = 1U << 4U;
  static constexpr uint32_t TX_EVENT_ERROR = 1U << 5U;
#if defined(LIBXR_STM32_UART_GPDMA)
  static constexpr uint32_t TX_EVENT_START_RX = 1U << 6U;
  static void DmaAbortCallback(DMA_HandleTypeDef* dma_handle);
#endif
 
  void BeginAbort(bool in_isr);
 
  void HandleTxService(uint32_t events, bool in_isr);
  void FillTx(bool in_isr);
  void HandleTxDone(bool in_isr);
  void StartTxDma(bool in_isr);
  void ApplyConfig(UART::Configuration config, bool in_isr);
  void TryApplyConfig(bool in_isr);
  void HandleRxData(bool in_isr);
 
  SerializedService tx_service_;
  bool abort_pending_ = false;
  std::atomic<ConfigState> config_state_{ConfigState::EMPTY};
  UART::Configuration pending_config_{};
#if defined(LIBXR_STM32_UART_GPDMA)
  STM32GpdmaUartAdapter gpdma_adapter_;
#endif
};
 
}  // namespace LibXR
 
#endif