Add colors to trace categories.

[loba.git] / deployment.cpp

diff --git a/deployment.cpp b/deployment.cpp
index 29de927fa45566845a2ea899e311b0674e210973..705c1ea859539b528810cad95040ac55c9789ca2 100644 (file)
--- a/deployment.cpp
+++ b/deployment.cpp
@@ -1,5 +1,10 @@
+#include <algorithm>
+#include <cstdlib>
+#include <tr1/functional>

 #include <iomanip>
 #include <iomanip>

+#include <numeric>

 #include <sstream>
 #include <sstream>

+#include <vector>

 #include <msg/msg.h>
 #include <xbt/dynar.h>
 #include <xbt/log.h>
 #include <msg/msg.h>
 #include <xbt/dynar.h>
 #include <xbt/log.h>


@@ -7,6 +12,7 @@
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(depl);
 
 #include "hostdata.h"
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(depl);
 
 #include "hostdata.h"

+#include "misc.h"

 #include "options.h"
 
 #include "deployment.h"
 #include "options.h"
 
 #include "deployment.h"


@@ -16,6 +22,7 @@ void MY_launch_application()
     deployment_generator* gen;
     gen = opt::topologies.new_instance(opt::auto_depl::topology);
     gen->generate();
     deployment_generator* gen;
     gen = opt::topologies.new_instance(opt::auto_depl::topology);
     gen->generate();

+    gen->distribute_load();

     gen->deploy();
     delete gen;
 }
     gen->deploy();
     delete gen;
 }


@@ -23,7 +30,6 @@ void MY_launch_application()
 deployment_generator::deployment_generator()
     : hosts(opt::auto_depl::nhosts)
 {
 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)
 }
 
 void deployment_generator::set_load(int host, double load)


@@ -42,24 +48,43 @@ void deployment_generator::set_link(int host1, int host2)
     set_neighbor(host2, host1);
 }
 
     set_neighbor(host2, host1);
 }
 

+void deployment_generator::distribute_load()
+{
+    using std::tr1::bind;
+    using std::tr1::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(),
+                   bind(std::multiplies<double>(), _1, factor));
+    for (unsigned i = 0 ; i < hosts.size() ; ++i)
+        set_load(i, loads[i]);
+}
+

 void deployment_generator::deploy()
 {
 void deployment_generator::deploy()
 {

-    const char* func = "loba";

     xbt_dynar_t args = xbt_dynar_new(sizeof(const char*), NULL);
     for (unsigned i = 0 ; i < hosts.size() ; ++i) {
         const char* hostname = hostdata::at(i).get_name();
         std::ostringstream oss;
         oss << std::setprecision(12) << hosts[i].load;
         std::string strload = oss.str();
     xbt_dynar_t args = xbt_dynar_new(sizeof(const char*), NULL);
     for (unsigned i = 0 ; i < hosts.size() ; ++i) {
         const char* hostname = hostdata::at(i).get_name();
         std::ostringstream oss;
         oss << std::setprecision(12) << hosts[i].load;
         std::string strload = oss.str();

-        DEBUG2("%s/load -> \"%s\"", hostname, strload.c_str());
+        XBT_DEBUG("%s/load -> \"%s\"", hostname, strload.c_str());

         xbt_dynar_push_as(args, const char*, strload.c_str());
         for (unsigned j = 0 ; j < hosts[i].neighbors.size() ; ++j) {
             int neighbor = hosts[i].neighbors[j];
             const char* neighbor_name = hostdata::at(neighbor).get_name();
         xbt_dynar_push_as(args, const char*, strload.c_str());
         for (unsigned j = 0 ; j < hosts[i].neighbors.size() ; ++j) {
             int neighbor = hosts[i].neighbors[j];
             const char* neighbor_name = hostdata::at(neighbor).get_name();

-            DEBUG2("%s/neighbor -> \"%s\"", hostname, neighbor_name);
+            XBT_DEBUG("%s/neighbor -> \"%s\"", hostname, neighbor_name);

             xbt_dynar_push_as(args, const char*, neighbor_name);
         }
             xbt_dynar_push_as(args, const char*, neighbor_name);
         }

-        MSG_set_function(hostname, func, args);
+        MSG_set_function(hostname, "compute", args);

         xbt_dynar_reset(args);
     }
     xbt_dynar_free(&args);
         xbt_dynar_reset(args);
     }
     xbt_dynar_free(&args);


@@ -138,10 +163,9 @@ void deployment_torus::generate()
     // here width == ceil(sqrt(size))
 
     unsigned first_on_last_line = (size() - 1) - (size() - 1) % width;
     // here width == ceil(sqrt(size))
 
     unsigned first_on_last_line = (size() - 1) - (size() - 1) % width;

-    DEBUG4("torus size = %u ; width = %u ; height = %u ; foll = %u",
-           (unsigned )size(), width,
-           (unsigned )(size() / width + !!(size() % width)),
-           first_on_last_line);
+    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;
         unsigned prev_line;
     for (unsigned i = 0; i < size(); i++) {
         unsigned next_line;
         unsigned prev_line;