MPIPortsCommunication.hpp

Go to the documentation of this file.

#pragma once
#ifndef PRECICE_NO_MPI
 
#include <map>
#include <mpi.h>
#include <set>
#include <stddef.h>
#include <string>
 
#include "com/MPICommunication.hpp"
#include "logging/Logger.hpp"
#include "precice/impl/Types.hpp"
 
namespace precice::com {
class MPIPortsCommunication : public MPICommunication {
public:
  explicit MPIPortsCommunication(std::string addressDirectory = ".");
 
  ~MPIPortsCommunication() override;
 
  size_t getRemoteCommunicatorSize() override;
 
  void acceptConnection(std::string const &acceptorName,
                        std::string const &requesterName,
                        std::string const &tag,
                        int                acceptorRank,
                        int                rankOffset = 0) override;
 
  void acceptConnectionAsServer(std::string const &acceptorName,
                                std::string const &requesterName,
                                std::string const &tag,
                                int                acceptorRank,
                                int                requesterCommunicatorSize) override;
 
  void requestConnection(std::string const &acceptorName,
                         std::string const &requesterName,
                         std::string const &tag,
                         int                requesterRank,
                         int                requesterCommunicatorSize) override;
 
  void requestConnectionAsClient(std::string const   &acceptorName,
                                 std::string const   &requesterName,
                                 std::string const   &tag,
                                 std::set<int> const &acceptorRanks,
                                 int                  requesterRank) override;
 
  void closeConnection() override;
 
  void prepareEstablishment(std::string const &acceptorName,
                            std::string const &requesterName) override;
 
  void cleanupEstablishment(std::string const &acceptorName,
                            std::string const &requesterName) override;
 
private:
  MPI_Comm &communicator(Rank rank) override;
 
  Rank rank(int rank) override;
 
  logging::Logger _log{"com::MPIPortsCommunication"};
 
  std::string _addressDirectory;

  std::map<int, MPI_Comm> _communicators;

  std::string _portName = std::string(MPI_MAX_PORT_NAME, '\0');
 
  bool _isAcceptor = false;
};
} // namespace precice::com
 
#endif // not PRECICE_NO_MPI