From 0b1c57ca5abb261b7d599fccdf3bdb60c07978ad Mon Sep 17 00:00:00 2001
From: michael <michael@localhost.localdomain>
Date: Fri, 5 Nov 2021 23:23:38 +0100
Subject: [PATCH] Added mpi4py in python module

---
 CoreFlows/swig/CMakeLists.txt | 10 ++++++++++
 CoreFlows/swig/CoreFlows.i    |  5 +++++
 2 files changed, 15 insertions(+)

diff --git a/CoreFlows/swig/CMakeLists.txt b/CoreFlows/swig/CMakeLists.txt
index 71cfdbb..553b8a1 100755
--- a/CoreFlows/swig/CMakeLists.txt
+++ b/CoreFlows/swig/CMakeLists.txt
@@ -11,6 +11,16 @@ INCLUDE_DIRECTORIES(
   ${SLEPC_INCLUDES}
 )
 
+IF(    ${SOLVERLAB_WITH_MPI} )
+  INCLUDE_DIRECTORIES(
+    ${Python_INCLUDE_DIRS}
+    ${CoreFlows_INCLUDES}											    #
+    ${PETSC_INCLUDES}
+    ${SLEPC_INCLUDES}
+    ${MPI4PY_ROOT_DIR}/include/
+  ) 
+ENDIF( ${SOLVERLAB_WITH_MPI} )
+
 IF   (${CMAKE_VERSION} VERSION_LESS "3.8.0") 
   SWIG_ADD_MODULE(CoreFlows python CoreFlows.i)
 ELSE (${CMAKE_VERSION} VERSION_LESS "3.8.0") 
diff --git a/CoreFlows/swig/CoreFlows.i b/CoreFlows/swig/CoreFlows.i
index a3f8ddf..d659bac 100755
--- a/CoreFlows/swig/CoreFlows.i
+++ b/CoreFlows/swig/CoreFlows.i
@@ -4,6 +4,11 @@
 %include std_vector.i
 %include std_map.i
 
+#ifdef MPI4PY_ROOT_DIR
+%include MPI4PY_ROOT_DIR/include/mpi4py/mpi4py.i
+%mpi4py_typemap(Comm, MPI_Comm);
+#endif
+
 %apply bool& INOUT {bool &stop}
 
 namespace std {
-- 
2.39.2