SimulationTimestampTypes.h

#ifndef WRENCH_SIMULATIONTIMESTAMPTYPES_H
#define WRENCH_SIMULATIONTIMESTAMPTYPES_H

#include "wrench/workflow/WorkflowTask.h"

namespace wrench {

    class WorkflowTask;
    class StorageService;

    class SimulationTimestampType {
    public:
        SimulationTimestampType();
        double getDate();

    private:
        double date = -0.1;
    };

    class SimulationTimestampPair : public SimulationTimestampType {
    public:
        /***********************/
        /***********************/
        SimulationTimestampPair();
        SimulationTimestampPair(SimulationTimestampPair *endpoint);
        virtual ~SimulationTimestampPair() {}
        /***********************/
        /***********************/

        virtual SimulationTimestampPair *getEndpoint();

    protected:
        SimulationTimestampPair *endpoint = nullptr;
    };

    class SimulationTimestampTask : public SimulationTimestampPair {

    public:

        /***********************/
        /***********************/
        SimulationTimestampTask(WorkflowTask *);
        /***********************/
        /***********************/

        WorkflowTask *getTask();
        SimulationTimestampTask *getEndpoint() override;

    protected:
        static std::map<std::string, SimulationTimestampTask *> pending_task_timestamps;

        void setEndpoints();

    private:
        WorkflowTask *task;
    };

    class SimulationTimestampTaskStart : public SimulationTimestampTask {
    public:
        /***********************/
        /***********************/
        SimulationTimestampTaskStart(WorkflowTask *);
        /***********************/
        /***********************/
    };

    class SimulationTimestampTaskFailure : public SimulationTimestampTask {
    public:
        /***********************/
        /***********************/
        SimulationTimestampTaskFailure(WorkflowTask *);
        /***********************/
        /***********************/

    };

    class SimulationTimestampTaskCompletion : public SimulationTimestampTask {
    public:
        /***********************/
        /***********************/
        SimulationTimestampTaskCompletion(WorkflowTask *);
        /***********************/
        /***********************/
    };

     class SimulationTimestampTaskTermination : public SimulationTimestampTask {
     public:
         /***********************/
         /***********************/
         SimulationTimestampTaskTermination(WorkflowTask *);
         /***********************/
         /***********************/
     };

    class SimulationTimestampFileCopyStart;
    class FileLocation;

    class SimulationTimestampFileCopy : public SimulationTimestampPair {
    public:

        /***********************/
        /***********************/
        SimulationTimestampFileCopy(WorkflowFile *file, std::shared_ptr<FileLocation>  src, std::shared_ptr<FileLocation> dst, SimulationTimestampFileCopyStart *start_timestamp = nullptr);
        /***********************/
        /***********************/

        SimulationTimestampFileCopy *getEndpoint() override;
        WorkflowFile *getFile();
        std::shared_ptr<FileLocation> getSource();
        std::shared_ptr<FileLocation> getDestination();

    protected:
        WorkflowFile *file;

        std::shared_ptr<FileLocation> source;

        std::shared_ptr<FileLocation> destination;
    };

    class SimulationTimestampFileCopyFailure;
    class SimulationTimestampFileCopyCompletion;

    class SimulationTimestampFileCopyStart : public SimulationTimestampFileCopy {
    public:
        /***********************/
        /***********************/
        SimulationTimestampFileCopyStart(WorkflowFile *file, std::shared_ptr<FileLocation> src, std::shared_ptr<FileLocation> dst);
        /***********************/
        /***********************/

        friend class SimulationTimestampFileCopyFailure;
        friend class SimulationTimestampFileCopyCompletion;
    };

    class SimulationTimestampFileCopyFailure : public SimulationTimestampFileCopy {
    public:
        /***********************/
        /***********************/
        SimulationTimestampFileCopyFailure(SimulationTimestampFileCopyStart *start_timestamp);
        /***********************/
        /***********************/
    };

    class SimulationTimestampFileCopyCompletion : public SimulationTimestampFileCopy {
    public:
        /***********************/
        /***********************/
        SimulationTimestampFileCopyCompletion(SimulationTimestampFileCopyStart *start_timestamp);
        /***********************/
        /***********************/
    };




    class SimulationTimestampPstateSet : public SimulationTimestampType {
    public:
        /***********************/
        /***********************/
        SimulationTimestampPstateSet(std::string hostname, int pstate);
        /***********************/
        /***********************/

        std::string getHostname();
        int getPstate();
    private:
        std::string hostname;
        int pstate;
    };

    class SimulationTimestampEnergyConsumption: public SimulationTimestampType {
    public:
        /***********************/
        /***********************/
        SimulationTimestampEnergyConsumption(std::string hostname, double joules);
        /***********************/
        /***********************/

        std::string getHostname();
        double getConsumption();

    private:
        std::string hostname;
        double joules;
    };
};

#endif //WRENCH_SIMULATIONTIMESTAMPTYPES_H