Proto_SPMD.H

#ifdef CH_LANG_CC
/*
 *      _______              __
 *     / ___/ /  ___  __ _  / /  ___
 *    / /__/ _ \/ _ \/  V \/ _ \/ _ \
 *    \___/_//_/\___/_/_/_/_.__/\___/
 *    Please refer to Copyright.txt, in Chombo's root directory.
 */
#endif
#ifndef _PROTO_SPMD_H_
#define _PROTO_SPMD_H_
#include <array>
#include "Proto.H"

#ifdef PR_MPI
#include "mpi.h"
#endif
// SPMD functions.
//When MPI is not defined, the following functions assume there is only one process.
using namespace std;
namespace Proto
{
#ifdef PR_MPI
  template <typename T=void>
  struct Proto_MPI
  {
    static MPI_Comm comm;
  };
  template <typename T>
  MPI_Comm Proto_MPI<T>::comm = MPI_COMM_WORLD;
#else
  // this can be changed for debugging parallel code in serial
  extern int num_procs;
#endif
#define PR_MAX_MPI_MESSAGE_SIZE 30*1024*1024  
  //extern unsigned long long PR_MaxMPISendSize;
  //extern unsigned long long PR_MaxMPIRecvSize;
#if 0
  inline int reportMPIStats()
  {
    pout()<<"Proto message size limit:"<< PR_MAX_MPI_MESSAGE_SIZE<<"\n"
          <<"Max send message size:"<<PR_MaxMPISendSize<<"\n"
          <<"Max recv message size:"<<PR_MaxMPIRecvSize<<std::endl;
  return 0;
  };
#endif

  inline int procID()
  {
#ifdef PR_MPI
    int retval;
    MPI_Comm_rank(Proto_MPI<void>::comm, &retval);
    return retval;
#else
    return 0;
#endif
  }

  inline int PRprocID()
  {
    return procID();
  }
  

  inline unsigned int numProc()
  {
#ifdef PR_MPI
    static int ret = -1;
    if (ret == -1)
      {
        MPI_Comm_size(Proto_MPI<void>::comm, &ret);
      }
    return ret;
#else
    return 1;
#endif
  }

  inline void barrier()
  {
#ifdef PR_MPI
    MPI_Barrier(Proto_MPI<>::comm);
#endif
  }

}
#endif