Gabriele Cosmo
78 lines
1.7 KiB
Plaintext
78 lines
1.7 KiB
Plaintext
// Copyright (C) 2010, Guy Barrand. All rights reserved.
|
|
// See the file tools.license for terms.
|
|
|
|
#ifndef tools_mpi_world
|
|
#define tools_mpi_world
|
|
|
|
// code to wrap MPI global things by having an interface (inlib::mpi).
|
|
|
|
#include <tools/impi_world>
|
|
|
|
#ifdef TOOLS_USE_NATIVE_MPI
|
|
#include <mpi.h>
|
|
#else
|
|
#include "dummy_mpi.h"
|
|
#endif
|
|
|
|
#ifdef TOOLS_MEM
|
|
#include <tools/mem>
|
|
#endif
|
|
|
|
namespace tools {
|
|
namespace mpi {
|
|
|
|
class world : public virtual tools::impi_world {
|
|
typedef tools::impi_world parent;
|
|
#ifdef TOOLS_MEM
|
|
protected:
|
|
static const std::string& s_class() {
|
|
static const std::string s_v("tools::mpi::world");
|
|
return s_v;
|
|
}
|
|
#endif
|
|
public:
|
|
virtual bool init(int* a_argc,char*** a_argv) {
|
|
if(::MPI_Init(a_argc,a_argv)!=MPI_SUCCESS) return false;
|
|
return true;
|
|
}
|
|
virtual bool rank(int& a_rank) const {
|
|
if(::MPI_Comm_rank(MPI_COMM_WORLD,&a_rank)!=MPI_SUCCESS) {a_rank=-1;return false;}
|
|
return true;
|
|
}
|
|
virtual bool size(int& a_size) const {
|
|
if(::MPI_Comm_size(MPI_COMM_WORLD,&a_size)!=MPI_SUCCESS) {a_size=0;return false;}
|
|
return true;
|
|
}
|
|
virtual bool processor_name(std::string& a_s) const {
|
|
char name[MPI_MAX_PROCESSOR_NAME];
|
|
int lname;
|
|
if(::MPI_Get_processor_name(name,&lname)!=MPI_SUCCESS) {a_s.clear();return false;}
|
|
a_s = std::string(name);
|
|
return true;
|
|
}
|
|
public:
|
|
world() {
|
|
#ifdef TOOLS_MEM
|
|
tools::mem::increment(s_class().c_str());
|
|
#endif
|
|
}
|
|
virtual ~world(){
|
|
#ifdef TOOLS_MEM
|
|
tools::mem::decrement(s_class().c_str());
|
|
#endif
|
|
}
|
|
protected:
|
|
world(const world& a_from):parent(a_from) {
|
|
#ifdef TOOLS_MEM
|
|
tools::mem::increment(s_class().c_str());
|
|
#endif
|
|
}
|
|
world& operator=(const world&) {return *this;}
|
|
};
|
|
|
|
}}
|
|
|
|
//exlib_build_use inlib mpi
|
|
|
|
#endif
|