dir structure for evaluator

1 files changed, 6 insertions, 4 deletions

diff --git a/creator/base.py b/creator/base.py
index 24b452c..1e8646d 100644
--- a/creator/base.py
+++ b/creator/base.py
@@ -13,17 +13,19 @@ logging.basicConfig(filename='log.txt',
 class Aircraft:
     """This class tracks all sub-components and is fed to the evaluator."""
     def __init__(self, evaluator, name):
-        evaluator.tree.update({"aircraft": self})
+        self.tree = {name: self}
+        evaluator.tree[type(self).__name__] = self.tree
         self.evaluator = evaluator
         self.name = name
-        self.tree = []  # Nested list of subcomponents
 
 
 class Component:
     """Basic component providing coordinates, tools and a component tree."""
     def __init__(self, parent, name):
-        self.tree = [name]
-        parent.tree.append(self.tree)
+        self.tree = {name: self}
+        # parent.tree.update(self.tree)
+        parent.tree[type(self).__name__] = self.tree
+        self.parent = parent
         self.name = name
         self.x = np.array([])
         self.z = np.array([])