fiqus 2025.12.0__py3-none-any.whl → 2026.1.0__py3-none-any.whl

fiqus/getdp_runners/RunGetdpConductorAC_CC.py

@@ -0,0 +1,123 @@
+import timeit
+import logging
+from enum import Enum
+import os
+import subprocess
+import re
+import pandas as pd
+
+import gmsh
+import numpy as np
+
+#from fiqus.data import RegionsModelFiQuS
+from fiqus.utils.Utils import GmshUtils, FilesAndFolders
+from fiqus.data.RegionsModelFiQuS import RegionsModel
+# import fiqus.data.DataConductorACGeom as geom
+
+from fiqus.pro_assemblers.ProAssembler import ASS_PRO
+
+logger = logging.getLogger('FiQuS')
+
+class Solve:
+    def __init__(self, fdm, GetDP_path, geometry_folder, mesh_folder, verbose=True):
+        self.fdm = fdm
+        self.cacdm = fdm.magnet
+        self.GetDP_path = GetDP_path
+        self.solution_folder = os.path.join(os.getcwd())
+        self.magnet_name = fdm.general.magnet_name
+        self.geometry_folder = geometry_folder
+        self.mesh_folder = mesh_folder
+        self.mesh_file = os.path.join(self.mesh_folder, f"{self.magnet_name}.msh")
+        self.pro_file = os.path.join(self.solution_folder, f"{self.magnet_name}.pro")
+        self.regions_file = os.path.join(mesh_folder, f"{self.magnet_name}.regions")
+        
+        self.verbose = verbose
+        self.gu = GmshUtils(self.solution_folder, self.verbose)
+        self.gu.initialize(verbosity_Gmsh=fdm.run.verbosity_Gmsh)
+
+        self.ass_pro = ASS_PRO(os.path.join(self.solution_folder, self.magnet_name))
+        self.regions_model = FilesAndFolders.read_data_from_yaml(self.regions_file, RegionsModel)
+        self.material_properties_model = None
+
+        self.ed = {} # excitation dictionary
+
+        gmsh.option.setNumber("General.Terminal", verbose)
+
+    def assemble_pro(self):
+        logger.info("Assembling .pro file")
+        self.ass_pro.assemble_combined_pro(template = self.cacdm.solve.pro_template, rm = self.regions_model, dm = self.fdm, ed=self.ed, mp=self.material_properties_model)
+
+    def read_excitation(self, inputs_folder_path):
+        """
+        Function for reading a CSV file for the 'piecewise' excitation case.
+
+        :param inputs_folder_path: The full path to the folder with input files.
+        :type inputs_folder_path: str
+        """
+        if self.cacdm.solve.source_parameters.source_type == 'piecewise' and self.cacdm.solve.source_parameters.piecewise.source_csv_file:
+            input_file = os.path.join(inputs_folder_path, self.cacdm.solve.source_parameters.piecewise.source_csv_file)
+            logger.info(f'Using excitation from file: {input_file}')
+            df = pd.read_csv(input_file, delimiter=',', engine='python')
+            excitation_time = df['time'].to_numpy(dtype='float').tolist()
+            self.ed['time'] = excitation_time
+            excitation_b = df['b'].to_numpy(dtype='float').tolist()
+            self.ed['b'] = excitation_b
+            excitation_I = df['I'].to_numpy(dtype='float').tolist()
+            self.ed['I'] = excitation_I
+
+    def run_getdp(self, solve = True, postOperation = True, gui = False):
+        command = ["-v2", "-verbose", "3"]
+        if solve: 
+            command += ["-solve", "MagDyn", "-mat_mumps_icntl_14","100"] # icntl for mumps just by precaution
+        command += ["-pos", "MagDyn"]
+
+        logger.info(f"Running GetDP with command: {command}")
+        startTime = timeit.default_timer()
+
+        if self.cacdm.solve.general_parameters.noOfMPITasks:
+            mpi_prefix = ["mpiexec", "-np", str(self.cacdm.solve.general_parameters.noOfMPITasks)]
+        else:
+            mpi_prefix = []
+
+        getdpProcess = subprocess.Popen(mpi_prefix + [self.GetDP_path] + [self.pro_file] + command + ["-msh"] + [self.mesh_file], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+
+        with getdpProcess.stdout:
+            for line in iter(getdpProcess.stdout.readline, b""):
+                line = line.decode("utf-8").rstrip()
+                line = line.split("\r")[-1]
+                if not "Test" in line:
+                    if line.startswith("Info"):
+                        parsedLine = re.sub(r"Info\s+:\s+", "", line)
+                        logger.info(parsedLine)
+                    elif line.startswith("Warning"):
+                        parsedLine = re.sub(r"Warning\s+:\s+", "", line)
+                        logger.warning(parsedLine)
+                    elif line.startswith("Error"):
+                        parsedLine = re.sub(r"Error\s+:\s+", "", line)
+                        logger.error(parsedLine)
+                        logger.error("Solving CAC failed.")
+                        # raise Exception(parsedLine)
+                    elif re.match("##", line):
+                        logger.critical(line)
+                    else:
+                        logger.info(line)
+        
+        simulation_time = timeit.default_timer()-startTime
+        
+        if gui and ((postOperation and not solve) or (solve and postOperation)): # and self.cacdm.postproc.generate_pos_files
+            # gmsh.option.setNumber("Geometry.Volumes", 1)
+            # gmsh.option.setNumber("Geometry.Surfaces", 1)
+            # gmsh.option.setNumber("Geometry.Curves", 1)
+            # gmsh.option.setNumber("Geometry.Points", 0)
+            posFiles = [
+                fileName
+                for fileName in os.listdir(self.solution_folder)
+                if fileName.endswith(".pos")
+            ]
+            for posFile in posFiles:
+                gmsh.open(os.path.join(self.solution_folder, posFile))
+            self.gu.launch_interactive_GUI()
+        else:
+            if gmsh.isInitialized():
+                gmsh.clear()
+                gmsh.finalize()

fiqus/mains/MainConductorAC_CC.py

@@ -0,0 +1,148 @@
+
+"""MainConductorAC_CC.py:"""
+
+import os
+import sys
+
+from fiqus.geom_generators.GeometryConductorAC_CC import Generate_geometry
+from fiqus.mesh_generators.MeshConductorAC_CC import Mesh
+from fiqus.getdp_runners.RunGetdpConductorAC_CC import Solve
+from fiqus.post_processors.PostProcessAC_CC import Post_Process
+
+if len(sys.argv) == 3:
+    sys.path.insert(0, os.path.join(os.getcwd(), 'steam-fiqus-dev'))
+from fiqus.data.DataFiQuS import FDM
+
+class MainConductorAC_CC:
+    def __init__(self, fdm, inputs_folder_path='', verbose=True):
+        """
+        Main class for working with simulations for CAC_CC type magnets
+        :param fdm: FiQuS data model
+        :type fdm: FDM
+        :param inputs_folder_path: full path to folder with input files, i.e. conductor and STEP files
+        :type inputs_folder_path: str
+        :param verbose: if True, more info is printed in the console
+        :type verbose: bool
+        :return: nothing, only saves files on disk
+        :rtype: none
+        """
+        self.verbose = verbose
+        self.fdm = fdm
+        self.inputs_folder_path = inputs_folder_path
+        self.GetDP_path = None
+        self.geom_folder = None
+        self.mesh_folder = None
+        self.solution_folder = None
+        self.model_file = None
+        self.model_folder = None
+
+    def generate_geometry(self, gui=False):
+        """
+        Main method for loading the geometry of CAC_CC models
+        :param gui: if true, graphical user interface (gui) of Gmsh is opened at the end
+        :type gui: bool
+        :return: nothing, only saves files on disk
+        :rtype: none
+        """
+        os.chdir(self.geom_folder)
+        gg = Generate_geometry(fdm=self.fdm, inputs_folder_path=self.inputs_folder_path, verbose=self.verbose)
+        gg.generate_HTS_layer()        
+        gg.generate_silver_top_layer()              
+        gg.generate_substrate_layer()        
+        gg.generate_copper_top_layer()
+        gg.generate_silver_bottom_layer()        
+        gg.generate_copper_bottom_layer()
+        gg.generate_copper_left_layer()
+        gg.generate_copper_right_layer()        
+        gg.generate_air_region()
+        gg.finalize_and_write()
+
+    def load_geometry(self, gui=False):
+        """
+        Main method for loading the geometry of CAC_CC models
+        """
+        os.chdir(self.geom_folder)
+        gg = Generate_geometry(fdm=self.fdm, inputs_folder_path=self.inputs_folder_path, verbose=self.verbose)
+        self.geometry = gg
+        gg.load_geometry(gui=gui)
+
+    def pre_process(self, gui=False):
+        """
+        Main method for preprocessing of CAC_CC models
+        :param gui: if true, graphical user interface (gui) of Gmsh is opened at the end
+        :type gui: bool
+        :return: nothing, only saves files on disk
+        :rtype: none
+        """
+        os.chdir(self.geom_folder)
+
+
+    def mesh(self, gui=False):
+        """
+        Main method for building the mesh of CAC_CC models
+        :param gui: if true, graphical user interface (gui) of Gmsh is opened at the end
+        :type gui: bool
+        :return: dictionary with mesh quality stats
+        :rtype: dict
+        """
+        os.chdir(self.mesh_folder)
+        m = Mesh(self.fdm)
+        m.generate_mesh(self.geom_folder)
+        return {"gamma": 0}
+
+
+    def load_mesh(self, gui=False):
+        """
+        Main method for loading the mesh of CAC_CC models
+        :param gui: if true, graphical user interface (gui) of Gmsh is opened at the end
+        :type gui: bool
+        :return: Nothing, only saves files on disk
+        :rtype: none
+        """
+        os.chdir(self.mesh_folder)
+
+    def solve_and_postprocess_getdp(self, gui: bool = False):
+        """
+        Assembles the .pro-file from the template, then runs the simulation and the post-processing steps using GetDP.
+        """
+        os.chdir(self.solution_folder)
+
+        s = Solve(self.fdm, self.GetDP_path, self.geom_folder, self.mesh_folder, self.verbose)
+        s.read_excitation(inputs_folder_path=self.inputs_folder_path)
+        s.assemble_pro()
+        s.run_getdp(solve = True, postOperation = True, gui = gui)
+    
+    def post_process_getdp(self, gui: bool = False):
+        """ 
+        Runs the post-processing steps trough GetDP.
+        """
+        os.chdir(self.solution_folder)
+
+        s = Solve(self.fdm, self.GetDP_path, self.geom_folder, self.mesh_folder, self.verbose)
+        s.read_excitation(inputs_folder_path=self.inputs_folder_path)
+        s.assemble_pro()
+        s.run_getdp(solve = False, postOperation = True, gui = gui)
+
+    def post_process_python(self, gui=False):
+        """
+        Main method for postprocessing using python (without solving) of CAC_CC models
+        :param gui: if true, graphical user interface (gui) of Gmsh is opened at the end
+        :type gui: bool
+        :return: Nothing, only saves files on disk
+        :rtype: none
+        """
+        os.chdir(self.solution_folder)
+        p=Post_Process(self.fdm, verbose=self.verbose)
+        p.cleanup()
+        return {'overall_error': 0}
+
+    def plot_python(self, gui=False):
+        """
+        Main method for making python plots related to CAC_CC models
+        :param gui: if true, graphical user interface (gui) of Gmsh is opened at the end
+        :type gui: bool
+        :return: Nothing, only saves files on disk
+        :rtype: none
+        """
+        os.chdir(self.solution_folder)
+