#include <avr/io.h>
#include <avr/pgmspace.h>
#include <util/delay.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdlib.h>

#include "calc.h"
#include "uart.h"
#include "stream.h"

// Shared stream instance
static STREAM _uart_singleton;
STREAM* uart;


void _uart_init_do(uint16_t ubrr) {
	/*Set baud rate */
	UBRR0H = (uint8_t) (ubrr >> 8);
	UBRR0L = (uint8_t) ubrr;

	// Enable Rx and Tx
	UCSR0B = (1 << RXEN0) | (1 << TXEN0);

	// 8-bit data, 1 stop bit
	UCSR0C = (0b11 << UCSZ00);

	_uart_singleton.tx = &uart_tx;
	_uart_singleton.rx = &uart_rx;

	uart = &_uart_singleton;
}


/** Enable or disable RX ISR */
void uart_isr_rx(bool yes)
{
	set_bit(UCSR0B, RXCIE0, yes);
}


/** Enable or disable TX ISR (1 byte is sent) */
void uart_isr_tx(bool yes)
{
	set_bit(UCSR0B, TXCIE0, yes);
}


/** Enable or disable DRE ISR (all is sent) */
void uart_isr_dre(bool yes)
{
	set_bit(UCSR0B, UDRIE0, yes);
}


/** Send byte over UART */
void uart_tx(uint8_t data)
{
	// Wait for transmit buffer
	while (!uart_tx_ready());
	// send it
	UDR0 = data;
}


/** Receive one byte over UART */
uint8_t uart_rx()
{
	// Wait for data to be received
	while (!uart_rx_ready());
	// Get and return received data from buffer
	return UDR0;
}


/** Send string over UART */
void uart_puts(const char* str)
{
	while (*str) {
		uart_tx(*str++);
	}
}


/** Send progmem string over UART */
void uart_puts_P(const char* str)
{
	char c;
	while ((c = pgm_read_byte(str++))) {
		uart_tx(c);
	}
}


/** Clear receive buffer */
void uart_flush()
{
	uint8_t dummy;
	while (bit_is_high(UCSR0A, RXC0)) {
		dummy = UDR0;
	}
}