From 4b36805b55c06ec15b65fbe45f4d038cad472b79 Mon Sep 17 00:00:00 2001
From: Robin Mueller <muellerr@irs.uni-stuttgart.de>
Date: Sun, 15 May 2022 10:26:55 +0200
Subject: [PATCH] continue obj manager

---
 .idea/runConfigurations/Format.xml            |  19 +++
 .../{Run_launchpad.xml => Run.xml}            |   3 +-
 .idea/runConfigurations/Test.xml              |  19 +++
 src/lib.rs                                    | 161 ++++++++++++++++++
 src/main.rs                                   |  13 +-
 5 files changed, 202 insertions(+), 13 deletions(-)
 create mode 100644 .idea/runConfigurations/Format.xml
 rename .idea/runConfigurations/{Run_launchpad.xml => Run.xml} (80%)
 create mode 100644 .idea/runConfigurations/Test.xml
 create mode 100644 src/lib.rs

diff --git a/.idea/runConfigurations/Format.xml b/.idea/runConfigurations/Format.xml
new file mode 100644
index 0000000..e6f696d
--- /dev/null
+++ b/.idea/runConfigurations/Format.xml
@@ -0,0 +1,19 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="Format" type="CargoCommandRunConfiguration" factoryName="Cargo Command">
+    <option name="command" value="fmt" />
+    <option name="workingDirectory" value="file://$PROJECT_DIR$" />
+    <option name="channel" value="DEFAULT" />
+    <option name="requiredFeatures" value="true" />
+    <option name="allFeatures" value="false" />
+    <option name="emulateTerminal" value="false" />
+    <option name="withSudo" value="false" />
+    <option name="buildTarget" value="REMOTE" />
+    <option name="backtrace" value="SHORT" />
+    <envs />
+    <option name="isRedirectInput" value="false" />
+    <option name="redirectInputPath" value="" />
+    <method v="2">
+      <option name="CARGO.BUILD_TASK_PROVIDER" enabled="true" />
+    </method>
+  </configuration>
+</component>
\ No newline at end of file
diff --git a/.idea/runConfigurations/Run_launchpad.xml b/.idea/runConfigurations/Run.xml
similarity index 80%
rename from .idea/runConfigurations/Run_launchpad.xml
rename to .idea/runConfigurations/Run.xml
index b265baa..342e53e 100644
--- a/.idea/runConfigurations/Run_launchpad.xml
+++ b/.idea/runConfigurations/Run.xml
@@ -1,5 +1,5 @@
 <component name="ProjectRunConfigurationManager">
-  <configuration default="false" name="Run launchpad" type="CargoCommandRunConfiguration" factoryName="Cargo Command">
+  <configuration default="false" name="Run" type="CargoCommandRunConfiguration" factoryName="Cargo Command">
     <option name="command" value="run --package launchpad --bin launchpad" />
     <option name="workingDirectory" value="file://$PROJECT_DIR$" />
     <option name="channel" value="DEFAULT" />
@@ -7,6 +7,7 @@
     <option name="allFeatures" value="false" />
     <option name="emulateTerminal" value="false" />
     <option name="withSudo" value="false" />
+    <option name="buildTarget" value="REMOTE" />
     <option name="backtrace" value="SHORT" />
     <envs />
     <option name="isRedirectInput" value="false" />
diff --git a/.idea/runConfigurations/Test.xml b/.idea/runConfigurations/Test.xml
new file mode 100644
index 0000000..9f2b663
--- /dev/null
+++ b/.idea/runConfigurations/Test.xml
@@ -0,0 +1,19 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="Test" type="CargoCommandRunConfiguration" factoryName="Cargo Command" nameIsGenerated="true">
+    <option name="command" value="test" />
+    <option name="workingDirectory" value="file://$PROJECT_DIR$" />
+    <option name="channel" value="DEFAULT" />
+    <option name="requiredFeatures" value="true" />
+    <option name="allFeatures" value="false" />
+    <option name="emulateTerminal" value="false" />
+    <option name="withSudo" value="false" />
+    <option name="buildTarget" value="REMOTE" />
+    <option name="backtrace" value="SHORT" />
+    <envs />
+    <option name="isRedirectInput" value="false" />
+    <option name="redirectInputPath" value="" />
+    <method v="2">
+      <option name="CARGO.BUILD_TASK_PROVIDER" enabled="true" />
+    </method>
+  </configuration>
+</component>
\ No newline at end of file
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..15f0d97
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,161 @@
+use std::any::Any;
+use std::collections::HashMap;
+use std::error::Error;
+// use thiserror::Error;
+
+pub struct Event {
+    event_id: u32,
+}
+
+#[derive(PartialEq, Eq, Hash, Copy, Clone)]
+pub struct ObjectId {
+    id: u32,
+    name: &'static str,
+}
+
+pub trait SystemObject {
+    fn as_any(&self) -> &dyn Any;
+    fn get_object_id(&self) -> &ObjectId;
+    fn initialize(&mut self) -> Result<(), Box<dyn Error>>;
+}
+
+pub trait ManagedSystemObject: SystemObject + Any {}
+
+pub struct ObjectManager {
+    obj_map: HashMap<ObjectId, Box<dyn ManagedSystemObject>>,
+}
+
+impl ObjectManager {
+    pub fn new() -> ObjectManager {
+        ObjectManager {
+            obj_map: HashMap::new(),
+        }
+    }
+    pub fn insert(&mut self, sys_obj: Box<dyn ManagedSystemObject>) -> bool {
+        let obj_id = sys_obj.get_object_id();
+        if self.obj_map.contains_key(&obj_id) {
+            return false;
+        }
+        self.obj_map.insert(obj_id.clone(), sys_obj).is_none()
+    }
+
+    /// Initializes all System Objects in the hash map and returns the number of successful
+    /// initializations
+    pub fn initialize(&mut self) -> Result<u32, Box<dyn Error>> {
+        let mut init_success = 0;
+        for val in self.obj_map.values_mut() {
+            if val.initialize().is_ok() {
+                init_success += 1
+            }
+        }
+        Ok(init_success)
+    }
+
+    pub fn get<T: Any>(&self, key: &ObjectId) -> Option<&T> {
+        self.obj_map
+            .get(key)
+            .and_then(|o| o.as_ref().as_any().downcast_ref::<T>())
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::{ManagedSystemObject, ObjectId, ObjectManager, SystemObject};
+    use std::any::Any;
+    use std::error::Error;
+
+    struct ExampleSysObj {
+        id: ObjectId,
+        dummy: u32,
+        was_initialized: bool,
+    }
+
+    impl ExampleSysObj {
+        fn new(id: ObjectId, dummy: u32) -> ExampleSysObj {
+            ExampleSysObj {
+                id,
+                dummy,
+                was_initialized: false,
+            }
+        }
+    }
+
+    impl SystemObject for ExampleSysObj {
+        fn as_any(&self) -> &dyn Any {
+            self
+        }
+
+        fn get_object_id(&self) -> &ObjectId {
+            &self.id
+        }
+
+        fn initialize(&mut self) -> Result<(), Box<dyn Error>> {
+            self.was_initialized = true;
+            Ok(())
+        }
+    }
+
+    impl ManagedSystemObject for ExampleSysObj {}
+
+    struct OtherExampleObject {
+        id: ObjectId,
+        string: String,
+        was_initialized: bool,
+    }
+
+    impl SystemObject for OtherExampleObject {
+        fn as_any(&self) -> &dyn Any {
+            self
+        }
+
+        fn get_object_id(&self) -> &ObjectId {
+            &self.id
+        }
+
+        fn initialize(&mut self) -> Result<(), Box<dyn Error>> {
+            self.was_initialized = true;
+            Ok(())
+        }
+    }
+
+    impl ManagedSystemObject for OtherExampleObject {}
+
+    #[test]
+    fn test_obj_manager_simple() {
+        let mut obj_manager = ObjectManager::new();
+        let expl_obj_id = ObjectId {
+            id: 0,
+            name: "Example 0",
+        };
+        let example_obj = ExampleSysObj::new(expl_obj_id, 42);
+        assert_eq!(obj_manager.insert(Box::new(example_obj)), true);
+        let res = obj_manager.initialize();
+        assert!(res.is_ok());
+        assert_eq!(res.unwrap(), 1);
+        let obj_back_casted: Option<&ExampleSysObj> = obj_manager.get(&expl_obj_id);
+        assert!(obj_back_casted.is_some());
+        let expl_obj_back_casted = obj_back_casted.unwrap();
+        assert_eq!(expl_obj_back_casted.dummy, 42);
+        assert_eq!(expl_obj_back_casted.was_initialized, true);
+
+        let second_obj_id = ObjectId {
+            id: 12,
+            name: "Example 1",
+        };
+        let second_example_obj = OtherExampleObject {
+            id: second_obj_id,
+            string: String::from("Hello Test"),
+            was_initialized: false,
+        };
+
+        assert_eq!(obj_manager.insert(Box::new(second_example_obj)), true);
+        let res = obj_manager.initialize();
+        assert!(res.is_ok());
+        assert_eq!(res.unwrap(), 2);
+        let obj_back_casted: Option<&OtherExampleObject> = obj_manager.get(&second_obj_id);
+        assert!(obj_back_casted.is_some());
+        let expl_obj_back_casted = obj_back_casted.unwrap();
+        assert_eq!(expl_obj_back_casted.string, String::from("Hello Test"));
+        assert_eq!(expl_obj_back_casted.was_initialized, true);
+    }
+}
diff --git a/src/main.rs b/src/main.rs
index d312f21..63f5441 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,15 +1,4 @@
-use std::fmt;
-use std::error::Error;
-use thiserror::Error;
-
-struct Event {
-    event_id: u32
-}
-
-trait SystemObject {
-    fn get_object_id(&self) -> u32;
-    fn initialize(&mut self) -> Result<(), Box<dyn Error>>;
-}
+// use launchpad::ObjectManager;
 
 fn main() {
     println!("Hello World");