StandardJob.h

 
#ifndef WRENCH_STANDARDJOB_H
#define WRENCH_STANDARDJOB_H
 
 
#include <map>
#include <set>
#include <vector>
 
#include "WorkflowJob.h"
 
#include "wrench/services/storage/storage_helpers/FileLocation.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 getMinimumRequiredMemory();
 
        unsigned long getNumCompletedTasks();
 
        unsigned long getNumTasks();
 
        StandardJob::State getState();
 
        std::map<WorkflowFile *, std::vector<std::shared_ptr<FileLocation>>> getFileLocations();
 
        unsigned long getPriority();
 
 
        std::vector<WorkflowTask *> tasks;
 
        double total_flops;
        unsigned long num_completed_tasks;
 
        std::map<WorkflowFile *, std::vector<std::shared_ptr<FileLocation>>> file_locations;
 
        std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  , std::shared_ptr<FileLocation>  >> pre_file_copies;
        std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  , std::shared_ptr<FileLocation>  >> post_file_copies;
        std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  >> cleanup_file_deletions;
 
        /***********************/
        /***********************/
 
        /***********************/
        /***********************/
 
        double getPreJobOverheadInSeconds();
        void setPreJobOverheadInSeconds(double overhead);
        double getPostJobOverheadInSeconds();
        void setPostJobOverheadInSeconds(double overhead);
 
 
    private:
 
        friend class StandardJobExecutor;
        friend class BareMetalComputeService;
        void incrementNumCompletedTasks();
 
        friend class JobManager;
 
        StandardJob(Workflow *workflow,
                    std::vector<WorkflowTask *> tasks, std::map<WorkflowFile *, std::vector<std::shared_ptr<FileLocation>>> &file_locations,
                    std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  , std::shared_ptr<FileLocation>>> &pre_file_copies,
                    std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  , std::shared_ptr<FileLocation>>> &post_file_copies,
                    std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>  >> &cleanup_file_deletions);
 
        State state;
        double pre_overhead = 0.0;
        double post_overhead = 0.0;
 
    };
 
    /***********************/
    /***********************/
 
};
 
#endif //WRENCH_MULTITASKJOB_H