source: XIOS/dev/branch_yushan_merged/extern/remap/src/parallel_tree.hpp @ 1146

#ifndef  __PARALLEL_TREE_HPP__
#define  __PARALLEL_TREE_HPP__

#include "tree.hpp" // for local tree and routing tree
//#include "sample_tree.hpp"
#include "mpi_cascade.hpp"
#include "mpi.hpp"
#ifdef _usingEP
#include "ep_declaration.hpp"
#endif

namespace sphereRemap {

class CParallelTree
{
public:
        CParallelTree(ep_lib::MPI_Comm comm);
        ~CParallelTree();

        void build(vector<Node>& node, vector<Node>& node2);

        void routeNodes(vector<int>& route, vector<Node>& nodes, int level = 0);
        void routeIntersections(vector<vector<int> >& route, vector<Node>& nodes, int level = 0);

        int nbLocalElements;
        Elt* localElements;

        CTree localTree;

private:
        void updateCirclesForRouting(Coord rootCentre, double rootRadius, int level = 0);
        void buildSampleTreeCascade(vector<Node>& sampleNodes, int level = 0);
        void buildLocalTree(const vector<Node>& node, const vector<int>& route);
        void buildRouteTree();

        //CSampleTree sampleTree;
        vector<CSampleTree> treeCascade; // first for sample tree, then for routing tree
        CMPICascade cascade;
        ep_lib::MPI_Comm communicator ;

};

void buildSampleTree(CSampleTree& tree, const vector<Node>& node, const CCascadeLevel& comm);

}
#endif