src/simdag/simdag_private.hpp

   1 /* Copyright (c) 2006-2021. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include "simgrid/s4u/Engine.hpp"
   7 #include "simgrid/simdag.h"
   8 #include "surf/surf.hpp"
   9 #include <set>
  10 #include <string>
  11 #include <vector>
  12
  13 #ifndef SIMDAG_PRIVATE_HPP
  14 #define SIMDAG_PRIVATE_HPP
  15 #if SIMGRID_HAVE_JEDULE
  16 #include "simgrid/jedule/jedule_sd_binding.h"
  17 #endif
  18
  19 namespace simgrid{
  20 namespace sd{
  21 class Global {
  22 public:
  23   explicit Global(int* argc, char** argv) : engine_(new simgrid::s4u::Engine(argc, argv)) {}
  24   bool watch_point_reached = false; /* has a task just reached a watch point? */
  25   std::set<SD_task_t> initial_tasks;
  26   std::set<SD_task_t> runnable_tasks;
  27   std::set<SD_task_t> completed_tasks;
  28   std::set<SD_task_t> return_set;
  29   s4u::Engine* engine_;
  30 };
  31
  32 std::set<SD_task_t>* simulate (double how_long);
  33 }
  34 }
  35
  36 extern XBT_PRIVATE std::unique_ptr<simgrid::sd::Global> sd_global;
  37
  38 /* Task */
  39 struct s_SD_task_t {
  40   e_SD_task_state_t state;
  41   void *data;                   /* user data */
  42   char *name;
  43   e_SD_task_kind_t kind;
  44   double amount;
  45   double alpha;          /* used by typed parallel tasks */
  46   double start_time;
  47   double finish_time;
  48   simgrid::kernel::resource::Action* surf_action;
  49   unsigned short watch_points;  /* bit field xor()ed with masks */
  50
  51   bool marked = false; /* used to check if the task DAG has some cycle*/
  52
  53   /* dependencies -- cannot be embedded in the struct since it's not handled as a real C++ class */
  54   std::set<SD_task_t> *inputs;
  55   std::set<SD_task_t> *outputs;
  56   std::set<SD_task_t> *predecessors;
  57   std::set<SD_task_t> *successors;
  58
  59   /* scheduling parameters (only exist in state SD_SCHEDULED) */
  60   std::vector<sg_host_t> *allocation;
  61   double *flops_amount;
  62   double *bytes_amount;
  63   double rate;
  64 };
  65
  66 /* SimDag private functions */
  67 XBT_PRIVATE void SD_task_set_state(SD_task_t task, e_SD_task_state_t new_state);
  68 XBT_PRIVATE void SD_task_run(SD_task_t task);
  69 XBT_PRIVATE bool acyclic_graph_detail(const_xbt_dynar_t dag);
  70 XBT_PRIVATE void uniq_transfer_task_name(SD_task_t task);
  71 XBT_PRIVATE const char *__get_state_name(e_SD_task_state_t state);
  72 #endif