communicator.hpp

namespace diy
{
namespace mpi
{

  class communicator
  {
    public:
                communicator(MPI_Comm comm = MPI_COMM_WORLD):
                  comm_(comm)                       { MPI_Comm_rank(comm_, &rank_); MPI_Comm_size(comm_, &size_); }

      int       rank() const                        { return rank_; }
      int       size() const                        { return size_; }

      //void      send(int dest,
      //               int tag,
      //               const void* buf,
      //               MPI_Datatype datatype) const   { }

      template<class T>
      void      send(int dest, int tag, const T& x) const   { detail::send<T>()(comm_, dest, tag, x); }

      template<class T>
      status    recv(int source, int tag, T& x) const       { return detail::recv<T>()(comm_, source, tag, x); }

      template<class T>
      request   isend(int dest, int tag, const T& x) const  { return detail::isend<T>()(comm_, dest, tag, x); }

      template<class T>
      request   irecv(int source, int tag, T& x) const      { return detail::irecv<T>()(comm_, source, tag, x); }

      status    probe(int source, int tag) const            { status s; MPI_Probe(source, tag, comm_, &s.s); return s; }

      inline
      optional<status>
                iprobe(int source, int tag) const;

      void      barrier() const                             { MPI_Barrier(comm_); }

                operator MPI_Comm() const                   { return comm_; }

    private:
      MPI_Comm  comm_;
      int       rank_;
      int       size_;
  };
}
}

diy::mpi::optional<diy::mpi::status>
diy::mpi::communicator::
iprobe(int source, int tag) const
{
  status s;
  int flag;
  MPI_Iprobe(source, tag, comm_, &flag, &s.s);
  if (flag)
    return s;
  return optional<status>();
}