X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=common%2Fserial.c;h=8874c3e07545f1d4b52cd1e60df6c003b6da1d64;hb=5d6f425971266accc2e3d1c6de96a0ba031adb8d;hp=2bf6901836064ece4517047dcb8d94f70838d99f;hpb=9c37d18b9c70fdb70dfec453398c4649e9e57586;p=model-railway-devices.git

diff --git a/common/serial.c b/common/serial.c
index 2bf6901..8874c3e 100644
--- a/common/serial.c
+++ b/common/serial.c
@@ -1,8 +1,23 @@
+#include <avr/interrupt.h>
 #include <avr/io.h>
+#include "ringbuffer.h"
 #include "serial.h"
 
 #define BIT(n) (1<<(n))
 
+#ifndef SERIAL_BUFSIZE
+#define SERIAL_BUFSIZE 16
+#endif
+
+#ifdef SERIAL_ASYNC
+static RINGBUFFER(rx, SERIAL_BUFSIZE);
+static RINGBUFFER(tx, SERIAL_BUFSIZE);
+static uint8_t rx_overrun;
+#define INTERRUPTS BIT(RXCIE0)
+#else
+#define INTERRUPTS 0
+#endif
+
 void serial_init(uint16_t baud)
 {
 	DDRD = (DDRD&~0x03) | 0x02;
@@ -11,12 +26,76 @@ void serial_init(uint16_t baud)
 	baud = (F_CPU/16+baud/2)/baud-1;
 	UBRR0H = baud>>8;
 	UBRR0L = baud;
+	UCSR0A = 0;
 	UCSR0C = BIT(UCSZ00) | BIT(UCSZ01);  // 8N1
-	UCSR0B = BIT(RXEN0) | BIT(TXEN0) | BIT(RXCIE0);
+	UCSR0B = BIT(RXEN0) | BIT(TXEN0) | INTERRUPTS;
+}
+
+uint8_t serial_read()
+{
+#ifdef SERIAL_ASYNC
+	return ringbuffer_blocking_pop(rx);
+#else
+	while(!(UCSR0A&BIT(RXC0))) ;
+	return UDR0;
+#endif
+}
+
+uint8_t serial_read_available()
+{
+#ifdef SERIAL_ASYNC
+	return ringbuffer_fill(rx);
+#else
+	return (UCSR0A&BIT(RXC0))!=0;
+#endif
+}
+
+uint8_t serial_read_overrun()
+{
+#ifdef SERIAL_ASYNC
+	uint8_t o = rx_overrun;
+	rx_overrun = 0;
+	return o;
+#else
+	return (UCSR0A&BIT(DOR0))!=0;
+#endif
 }
 
 void serial_write(uint8_t c)
 {
-	while(!(UCSR0A&(1<<UDRE0))) ;
+#ifdef SERIAL_ASYNC
+	ringbuffer_blocking_push(tx, c);
+	UCSR0B |= BIT(UDRIE0);
+#else
+	while(!(UCSR0A&BIT(UDRE0))) ;
 	UDR0 = c;
+#endif
+}
+
+uint8_t serial_write_space()
+{
+#ifdef SERIAL_ASYNC
+	return ringbuffer_space(tx);
+#else
+	return (UCSR0A&BIT(UDRE0))!=0;
+#endif
+}
+
+#ifdef SERIAL_ASYNC
+ISR(USART_UDRE_vect)
+{
+	if(ringbuffer_fill(tx))
+		UDR0 = ringbuffer_pop(tx);
+	else
+		UCSR0B &= ~BIT(UDRIE0);
+}
+
+ISR(USART_RX_vect)
+{
+	uint8_t c = UDR0;
+	if(!ringbuffer_space(rx))
+		rx_overrun = 1;
+	else
+		ringbuffer_push(rx, c);
 }
+#endif