fiqus 2025.2.0__py3-none-any.whl → 2025.11.0__py3-none-any.whl

fiqus/MainFiQuS.py

@@ -19,7 +19,7 @@ from fiqus.utils.Utils import CheckForExceptions as Check
 from fiqus.utils.Utils import create_json_schema
 from fiqus.utils.Utils import get_data_settings
 from fiqus.utils.Utils import initialize_logger
-from fiqus.data.DataFiQuS import FDM
+from fiqus.data.DataFiQuS import FDM, SolveDumpDataModel
 from fiqus.data.DataSettings import DataSettings
 from fiqus.mains.MainCCT import MainCCT
 from fiqus.mains.MainMultipole import MainMultipole
@@ -86,22 +86,10 @@ class MainFiQuS:
         # Initialize Main object
         if self.fdm.magnet.type == "CCT_straight":
             self.main_magnet = MainCCT(fdm=self.fdm, verbose=verbose)
-        elif self.fdm.magnet.type == "CWS":
-            self.main_magnet = MainCWS(
-                fdm=self.fdm,
-                inputs_folder_path=pathlib.Path(input_file_path).parent,
-                verbose=self.fdm.run.verbosity_FiQuS,
-            )
         elif self.fdm.magnet.type == "Pancake3D":
             self.main_magnet = MainPancake3D(fdm=self.fdm, verbose=verbose)
         elif self.fdm.magnet.type == "CACStrand":
             self.main_magnet = MainConductorAC_Strand(fdm=self.fdm, inputs_folder_path=pathlib.Path(input_file_path).parent, outputs_folder_path=model_folder, verbose=verbose)
-        elif self.fdm.magnet.type == "CACRutherford":
-            self.main_magnet = MainConductorAC_Rutherford(fdm=self.fdm, inputs_folder_path=pathlib.Path(input_file_path).parent, verbose=verbose)
-        elif self.fdm.magnet.type == "Racetrack":
-            self.main_magnet = MainRacetrack(fdm=self.fdm, verbose=verbose)
-        elif self.fdm.magnet.type == "Racetrack3D":
-            self.main_magnet = MainRacetrack3D(fdm=self.fdm, inputs_folder_path=pathlib.Path(input_file_path).parent, verbose=verbose)
         elif self.fdm.magnet.type == "multipole":
             self.file_name = os.path.basename(input_file_path)[:-5]
             if not self.fdm.magnet.geometry.geom_file_path:
@@ -230,7 +218,7 @@ class MainFiQuS:
         else:  # post_process_getdp_only or post_process_python_only or plot_python
             required_folders = []
 
-        fdm = self.main_magnet.fdm.magnet
+
 
         self.main_magnet.geom_folder = _check_and_generate_path(folder_type="Geometry")
         if not self.fdm.run.type in ["geometry_only"]:
@@ -250,7 +238,7 @@ class MainFiQuS:
         ]:
             Util.write_data_model_to_yaml(
                 os.path.join(self.main_magnet.geom_folder, "geometry.yaml"),
-                fdm.geometry,
+                self.fdm.magnet.geometry,
                 by_alias=True,
                 with_comments=True,
             )
@@ -262,7 +250,7 @@ class MainFiQuS:
         ]:
             Util.write_data_model_to_yaml(
                 os.path.join(self.main_magnet.mesh_folder, "mesh.yaml"),
-                fdm.mesh,
+                self.fdm.magnet.mesh,
                 by_alias=True,
                 with_comments=True,
             )
@@ -272,11 +260,20 @@ class MainFiQuS:
             "solve_with_post_process_python",
             "solve_only",
             "post_process",
-            "plot_python"
+            "plot_python",
+            "postprocess_veusz"
         ]:
+            solve_dump_data = SolveDumpDataModel(
+                solve=self.fdm.magnet.solve,
+                circuit=self.fdm.circuit,
+                power_supply=self.fdm.power_supply,
+                quench_protection=self.fdm.quench_protection,
+                quench_detection=self.fdm.quench_detection,
+                conductors=self.fdm.conductors
+            )
             Util.write_data_model_to_yaml(
                 os.path.join(self.main_magnet.solution_folder, "solve.yaml"),
-                fdm.solve,
+                solve_dump_data,
                 by_alias=True,
                 with_comments=True,
             )
@@ -287,11 +284,12 @@ class MainFiQuS:
             "post_process_python_only",
             "post_process_getdp_only",
             "post_process",
+            "postprocess_veusz",
             "plot_python"
         ]:
             Util.write_data_model_to_yaml(
                 os.path.join(self.main_magnet.solution_folder, "postproc.yaml"),
-                fdm.postproc,
+                self.fdm.magnet.postproc,
                 by_alias=True,
                 with_comments=True,
             )
@@ -337,6 +335,7 @@ class MainFiQuS:
         except:
             self.logger.warning("Run log could not be completed.")
 
+
     def build_magnet(self):
         """
         Main method to build magnets, i.e. to run various fiqus run types and magnet types
@@ -389,6 +388,8 @@ class MainFiQuS:
             self.summary["solution_time"] = (
                 self.main_magnet.solve_and_postprocess_getdp(gui=self.main_magnet.fdm.run.launch_gui)
             )
+        elif self.fdm.run.type == "postprocess_veusz":
+            self.main_magnet.post_process_veusz(gui=self.main_magnet.fdm.run.launch_gui)
         elif self.fdm.run.type == "post_process_getdp_only":
             self.main_magnet.post_process_getdp(gui=self.main_magnet.fdm.run.launch_gui)
         elif self.fdm.run.type == "post_process_python_only":
@@ -405,15 +406,10 @@ class MainFiQuS:
             self.main_magnet.batch_post_process_python()
         os.chdir(self.start_folder)
 
-        if self.file_name: json.dump(self.summary, open(f"{os.path.join(self.wrk_folder, self.file_name)}.json", 'w'))
-        # mesh_par = self.fdm.magnet.mesh
-        # if self.summary['solution_time']:
-        #     with open(r"C:\Users\avitrano\PycharmProjects\steam_sdk\tests\parsims\FiQuS_run\summary.dat", 'a') as f:
-        #         content = (f"{mesh_par.mesh_coil.SizeMin}     {mesh_par.mesh_coil.SizeMax}      {mesh_par.mesh_iron.SizeMin}      {mesh_par.mesh_iron.SizeMax}       "
-        #                    f"{self.summary['solution_time']}    {self.summary['overall_error']}       {self.summary['overall_error'] * 0.999 + self.summary['solution_time'] * 0.001}       "
-        #                    f"{self.summary['SJ']}     {self.summary['SICN']}      {self.summary['SIGE']}      {self.summary['Gamma']}       "
-        #                    f"{self.summary['nodes']}    {self.summary['minimum_diff']}    {self.summary['maximum_diff']}\n")
-        #         f.writelines(content)
+        if self.file_name:
+            file_path = os.path.join(self.wrk_folder, f"{self.file_name}.json")
+            with open(file_path, 'w', encoding='utf-8') as f:
+                json.dump(self.summary, f, indent=2)
 
     @staticmethod
     def add_to_run_log(path_to_csv, run_log_row):