Workflow.h

 
#ifndef WRENCH_WORKFLOW_H
#define WRENCH_WORKFLOW_H
 
#include <map>
#include <set>
 
#include "wrench/execution_events/ExecutionEvent.h"
#include "wrench/data_file/DataFile.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 std::enable_shared_from_this<Workflow>  {
 
    public:
 
 
        static std::shared_ptr<Workflow> createWorkflow();
        void clear();
 
        std::shared_ptr<Workflow> getSharedPtr() { return this->shared_from_this(); }
 
 
        std::shared_ptr<WorkflowTask> addTask(std::string, double flops,
                              unsigned long min_num_cores,
                              unsigned long max_num_cores,
                              double memory_requirement);
 
        void removeTask(std::shared_ptr<WorkflowTask>task);
 
        void removeFile(std::shared_ptr<DataFile>file);
        std::map<std::string, std::shared_ptr<DataFile>> &getFileMap();
        std::shared_ptr<DataFile> addFile(std::string id, double size);
        std::shared_ptr<DataFile> getFileByID(const std::string &id);
        std::shared_ptr<WorkflowTask> getTaskByID(const std::string &id);
 
 
        static double getSumFlops(const std::vector<std::shared_ptr<WorkflowTask>> tasks);
 
        void addControlDependency(std::shared_ptr<WorkflowTask>src, std::shared_ptr<WorkflowTask>dest, bool redundant_dependencies = false);
        void removeControlDependency(std::shared_ptr<WorkflowTask>src, std::shared_ptr<WorkflowTask>dest);
 
        unsigned long getNumberOfTasks();
 
        unsigned long getNumLevels();
 
        double getCompletionDate();
 
        void exportToEPS(std::string);
 
        std::vector<std::shared_ptr<DataFile>> getInputFiles() const;
        std::map<std::string, std::shared_ptr<DataFile>> getInputFileMap() const;
        std::vector<std::shared_ptr<DataFile>> getOutputFiles() const;
        std::map<std::string, std::shared_ptr<DataFile>> getOutputFileMap() const;
 
        std::vector<std::shared_ptr<WorkflowTask>> getTasks();
        std::map<std::string, std::shared_ptr<WorkflowTask>> getTaskMap();
        std::map<std::string, std::shared_ptr<WorkflowTask>> getEntryTaskMap() const;
        std::vector<std::shared_ptr<WorkflowTask>> getEntryTasks() const;
        std::map<std::string, std::shared_ptr<WorkflowTask>> getExitTaskMap() const;
        std::vector<std::shared_ptr<WorkflowTask>> getExitTasks() const;
 
        std::vector<std::shared_ptr<WorkflowTask>> getTaskParents(const std::shared_ptr<WorkflowTask>task);
        long getTaskNumberOfParents(const  std::shared_ptr<WorkflowTask>task);
        std::vector<std::shared_ptr<WorkflowTask>> getTaskChildren(const std::shared_ptr<WorkflowTask>task);
        long getTaskNumberOfChildren(const std::shared_ptr<WorkflowTask>task);
 
        bool pathExists(const std::shared_ptr<WorkflowTask>src, const std::shared_ptr<WorkflowTask>dst);
 
        std::shared_ptr<WorkflowTask> getTaskThatOutputs(std::shared_ptr<DataFile> file);
        bool isFileOutputOfSomeTask(std::shared_ptr<DataFile> file);
 
        std::set<std::shared_ptr<WorkflowTask>> getTasksThatInput(std::shared_ptr<DataFile> file);
        bool isDone();
 
        /***********************/
        /***********************/
 
        std::vector<std::shared_ptr<WorkflowTask>> getTasksInTopLevelRange(unsigned long min, unsigned long max);
 
        std::vector<std::shared_ptr<WorkflowTask>> getReadyTasks();
 
        std::map<std::string, std::vector<std::shared_ptr<WorkflowTask>>> getReadyClusters();
 
        /***********************/
        /***********************/
 
    private:
        friend class WMS;
        friend class Simulation;
        friend class WorkflowTask;
 
        Workflow();
 
        struct Vertex{ std::shared_ptr<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;
 
        /* Map to find tasks by name */
        std::map<std::string, std::shared_ptr<WorkflowTask>> tasks;
 
        /* Set of ready tasks */
        std::set<std::shared_ptr<WorkflowTask>> ready_tasks;
 
        /* Map of output files */
        std::map<std::shared_ptr<DataFile>, std::shared_ptr<WorkflowTask>> task_output_files;
        std::map<std::shared_ptr<DataFile>, std::set<std::shared_ptr<WorkflowTask>>> task_input_files;
 
        /* Set of files (also kept track by the simulation!) */
        std::set<std::shared_ptr<DataFile>> data_files;
 
    };
};
 
#endif //WRENCH_WORKFLOW_H