gb-emu/src/processor/memory.rs

use std::io::{stdout, Write};

use crate::verbose_println;

pub(crate) type Address = u16;
pub(crate) type ROM = Vec<u8>;

#[allow(dead_code)]
pub struct Memory {
    pub(super) bootrom: ROM,
    pub(super) bootrom_enabled: bool,
    pub(super) rom: ROM,
    pub(super) vram: [u8; 8192],
    pub(super) ram: [u8; 8192],
    pub(super) switchable_ram: [u8; 8192],
    pub(super) cpu_ram: [u8; 128],
    pub(super) oam: [u8; 160],
    pub(super) interrupts: u8,
    pub(super) ime: bool,
    pub(super) ime_scheduled: u8,
    pub(super) io: [u8; 76],
    pub(super) user_mode: bool,
}

impl Memory {
    pub fn init(bootrom: ROM, bootrom_enabled: bool, rom: ROM) -> Self {
        Self {
            bootrom,
            bootrom_enabled,
            rom,
            vram: [0x0; 8192],
            ram: [0x0; 8192],
            switchable_ram: [0x0; 8192],
            cpu_ram: [0x0; 128],
            oam: [0x0; 160],
            interrupts: 0x0,
            ime: false,
            ime_scheduled: 0x0,
            io: [0xFF; 76],
            user_mode: false,
        }
    }

    pub fn get(&self, address: Address) -> u8 {
        match address {
            0x0..0x8000 => {
                // rom access
                // todo - switchable rom banks
                if self.bootrom_enabled && ((address as usize) < self.bootrom.len()) {
                    return self.bootrom[address as usize];
                } else {
                    return self.rom[address as usize];
                }
            }
            0x8000..0xA000 => {
                return self.vram[(address - 0x8000) as usize];
            }
            0xA000..0xC000 => 0xFF,
            0xC000..0xE000 => {
                return self.ram[(address - 0xC000) as usize];
            }
            0xE000..0xFE00 => {
                return self.ram[(address - 0xE000) as usize];
            }
            0xFE00..0xFEA0 => {
                return self.oam[(address - 0xFE00) as usize];
            }
            0xFEA0..0xFF00 => {
                return 0x0;
            }
            0xFF00..0xFF4C => {
                if address == 0xFF00 {
                    return 0xFF;
                }
                return self.io[(address - 0xFF00) as usize];
            }
            0xFF4C..0xFF80 => {
                // println!("empty space 2 read");
                return 0xFF;
            }
            0xFF80..0xFFFF => {
                return self.cpu_ram[(address - 0xFF80) as usize];
            }
            0xFFFF => {
                return self.interrupts;
            }
        }
    }

    pub fn set(&mut self, address: Address, data: u8) {
        // verbose_println!("write addr: {:#X}, data: {:#X}", address, data);

        match address {
            0x0..0x8000 => {
                // change this with MBC code...
                // println!("tried to write {:#5X} at {:#X}", data, address);
            }
            0x8000..0xA000 => {
                self.vram[(address - 0x8000) as usize] = data;
            }
            0xA000..0xC000 => {
                // panic!("switchable write");
                // self.switchable_ram[(address - 0xA000) as usize] = data;
            }
            0xC000..0xE000 => {
                self.ram[(address - 0xC000) as usize] = data;
            }
            0xE000..0xFE00 => {
                self.ram[(address - 0xE000) as usize] = data;
            }
            0xFE00..0xFEA0 => {
                self.oam[(address - 0xFE00) as usize] = data;
            }
            0xFEA0..0xFF00 => {
                // println!("empty space write: {:#X} to addr {:#X}", data, address);
            }
            0xFF04 => {
                if self.user_mode {
                    self.io[0xFF04 - 0xFF00] = 0;
                } else {
                    self.io[0xFF04 - 0xFF00] = data;
                }
            }
            0xFF00..0xFF04 | 0xFF05..0xFF4C => {
                // verbose_print!("writing to addr {:#X}\r", address);
                stdout().flush().unwrap();

                if address == 0xFF02 && data == 0x81 {
                    print!("{}", self.get(0xFF01) as char);
                    stdout().flush().unwrap();
                }
                self.io[(address - 0xFF00) as usize] = data;
            }
            0xFF50 => {
                self.bootrom_enabled = false;
            }
            0xFF4C..0xFF50 | 0xFF51..0xFF80 => {
                // println!("empty space 2 write: {:#X} to addr {:#X}", data, address);
            }
            0xFF80..0xFFFF => {
                self.cpu_ram[(address - 0xFF80) as usize] = data;
            }
            0xFFFF => {
                verbose_println!("interrupts set to {:#b}", data);
                verbose_println!("                / {:#X}", data);
                self.interrupts = data;
            }
        }
    }
}