Add grids of 16, 64, and 256 nodes, based on g5k.xml

[loba.git] / deployment.cpp

diff --git a/deployment.cpp b/deployment.cpp
index 70bd8b3f15bf4c670703b04509d25f6e307eef8b..8f52806bb15fc591f28b6c5c6215466fb4770627 100644 (file)
--- a/deployment.cpp
+++ b/deployment.cpp
@@ -1,5 +1,10 @@
+#include <algorithm>
+#include <cstdlib>
+#include <functional>
 #include <iomanip>
+#include <numeric>
 #include <sstream>
+#include <vector>
 #include <msg/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,25 @@ void deployment_generator::set_link(int host1, int host2)
     set_neighbor(host2, host1);
 }
 
+void deployment_generator::distribute_load()
+{
+    using std::placeholders::_1;
+
+    if (!opt::auto_depl::random_distribution) {
+        set_load(0, opt::auto_depl::load);
+        return;
+    }
+    srand48(opt::auto_depl::random_seed);
+    std::vector<double> loads(hosts.size());
+    std::generate(loads.begin(), loads.end(), drand48);
+    double factor = opt::auto_depl::load /
+        std::accumulate(loads.begin(), loads.end(), 0.0);
+    std::transform(loads.begin(), loads.end(), loads.begin(),
+                   std::bind(std::multiplies<double>(), _1, factor));
+    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);