68 lines
1.7 KiB
Rust
68 lines
1.7 KiB
Rust
use crate::static_data::libraries::data_library::DataLibrary;
|
|
use crate::static_data::moves::move_data::MoveData;
|
|
use std::collections::HashMap;
|
|
|
|
#[derive(Debug)]
|
|
pub struct MoveLibrary {
|
|
map: HashMap<String, MoveData>,
|
|
list: Vec<String>,
|
|
}
|
|
|
|
impl MoveLibrary {
|
|
pub fn new(capacity: usize) -> MoveLibrary {
|
|
MoveLibrary {
|
|
map: HashMap::with_capacity(capacity),
|
|
list: Vec::with_capacity(capacity),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl DataLibrary<'_, MoveData> for MoveLibrary {
|
|
fn map(&self) -> &HashMap<String, MoveData> {
|
|
&self.map
|
|
}
|
|
|
|
fn list_values(&self) -> &Vec<String> {
|
|
&self.list
|
|
}
|
|
|
|
fn get_modify(&mut self) -> (&mut HashMap<String, MoveData>, &mut Vec<String>) {
|
|
(&mut self.map, &mut self.list)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
pub mod tests {
|
|
use crate::static_data::libraries::data_library::DataLibrary;
|
|
use crate::static_data::libraries::move_library::MoveLibrary;
|
|
use crate::static_data::moves::move_data::{MoveCategory, MoveData, MoveTarget};
|
|
use crate::static_data::moves::secondary_effect::SecondaryEffect;
|
|
use std::collections::HashSet;
|
|
|
|
fn build_move() -> MoveData {
|
|
MoveData::new(
|
|
"foo",
|
|
0,
|
|
MoveCategory::Physical,
|
|
100,
|
|
100,
|
|
30,
|
|
MoveTarget::Any,
|
|
0,
|
|
SecondaryEffect::empty(),
|
|
HashSet::new(),
|
|
)
|
|
}
|
|
|
|
pub fn build() -> MoveLibrary {
|
|
let mut lib = MoveLibrary::new(1);
|
|
let m = build_move();
|
|
// Borrow as mut so we can insert
|
|
let w = &mut lib;
|
|
w.add("foo", m);
|
|
// Drops borrow as mut
|
|
|
|
lib
|
|
}
|
|
}
|