output.h

Go to the documentation of this file.

#pragma once
 
#include "fpmas/io/json_output.h"
#include "agent.h"
 
class BasicMetaModelFactory;
class BasicMetaModel;
 
 
class MetaModelCsvOutput :
    public fpmas::io::FileOutput,
    public fpmas::io::CsvOutput<
        fpmas::scheduler::Date, // Time Step
        unsigned int, // Partitioning time
        unsigned int, // Distribution time
        float, // Local Agents
        float, // Local cells
        float, // Distant agent->agent edges
        float, // Distant agent->cell edges
        float, // Distant cell->cell edges
        unsigned int, // LOCAL Cell->Cell read time
        unsigned int, // LOCAL Cell->Cell read count
        unsigned int, // LOCAL Cell->Cell write time
        unsigned int, // LOCAL Cell->Cell write count
        unsigned int, // DISTANT Cell->Cell read time
        unsigned int, // DISTANT Cell->Cell read count
        unsigned int, // DISTANT Cell->Cell write time
        unsigned int, // DISTANT Cell->Cell write count
        unsigned int // Sync time
    > {
        private:
            fpmas::scheduler::detail::LambdaTask commit_probes_task;
            fpmas::scheduler::detail::LambdaTask clear_monitor_task;
            fpmas::scheduler::Job commit_probes_job {{commit_probes_task}};
            fpmas::scheduler::Job clear_monitor_job {{clear_monitor_task}};
            fpmas::scheduler::JobList _jobs;
 
        public:
            MetaModelCsvOutput(
                    BasicMetaModel& meta_model,
                    fpmas::api::utils::perf::Probe& balance_probe,
                    fpmas::api::utils::perf::Probe& distribute_probe,
                    fpmas::api::utils::perf::Probe& local_read_probe,
                    fpmas::api::utils::perf::Probe& local_write_probe,
                    fpmas::api::utils::perf::Probe& distant_read_probe,
                    fpmas::api::utils::perf::Probe& distant_write_probe,
                    fpmas::api::utils::perf::Probe& sync_probe,
                    fpmas::api::utils::perf::Monitor& monitor
                    );
 
            const fpmas::scheduler::JobList& jobs() {
                return _jobs;
            }
    };
 
 
class CellsUtilityOutput : public fpmas::io::OutputBase {
    private:
        fpmas::io::FileOutput output_file;
        BasicMetaModel& model;
        std::size_t grid_width;
        std::size_t grid_height;
 
    public:
        CellsUtilityOutput(
                BasicMetaModel& model,
                std::size_t grid_width,
                std::size_t grid_height) :
            fpmas::io::OutputBase(output_file),
            output_file("grid.json"),
            model(model), grid_width(grid_width), grid_height(grid_height) {
        }
 
        void dump() override;
};
 
class CellsLocationOutput : public fpmas::io::OutputBase {
    private:
        fpmas::io::DynamicFileOutput output_file;
        BasicMetaModel& meta_model;
        std::size_t grid_width;
        std::size_t grid_height;
 
        std::vector<std::vector<int>> gather_cells();
 
    public:
        CellsLocationOutput(
                BasicMetaModel& meta_model,
                std::string filename,
                std::size_t grid_width, std::size_t grid_height
                );
 
        void dump() override;
};
 
class GraphStatsOutput : public fpmas::io::DistributedCsvOutput<
                         // Two local fields, dumped only on process 0
                         fpmas::io::Local<float>,
                         fpmas::io::Local<float>
                              > {
    private:
        fpmas::io::FileOutput output_file;
 
    public:
        GraphStatsOutput(
                BasicMetaModel& model,
                std::string file_name
                );
};
 
struct MetaAgentView {
    DistributedId id;
    std::deque<DistributedId> contacts;
    std::vector<DistributedId> perceptions;
 
    MetaAgentView(const MetaAgentBase* agent);
};
 
struct MetaGridAgentView : public MetaAgentView {
    DiscretePoint location;
 
    MetaGridAgentView(const MetaGridAgent* agent);
};
 
struct DistantAgentView {
    DistributedId id;
    int rank;
 
    DistantAgentView(const fpmas::api::model::Agent* agent);
};
 
struct AgentsOutputView {
    int rank;
    std::size_t grid_width;
    std::size_t grid_height;
    std::vector<MetaGridAgentView> agents;
    std::vector<DistantAgentView> distant_agents;
 
    AgentsOutputView(
            int rank, std::size_t grid_width, std::size_t grid_height,
            std::vector<MetaGridAgentView> agents,
            std::vector<DistantAgentView> distant_agents
            );
};
 
 
class AgentsOutput : public fpmas::io::JsonOutput<AgentsOutputView> {
    private:
        fpmas::io::DynamicFileOutput output_file;
 
    public:
        AgentsOutput(
                BasicMetaModel& model,
                std::size_t grid_width, std::size_t grid_height
                );
};
 
namespace nlohmann {
    template<>
        struct adl_serializer<MetaAgentView> {
            static void to_json(nlohmann::json& json, const MetaAgentView& agent);
        };
    template<>
        struct adl_serializer<MetaGridAgentView> {
            static void to_json(nlohmann::json& json, const MetaGridAgentView& agent);
        };
    template<>
        struct adl_serializer<DistantAgentView> {
            static void to_json(nlohmann::json& json, const DistantAgentView& agent);
        };
    template<>
        struct adl_serializer<AgentsOutputView> {
            static void to_json(nlohmann::json& json, const AgentsOutputView& agent_output);
        };
}