//Emil Fihlman
#include <stdtypes.h>
#define F_CPU 16000000UL
#include <avr/io.h>
#include <util/delay.h>
#include <string.h>
#include "font.c"

u8 i, j, k=0, rxb[5]={0}, c, b=0x00;

u8 d3[8]=
{
	0b10000000,
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000001
};

u8 d2[8]=
{
	0b00000001,
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000000,
	0b10000000
};

u8 d1[8]=
{
	0b00000000,
	0b00000000,
	0b00000000,
	0b10000000,
	0b10000000,
	0b00000000,
	0b00000000,
	0b00000000
};

u8 d0[8]=
{
	0b00000000,
	0b00000000,
	0b00000000,
	0b00000001,
	0b00000001,
	0b00000000,
	0b00000000,
	0b00000000
};

u8 *d[4]=
{
	d0,
	d1,
	d2,
	d3
};

void uart_init(void)
{
	UCSR0A=0b00000000;
	UCSR0B=0b00011000;
	UCSR0C=0b00000110;
	UBRR0H=0b00000000;
	UBRR0L=103;
}

u8 uart_getc(void)
{
	loop_until_bit_is_set(UCSR0A, RXC0);
	return UDR0;
}

void uart_putc(u8 c)
{
	loop_until_bit_is_set(UCSR0A, UDRE0);
	UDR0=c;
}

void uart_puts(u8 *s)
{
	while(*s)
	{
		uart_putc(*s++);
	}
}

u8 spit(u8 c)
{
	SPDR=c;
	while(!(SPSR&0b10000000)){};
	return(SPDR);
}

void spitc(u8 addr, u8 comm, u8 chip)
{
	u8 i;
	PORTB&=0b11111011;
	for(i=0; i<(3-chip); i++)
	{
		spit(0x00);
		spit(0x00);
	}
	spit(addr);
	spit(comm);
	for(i=0; i<chip; i++)
	{
		spit(0x00);
		spit(0x00);
	}
	PORTB|=0b00000100;
}

void spitd(u8 addr, u8 data, u8 chip)
{
	u8 i;
	PORTB&=0b11111011;
	for(i=0; i<(3-chip); i++)
	{
		spit(0x00);
		spit(0x00);
	}
	spit(addr);
	spit(data);
	for(i=0; i<chip; i++)
	{
		spit(0x00);
		spit(0x00);
	}
	PORTB|=0b00000100;
}

void setledsymbol(u8 c, u8 i)
{
	if('0'<=c && c<='9')
	{
		memcpy(d[i], symbols+8*(c-'0'), 8);
	}
	else if('a'<=c && c<='z')
	{
		memcpy(d[i], symbols+8*(c+10-'a'), 8);
	}
	else if('A'<=c && c<='Z')
	{
		memcpy(d[i], symbols+8*(c+10-'A'), 8);
	}
	else
	{
		memset(d[i], 0, 8);
	}
}

void setleds(char *c)
{
	u8 i, j;
	for(i=0; i<4; i++)
	{
		setledsymbol(*(c+i), i);
	}
	for(j=0; j<4; j++)
	{
		for(i=0; i<8; i++)
		{
			spitd(i+1, d[3-j][i], j);
		}
	}
}

void led_init(void)
{
	spitc(0x0f, 0x00, 0);
	spitc(0x09, 0x00, 0);
	spitc(0x0a, 0x00, 0);
	spitc(0x0b, 0x07, 0);
	spitc(0x0c, 0x01, 0);
	spitc(0x0f, 0x00, 1);
	spitc(0x09, 0x00, 1);
	spitc(0x0a, 0x00, 1);
	spitc(0x0b, 0x07, 1);
	spitc(0x0c, 0x01, 1);
	spitc(0x0f, 0x00, 2);
	spitc(0x09, 0x00, 2);
	spitc(0x0a, 0x00, 2);
	spitc(0x0b, 0x07, 2);
	spitc(0x0c, 0x01, 2);
	spitc(0x0f, 0x00, 3);
	spitc(0x09, 0x00, 3);
	spitc(0x0a, 0x00, 3);
	spitc(0x0b, 0x07, 3);
	spitc(0x0c, 0x01, 3);
	for(j=0; j<4; j++)
	{
		spitc(0x0a, b, j);
		for(i=0; i<8; i++)
		{
			spitd(i+1, d[j][i], j);
		}
	}
}

s16 main(void)
{
	DDRB=0b11111111;
	PORTB=0b11111111;
	SPCR=0b01010000;
	SPSR=0b00000001;

	uart_init();
	led_init();

	_delay_ms(3000);
	while(1)
	{
		spitc(0x0c, 0x01, 0);
		spitc(0x0c, 0x01, 1);
		spitc(0x0c, 0x01, 2);
		spitc(0x0c, 0x01, 3);
		setleds("NEXT");
		_delay_ms(2000);
		setleds("102T");
		_delay_ms(2000);
		setleds("0 10");
		_delay_ms(2000);
		setleds(" 195");
		_delay_ms(2000);
		setleds("0 15");
		_delay_ms(2000);
		setleds("  10");
		_delay_ms(2000);
		setleds("0 03");
		_delay_ms(2000);
		setleds("TIME");
		_delay_ms(2000);
		setleds("1536");
		_delay_ms(2000);
		setleds("    ");
		_delay_ms(18000);
		spitc(0x0c, 0x00, 0);
		spitc(0x0c, 0x00, 1);
		spitc(0x0c, 0x00, 2);
		spitc(0x0c, 0x00, 3);
	}
}