A lot more work on a bunch of different parts of the system.
This commit is contained in:
56
src/static_data/libraries/ability_library.rs
Normal file
56
src/static_data/libraries/ability_library.rs
Normal file
@@ -0,0 +1,56 @@
|
||||
use crate::static_data::libraries::data_library::DataLibrary;
|
||||
use crate::static_data::species_data::ability::Ability;
|
||||
use crate::StringKey;
|
||||
use hashbrown::HashMap;
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct AbilityLibrary {
|
||||
map: HashMap<StringKey, Box<Ability>>,
|
||||
list: Vec<StringKey>,
|
||||
}
|
||||
|
||||
impl AbilityLibrary {
|
||||
pub fn new(capacity: usize) -> AbilityLibrary {
|
||||
AbilityLibrary {
|
||||
map: HashMap::with_capacity(capacity),
|
||||
list: Vec::with_capacity(capacity),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl DataLibrary<'_, Box<Ability>> for AbilityLibrary {
|
||||
fn map(&self) -> &HashMap<StringKey, Box<Ability>> {
|
||||
&self.map
|
||||
}
|
||||
|
||||
fn list_values(&self) -> &Vec<StringKey> {
|
||||
&self.list
|
||||
}
|
||||
|
||||
fn get_modify(&mut self) -> (&mut HashMap<StringKey, Box<Ability>>, &mut Vec<StringKey>) {
|
||||
(&mut self.map, &mut self.list)
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
pub mod tests {
|
||||
use crate::static_data::libraries::ability_library::AbilityLibrary;
|
||||
use crate::static_data::libraries::data_library::DataLibrary;
|
||||
use crate::static_data::species_data::ability::Ability;
|
||||
use crate::StringKey;
|
||||
|
||||
pub fn build() -> AbilityLibrary {
|
||||
let mut lib = AbilityLibrary::new(1);
|
||||
lib.add(
|
||||
&StringKey::new("test_ability"),
|
||||
Box::new(Ability::new(
|
||||
&"test_ability".into(),
|
||||
&"test_ability".into(),
|
||||
Vec::new(),
|
||||
)),
|
||||
);
|
||||
// Drops borrow as mut
|
||||
|
||||
lib
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user