SimulationOutput.h

 
#ifndef WRENCH_SIMULATIONOUTPUT_H
#define WRENCH_SIMULATIONOUTPUT_H
 
#include <typeinfo>
#include <typeindex>
#include <iostream>
 
#include "wrench/simulation/SimulationTimestamp.h"
#include "wrench/simulation/SimulationTrace.h"
 
 
namespace wrench {
    class Simulation;
 
    class SimulationOutput {
 
    public:
        template<class T>
        std::vector<SimulationTimestamp<T> *> getTrace() {
            std::type_index type_index = std::type_index(typeid(T));
 
            // Is the trace empty?
            if (this->traces.find(type_index) == this->traces.end()) {
                return {};
            }
 
            std::vector<SimulationTimestamp<T> *> non_generic_vector;
            auto trace = (SimulationTrace<T> *) (this->traces[type_index]);
            for (auto ts : trace->getTrace()) {
                non_generic_vector.push_back((SimulationTimestamp<T> *) ts);
            }
            return non_generic_vector;
        }
 
        void dumpWorkflowExecutionJSON(Workflow *workflow, std::string file_path,
                                       bool generate_host_utilization_layout = false, bool writing_file = true);
 
        void dumpWorkflowGraphJSON(wrench::Workflow *workflow, std::string file_path, bool writing_file = true);
 
        void dumpHostEnergyConsumptionJSON(std::string file_path, bool writing_file = true);
 
        void dumpPlatformGraphJSON(std::string file_path, bool writing_file = true);
 
        void dumpDiskOperationsJSON(std::string file_path, bool writing_file = true);
 
        void dumpLinkUsageJSON(std::string file_path, bool writing_file = true);
 
        void dumpUnifiedJSON(Workflow *workflow, std::string file_path,
                             bool include_platform = false,
                             bool include_workflow_exec = true,
                             bool include_workflow_graph = false,
                             bool include_energy = false,
                             bool generate_host_utilization_layout = false,
                             bool include_disk = false,
                             bool include_bandwidth = false);
 
        void enableWorkflowTaskTimestamps(bool enabled);
 
        void enableFileReadWriteCopyTimestamps(bool enabled);
 
        void enableEnergyTimestamps(bool enabled);
 
        void enableDiskTimestamps(bool enabled);
 
        void enableBandwidthTimestamps(bool enabled);
 
        /***********************/
        /***********************/
 
        void addTimestampTaskStart(WorkflowTask *task);
 
        void addTimestampTaskFailure(WorkflowTask *task);
 
        void addTimestampTaskCompletion(WorkflowTask *task);
 
        void addTimestampTaskTermination(WorkflowTask *task);
 
        void addTimestampFileReadStart(WorkflowFile *file, FileLocation *src, StorageService *service,
                                       WorkflowTask *task = nullptr);
 
        void addTimestampFileReadFailure(WorkflowFile *file, FileLocation *src, StorageService *service,
                                         WorkflowTask *task = nullptr);
 
        void addTimestampFileReadCompletion(WorkflowFile *file, FileLocation *src, StorageService *service,
                                            WorkflowTask *task = nullptr);
 
        void addTimestampFileWriteStart(WorkflowFile *file, FileLocation *src, StorageService *service,
                                        WorkflowTask *task = nullptr);
 
        void addTimestampFileWriteFailure(WorkflowFile *file, FileLocation *src, StorageService *service,
                                          WorkflowTask *task = nullptr);
 
        void addTimestampFileWriteCompletion(WorkflowFile *file, FileLocation *src, StorageService *service,
                                             WorkflowTask *task = nullptr);
 
        void addTimestampFileCopyStart(WorkflowFile *file, std::shared_ptr<FileLocation> src,
                                       std::shared_ptr<FileLocation> dst);
 
        void addTimestampFileCopyFailure(WorkflowFile *file, std::shared_ptr<FileLocation> src,
                                         std::shared_ptr<FileLocation> dst);
 
        void addTimestampFileCopyCompletion(WorkflowFile *file, std::shared_ptr<FileLocation> src,
                                            std::shared_ptr<FileLocation> dst);
 
        void
        addTimestampDiskReadStart(std::string hostname, std::string mount, double bytes, int unique_sequence_number);
 
        void
        addTimestampDiskReadFailure(std::string hostname, std::string mount, double bytes, int unique_sequence_number);
 
        void addTimestampDiskReadCompletion(std::string hostname, std::string mount, double bytes,
                                            int unique_sequence_number);
 
        void
        addTimestampDiskWriteStart(std::string hostname, std::string mount, double bytes, int unique_sequence_number);
 
        void
        addTimestampDiskWriteFailure(std::string hostname, std::string mount, double bytes, int unique_sequence_number);
 
        void addTimestampDiskWriteCompletion(std::string hostname, std::string mount, double bytes,
                                             int unique_sequence_number);
 
        void addTimestampPstateSet(std::string hostname, int pstate);
 
        void addTimestampEnergyConsumption(std::string hostname, double joules);
 
        void addTimestampLinkUsage(std::string linkname, double bytes_per_second);
 
        template<class T>
        void addTimestamp(T *timestamp) {
            std::type_index type_index = std::type_index(typeid(T));
            if (this->traces.find(type_index) == this->traces.end()) {
                this->traces[type_index] = new SimulationTrace<T>();
            }
            ((SimulationTrace<T> *) (this->traces[type_index]))->addTimestamp(new SimulationTimestamp<T>(timestamp));
        }
 
        /***********************/
        /***********************/
 
        /***********************/
        /***********************/
 
        ~SimulationOutput();
 
        SimulationOutput();
 
        /***********************/
        /***********************/
 
    private:
        std::map<std::type_index, GenericSimulationTrace *> traces;
        nlohmann::json platform_json_part;
        nlohmann::json workflow_exec_json_part;
        nlohmann::json workflow_graph_json_part;
        nlohmann::json energy_json_part;
        nlohmann::json disk_json_part;
        nlohmann::json bandwidth_json_part;
 
        std::map<std::type_index, bool> enabledStatus;
 
        template<class T>
        bool isEnabled() {
            std::type_index type_index = std::type_index(typeid(T));
            return this->enabledStatus[type_index];
        }
 
        template<class T>
        void setEnabled(bool enabled) {
            std::type_index type_index = std::type_index(typeid(T));
            this->enabledStatus[type_index] = enabled;
        }
    };
};
 
#endif //WRENCH_SIMULATIONOUTPUT_H