mindustry logic execution, map- and schematic- parsing and rendering
Diffstat (limited to 'src/registry.rs')
-rw-r--r--src/registry.rs64
1 files changed, 31 insertions, 33 deletions
diff --git a/src/registry.rs b/src/registry.rs
index a22682d..240fb65 100644
--- a/src/registry.rs
+++ b/src/registry.rs
@@ -1,50 +1,48 @@
use std::any::type_name;
-use std::collections::HashMap;
use std::collections::hash_map::Entry;
+use std::collections::HashMap;
use std::error::Error;
use std::fmt;
-pub trait RegistryEntry
-{
- fn get_name(&self) -> &str;
+pub trait RegistryEntry {
+ fn get_name(&self) -> &str;
}
-pub struct Registry<'l, E: RegistryEntry + fmt::Debug + 'static>
-{
- by_name: HashMap<&'l str, &'l E>,
+pub struct Registry<'l, E: RegistryEntry + fmt::Debug + 'static> {
+ by_name: HashMap<&'l str, &'l E>,
}
-impl<'l, E: RegistryEntry + fmt::Debug + 'static> Registry<'l, E>
-{
- pub fn new() -> Self
- {
- Self{by_name: HashMap::new()}
- }
-
- pub fn register(&mut self, val: &'l E) -> Result<&'l E, RegisterError<'l, E>>
- {
- match self.by_name.entry(&val.get_name())
- {
- Entry::Occupied(e) => Err(RegisterError(e.get())),
- Entry::Vacant(e) => Ok(e.insert(val)),
- }
- }
-
- pub fn get(&self, name: &str) -> Option<&'l E>
- {
- self.by_name.get(name).map(|&r| r)
- }
+impl<'l, E: RegistryEntry + fmt::Debug + 'static> Registry<'l, E> {
+ pub fn new() -> Self {
+ Self {
+ by_name: HashMap::new(),
+ }
+ }
+
+ pub fn register(&mut self, val: &'l E) -> Result<&'l E, RegisterError<'l, E>> {
+ match self.by_name.entry(&val.get_name()) {
+ Entry::Occupied(e) => Err(RegisterError(e.get())),
+ Entry::Vacant(e) => Ok(e.insert(val)),
+ }
+ }
+
+ pub fn get(&self, name: &str) -> Option<&'l E> {
+ self.by_name.get(name).map(|&r| r)
+ }
}
#[derive(Clone, Copy, Debug)]
pub struct RegisterError<'l, E: RegistryEntry + fmt::Debug + 'static>(pub &'l E);
-impl<'l, E: RegistryEntry + fmt::Debug + 'static> fmt::Display for RegisterError<'l, E>
-{
- fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
- {
- write!(f, "{} {:?} already exists", type_name::<E>(), self.0.get_name())
- }
+impl<'l, E: RegistryEntry + fmt::Debug + 'static> fmt::Display for RegisterError<'l, E> {
+ fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+ write!(
+ f,
+ "{} {:?} already exists",
+ type_name::<E>(),
+ self.0.get_name()
+ )
+ }
}
impl<'l, E: RegistryEntry + fmt::Debug + 'static> Error for RegisterError<'l, E> {}