add option to select mpi module

diff --git a/bin/salomeContext.py b/bin/salomeContext.py
index 11a980bdb22b5069bbd1cba27656ce49a798bb81..3dfaa46d9bde52459719a006d1a7ca813b6e3411 100644 (file)
--- a/bin/salomeContext.py
+++ b/bin/salomeContext.py
@@ -237,6 +237,21 @@ class SalomeContext:
       # try to default to "start" command
       command = "_runAppli"
 
+    mpi_module_option = "--with_mpi_module="
+    mpi_module = [x for x in args if x.startswith(mpi_module_option)]
+    if mpi_module:
+      mpi_module = mpi_module[0][len(mpi_module_option):]
+      print "Trying to load MPI module: %s..."%mpi_module,
+      import subprocess
+      try:
+        subprocess.call(["module", "load", mpi_module])
+        print " OK"
+      except:
+        print " ** Failed **"
+        pass
+      options = [x for x in options if not x.startswith(mpi_module_option)]
+      pass
+
     try:
       res = getattr(self, command)(options) # run appropriate method
       return res or (None, None)