Workflow.h

 
#ifndef WRENCH_WORKFLOW_H
#define WRENCH_WORKFLOW_H
 
#include <map>
#include <set>
 
#include "wrench/workflow/execution_events/WorkflowExecutionEvent.h"
#include "WorkflowFile.h"
#include "WorkflowTask.h"
#include "DagOfTasks.h"
 
#include <boost/graph/adjacency_list.hpp>
#include <wrench/workflow/parallel_model/ParallelModel.h>
 
class WorkflowTask;
 
namespace wrench {
 
    class Simulation;
 
 
    class Workflow {
 
    public:
        Workflow();
 
        WorkflowTask *addTask(std::string, double flops,
                              unsigned long min_num_cores,
                              unsigned long max_num_cores,
                              double memory_requirement);
 
        void removeTask(WorkflowTask *task);
 
        void removeFile(WorkflowFile *file);
 
        WorkflowTask *getTaskByID(const std::string &id);
 
        WorkflowFile *addFile(std::string, double);
 
        WorkflowFile *getFileByID(const std::string &id);
 
        static double getSumFlops(const std::vector<WorkflowTask *> tasks);
 
        void addControlDependency(WorkflowTask *src, WorkflowTask *dest, bool redundant_dependencies = false);
        void removeControlDependency(WorkflowTask *src, WorkflowTask *dest);
 
        unsigned long getNumberOfTasks();
 
        unsigned long getNumLevels();
 
        double getCompletionDate();
 
        void exportToEPS(std::string);
 
        std::vector<WorkflowFile *> getFiles() const;
        std::map<std::string, WorkflowFile *> getFileMap() const;
        std::vector<WorkflowFile *> getInputFiles() const;
        std::map<std::string, WorkflowFile *> getInputFileMap() const;
        std::vector<WorkflowFile *> getOutputFiles() const;
        std::map<std::string, WorkflowFile *> getOutputFileMap() const;
 
        std::vector<WorkflowTask *> getTasks();
        std::map<std::string, WorkflowTask *> getTaskMap();
        std::map<std::string, WorkflowTask *> getEntryTaskMap() const;
        std::vector<WorkflowTask *> getEntryTasks() const;
        std::map<std::string, WorkflowTask *> getExitTaskMap() const;
        std::vector<WorkflowTask *> getExitTasks() const;
 
        std::vector<WorkflowTask *> getTaskParents(const WorkflowTask *task);
        long getTaskNumberOfParents(const  WorkflowTask *task);
        std::vector<WorkflowTask *> getTaskChildren(const WorkflowTask *task);
        long getTaskNumberOfChildren(const WorkflowTask *task);
 
        bool pathExists(const WorkflowTask *src, const WorkflowTask *dst);
 
 
        bool isDone();
 
        /***********************/
        /***********************/
 
        std::vector<WorkflowTask *> getTasksInTopLevelRange(unsigned long min, unsigned long max);
 
        std::vector<WorkflowTask *> getReadyTasks();
 
        std::map<std::string, std::vector<WorkflowTask *>> getReadyClusters();
 
        /***********************/
        /***********************/
 
 
        /***********************/
        /***********************/
        std::shared_ptr<WorkflowExecutionEvent> waitForNextExecutionEvent();
        std::shared_ptr<WorkflowExecutionEvent> waitForNextExecutionEvent(double timeout);
 
        std::string getCallbackMailbox();
 
//        void updateTaskState(WorkflowTask *task, WorkflowTask::State state);
 
        /***********************/
        /***********************/
 
    private:
        friend class WMS;
 
        friend class WorkflowTask;
 
        struct Vertex{ WorkflowTask *task;};
        typedef boost::adjacency_list<boost::listS, boost::vecS, boost::directedS, Vertex> DAG;
        typedef boost::graph_traits<DAG>::vertex_descriptor vertex_t;
 
        DagOfTasks dag;
 
        std::map<std::string, std::unique_ptr<WorkflowTask>> tasks;
        std::map<std::string, std::unique_ptr<WorkflowFile>> files;
 
 
        std::string callback_mailbox;
        ComputeService *parent_compute_service; // The compute service to which the job was submitted, if any
        Simulation *simulation; // a ptr to the simulation so that the simulation can obtain simulation timestamps for workflow tasks
    };
};
 
#endif //WRENCH_WORKFLOW_H