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 Workunit {

    public:

        Workunit(StandardJob *job,
                 std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> pre_file_copies,
                 WorkflowTask * task,
                 std::map<WorkflowFile *, StorageService *> file_locations,
                 std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> post_file_copies,
                 std::set<std::tuple<WorkflowFile *, StorageService *>> cleanup_file_deletions);

        static void addDependency(Workunit *parent, Workunit *child);

        static std::set<std::unique_ptr<Workunit>> createWorkunits(StandardJob *job);

        StandardJob *getJob();

        StandardJob *job;

        std::set<Workunit*> children;
        unsigned long num_pending_parents;

        std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> pre_file_copies;
        WorkflowTask *task = nullptr;
        std::map<WorkflowFile *, StorageService *> file_locations;
        std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> post_file_copies;
        std::set<std::tuple<WorkflowFile *, StorageService *>> cleanup_file_deletions;


        ~Workunit();
    };

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

};


#endif //WRENCH_WORKUNIT_H