feature: allow logger to be saved from any class, by passing the material folder + forward the logger to solver if wake object is passed to constructor

Author: elenafuengar

wakis/logger.py

@@ -24,10 +24,13 @@ def __init__(self):
         self.solver = {}
         self.wakeSolver = {}
 
-    def save_logs(self):
+    def save_logs(self, results_folder=None):
         """
         Save all logs (grid, solver, wakeSolver) into log-file inside the results folder.
         """
+        if results_folder is not None:
+            self.wakeSolver["results_folder"] = results_folder
+            
         logfile = os.path.join(self.wakeSolver["results_folder"], "wakis.log")
 
         # Write sections

wakis/solverFIT3D.py

@@ -137,6 +137,7 @@ def __init__(self, grid, wake=None, cfln=0.5, dt=None,
 
         # Wake computation
         self.wake = wake
+        self.logger.wakeSolver=self.wake.logger.wakeSolver
 
         # Fields
         self.dtype = dtype
@@ -952,6 +953,7 @@ def apply_stl(self):
         grid = self.grid.grid
         self.stl_solids = self.grid.stl_solids
         self.stl_materials = self.grid.stl_materials
+        self.stl_colors = self.grid.stl_colors
 
         for key in self.stl_solids.keys():