From 6d3e7fa3d7143b7d7cdfa0fc495284b2aa9b784c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Magalh=C3=A3es?= <joamag@gmail.com>
Date: Sun, 18 Jun 2023 16:38:35 +0100
Subject: [PATCH] chore: support for has_rumble
---
src/rom.rs | 65 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 65 insertions(+)
diff --git a/src/rom.rs b/src/rom.rs
index 1ab7975f..d8b5aa2e 100644
--- a/src/rom.rs
+++ b/src/rom.rs
@@ -335,6 +335,13 @@ impl Cartridge {
}
}
+ pub fn has_rumble(&mut self) -> bool {
+ matches!(
+ self.rom_type(),
+ RomType::Mbc5Rumble | RomType::Mbc5RumbleRam | RomType::Mbc5RumbleRamBattery
+ )
+ }
+
pub fn set_rom_bank(&mut self, rom_bank: u8) {
self.rom_offset = rom_bank as usize * ROM_BANK_SIZE;
}
@@ -458,6 +465,40 @@ impl Cartridge {
}
}
+ pub fn set_rom_type(&mut self, rom_type: RomType) {
+ self.rom_data[0x0147] = match rom_type {
+ RomType::RomOnly => 0x00,
+ RomType::Mbc1 => 0x01,
+ RomType::Mbc1Ram => 0x02,
+ RomType::Mbc1RamBattery => 0x03,
+ RomType::Mbc2 => 0x05,
+ RomType::Mbc2Battery => 0x06,
+ RomType::RomRam => 0x08,
+ RomType::RomRamBattery => 0x09,
+ RomType::Mmm01 => 0x0b,
+ RomType::Mmm01Ram => 0x0c,
+ RomType::Mmm01RamBattery => 0x0d,
+ RomType::Mbc3TimerBattery => 0x0f,
+ RomType::Mbc3TimerRamBattery => 0x10,
+ RomType::Mbc3 => 0x11,
+ RomType::Mbc3Ram => 0x12,
+ RomType::Mbc3RamBattery => 0x13,
+ RomType::Mbc5 => 0x19,
+ RomType::Mbc5Ram => 0x1a,
+ RomType::Mbc5RamBattery => 0x1b,
+ RomType::Mbc5Rumble => 0x1c,
+ RomType::Mbc5RumbleRam => 0x1d,
+ RomType::Mbc5RumbleRamBattery => 0x1e,
+ RomType::Mbc6 => 0x20,
+ RomType::Mbc7SensorRumbleRamBattery => 0x22,
+ RomType::PocketCamera => 0xfc,
+ RomType::BandaiTama5 => 0xfd,
+ RomType::HuC3 => 0xfe,
+ RomType::HuC1RamBattery => 0xff,
+ RomType::Unknown => panic!("Unknown ROM type"),
+ };
+ }
+
pub fn rom_size(&self) -> RomSize {
match self.rom_data[0x0148] {
0x00 => RomSize::Size32K,
@@ -751,3 +792,27 @@ pub static MBC5: Mbc = Mbc {
rom.ram_data[rom.ram_offset + (addr - 0xa000) as usize] = value;
},
};
+
+#[cfg(test)]
+mod tests {
+ use super::{Cartridge, RomType};
+
+ #[test]
+ fn test_has_rumble() {
+ let mut rom = Cartridge::new();
+ rom.set_data(&vec![0; 0x8000]);
+ assert!(!rom.has_rumble());
+
+ rom.set_rom_type(RomType::Mbc5Rumble);
+ assert!(rom.has_rumble());
+
+ rom.set_rom_type(RomType::Mbc5RumbleRam);
+ assert!(rom.has_rumble());
+
+ rom.set_rom_type(RomType::Mbc5RumbleRamBattery);
+ assert!(rom.has_rumble());
+
+ rom.set_rom_type(RomType::Mbc1);
+ assert!(!rom.has_rumble());
+ }
+}
--
GitLab