mpiguard.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
 
#ifndef DUNE_COMMON_MPIGUARD_HH
#define DUNE_COMMON_MPIGUARD_HH
 
#include "mpihelper.hh"
#include "collectivecommunication.hh"
#include "mpicollectivecommunication.hh"
#include <dune/common/exceptions.hh>
 
namespace Dune
{
 
#ifndef DOXYGEN
 
  /*
     Interface class for the communication needed by MPIGuard
   */
  struct GuardCommunicator
  {
    // cleanup
    virtual ~GuardCommunicator() {};
    // all the communication methods we need
    virtual int rank() = 0;
    virtual int size() = 0;
    virtual int sum(int i) = 0;
    // create a new GuardCommunicator pointer
    template <class C>
    static GuardCommunicator * create(const C & c);
  };
 
  namespace {
    /*
       templated implementation of different communication classes
     */
    // the default class will always fail, due to the missing implementation of "sum"
    template <class Imp>
    struct GenericGuardCommunicator
      : public GuardCommunicator
    {};
    // specialization for CollectiveCommunication
    template <class T>
    struct GenericGuardCommunicator< CollectiveCommunication<T> >
      : public GuardCommunicator
    {
      const CollectiveCommunication<T> comm;
      GenericGuardCommunicator(const CollectiveCommunication<T> & c) :
        comm(c) {}
      virtual int rank() { return comm.rank(); };
      virtual int size() { return comm.size(); };
      virtual int sum(int i) { return comm.sum(i); }
    };
 
#if HAVE_MPI
    // specialization for MPI_Comm
    template <>
    struct GenericGuardCommunicator<MPI_Comm>
      : public GenericGuardCommunicator< CollectiveCommunication<MPI_Comm> >
    {
      GenericGuardCommunicator(const MPI_Comm & c) :
        GenericGuardCommunicator< CollectiveCommunication<MPI_Comm> >(
          CollectiveCommunication<MPI_Comm>(c)) {}
    };
#endif
  }   // anonymous namespace
 
  template<class C>
  GuardCommunicator * GuardCommunicator::create(const C & comm)
  {
    return new GenericGuardCommunicator<C>(comm);
  }
#endif
 
  class MPIGuardError : public ParallelError {};
 
  class MPIGuard
  {
    GuardCommunicator * comm_;
    bool active_;
 
    // we don't want to copy this class
    MPIGuard (const MPIGuard &);
 
  public:
    MPIGuard (bool active=true) :
      comm_(GuardCommunicator::create(
              MPIHelper::getCollectiveCommunication())),
      active_(active)
    {}
 
    MPIGuard (MPIHelper & m, bool active=true) :
      comm_(GuardCommunicator::create(
              m.getCollectiveCommunication())),
      active_(active)
    {}
 
    template <class C>
    MPIGuard (const C & comm, bool active=true) :
      comm_(GuardCommunicator::create(comm)),
      active_(active)
    {}
 
    ~MPIGuard()
    {
      if (active_)
      {
        active_ = false;
        finalize(false);
      }
      delete comm_;
    }
 
    void reactivate() {
      if (active_ == true)
        finalize();
      active_ = true;
    }
 
    void finalize(bool success = true)
    {
      int result = success ? 0 : 1;
      bool was_active = active_;
      active_ = false;
      result = comm_->sum(result);
      if (result>0 && was_active)
      {
        DUNE_THROW(MPIGuardError, "Terminating process "
                   << comm_->rank() << " due to "
                   << result << " remote error(s)");
      }
    }
  };
 
}
 
#endif // DUNE_COMMON_MPIGUARD_HH