/*
	Skelton for retropc emulator

	Author : Takeda.Toshiya
	Date   : 2010.09.16-

	[ memory ]
*/

#include "memory.h"

#define ADDR_MASK (addr_max - 1)
#define BANK_MASK (bank_size - 1)

void MEMORY::initialize()
{
	// allocate tables here to support multiple instances with different address range
	if(rd_table == NULL) {
		int64_t bank_num = addr_max / bank_size;
		
		rd_dummy = (uint8_t *)malloc(bank_size);
		wr_dummy = (uint8_t *)malloc(bank_size);
		
		rd_table = (bank_t *)calloc(bank_num, sizeof(bank_t));
		wr_table = (bank_t *)calloc(bank_num, sizeof(bank_t));
		
		for(int i = 0; i < bank_num; i++) {
			rd_table[i].dev = NULL;
			rd_table[i].memory = rd_dummy;
			rd_table[i].wait = 0;
			
			wr_table[i].dev = NULL;
			wr_table[i].memory = wr_dummy;
			rd_table[i].wait = 0;
		}
		for(int i = 0;; i++) {
			if(bank_size == (int64_t)(1 << i)) {
				addr_shift = i;
				break;
			}
		}
		memset(rd_dummy, 0xff, bank_size);
	}
}

void MEMORY::release()
{
	free(rd_table);
	free(wr_table);
	free(rd_dummy);
	free(wr_dummy);
}

uint32_t MEMORY::read_data8(uint32_t addr)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(rd_table[bank].dev != NULL) {
		return rd_table[bank].dev->read_memory_mapped_io8(addr);
	} else {
		return rd_table[bank].memory[addr & BANK_MASK];
	}
}

void MEMORY::write_data8(uint32_t addr, uint32_t data)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(wr_table[bank].dev != NULL) {
		wr_table[bank].dev->write_memory_mapped_io8(addr, data);
	} else {
		wr_table[bank].memory[addr & BANK_MASK] = data;
	}
}

uint32_t MEMORY::read_data16(uint32_t addr)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(rd_table[bank].dev != NULL) {
		return rd_table[bank].dev->read_memory_mapped_io16(addr);
	} else {
		int bank1 = ((addr + 1) & ADDR_MASK) >> addr_shift;
		if(bank != bank1) {
			uint32_t val = read_data8(addr);
			val |= read_data8(addr + 1) << 8;
			return val;
		} else {
			uint8_t* p = (uint8_t*)(&(rd_table[bank].memory[addr & BANK_MASK]));
			uint32_t val;
#if defined(__LITTLE_ENDIAN__)
			uint16_t* pp = (uint16_t*)p;
			val = (uint32_t)(*pp);
#else			
			val = p[0];
			val =  (val << 8) | ((uint32_t)(p[1]));
#endif
			return val;
		}
	}
}

void MEMORY::write_data16(uint32_t addr, uint32_t data)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(wr_table[bank].dev != NULL) {
		wr_table[bank].dev->write_memory_mapped_io16(addr, data);
	} else {
		int bank1 = ((addr + 1) & ADDR_MASK) >> addr_shift;
		if(bank != bank1) {
			write_data8(addr, data & 0xff);
			write_data8(addr + 1, (data >> 8) & 0xff);
		} else {
			uint8_t* p = (uint8_t*)(&(wr_table[bank].memory[addr & BANK_MASK]));
#if defined(__LITTLE_ENDIAN__)
			uint16_t* pp = (uint16_t*)p;
			*pp = (uint16_t)data;
#else
			p[0] = data & 0xff;
			p[1] = (data >> 8) & 0xff;
#endif
		}			
	}
}

uint32_t MEMORY::read_data32(uint32_t addr)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(rd_table[bank].dev != NULL) {
		return rd_table[bank].dev->read_memory_mapped_io32(addr);
	} else {
		int bank1 = ((addr + 3) & ADDR_MASK) >> addr_shift;
		if(bank != bank1) {
			uint32_t val = read_data16(addr);
			val |= read_data16(addr + 2) << 16;
			return val;
		} else {
			uint8_t* p = (uint8_t*)(&(rd_table[bank].memory[addr & BANK_MASK]));
			uint32_t val;
#if defined(__LITTLE_ENDIAN__)
			uint32_t* pp = (uint32_t*)p;
			val = *pp;
#else
			val = ((uint32_t)p[0]) | (((uint32_t)p[1]) << 8) | (((uint32_t)p[2]) << 16) |(((uint32_t)p[3]) << 24);
#endif
			return val;
		}
	}
}

void MEMORY::write_data32(uint32_t addr, uint32_t data)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(wr_table[bank].dev != NULL) {
		wr_table[bank].dev->write_memory_mapped_io32(addr, data);
	} else {
		int bank1 = ((addr + 3) & ADDR_MASK) >> addr_shift;
		if(bank != bank1) {
			write_data16(addr, data & 0xffff);
			write_data16(addr + 2, (data >> 16) & 0xffff);
		} else {
			uint8_t* p = (uint8_t*)(&(wr_table[bank].memory[addr & BANK_MASK]));
#if defined(__LITTLE_ENDIAN__)
			uint32_t* pp = (uint32_t*)p;
			*pp = data;
#else
			p[0] = data & 0xff;
			p[1] = (data >> 8) & 0xff;
			p[2] = (data >> 16) & 0xff;
			p[3] = (data >> 24) & 0xff;
#endif
		}
	}
}

uint32_t MEMORY::read_data8w(uint32_t addr, int* wait)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	*wait = rd_table[bank].wait;
	if(rd_table[bank].dev != NULL) {
		return rd_table[bank].dev->read_memory_mapped_io8(addr);
	} else {
		return rd_table[bank].memory[addr & BANK_MASK];
	}
}

void MEMORY::write_data8w(uint32_t addr, uint32_t data, int* wait)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	*wait = wr_table[bank].wait;
	if(wr_table[bank].dev != NULL) {
		wr_table[bank].dev->write_memory_mapped_io8(addr, data);
	} else {
		wr_table[bank].memory[addr & BANK_MASK] = data;
	}
}

uint32_t MEMORY::read_data16w(uint32_t addr, int* wait)
{
	int wait_l, wait_h;
	uint32_t val = read_data8w(addr, &wait_l);
	val |= read_data8w(addr + 1, &wait_h) << 8;
	*wait = wait_l + wait_h;
	return val;
}

void MEMORY::write_data16w(uint32_t addr, uint32_t data, int* wait)
{
	int wait_l, wait_h;
	write_data8w(addr, data & 0xff, &wait_l);
	write_data8w(addr + 1, (data >> 8) & 0xff, &wait_h);
	*wait = wait_l + wait_h;
}

uint32_t MEMORY::read_data32w(uint32_t addr, int* wait)
{
	int wait_l, wait_h;
	uint32_t val = read_data16w(addr, &wait_l);
	val |= read_data16w(addr + 2, &wait_h) << 16;
	*wait = wait_l + wait_h;
	return val;
}

void MEMORY::write_data32w(uint32_t addr, uint32_t data, int* wait)
{
	int wait_l, wait_h;
	write_data16w(addr, data & 0xffff, &wait_l);
	write_data16w(addr + 2, (data >> 16) & 0xffff, &wait_h);
	*wait = wait_l + wait_h;
}

#ifdef MEMORY_DISABLE_DMA_MMIO
uint32_t MEMORY::read_dma_data8(uint32_t addr)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(rd_table[bank].dev != NULL) {
//		return rd_table[bank].dev->read_memory_mapped_io8(addr);
		return 0xff;
	} else {
		return rd_table[bank].memory[addr & BANK_MASK];
	}
}

void MEMORY::write_dma_data8(uint32_t addr, uint32_t data)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(wr_table[bank].dev != NULL) {
//		wr_table[bank].dev->write_memory_mapped_io8(addr, data);
	} else {
		wr_table[bank].memory[addr & BANK_MASK] = data;
	}
}

uint32_t MEMORY::read_dma_data16(uint32_t addr)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(rd_table[bank].dev != NULL) {
//		return rd_table[bank].dev->read_memory_mapped_io16(addr);
		return 0xffff;
	} else {
		uint32_t val = read_dma_data8(addr);
		val |= read_dma_data8(addr + 1) << 8;
		return val;
	}
}

void MEMORY::write_dma_data16(uint32_t addr, uint32_t data)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(wr_table[bank].dev != NULL) {
//		wr_table[bank].dev->write_memory_mapped_io16(addr, data);
	} else {
		write_dma_data8(addr, data & 0xff);
		write_dma_data8(addr + 1, (data >> 8) & 0xff);
	}
}

uint32_t MEMORY::read_dma_data32(uint32_t addr)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(rd_table[bank].dev != NULL) {
//		return rd_table[bank].dev->read_memory_mapped_io32(addr);
		return 0xffffffff;
	} else {
		uint32_t val = read_dma_data16(addr);
		val |= read_dma_data16(addr + 2) << 16;
		return val;
	}
}

void MEMORY::write_dma_data32(uint32_t addr, uint32_t data)
{
	int bank = (addr & ADDR_MASK) >> addr_shift;
	
	if(wr_table[bank].dev != NULL) {
//		wr_table[bank].dev->write_memory_mapped_io32(addr, data);
	} else {
		write_dma_data16(addr, data & 0xffff);
		write_dma_data16(addr + 2, (data >> 16) & 0xffff);
	}
}
#endif

// register

void MEMORY::set_memory_r(uint32_t start, uint32_t end, uint8_t *memory)
{
	MEMORY::initialize(); // subclass may overload initialize()
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		rd_table[i].dev = NULL;
		rd_table[i].memory = memory + bank_size * (i - start_bank);
	}
}

void MEMORY::set_memory_w(uint32_t start, uint32_t end, uint8_t *memory)
{
	MEMORY::initialize();
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		wr_table[i].dev = NULL;
		wr_table[i].memory = memory + bank_size * (i - start_bank);
	}
}

void MEMORY::set_memory_mapped_io_r(uint32_t start, uint32_t end, DEVICE *device)
{
	MEMORY::initialize();
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		rd_table[i].dev = device;
	}
}

void MEMORY::set_memory_mapped_io_w(uint32_t start, uint32_t end, DEVICE *device)
{
	MEMORY::initialize();
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		wr_table[i].dev = device;
	}
}

void MEMORY::set_wait_r(uint32_t start, uint32_t end, int wait)
{
	MEMORY::initialize();
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		rd_table[i].wait = wait;
	}
}

void MEMORY::set_wait_w(uint32_t start, uint32_t end, int wait)
{
	MEMORY::initialize();
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		wr_table[i].wait = wait;
	}
}

void MEMORY::unset_memory_r(uint32_t start, uint32_t end)
{
	MEMORY::initialize();
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		rd_table[i].dev = NULL;
		rd_table[i].memory = rd_dummy;
	}
}

void MEMORY::unset_memory_w(uint32_t start, uint32_t end)
{
	MEMORY::initialize();
	
	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		wr_table[i].dev = NULL;
		wr_table[i].memory = wr_dummy;
	}
}

void MEMORY::copy_table_w(uint32_t to, uint32_t start, uint32_t end)
{
	MEMORY::initialize();

	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	uint32_t to_bank = to >> addr_shift;
	int blocks = (int)(addr_max / bank_size);
	
	for(uint32_t i = start_bank; i <= end_bank; i++) {
		if(to_bank >= blocks) break;
		wr_table[to_bank].dev = wr_table[i].dev;
		wr_table[to_bank].memory = wr_table[i].memory;
		wr_table[to_bank].wait = wr_table[i].wait;
		to_bank++;
	}
}

void MEMORY::copy_table_r(uint32_t to, uint32_t start, uint32_t end)
{
	MEMORY::initialize();

	uint32_t start_bank = start >> addr_shift;
	uint32_t end_bank = end >> addr_shift;
	uint32_t to_bank = to >> addr_shift;
	int blocks = (int)(addr_max / bank_size);

	for(uint32_t i = start_bank; i <= end_bank; i++) {
		if(to_bank >= blocks) break;
		rd_table[to_bank].dev = rd_table[i].dev;
		rd_table[to_bank].memory = rd_table[i].memory;
		rd_table[to_bank].wait = rd_table[i].wait;
		to_bank++;
	}
}

// load/save image

int MEMORY::read_bios(const _TCHAR *file_name, uint8_t *buffer, int size)
{
	FILEIO* fio = new FILEIO();
	int length = 0;
	
	if(fio->Fopen(create_local_path(file_name), FILEIO_READ_BINARY)) {
		fio->Fread(buffer, size, 1);
		length = fio->Ftell();
		fio->Fclose();
	}
	delete fio;
	return length;
}

bool MEMORY::write_bios(const _TCHAR *file_name, uint8_t *buffer, int size)
{
	FILEIO* fio = new FILEIO();
	bool result = false;
	
	if(fio->Fopen(create_local_path(file_name), FILEIO_WRITE_BINARY)) {
		fio->Fwrite(buffer, size, 1);
		fio->Fclose();
		result = true;
	}
	delete fio;
	return result;
}

bool MEMORY::read_image(const _TCHAR *file_path, uint8_t *buffer, int size)
{
	FILEIO* fio = new FILEIO();
	bool result = false;
	
	if(fio->Fopen(file_path, FILEIO_READ_BINARY)) {
		fio->Fread(buffer, size, 1);
		fio->Fclose();
		result = true;
	}
	delete fio;
	return result;
}

bool MEMORY::write_image(const _TCHAR* file_path, uint8_t* buffer, int size)
{
	FILEIO* fio = new FILEIO();
	bool result = false;
	
	if(fio->Fopen(file_path, FILEIO_WRITE_BINARY)) {
		fio->Fwrite(buffer, size, 1);
		fio->Fclose();
		result = true;
	}
	delete fio;
	return result;
}
