MPIPrint for raising errors only on master node

Author: Nicola-Fonzi

SU2_PY/FSI_tools/FSI_config.py

@@ -54,8 +54,9 @@ class FSIConfig:
     Read the file and store all the options into a dictionary.
     """
 
-    def __init__(self,FileName):
+    def __init__(self,FileName,comm):
         self.ConfigFileName = FileName
+        self.comm = comm
         self._ConfigContent = {}
         self.readConfig()
         self.applyDefaults()
@@ -124,19 +125,35 @@ def applyDefaults(self):
 
         if "MAPPING_MODES" not in self._ConfigContent:
             self._ConfigContent["MAPPING_MODES"] = "NO"
-            print("MAPPING_MODES keyword was not found in the configuration file of the interface, setting to NO")
+            MPIPrint("MAPPING_MODES keyword was not found in the configuration file of the interface, setting to NO",False)
 
         if "IMPOSED_MOTION" not in self._ConfigContent:
             self._ConfigContent["IMPOSED_MOTION"] = "NO"
-            print("IMPOSED_MOTION keyword was not found in the configuration file of the interface, setting to NO")
+            MPIPrint("IMPOSED_MOTION keyword was not found in the configuration file of the interface, setting to NO",False)
 
         if self._ConfigContent["IMPOSED_MOTION"] == "YES":
             if self._ConfigContent["AITKEN_RELAX"] != "STATIC" or self._ConfigContent["AITKEN_PARAM"] != 1.0:
-                raise Exception("When imposing motion, the Aitken parameter must be static and equal to 1")
+                MPIPrint("When imposing motion, the Aitken parameter must be static and equal to 1",True)
 
         if self._ConfigContent["RESTART_SOL"] == "YES":
             if self._ConfigContent["TIME_TRESHOLD"] != -1:
-                raise Exception("When restarting a simulation, the time threshold must be -1 for immediate coupling")
+                MPIPrint("When restarting a simulation, the time threshold must be -1 for immediate coupling",True)
 
         if self._ConfigContent["MAPPING_MODES"] == "YES" and self._ConfigContent["CSD_SOLVER"]!="NATIVE":
-            raise Exception("Mapping modes only works with the native solver")
+            MPIPrint("Mapping modes only works with the native solver",True)
+
+    def MPIPrint(self, message, error):
+        """
+        Print a message, or raise error, on screen only from the master process.
+        """
+
+        if self.comm:
+            myid = self.comm.Get_rank()
+        else:
+            myid = 0
+
+        if not myid:
+            if error:
+                raise Exception(message)
+            else:
+                print(message)

SU2_PY/fsi_computation.py

@@ -86,7 +86,7 @@ def main():
 
   confFile = str(options.filename)
 
-  FSI_config = FSI.FSIConfig(confFile) 		# FSI configuration file
+  FSI_config = FSI.FSIConfig(confFile, comm) 		# FSI configuration file
   CFD_ConFile = FSI_config['CFD_CONFIG_FILE_NAME']	# CFD configuration file
   CSD_ConFile = FSI_config['CSD_CONFIG_FILE_NAME']	# CSD configuration file