loadBalanceSimulator.h

Go to the documentation of this file.

#ifndef included_AMP_loadBalanceSimulator
#define included_AMP_loadBalanceSimulator
 
#include "AMP/utils/Database.h"
 
 
namespace AMP::Mesh {
 
 
class loadBalanceSimulator
{
public:
    loadBalanceSimulator( std::shared_ptr<const AMP::Database> params );
 
    loadBalanceSimulator( double cost, int maxProc = 0, const std::string &name = "" );
 
    loadBalanceSimulator( const std::vector<loadBalanceSimulator> &meshes,
                          int method              = 2,
                          const std::string &name = "" );
 
    loadBalanceSimulator();
 
    size_t getMethod() const { return d_method; }
 
    inline int nRanks() const { return d_end - d_begin; }
 
    std::vector<int> getRanks() const;
 
    inline std::vector<int> getRanks( int i ) const { return d_submeshes[i].getRanks(); }
 
    inline int maxProcs() const { return d_max_procs; }
 
    void setProcs( int N_ranks );
 
    inline const auto &getSubmeshes() const { return d_submeshes; }
 
    int getMeshCount() const;
 
    double getCost() const { return d_cost; }
 
    std::vector<double> getRankCost() const;
 
    void print( uint8_t detail = 0, uint8_t indent = 0 );
 
 
private:                                           // Internal data
    std::string d_name;                            // Name of mesh
    double d_cost;                                 // Total cost of the mesh
    double d_maxCostRank;                          // Current maximum cost of any rank
    int d_max_procs;                               // Maximum number of allowed ranks
    int d_method;                                  // Load balance method (see MultiMesh)
    bool d_allEqual;                               // Is the cost of all meshes equal
    int d_begin;                                   // Begin assigned rank
    int d_end;                                     // End assigned rank
    std::vector<loadBalanceSimulator> d_submeshes; // list of sub-meshes
 
 
private: // Private functions
    void setRanks( int begin, int end );
    inline void setRanks( int rank ) { setRanks( rank, rank + 1 ); }
    void addRankCost( std::vector<double> &cost ) const;
    void addRank();
    void addRanks( int N );
    void copyRanks( const loadBalanceSimulator &x, int offset );
    void loadBalance( int, std::vector<int> &N );
};
 
 
} // namespace AMP::Mesh
 
#endif