osiris-process 0.3.1

//! Implements a integral register bank.

use osiris_data::data::atomic::Word;
use osiris_data::data::identification::Identifier;

use crate::register::{RangeApplication, RegisterBank, RegisterId, RegisterRange, REGISTERS_IN_BANK};


/// A function on a range which returns a [Word].
pub type BankApplication = RangeApplication<Word>;

/// The register bank.
/// 
/// The struct contains the whole register array.
#[derive(Clone, Debug)]
pub struct Bank {
    registers: Vec<Word>,
}

impl RegisterBank<Word> for Bank {
    fn new() -> Self {
        Self { registers: vec![Word::default(); REGISTERS_IN_BANK] }
    }

    fn set(&mut self, register: RegisterId, word: Word) {
        self.registers[register.to_usize()] = word;
    }

    fn get(&self, register: RegisterId) -> Word {
        self.registers[register.to_usize()]
    }

    fn apply(&self, range: RegisterRange, function: BankApplication) -> Word {
        let slice = self.slice(range);
        function(slice)
    }
    
    fn slice(&self, range: RegisterRange) -> &[Word] {
        &self.registers[range.to_usize_range()]
    }

    fn copy(&mut self, to: RegisterId, slice: &[Word]) {
        let mut index = to.to_usize();
        for w in slice {
            if index > REGISTERS_IN_BANK {
                break;
            }
            self.registers[index] = *w;
            index += 1;
        }
    }
}

impl Default for Bank {
    fn default() -> Self { Self::new() }
}