Files
micropython/ports/alif/mpuart.c
Damien George 737acef5cb alif: Support more fine-grained pin alternate function selection.
Now raises an exception if the pin doesn't support the alternate function
unit number and line type, eg UART0_TX (previously it only checked the
peripheral).

Signed-off-by: Damien George <damien@micropython.org>
2025-04-09 00:22:33 +10:00

347 lines
12 KiB
C

/*
* This file is part of the MicroPython project, http://micropython.org/
*
* The MIT License (MIT)
*
* Copyright (c) 2024 OpenMV LLC.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "py/mphal.h"
#include "py/runtime.h"
#include "shared/runtime/softtimer.h"
#include "mpuart.h"
#include "sys_ctrl_uart.h"
#define mp_container_of(ptr, structure, member) (void *)((uintptr_t)(ptr) - offsetof(structure, member))
#define UART_LSR_TEMT_Pos (6)
#define SYST_PCLK (100000000)
#define CALCULATE_BITS_PER_CHAR(data_bits, parity, stop_bits) \
(1 + ((data_bits) + 5) + ((parity) != UART_PARITY_NONE) + ((stop_bits) + 1))
typedef struct _uart_state_t {
UART_TRANSFER_STATUS status;
uint32_t baudrate;
uint32_t bits_per_char;
ringbuf_t *rx_ringbuf;
const uint8_t *tx_src;
const uint8_t *tx_src_max;
soft_timer_entry_t rx_idle_timer;
unsigned int irq_trigger;
void (*irq_callback)(unsigned int uart_id, unsigned int trigger);
} uart_state_t;
static const uint8_t uart_irqn[UART_MAX] = {
UART0_IRQ_IRQn,
UART1_IRQ_IRQn,
UART2_IRQ_IRQn,
UART3_IRQ_IRQn,
UART4_IRQ_IRQn,
UART5_IRQ_IRQn,
UART6_IRQ_IRQn,
UART7_IRQ_IRQn,
};
static UART_Type *const uart_periph[UART_MAX] = {
(UART_Type *)UART0_BASE,
(UART_Type *)UART1_BASE,
(UART_Type *)UART2_BASE,
(UART_Type *)UART3_BASE,
(UART_Type *)UART4_BASE,
(UART_Type *)UART5_BASE,
(UART_Type *)UART6_BASE,
(UART_Type *)UART7_BASE,
};
static uart_state_t uart_state[UART_MAX];
// This is called by soft_timer and executes at IRQ_PRI_PENDSV.
static void rx_idle_timer_callback(soft_timer_entry_t *self) {
uart_state_t *state = mp_container_of(self, uart_state_t, rx_idle_timer);
if (state->irq_callback != NULL && state->irq_trigger & MP_UART_IRQ_RXIDLE) {
unsigned int uart_id = state - &uart_state[0];
state->irq_callback(uart_id, MP_UART_IRQ_RXIDLE);
}
}
void mp_uart_init(unsigned int uart_id, uint32_t baudrate,
UART_DATA_BITS data_bits, UART_PARITY parity, UART_STOP_BITS stop_bits,
mp_hal_pin_obj_t tx, mp_hal_pin_obj_t rx, ringbuf_t *rx_ringbuf) {
UART_Type *uart = uart_periph[uart_id];
uart_state_t *state = &uart_state[uart_id];
// Configure TX/RX pins.
mp_hal_pin_config(tx, MP_HAL_PIN_MODE_ALT, MP_HAL_PIN_PULL_NONE,
MP_HAL_PIN_SPEED_HIGH, MP_HAL_PIN_DRIVE_8MA, MP_HAL_PIN_ALT(UART_TX, uart_id), false);
mp_hal_pin_config(rx, MP_HAL_PIN_MODE_ALT, MP_HAL_PIN_PULL_NONE,
MP_HAL_PIN_SPEED_HIGH, MP_HAL_PIN_DRIVE_8MA, MP_HAL_PIN_ALT(UART_RX, uart_id), true);
// Configure the UART peripheral.
select_uart_clock_syst_pclk(uart_id);
enable_uart_clock(uart_id);
uart_software_reset(uart);
uart_enable_fifo(uart);
uart_disable_tx_irq(uart);
uart_disable_rx_irq(uart);
uart_set_baudrate(uart, SYST_PCLK, baudrate);
uart_set_data_parity_stop_bits(uart, data_bits, parity, stop_bits);
uart_set_flow_control(uart, UART_FLOW_CONTROL_NONE);
uart->UART_FCR |= UART_FCR_RCVR_FIFO_RESET;
uart_set_tx_trigger(uart, UART_TX_FIFO_EMPTY);
uart_set_rx_trigger(uart, UART_RX_FIFO_QUARTER_FULL);
// Initialise the state.
state->status = UART_TRANSFER_STATUS_NONE;
state->baudrate = baudrate;
state->bits_per_char = CALCULATE_BITS_PER_CHAR(data_bits, parity, stop_bits);
state->rx_ringbuf = rx_ringbuf;
state->tx_src = NULL;
state->tx_src_max = NULL;
state->irq_trigger = 0;
state->irq_callback = NULL;
// Enable interrupts.
NVIC_ClearPendingIRQ(uart_irqn[uart_id]);
NVIC_SetPriority(uart_irqn[uart_id], IRQ_PRI_UART_REPL);
NVIC_EnableIRQ(uart_irqn[uart_id]);
uart_enable_rx_irq(uart);
soft_timer_static_init(&state->rx_idle_timer, SOFT_TIMER_MODE_ONE_SHOT, 0, rx_idle_timer_callback);
}
void mp_uart_deinit(unsigned int uart_id) {
UART_Type *uart = uart_periph[uart_id];
uart_disable_rx_irq(uart);
NVIC_DisableIRQ(uart_irqn[uart_id]);
}
void mp_uart_set_irq_callback(unsigned int uart_id, unsigned int trigger, void (*callback)(unsigned int uart_id, unsigned int trigger)) {
UART_Type *uart = uart_periph[uart_id];
uart_state_t *state = &uart_state[uart_id];
state->irq_trigger = trigger;
state->irq_callback = callback;
if (trigger & MP_UART_IRQ_RX) {
uart_set_rx_trigger(uart, UART_RX_ONE_CHAR_IN_FIFO);
} else {
uart_set_rx_trigger(uart, UART_RX_FIFO_QUARTER_FULL);
}
}
void mp_uart_set_flow(unsigned int uart_id, mp_hal_pin_obj_t rts, mp_hal_pin_obj_t cts) {
UART_Type *uart = uart_periph[uart_id];
unsigned int flow = UART_FLOW_CONTROL_NONE;
if (rts != NULL) {
flow |= UART_FLOW_CONTROL_RTS;
mp_hal_pin_config(rts, MP_HAL_PIN_MODE_ALT, MP_HAL_PIN_PULL_NONE,
MP_HAL_PIN_SPEED_HIGH, MP_HAL_PIN_DRIVE_8MA, MP_HAL_PIN_ALT(UART_RTS, uart_id), false);
}
if (cts != NULL) {
flow |= UART_FLOW_CONTROL_CTS;
mp_hal_pin_config(cts, MP_HAL_PIN_MODE_ALT, MP_HAL_PIN_PULL_NONE,
MP_HAL_PIN_SPEED_HIGH, MP_HAL_PIN_DRIVE_8MA, MP_HAL_PIN_ALT(UART_CTS, uart_id), true);
}
uart_set_flow_control(uart, flow);
}
void mp_uart_set_baudrate(unsigned int uart_id, uint32_t baudrate) {
UART_Type *uart = uart_periph[uart_id];
uart_state_t *state = &uart_state[uart_id];
state->baudrate = baudrate;
uart_set_baudrate(uart, SYST_PCLK, baudrate);
}
void mp_uart_set_bits_parity_stop(unsigned int uart_id, UART_DATA_BITS data_bits, UART_PARITY parity, UART_STOP_BITS stop_bits) {
UART_Type *uart = uart_periph[uart_id];
uart_state_t *state = &uart_state[uart_id];
state->bits_per_char = CALCULATE_BITS_PER_CHAR(data_bits, parity, stop_bits);
uart_set_data_parity_stop_bits(uart, data_bits, parity, stop_bits);
}
size_t mp_uart_rx_any(unsigned int uart_id) {
uart_state_t *state = &uart_state[uart_id];
if (state->rx_ringbuf != NULL) {
return ringbuf_avail(state->rx_ringbuf);
}
return 0;
}
size_t mp_uart_tx_any(unsigned int uart_id) {
UART_Type *uart = uart_periph[uart_id];
return uart->UART_TFL + !((uart->UART_LSR >> UART_LSR_TEMT_Pos) & 1);
}
int mp_uart_rx_char(unsigned int uart_id) {
uart_state_t *state = &uart_state[uart_id];
if (state->rx_ringbuf != NULL && ringbuf_avail(state->rx_ringbuf)) {
return ringbuf_get(state->rx_ringbuf);
}
return -1;
}
void mp_uart_tx_data_blocking(unsigned int uart_id, const uint8_t *src, size_t len) {
UART_Type *uart = uart_periph[uart_id];
for (size_t i = 0; i < len; ++i) {
for (;;) {
size_t tx_avail = UART_FIFO_DEPTH - uart->UART_TFL;
if (tx_avail > 0) {
break;
}
mp_event_handle_nowait();
}
uart->UART_THR = src[i];
}
}
void mp_uart_tx_data(unsigned int uart_id, const uint8_t *src, size_t len) {
UART_Type *uart = uart_periph[uart_id];
// Configure transmission, to be handled by IRQ.
uart_state_t *state = &uart_state[uart_id];
state->status = UART_TRANSFER_STATUS_NONE;
state->tx_src = src;
state->tx_src_max = src + len;
// Enable TX IRQ to start transmission.
uart_enable_tx_irq(uart);
// Wait for transfer to complete.
const uint32_t total_timeout_ms = 100 * len;
uint32_t start = mp_hal_ticks_ms();
while (state->status == UART_TRANSFER_STATUS_NONE) {
if (mp_hal_ticks_ms() - start > total_timeout_ms) {
break;
}
mp_event_wait_indefinite();
}
// Disable TX IRQ.
uart_disable_tx_irq(uart);
}
static void mp_uart_irq_handler(unsigned int uart_id) {
UART_Type *uart = uart_periph[uart_id];
uart_state_t *state = &uart_state[uart_id];
// Process pending interrupt (below is order of highest to lowest priority).
uint32_t iir = uart->UART_IIR & UART_IIR_INTERRUPT_ID_MASK;
switch (iir) {
case UART_IIR_RECEIVER_LINE_STATUS: {
uint32_t lsr = uart->UART_LSR;
if (lsr & (UART_LSR_RECEIVER_FIFO_ERR | UART_LSR_OVERRUN_ERR)) {
state->status = UART_TRANSFER_STATUS_ERROR;
}
break;
}
case UART_IIR_RECEIVED_DATA_AVAILABLE:
case UART_IIR_CHARACTER_TIMEOUT: {
bool had_char = false;
while (uart->UART_USR & UART_USR_RECEIVE_FIFO_NOT_EMPTY) {
for (uint32_t rfl = uart->UART_RFL; rfl; --rfl) {
had_char = true;
int c = uart->UART_RBR;
#if MICROPY_HW_ENABLE_UART_REPL && MICROPY_KBD_EXCEPTION
if (uart_id == MICROPY_HW_UART_REPL) {
if (c == mp_interrupt_char) {
mp_sched_keyboard_interrupt();
continue;
}
}
#endif
if (state->rx_ringbuf != NULL) {
ringbuf_put(state->rx_ringbuf, c);
}
}
}
if (had_char) {
if (state->irq_callback != NULL && state->irq_trigger & MP_UART_IRQ_RX) {
state->irq_callback(uart_id, MP_UART_IRQ_RX);
}
}
if (state->irq_trigger & MP_UART_IRQ_RXIDLE) {
// Wait for 2 characters worth of time before triggering the RXIDLE event.
soft_timer_reinsert(&state->rx_idle_timer, 2000 * state->bits_per_char / state->baudrate + 1);
}
break;
}
case UART_IIR_TRANSMIT_HOLDING_REG_EMPTY:
while (uart->UART_USR & UART_USR_TRANSMIT_FIFO_NOT_FULL) {
if (state->tx_src < state->tx_src_max) {
uart->UART_THR = *state->tx_src++;
} else {
uart_disable_tx_irq(uart);
state->status = UART_TRANSFER_STATUS_SEND_COMPLETE;
if (state->irq_callback != NULL && state->irq_trigger & MP_UART_IRQ_TXIDLE) {
state->irq_callback(uart_id, MP_UART_IRQ_TXIDLE);
}
break;
}
}
break;
case UART_IIR_MODEM_STATUS:
(void)uart->UART_MSR;
break;
}
__SEV();
}
#define DEFINE_IRQ_HANDLER(id) \
void UART##id##_IRQHandler(void) { \
mp_uart_irq_handler(id); \
}
DEFINE_IRQ_HANDLER(0)
DEFINE_IRQ_HANDLER(1)
DEFINE_IRQ_HANDLER(2)
DEFINE_IRQ_HANDLER(3)
DEFINE_IRQ_HANDLER(4)
DEFINE_IRQ_HANDLER(5)
DEFINE_IRQ_HANDLER(6)
DEFINE_IRQ_HANDLER(7)
#if MICROPY_HW_ENABLE_UART_REPL
#define REPL_BAUDRATE (115200)
void mp_uart_init_repl(void) {
mp_uart_init(MICROPY_HW_UART_REPL,
REPL_BAUDRATE, UART_DATA_BITS_8, UART_PARITY_NONE, UART_STOP_BITS_1,
pin_REPL_UART_TX, pin_REPL_UART_RX, &stdin_ringbuf);
}
void mp_uart_write_strn_repl(const char *str, size_t len) {
mp_uart_tx_data(MICROPY_HW_UART_REPL, (const uint8_t *)str, len);
}
#endif