WMS.h

#ifndef WRENCH_WMS_H
#define WRENCH_WMS_H

#include <wrench/managers/EnergyMeter.h>
#include "wrench/simgrid_S4U_util/S4U_Daemon.h"
#include "wrench/services/Service.h"
#include "wrench/wms/DynamicOptimization.h"
#include "wrench/wms/StaticOptimization.h"
#include "wrench/wms/scheduler/PilotJobScheduler.h"
#include "wrench/wms/scheduler/StandardJobScheduler.h"
#include "wrench/workflow/Workflow.h"

namespace wrench {

    class Simulation;
    class ComputeService;
    class StorageService;
    class NetworkProximityService;
    class FileRegistryService;

    class WMS : public Service {

    public:
        void addWorkflow(Workflow *workflow, double start_time = 0);
        Workflow *getWorkflow();
        PilotJobScheduler *getPilotJobScheduler();
        StandardJobScheduler *getStandardJobScheduler();

        void addStaticOptimization(std::unique_ptr<StaticOptimization>);

        void addDynamicOptimization(std::unique_ptr<DynamicOptimization>);


    protected:

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

        WMS(std::unique_ptr<StandardJobScheduler> standard_job_scheduler,
            std::unique_ptr<PilotJobScheduler> pilot_job_scheduler,
            const std::set<ComputeService *> &compute_services,
            const std::set<StorageService *> &storage_services,
            const std::set<NetworkProximityService *> &network_proximity_services,
            FileRegistryService *file_registry_service,
            const std::string &hostname,
            const std::string suffix);


        void checkDeferredStart();

        std::shared_ptr<JobManager> createJobManager();
        std::shared_ptr<DataMovementManager> createDataMovementManager();
        std::shared_ptr<EnergyMeter> createEnergyMeter(const std::map<std::string, double> &measurement_periods);
        std::shared_ptr<EnergyMeter> createEnergyMeter(const std::vector<std::string> &hostnames, double measurement_period);

        void runDynamicOptimizations();

        void runStaticOptimizations();

        std::set<ComputeService *> getAvailableComputeServices();
        std::set<StorageService *> getAvailableStorageServices();
        std::set<NetworkProximityService *> getAvailableNetworkProximityServices();
        FileRegistryService * getAvailableFileRegistryService();

        void waitForAndProcessNextEvent();

        virtual void processEventStandardJobCompletion(std::unique_ptr<StandardJobCompletedEvent>);

        virtual void processEventStandardJobFailure(std::unique_ptr<StandardJobFailedEvent>);

        virtual void processEventPilotJobStart(std::unique_ptr<PilotJobStartedEvent>);

        virtual void processEventPilotJobExpiration(std::unique_ptr<PilotJobExpiredEvent>);

        virtual void processEventFileCopyCompletion(std::unique_ptr<FileCopyCompletedEvent>);

        virtual void processEventFileCopyFailure(std::unique_ptr<FileCopyFailedEvent>);

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

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

    private:
        friend class Simulation;
        friend class DataMovementManager;
        friend class JobManager;

        Workflow *workflow;
        double start_time;
        std::set<ComputeService *> compute_services;
        std::set<StorageService *> storage_services;
        std::set<NetworkProximityService *> network_proximity_services;
        FileRegistryService * file_registry_service;

        std::unique_ptr<StandardJobScheduler> standard_job_scheduler;
        std::unique_ptr<PilotJobScheduler> pilot_job_scheduler;

        std::vector<std::unique_ptr<DynamicOptimization>> dynamic_optimizations;
        std::vector<std::unique_ptr<StaticOptimization>> static_optimizations;

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

    private:
        virtual int main() = 0;

    };
};


#endif //WRENCH_WMS_H