Workunit.h

 
#ifndef WRENCH_WORKUNIT_H
#define WRENCH_WORKUNIT_H
 
 
#include <tuple>
#include <set>
#include <map>
#include <vector>
#include <memory>
 
namespace wrench {
 
    class StandardJob;
    class WorkflowFile;
    class StorageService;
    class WorkflowTask;
    class FileLocation;
 
    /***********************/
    /***********************/
 
    class Workunit {
 
    public:
 
        Workunit(
                std::shared_ptr<StandardJob> job,
                double sleep_time,
                std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>, std::shared_ptr<FileLocation>>> pre_file_copies,
                WorkflowTask *task,
                std::map<WorkflowFile *, std::vector<std::shared_ptr<FileLocation>>> file_locations,
                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);
 
        static void addDependency(std::shared_ptr<Workunit> parent, std::shared_ptr<Workunit> child);
 
        static std::set<std::shared_ptr<Workunit>> createWorkunits(std::shared_ptr<StandardJob> job);
 
        std::shared_ptr<StandardJob> getJob();
 
        std::shared_ptr<StandardJob> job;
 
        std::set<std::shared_ptr<Workunit>> children;
        unsigned long num_pending_parents;
 
        double sleep_time;
        std::vector<std::tuple<WorkflowFile *, std::shared_ptr<FileLocation>, std::shared_ptr<FileLocation>>> pre_file_copies;
        WorkflowTask *task = nullptr;
        std::map<WorkflowFile *, std::vector<std::shared_ptr<FileLocation>>> file_locations;
        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;
 
 
        ~Workunit();
    };
 
    /***********************/
    /***********************/
 
};
 
 
#endif //WRENCH_WORKUNIT_H