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

#define NOP()	asm("nop")
#define BIT(x)	(1<<(x))
#define cbi(PORT,x) (PORT &= ~(1 << x))
#define sbi(PORT,x)	(PORT |= (1 << x))

#define SN74HC595_PORT		PORTB 
#define SN74HC595_DATA		PINB3 		//数据-------SN74HC595 14# 
#define ST			    	PINB2		//锁存时钟---SN74HC595 12# 
#define SH					PINB5       //移位时钟---SN74HC595 11# 
#define SN74HC595_OE_PORT	PORTD
#define OE					PIND3

#define set_oe()			sbi(SN74HC595_OE_PORT,OE)
#define clr_oe()			cbi(SN74HC595_OE_PORT,OE)

volatile unsigned char dat = 0;
//volatile unsigned char temp_buf[8] = {0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};
//volatile unsigned char led_buf[8] = {0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};
volatile unsigned char led_data[64] = {0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
									   0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
									   0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
									   0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
									   0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
									   0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,
									   0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF};
volatile unsigned char disp_data[8] = {0x7f,0xff,0xff,0xff,0xff,0xff,0xff,0xff};

void wdt_off(void)
{
	MCUSR = 0x00;
	WDTCSR = (1 <<WDCE) | (1 << WDE);
	WDTCSR = 0x00;
}

void delay_ms(unsigned int ms)
{
	for(;ms;ms--)
	{
		_delay_ms(1);
	}
}

void port_init(void)
{
	PORTB = 0x00;
	DDRB  = 0x2C;
	PORTC = 0x00;
	DDRC  = 0x00;
	PORTD = 0x00;
	DDRD  = 0x08;
}

//TIMER2 initialize - prescale:1024
// WGM: PWM Fast
// desired value: 61.03515625Hz
// actual value: 488.28125Hz / 8 = 61.03515625Hz
void timer2_init(void)
{
	TCCR2B = 0x00; //stop
	ASSR  = 0x00; //set async mode
	TCNT2 = 0x01; //setup
	OCR2A = 0xFF;
	OCR2B = 0xFF;
	TCCR2A = 0x23; 
	TCCR2B = 0x05; //start
}

//SPI initialize
// SPI clock rate: 12000000hz
void spi_init(void)
{
	SPCR = 0x50; //setup SPI
	SPSR = 0x01; //setup SPI
}

void spi_write(unsigned char *s_buf)
{
	unsigned char i = 0;
	cbi(SN74HC595_PORT,ST);
	for(i = 0;i < 8;i++)
	{
		SPDR = *s_buf;
		s_buf++;
		loop_until_bit_is_set(SPSR,SPIF);
		//while(!(SPSR & BIT(SPIF)));
	}
	sbi(SN74HC595_PORT,ST);
}
//call this routine to initialize all peripherals
void init_devices(void)
{
	//stop errant interrupts until set up
	cli(); //disable all interrupts
	port_init();
	set_oe();
	timer2_init();
	spi_init();

	MCUCR = 0x00;
	EICRA = 0x00; //extended ext ints
	EIMSK = 0x00;

	TIMSK0 = 0x00; //timer 0 interrupt sources
	TIMSK1 = 0x00; //timer 1 interrupt sources
	TIMSK2 = 0x01; //timer 2 interrupt sources

	PCMSK0 = 0x00; //pin change mask 0 
	PCMSK1 = 0x00; //pin change mask 1 
	PCMSK2 = 0x00; //pin change mask 2
	PCICR = 0x00; //pin change enable 
	PRR = 0x00; //power controller
	sei(); //re-enable interrupts
	//all peripherals are now initialized
}

int main(void)
{
	unsigned char i = 0;
	unsigned char j = 0;
	unsigned char k = 0;
	init_devices();

	for(i = 0;i < 64;i++)
	{
		led_data[i] = 255 - i * 2 - 1;
	}

	while(1)
	{
		for(k = 0;k < 8;k++)
		{
			for(j = 0;j < 255;j++)
			{
				for(i = 0;i < 64;i++)
				{
					led_data[i]--;
				}
				delay_ms(5);
			}
		}
		for(k = 0;k < 8;k++)
		{
			for(j = 255;j > 0;j--)
			{
				for(i = 0;i < 64;i++)
				{
					led_data[i]++;
				}
				delay_ms(5);
			}
		}
	}
	return(0);
}

void process(unsigned char p_bit)
{
	unsigned char i = 0;
	unsigned char j = 0;
	unsigned char temp1 = 0;
	unsigned char temp2 = 0;
	unsigned char temp_buf[8] = {0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};

	temp2 = (unsigned char)(0x01 << p_bit);

	for(j = 0;j < 8;j++)
	{
		for(i = 0;i < 8;i++)
		{
			temp_buf[j] <<= 1;
			temp1 = 8 * j + i;
			if((led_data[temp1] & temp2) == temp2)
			{
				temp_buf[j] |= 0x01;
			}
		}
	}
	spi_write(temp_buf);
}

ISR(TIMER2_OVF_vect)
{
	static unsigned char duty = 128;
	static unsigned char jie = 7;
	OCR2B = duty;
	process(jie);
	switch (duty)
	{
		case 128:
			duty = 192;
			jie = 6;
			break;
		case 192:
			duty = 224;
			jie = 5;
			break;
		case 224:
			duty = 240;
			jie = 4;
			break;
		case 240:
			duty = 248;
			jie = 3;
			break;
		case 248:
			duty = 252;
			jie = 2;
			break;
		case 252:
			duty = 254;
			jie = 1;
			break;
		case 254:
			duty = 255;
			jie = 0;
			break;
		case 255:
			duty = 128;
			jie = 7;
			break;
		default:
			duty = 128;
			jie = 7;
			break;
	}
}