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 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;
          SimulationTrace<T> *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);
        void dumpWorkflowGraphJSON(wrench::Workflow *workflow, std::string file_path);
        void dumpHostEnergyConsumptionJSON(std::string file_path);
        void dumpPlatformGraphJSON(std::string file_path);

        /***********************/
        /***********************/

        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() {
          for (auto t : this->traces) {
            delete t.second;
          }
          this->traces.clear();
        }
        /***********************/
        /***********************/

    private:
        std::map<std::type_index, GenericSimulationTrace*> traces;
    };

};


#endif //WRENCH_SIMULATIONOUTPUT_H