use crate::nes::cartridge::base_mapper::BaseMapper;
use crate::nes::cartridge::mapper::{Mapper, MapperCapabilities};
pub struct Mapper244 {
base: BaseMapper,
}
impl Mapper244 {
const PRG_LUT: [u8; 4] = [0, 1, 2, 3];
const CHR_LUT: [u8; 8] = [0, 2, 1, 3, 4, 6, 5, 7];
pub fn new(ctx: crate::nes::cartridge::mapper::MapperContext) -> Self {
let capabilities = MapperCapabilities {
has_chr_banking: true,
max_prg_ram_kb: 8,
prg_bank_size_kb: 32,
chr_bank_size_kb: 8,
..Default::default()
};
let mut base = BaseMapper::new(&ctx, capabilities);
base.configure_prg_banking(32 * 1024);
base.configure_chr_banking(8 * 1024);
Self { base }
}
}
impl Mapper for Mapper244 {
fn base(&self) -> &BaseMapper {
&self.base
}
fn base_mut(&mut self) -> &mut BaseMapper {
&mut self.base
}
fn write_prg(&mut self, addr: u16, value: u8) {
if self.base.try_write_prg_ram(addr, value) {
return;
}
if (0x8000..=0xFFFF).contains(&addr) {
if value & 0x08 == 0 {
let index = (value & 0x03) as usize;
self.base.select_prg_page(0, Self::PRG_LUT[index] as i16);
} else {
let index = (value & 0x07) as usize;
self.base.select_chr_page(0, Self::CHR_LUT[index] as i16);
}
}
}
fn registers_snapshot(&self) -> Vec<u8> {
self.base.banking_snapshot()
}
fn restore_registers(&mut self, data: &[u8]) {
self.base.restore_banking(data);
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::nes::cartridge::NametableLayout;
use crate::nes::cartridge::mapper::{MapperContext, create_mapper};
use crate::nes::cartridge::test_helpers::banked_data;
fn create_mapper244(
prg_rom: Vec<u8>,
chr_rom: Vec<u8>,
mirroring: NametableLayout,
) -> std::io::Result<Box<dyn Mapper>> {
create_mapper(MapperContext::new_for_test(
244, prg_rom, chr_rom, mirroring,
))
}
#[test]
fn test_factory_creates_mapper_244() {
let prg_rom = banked_data(32 * 1024, 4);
let chr_rom = banked_data(8 * 1024, 8);
let mapper = create_mapper244(prg_rom, chr_rom, NametableLayout::Vertical);
assert!(mapper.is_ok(), "Mapper 244 should be creatable via factory");
}
#[test]
fn test_prg_bank_select_via_lut() {
let prg_rom = banked_data(32 * 1024, 4);
let chr_rom = banked_data(8 * 1024, 8);
let mut mapper = create_mapper244(prg_rom, chr_rom, NametableLayout::Vertical).unwrap();
mapper.write_prg(0x8000, 0); assert_eq!(mapper.read_prg(0x8000), 0);
mapper.write_prg(0x8000, 1); assert_eq!(mapper.read_prg(0x8000), 1);
mapper.write_prg(0x8000, 2); assert_eq!(mapper.read_prg(0x8000), 2);
mapper.write_prg(0x8000, 3); assert_eq!(mapper.read_prg(0x8000), 3);
}
#[test]
fn test_chr_bank_select_via_scrambled_lut() {
let prg_rom = banked_data(32 * 1024, 4);
let chr_rom = banked_data(8 * 1024, 8);
let mut mapper = create_mapper244(prg_rom, chr_rom, NametableLayout::Vertical).unwrap();
mapper.write_prg(0x8000, 0x08); assert_eq!(mapper.read_chr(0x0000), 0);
mapper.write_prg(0x8000, 0x09); assert_eq!(mapper.read_chr(0x0000), 2);
mapper.write_prg(0x8000, 0x0A); assert_eq!(mapper.read_chr(0x0000), 1);
mapper.write_prg(0x8000, 0x0B); assert_eq!(mapper.read_chr(0x0000), 3);
mapper.write_prg(0x8000, 0x0C); assert_eq!(mapper.read_chr(0x0000), 4);
mapper.write_prg(0x8000, 0x0D); assert_eq!(mapper.read_chr(0x0000), 6);
mapper.write_prg(0x8000, 0x0E); assert_eq!(mapper.read_chr(0x0000), 5);
mapper.write_prg(0x8000, 0x0F); assert_eq!(mapper.read_chr(0x0000), 7);
}
#[test]
fn test_mirroring_is_fixed() {
let prg_rom = banked_data(32 * 1024, 4);
let chr_rom = banked_data(8 * 1024, 8);
let mut mapper = create_mapper244(prg_rom, chr_rom, NametableLayout::Horizontal).unwrap();
assert_eq!(mapper.get_mirroring(), NametableLayout::Horizontal);
mapper.write_prg(0x8000, 0xFF);
assert_eq!(mapper.get_mirroring(), NametableLayout::Horizontal);
}
#[test]
fn test_prg_and_chr_banks_are_independent() {
let prg_rom = banked_data(32 * 1024, 4);
let chr_rom = banked_data(8 * 1024, 8);
let mut mapper = create_mapper244(prg_rom, chr_rom, NametableLayout::Vertical).unwrap();
mapper.write_prg(0x8000, 2);
assert_eq!(mapper.read_prg(0x8000), 2);
assert_eq!(mapper.read_chr(0x0000), 0);
mapper.write_prg(0x8000, 0x09);
assert_eq!(mapper.read_prg(0x8000), 2); assert_eq!(mapper.read_chr(0x0000), 2);
}
#[test]
fn test_registers_snapshot_and_restore() {
let prg_rom = banked_data(32 * 1024, 4);
let chr_rom = banked_data(8 * 1024, 8);
let mut mapper =
create_mapper244(prg_rom.clone(), chr_rom.clone(), NametableLayout::Vertical).unwrap();
mapper.write_prg(0x8000, 2); mapper.write_prg(0x8000, 0x0D);
let regs = mapper.registers_snapshot();
let mut restored = create_mapper244(prg_rom, chr_rom, NametableLayout::Vertical).unwrap();
restored.restore_registers(®s);
assert_eq!(restored.read_prg(0x8000), 2);
assert_eq!(restored.read_chr(0x0000), 6);
}
}