SpreadOutHierarchically.h

Go to the documentation of this file.

// This file is part of the Peano project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org
#pragma once
 
 
 
#include "tarch/logging/Log.h"
 
#include "peano4/parallel/Node.h"
 
#include <map>
 
#include "toolbox/loadbalancing/AbstractLoadBalancing.h"
 
#include "toolbox/loadbalancing/metrics/CellCount.h"
 
 
namespace toolbox {
  namespace loadbalancing {
    namespace strategies {
      class SpreadOutHierarchically;
    }
  }
}
 
 
class toolbox::loadbalancing::strategies::SpreadOutHierarchically: public toolbox::loadbalancing::AbstractLoadBalancing {
  public:
    enum class Action {
      Unspecified,
      None,
      SpreadEquallyOverAllRanks,
      SpreadEquallyOverAllThreads
    };
 
    static std::string toString( Action action );
 
    SpreadOutHierarchically(Configuration* configuration = new DefaultConfiguration(), CostMetrics*   costMetrics = new toolbox::loadbalancing::metrics::CellCount());
    virtual ~SpreadOutHierarchically();
 
    virtual void finishStep() override;
 
  private:
    static tarch::logging::Log  _log;
 
    int _stepsToWaitForNextLoadBalancingDecision;
 
    void updateState();
    void updateLoadBalancing();
    void triggerSplit( int numberOfCells, int targetRank );
    Action getAction() const;
    int getNumberOfSplitsOnLocalRank() const;
};