#include <avr/io.h>
#include <stdio.h>
#include "UART.h"

void Uart_Init(  uint8_t Ubrr, uint8_t Uart0_Enable, uint8_t Uart1_Enable)
{
	/* Set baud rate */
	UBRR0H = (uint8_t)(Ubrr>>8);
	UBRR0L = (uint8_t)Ubrr;
	UBRR1H = (uint8_t)(Ubrr>>8);
	UBRR1L = (uint8_t)Ubrr;
	if (Uart0_Enable)
	{
		UCSR0B |= (1<<RXEN0)|(1<<TXEN0);	/* Enable rx, tx */
		UCSR0B |= (1<<RXCIE0);				/* Enable rx interrupt */
		UCSR0C = (3<<UCSZ00);				/* Set frame format: 8 data, 1 stop bit */
		DDRD |= (1<<PORTD1);				/* set TX0 pin to output */
	}
	if (Uart1_Enable)
	{
		UCSR1B |= (1<<RXEN1)|(1<<TXEN1);	/* Enable rx, tx */
		UCSR1B |= (1<<RXCIE1);				/* Enable rx interrupt */
		UCSR1C = (3<<UCSZ10);				/* Set frame format: 8 data, 1 stop bit */
		DDRD |= (1<<PORTD3);				/* set TX0 pin to output */
	}
}

void Uart0_Transmit8bit(uint8_t Data)
{
	while (!(UCSR0A & (1<<UDRE0)));			/* Wait for empty transmit buffer */
	UDR0 = Data;							/* Put data into buffer, sends the data */
}

void Uart1_Transmit8bit(uint8_t Data)
{
	while (!(UCSR1A & (1<<UDRE1)));			/* Wait for empty transmit buffer */
	UDR1 = Data;							/* Put data into buffer, sends the data */
}