SplitStrategy.h

Go to the documentation of this file.

#ifndef __SPLIT_STRATEGY__
#define __SPLIT_STRATEGY__

#include "mpi.h"

#include "Util/CmdArguments.h"
#include "Util/OptPilotException.h"


enum commGroupColorings_t {Internal, ExternalToOpt, ExternalToWorker};

class SplitStrategy {

public:

    SplitStrategy(CmdArguments_t args,
                  MPI_Comm comm = MPI_COMM_WORLD)
        : comm_(comm)
        , cmd_args_(args)
        , role_(UNASSIGNED)
    {

        MPI_Comm_rank(comm, &rank_);
        MPI_Comm_rank(MPI_COMM_WORLD, &global_rank_);
        MPI_Comm_size(comm, &num_procs_);

        group_id_ = 0;

        if(num_procs_ < 3)
            throw OptPilotException("SplitStrategy::SplitStrategy",
                                    "We need 3 or more cores to split!");
    }


    virtual ~SplitStrategy()
    {}


    virtual void split() = 0;

    virtual int getNrWorkerGroups() const = 0;

    MPI_Comm getComm()  const { return comm_; }

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

    Role_t getRole()    const { return role_; }
    int getLeader()     const { return leader_; }
    int getPoller()     const { return poller_; }


private:

    MPI_Comm comm_;


protected:

    int rank_;
    int global_rank_;
    int num_procs_;
    int group_id_;

    CmdArguments_t cmd_args_;

    Role_t role_;

    std::vector<unsigned int> colorings_;

    int leader_;

    int poller_;
};

#endif