#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