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