StandardJob.h

#ifndef WRENCH_MULTITASKJOB_H
#define WRENCH_MULTITASKJOB_H


#include <vector>
#include <map>
#include <set>
#include <vector>

#include "WorkflowJob.h"

namespace wrench {

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

    class StorageService;

    class WorkflowFile;

    class WorkflowTask;

    class StandardJob : public WorkflowJob {

    public:
        enum State {
                    NOT_SUBMITTED,
                    PENDING,
                    RUNNING,
                    COMPLETED,
                    FAILED,
                    TERMINATED
        };

        std::vector<WorkflowTask *> getTasks();

        unsigned long getMinimumRequiredNumCores();

        unsigned long getNumCompletedTasks();

        unsigned long getNumTasks();

        StandardJob::State getState();

        std::map<WorkflowFile *, StorageService *> getFileLocations();

        // Tasks to run


        std::vector<WorkflowTask *> tasks;
        double total_flops;
        unsigned long num_completed_tasks;
        std::map<WorkflowFile *, StorageService *> file_locations;
        std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> pre_file_copies;

        std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> post_file_copies;
        std::set<std::tuple<WorkflowFile *, StorageService *>> cleanup_file_deletions;

    private:

        friend class StandardJobExecutor;
        friend class BareMetalComputeService;

        void incrementNumCompletedTasks();

        friend class JobManager;

        StandardJob(Workflow *workflow,
                    std::vector<WorkflowTask *> tasks, std::map<WorkflowFile *, StorageService *> &file_locations,
                    std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> &pre_file_copies,
                    std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> &post_file_copies,
                    std::set<std::tuple<WorkflowFile *, StorageService *>> &cleanup_file_deletions);

        State state;

    };

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

};

#endif //WRENCH_MULTITASKJOB_H