diff --git a/LoopDataConverter/input/input_data.py b/LoopDataConverter/input/input_data.py new file mode 100644 index 0000000..2fa7cdc --- /dev/null +++ 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) \ No newline at end of file