Convert clock_type to enum class.

[loba.git] / deployment.cpp

diff --git a/deployment.cpp b/deployment.cpp
index 5659db655f4ba75780f74fc21d2d3dbbff9df50d..287ca4f88fb2432a5c7f710b52ed2dc7606bb379 100644 (file)
--- a/deployment.cpp
+++ b/deployment.cpp
@@ -1,6 +1,11 @@
+#include <algorithm>
+#include <cmath>
+#include <cstdlib>
 #include <iomanip>
+#include <numeric>
 #include <sstream>
-#include <msg/msg.h>
+#include <vector>
+#include <simgrid/msg.h>
 #include <xbt/dynar.h>
 #include <xbt/log.h>
 
@@ -17,6 +22,7 @@ void MY_launch_application()
     deployment_generator* gen;
     gen = opt::topologies.new_instance(opt::auto_depl::topology);
     gen->generate();
+    gen->distribute_load();
     gen->deploy();
     delete gen;
 }
@@ -24,7 +30,6 @@ void MY_launch_application()
 deployment_generator::deployment_generator()
     : hosts(opt::auto_depl::nhosts)
 {
-    set_load(0, opt::auto_depl::load);
 }
 
 void deployment_generator::set_load(int host, double load)
@@ -43,6 +48,49 @@ void deployment_generator::set_link(int host1, int host2)
     set_neighbor(host2, host1);
 }
 
+void deployment_generator::distribute_load()
+{
+    if (!opt::auto_depl::random_distribution) {
+        set_load(0, opt::auto_depl::load);
+        return;
+    }
+    std::vector<double> loads(hosts.size());
+    if (opt::auto_depl::random_seed != 1) {
+        srand48(opt::auto_depl::random_seed);
+        std::generate(loads.begin(), loads.end(), drand48);
+    } else {
+        std::fill(loads.begin(), loads.end(), 1.0);
+    }
+    double factor = opt::auto_depl::load /
+        std::accumulate(loads.begin(), loads.end(), 0.0);
+    std::transform(loads.begin(), loads.end(), loads.begin(),
+                   [&factor](double a) { return factor * a; });
+    if (opt::integer_transfer) {
+        double iload;
+        double residue = 0.0;
+        unsigned i;
+        for (i = 0 ; i < hosts.size() - 1; ++i) {
+            if (residue < 0.0)
+                iload = std::floor(loads[i]);
+            else if (residue > 0.0)
+                iload = std::ceil(loads[i]);
+            else // residue == 0.0
+                iload = std::round(loads[i]);
+            residue += (loads[i] - iload);
+            loads[i] = iload;
+        }
+        // abs(round(...)) to avoid rounding errors, or a value of -0
+        // Note: i == hosts.size() - 1
+        iload = std::abs(std::round(loads[i] + residue));
+        loads[i] = iload;
+        // final sanity check
+        xbt_assert(opt::auto_depl::load ==
+                   std::accumulate(loads.begin(), loads.end(), 0.0));
+    }
+    for (unsigned i = 0 ; i < hosts.size() ; ++i)
+        set_load(i, loads[i]);
+}
+
 void deployment_generator::deploy()
 {
     xbt_dynar_t args = xbt_dynar_new(sizeof(const char*), NULL);
@@ -59,7 +107,7 @@ void deployment_generator::deploy()
             XBT_DEBUG("%s/neighbor -> \"%s\"", hostname, neighbor_name);
             xbt_dynar_push_as(args, const char*, neighbor_name);
         }
-        MSG_set_function(hostname, "loba", args);
+        MSG_set_function(hostname, "compute", args);
         xbt_dynar_reset(args);
     }
     xbt_dynar_free(&args);
@@ -135,12 +183,11 @@ void deployment_torus::generate()
             b = c;
     }
     unsigned width = b;
-    // here width == ceil(sqrt(size))
+    // here width == std::ceil(std::sqrt(size))
 
     unsigned first_on_last_line = (size() - 1) - (size() - 1) % width;
-    XBT_DEBUG("torus size = %u ; width = %u ; height = %u ; foll = %u",
-              (unsigned )size(), width,
-              (unsigned )(size() / width + !!(size() % width)),
+    XBT_DEBUG("torus size = %zu ; width = %u ; height = %zu ; foll = %u",
+              size(), width, size() / width + !!(size() % width),
               first_on_last_line);
     for (unsigned i = 0; i < size(); i++) {
         unsigned next_line;