Enforce integer initial load when integer transfers are activated.

[loba.git] / deployment.cpp

diff --git a/deployment.cpp b/deployment.cpp
index 705c1ea859539b528810cad95040ac55c9789ca2..2befb00bb9b7af1d506fcc6c07178490fea8885f 100644 (file)
--- a/deployment.cpp
+++ b/deployment.cpp
@@ -1,6 +1,6 @@
 #include <algorithm>
 #include <cstdlib>
-#include <tr1/functional>
+#include <functional>
 #include <iomanip>
 #include <numeric>
 #include <sstream>
@@ -50,8 +50,7 @@ void deployment_generator::set_link(int host1, int host2)
 
 void deployment_generator::distribute_load()
 {
-    using std::tr1::bind;
-    using std::tr1::placeholders::_1;
+    using std::placeholders::_1;
 
     if (!opt::auto_depl::random_distribution) {
         set_load(0, opt::auto_depl::load);
@@ -63,9 +62,36 @@ void deployment_generator::distribute_load()
     double factor = opt::auto_depl::load /
         std::accumulate(loads.begin(), loads.end(), 0.0);
     std::transform(loads.begin(), loads.end(), loads.begin(),
-                   bind(std::multiplies<double>(), _1, factor));
-    for (unsigned i = 0 ; i < hosts.size() ; ++i)
-        set_load(i, loads[i]);
+                   std::bind(std::multiplies<double>(), _1, factor));
+    if (opt::integer_transfer) {
+        // round the loads
+        std::vector<double> iloads(hosts.size());
+        std::transform(loads.begin(), loads.end(), iloads.begin(), round);
+        // compute the differences between each load and its rounded value
+        std::vector<double> diffs(hosts.size());
+        std::transform(loads.begin(), loads.end(), iloads.begin(),
+                       diffs.begin(), std::minus<double>());
+        // compute the absolute values of the diffs
+        std::vector<double> adiffs(hosts.size());
+        std::transform(diffs.begin(), diffs.end(), adiffs.begin(), fabs);
+        // find i, index of the element farthest from its rounded value
+        unsigned i;
+        i = std::max_element(adiffs.begin(), adiffs.end()) - adiffs.begin();
+        // remove element i from diffs, and compute the residual part...
+        diffs[i] = diffs.back();
+        diffs.pop_back();
+        double residue = std::accumulate(diffs.begin(), diffs.end(), 0.0);
+        // ... and compute element i (rounded to avoid numerical errors)
+        iloads[i] = fabs(round(loads[i] + residue));
+        // final sanity check
+        xbt_assert(opt::auto_depl::load ==
+                   std::accumulate(iloads.begin(), iloads.end(), 0.0));
+        for (unsigned i = 0 ; i < hosts.size() ; ++i)
+            set_load(i, iloads[i]);
+    } else {
+        for (unsigned i = 0 ; i < hosts.size() ; ++i)
+            set_load(i, loads[i]);
+    }
 }
 
 void deployment_generator::deploy()