chore: added dataclass for input data

Loop3D · Jul 8, 2024 · f32e3f2 · f32e3f2
1 parent fb2b6d7
commit f32e3f2
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/LoopDataConverter/input/input_data.py b/LoopDataConverter/input/input_data.py
@@ -0,0 +1,22 @@
+from dataclasses import dataclass
+from ..datatypes import Datatype
+
+
+
+@dataclass
+class InputData:
+    geology: Datatype.GEOLOGY = None
+    structure: Datatype.STRUCTURE = None
+    fault: Datatype.FAULT = None
+    fold: Datatype.FOLD = None
+
+
+    def __getitem__(self, datatype: Datatype):
+
+        return self.__dict__[datatype]
+
+@dataclass 
+class OutputData(InputData):
+
+    def __getitem__(self, datatype: Datatype):
+        return super().__getitem__(datatype)