From db1c94e2a56a1176f70af191c62f3cde0bbf883f Mon Sep 17 00:00:00 2001
From: Ava Silver <ava.silver@datadoghq.com>
Date: Fri, 29 Dec 2023 22:52:11 -0800
Subject: [PATCH] fixed the build post merge :3

---
 src/systems/aiie/mod.rs      | 17 +++++++++++------
 src/systems/aiie/switches.rs |  4 ++--
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/src/systems/aiie/mod.rs b/src/systems/aiie/mod.rs
index 2b8f12e..d751879 100644
--- a/src/systems/aiie/mod.rs
+++ b/src/systems/aiie/mod.rs
@@ -1,7 +1,10 @@
 use std::sync::Arc;
 
 use crate::{
-  cpu::{MemoryIO, Mos6502, Mos6502Variant},
+  cpu::{
+    mos6502::{MemoryIO, Mos6502, Mos6502Variant},
+    Cpu,
+  },
   memory::{BankedMemory, BlockMemory, BranchMemory, LoggingMemory, NullMemory},
   platform::{Color, PlatformProvider, WindowConfig},
   systems::System,
@@ -20,7 +23,7 @@ use instant::Duration;
 
 use self::switches::{AiieBankSelectors, AiieSoftSwitches};
 
-use super::SystemBuilder;
+use super::BuildableSystem;
 
 const WIDTH: u32 = 40;
 const HEIGHT: u32 = 24;
@@ -30,10 +33,8 @@ const CHAR_WIDTH: u32 = 7;
 /// Configuration for a Apple IIe system.
 pub struct AiieSystemConfig {}
 
-/// A factory for creating a Commodore 64 system.
-pub struct AiieSystemBuilder;
-
-impl SystemBuilder<AiieSystem, AiieSystemRoms, AiieSystemConfig> for AiieSystemBuilder {
+/// A factory for creating an Apple IIe system.
+impl BuildableSystem<AiieSystemRoms, AiieSystemConfig> for AiieSystem {
   fn build(
     roms: AiieSystemRoms,
     _config: AiieSystemConfig,
@@ -146,6 +147,10 @@ pub struct AiieSystem {
 }
 
 impl System for AiieSystem {
+  fn get_cpu_mut(&mut self) -> Box<&mut dyn crate::cpu::Cpu> {
+    Box::new(&mut self.cpu)
+  }
+
   fn tick(&mut self) -> Duration {
     if self.cpu.registers.pc.address() < 0xF800 {
       // println!("{:#04x}", self.cpu.registers.pc.address());
diff --git a/src/systems/aiie/switches.rs b/src/systems/aiie/switches.rs
index 4921786..f12f111 100644
--- a/src/systems/aiie/switches.rs
+++ b/src/systems/aiie/switches.rs
@@ -2,7 +2,7 @@ use std::{cell::Cell, io::Write, rc::Rc, sync::Arc};
 
 use crate::{
   keyboard::{KeyAdapter, SymbolAdapter},
-  memory::{ActiveInterrupt, Memory, SystemInfo},
+  memory::{ActiveInterrupt, Memory},
   platform::PlatformProvider,
   systems::aiie::keyboard::AppleIISymbolAdapter,
 };
@@ -254,7 +254,7 @@ impl Memory for AiieSoftSwitches {
     self.bank_ram_select = false;
   }
 
-  fn poll(&mut self, _cycles: u32, _info: &SystemInfo) -> ActiveInterrupt {
+  fn poll(&mut self, _cycles_since_poll: u64, _total_cycle_count: u64) -> ActiveInterrupt {
     ActiveInterrupt::None
   }
 }