#include "AT91RM9200.h"
#include "lib_AT91RM9200.h"
#include "embedded_services.h"
#include "def.h"
#include "config.h"
#include "console.h"
#include "params.h"

#ifdef	XMODEM_SUPPORT

#define AT91C_UBOOT_BASE_ADDRESS	BOOT_PORG_ADDR
#define AT91C_UBOOT_MAXSIZE	        0x20000
#define CLRSCREEN	                "\033[2J"
#define	AT91C_PAUSE					100

#define LOADER_VERSION	"loader 1.0"

/* Global variables */
volatile char		XmodemComplete =0;
extern volatile unsigned int StTicks;
extern void (*StIrqHandler)(void);
extern void (*DbguIrqHandler)(void);
void start_kernel(U32 address, U32 initrd);

AT91S_RomBoot const *pAT91;
AT91S_SBuffer   	sXmBuffer;
AT91S_SvcXmodem 	svcXmodem;
AT91S_Pipe      	xmodemPipe;
AT91S_CtlTempo  	ctlTempo;

/*--------------------------------------------------------------------------------------*/
/* Function Name       : AT91_XmodemComplete()						*/
/* Object              : Perform the remap and jump to appli in RAM			*/
/*--------------------------------------------------------------------------------------*/
static void AT91F_XmodemComplete(AT91S_PipeStatus status, void *pVoid)
{
	/* stop the Xmodem tempo */
	svcXmodem.tempo.Stop(&(svcXmodem.tempo));
	XmodemComplete = 1;
}

/*--------------------------------------------------------------------------------------*/
/* Function Name       : AT91F_XmodemProtocol(AT91S_PipeStatus status, void *pVoid)	*/
/* Object              : Xmodem dispatcher						*/
/*--------------------------------------------------------------------------------------*/
static void AT91F_XmodemProtocol(AT91S_PipeStatus status, void *pVoid)
{
	AT91PS_SBuffer pSBuffer = (AT91PS_SBuffer) xmodemPipe.pBuffer->pChild;
	AT91PS_USART   pUsart     = svcXmodem.pUsart;
			
	if (pSBuffer->szRdBuffer == 0) {
		svcXmodem.tempo.Start(&(svcXmodem.tempo), 10, 0, AT91F_XmodemComplete, pUsart);								
	}
}

/*-------------------------- Interrupt handlers ----------------------------------------*/
/*--------------------------------------------------------------------------------------*/
/* Function Name       : AT91F_ST_HANDLER()						*/
/* Object              : C Interrupt handler for the system timer			*/
/*--------------------------------------------------------------------------------------*/
/*
static void AT91F_ST_HANDLER(void)
{
	volatile unsigned int csr = *AT91C_DBGU_CSR;
	unsigned int error;
	
	// ========== Systimer interrupt ============== 
	if (AT91C_BASE_ST->ST_SR & 0x01) {
		StTicks++;
		ctlTempo.CtlTempoTick(&ctlTempo);
		return;
	}

	error = AT91F_US_Error((AT91PS_USART)AT91C_BASE_DBGU);
	if (csr & error) {
		// Stop previous Xmodem transmition 
		*(AT91C_DBGU_CR) = AT91C_US_RSTSTA;
		AT91F_US_DisableIt((AT91PS_USART)AT91C_BASE_DBGU, AT91C_US_ENDRX);
		AT91F_US_EnableIt((AT91PS_USART)AT91C_BASE_DBGU, AT91C_US_RXRDY);

	}
	
	else if (csr & (AT91C_US_TXRDY | AT91C_US_ENDTX | AT91C_US_TXEMPTY | 
	                AT91C_US_RXRDY | AT91C_US_ENDRX | AT91C_US_TIMEOUT | 
	                AT91C_US_RXBUFF)) {
		if ( !(svcXmodem.eot) )
			svcXmodem.Handler(&svcXmodem, csr);
	}
}
*/
static void XmodemStIrqHandler(void)
{
/*	static tick;
	if(tick++>=100) {
		tick = 0;
		putch('a');
	}*/
	ctlTempo.CtlTempoTick(&ctlTempo);
}

static void XmodemDbguIrqHandler(void)
{
	volatile unsigned int csr = *AT91C_DBGU_CSR;
	unsigned int error;
	
	error = AT91F_US_Error((AT91PS_USART)AT91C_BASE_DBGU);
	if (csr & error) {
		/* Stop previous Xmodem transmition */
		*(AT91C_DBGU_CR) = AT91C_US_RSTSTA;
		AT91F_US_DisableIt((AT91PS_USART)AT91C_BASE_DBGU, AT91C_US_ENDRX);
		AT91F_US_EnableIt((AT91PS_USART)AT91C_BASE_DBGU, AT91C_US_RXRDY);

	} else if (csr & (AT91C_US_TXRDY | AT91C_US_ENDTX | AT91C_US_TXEMPTY | 
					AT91C_US_RXRDY | AT91C_US_ENDRX | AT91C_US_TIMEOUT | 
					AT91C_US_RXBUFF)) {
		if ( !(svcXmodem.eot) )
			svcXmodem.Handler(&svcXmodem, csr);
	}
}

void XmodemDownload(U32 a, U32 b)
{
	AT91PS_Buffer  pXmBuffer;
	AT91PS_SvcComm pSvcXmodem;
	
	/* Init of ROM services structure */
	pAT91 = AT91C_ROM_BOOT_ADDRESS;

	/* Tempo Initialization */
	pAT91->OpenCtlTempo(&ctlTempo, (void *) &(pAT91->SYSTIMER_DESC));
	ctlTempo.CtlTempoStart((void *) &(pAT91->SYSTIMER_DESC));

	/* Xmodem Initialization */
	pXmBuffer  = pAT91->OpenSBuffer(&sXmBuffer);
	pSvcXmodem = pAT91->OpenSvcXmodem(&svcXmodem, (AT91PS_USART)AT91C_BASE_DBGU, &ctlTempo);
	pAT91->OpenPipe(&xmodemPipe, pSvcXmodem, pXmBuffer);

	/* System Timer initialization */
/*	AT91F_AIC_ConfigureIt (	AT91C_BASE_AIC,				// AIC base address
				AT91C_ID_SYS,							// System peripheral ID
				AT91C_AIC_PRIOR_HIGHEST,				// Max priority
				AT91C_AIC_SRCTYPE_INT_LEVEL_SENSITIVE,	// Level sensitive
//				AT91F_ST_ASM_HANDLER );
				AT91F_ST_HANDLER );*/
	//片內下載程序設置值為30,可在上電時讀出!時間間隔為(30/32768)S
	AT91C_BASE_ST->ST_PIMR = 30;
	StIrqHandler = XmodemStIrqHandler;
	DbguIrqHandler = XmodemDbguIrqHandler;
	AT91F_ST_EnableIt(AT91C_BASE_ST, AT91C_ST_PITS);

	/* Enable ST interrupt */
//	AT91F_AIC_EnableIt(AT91C_BASE_AIC, AT91C_ID_SYS);

	puts(CLRSCREEN);

	puts(""LOADER_VERSION" ("__DATE__ " - " __TIME__ ")\n");
	
	puts("XMODEM: Download U-BOOT\n");

	xmodemPipe.Read(&xmodemPipe, (char *)AT91C_UBOOT_BASE_ADDRESS, AT91C_UBOOT_MAXSIZE, AT91F_XmodemProtocol, 0);
	while(XmodemComplete !=1);
		
	puts("\nU-Boot downloaded successfully\n");

	delay(AT91C_PAUSE);

//	AT91F_SetPLL();

	AT91F_AIC_DisableIt(AT91C_BASE_AIC, 0xFFFFFFFF);
	AT91F_AIC_AcknowledgeIt(AT91C_BASE_AIC);
	
	start_kernel((unsigned int)AT91C_UBOOT_BASE_ADDRESS, 0);
}

#else

void XmodemDownload(U32 a, U32 b) {}

#endif