CommTopology.h

Go to the documentation of this file.

#ifndef __COMM_TOPOLOGY__
#define __COMM_TOPOLOGY__

#include <vector>
#include "mpi.h"


class CommTopology {

public:

    CommTopology(MPI_Comm comm = MPI_COMM_WORLD) : comm_(comm) {
        MPI_Comm_rank(comm_, &rank_);
        MPI_Comm_size(comm_, &num_procs_);
    }

    virtual ~CommTopology()
    {}

    virtual void discover() = 0;

    int getRank() const { return rank_; }
    int getNP()   const { return num_procs_; }

    unsigned int getNumDimensions() const { return num_dims_; }
    unsigned int getCoreID()        const { return my_core_id_; }

    std::vector<unsigned int> getCoordinates() const { return coords_; }
    std::vector<unsigned int> getDimensions()  const { return dims_; }


private:

    MPI_Comm comm_;
    int rank_;
    int num_procs_;


protected:

    unsigned int hwID_;
    unsigned int num_dims_;
    unsigned int my_core_id_;

    std::vector<unsigned int> coords_;
    std::vector<unsigned int> dims_;

};

#endif