#include <mega48.h>

// Alphanumeric LCD Module functions
#include <alcd.h>

// Standard Input/Output functions
#include <stdio.h>

#include <delay.h>


#define LCD_CHARS_LINE 16

// Bit definitions from the USART control registers
#define RXB8 1
#define TXB8 0
#define UPE 2
#define OVR 3
#define FE 4
#define UDRE 5
#define RXC 7

#define FRAMING_ERROR (1<<FE)
#define PARITY_ERROR (1<<UPE)
#define DATA_OVERRUN (1<<OVR)
#define DATA_REGISTER_EMPTY (1<<UDRE)
#define RX_COMPLETE (1<<RXC)

// Specify that a new putchar function will be used instead of the one from stdio.h
#define _ALTERNATE_PUTCHAR_


// Specify the output types
#define USART0 0
#define LCD 2

// This variable will specify to which peripheral the output of putchar will be directed
unsigned char poutput;

void putchar(char c)
{
switch (poutput)
       {
       case USART0: // the output will be directed to USART0
            while ((UCSR0A & DATA_REGISTER_EMPTY)==0);
            UDR0=c;
       break;

      
       case LCD:    // the output will be directed to the LCD
            lcd_putchar©;
       };
}

void main(void)
{

char k;


// Crystal Oscillator division factor: 1
#pragma optsize-
CLKPR=0x80;
CLKPR=0x00;
#ifdef _OPTIMIZE_SIZE_
#pragma optsize+
#endif


// USART initialization
// Communication Parameters: 8 Data, 1 Stop, No Parity
// USART Receiver: On
// USART Transmitter: On
// USART0 Mode: Asynchronous
// USART Baud Rate: 9600
UCSR0A=0x00;
UCSR0B=0x18;
UCSR0C=0x06;
UBRR0H=0x00;
UBRR0L=0x33;

lcd_init(16);
  
while (1)
    {
    poutput=USART0;
    //printf("This is a test of USART%u.\r",0);
    k=getchar();
    putchar(k);
    //UDR0=3;
    delay_ms(100);
    poutput=LCD;
    lcd_clear();
    printf("-- %c --\n%u characters/line.",k,LCD_CHARS_LINE);
    }
}

#include <avr\io.h>
#include <util\delay.h>
#include <stdlib.h>
//#include <avr\interrupt.h>    

void USART_Transmit( unsigned char data )
{
// Wait for empty transmit buffer
while ( !( UCSR0A & (1<<UDRE0)) )
;
// Put data into buffer, sends the data
UDR0 = data;
}

unsigned char USART_Receive( void )
{
/* Wait for data to be received */
while ( !(UCSR0A & (1<<RXC0)) )
;
/* Get and return received data from buffer */
return UDR0;
}

void USART_Init( unsigned int baud )
{
// Set baud rate
UBRR0H = (unsigned char)(baud>>8);
UBRR0L = (unsigned char)baud;
// Enable receiver and transmitter
UCSR0B = (1<<RXEN0)|(1<<TXEN0);
// Set frame format: 8data, 1stop bit
UCSR0C = (0<<USBS0)|(3<<UCSZ00);
}

int main (void)
{

DDRB=0x00;

USART_Init(9600);

while(1)
    {
    //USART_Transmit(USART_Receive());
    USART_Transmit('T');
    PORTB=0x00;
    _delay_ms(500);
    PORTB=0xFF;
    _delay_ms(500);
    }
}