communicator.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2025 VTT Technical Research Centre of Finland Ltd
// SPDX-License-Identifier: AGPL-3.0-or-later
 
#ifndef PFC_MPI_COMMUNICATOR_HPP
#define PFC_MPI_COMMUNICATOR_HPP
 
#include <memory>
#include <mpi.h>
 
namespace pfc {
namespace mpi {
 
class communicator {
 
public:
  communicator();
  operator MPI_Comm() const;
  int rank() const;
  int size() const;
 
protected:
  std::shared_ptr<MPI_Comm> comm_ptr;
};
 
inline communicator::communicator() { comm_ptr.reset(new MPI_Comm(MPI_COMM_WORLD)); }
 
inline communicator::operator MPI_Comm() const {
  if (comm_ptr)
    return *comm_ptr;
  else
    return MPI_COMM_NULL;
}
 
inline int communicator::size() const {
  int size_;
  MPI_Comm_size(MPI_Comm(*this), &size_);
  return size_;
}
 
inline int communicator::rank() const {
  int rank_;
  MPI_Comm_rank(MPI_Comm(*this), &rank_);
  return rank_;
}
 
} // namespace mpi
} // namespace pfc
 
#endif // PFC_MPI_COMMUNICATOR_HPP