Small changes. Convert TABs to spaces.

[irmp-demo.git] / serial.c

/*
 * (C) Copyright 2014 - 2017 Leo C. <erbl259-lmu@yahoo.de>
 *
 * SPDX-License-Identifier:	GPL-2.0
 */

#include "serial.h"
#include "config.h"
#include <errno.h>
#include <stdio.h>
#include <unistd.h>
#include <libopencm3/stm32/rcc.h>
#include <libopencm3/stm32/gpio.h>
#include <libopencm3/stm32/usart.h>
#include <libopencm3/cm3/nvic.h>

#if (USART_CONSOLE == 1)
#define CON_PORT A
#define CON_GPIO GPIOA
#elif (USART_CONSOLE == 2)
#define CON_PORT A
#define CON_GPIO GPIOA
#elif (USART_CONSOLE == 3)
#define CON_PORT B
#define CON_GPIO GPIOB
#else
#error  USART_CONSOLE undefined or invalid value
#endif

#define CONCAT2(a,b)    a##b
#define CONCAT(a,b)     CONCAT2(a,b)


#define CON_USART       CONCAT(USART,USART_CONSOLE)
#define CON_IRQ         CONCAT(CONCAT(NVIC_USART, USART_CONSOLE), _IRQ)
#define CON_isr         CONCAT(CONCAT(usart, USART_CONSOLE), _isr)
#define CON_GPIO_RX     CONCAT(CONCAT(GPIO_USART, USART_CONSOLE), _RX)
#define CON_GPIO_TX     CONCAT(CONCAT(GPIO_USART, USART_CONSOLE), _TX)
#define RCC_USART_CON   CONCAT(RCC_USART, USART_CONSOLE)
#define RCC_GPIO_CON    CONCAT(RCC_GPIO, CON_PORT)


int _write(int fd, char *ptr, int len)  __attribute__((used));


struct ring {
	uint8_t *data;
	int size;
	volatile int begin;
	volatile int end;
};


#define BUFFER_SIZE 256

struct ring rx_ring;
struct ring tx_ring;
uint8_t rx_ring_buffer[BUFFER_SIZE];
uint8_t tx_ring_buffer[BUFFER_SIZE];


static void ring_init(struct ring *ring, uint8_t *buf, int size)
{
	ring->data = buf;
	ring->size = size;
	ring->begin = 0;
	ring->end = 0;
}

static int ring_write_ch(struct ring *ring, uint8_t ch)
{
	int ep = (ring->end + 1) % ring->size;

	if ((ep) != ring->begin) {
		ring->data[ring->end] = ch;
		ring->end = ep;
		return 1;
	}

	return -1;
}

#if 0 /* unused */
static int ring_write(struct ring *ring, uint8_t *data, int size)
{
	int i;

	for (i = 0; i < size; i++) {
		if (ring_write_ch(ring, data[i]) < 0)
			return -i;
	}

	return i;
}
#endif

static int ring_read_ch(struct ring *ring)
{
	int ret = -1;

	if (ring->begin != ring->end) {
		ret = ring->data[ring->begin];
		ring->begin = (ring->begin +1) % ring->size;
	}

	return ret;
}

void usart_setup(int baud)
{
	/* Initialize output ring buffer. */
	ring_init(&rx_ring, rx_ring_buffer, BUFFER_SIZE);
	ring_init(&tx_ring, tx_ring_buffer, BUFFER_SIZE);

    rcc_periph_clock_enable(RCC_USART_CON);

	/* Enable the USART_CONSOLE interrupt. */
	nvic_enable_irq(CON_IRQ);
    nvic_set_priority(CON_IRQ, 6*16);

	/* Setup GPIO pins for CONSOLE TX/RX */
    rcc_periph_clock_enable(RCC_GPIO_CON);
	gpio_set_mode(CON_GPIO, GPIO_MODE_OUTPUT_2_MHZ,
                GPIO_CNF_OUTPUT_ALTFN_PUSHPULL, CON_GPIO_TX);
	gpio_set_mode(CON_GPIO, GPIO_MODE_INPUT,
                GPIO_CNF_INPUT_FLOAT, CON_GPIO_RX);

	/* Setup UART parameters. */
	usart_set_baudrate(CON_USART, baud);
	usart_set_databits(CON_USART, 8);
	usart_set_stopbits(CON_USART, USART_STOPBITS_1);
	usart_set_parity(CON_USART, USART_PARITY_NONE);
	usart_set_flow_control(CON_USART, USART_FLOWCONTROL_NONE);
	usart_set_mode(CON_USART, USART_MODE_TX_RX);

	/* Enable CONSOLE Receive interrupt. */
	USART_CR1(CON_USART) |= USART_CR1_RXNEIE;

	/* Finally enable the USART. */
	usart_enable(CON_USART);
}


void CON_isr(void)
{
	/* Check if we were called because of RXNE. */
	if (((USART_CR1(CON_USART) & USART_CR1_RXNEIE) != 0) &&
	    ((USART_SR(CON_USART) & USART_SR_RXNE) != 0)) {

		/* Retrieve the data from the peripheral. */
		ring_write_ch(&rx_ring, USART_DR(CON_USART) & USART_DR_MASK);

	}

	/* Check if we were called because of TXE. */
	if (((USART_CR1(CON_USART) & USART_CR1_TXEIE) != 0) &&
	    ((USART_SR(CON_USART) & USART_SR_TXE) != 0)) {

		int data;

		data = ring_read_ch(&tx_ring);

		if (data == -1) {
			/* Disable the TXE interrupt, it's no longer needed. */
			USART_CR1(CON_USART) &= ~USART_CR1_TXEIE;
		} else {
			/* Put data into the transmit register. */
			usart_send(CON_USART, data);
		}
	}
}

/*--------------------------------------------------------------------------*/

void serial_setup(int baud)
{
	usart_setup(baud);
}

void serial_setbaudrate(int baud)
{
	usart_set_baudrate(CON_USART, baud);
}

/*--------------------------------------------------------------------------*/

/**
 * Use USART_CONSOLE as a console.
 * This is a syscall for newlib
 * @param fd
 * @param ptr
 * @param len
 * @return
 */
int _write(int fd, char *ptr, int len)
{
	int i;

	if (fd == STDOUT_FILENO || fd == STDERR_FILENO) {
		for (i = 0; i < len; i++) {
			if (ptr[i] == '\n') {
				serial_putc('\r');
			}
			serial_putc(ptr[i]);
		}
		return i;
	}
	errno = EIO;
	return -1;
}

int serial_getc(void)
{
	return ring_read_ch(&rx_ring);
}

void serial_putc(int data)
{
	while (ring_write_ch(&tx_ring, data) < 0)
		;

	/* Enable the TXE interrupt. */
	USART_CR1(CON_USART) |= USART_CR1_TXEIE;
}