cell_load_balancing.h

Go to the documentation of this file.

#ifndef FPMAS_CELL_LB_H
#define FPMAS_CELL_LB_H
 
#include "fpmas/api/graph/load_balancing.h"
#include "fpmas/api/model/spatial/spatial_model.h"
 
namespace fpmas { namespace model {
 
    class CellLoadBalancing : public api::graph::LoadBalancing<api::model::AgentPtr> {
        private:
            api::communication::MpiCommunicator& comm;
            api::graph::LoadBalancing<api::model::AgentPtr>& cell_lb;
 
        public:
            CellLoadBalancing(
                    api::communication::MpiCommunicator& comm,
                    api::graph::LoadBalancing<api::model::AgentPtr>& cell_lb)
                : comm(comm), cell_lb(cell_lb) {
                }
 
 
            api::graph::PartitionMap balance(
                    api::graph::NodeMap<api::model::AgentPtr> nodes
                    ) override;
 
            api::graph::PartitionMap balance(
                    api::graph::NodeMap<api::model::AgentPtr> nodes,
                    api::graph::PartitionMode partition_mode
                    ) override;
    };
 
    class StaticCellLoadBalancing : public api::graph::LoadBalancing<api::model::AgentPtr> {
        private:
            api::communication::MpiCommunicator& comm;
            api::graph::LoadBalancing<api::model::AgentPtr>& cell_lb;
 
        public:
            StaticCellLoadBalancing(
                    api::communication::MpiCommunicator& comm,
                    api::graph::LoadBalancing<api::model::AgentPtr>& cell_lb)
                : comm(comm), cell_lb(cell_lb) {
                }
 
 
            api::graph::PartitionMap balance(
                    api::graph::NodeMap<api::model::AgentPtr> nodes
                    ) override;
 
            api::graph::PartitionMap balance(
                    api::graph::NodeMap<api::model::AgentPtr> nodes,
                    api::graph::PartitionMode partition_mode
                    ) override;
    };
 
}}
#endif