probe.h

Go to the documentation of this file.

#pragma once
 
#include "fpmas.h"
#include "fpmas/utils/perf.h"
 
class LoadBalancingProbe : public fpmas::api::model::LoadBalancing {
    private:
        // Probes only the load balancing algorithm execution time
        fpmas::api::utils::perf::Probe& lb_algorithm_probe;
        fpmas::api::model::LoadBalancing& lb;
 
    public:
        
        LoadBalancingProbe(
                fpmas::api::utils::perf::Probe& lb_algorithm_probe,
                fpmas::api::model::LoadBalancing& lb
                ) :
            lb_algorithm_probe(lb_algorithm_probe),
            lb(lb) {
        }
 
        fpmas::graph::PartitionMap balance(
                fpmas::graph::NodeMap<fpmas::model::AgentPtr> node_map,
                fpmas::api::graph::PartitionMode mode) override;
 
        fpmas::graph::PartitionMap balance(
                fpmas::graph::NodeMap<fpmas::model::AgentPtr> node_map
                ) override {
            return this->balance(node_map, fpmas::api::graph::PARTITION);
        }
};
 
class GraphBalanceProbe : public fpmas::scheduler::Task {
    private:
        // Wrapper for the existing load balancing algorithm, that adds probes
        LoadBalancingProbe load_balancing_algorithm;
        fpmas::model::detail::LoadBalancingTask lb_task;
        // Probes the complete load balancing + distribution process
        fpmas::api::utils::perf::Probe& graph_balance_probe;
 
    public:
        fpmas::scheduler::Job job;
 
        GraphBalanceProbe(
                fpmas::api::model::AgentGraph& graph,
                fpmas::api::model::LoadBalancing& probed_lb,
                fpmas::api::utils::perf::Probe& lb_algorithm_probe,
                fpmas::api::utils::perf::Probe& graph_balance_probe
                ) : 
            load_balancing_algorithm(lb_algorithm_probe, probed_lb),
            // The FPMAS LoadBalancingTask will call the
            // DistributedGraph::balance() method, that implicitly call this
            // class balance() methods, but also the distribution process,
            // implicitly feeding the probes 
            lb_task(graph, load_balancing_algorithm),
            graph_balance_probe(graph_balance_probe),
            // The job only contains this task
            job({*this}) {
        }
 
        void run() override;
 
};
 
class SyncProbeTask : public fpmas::scheduler::Task {
    private:
        fpmas::api::utils::perf::Probe& sync_probe;
        fpmas::model::detail::SynchronizeGraphTask sync_task;
 
    public:
        SyncProbeTask(
                fpmas::api::utils::perf::Probe& sync_probe,
                fpmas::api::model::AgentGraph& graph);
 
        void run() override;
};