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

#include "pin2410.h"
#include "Jtag.h"
#include "sjf2410.h"
#include "an29lv160dt.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;

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

    _RESET();
	
    _WR(0x555,0xaa);
    _WR(0x2aa,0x55);
    _WR(0x555,0x90);
    manId=_RD(0x0);

    _RESET();  
    _WR(0x555,0xaa);
    _WR(0x2aa,0x55);
    _WR(0x555,0x90);
    devId=_RD(0x1);

    _RESET();   

    printf("Manufacture ID=%4x(0x0001), Device ID(0x22c4)=%4x\n",manId,devId);

    if(manId == 0x1 && devId == 0x22c4)
		return 1;
    else 
		return 0;
}

int AM_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, 0x10000,0x20000,0x30000,0x40000,\n 0x50000,0x60000,0x70000,0x80000,0x90000,0xa0000,0xb0000,0xc0000,\n");
    printf("0xd0000,0xe0000,0xf0000\n");

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


void AMGetSectorSize()
{
	sectorSize=0x10000;

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

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

    _RESET();

    _WR(0x555,0xaa);
    _WR(0x2aa,0x55);
    _WR(0x555,0x80);
    _WR(0x555,0xaa);
    _WR(0x2aa,0x55);
    _WR(BADDR2WADDR(targetAddr),0x30);
        
	_RESET();
}

int AM29lv160dt_ProgFlash(U32 realAddr,U16 data)
{
        _WR(0x555,0xaa);
        _WR(0x2aa,0x55);
        _WR(0x555,0xa0);

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

void ProgramAM29lv160dt(void)
{
	int i;

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

    printf("[Check AM29lv160dt]\n");
    if(!AM29lv160dt_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");
//AM29lv160dt_ProgFlash(0x0,0x74);

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

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

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

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

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