parallelPartition.hh

#include <array>
 
#ifndef PARALLEL_PARTITION__HH
#define PARALLEL_PARTITION__HH
 
namespace Dune{
    namespace Composites{
 
        template<int d>
            class YaspPartition : public Dune::YLoadBalance<d>
        {
            private:
                int numProc ; // number of processes
            public:
                YaspPartition(int num_proc_,std::array<int, 3> myDim_) : numProc(num_proc_), myDim(myDim_) {};
                typedef std::array<int, d> iTupel;
                iTupel myDim;
 
                virtual ~YaspPartition() {}
 
                virtual void loadbalance (const iTupel& size, int P, iTupel& dims) const
                {
                    dims = myDim;
 
                    if(dims[0]*dims[1]*dims[2]!=numProc){
 
                        if(int(sqrt(numProc))*int(sqrt(numProc)) == numProc){
                            dims[0] = sqrt(numProc);
                            dims[1] = sqrt(numProc);
                        }
                        else{
                            if(int(numProc/8)*8 == numProc){
                                dims[0] = 8;
                                dims[1] = int(numProc/8);
                                dims[2] = 1;
                            }
                            else{
                                dims[0] = numProc;
                            }
                        }
                    }
                }
        };
 
    }
}
 
#endif