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

309 lines
9.4 KiB
C
Raw Normal View History

/*
* This file is part of the MicroPython project, http://micropython.org/
*
* The MIT License (MIT)
*
* Copyright (c) 2016 Damien P. George
*
* 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 "mpconfigport.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 ) {\
mp_raise_msg_varg(&mp_type_AssertionError, translate("error = 0x%08lX "), _err);\
}\
}while(0)
static uint32_t get_nrf_baud (uint32_t baudrate);
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 ) {
case NRFX_UART_EVT_TX_DONE:
break;
2018-09-19 14:07:45 -04:00
case NRFX_UART_EVT_RX_DONE:
2018-09-24 04:54:32 -04:00
self->rx_count = event->data.rxtx.bytes;
2018-09-19 14:07:45 -04:00
break;
2018-09-19 06:59:15 -04:00
default:
break;
}
}
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, uint32_t timeout,
uint8_t receiver_buffer_size) {
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
2018-09-19 06:59:15 -04:00
#else
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 ) {
mp_raise_ValueError(translate("busio.UART odd parity is not supported"));
}
2018-09-24 03:37:28 -04:00
nrfx_uarte_config_t config = {
2018-09-24 05:06:44 -04:00
.pseltxd = (tx == mp_const_none) ? NRF_UART_PSEL_DISCONNECTED : tx->number,
.pselrxd = (rx == mp_const_none) ? NRF_UART_PSEL_DISCONNECTED : rx->number,
2018-09-19 06:59:15 -04:00
.pselcts = NRF_UART_PSEL_DISCONNECTED,
.pselrts = NRF_UART_PSEL_DISCONNECTED,
.p_context = self,
.hwfc = NRF_UART_HWFC_DISABLED,
.parity = (parity == PARITY_NONE) ? NRF_UART_PARITY_EXCLUDED : NRF_UART_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
};
2018-09-24 05:06:44 -04:00
// support only 1 instance for now
self->uarte = (nrfx_uarte_t ) NRFX_UARTE_INSTANCE(0);
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 ) {
self->buffer = (uint8_t *) gc_alloc(receiver_buffer_size, false, false);
if ( !self->buffer ) {
nrfx_uarte_uninit(&self->uarte);
mp_raise_msg(&mp_type_MemoryError, translate("Failed to allocate RX buffer"));
}
self->bufsize = receiver_buffer_size;
claim_pin(rx);
}
if ( tx != mp_const_none ) {
claim_pin(tx);
2018-09-19 06:59:15 -04:00
}
self->baudrate = baudrate;
self->timeout_ms = timeout;
#endif
}
bool common_hal_busio_uart_deinited(busio_uart_obj_t *self) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
2018-09-19 06:59:15 -04:00
#else
2018-09-24 05:06:44 -04:00
return (nrf_uarte_rx_pin_get(self->uarte.p_reg) == NRF_UART_PSEL_DISCONNECTED) &&
(nrf_uarte_tx_pin_get(self->uarte.p_reg) == NRF_UART_PSEL_DISCONNECTED);
2018-09-19 06:59:15 -04:00
#endif
}
void common_hal_busio_uart_deinit(busio_uart_obj_t *self) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
2018-09-19 06:59:15 -04:00
#else
if ( !common_hal_busio_uart_deinited(self) ) {
2018-09-24 05:06:44 -04:00
nrfx_uarte_uninit(&self->uarte);
2018-09-24 03:37:28 -04:00
gc_free(self->buffer);
}
2018-09-19 06:59:15 -04:00
#endif
}
// Read characters.
size_t common_hal_busio_uart_read(busio_uart_obj_t *self, uint8_t *data, size_t len, int *errcode) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
return 0;
2018-09-19 06:59:15 -04:00
#else
2018-09-24 05:06:44 -04:00
if ( nrf_uarte_rx_pin_get(self->uarte.p_reg) == NRF_UART_PSEL_DISCONNECTED ) {
mp_raise_ValueError(translate("No RX pin"));
}
2018-09-20 16:48:13 -04:00
size_t remain = len;
uint64_t start_ticks = ticks_ms;
2018-09-24 04:54:32 -04:00
while ( remain && (ticks_ms - start_ticks < self->timeout_ms) ) {
const size_t cnt = MIN(self->bufsize, len);
2018-09-20 14:27:52 -04:00
2018-09-24 04:54:32 -04:00
self->rx_count = -1;
2018-09-24 05:06:44 -04:00
_VERIFY_ERR(nrfx_uarte_rx(&self->uarte, self->buffer, cnt));
2018-09-20 16:48:13 -04:00
2018-09-24 04:54:32 -04:00
while ( (self->rx_count == -1) && (ticks_ms - start_ticks < self->timeout_ms) ) {
2018-09-20 16:48:13 -04:00
#ifdef MICROPY_VM_HOOK_LOOP
MICROPY_VM_HOOK_LOOP
#endif
2018-09-24 04:54:32 -04:00
}
2018-09-20 16:48:13 -04:00
2018-09-24 04:54:32 -04:00
// Time up, abort rx use received so far
if ( self->rx_count == -1 ) {
2018-09-24 05:06:44 -04:00
nrfx_uarte_rx_abort(&self->uarte);
2018-09-24 04:54:32 -04:00
while ( self->rx_count == -1 ) {
}
}
2018-09-20 16:48:13 -04:00
2018-09-24 04:54:32 -04:00
if ( self->rx_count > 0 ) {
memcpy(data, self->buffer, self->rx_count);
data += self->rx_count;
remain -= self->rx_count;
2018-09-20 16:48:13 -04:00
}
}
2018-09-24 04:54:32 -04:00
2018-09-19 14:07:45 -04:00
return len - remain;
2018-09-19 06:59:15 -04:00
#endif
}
// Write characters.
size_t common_hal_busio_uart_write(busio_uart_obj_t *self, const uint8_t *data, size_t len, int *errcode) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
return 0;
2018-09-19 06:59:15 -04:00
#else
2018-09-24 05:06:44 -04:00
if ( nrf_uarte_tx_pin_get(self->uarte.p_reg) == NRF_UART_PSEL_DISCONNECTED ) {
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-24 05:06:44 -04:00
if ( !nrfx_uarte_tx_in_progress(&self->uarte) ) {
nrfx_uarte_tx_abort(&self->uarte);
2018-09-24 04:54:32 -04:00
}
2018-09-24 05:06:44 -04:00
(*errcode) = nrfx_uarte_tx(&self->uarte, data, len);
2018-09-19 14:07:45 -04:00
_VERIFY_ERR(*errcode);
2018-09-19 06:59:15 -04:00
(*errcode) = 0;
uint64_t start_ticks = ticks_ms;
2018-09-24 05:06:44 -04:00
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
}
return len;
#endif
}
uint32_t common_hal_busio_uart_get_baudrate(busio_uart_obj_t *self) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
2018-09-19 06:59:15 -04:00
#endif
return self->baudrate;
}
void common_hal_busio_uart_set_baudrate(busio_uart_obj_t *self, uint32_t baudrate) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
2018-09-19 06:59:15 -04:00
#else
self->baudrate = baudrate;
2018-09-24 05:06:44 -04:00
nrf_uarte_baudrate_set(self->uarte.p_reg, get_nrf_baud(baudrate));
2018-09-19 06:59:15 -04:00
#endif
}
uint32_t common_hal_busio_uart_rx_characters_available(busio_uart_obj_t *self) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
2018-09-19 06:59:15 -04:00
#else
2018-09-24 05:06:44 -04:00
return 1; // nrf_uart_event_check(self->uarte.p_reg, NRF_UART_EVENT_RXDRDY) ? 1 : 0;
2018-09-19 06:59:15 -04:00
#endif
}
void common_hal_busio_uart_clear_rx_buffer(busio_uart_obj_t *self) {
2018-09-24 05:06:44 -04:00
}
bool common_hal_busio_uart_ready_to_tx(busio_uart_obj_t *self) {
2018-09-19 06:59:15 -04:00
#ifndef NRF52840_XXAA
2018-08-16 17:21:44 -04:00
mp_raise_NotImplementedError(translate("busio.UART not yet implemented"));
return false;
2018-09-19 06:59:15 -04:00
#else
2018-09-24 05:06:44 -04:00
return !nrfx_uarte_tx_in_progress(&self->uarte);
2018-09-19 06:59:15 -04:00
#endif
}
static uint32_t get_nrf_baud (uint32_t baudrate)
{
if ( baudrate <= 1200 ) {
return NRF_UART_BAUDRATE_1200;
}
else if ( baudrate <= 2400 ) {
return NRF_UART_BAUDRATE_2400;
}
else if ( baudrate <= 4800 ) {
return NRF_UART_BAUDRATE_4800;
}
else if ( baudrate <= 9600 ) {
return NRF_UART_BAUDRATE_9600;
}
else if ( baudrate <= 14400 ) {
return NRF_UART_BAUDRATE_14400;
}
else if ( baudrate <= 19200 ) {
return NRF_UART_BAUDRATE_19200;
}
else if ( baudrate <= 28800 ) {
return NRF_UART_BAUDRATE_28800;
}
else if ( baudrate <= 38400 ) {
return NRF_UART_BAUDRATE_38400;
}
else if ( baudrate <= 57600 ) {
return NRF_UART_BAUDRATE_57600;
}
else if ( baudrate <= 76800 ) {
return NRF_UART_BAUDRATE_76800;
}
else if ( baudrate <= 115200 ) {
return NRF_UART_BAUDRATE_115200;
}
else if ( baudrate <= 230400 ) {
return NRF_UART_BAUDRATE_230400;
}
else if ( baudrate <= 250000 ) {
return NRF_UART_BAUDRATE_250000;
}
else if ( baudrate <= 460800 ) {
return NRF_UART_BAUDRATE_460800;
}
else if ( baudrate <= 921600 ) {
return NRF_UART_BAUDRATE_921600;
}
else {
return NRF_UART_BAUDRATE_1000000;
}
}