diff options
Diffstat (limited to 'stmhal/usart.c')
| -rw-r--r-- | stmhal/usart.c | 269 |
1 files changed, 269 insertions, 0 deletions
diff --git a/stmhal/usart.c b/stmhal/usart.c new file mode 100644 index 0000000000..16b23240a8 --- /dev/null +++ b/stmhal/usart.c @@ -0,0 +1,269 @@ +#include <stdio.h> +#include <string.h> +#include <stm32f4xx_hal.h> + +#include "misc.h" +#include "mpconfig.h" +#include "qstr.h" +#include "obj.h" +#include "usart.h" + +pyb_usart_t pyb_usart_global_debug = PYB_USART_NONE; + +#if 0 +#else +// This needs to be fixed. Right now its just a hack to get REPL working +static UART_HandleTypeDef UartHandle; +#endif + +#if 0 +static USART_TypeDef *usart_get_base(pyb_usart_t usart_id) { + USART_TypeDef *USARTx=NULL; + + switch (usart_id) { + case PYB_USART_NONE: + break; + case PYB_USART_1: + USARTx = USART1; + break; + case PYB_USART_2: + USARTx = USART2; + break; + case PYB_USART_3: + USARTx = USART3; + break; + case PYB_USART_6: + USARTx = USART6; + break; + } + + return USARTx; +} +#endif + +void usart_init(pyb_usart_t usart_id, uint32_t baudrate) { + USART_TypeDef *USARTx=NULL; + + uint32_t GPIO_Pin=0; + uint8_t GPIO_AF_USARTx=0; + GPIO_TypeDef* GPIO_Port=NULL; + + switch (usart_id) { + case PYB_USART_NONE: + return; + + case PYB_USART_1: + USARTx = USART1; + + GPIO_Port = GPIOA; + GPIO_AF_USARTx = GPIO_AF7_USART1; + GPIO_Pin = GPIO_PIN_9 | GPIO_PIN_10; + + __USART1_CLK_ENABLE(); + break; + case PYB_USART_2: + USARTx = USART2; + + GPIO_Port = GPIOD; + GPIO_AF_USARTx = GPIO_AF7_USART2; + GPIO_Pin = GPIO_PIN_5 | GPIO_PIN_6; + + __USART2_CLK_ENABLE(); + break; + case PYB_USART_3: + USARTx = USART3; + +#if defined(PYBOARD3) || defined(PYBOARD4) + GPIO_Port = GPIOB; + GPIO_AF_USARTx = GPIO_AF7_USART3; + GPIO_Pin = GPIO_PIN_10 | GPIO_PIN_11; +#else + GPIO_Port = GPIOD; + GPIO_AF_USARTx = GPIO_AF7_USART3; + GPIO_Pin = GPIO_PIN_8 | GPIO_PIN_9; +#endif + __USART3_CLK_ENABLE(); + break; + case PYB_USART_6: + USARTx = USART6; + + GPIO_Port = GPIOC; + GPIO_AF_USARTx = GPIO_AF8_USART6; + GPIO_Pin = GPIO_PIN_6 | GPIO_PIN_7; + + __USART6_CLK_ENABLE(); + break; + } + + /* Initialize USARTx */ + + GPIO_InitTypeDef GPIO_InitStructure; + GPIO_InitStructure.Pin = GPIO_Pin; + GPIO_InitStructure.Speed = GPIO_SPEED_HIGH; + GPIO_InitStructure.Mode = GPIO_MODE_AF_PP; + GPIO_InitStructure.Pull = GPIO_PULLUP; + GPIO_InitStructure.Alternate = GPIO_AF_USARTx; + HAL_GPIO_Init(GPIO_Port, &GPIO_InitStructure); + + memset(&UartHandle, 0, sizeof(UartHandle)); + UartHandle.Instance = USARTx; + UartHandle.Init.BaudRate = baudrate; + UartHandle.Init.WordLength = USART_WORDLENGTH_8B; + UartHandle.Init.StopBits = USART_STOPBITS_1; + UartHandle.Init.Parity = USART_PARITY_NONE; + UartHandle.Init.Mode = USART_MODE_TX_RX; + UartHandle.Init.HwFlowCtl = UART_HWCONTROL_NONE; + UartHandle.Init.OverSampling = UART_OVERSAMPLING_16; + HAL_UART_Init(&UartHandle); +} + +bool usart_rx_any(pyb_usart_t usart_id) { +#if 0 + USART_TypeDef *USARTx = usart_get_base(usart_id); + return USART_GetFlagStatus(USARTx, USART_FLAG_RXNE) == SET; +#else + return __HAL_UART_GET_FLAG(&UartHandle, USART_FLAG_RXNE); +#endif +} + +int usart_rx_char(pyb_usart_t usart_id) { +#if 0 + USART_TypeDef *USARTx = usart_get_base(usart_id); + return USART_ReceiveData(USARTx); +#else + uint8_t ch; + if (HAL_UART_Receive(&UartHandle, &ch, 1, 0) != HAL_OK) { + ch = 0; + } + return ch; +#endif +} + +void usart_tx_char(pyb_usart_t usart_id, int c) { +#if 0 + USART_TypeDef *USARTx = usart_get_base(usart_id); + // wait until the end of any previous transmission + uint32_t timeout = 100000; + while (USART_GetFlagStatus(USARTx, USART_FLAG_TC) == RESET && --timeout > 0) { + } + USART_SendData(USARTx, c); +#else + uint8_t ch = c; + HAL_UART_Transmit(&UartHandle, &ch, 1, 100000); +#endif +} + +void usart_tx_str(pyb_usart_t usart_id, const char *str) { + for (; *str; str++) { + usart_tx_char(usart_id, *str); + } +} + +void usart_tx_bytes(pyb_usart_t usart_id, const char *data, uint len) { + for (; len > 0; data++, len--) { + usart_tx_char(usart_id, *data); + } +} + +void usart_tx_strn_cooked(pyb_usart_t usart_id, const char *str, int len) { + for (const char *top = str + len; str < top; str++) { + if (*str == '\n') { + usart_tx_char(usart_id, '\r'); + } + usart_tx_char(usart_id, *str); + } +} + +#if 0 +/******************************************************************************/ +/* Micro Python bindings */ + +typedef struct _pyb_usart_obj_t { + mp_obj_base_t base; + pyb_usart_t usart_id; + bool is_enabled; +} pyb_usart_obj_t; + +static mp_obj_t usart_obj_status(mp_obj_t self_in) { + pyb_usart_obj_t *self = self_in; + if (usart_rx_any(self->usart_id)) { + return mp_const_true; + } else { + return mp_const_false; + } +} + +static mp_obj_t usart_obj_rx_char(mp_obj_t self_in) { + mp_obj_t ret = mp_const_none; + pyb_usart_obj_t *self = self_in; + + if (self->is_enabled) { + ret = mp_obj_new_int(usart_rx_char(self->usart_id)); + } + return ret; +} + +static mp_obj_t usart_obj_tx_char(mp_obj_t self_in, mp_obj_t c) { + pyb_usart_obj_t *self = self_in; + uint len; + const char *str = mp_obj_str_get_data(c, &len); + if (len == 1 && self->is_enabled) { + usart_tx_char(self->usart_id, str[0]); + } + return mp_const_none; +} + +static mp_obj_t usart_obj_tx_str(mp_obj_t self_in, mp_obj_t s) { + pyb_usart_obj_t *self = self_in; + if (self->is_enabled) { + if (MP_OBJ_IS_STR(s)) { + uint len; + const char *data = mp_obj_str_get_data(s, &len); + usart_tx_bytes(self->usart_id, data, len); + } + } + return mp_const_none; +} + +static void usart_obj_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in, mp_print_kind_t kind) { + pyb_usart_obj_t *self = self_in; + print(env, "<Usart %lu>", self->usart_id); +} + +static MP_DEFINE_CONST_FUN_OBJ_1(usart_obj_status_obj, usart_obj_status); +static MP_DEFINE_CONST_FUN_OBJ_1(usart_obj_rx_char_obj, usart_obj_rx_char); +static MP_DEFINE_CONST_FUN_OBJ_2(usart_obj_tx_char_obj, usart_obj_tx_char); +static MP_DEFINE_CONST_FUN_OBJ_2(usart_obj_tx_str_obj, usart_obj_tx_str); + +STATIC const mp_method_t usart_methods[] = { + { "status", &usart_obj_status_obj }, + { "recv_chr", &usart_obj_rx_char_obj }, + { "send_chr", &usart_obj_tx_char_obj }, + { "send", &usart_obj_tx_str_obj }, + { NULL, NULL }, +}; + +STATIC const mp_obj_type_t usart_obj_type = { + { &mp_type_type }, + .name = MP_QSTR_Usart, + .print = usart_obj_print, + .methods = usart_methods, +}; + +STATIC mp_obj_t pyb_Usart(mp_obj_t usart_id, mp_obj_t baudrate) { + if (mp_obj_get_int(usart_id)>PYB_USART_MAX) { + return mp_const_none; + } + + /* init USART */ + usart_init(mp_obj_get_int(usart_id), mp_obj_get_int(baudrate)); + + pyb_usart_obj_t *o = m_new_obj(pyb_usart_obj_t); + o->base.type = &usart_obj_type; + o->usart_id = mp_obj_get_int(usart_id); + o->is_enabled = true; + return o; +} + +MP_DEFINE_CONST_FUN_OBJ_2(pyb_Usart_obj, pyb_Usart); +#endif |