Add functions SD_workstation_get_computation_time, SD_workstation_route_get_communica...

[simgrid.git] / examples / simdag / sd_test.c

diff --git a/examples/simdag/sd_test.c b/examples/simdag/sd_test.c
index b2c435a..e5f6f15 100644 (file)
--- a/examples/simdag/sd_test.c
+++ b/examples/simdag/sd_test.c
@@ -8,12 +8,14 @@ XBT_LOG_NEW_DEFAULT_CATEGORY(sd_test,
                             "Logging specific to this SimDag example");
 
 int main(int argc, char **argv) {
+  int i;
   
   /* initialisation of SD */
   SD_init(&argc, argv);
 
-  /*  xbt_log_control_set("sd.thres=debug");*/
-
+/*   xbt_log_control_set("sd_test.thres=debug"); */
+/*   xbt_log_control_set("sd.thres=debug"); */
+  
   if (argc < 2) {
      INFO1("Usage: %s platform_file", argv[0]);
      INFO1("example: %s sd_platform.xml", argv[0]);
@@ -24,6 +26,34 @@ int main(int argc, char **argv) {
   char * platform_file = argv[1];
   SD_create_environment(platform_file);
 
+  /* test the estimation functions (use small_platform.xml) */
+  SD_workstation_t w1 = SD_workstation_get_by_name("Jupiter");
+  SD_workstation_t w2 = SD_workstation_get_by_name("Ginette");
+  const char *name1 = SD_workstation_get_name(w1);
+  const char *name2 = SD_workstation_get_name(w2);
+  const double computation_amount1 = 2000000;
+  const double computation_amount2 = 1000000;
+  const double communication_amount12 = 2000000;
+  const double communication_amount21 = 3000000;
+  INFO3("Computation time for %f flops on %s: %f", computation_amount1, name1,
+       SD_workstation_get_computation_time(w1, computation_amount1));
+  INFO3("Computation time for %f flops on %s: %f", computation_amount2, name2,
+       SD_workstation_get_computation_time(w2, computation_amount2));
+
+  INFO2("Route between %s and %s:", name1, name2);
+  SD_link_t *route = SD_workstation_route_get_list(w1, w2);
+  int route_size = SD_workstation_route_get_size(w1, w2);
+  for (i = 0; i < route_size; i++) {
+    INFO3("\tLink %s: latency = %f, bandwidth = %f", SD_link_get_name(route[i]),
+         SD_link_get_current_latency(route[i]), SD_link_get_current_bandwidth(route[i]));
+  }
+  INFO2("Route latency = %f, route bandwidth = %f", SD_workstation_route_get_latency(w1, w2),
+       SD_workstation_route_get_bandwidth(w1, w2));
+  INFO4("Communication time for %f bytes between %s and %s: %f", communication_amount12, name1, name2,
+       SD_workstation_route_get_communication_time(w1, w2, communication_amount12));
+  INFO4("Communication time for %f bytes between %s and %s: %f", communication_amount21, name2, name1,
+       SD_workstation_route_get_communication_time(w2, w1, communication_amount21));
+
   /* creation of the tasks and their dependencies */
   SD_task_t taskA = SD_task_create("Task A", NULL, 10.0);
   SD_task_t taskB = SD_task_create("Task B", NULL, 40.0);
@@ -34,19 +64,28 @@ int main(int argc, char **argv) {
   SD_task_dependency_add(NULL, NULL, taskC, taskA);
   SD_task_dependency_add(NULL, NULL, taskD, taskB);
   SD_task_dependency_add(NULL, NULL, taskD, taskC);
-  /*  SD_task_dependency_add(NULL, NULL, taskA, taskD); /\* deadlock */
+  /*  SD_task_dependency_add(NULL, NULL, taskA, taskD); /\* deadlock *\/ */
 
-  /* let's launch the simulation! */
+  /* scheduling parameters */
 
-  int workstation_number = 2;
-  const SD_workstation_t *workstation_list = SD_workstation_get_list();
-  double computation_amount[] = {100, 200};
+  const int workstation_number = 2;
+  /*  const SD_workstation_t *workstation_list = SD_workstation_get_list();*/
+  const SD_workstation_t workstation_list[] = {w1, w2};
+  double computation_amount[] = {computation_amount1, computation_amount2};
   double communication_amount[] =
     {
-      0, 30,
-      20, 0
+      0, communication_amount12,
+      communication_amount21, 0
     };
-  double rate = 1;
+  double rate = -1.0;
+
+  /* estimated time */
+  SD_task_t task = taskD;
+  INFO2("Estimated time for '%s': %f", SD_task_get_name(task),
+       SD_task_get_execution_time(task, workstation_number, workstation_list,
+                                  computation_amount, communication_amount, rate));
+
+  /* let's launch the simulation! */
 
   SD_task_schedule(taskA, workstation_number, workstation_list,
                   computation_amount, communication_amount, rate);
@@ -60,7 +99,6 @@ int main(int argc, char **argv) {
   SD_task_watch(taskC, SD_DONE);
 
   SD_task_t *changed_tasks;
-  int i;
 
   changed_tasks = SD_simulate(0.001);
   
@@ -96,10 +134,15 @@ int main(int argc, char **argv) {
 
   free(changed_tasks);
 
+  DEBUG0("Destroying tasks...");
+
   SD_task_destroy(taskA);
   SD_task_destroy(taskB);
   SD_task_destroy(taskC);
   SD_task_destroy(taskD);
+
+  DEBUG0("Tasks destroyed. Exiting SimDag...");
+
   SD_exit();
   return 0;
 }