source: XIOS/trunk/extern/remap/src/mpi_cascade.hpp @ 1638

#ifndef  __MPI_CASCADE_HPP__
#define  __MPI_CASCADE_HPP__

#include "mpi.hpp"
#include <vector>
#include <cassert>
#include "misc.hpp"

namespace sphereRemap {

class CCascadeLevel
{
public:
        CCascadeLevel(ep_lib::MPI_Comm comm) : comm(comm)
        {
                ep_lib::MPI_Comm_size(comm, &size);
                ep_lib::MPI_Comm_rank(comm, &rank);
        }
        int colour() const { return rank % group_size; };
        int key() const { return p_colour() + rank/(p_grp_size*group_size)*p_grp_size; }

        // perpendicular group
        int p_colour() const { return (rank%group_size + rank/group_size) % p_grp_size; }
        int p_key() const { return colour() + rank/(p_grp_size*group_size)*group_size; }

        ep_lib::MPI_Comm comm, pg_comm;
        int rank;
        int size;
        int group_size; // group_size and p_grp_size are interchanged?? FIXME
        int p_grp_size;
};

class CMPICascade
{
public:
        // 
        CMPICascade(int nodes_per_level, ep_lib::MPI_Comm comm);

        int num_levels;
        std::vector<CCascadeLevel> level;
};

}
#endif