Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
prevent stings to be destroyed (dumb me)
[simgrid.git] / src / simdag / sd_global.cpp
index 4f55b83..6879f57 100644 (file)
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
-#include "instr/instr_interface.h"
 #include "simgrid/sg_config.h"
 #include "simgrid/host.h"
-#include "src/simdag/simdag_private.h"
 #include "src/surf/surf_interface.hpp"
 #include "simgrid/s4u/engine.hpp"
-
-#if HAVE_JEDULE
-#include "simgrid/jedule/jedule_sd_binding.h"
-#endif
+#include "simdag_private.hpp"
 
 XBT_LOG_NEW_CATEGORY(sd, "Logging specific to SimDag");
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(sd_kernel, sd, "Logging specific to SimDag (kernel)");
 
-SD_global_t sd_global = nullptr;
+simgrid::sd::Global *sd_global = nullptr;
+
+namespace simgrid{
+namespace sd{
+Global::Global(){
+  watch_point_reached = false;
+  initial_tasks = new std::set<SD_task_t>();
+  runnable_tasks = new std::set<SD_task_t>();
+  completed_tasks = new std::set<SD_task_t>();
+  return_set = new std::set<SD_task_t>();
+}
+
+Global::~Global(){
+  delete initial_tasks;
+  delete runnable_tasks;
+  delete completed_tasks;
+  delete return_set;
+}
+
+std::set<SD_task_t>* simulate(double how_long){
+  XBT_VERB("Run simulation for %f seconds", how_long);
+
+  sd_global->watch_point_reached = false;
+  sd_global->return_set->clear();
+
+  /* explore the runnable tasks */
+  while(!sd_global->runnable_tasks->empty())
+    SD_task_run(*(sd_global->runnable_tasks->begin()));
+
+  double elapsed_time = 0.0;
+  double total_time = 0.0;
+  /* main loop */
+  while (elapsed_time >= 0 && (how_long < 0 || 0.00001 < (how_long -total_time)) && !sd_global->watch_point_reached) {
+    surf_model_t model = nullptr;
+
+    XBT_DEBUG("Total time: %f", total_time);
+
+    elapsed_time = surf_solve(how_long > 0 ? surf_get_clock() + how_long - total_time: -1.0);
+    XBT_DEBUG("surf_solve() returns %f", elapsed_time);
+    if (elapsed_time > 0.0)
+      total_time += elapsed_time;
+
+    /* let's see which tasks are done */
+    unsigned int iter;
+    xbt_dynar_foreach(all_existing_models, iter, model) {
+      surf_action_t action = surf_model_extract_done_action_set(model);
+      while (action != nullptr) {
+        SD_task_t task = static_cast<SD_task_t>(action->getData());
+        XBT_VERB("Task '%s' done", SD_task_get_name(task));
+        SD_task_set_state(task, SD_DONE);
+
+        /* the state has changed. Add it only if it's the first change */
+        if (sd_global->return_set->find(task) == sd_global->return_set->end())
+          sd_global->return_set->insert(task);
+
+        /* remove the dependencies after this task */
+        for (auto succ : *task->successors) {
+          succ->predecessors->erase(task);
+          succ->inputs->erase(task);
+          XBT_DEBUG("Release dependency on %s: %zu remain(s). Becomes schedulable if %zu=0", SD_task_get_name(succ),
+              succ->predecessors->size()+succ->inputs->size(), succ->predecessors->size());
+
+          if (SD_task_get_state(succ) == SD_NOT_SCHEDULED && succ->predecessors->empty())
+            SD_task_set_state(succ, SD_SCHEDULABLE);
+
+          if (SD_task_get_state(succ) == SD_SCHEDULED && succ->predecessors->empty() && succ->inputs->empty())
+            SD_task_set_state(succ, SD_RUNNABLE);
+
+          if (SD_task_get_state(succ) == SD_RUNNABLE && !sd_global->watch_point_reached)
+            SD_task_run(succ);
+        }
+        task->successors->clear();
+
+        for (auto output : *task->outputs) {
+          output->start_time = task->finish_time;
+          output->predecessors->erase(task);
+          if (SD_task_get_state(output) == SD_SCHEDULED)
+             SD_task_set_state(output, SD_RUNNABLE);
+          else
+             SD_task_set_state(output, SD_SCHEDULABLE);
+
+          SD_task_t comm_dst = *(output->successors->begin());
+          if (SD_task_get_state(comm_dst) == SD_NOT_SCHEDULED && comm_dst->predecessors->empty()){
+            XBT_DEBUG("%s is a transfer, %s may be ready now if %zu=0",
+                SD_task_get_name(output), SD_task_get_name(comm_dst), comm_dst->predecessors->size());
+            SD_task_set_state(comm_dst, SD_SCHEDULABLE);
+          }
+          if (SD_task_get_state(output) == SD_RUNNABLE && !sd_global->watch_point_reached)
+            SD_task_run(output);
+        }
+        task->outputs->clear();
+        action = surf_model_extract_done_action_set(model);
+      }
+
+      /* let's see which tasks have just failed */
+      action = surf_model_extract_failed_action_set(model);
+      while (action != nullptr) {
+        SD_task_t task = static_cast<SD_task_t>(action->getData());
+        XBT_VERB("Task '%s' failed", SD_task_get_name(task));
+        SD_task_set_state(task, SD_FAILED);
+        sd_global->return_set->insert(task);
+        action = surf_model_extract_failed_action_set(model);
+      }
+    }
+  }
+
+  if (!sd_global->watch_point_reached && how_long < 0 && !sd_global->initial_tasks->empty()) {
+    XBT_WARN("Simulation is finished but %zu tasks are still not done", sd_global->initial_tasks->size());
+    for (auto t : *sd_global->initial_tasks)
+      XBT_WARN("%s is in %s state", SD_task_get_name(t), __get_state_name(SD_task_get_state(t)));
+  }
+
+  XBT_DEBUG("elapsed_time = %f, total_time = %f, watch_point_reached = %d",
+             elapsed_time, total_time, sd_global->watch_point_reached);
+  XBT_DEBUG("current time = %f", surf_get_clock());
+
+  return sd_global->return_set;
+}
+}
+}
+
+/**
+ * \brief helper for pretty printing of task state
+ * \param state the state of a task
+ * \return the equivalent as a readable string
+ */
+const char *__get_state_name(e_SD_task_state_t state){
+  static std::string state_names[7] =
+    { "not scheduled", "schedulable", "scheduled", "runnable","running", "done", "failed" };
+  return state_names[static_cast<int>(log2(static_cast<double>(state)))].data();
+}
 
 /**
  * \brief Initializes SD internal data
@@ -33,15 +158,7 @@ void SD_init(int *argc, char **argv)
 {
   xbt_assert(sd_global == nullptr, "SD_init() already called");
 
-  sd_global = xbt_new(s_SD_global_t, 1);
-  sd_global->watch_point_reached = false;
-
-  sd_global->task_mallocator=xbt_mallocator_new(65536, SD_task_new_f, SD_task_free_f, SD_task_recycle_f);
-
-  sd_global->initial_tasks = new std::set<SD_task_t>();
-  sd_global->runnable_tasks = new std::set<SD_task_t>();
-  sd_global->completed_tasks = new std::set<SD_task_t>();
-  sd_global->return_set = xbt_dynar_new(sizeof(SD_task_t), nullptr);
+  sd_global = new simgrid::sd::Global();
 
   surf_init(argc, argv);
 
@@ -90,7 +207,7 @@ void SD_create_environment(const char *platform_file)
 {
   simgrid::s4u::Engine::instance()->loadPlatform(platform_file);
 
-  XBT_DEBUG("Workstation number: %zu, link number: %d", sg_host_count(), sg_link_count());
+  XBT_DEBUG("Host number: %zu, link number: %d", sg_host_count(), sg_link_count());
 #if HAVE_JEDULE
   jedule_setup_platform();
 #endif
@@ -114,102 +231,11 @@ void SD_create_environment(const char *platform_file)
  */
 
 xbt_dynar_t SD_simulate(double how_long) {
-  XBT_VERB("Run simulation for %f seconds", how_long);
-
-  sd_global->watch_point_reached = false;
-  xbt_dynar_reset(sd_global->return_set);
-
-  /* explore the runnable tasks */
-  while(!sd_global->runnable_tasks->empty())
-    SD_task_run(*(sd_global->runnable_tasks->begin()));
-
-  double elapsed_time = 0.0;
-  double total_time = 0.0;
-  /* main loop */
-  while (elapsed_time >= 0 && (how_long < 0 || 0.00001 < (how_long -total_time)) && !sd_global->watch_point_reached) {
-    surf_model_t model = nullptr;
-
-    XBT_DEBUG("Total time: %f", total_time);
-
-    elapsed_time = surf_solve(how_long > 0 ? surf_get_clock() + how_long - total_time: -1.0);
-    XBT_DEBUG("surf_solve() returns %f", elapsed_time);
-    if (elapsed_time > 0.0)
-      total_time += elapsed_time;
-
-    /* let's see which tasks are done */
-    unsigned int iter;
-    xbt_dynar_foreach(all_existing_models, iter, model) {
-      surf_action_t action;
-      while ((action = surf_model_extract_done_action_set(model))) {
-        SD_task_t task = static_cast<SD_task_t>(action->getData());
-        XBT_VERB("Task '%s' done", SD_task_get_name(task));
-        SD_task_set_state(task, SD_DONE);
-
-        /* the state has changed. Add it only if it's the first change */
-        if (xbt_dynar_member(sd_global->return_set, &task) == 0)
-          xbt_dynar_push(sd_global->return_set, &task);
-
-        /* remove the dependencies after this task */
-        for (std::set<SD_task_t>::iterator succ = task->successors->begin(); succ != task->successors->end(); ++succ){
-          (*succ)->predecessors->erase(task);
-          (*succ)->inputs->erase(task);
-          XBT_DEBUG("Release dependency on %s: %zu remain(s). Becomes schedulable if %zu=0", SD_task_get_name((*succ)),
-              (*succ)->predecessors->size()+(*succ)->inputs->size(), (*succ)->predecessors->size());
-
-          if (SD_task_get_state((*succ)) == SD_NOT_SCHEDULED && (*succ)->predecessors->empty())
-            SD_task_set_state((*succ), SD_SCHEDULABLE);
-
-          if (SD_task_get_state((*succ)) == SD_SCHEDULED && (*succ)->predecessors->empty() && (*succ)->inputs->empty())
-            SD_task_set_state((*succ), SD_RUNNABLE);
-
-          if (SD_task_get_state((*succ)) == SD_RUNNABLE && !sd_global->watch_point_reached)
-            SD_task_run((*succ));
-        }
-        task->successors->clear();
-
-        for (std::set<SD_task_t>::iterator output=task->outputs->begin(); output!=task->outputs->end(); ++output){
-          (*output)->start_time = task->finish_time;
-          (*output)->predecessors->erase(task);
-          if (SD_task_get_state(*output) == SD_SCHEDULED)
-             SD_task_set_state(*output, SD_RUNNABLE);
-          else
-             SD_task_set_state(*output, SD_SCHEDULABLE);
-
-          SD_task_t comm_dst = *((*output)->successors->begin());
-          if (SD_task_get_state(comm_dst) == SD_NOT_SCHEDULED && comm_dst->predecessors->empty()){
-            XBT_DEBUG("%s is a transfer, %s may be ready now if %zu=0",
-                SD_task_get_name(*output), SD_task_get_name(comm_dst), comm_dst->predecessors->size());
-            SD_task_set_state(comm_dst, SD_SCHEDULABLE);
-          }
-          if (SD_task_get_state(*output) == SD_RUNNABLE && !sd_global->watch_point_reached)
-            SD_task_run(*output);
-        }
-        task->outputs->clear();
-      }
-
-      /* let's see which tasks have just failed */
-      while ((action = surf_model_extract_failed_action_set(model))) {
-        SD_task_t task = static_cast<SD_task_t>(action->getData());
-        XBT_VERB("Task '%s' failed", SD_task_get_name(task));
-        SD_task_set_state(task, SD_FAILED);
-        xbt_dynar_push(sd_global->return_set, &task);
-      }
-    }
-  }
-
-  if (!sd_global->watch_point_reached && how_long < 0 && !sd_global->initial_tasks->empty()) {
-    XBT_WARN("Simulation is finished but %zu tasks are still not done", sd_global->initial_tasks->size());
-    static const char* state_names[] =
-      { "SD_NOT_SCHEDULED", "SD_SCHEDULABLE", "SD_SCHEDULED", "SD_RUNNABLE", "SD_RUNNING", "SD_DONE","SD_FAILED" };
-    for (std::set<SD_task_t>::iterator t = sd_global->initial_tasks->begin(); t != sd_global->initial_tasks->end(); ++t)
-      XBT_WARN("%s is in %s state", SD_task_get_name(*t), state_names[SD_task_get_state(*t)]);
-  }
-
-  XBT_DEBUG("elapsed_time = %f, total_time = %f, watch_point_reached = %d",
-             elapsed_time, total_time, sd_global->watch_point_reached);
-  XBT_DEBUG("current time = %f", surf_get_clock());
-
-  return sd_global->return_set;
+  std::set<SD_task_t> *changed_tasks = simgrid::sd::simulate(how_long);
+  xbt_dynar_t changed_tasks_dynar = xbt_dynar_new(sizeof(SD_task_t), nullptr);
+  for (auto task: *changed_tasks)
+    xbt_dynar_push(changed_tasks_dynar, &task);
+  return changed_tasks_dynar;
 }
 
 /** @brief Returns the current clock, in seconds */
@@ -224,18 +250,9 @@ double SD_get_clock() {
  */
 void SD_exit()
 {
-  TRACE_surf_resource_utilization_release();
-
 #if HAVE_JEDULE
   jedule_sd_cleanup();
   jedule_sd_exit();
 #endif
-
-  xbt_mallocator_free(sd_global->task_mallocator);
-  delete sd_global->initial_tasks;
-  delete sd_global->runnable_tasks;
-  delete sd_global->completed_tasks;
-  xbt_dynar_free_container(&(sd_global->return_set));
-  xbt_free(sd_global);
-  sd_global = nullptr;
+  delete sd_global;
 }