#include <stdio.h>
#include "..\include\pin6400.h"
#include "..\include\Jtag.h"
#include "..\include\K9f1g08.h"
#include "..\include\sjf6400.h"


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

//*************** JTAG dependent functions ***************
void K9f1g08_JtagInit(void);
static void NF_CMD(U8 cmd);
static void NF_RADDR(U8 raddr);
static void NF_nFCE_L(void);
static void NF_nFCE_H(void);
static U8 NF_RDRDATA(void);
static void NF_WRRDATA(U8 rdata);
static void NF_WAITRB(void);
//*************** 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 K9f1g08_PrintBlock(void);
void K9f1g08_Program(void);


static U32 targetBlock;	    // Block number (0 ~ 4095)
static U32 targetSize;	    // Total byte size 
static U8 blockBuf[0x20000];   
static void *function[][2]=
{
    (void *)K9f1g08_Program,		"K9f1g08 Program     ",
    (void *)K9f1g08_PrintBlock,		"K9f1g08 Pr BlkPage  ",
    (void *)1,			    	"Exit                ",
    0,0
};


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

    printf("\n[K9f1g08 NAND Flash JTAG Programmer]\n");
    K9f1g08_JtagInit();
    NF_Init();

    id=NF_CheckId();
    
	if(id!=0xeca1) // K9F1G08Q0A Device Id = A1
    {
	printf("ERROR: K9f1g08(0xeca1) is not detected. Detected ID=0x%x.\n",id);
	return;
    }
    else
    {
    	printf("K9f1g08(0xeca1) is detected. ID=0x%x\n",id);
    }

    while(1)
    {

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

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



void K9f1g08_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 spareBuf[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,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};

    U8 *srcPt;
    U32 progSize=0;

    printf("\n[SMC(K9f1g08Q0M) NAND Flash Writing Program]\n");
    
    printf("\nSource size:0h~%xh\n",imageSize-1);
    printf("\nAvailable target block number: 0~1024\n");
    printf("Input target block number:");
    scanf("%d",&targetBlock);
    targetSize=((imageSize+0x20000-1)/0x20000)*0x20000;
    printf("target start block number     =%d\n",targetBlock);
    printf("target size        (0x20000*n) =0x%x\n",targetSize);
    printf("STATUS:");
    blockIndex=targetBlock;
    while(1)
    {
	if(noLoad==0)
	{
	    LoadImageFile(blockBuf,0x20000);
	}
	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;
	}

	printf("E");
	srcPt=blockBuf;

	for(i=0;i<64;i++)    // if y download the testcode or OS image, y replace 4 to 64.
	{
	    if(!NF_WritePage(blockIndex,i,srcPt,NULL/*spareBuf*/))// block num, page num, buffer
	    {
	        programError=1;
	        break;
	    }

	   // srcPt+=512;	// Increase buffer raddr one pase size
		srcPt+=2048;	// Increase buffer raddr one pase size
	    printf("p");
	}
	printf("\n");

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




void K9f1g08_PrintBlock(void)// Printf one page
{
    int i;
    U16 id;
    U32 block,page;
    //U8	buffer[512+16];
	U8	buffer[2048+64];

    printf("\n[SMC(K9f1g08) NAND Flash block read]\n");	
    
    NF_Init();
    id=NF_CheckId();
    printf("ID=%x(0xeca1)\n",id);
    if(id!=0xeca1)
	return;

    printf("Input target block number:");
    scanf("%d",&block);
    printf("Input target page number:");   
    scanf("%d",&page);
    
    NF_ReadPage(block,page,buffer,buffer+2048);
    
    printf("block=%d,page=%d:",block,page);
    for(i=0;i<2048;i++)
    {
        if(i%16==0)
	    printf("\n%3xh:",i);
        printf("%02x ",buffer[i]);
    }
    printf("\nS.A.:",i);

    for(i=2048;i<2048+64;i++)
    {
        printf("%02x ",buffer[i]);
    }

    printf("\n");    	
}

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

// NAND Flash Memory Commands
#define	SEQ_RDATA_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_RADDR(raddr)	{rNFRADDR=raddr;}	
#define NF_nFCE_L()	{rNFCONF&=~(1<<11);}
#define NF_nFCE_H()	{rNFCONF|=(1<<11);}
#define NF_RSTECC()	{rNFCONF|=(1<<12);}
#define NF_RDRDATA() 	(rNFRDATA)
#define NF_WRRDATA(rdata) {rNFRDATA=rdata;}

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

static U8 seBuf[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,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};


// 1block=(2048+64)bytes x 64pages
// 1024block

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

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

#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_CMD(0x60);   // Erase one block 1st command

    NF_RADDR( (U8)(blockPage&0xff) );	    // Page number=A[7:0]=0
	NF_RADDR( (U8)((blockPage>>8)&0xff) );	    // Page number=A[11:8]=0
    //NF_RADDR((blockPage>>16)&0xff);   
    //NF_RADDR((blockPage>>24)&0xff);

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

    //NF_WAITRB();    // Wait tBERS max 3ms.
    NF_CMD(0x70);   // Read status command

    if (NF_RDRDATA()&0x1) // Erase error
    {	
    	NF_nFCE_H();
	printf("[ERASE_ERROR:block#=%d]\n",block);
	NF_MarkBadBlock(block);
	return 0;
    }
    else 
    {
    	NF_nFCE_H();
        return 1;
    }
}


static int NF_IsBadBlock(U32 block)
{
    unsigned int blockPage;
    U8 rdata;
    
    
    blockPage=(block<<6);	// For 2'nd cycle I/O[7:5] 
    
    NF_nFCE_L();        // 


    //NF_CMD(0x50);		// Spare array read command
	NF_CMD(0x0);		// read command
    NF_RADDR((1024+0) & 0xf);		// 
    NF_RADDR(((1024+0)>>8) & 0xff);	// 
	NF_RADDR((blockPage) & 0xff);	// 
    NF_RADDR((blockPage>>8)&0xff);   // 
  
	NF_CMD(0x30);		// read command