Make Nature a trait

src/static_data/libraries/nature_library.rs

@@ -9,7 +9,7 @@ pub struct NatureLibrary {
     /// A unique identifier so we know what value this is.
     identifier: ValueIdentifier,
     /// The underlying data structure.
-    map: HashMap<StringKey, Arc<Nature>>,
+    map: HashMap<StringKey, Arc<dyn Nature>>,
 }
 
 impl NatureLibrary {
@@ -22,21 +22,21 @@ impl NatureLibrary {
     }
 
     /// Adds a new nature with name to the library.
-    pub fn load_nature(&mut self, name: StringKey, nature: Arc<Nature>) {
+    pub fn load_nature(&mut self, name: StringKey, nature: Arc<dyn Nature>) {
         self.map.insert(name, nature);
     }
 
     /// Gets a nature by name.
-    pub fn get_nature(&self, key: &StringKey) -> Option<&Arc<Nature>> {
+    pub fn get_nature(&self, key: &StringKey) -> Option<&Arc<dyn Nature>> {
         self.map.get(key)
     }
 
     /// Finds a nature name by nature.
-    pub fn get_nature_name(&self, nature: &Nature) -> StringKey {
+    pub fn get_nature_name(&self, nature: &Arc<dyn Nature>) -> StringKey {
         for kv in &self.map {
             // As natures can't be copied, and should always be the same reference as the value
             // in the map, we just compare by reference.
-            if std::ptr::eq(Arc::as_ptr(kv.1), nature) {
+            if kv.1.value_identifier() == nature.value_identifier() {
                 return kv.0.clone();
             }
         }
@@ -53,14 +53,14 @@ impl ValueIdentifiable for NatureLibrary {
 #[cfg(test)]
 pub mod tests {
     use crate::static_data::statistics::Statistic;
-    use crate::static_data::{Nature, NatureLibrary};
+    use crate::static_data::{NatureImpl, NatureLibrary};
 
     pub fn build() -> NatureLibrary {
         let mut lib = NatureLibrary::new(2);
 
         lib.load_nature(
             "test_nature".into(),
-            Nature::new(Statistic::HP, Statistic::Attack, 1.1, 0.9),
+            NatureImpl::new(Statistic::HP, Statistic::Attack, 1.1, 0.9),
         );
 
         lib
@@ -69,10 +69,13 @@ pub mod tests {
     #[test]
     fn create_nature_library_insert_and_retrieve() {
         let mut lib = NatureLibrary::new(2);
-        lib.load_nature("foo".into(), Nature::new(Statistic::HP, Statistic::Attack, 1.1, 0.9));
+        lib.load_nature(
+            "foo".into(),
+            NatureImpl::new(Statistic::HP, Statistic::Attack, 1.1, 0.9),
+        );
         lib.load_nature(
             "bar".into(),
-            Nature::new(Statistic::Attack, Statistic::Defense, 1.1, 0.9),
+            NatureImpl::new(Statistic::Attack, Statistic::Defense, 1.1, 0.9),
         );
         let n1 = lib.get_nature(&"foo".into()).expect("Nature was not found");
         assert_eq!(n1.increased_stat(), Statistic::HP);
@@ -84,10 +87,13 @@ pub mod tests {
     #[test]
     fn create_nature_library_insert_and_get_name() {
         let mut lib = NatureLibrary::new(2);
-        lib.load_nature("foo".into(), Nature::new(Statistic::HP, Statistic::Attack, 1.1, 0.9));
+        lib.load_nature(
+            "foo".into(),
+            NatureImpl::new(Statistic::HP, Statistic::Attack, 1.1, 0.9),
+        );
         lib.load_nature(
             "bar".into(),
-            Nature::new(Statistic::Attack, Statistic::Defense, 1.1, 0.9),
+            NatureImpl::new(Statistic::Attack, Statistic::Defense, 1.1, 0.9),
         );
 
         let n1 = lib.get_nature(&"foo".into()).expect("Nature was not found");

src/static_data/natures.rs

@@ -1,11 +1,26 @@
 use crate::static_data::Statistic;
 use crate::{ValueIdentifiable, ValueIdentifier};
+use std::fmt::Debug;
 use std::sync::Arc;
 
+/// A nature is an attribute on a Pokemon that modifies the effective base stats on a Pokemon. They
+/// can have an increased statistic and a decreased statistic, or be neutral.
+pub trait Nature: ValueIdentifiable + Debug {
+    /// The stat that should receive the increased modifier.
+    fn increased_stat(&self) -> Statistic;
+
+    /// The stat that should receive the decreased modifier.
+    fn decreased_stat(&self) -> Statistic;
+
+    /// Calculates the modifier for a given stat. If it's the increased stat, returns the increased
+    /// modifier, if it's the decreased stat, returns the decreased modifier. Otherwise returns 1.0
+    fn get_stat_modifier(&self, stat: Statistic) -> f32;
+}
+
 /// A nature is an attribute on a Pokemon that modifies the effective base stats on a Pokemon. They
 /// can have an increased statistic and a decreased statistic, or be neutral.
 #[derive(Debug)]
-pub struct Nature {
+pub struct NatureImpl {
     /// A unique identifier so we know what value this is.
     identifier: ValueIdentifier,
     /// The stat that should receive the increased modifier.
@@ -18,7 +33,7 @@ pub struct Nature {
     decrease_modifier: f32,
 }
 
-impl Nature {
+impl NatureImpl {
     /// Instantiates a new statistic.
     pub fn new(
         increase_stat: Statistic,
@@ -34,20 +49,22 @@ impl Nature {
             decrease_modifier,
         })
     }
+}
 
+impl Nature for NatureImpl {
     /// The stat that should receive the increased modifier.
-    pub fn increased_stat(&self) -> Statistic {
+    fn increased_stat(&self) -> Statistic {
         self.increase_stat
     }
 
     /// The stat that should receive the decreased modifier.
-    pub fn decreased_stat(&self) -> Statistic {
+    fn decreased_stat(&self) -> Statistic {
         self.decrease_stat
     }
 
     /// Calculates the modifier for a given stat. If it's the increased stat, returns the increased
     /// modifier, if it's the decreased stat, returns the decreased modifier. Otherwise returns 1.0
-    pub fn get_stat_modifier(&self, stat: Statistic) -> f32 {
+    fn get_stat_modifier(&self, stat: Statistic) -> f32 {
         if stat == self.increase_stat && stat != self.decrease_stat {
             self.increase_modifier
         } else if stat == self.decrease_stat && stat != self.increase_stat {
@@ -58,7 +75,7 @@ impl Nature {
     }
 }
 
-impl ValueIdentifiable for Nature {
+impl ValueIdentifiable for NatureImpl {
     fn value_identifier(&self) -> ValueIdentifier {
         self.identifier
     }