PkmnLib_rs/src/static_data/libraries/growth_rate_library.rs

use crate::defines::LevelInt;
use crate::static_data::GrowthRate;
use crate::StringKey;
use hashbrown::HashMap;
use std::fmt;
use std::fmt::{Debug, Formatter};

pub struct GrowthRateLibrary {
    growth_rates: HashMap<StringKey, Box<dyn GrowthRate>>,
}

impl GrowthRateLibrary {
    pub fn new(capacity: usize) -> GrowthRateLibrary {
        GrowthRateLibrary {
            growth_rates: HashMap::with_capacity(capacity),
        }
    }

    pub fn calculate_level(&self, growth_rate: &StringKey, experience: u32) -> LevelInt {
        self.growth_rates[growth_rate].calculate_level(experience)
    }
    pub fn calculate_experience(&self, growth_rate: &StringKey, level: LevelInt) -> u32 {
        self.growth_rates[growth_rate].calculate_experience(level)
    }
    pub fn add_growth_rate(&mut self, key: &StringKey, value: Box<dyn GrowthRate>) {
        self.growth_rates.insert(key.clone(), value);
    }
}

impl Debug for GrowthRateLibrary {
    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
        f.debug_struct("GrowthRateLibrary").finish()
    }
}

#[cfg(test)]
pub mod tests {
    use crate::static_data::growth_rates::lookup_growth_rate::LookupGrowthRate;
    use crate::static_data::libraries::growth_rate_library::GrowthRateLibrary;

    pub fn build() -> GrowthRateLibrary {
        let mut lib = GrowthRateLibrary::new(1);

        // Borrow as mut so we can insert
        let w = &mut lib;
        w.add_growth_rate(
            &"test_growthrate".into(),
            Box::new(LookupGrowthRate::new(vec![0, 5, 10, 100])),
        );
        // Drops borrow as mut

        lib
    }

    #[test]
    fn add_growth_rate_to_library_and_calculate_level() {
        let lib = build();
        assert_eq!(lib.calculate_level(&"test_growthrate".into(), 3), 1);
        assert_eq!(lib.calculate_level(&"test_growthrate".into(), 50), 3);
    }

    #[test]
    fn add_growth_rate_to_library_and_calculate_experience() {
        let lib = build();
        assert_eq!(lib.calculate_experience(&"test_growthrate".into(), 1), 0);
        assert_eq!(lib.calculate_experience(&"test_growthrate".into(), 3), 10);
    }
}