#include "AT91RM9200.h"
#include "lib_AT91RM9200.h"
#include "def.h"
#include "config.h"
#include "slib.h"
#include "console.h"
#include "twi.h"
#include "params.h"

// The following DBGU ASM handler is defined in asm_isr.s
//extern void AT91F_ASM_DBGU_Handler(void);

U32 downloadAddress, downloadFileSize;

//char hex_ch[16] = {'0','1','2','3','4','5','6','7',
//				   '8','9','a','b','c','d','e','f'};
static EnvParams Env;

static U8 OurEmacAddr[6] = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06};

volatile U32 StTick;
void (*StIrqHandler)(void);
void (*DbguIrqHandler)(void);

static void SysIrqHandler(void)
{
	/* ========== Systimer interrupt ============== */
	if(AT91F_ST_GetInterruptMaskStatus(AT91C_BASE_ST) & AT91C_ST_PITS) {//周期間隔(PITS)定時器中斷使能
		if (AT91C_BASE_ST->ST_SR & AT91C_ST_PITS) {//產生PITS中斷了
			StTick++;//時間片自增
			if(*StIrqHandler)
				(*StIrqHandler)();
			return;
		}
	}
	
	if(*DbguIrqHandler)
		(*DbguIrqHandler)();
}

//統一的中斷入口處理
void __irq irq_handler(void)
{	
	void (*svr)(void);
	AT91PS_AIC ptr = AT91C_BASE_AIC;
	U32 i;
//	U8 irq_idx;

//取得中斷入口地址可用AIC_IVR或用AIC_ISR作索引得到AIC_SVR數組中的地址
	i = ptr->AIC_IVR;		//read AIC_IVR
//	irq_idx = ptr->AIC_ISR&0x1f;
//邊沿觸發中斷必須以此清中斷
//	AT91F_AIC_ClearIt(AT91C_BASE_AIC, irq_idx);	

// Write in the IVR to support Protect Mode
// No effect in Normal Mode
// De-assert the NIRQ and clear the source in Protect Mode	
	ptr->AIC_IVR = (AT91_REG)ptr;
	
//	putch('I');
//	printf("%x,%x,%x\n", irq_idx, ptr->AIC_IPR, ptr->AIC_CISR);

	svr = (void (*)(void))i;//ptr->AIC_SVR[irq_idx];
	(*svr)();
	AT91F_AIC_AcknowledgeIt(ptr);	//退出中斷前必須應答
}

static void InitPio(void)
{
		
	//led
	AT91F_PIO_CfgOutput(AT91C_BASE_PIOC, AT91C_PIO_PC0|AT91C_PIO_PC2|AT91C_PIO_PC4|AT91C_PIO_PC5);
	AT91F_PIO_ClearOutput(AT91C_BASE_PIOC, AT91C_PIO_PC0|AT91C_PIO_PC2|AT91C_PIO_PC4|AT91C_PIO_PC5);
	//key
	//AT91F_PIO_CfgInput(AT91C_BASE_PIOB, AT91C_PIO_PB6|AT91C_PIO_PB7|AT91C_PIO_PB8|AT91C_PIO_PB9);
	AT91F_PIO_CfgInput(AT91C_BASE_PIOA, AT91C_PIO_PA24);
	AT91F_PIO_CfgInput(AT91C_BASE_PIOB, AT91C_PIO_PB1|AT91C_PIO_PB2|AT91C_PIO_PB6);
	//AT91F_PIO_CfgOutput(AT91C_BASE_PIOB, AT91C_PIO_PB1|AT91C_PIO_PB2|AT91C_PIO_PB6);
	///AT91F_PIO_ClearOutput(AT91C_BASE_PIOB, AT91C_PIO_PB1|AT91C_PIO_PB2|AT91C_PIO_PB6);
	AT91F_PIO_CfgOutput(AT91C_BASE_PIOB, AT91C_PIO_PB7);
	AT91F_PIO_ClearOutput(AT91C_BASE_PIOB, AT91C_PIO_PB7);
	
	AT91F_PMC_EnablePeriphClock(AT91C_BASE_PMC, ((U32) 1 << AT91C_ID_PIOB));
	AT91F_PMC_EnablePeriphClock(AT91C_BASE_PMC, ((U32) 1 <<AT91C_ID_PIOA ));
	//led
	/*AT91F_PIO_CfgOutput(AT91C_BASE_PIOC, AT91C_PIO_PC0|AT91C_PIO_PC2|AT91C_PIO_PC4|AT91C_PIO_PC5);
	AT91F_PIO_ClearOutput(AT91C_BASE_PIOC, AT91C_PIO_PC0|AT91C_PIO_PC2|AT91C_PIO_PC4|AT91C_PIO_PC5);
	//key
	AT91F_PIO_CfgInput(AT91C_BASE_PIOB, AT91C_PIO_PB6|AT91C_PIO_PB7|AT91C_PIO_PB8|AT91C_PIO_PB9);
	
	//buzzer
	AT91F_PIO_CfgPeriph(
		AT91C_BASE_PIOB, // PIO controller base address
		((unsigned int) AT91C_PB27_PCK0    ), // Peripheral A
		0); // Peripheral B
	AT91F_PMC_EnablePCK(AT91C_BASE_PMC,
		0, 	//PCK0
		AT91C_PMC_CSS_SLOW_CLK,	//32768
		AT91C_PMC_PRES_CLK_16);	//1/16
	delay(200);
	AT91F_PIO_CfgInput(AT91C_BASE_PIOB, AT91C_PIO_PB27);
	*/
	//delay(200);
}

U32 GetFlashID(void);
int SectorProg(U32 begin, U16 *data, U32 size);
int SectorRead(U32 begin, U16 *data, U32 size);

void start_kernel(U32, U32);

static void InitNorFlash(void)
{
#ifdef	INTEL_FLASH_SUPPORT
	// Setup MEMC to support CS0 = static memory
	AT91C_BASE_EBI->EBI_CSA &= ~1;
	// [D15:0] pull-up
	AT91C_BASE_EBI->EBI_CFGR = (AT91C_EBI_DBPUC & 0x00) | (AT91C_EBI_EBSEN & 0x00);

	// Setup Flash 存儲器參數設置CS0-CS7對應SMC2_CSR[0]-SMC2_CSR[7]
	AT91C_BASE_SMC2->SMC2_CSR[0] = (AT91C_SMC2_NWS & 0x7f)  | AT91C_SMC2_WSEN |
	                               (AT91C_SMC2_TDF & 0x200)| AT91C_SMC2_BAT  |
	                                AT91C_SMC2_DBW_16 | AT91C_SMC2_ACSS_STANDARD;/* |
	                                AT91C_SMC2_RWSETUP | AT91C_SMC2_RWHOLD; */
	printf("FLASH ID = %x\n", GetFlashID());
#endif
}

static void RunProgFrNor(U32 addr, U32 size)
{
#ifdef	INTEL_FLASH_SUPPORT	
	downloadAddress  = 0x20800000;
	downloadFileSize = 0x200000;
	printf("Read Program From Nor Flash to 0x%8x, Size 0x%x\n", 
			downloadAddress, downloadFileSize);
	SectorRead(0x100000, (U16 *)downloadAddress, downloadFileSize);
	start_kernel(downloadAddress, 0);
#endif
}

static void WrFileToNor(U32 addr, U32 size)
{
#ifdef	INTEL_FLASH_SUPPORT
	printf("Write Data From 0x%8x, Size 0x%x to Nor Flash\n", addr, size);
	ProgFlash();
	SectorProg(0x100000, (U16 *)addr, size);
#endif
}


void GetNandFlashChip(void);

static void InitNandFlash(void)
{
#ifdef	NAND_SUPPORT
	AT91F_PIO_CfgPeriph(
		AT91C_BASE_PIOC, // PIO controller base address
		(unsigned int) AT91C_PC1_BFRDY_SMOE | (unsigned int) AT91C_PC3_BFBAA_SMWE,	// Peripheral A
		0);	// Peripheral B
	
	// Setup MEMC to support CS3 = NAND Flash
	AT91C_BASE_EBI->EBI_CSA |= AT91C_EBI_CS3A_SMC_SmartMedia;
	// [D15:0] pull-up
	AT91C_BASE_EBI->EBI_CFGR = (AT91C_EBI_DBPUC & 0x00) | (AT91C_EBI_EBSEN & 0x00);

	// Setup Flash 存儲器參數設置CS0-CS7對應SMC2_CSR[0]-SMC2_CSR[7]
	AT91C_BASE_SMC2->SMC2_CSR[3] = (AT91C_SMC2_NWS & 0x4)  | AT91C_SMC2_WSEN |
	                               (AT91C_SMC2_TDF & 0x200)| AT91C_SMC2_BAT  |
	                                AT91C_SMC2_DBW_8;
	// Enable PIOC clock for input
	AT91F_PMC_EnablePeriphClock(AT91C_BASE_PMC, ((unsigned int) 1 << AT91C_ID_PIOC));
	AT91F_PIO_CfgInput (AT91C_BASE_PIOC, AT91C_PIO_PC14);
	AT91F_PIO_CfgPullup(AT91C_BASE_PIOC, AT91C_PIO_PC14);
	AT91F_PIO_CfgOutput (AT91C_BASE_PIOC, AT91C_PIO_PC15);
//	AT91F_PIO_GetInput(AT91C_BASE_PIOC) & AT91C_PIO_PC14;

	GetNandFlashChip();
#endif
}


#ifdef	IIC_SUPPORT
#define	EEP_RW_CHK_CNT	32
#endif	/* IIC_SUPPORT */

static void InitTwi(void)
{
#ifdef	IIC_SUPPORT
	int loop;
	char data[EEP_RW_CHK_CNT];

	// Configure TWI PIOs
	AT91F_TWI_CfgPIO ();
	AT91F_PIO_CfgOpendrain(AT91C_BASE_PIOA, (unsigned int) AT91C_PA25_TWD);

	// Configure PMC by enabling TWI clock
	AT91F_TWI_CfgPMC ();
	
	// Configure TWI in master mode
	AT91F_TWI_Configure (AT91C_BASE_TWI);
		
	// Set TWI Clock Waveform Generator Register	
	AT91F_SetTwiClock(AT91C_BASE_TWI);

	// IIC啟動,先測試IIC讀
	printf("AT91RM9200 TWI EEPREOM Test, address is 0x%x\n", AT91C_EEPROM_I2C_ADDRESS);
	
//	for(loop=0; loop<EEP_RW_CHK_CNT; loop++)
//		data[loop] = loop;

	// Write and read iic
//	AT91F_TWI_Write(AT91C_BASE_TWI, 0x0, data, EEP_RW_CHK_CNT);
	
	// Wait 10 ms before data is written into EEPROM
//	puts("Wait at least 10 ms before value is written into EEPROM\n");
//	for (loop=0; loop<100000; loop++);
	
	for(loop=0; loop<EEP_RW_CHK_CNT; loop++)
		data[loop] = 0;	

	AT91F_TWI_Read(AT91C_BASE_TWI, 0x0, data, EEP_RW_CHK_CNT);
	
	for(loop=0; loop<EEP_RW_CHK_CNT; loop++)
		if(data[loop]!=loop)
			break;
	
	printf("Check data %s\n", (loop==EEP_RW_CHK_CNT)?"success":"fail");
	
#endif	/* IIC_SUPPORT */
}

static void ShowLed(U32 led)
{
	U8 set = 0, clr = 0;
	
	if(led&1)
		set |= 1;
	else
		clr |= 1;
		
	if(led&2)
		set |= 4;
	else
		clr |= 4;
		
	if(led&4)
		set |= 0x10;
	else
		clr |= 0x10;
		
	if(led&8)
		set |= 0x20;
	else
		clr |= 0x20;

	AT91F_PIO_SetOutput (AT91C_BASE_PIOC, set);
	AT91F_PIO_ClearOutput (AT91C_BASE_PIOC, clr);
}


U16 GetKeyStatus(void)
{
	U16 r=0;
	U32 k=0;
	k = AT91F_PIO_GetInput(AT91C_BASE_PIOB);
	//printf("PB = %d\n",k);
	if(!(k&(1<<1)))
	{
		r |= 2;
	//	printf("r1=%x\n",r);
	}
	if(!(k&(1<<2)))
	{
		r |= 3;
	//	printf("r2=%x\n",r);
	}
	if(!(k&(1<<6)))
	{
		r |= 4;
	//	printf("r3=%x\n",r);
	}
	if(!(AT91F_PIO_GetInput(AT91C_BASE_PIOA)&(1<<24)))
	{	
		r |= 1;
	//	printf("r3=%x\n",r);
	}
		
	//printf("r4=%d\n",r);
	return r;
	//return (AT91F_PIO_GetInput(AT91C_BASE_PIOB));
}

void ComDownload(U32 a1, U32 a2);
void Program_eep(U32 addr, U32 size);
//void WrFileToNF(U32 FileAddr, U32 FileSize);
//void RdFileFrNF(U32 FileAddr, U32 FileSize);
void XmodemDownload(U32 a1, U32 a2);
//void TestNandFlash(U32 a1, U32 a2);
//void EraseNandPart(U32 a1, U32 a2);
void RunProgFrNor(U32 addr, U32 size);
//void tftp_main(U32 addr, U32 dummy);

static struct {
	void (*FuncAddr)(U32, U32);
	char *str;
} Functions[] = {
		{ComDownload,    "Download and Run File By Uart55555(DNW)"},
//		{tftp_main, "Download File By Tftp"},
		{XmodemDownload, "Download and Run File By Xmodem888888"},
//		{Program_eep, "Write File To IIC Rom"},
//		{WrFileToNF, "Write File To Nand Flash"},
//		{RdFileFrNF, "Run Pragram From Nand Flash"},		
//		{ProgFlash, "ProgFlash"},
//		{WrFileToNor, "Write File To Nor Flash"},
//		{RunProgFrNor, "Run Program From Nor Flash"},
//		{TestNandFlash, "Test nand flash"},
		{0, 0}
	};

//*----------------------------------------------------------------------------
//* \fn    main
//* \brief 
//*