#include  <5410ad.h>
#include <intrins.h>        /* use _nop_() function */

/* 定義命令 */
#define READ_AP_and_Data_Memory_Command				0x01        /*  字節讀數據存儲區   */
#define PROGRAM_AP_and_Data_Memory_Command          0x02        /*  字節編程數據存儲區 */
#define SECTOR_ERASE_AP_and_Data_Memory_Command     0x03        /*  扇區擦除數據存儲區 */

typedef     unsigned char	INT8U;		/* 8 bit 無符號整型  */
typedef     unsigned int    INT16U;     /* 16 bit 無符號整型 */

/* 定義常量 */
#define ERROR   0
#define OK      1

sbit        Begin_LED       =		P1^0;
sbit        ERROR_LED       =       P1^3;
sbit        OK_LED			=       P1^7;

/* 定義Flash 操作等待時間 */
#define        MCU_CLOCK_40MHz
//#define         MCU_CLOCK_20MHz
//#define        MCU_CLOCK_10MHz
//#define        MCU_CLOCK_5MHz
#ifdef MCU_CLOCK_40MHz
        #define WAIT_TIME        0x00
#endif
#ifdef MCU_CLOCK_20MHz
        #define WAIT_TIME        0x01
#endif
#ifdef MCU_CLOCK_10MHz
        #define WAIT_TIME        0x02
#endif
#ifdef MCU_CLOCK_5MHz
        #define WAIT_TIME        0x03
#endif

//#define USED_BYTE_QTY_IN_ONE_SECTOR                1
//#define USED_BYTE_QTY_IN_ONE_SECTOR                2
//#define USED_BYTE_QTY_IN_ONE_SECTOR                4
//#define USED_BYTE_QTY_IN_ONE_SECTOR                8
//#define USED_BYTE_QTY_IN_ONE_SECTOR                16
#define USED_BYTE_QTY_IN_ONE_SECTOR                32
//#define USED_BYTE_QTY_IN_ONE_SECTOR                64
//#define USED_BYTE_QTY_IN_ONE_SECTOR                128
//#define USED_BYTE_QTY_IN_ONE_SECTOR                256
//#define USED_BYTE_QTY_IN_ONE_SECTOR                512

INT8U xdata protect_buffer[USED_BYTE_QTY_IN_ONE_SECTOR];

/* 打開 ISP,IAP 功能 */
void ISP_IAP_enable(void)
{
	EA	=	0;	/* 關中斷 */
	ISP_CONTR	=	ISP_CONTR & 0x18;       /* 0001,1000 */
	ISP_CONTR	=	ISP_CONTR | WAIT_TIME;
	ISP_CONTR	=	ISP_CONTR | 0x80;       /* 1000,0000 */
}

/* 關閉 ISP,IAP 功能 */
void ISP_IAP_disable(void)
{
	ISP_CONTR	=	ISP_CONTR & 0x7f;	/* 0111,1111 */
	ISP_TRIG	=	0x00;
	EA			=   1;                	/* 開中斷 */
}

/* 字節讀 */
INT8U byte_read(INT16U byte_addr)
{
	ISP_ADDRH	=	(INT8U)(byte_addr >> 8);
	ISP_ADDRL	=	(INT8U)(byte_addr & 0x00ff);

	ISP_CMD		=	ISP_CMD	&	0xf8;        /* 1111,1000 */
	ISP_CMD		=	ISP_CMD	|	READ_AP_and_Data_Memory_Command;        /* 0000,0001 */

	ISP_IAP_enable();

	ISP_TRIG	=	0x46;
	ISP_TRIG	=	0xb9;
	_nop_();

	ISP_IAP_disable();
	return (ISP_DATA);
}

/* 寫數據進數據Flash存儲器(EEPROM), 只在同一個扇區內寫，保留同一扇區中不需修改的數據	*/
/* begin_addr,被寫數據Flash開始地址；counter,連續寫多少個字節； array[]，數據來源		*/
INT8U write_flash_with_protect_in_one_sector(INT16U begin_addr, INT16U counter, INT8U array[])
{
	INT16U	i	=	0;
	INT16U	in_sector_begin_addr	=	0;
	INT16U	sector_addr	=	0;
	INT16U	byte_addr	=	0;

	/* 判是否是有效范圍,此函數不允許跨扇區操作 */
	if(counter > USED_BYTE_QTY_IN_ONE_SECTOR)
		return ERROR;
	in_sector_begin_addr =        begin_addr & 0x01ff;         /* 0000,0001,1111,1111 */
	/* 假定從扇區的第0個字節開始，到USED_BYTE_QTY_IN_ONE_SECTOR-1個字節結束,后面部分不用,程序易編寫	*/
	if( (in_sector_begin_addr + counter) > USED_BYTE_QTY_IN_ONE_SECTOR )
		return ERROR;

	/* 將該扇區數據 0 - (USED_BYTE_QTY_IN_ONE_SECTOR-1) 字節數據讀入緩沖區保護 */
	sector_addr		=	(begin_addr & 0xfe00); 	/* 1111,1110,0000,0000; 取扇區地址		*/
	byte_addr		=   sector_addr;			/* 扇區地址為扇區首字節地址			 	*/

	ISP_IAP_enable();
	for(i = 0; i < USED_BYTE_QTY_IN_ONE_SECTOR; i++)
	{
		ISP_ADDRH	=	(INT8U)(byte_addr >> 8);
		ISP_ADDRL	=	(INT8U)(byte_addr & 0x00ff);

		ISP_CMD		=	ISP_CMD	&	0xf8;        /* 1111,1000 */
		ISP_CMD		=	ISP_CMD	|	READ_AP_and_Data_Memory_Command;	/* 0000,0001 */

		ISP_TRIG	=	0x46;
		ISP_TRIG	=	0xb9;
		_nop_();

		protect_buffer[i]	=	ISP_DATA;
		byte_addr++;
	}

	/* 將要寫入的數據寫入保護緩沖區的相應區域,其余部分保留 */
	for(i = 0; i < counter; i++)
	{
		protect_buffer[in_sector_begin_addr] = array[i];
		in_sector_begin_addr++;
	}

	/* 擦除 要修改/寫入 的扇區 */
	ISP_ADDRH	=	(INT8U)(sector_addr >> 8);
	ISP_ADDRL	=	0x00;
	ISP_CMD		=	ISP_CMD	&	0xf8;        /* 1111,1000 */
	ISP_CMD		=	ISP_CMD	|	SECTOR_ERASE_AP_and_Data_Memory_Command;	/* 0000,0011 */

	ISP_TRIG	=	0x46;        /* 觸發ISP_IAP命令 */
	ISP_TRIG	=	0xb9;        /* 觸發ISP_IAP命令 */
	_nop_();

	/* 將保護緩沖區的數據寫入 Data Flash, EEPROM */
	byte_addr	=   sector_addr;			/* 扇區地址為扇區首字節地址	*/
	for(i = 0; i< USED_BYTE_QTY_IN_ONE_SECTOR; i++)
	{
		/* 寫一個字節 */
		ISP_ADDRH	=	(INT8U)(byte_addr >> 8);
		ISP_ADDRL	=	(INT8U)(byte_addr & 0x00ff);
		ISP_DATA	=	protect_buffer[i];
		ISP_CMD		=	ISP_CMD	&	0xf8;        /* 1111,1000 */
		ISP_CMD		=	ISP_CMD	|	PROGRAM_AP_and_Data_Memory_Command;		/* 0000,0010 */

		ISP_TRIG	=	0x46;        /* 觸發ISP_IAP命令 */
		ISP_TRIG	=	0xb9;        /* 觸發ISP_IAP命令 */
		_nop_();

		/* 讀回來 */
		ISP_DATA	=	0x00;

		ISP_CMD		=	ISP_CMD	&	0xf8;        /* 1111,1000 */
		ISP_CMD		=	ISP_CMD	|	READ_AP_and_Data_Memory_Command;	/* 0000,0001 */

		ISP_TRIG	=	0x46;        /* 觸發ISP_IAP命令 */
		ISP_TRIG	=	0xb9;        /* 觸發ISP_IAP命令 */
		_nop_();

		/*  比較對錯 */
		if(ISP_DATA != protect_buffer[i])
		{
			ISP_IAP_disable();
			return ERROR;
        }
        byte_addr++;
	}
	ISP_IAP_disable();
	return OK;
}