circuitpython/ports/nrf/common-hal/busio/UART.c

333 lines
11 KiB
C
Raw Normal View History

/*
* This file is part of the MicroPython project, http://micropython.org/
*
* The MIT License (MIT)
*
2018-10-16 11:05:02 -04:00
* Copyright (c) 2018 Ha Thach for Adafruit Industries
*
* 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 "shared-bindings/microcontroller/__init__.h"
#include "shared-bindings/busio/UART.h"
#include "lib/utils/interrupt_char.h"
#include "py/mpconfig.h"
#include "py/gc.h"
#include "py/mperrno.h"
#include "py/runtime.h"
#include "py/stream.h"
2018-08-16 17:21:44 -04:00
#include "supervisor/shared/translate.h"
#include "tick.h"
2018-09-24 03:37:28 -04:00
#include "nrfx_uarte.h"
2018-09-20 16:48:13 -04:00
#include <string.h>
2018-09-19 06:59:15 -04:00
// expression to examine, and return value in case of failing
2018-09-19 14:07:45 -04:00
#define _VERIFY_ERR(_exp) \
2018-09-19 06:59:15 -04:00
do {\
uint32_t _err = (_exp);\
if (NRFX_SUCCESS != _err ) {\
2018-09-25 15:06:32 -04:00
mp_raise_msg_varg(&mp_type_RuntimeError, translate("error = 0x%08lX"), _err);\
2018-09-19 06:59:15 -04:00
}\
}while(0)
static nrfx_uarte_t nrfx_uartes[] = {
#if NRFX_CHECK(NRFX_UARTE0_ENABLED)
NRFX_UARTE_INSTANCE(0),
#endif
#if NRFX_CHECK(NRFX_UARTE1_ENABLED)
NRFX_UARTE_INSTANCE(1),
#endif
};
static uint32_t get_nrf_baud (uint32_t baudrate) {
static const struct {
const uint32_t boundary;
nrf_uarte_baudrate_t uarte_baudraute;
} baudrate_map[] = {
{ 1200, NRF_UARTE_BAUDRATE_1200 },
{ 2400, NRF_UARTE_BAUDRATE_2400 },
{ 4800, NRF_UARTE_BAUDRATE_4800 },
{ 9600, NRF_UARTE_BAUDRATE_9600 },
{ 14400, NRF_UARTE_BAUDRATE_14400 },
{ 19200, NRF_UARTE_BAUDRATE_19200 },
{ 28800, NRF_UARTE_BAUDRATE_28800 },
{ 38400, NRF_UARTE_BAUDRATE_38400 },
{ 57600, NRF_UARTE_BAUDRATE_57600 },
{ 76800, NRF_UARTE_BAUDRATE_76800 },
{ 115200, NRF_UARTE_BAUDRATE_115200 },
{ 230400, NRF_UARTE_BAUDRATE_230400 },
{ 250000, NRF_UARTE_BAUDRATE_250000 },
{ 460800, NRF_UARTE_BAUDRATE_460800 },
{ 921600, NRF_UARTE_BAUDRATE_921600 },
{ 0, NRF_UARTE_BAUDRATE_1000000 },
};
size_t i = 0;
uint32_t boundary;
do {
boundary = baudrate_map[i].boundary;
if (baudrate <= boundary || boundary == 0) {
return baudrate_map[i].uarte_baudraute;
}
i++;
} while (true);
}
2018-09-19 06:59:15 -04:00
2018-09-24 03:37:28 -04:00
static void uart_callback_irq (const nrfx_uarte_event_t * event, void * context) {
2018-09-19 06:59:15 -04:00
busio_uart_obj_t* self = (busio_uart_obj_t*) context;
switch ( event->type ) {
2018-09-25 03:29:45 -04:00
case NRFX_UARTE_EVT_RX_DONE:
2018-12-27 13:05:30 -05:00
ringbuf_put_n(&self->rbuf, event->data.rxtx.p_data, event->data.rxtx.bytes);
2018-12-13 11:48:53 -05:00
// keep receiving
(void) nrfx_uarte_rx(self->uarte, &self->rx_char, 1);
2018-09-25 01:37:31 -04:00
break;
2018-09-25 03:29:45 -04:00
case NRFX_UARTE_EVT_TX_DONE:
2018-12-13 11:48:53 -05:00
// nothing to do
2018-09-19 06:59:15 -04:00
break;
2018-09-25 03:29:45 -04:00
case NRFX_UARTE_EVT_ERROR:
// Possible Error source is Overrun, Parity, Framing, Break
// uint32_t errsrc = event->data.error.error_mask;
2018-12-27 13:05:30 -05:00
ringbuf_put_n(&self->rbuf, event->data.error.rxtx.p_data, event->data.error.rxtx.bytes);
// Keep receiving
(void) nrfx_uarte_rx(self->uarte, &self->rx_char, 1);
2018-09-19 14:07:45 -04:00
break;
2018-09-19 06:59:15 -04:00
default:
break;
}
}
void uart_reset(void) {
for (size_t i = 0 ; i < MP_ARRAY_SIZE(nrfx_uartes); i++) {
nrf_uarte_disable(nrfx_uartes[i].p_reg);
}
}
2018-09-19 06:59:15 -04:00
void common_hal_busio_uart_construct (busio_uart_obj_t *self,
const mcu_pin_obj_t * tx, const mcu_pin_obj_t * rx, uint32_t baudrate,
uint8_t bits, uart_parity_t parity, uint8_t stop, mp_float_t timeout,
2018-09-19 06:59:15 -04:00
uint8_t receiver_buffer_size) {
// Find a free UART peripheral.
self->uarte = NULL;
for (size_t i = 0 ; i < MP_ARRAY_SIZE(nrfx_uartes); i++) {
if ((nrfx_uartes[i].p_reg->ENABLE & UARTE_ENABLE_ENABLE_Msk) == 0) {
self->uarte = &nrfx_uartes[i];
break;
}
}
if (self->uarte == NULL) {
mp_raise_ValueError(translate("All UART peripherals are in use"));
}
2018-09-24 05:06:44 -04:00
if ( (tx == mp_const_none) && (rx == mp_const_none) ) {
mp_raise_ValueError(translate("tx and rx cannot both be None"));
2018-09-19 06:59:15 -04:00
}
if ( receiver_buffer_size == 0 ) {
mp_raise_ValueError(translate("Invalid buffer size"));
}
2018-09-20 14:27:52 -04:00
if ( parity == PARITY_ODD ) {
2018-09-25 15:10:44 -04:00
mp_raise_ValueError(translate("Odd parity is not supported"));
2018-09-20 14:27:52 -04:00
}
2018-09-24 03:37:28 -04:00
nrfx_uarte_config_t config = {
2018-09-25 05:14:44 -04:00
.pseltxd = (tx == mp_const_none) ? NRF_UARTE_PSEL_DISCONNECTED : tx->number,
.pselrxd = (rx == mp_const_none) ? NRF_UARTE_PSEL_DISCONNECTED : rx->number,
.pselcts = NRF_UARTE_PSEL_DISCONNECTED,
.pselrts = NRF_UARTE_PSEL_DISCONNECTED,
2018-09-19 06:59:15 -04:00
.p_context = self,
2018-09-25 05:14:44 -04:00
.hwfc = NRF_UARTE_HWFC_DISABLED,
.parity = (parity == PARITY_NONE) ? NRF_UARTE_PARITY_EXCLUDED : NRF_UARTE_PARITY_INCLUDED,
2018-09-20 14:27:52 -04:00
.baudrate = get_nrf_baud(baudrate),
2018-09-19 06:59:15 -04:00
.interrupt_priority = 7
};
nrfx_uarte_uninit(self->uarte);
_VERIFY_ERR(nrfx_uarte_init(self->uarte, &config, uart_callback_irq));
2018-09-19 06:59:15 -04:00
2018-09-20 14:27:52 -04:00
// Init buffer for rx
2018-09-24 05:06:44 -04:00
if ( rx != mp_const_none ) {
// Initially allocate the UART's buffer in the long-lived part of the
// heap. UARTs are generally long-lived objects, but the "make long-
// lived" machinery is incapable of moving internal pointers like
// self->buffer, so do it manually. (However, as long as internal
// pointers like this are NOT moved, allocating the buffer
// in the long-lived pool is not strictly necessary)
// (This is a macro.)
ringbuf_alloc(&self->rbuf, receiver_buffer_size, true);
2018-12-13 11:48:53 -05:00
if ( !self->rbuf.buf ) {
nrfx_uarte_uninit(self->uarte);
2018-09-24 05:06:44 -04:00
mp_raise_msg(&mp_type_MemoryError, translate("Failed to allocate RX buffer"));
}
2018-12-13 11:48:53 -05:00
self->rx_pin_number = rx->number;
2018-09-24 05:06:44 -04:00
claim_pin(rx);
}
if ( tx != mp_const_none ) {
self->tx_pin_number = tx->number;
2018-09-24 05:06:44 -04:00
claim_pin(tx);
} else {
self->tx_pin_number = NO_PIN;
2018-09-19 06:59:15 -04:00
}
self->baudrate = baudrate;
self->timeout_ms = timeout * 1000;
2018-09-25 01:37:31 -04:00
2018-12-13 11:48:53 -05:00
// Initial wait for incoming byte
_VERIFY_ERR(nrfx_uarte_rx(self->uarte, &self->rx_char, 1));
}
bool common_hal_busio_uart_deinited(busio_uart_obj_t *self) {
return self->rx_pin_number == NO_PIN;
}
void common_hal_busio_uart_deinit(busio_uart_obj_t *self) {
2018-09-19 06:59:15 -04:00
if ( !common_hal_busio_uart_deinited(self) ) {
nrfx_uarte_uninit(self->uarte);
reset_pin_number(self->tx_pin_number);
reset_pin_number(self->rx_pin_number);
self->tx_pin_number = NO_PIN;
self->rx_pin_number = NO_PIN;
2018-12-13 11:48:53 -05:00
gc_free(self->rbuf.buf);
self->rbuf.size = 0;
self->rbuf.iput = self->rbuf.iget = 0;
}
}
// Read characters.
size_t common_hal_busio_uart_read(busio_uart_obj_t *self, uint8_t *data, size_t len, int *errcode) {
if ( nrf_uarte_rx_pin_get(self->uarte->p_reg) == NRF_UARTE_PSEL_DISCONNECTED ) {
2018-09-24 05:06:44 -04:00
mp_raise_ValueError(translate("No RX pin"));
}
2018-12-13 11:48:53 -05:00
size_t rx_bytes = 0;
2018-09-20 16:48:13 -04:00
uint64_t start_ticks = ticks_ms;
2018-12-13 11:48:53 -05:00
// Wait for all bytes received or timeout
while ( (ringbuf_count(&self->rbuf) < len) && (ticks_ms - start_ticks < self->timeout_ms) ) {
2018-09-20 16:48:13 -04:00
#ifdef MICROPY_VM_HOOK_LOOP
2018-12-13 11:48:53 -05:00
MICROPY_VM_HOOK_LOOP ;
// Allow user to break out of a timeout with a KeyboardInterrupt.
if ( mp_hal_is_interrupted() ) {
return 0;
2018-09-20 16:48:13 -04:00
}
2018-12-13 11:48:53 -05:00
#endif
}
2018-09-25 01:37:31 -04:00
2018-12-13 11:48:53 -05:00
// prevent conflict with uart irq
NVIC_DisableIRQ(nrfx_get_irq_number(self->uarte->p_reg));
2018-09-25 01:37:31 -04:00
2018-12-13 11:48:53 -05:00
// copy received data
rx_bytes = ringbuf_count(&self->rbuf);
rx_bytes = MIN(rx_bytes, len);
for ( uint16_t i = 0; i < rx_bytes; i++ ) {
data[i] = ringbuf_get(&self->rbuf);
2018-09-20 16:48:13 -04:00
}
2018-10-16 11:05:02 -04:00
NVIC_EnableIRQ(nrfx_get_irq_number(self->uarte->p_reg));
2018-09-25 01:37:31 -04:00
2018-12-13 11:48:53 -05:00
return rx_bytes;
}
// Write characters.
size_t common_hal_busio_uart_write (busio_uart_obj_t *self, const uint8_t *data, size_t len, int *errcode) {
if ( nrf_uarte_tx_pin_get(self->uarte->p_reg) == NRF_UARTE_PSEL_DISCONNECTED ) {
2018-09-24 05:06:44 -04:00
mp_raise_ValueError(translate("No TX pin"));
}
2018-09-24 03:37:28 -04:00
if ( len == 0 ) return 0;
2018-09-19 06:59:15 -04:00
2018-09-25 01:48:48 -04:00
uint64_t start_ticks = ticks_ms;
// Wait for on-going transfer to complete
while ( nrfx_uarte_tx_in_progress(self->uarte) && (ticks_ms - start_ticks < self->timeout_ms) ) {
2018-09-25 01:48:48 -04:00
#ifdef MICROPY_VM_HOOK_LOOP
MICROPY_VM_HOOK_LOOP
#endif
}
// Time up
if ( !(ticks_ms - start_ticks < self->timeout_ms) ) {
*errcode = MP_EAGAIN;
return MP_STREAM_ERROR;
2018-09-24 04:54:32 -04:00
}
2018-09-24 05:18:49 -04:00
// EasyDMA can only access SRAM
uint8_t * tx_buf = (uint8_t*) data;
if ( !nrfx_is_in_ram(data) ) {
// TODO: If this is not too big, we could allocate it on the stack.
tx_buf = (uint8_t *) gc_alloc(len, false, false);
2018-09-24 05:18:49 -04:00
memcpy(tx_buf, data, len);
}
(*errcode) = nrfx_uarte_tx(self->uarte, tx_buf, len);
2018-09-19 14:07:45 -04:00
_VERIFY_ERR(*errcode);
2018-09-19 06:59:15 -04:00
(*errcode) = 0;
while ( nrfx_uarte_tx_in_progress(self->uarte) && (ticks_ms - start_ticks < self->timeout_ms) ) {
2018-09-19 06:59:15 -04:00
#ifdef MICROPY_VM_HOOK_LOOP
MICROPY_VM_HOOK_LOOP
#endif
}
2018-09-24 05:18:49 -04:00
if ( !nrfx_is_in_ram(data) ) {
gc_free(tx_buf);
}
2018-09-19 06:59:15 -04:00
return len;
}
uint32_t common_hal_busio_uart_get_baudrate(busio_uart_obj_t *self) {
return self->baudrate;
}
void common_hal_busio_uart_set_baudrate(busio_uart_obj_t *self, uint32_t baudrate) {
self->baudrate = baudrate;
nrf_uarte_baudrate_set(self->uarte->p_reg, get_nrf_baud(baudrate));
}
uint32_t common_hal_busio_uart_rx_characters_available(busio_uart_obj_t *self) {
2018-12-13 11:48:53 -05:00
return ringbuf_count(&self->rbuf);
}
void common_hal_busio_uart_clear_rx_buffer(busio_uart_obj_t *self) {
2018-12-13 11:48:53 -05:00
// prevent conflict with uart irq
NVIC_DisableIRQ(nrfx_get_irq_number(self->uarte->p_reg));
2018-12-13 11:48:53 -05:00
ringbuf_clear(&self->rbuf);
NVIC_EnableIRQ(nrfx_get_irq_number(self->uarte->p_reg));
}
bool common_hal_busio_uart_ready_to_tx(busio_uart_obj_t *self) {
return !nrfx_uarte_tx_in_progress(self->uarte);
2018-09-25 03:29:45 -04:00
}