/*
 * (C) 2008 Dirk W. Hoffmann. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "C64.h"

VC1541Memory::VC1541Memory()
{
    name = "1541MEM";
	debug(2, "  Creating VC1541 memory at %p...\n", this);

    // Register snapshot items
    SnapshotItem items[] = {

    { mem,              0xC000,     CLEAR_ON_RESET },
    { &mem[0xA000],     0x4000,     KEEP_ON_RESET  }, /* VC1541 Rom */
    { NULL,             0,          0 }};

    registerSnapshotItems(items, sizeof(items));

	romFile = NULL;
}

VC1541Memory::~VC1541Memory()
{
	debug(2, "  Releasing VC1541 memory at %p...\n", this);
}

void 
VC1541Memory::reset()
{
    VirtualComponent::reset();
    
    // Establish bindings
    cpu = c64->cpu;
    iec = c64->iec;
    floppy = c64->floppy;    
}

bool 
VC1541Memory::is1541Rom(const char *filename)
{
	int magic_bytes1[] = { 0x97, 0xAA, 0xAA, EOF };
	int magic_bytes2[] = { 0x97, 0xE0, 0x43, EOF };

	if (filename == NULL)
		return false;
	
	if (!checkFileSize(filename, 0x4000, 0x4000))
		return false;
	
	if (!checkFileHeader(filename, magic_bytes1) && !checkFileHeader(filename, magic_bytes2))
		return false;
		
	return true;
}

bool 
VC1541Memory::loadRom(const char *filename)
{
	if (is1541Rom(filename)) {
		romFile = strdup(filename);
		flashRom(filename, 0xC000);
		return true;
	}
	return false;
}

void 
VC1541Memory::dumpState()
{
	msg("VC1541 Memory:\n");
	msg("--------------\n\n");
	msg("VC1541 ROM :%s loaded\n", romIsLoaded() ? "" : " not");
	for (uint16_t i = 0; i < 0xFFFF; i++) {
		uint8_t tag = cpu->getBreakpointTag(i);
		if (tag != CPU::NO_BREAKPOINT) {
			msg("Breakpoint at %0x4X %s\n", i, tag == CPU::SOFT_BREAKPOINT ? "(soft)" : "");
		}
	}
	msg("\n");
}

bool 
VC1541Memory::isValidAddr(uint16_t addr, MemoryType type)
{
	switch (type) {
		case MEM_RAM: 
			return (addr < 0x800);
		case MEM_ROM:
			return (addr >= 0xC000);
		case MEM_IO:
			return (addr >= 0x1800 && addr <= 0x1c00) || (addr >= 0x1c00 && addr <= 0x2000);
		default:
			assert(false);
			return false;
	}
}

uint8_t 
VC1541Memory::peekRam(uint16_t addr)
{
	return mem[addr];
}

uint8_t 
VC1541Memory::peekRom(uint16_t addr)
{
	return mem[addr];
}

uint8_t 
VC1541Memory::peekIO(uint16_t addr)
{	
	if ((addr & 0xFC00) == 0x1800) {
		return floppy->via1.peek(addr & 0x000F);
	} else if ((addr & 0xFC00) == 0x1c00) {
		return floppy->via2.peek(addr & 0x000F);
	} else {
		// Return high byte of addr 
		// VICE and Frodo are doing it that way
		return (addr >> 8);
	}
}

uint8_t 
VC1541Memory::peek(uint16_t addr)
{
	uint8_t result;
	
	if (addr >= 0xc000) { 
		// ROM
		result = mem[addr];
	} else if (addr < 0x1000) { // TODO: Check if 0x1000 is the correct value
		// RAM (bitmask is applied because RAM repeats multiple times)
		result = mem[addr & 0x07ff]; 		
	} else { 
		// IO space
		result = peekIO(addr);
	}
	
	return result;
}
	
void 
VC1541Memory::pokeRam(uint16_t addr, uint8_t value)
{
	mem[addr] = value;
}

void 
VC1541Memory::pokeRom(uint16_t addr, uint8_t value)
{
	mem[addr] = value;
}
             
void 
VC1541Memory::pokeIO(uint16_t addr, uint8_t value)
{	
	if ((addr & 0xFC00) == 0x1800) {
		floppy->via1.poke(addr & 0x000F, value);
	} else if ((addr & 0xFC00) == 0x1c00) {
		floppy->via2.poke(addr & 0X000F, value);
	} else {
		// No memory here, nothing happens
	}
}
			 			 
void 
VC1541Memory::poke(uint16_t addr, uint8_t value)
{
	if (addr < 0x1000) {
		// RAM (repeats multiply times, hence we apply a bitmask)
		mem[addr & 0x7ff] = value;
	} else if (addr >= 0xc000) { 
		// ROM (poking to ROM has no effect)
	} else {
		// IO space
		pokeIO(addr, value);
	}
}