communications: add getRank and getProcessorCount to DataCommunicator…

… class
optimad · Apr 16, 2024 · 3b2ab1a · 3b2ab1a
1 parent 3ae33ab
commit 3b2ab1a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/communications/communications.cpp b/src/communications/communications.cpp
@@ -75,6 +75,32 @@ DataCommunicator::~DataCommunicator()
     }
 }
 
+/*!
+    Gets the rank in the MPI communicator
+
+    \return the MPI rank.
+*/
+int DataCommunicator::getRank() const
+{
+    int rank;
+    MPI_Comm_rank(m_communicator, &rank);
+
+    return rank;
+}
+
+/*!
+    Gets the number of processes in the MPI communicator
+
+    \return The number of processes in the MPI communicator.
+*/
+int DataCommunicator::getProcessorCount() const
+{
+    int nProcs;
+    MPI_Comm_size(m_communicator, &nProcs);
+
+    return nProcs;
+}
+
 /*!
 	Gets the MPI communicator
 

diff --git a/src/communications/communications.hpp b/src/communications/communications.hpp
@@ -49,6 +49,9 @@ class DataCommunicator
 
     const MPI_Comm & getCommunicator() const;
 
+    int getRank() const;
+    int getProcessorCount() const;
+
     void finalize(bool synchronous = false);
 
     void setTag(int exchangeTag);