#include <stdio.h>
#include <stdlib.h>
#include "def.h"

#include "pin2410.h"
#include "Jtag.h"
#include "sjf2410.h"

#include "mem_rdwr.h"

#define TARGET_ADDR_29LV160     0x0  // nGCS0, 128MB area

#define _WR(addr,data)	MRW_Wr16QQ(addr<<1,data,0,0x3)
#define _RD(addr)	    MRW_Rd16Q(addr<<1,0,0x3)
#define _RESET()	    _WR(0x0,0xf0f0)
#define BADDR2WADDR(addr)   (addr>>1)  



static U32 srcAddress;
static U32 srcOffset;

static U32 targetAddress;
static U32 targetOffset;
static U32 targetSize;

static U32 sectorOffset;
static U32 sectorSize;
static U32 totalSize;

HY29lv160_CheckId()
{
	U8 tmp;
	U16 manId,devId;
	int i;

    _RESET();
	
	 _WR(0x555,0xaaaa);
    _WR(0x2aa,0x5555);
    _WR(0x555,0x9090);
    manId=_RD(0x0);

    _RESET(); // New 5V AM29F800 needs this command. 
    _WR(0x555,0xaaaa);
    _WR(0x2aa,0x5555);
    _WR(0x555,0x9090);
    devId=_RD(0x1);

    _RESET();   

    //_RESET();   

    printf("Manufacture ID=%4x(0x0001), Device ID(0x2249)=%4x\n",manId,devId);
		return 1;
	if(manId == 0xad && devId == 0x2249 )
		return 1;
    else 
		return 0;
}

int _WAIT(void) //Check if the bit6 toggle ends.
{
        volatile U16 flashStatus,old;

	old=_RD(BADDR2WADDR(0x0));

        while(1)
	{
	    flashStatus=_RD(BADDR2WADDR(0x0));
            if( (old&0x40) == (flashStatus&0x40) )break;
            if( flashStatus&0x20 )
	    {
		//printf("[DQ5=1:%x]\n",flashStatus);
		old=_RD(BADDR2WADDR(0x0));
		flashStatus=_RD(BADDR2WADDR(0x0));
		if( (old&0x40) == (flashStatus&0x40) )
		    return 0;
		else return 1;
	    }
	    //printf(".");
	    old=flashStatus;
        }
        //printf("!\n");
	return 1;
}

static void InputTargetOffset(void)
{
    printf("\nImage Size:0h~%xh\n",targetSize);
    printf("\nAvailable Target Offset:\n"); 
    printf("	 0x0, 0x4000, 0x6000, 0x8000,0x10000,0x20000,0x30000,0x40000,\n");
    printf("0x50000,0x60000,0x70000,0x80000,0x90000,0xa0000,0xb0000,0xc0000,\n");
    printf("0xd0000,0xe0000,0xf0000\n");

    printf("Input target offset:");
    scanf("%x",&targetOffset);
}


void GetSectorSize()
{
	if(sectorOffset<0x4000){sectorSize=0x4000;}
    else if(sectorOffset<0x6000)sectorSize=0x2000;
    else if(sectorOffset<0x8000)sectorSize=0x2000;
    else if(sectorOffset<0x10000)sectorSize=0x8000; 
    else sectorSize=0x10000;

    printf("SectorOffset=0x%x\n",sectorOffset);
    printf("SectorSize  =0x%x\n",sectorSize);
}

void HY29lv160_EraseSector(int targetAddr)
{
	printf("Sector Erase is started!\n");

        _RESET();

        _WR(0x555,0xaaaa);
        _WR(0x2aa,0x5555);
        _WR(0x555,0x8080);
        _WR(0x555,0xaaaa);
        _WR(0x2aa,0x5555);
        _WR(BADDR2WADDR(targetAddr),0x3030);
        
		_WAIT(); 
        
		_RESET();
}

int HY29lv160_ProgFlash(U32 realAddr,U16 data)
{
        _WR(0x555,0xaaaa);
        _WR(0x2aa,0x5555);
        _WR(0x555,0xa0a0);

        _WR(BADDR2WADDR(realAddr),data);
	//return _WAIT(); //not needed at JTAG access
	return 1;
}

void ProgramHY29lv160(void)
{
	int i;

    printf("\n[HY29lv160 Writing Program]\n");
    MRW_JtagInit();

    printf("[Check HY29lv160]\n");
    if(!HY29lv160_CheckId())
    {
		printf("ID Check Error!!!\n");
		return;
    }
	
	targetSize=imageSize;

    srcAddress=(U32)malloc(targetSize+4);
    if(srcAddress==0)return;
    LoadImageFile((U8 *)srcAddress,targetSize);

    srcOffset=0;
    
    targetAddress=TARGET_ADDR_29LV160;
    InputTargetOffset();
    
    totalSize=0;
    sectorOffset=targetOffset;

    printf("\n");

    while(totalSize<imageSize)
    {
        GetSectorSize(); //sectorSize is updated.

		printf("Erase the sector:0x%x.\n",targetAddress+sectorOffset);
    
		HY29lv160_EraseSector(targetAddress+sectorOffset);

		printf("Start of the sector data writing.\n");

		for(i=0x0;i<sectorSize;i+=2) 
		{
			HY29lv160_ProgFlash(targetAddress+sectorOffset+i,*( (U16 *)(srcAddress+srcOffset+i) )  );
			if((i%0x100)==0)printf("%x ",i);
			totalSize+=2;
			if(totalSize>=targetSize) break;
		}
	
		printf("\nEnd of the sector data writing!!!\n\n");

		_RESET();
		sectorOffset+=sectorSize;
		srcOffset+=sectorSize;
    }
}