Environment.hpp

#pragma once
 
// std c++ headers
#include <cassert>
#include <string>
 
#include <dune/common/parallel/mpihelper.hh>
 
namespace AMDiS
{
 
  class Environment
  {
    // internal static container holding a pointer to the Dune::MPIHelper.
    struct Mpi
    {
      static Mpi& instance()
      {
        static Mpi mpi;
        return mpi;
      }
 
      void registerMpiHelper(Dune::MPIHelper& mpiHelper)
      {
        mpiHelper_ = &mpiHelper;
      }
 
      int rank()
      {
        assert(mpiHelper_ != nullptr);
        return mpiHelper_->rank();
      }
 
      int size()
      {
        assert(mpiHelper_ != nullptr);
        return mpiHelper_->size();
      }
 
      Dune::MPIHelper& mpiHelper()
      {
        return *mpiHelper_;
      }
 
    private:
      Dune::MPIHelper* mpiHelper_ = nullptr;
    };
 
  public:
    Environment(std::string const& initFileName = "");
 
    Environment(int& argc, char**& argv, std::string const& initFileName = "");
 
    ~Environment();
 
    static int mpiRank()
    {
      return Mpi::instance().rank();
    }
 
    static int mpiSize()
    {
      return Mpi::instance().size();
    }
 
    static Dune::MPIHelper& mpiHelper()
    {
      return Mpi::instance().mpiHelper();
    }
 
    static typename Dune::MPIHelper::MPICommunicator comm()
    {
      return Dune::MPIHelper::getCommunicator();
    }
 
    static int infoLevel();
 
    static bool msgAllRanks();
 
  private:
#if AMDIS_HAS_PETSC
    bool petscInitialized_ = false;
#endif
  };
 
} // end namespace AMDiS