/* FCE Ultra - NES/Famicom Emulator
 *
 * Copyright notice for this file:
 *  Copyright (C) 2017 CaH4e3
 *
 * 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include "mapinc.h"
#include "mmc3.h"
#include "../ines.h"

/* added on 2019-5-23 - NES 2.0 Mapper 260
 * HP10xx/HP20xx - a simplified version of FK23C mapper with pretty strict and better
 * organized banking behaviour. It seems that some 176 mapper assigned game may be
 * actually this kind of board instead but in common they aren't compatible at all,
 * the games on the regular FK23C boards couldn't run on this mapper and vice versa...
 */

static uint8 unromchr, lock;
static uint32 dipswitch;

static void BMCHPxxCW(uint32 A, uint8 V) {
	if (EXPREGS[0] & 4) {		/* custom banking */
		switch(EXPREGS[0] & 3) {
		case 0:
		case 1:
			setchr8(EXPREGS[2] & 0x3F);
/*			FCEU_printf("\tCHR8 %02X\n",EXPREGS[2]&0x3F); */
			break;
		case 2:
			setchr8((EXPREGS[2] & 0x3E) | (unromchr & 1));
/*			FCEU_printf("\tCHR8 %02X\n",(EXPREGS[2]&0x3E)|(unromchr&1)); */
			break;
		case 3:
			setchr8((EXPREGS[2] & 0x3C) | (unromchr & 3));
/*			FCEU_printf("\tCHR8 %02X\n",(EXPREGS[2]&0x3C)|(unromchr&3)); */
			break;
		}
	} else {				/* mmc3 banking */
		int base, mask;
		if(EXPREGS[0] & 1) {	/* 128K mode */
			base = EXPREGS[2] & 0x30;
			mask = 0x7F;
		} else {			/* 256K mode */
			base = EXPREGS[2] & 0x20;
			mask = 0xFF;
		}
/*		FCEU_printf("\tCHR1 %04x:%02X\n",A,(V&mask)|(base<<3)); */
		setchr1(A, (V & mask) | (base << 3));
	}
}

/* PRG wrapper */
static void BMCHPxxPW(uint32 A, uint8 V) {
	if(EXPREGS[0] & 4) {		/* custom banking */
		if((EXPREGS[0] & 0xF) == 4) {	/* 16K mode */
/*			FCEU_printf("\tPRG16 %02X\n",EXPREGS[1]&0x1F); */
			setprg16(0x8000, EXPREGS[1] & 0x1F);
			setprg16(0xC000, EXPREGS[1] & 0x1F);
		} else {			/* 32K modes */
/*			FCEU_printf("\tPRG32 %02X\n",(EXPREGS[1]&0x1F)>>1); */
			setprg32(0x8000, (EXPREGS[1] & 0x1F) >> 1);
		}
	} else {				/* mmc3 banking */
		uint8 base, mask;
		if(EXPREGS[0] & 2) {	/* 128K mode */
			base = EXPREGS[1] & 0x18;
			mask = 0x0F;
		} else {			/* 256K mode */
			base = EXPREGS[1] & 0x10;
			mask = 0x1F;
		}
/*		FCEU_printf("\tPRG8 %02X\n",(V&mask)|(base<<1)); */
		setprg8(A, (V & mask) | (base << 1));
		setprg8r(0x10, 0x6000, A001B & 3);
	}
}

/* MIRROR wrapper */
static void BMCHPxxMW(uint8 V) {
	if(EXPREGS[0] & 4) {		/* custom banking */
/*		FCEU_printf("CUSTOM MIRR: %d\n",(unromchr>>2)&1); */
		setmirror(((unromchr >> 2) & 1) ^ 1);
	} else {				/* mmc3 banking */
/*		FCEU_printf("MMC3 MIRR: %d\n",(V&1)^1); */
		A000B = V;
		setmirror((A000B & 1) ^ 1);
	}
}

/* PRG handler ($8000-$FFFF) */
static DECLFW(BMCHPxxHiWrite) {
/*	FCEU_printf("HI WRITE %04X:%02X\n",A,V); */
	if(EXPREGS[0] & 4) {		/* custom banking */
/*		FCEU_printf("CUSTOM\n"); */
		unromchr = V;
		FixMMC3CHR(MMC3_cmd);
	} else {				/* mmc3 banking */
/*		FCEU_printf("MMC3\n"); */
		if(A<0xC000) {
			MMC3_CMDWrite(A, V);
			FixMMC3PRG(MMC3_cmd);
			FixMMC3CHR(MMC3_cmd);
		} else {
			MMC3_IRQWrite(A, V);
		}
	}
}

/* EXP handler ($5000-$5FFF) */
static DECLFW(BMCHPxxWrite) {
	if (!lock) {
/*		FCEU_printf("LO WRITE %04X:%02X\n",A,V); */
		EXPREGS[A & 3] = V;
		lock = V & 0x80;
		FixMMC3PRG(MMC3_cmd);
		FixMMC3CHR(MMC3_cmd);
	}
}

static DECLFR(BMCHPxxRead) {
	return dipswitch;
}

static void BMCHPxxReset(void) {
	dipswitch++;
	dipswitch &= 0xF;
	lock = 0;
/*	FCEU_printf("BMCHPxx dipswitch set to %d\n",dipswitch); */
	EXPREGS[0] = EXPREGS[1] = EXPREGS[2] = EXPREGS[3] = 0;
	MMC3RegReset();
	FixMMC3PRG(MMC3_cmd);
	FixMMC3CHR(MMC3_cmd);
}

static void BMCHPxxPower(void) {
	GenMMC3Power();
	dipswitch = lock = 0;
	EXPREGS[0] = EXPREGS[1] = EXPREGS[2] = EXPREGS[3] = 0;
	FixMMC3PRG(MMC3_cmd);
	FixMMC3CHR(MMC3_cmd);
	SetReadHandler(0x5000, 0x5fff, BMCHPxxRead);
	SetWriteHandler(0x5000, 0x5fff, BMCHPxxWrite);
	SetWriteHandler(0x8000, 0xffff, BMCHPxxHiWrite);
}

void BMCHPxx_Init(CartInfo *info) {
	GenMMC3_Init(info, 256, 256, 8, 0);
	cwrap = BMCHPxxCW;
	pwrap = BMCHPxxPW;
	mwrap = BMCHPxxMW;
	info->Power = BMCHPxxPower;
	info->Reset = BMCHPxxReset;
	AddExState(EXPREGS, 8, 0, "EXPR");
	AddExState(&unromchr, 1, 0, "UCHR");
	AddExState(&dipswitch, 1, 0, "DPSW");
	AddExState(&lock, 1, 0, "LOCK");
}