//C library for W25Q64 Flash IC SPI


#define MAX_BLOCKSIZE         128   
#define MAX_SECTORSIZE        2048  

#define CMD_WRIRE_ENABLE      0x06
#define CMD_WRITE_DISABLE     0x04
#define CMD_READ_STATUS_R1    0x05
#define CMD_READ_STATUS_R2    0x35
#define CMD_WRITE_STATUS_R    0x01 
#define CMD_PAGE_PROGRAM      0x02
#define CMD_QUAD_PAGE_PROGRAM 0x32 
#define CMD_BLOCK_ERASE64KB   0xd8
#define CMD_BLOCK_ERASE32KB   0x52
#define CMD_SECTOR_ERASE      0x20
#define CMD_CHIP_ERASE        0xC7
#define CMD_ERASE_SUPPEND     0x75 
#define CMD_ERASE_RESUME      0x7A 
#define CMD_POWER_DOWN        0xB9
#define CMD_HIGH_PERFORM_MODE 0xA3 
#define CMD_CNT_READ_MODE_RST 0xFF 
#define CMD_RELEASE_PDOWN_ID  0xAB 
#define CMD_MANUFACURER_ID    0x90
#define CMD_READ_UNIQUE_ID    0x4B
#define CMD_JEDEC_ID          0x9f
#define CMD_RESET	          0x99

#define CMD_READ_DATA         0x03
#define CMD_FAST_READ         0x0B
#define CMD_READ_DUAL_OUTPUT  0x3B 
#define CMD_READ_DUAL_IO      0xBB 
#define CMD_READ_QUAD_OUTPUT  0x6B 
#define CMD_READ_QUAD_IO      0xEB 
#define CMD_WORD_READ         0xE3 

#define SR1_BUSY_MASK	0x01
#define SR1_WEN_MASK	0x02


PORT_t	*csport;
register8_t cspin;


inline void W25Q64_init(SPI_t *spiSPI, PORT_t *spiPort, PORT_t *cs_port, uint8_t cs_pin)
{
	//done by other functions
	//SPI_MasterInit(&spiMasterD, spiSPI, spiPort, FALSE, SPI_MODE_0_gc, SPI_INTLVL_OFF_gc, FALSE, SPI_PRESCALER_DIV4_gc); 

	cs_port->DIR |= (1<<cs_pin);
	cs_port->OUT |= (1<<cs_pin);

	cspin = cs_pin;
	csport = cs_port;
}

uint8_t W25Q64_readStatusReg1(void) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_READ_STATUS_R1);
	uint8_t rc = SPI_MasterTransceiveByte(&spiMasterD, 0x00);
	csport->OUT |= (1<<cspin);

	return rc;
}

uint8_t W25Q64_readStatusReg2(void) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_READ_STATUS_R2);
	uint8_t rc = SPI_MasterTransceiveByte(&spiMasterD, 0x00);
	csport->OUT |= (1<<cspin);

	return rc;
}

void W25Q64_readManufacturer(uint8_t* d3) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_JEDEC_ID);
	
	for (uint8_t i = 0; i < 3; i++) d3[i] = SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	csport->OUT |= (1<<cspin);
}

void W25Q64_readDeviceID(uint8_t* d2) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_MANUFACURER_ID);
	
	for (uint8_t i = 0; i < 3; i++) SPI_MasterTransceiveByte(&spiMasterD, 0x00);
	for (uint8_t i = 0; i < 2; i++) d2[i] = SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	csport->OUT |= (1<<cspin);
}

void W25Q64_readUniqieID(uint8_t* d8)
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_READ_UNIQUE_ID);
	
	for (uint8_t i = 0; i < 4; i++) SPI_MasterTransceiveByte(&spiMasterD, 0x00);
	for (uint8_t i = 0; i < 8; i++) d8[i] = SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	csport->OUT |= (1<<cspin);
}

uint8_t W25Q64_IsBusy(void) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_READ_STATUS_R1);
	uint8_t rc = SPI_MasterTransceiveByte(&spiMasterD, 0x00);
	csport->OUT |= (1<<cspin);

	return (rc & SR1_BUSY_MASK);
}

void W25Q64_powerDown(void) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_POWER_DOWN);
	csport->OUT |= (1<<cspin);
}

void W25Q64_reset(void)
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_RESET);
	csport->OUT |= (1<<cspin);
}

void W25Q64_WriteEnable(void) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_WRIRE_ENABLE);
	csport->OUT |= (1<<cspin);
}

void W25Q64_WriteDisable(void) 
{
	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_WRITE_DISABLE);
	csport->OUT |= (1<<cspin);
}

void W25Q64_read(uint32_t addr, uint8_t *buf, uint16_t n)
{
	while(W25Q64_IsBusy());

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_READ_DATA);
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>16));
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>8));
	SPI_MasterTransceiveByte(&spiMasterD, addr);
	SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	for (uint16_t i = 0; i < n; i++) buf[i] = SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	csport->OUT |= (1<<cspin);
}

void W25Q64_fastread(uint32_t addr, uint8_t *buf, uint16_t n) 
{
	while(W25Q64_IsBusy());

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_FAST_READ);
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>16));
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>8));
	SPI_MasterTransceiveByte(&spiMasterD, addr);
	SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	for (uint16_t i = 0; i < n; i++) buf[i] = SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	csport->OUT |= (1<<cspin);
}

uint8_t W25Q64_eraseSector(uint16_t sect_no, uint8_t flgwait) 
{
	uint32_t addr = (sect_no << 12);

	W25Q64_WriteEnable();

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_SECTOR_ERASE);
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>16));
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>8));
	SPI_MasterTransceiveByte(&spiMasterD, addr);
	csport->OUT |= (1<<cspin);

	while(W25Q64_IsBusy() & flgwait) 
	{
		_delay_ms(10);
	}
	return 1;
}

uint8_t W25Q64_erase64Block(uint16_t blk_no, uint8_t flgwait) 
{
	uint32_t addr = (blk_no << 16);

	W25Q64_WriteEnable();

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_BLOCK_ERASE64KB);
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>16));
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>8));
	SPI_MasterTransceiveByte(&spiMasterD, addr);
	csport->OUT |= (1<<cspin);
	
	while(W25Q64_IsBusy() & flgwait) 
	{
		_delay_ms(50);
	}
	return 1;
}

uint8_t W25Q64_erase32Block(uint16_t blk_no, uint8_t flgwait) 
{
	uint32_t addr = (blk_no << 15);

	W25Q64_WriteEnable();

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_BLOCK_ERASE32KB);
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>16));
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>8));
	SPI_MasterTransceiveByte(&spiMasterD, addr);
	csport->OUT |= (1<<cspin);
	
	while(W25Q64_IsBusy() & flgwait) 
	{
		_delay_ms(50);
	}

	return 1;
}

uint8_t W25Q64_eraseAll(uint8_t flgwait) 
{
	W25Q64_WriteEnable();

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_CHIP_ERASE);
	csport->OUT |= (1<<cspin);

	while(W25Q64_IsBusy() & flgwait) 
	{
		_delay_ms(500);
	}

	return 1;
}

void W25Q64_pageWrite(uint16_t sect_no, uint16_t inaddr, uint8_t* buf, uint8_t n) 
{
	uint32_t addr = (sect_no << 12);
	addr += inaddr;

	W25Q64_WriteEnable();
	while(W25Q64_IsBusy());

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_PAGE_PROGRAM);
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>16));
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>8));
	SPI_MasterTransceiveByte(&spiMasterD, addr);
	SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	for (uint8_t i = 0; i < n; i++) SPI_MasterTransceiveByte(&spiMasterD, buf[i]);

	csport->OUT |= (1<<cspin);
}

void W25Q64_pageWriteManipulate(uint16_t sect_no, uint16_t inaddr, uint8_t* buf, uint8_t nPageSplitStart, uint8_t nPageSplitSize, uint8_t pageSSplitAdr)
{
	uint8_t readData[0xFF];

	uint32_t addr = (sect_no << 12);
	addr += inaddr;

	W25Q64_fastread(addr, &readData, 0xFF);

	//manipulate the section data, delete/replace data and copy back

	for (uint8_t i = nPageSplitStart; i < nPageSplitSize; i++)	readData[i] = buf[(i - nPageSplitStart)];

	while(W25Q64_IsBusy());

	W25Q64_WriteEnable();
	while(W25Q64_IsBusy());

	csport->OUT &= ~(1<<cspin);
	SPI_MasterTransceiveByte(&spiMasterD, CMD_PAGE_PROGRAM);
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>16));
	SPI_MasterTransceiveByte(&spiMasterD, (addr>>8));
	SPI_MasterTransceiveByte(&spiMasterD, addr);
	SPI_MasterTransceiveByte(&spiMasterD, 0x00);

	for (uint8_t i = 0; i < 0xFF; i++) SPI_MasterTransceiveByte(&spiMasterD, readData[i]);

	csport->OUT |= (1<<cspin);
}