ProcessAdm.h

Go to the documentation of this file.

//==============================================================================
//==============================================================================
 
#ifndef _PROCESS_ADM_H
#define _PROCESS_ADM_H
 
#ifdef HAS_PETSC
#include "petscsys.h"
#elif defined(HAVE_MPI)
#include "mpi.h"
#endif
#include <vector>
 
#include "DomainDecomposition.h"
#include "LogStream.h"
 
 
class ProcessAdm
{
  int myPid;       
  int nProc;       
  bool parallel;   
 
#if defined(HAS_PETSC) || defined(HAVE_MPI)
  MPI_Comm comm;   
#endif
 
public:
  mutable utl::LogStream cout; 
 
  DomainDecomposition dd; 
 
  ProcessAdm();
#if defined(HAS_PETSC) || defined(HAVE_MPI)
  explicit ProcessAdm(MPI_Comm& mpi_comm);
#endif
  ProcessAdm(const ProcessAdm& adm);
 
#ifdef HAVE_MPI
  explicit ProcessAdm(bool hack);
#endif
 
  ~ProcessAdm();
 
  int getProcId() const  { return myPid; }
  int getNoProcs() const { return nProc; }
  int isParallel() const { return parallel; }
 
  void broadcast(std::vector<double>& vec, int root) const;
 
  void broadcast(std::vector<int>& vec, int root) const;
 
#if defined(HAS_PETSC) || defined(HAVE_MPI)
  MPI_Comm* getCommunicator() { return &comm; }
  const MPI_Comm* getCommunicator() const { return &comm; }
 
  void setCommunicator(const MPI_Comm* comm2);
 
  ProcessAdm& operator=(const ProcessAdm&);
 
  void send(int value, int dest) const;
  void send(std::vector<int>& ivec, int dest) const;
  void send(double value, int dest) const;
  void send(std::vector<double>& rvec, int dest) const;
 
  void receive(int& value, int source) const;
  void receive(std::vector<int>& ivec, int source) const;
  void receive(double& value, int source) const;
  void receive(std::vector<double>& rvec, int source) const;
 
  int allReduce(int value, MPI_Op oper) const;
  void allReduce(std::vector<int>& vec, MPI_Op oper) const;
 
  double allReduce(double value, MPI_Op oper) const;
  void allReduce(std::vector<double>& vec, MPI_Op oper) const;
#endif
 
  template<class T> void allReduceAsSum(std::vector<T>& vec) const
  {
#ifdef HAS_PETSC
    if (parallel)
      this->allReduce(vec,MPI_SUM);
#endif
  }
};
 
#endif