#include <stdio.h>
//#include "pin2410.h"
//#include "Jtag.h"
#include "K9S1208.h"
//#include "sjf2410.h"
#include "nand.h"
#include "2440lib.h"
#include "def.h"
#include "2440addr.h"

#define BAD_CHECK	(0)
#define ECC_CHECK	(0)

//#define imageSize 0x4000

void Uart_Printf(char *fmt,...);
U32 GETVALUE(char *buf);
 int LoadImageFile(U8 *buf,int size);
//*************** JTAG dependent functions ***************
/*
void K9S1208_JtagInit(void);
static void NF_CMD(U8 cmd);
static void NF_ADDR(U8 addr);
static void NF_nFCE_L(void);
static void NF_nFCE_H(void);
static U8 NF_RDDATA(void);
static void NF_WRDATA(U8 data);
static void NF_WAITRB(void);*/

#define NF_MECC_UnLock()         {rNFCONT&=~(1<<5);}
#define NF_MECC_Lock()         {rNFCONT|=(1<<5);}

#define NF_CMD(cmd)			{rNFCMD=cmd;}
#define NF_ADDR(addr)		{rNFADDR=addr;}	
#define NF_nFCE_L()			{rNFCONT&=~(1<<1);}
#define NF_nFCE_H()			{rNFCONT|=(1<<1);}
#define NF_RSTECC()			{rNFCONT|=(1<<4);}
#define NF_RDDATA() 		(rNFDATA8)
#define NF_RDDATA8() 		((*(volatile unsigned char*)0x4E000010) )

#define NF_WRDATA(data) 	{rNFDATA8=data;}//by xh

#define NF_WAITRB()    		{while(!(rNFSTAT&(1<<0)));} 
	   						 //wait tWB and check F_RNB pin.
	   						 
#define TACLS		0	// 1-clk(0ns) 
#define TWRPH0		6	// 3-clk(25ns)
#define TWRPH1		0	// 1-clk(10ns)  //TACLS+TWRPH0+TWRPH1>=50ns
	   						 
// RnB Signal
#define NF_CLEAR_RB()    		{rNFSTAT |= (1<<2);}	// Have write '1' to clear this bit.
#define NF_DETECT_RB()    		{while(!(rNFSTAT&(1<<2)));}

#define ID_K9S1208V0M	0xec76
#define ID_K9K2G16U0M	0xecca

#define	NF16_BAD_OFFSET		(12)

//#define blockBuf ((volatile unsigned char *)0x31000000)
//*************** H/W dependent functions ***************
static U16 NF_CheckId(void);
static int NF_EraseBlock(U32 blockNum);
static int NF_ReadPage(U32 block,U32 page,U8 *buffer,U8 *spareBuf);
static int NF_WritePage(U32 block,U32 page,U8 *buffer,U8 *spareBuf);
	//buffer size is 512 bytes
static int NF_IsBadBlock(U32 block);
static int NF_MarkBadBlock(U32 block);
static void NF_Reset(void);
static void NF_Init(void);
//*******************************************************

void K9S1208_PrintBlock(void);
void K9S1208_Program(void);


static U32 targetBlock;	    // Block number (0 ~ 4095)
static U32 targetSize;	    // Total byte size 
static U8 blockBuf[0x4000];


static char RCVBUF[10];//by xh
static U32 imageSize;
volatile unsigned char *BufBase;

static void *function[][2]=
{
    (void *)K9S1208_Program,		"K9S1208 Program     ",
    (void *)K9S1208_PrintBlock,		"K9S1208 Pr BlkPage  ",
    (void *)1,			    	"Exit                ",
    0,0
};


void K9S1208_Menu(void)
{
    int i;
    U16 id;

    Uart_Printf("\n[K9S1208 NAND Flash JTAG Programmer]\n");
    //K9S1208_JtagInit();
    NF_Init();
    //NF8_Init();//by xh

    id=NF_CheckId();
    if(id!=0xec76)
    {
	Uart_Printf("ERROR: K9S1208 is not detected. Detected ID=0x%x.\n",id);
	return;
    }
    else
    {
    	Uart_Printf("K9S1208 is detected. ID=0x%x\n",id);
    }

    while(1)
    {

	i=0;
    	while(1)
	{   //display menu
	    Uart_Printf("%2d:%s",i,function[i][1]);
	    i++;
	    if((int)(function[i][0])==0)
	    {
		Uart_Printf("\n");
		break;
	    }
	    if((i%4)==0)
		Uart_Printf("\n");
	}

	Uart_Printf("Select the function to test :");
	//scanf("%d",&i);// by xh
	i=GETVALUE(RCVBUF);
	if( i>=0 && (i<((sizeof(function)/8)-2)) ) 
	    ( (void (*)(void)) (function[i][0]) )();  
	else
	    break; //Exit menu
	    
    }
}



void K9S1208_Program(void)
{
    int i;
    int programError=0;
    U32 blockIndex;
    int noLoad=0;
    U8 spareBuf[16]=
	{0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};
    U8 *srcPt;
    U32 progSize=0;

    Uart_Printf("\n[SMC(K9S1208V0M) NAND Flash Writing Program]\n");
    
    Uart_Printf("\nPlease input code size:");//by xhe
    imageSize = GETVALUE(RCVBUF);
    
    Uart_Printf("\nSource size:0h~%xh\n",imageSize-1);
    
    Uart_Printf("\nAvailable target block number: 0~4095\n");
    Uart_Printf("Input target block number:");
    //scanf("%d",&targetBlock);// by xh
    targetBlock = GETVALUE(RCVBUF);
    targetSize=((imageSize+0x4000-1)/0x4000)*0x4000;
    Uart_Printf("target start block number     =%d\n",targetBlock);
    Uart_Printf("target size        (0x4000*n) =0x%x\n",targetSize);
    Uart_Printf("STATUS:");
    blockIndex=targetBlock;
    BufBase = ((volatile unsigned char *)0x31000000);
    while(1)
    {
	if(noLoad==0)
	{
	    LoadImageFile(blockBuf,0x4000);
	}
	noLoad=0;
	
#if BAD_CHECK       
	if(NF_IsBadBlock(blockIndex) && blockIndex!=0 )	// 1:bad 0:good
        {
	    blockIndex++;   // for next block
	    noLoad=1;
	    continue;
	}
#endif
	if(!NF_EraseBlock(blockIndex))
	{
	    blockIndex++;   // for next block
	    noLoad=1;
	    continue;
	}

	Uart_Printf("E");
	srcPt=blockBuf;
	//srcPt = ((U8 *)0x31000000);//by xh

	for(i=0;i<32;i++)
	{
	    if(!NF_WritePage(blockIndex,i,srcPt,NULL/*spareBuf*/))// block num, page num, buffer
	    {
	        programError=1;
	        break;
	    }

	    srcPt+=512;	// Increase buffer addr one pase size
	    Uart_Printf("p");
	}
	Uart_Printf("\n");

        if(programError==1)
	{
	    blockIndex++;
	    noLoad=1;
	    programError=0;
	    continue;
	}
	progSize+=0x4000;
	if(progSize>=imageSize)
	    break;	// Exit while loop
	blockIndex++;
    }
}




void K9S1208_PrintBlock(void)// Printf one page
{
    int i,j,ecc;
    U16 id;
    U32 block,page;
    U8	buffer[512+16];
    //volatile unsigned char buffer[512+16];

    Uart_Printf("\n[SMC(K9S1208) NAND Flash block read]\n");	
    
    NF_Init();//by xh
    id=NF_CheckId();
    Uart_Printf("ID=%x(0xec76)\n",id);
    if(id!=0xec76)
	return;

    Uart_Printf("Input target block number:");
    //scanf("%d",&block);//by xh
    block = GETVALUE(RCVBUF);
    Uart_Printf("Input target page number:");   
    //scanf("%d",&page);// by xh
    page = GETVALUE(RCVBUF);
    
    NF_ReadPage(block,page,buffer,buffer+512);
    
    Uart_Printf("block=%d,page=%d:",block,page);
    for(i=0;i<512;i++)
    {
        if(i%16==0)
	    Uart_Printf("\n%3xh:",i);
        Uart_Printf("%02x ",buffer[i]);
        for(j=0;j<100;j++);//by xh
    }
    Uart_Printf("\nS.A.:",i);

    for(i=512;i<512+16;i++)
    {
        Uart_Printf("%02x ",buffer[i]);
        for(j=0;j<100;j++);//by xh
    }

    Uart_Printf("\n");    	
}

//*************************************************
//*************************************************
//**           H/W dependent functions           **
//************************************************* 
//*************************************************

// NAND Flash Memory Commands
#define	SEQ_DATA_INPUT			(0x80)
#define	READ_ID				(0x90)
#define	RESET				(0xFF)
#define	READ_1_1			(0x00)
#define	READ_1_2			(0x01)
#define	READ_2				(0x50)
#define	PAGE_PROGRAM			(0x10)
#define	BLOCK_ERASE			(0x60)
#define	BLOCK_ERASE_CONFIRM		(0xD0)
#define	READ_STATUS			(0x70)


// block0: reserved for boot strap
// block1~4095: used for OS image
// badblock SE: xx xx xx xx xx 00 ....
// good block SE: ECC0 ECC1 ECC2 FF FF FF ....

#define WRITEVERIFY  (0)  //verifing is enable at writing.

/*
#define NF_CMD(cmd)	{rNFCMD=cmd;}
#define NF_ADDR(addr)	{rNFADDR=addr;}	
#define NF_nFCE_L()	{rNFCONF&=~(1<<11);}
#define NF_nFCE_H()	{rNFCONF|=(1<<11);}
#define NF_RSTECC()	{rNFCONF|=(1<<12);}
#define NF_RDDATA() 	(rNFDATA)
#define NF_WRDATA(data) {rNFDATA=data;}

#define NF_WAITRB()    {while(!(rNFSTAT&(1<<0)));} 
	    //wait tWB and check F_RNB pin.   
*/
#define ID_K9S1208V0M	0xec76

static U8 seBuf[16]={0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};

// 1block=(512+16)bytes x 32pages
// 4096block

// A[23:14][13:9]
//  block   page

static int NF_EraseBlock(U32 block)
{
    U32 blockPage=(block<<5);

#if BAD_CHECK
    if(NF_IsBadBlock(block) && block!=0) //block #0 can't be bad block for NAND boot
	return 0;
#endif

    NF_nFCE_L();
    //NF_CLEAR_RB();
    NF_CMD(0x60);   // Erase one block 1st command

    NF_ADDR(blockPage&0xff);	    // Page number=0
    NF_ADDR((blockPage>>8)&0xff);   
    NF_ADDR((blockPage>>16)&0xff);

    NF_CMD(0xd0);   // Erase one blcok 2nd command
    
    Delay(1); //wait tWB(100ns)

    NF_WAITRB();    // Wait tBERS max 3ms.
    //NF_DETECT_RB();//by xh