Start work on optimized multiprocessing random a/c gen. & eval.HEADmaster

1 files changed, 112 insertions, 0 deletions

diff --git a/aircraftstudio/creator/base.py b/aircraftstudio/creator/base.py
new file mode 100644
index 0000000..92fe421
--- /dev/null
+++ b/aircraftstudio/creator/base.py
@@ -0,0 +1,112 @@
+"""The base.py module contains parent classes for components."""
+
+import numpy as np
+import os.path
+import random
+import logging
+
+from aircraftstudio import creator
+
+logging.basicConfig(filename='log_base.txt',
+                    level=logging.DEBUG,
+                    format='%(asctime)s - %(levelname)s - %(message)s')
+
+
+class Aircraft:
+    """This class tracks all sub-components and is fed to the evaluator."""
+    name = None
+    fuselage = None
+    propulsion = None
+    wing = None
+    properties = {}
+
+    naca = [2412, 3412, 2420]
+
+    def __init__(self):
+        self.results = {}
+
+    def __str__(self):
+        return self.name
+
+    @classmethod
+    def from_default(cls):
+        aircraft = Aircraft()
+        aircraft.name = 'default_aircraft_' + str(random.randrange(1000, 9999))
+        airfoil = creator.wing.Airfoil(aircraft, 'default_airfoil')
+        airfoil.add_naca(2412)
+        soar1 = creator.wing.Spar(airfoil, 'default_spar_1', 0.30)
+        soar2 = creator.wing.Spar(airfoil, 'default_spar_2', 0.60)
+        stringer = creator.wing.Stringer(airfoil, 'default_stringer')
+        return aircraft
+
+    @classmethod
+    def from_random(cls):
+        aircraft = Aircraft()
+        aircraft.name = 'random_aircraft_' + str(random.randrange(1000, 9999))
+        airfoil = creator.wing.Airfoil(aircraft, 'random_airfoil')
+        airfoil.add_naca(random.choice(cls.naca))
+        soar1 = creator.wing.Spar(airfoil, 'random_spar_1',
+                                  random.randrange(20, 80) / 100)
+        soar2 = creator.wing.Spar(airfoil, 'random_spar_2',
+                                  random.randrange(20, 80) / 100)
+        stringer = creator.wing.Stringer(airfoil, 'random_stringer',
+                                         random.randint(1, 10),
+                                         random.randint(1, 10),
+                                         random.randint(1, 10),
+                                         random.randint(1, 10))
+        return aircraft
+
+
+class Component:
+    """Basic component providing coordinates, tools and a component tree."""
+    def __init__(self, parent, name):
+        self.parent = parent
+        self.name = name
+        self.x = np.array([])
+        self.z = np.array([])
+        self.y = np.array([])
+        self.material = None
+        self.mass = float()
+        self.properties = {}
+
+    def __str__(self):
+        return self.name
+
+    def info_print(self, round):
+        """Print all the component's coordinates to the terminal."""
+        name = f'    CREATOR DATA FOR {str(self).upper()}    '
+        num_of_dashes = len(name)
+        print(num_of_dashes * '-')
+        print(name)
+        for k, v in self.__dict__.items():
+            if type(v) is not np.ndarray:
+                print(f'{k}:\n', v)
+                print(num_of_dashes * '-')
+        for k, v in self.__dict__.items():
+            if type(v) is np.ndarray:
+                print(f'{k}:\n', np.around(v, round))
+        return None
+
+    def info_save(self,
+                  save_path='/home/blendux/Projects/Aircraft_Studio/save'):
+        """Save all the object's coordinates (must be full path)."""
+        file_name = f'{self.name}_info.txt'
+        full_path = os.path.join(save_path, file_name)
+        try:
+            with open(full_path, 'w') as f:
+                for k, v in self.__dict__.items():
+                    if type(v) is not np.ndarray:
+                        f.write(f'{k}=\n')
+                        f.write(str(v))
+                        f.write("\n")
+                        # print(num_of_dashes * '-')
+                for k, v in self.__dict__.items():
+                    if type(v) is np.ndarray:
+                        f.write(f'{k}=\n')
+                        f.write(str(v))
+                        f.write("\n")
+            logging.debug(f'Successfully wrote to file {full_path}')
+        except IOError:
+            print(f'Unable to write {file_name} to specified directory.\n',
+                  'Was the full path passed to the function?')
+        return None