2022-06-19 19:34:08 +00:00
|
|
|
use hashbrown::HashMap;
|
|
|
|
|
2022-06-11 18:51:37 +00:00
|
|
|
use crate::static_data::DataLibrary;
|
|
|
|
use crate::static_data::Species;
|
2022-06-11 15:22:46 +00:00
|
|
|
use crate::StringKey;
|
2021-01-30 21:29:59 +00:00
|
|
|
|
|
|
|
#[derive(Debug)]
|
2022-06-17 17:53:33 +00:00
|
|
|
pub struct SpeciesLibrary {
|
|
|
|
map: HashMap<StringKey, Box<Species>>,
|
2022-06-11 15:22:46 +00:00
|
|
|
list: Vec<StringKey>,
|
2021-01-30 21:29:59 +00:00
|
|
|
}
|
|
|
|
|
2022-06-17 17:53:33 +00:00
|
|
|
impl SpeciesLibrary {
|
|
|
|
pub fn new(capacity: usize) -> SpeciesLibrary {
|
2021-01-30 21:29:59 +00:00
|
|
|
SpeciesLibrary {
|
|
|
|
map: HashMap::with_capacity(capacity),
|
|
|
|
list: Vec::with_capacity(capacity),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-06-17 17:53:33 +00:00
|
|
|
impl<'a> DataLibrary<'a, Box<Species>> for SpeciesLibrary {
|
|
|
|
fn map(&self) -> &HashMap<StringKey, Box<Species>> {
|
2021-01-30 21:29:59 +00:00
|
|
|
&self.map
|
|
|
|
}
|
|
|
|
|
2022-06-11 15:22:46 +00:00
|
|
|
fn list_values(&self) -> &Vec<StringKey> {
|
2021-01-30 21:29:59 +00:00
|
|
|
&self.list
|
|
|
|
}
|
|
|
|
|
2022-06-17 17:53:33 +00:00
|
|
|
fn get_modify(&mut self) -> (&mut HashMap<StringKey, Box<Species>>, &mut Vec<StringKey>) {
|
2021-01-30 21:29:59 +00:00
|
|
|
(&mut self.map, &mut self.list)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
2021-01-31 16:31:22 +00:00
|
|
|
pub mod tests {
|
2022-06-19 19:34:08 +00:00
|
|
|
use hashbrown::HashSet;
|
|
|
|
|
2021-01-30 21:29:59 +00:00
|
|
|
use crate::static_data::libraries::data_library::DataLibrary;
|
|
|
|
use crate::static_data::libraries::species_library::SpeciesLibrary;
|
2022-06-19 19:34:08 +00:00
|
|
|
use crate::static_data::Form;
|
|
|
|
use crate::static_data::LearnableMoves;
|
|
|
|
use crate::static_data::Species;
|
2022-06-18 12:17:29 +00:00
|
|
|
use crate::static_data::StaticStatisticSet;
|
2021-01-30 21:29:59 +00:00
|
|
|
|
2022-06-18 14:06:54 +00:00
|
|
|
fn build_species() -> Species {
|
2021-01-30 21:29:59 +00:00
|
|
|
Species::new(
|
|
|
|
0,
|
2022-06-11 15:22:46 +00:00
|
|
|
&"foo".into(),
|
2021-01-30 21:29:59 +00:00
|
|
|
0.5,
|
2022-06-11 15:22:46 +00:00
|
|
|
&"test_growthrate".into(),
|
2021-01-30 21:29:59 +00:00
|
|
|
0,
|
|
|
|
Form::new(
|
2022-06-11 15:22:46 +00:00
|
|
|
&"default".into(),
|
2021-01-30 21:29:59 +00:00
|
|
|
0.0,
|
|
|
|
0.0,
|
|
|
|
0,
|
|
|
|
Vec::new(),
|
2022-06-18 12:17:29 +00:00
|
|
|
StaticStatisticSet::default(),
|
2021-01-30 21:29:59 +00:00
|
|
|
Vec::new(),
|
|
|
|
Vec::new(),
|
|
|
|
LearnableMoves::new(),
|
|
|
|
HashSet::new(),
|
|
|
|
),
|
|
|
|
HashSet::new(),
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2022-06-18 14:06:54 +00:00
|
|
|
pub fn build() -> SpeciesLibrary {
|
2021-01-30 21:29:59 +00:00
|
|
|
let mut lib = SpeciesLibrary::new(1);
|
|
|
|
let species = build_species();
|
|
|
|
// Borrow as mut so we can insert
|
|
|
|
let w = &mut lib;
|
2022-06-11 15:22:46 +00:00
|
|
|
w.add(&"foo".into(), Box::from(species));
|
2021-01-30 21:29:59 +00:00
|
|
|
// Drops borrow as mut
|
|
|
|
|
2021-01-31 16:31:22 +00:00
|
|
|
lib
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn add_species_to_library_and_fetch() {
|
|
|
|
let lib = build();
|
|
|
|
|
2021-01-30 21:29:59 +00:00
|
|
|
// Borrow as read so we can read
|
|
|
|
let r = &lib;
|
2022-06-11 15:22:46 +00:00
|
|
|
let mon = r.get(&"foo".into());
|
2021-01-30 21:29:59 +00:00
|
|
|
assert!(mon.is_some());
|
2022-06-06 12:43:41 +00:00
|
|
|
assert_eq!(mon.unwrap().id(), 0_u16);
|
2022-06-11 15:22:46 +00:00
|
|
|
assert_eq!(mon.unwrap().as_ref().name(), &"foo".into());
|
2021-01-30 21:29:59 +00:00
|
|
|
assert_eq!(r.len(), 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn add_species_to_library_then_remove() {
|
2021-01-31 16:31:22 +00:00
|
|
|
let mut lib = build();
|
2021-01-30 21:29:59 +00:00
|
|
|
|
2022-06-11 15:22:46 +00:00
|
|
|
lib.remove(&"foo".into());
|
2021-01-30 21:29:59 +00:00
|
|
|
|
|
|
|
// Borrow as read so we can read
|
|
|
|
let r = &lib;
|
2022-06-11 15:22:46 +00:00
|
|
|
let mon = r.get(&"foo".into());
|
2021-01-30 21:29:59 +00:00
|
|
|
assert!(mon.is_none());
|
|
|
|
assert_eq!(r.len(), 0);
|
|
|
|
}
|
|
|
|
}
|