StandardJob.h

#ifndef WRENCH_MULTITASKJOB_H
#define WRENCH_MULTITASKJOB_H


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

#include "WorkflowJob.h"

#include "wrench/services/storage/StorageService.h"

namespace wrench {

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

    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 *, std::shared_ptr<FileLocation>> getFileLocations();

        unsigned long getPriority();

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

        std::set<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  , std::shared_ptr<FileLocation>  >> post_file_copies;
        std::set<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  >> 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 *, std::shared_ptr<FileLocation>  > &file_locations,
                    std::set<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  , std::shared_ptr<FileLocation>  >> &pre_file_copies,
                    std::set<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  , std::shared_ptr<FileLocation>  >> &post_file_copies,
                    std::set<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  >> &cleanup_file_deletions);

        State state;

    };

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

};

#endif //WRENCH_MULTITASKJOB_H