deployment.cpp

   1 #include <iomanip>
   2 #include <sstream>
   3 #include <msg/msg.h>
   4 #include <xbt/dynar.h>
   5 #include <xbt/log.h>
   6
   7 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(depl);
   8
   9 #include "hostdata.h"
  10 #include "options.h"
  11
  12 #include "deployment.h"
  13
  14 void MY_launch_application()
  15 {
  16     deployment_generator* gen;
  17     gen = opt::topologies.new_instance(opt::auto_depl::topology);
  18     gen->generate();
  19     gen->deploy();
  20     delete gen;
  21 }
  22
  23 deployment_generator::deployment_generator()
  24     : hosts(opt::auto_depl::nhosts)
  25 {
  26     set_load(0, opt::auto_depl::load);
  27 }
  28
  29 void deployment_generator::set_load(int host, double load)
  30 {
  31     hosts[host].load = load;
  32 }
  33
  34 void deployment_generator::set_neighbor(int host, int neighbor)
  35 {
  36     hosts[host].neighbors.push_back(neighbor);
  37 }
  38
  39 void deployment_generator::set_link(int host1, int host2)
  40 {
  41     set_neighbor(host1, host2);
  42     set_neighbor(host2, host1);
  43 }
  44
  45 void deployment_generator::deploy()
  46 {
  47     xbt_dynar_t args = xbt_dynar_new(sizeof(const char*), NULL);
  48     for (unsigned i = 0 ; i < hosts.size() ; ++i) {
  49         const char* hostname = hostdata::at(i).get_name();
  50         std::ostringstream oss;
  51         oss << std::setprecision(12) << hosts[i].load;
  52         std::string strload = oss.str();
  53         DEBUG2("%s/load -> \"%s\"", hostname, strload.c_str());
  54         xbt_dynar_push_as(args, const char*, strload.c_str());
  55         for (unsigned j = 0 ; j < hosts[i].neighbors.size() ; ++j) {
  56             int neighbor = hosts[i].neighbors[j];
  57             const char* neighbor_name = hostdata::at(neighbor).get_name();
  58             DEBUG2("%s/neighbor -> \"%s\"", hostname, neighbor_name);
  59             xbt_dynar_push_as(args, const char*, neighbor_name);
  60         }
  61         MSG_set_function(hostname, "loba", args);
  62         xbt_dynar_reset(args);
  63     }
  64     xbt_dynar_free(&args);
  65 }
  66
  67 void deployment_btree::generate()
  68 {
  69     for (unsigned i = 0 ; i < size() / 2 ; ++i) {
  70         unsigned left_child = 2 * i + 1;
  71         unsigned right_child = 2 * i + 2;
  72         if (left_child < size()) {
  73             set_link(i, left_child);
  74             if (right_child < size())
  75                 set_link(i, right_child);
  76         }
  77     }
  78 }
  79
  80 void deployment_clique::generate()
  81 {
  82     for (unsigned i = 0 ; i < size() ; ++i)
  83         for (unsigned j = 0 ; j < i ; ++j)
  84             set_link(i, j);
  85 }
  86
  87 void deployment_hcube::generate()
  88 {
  89     for (unsigned i = 0 ; i < size() ; ++i)
  90         for (unsigned j = 0 ; j < i ; ++j) {
  91             // Adapted from rom http://en.wikipedia.org/wiki/Hamming_distance
  92             unsigned dist = 0;
  93             unsigned val = i ^ j;
  94
  95             // Count the number of set bits
  96             while (val && dist < 2) {
  97                 ++dist;
  98                 val &= val - 1;
  99             }
 100             if (dist == 1)
 101                 set_link(i, j);
 102         }
 103 }
 104
 105 void deployment_line::generate()
 106 {
 107     for (unsigned i = 0 ; i < size() - 1 ; ++i)
 108         set_link(i, i + 1);
 109 }
 110
 111 void deployment_ring::generate()
 112 {
 113     set_neighbor(0, size() - 1);
 114     for (unsigned i = 0 ; i < size() - 1 ; ++i)
 115         set_link(i, i + 1);
 116     set_neighbor(size() - 1, 0);
 117 }
 118
 119 void deployment_star::generate()
 120 {
 121     for (unsigned i = 1 ; i < size() ; ++i)
 122         set_link(0, i);
 123 }
 124
 125 void deployment_torus::generate()
 126 {
 127     unsigned a = 0;
 128     unsigned b = size();
 129     while (a + 1 < b) {
 130         unsigned c = (a + b) / 2;
 131         if (c * c < size())
 132             a = c;
 133         else
 134             b = c;
 135     }
 136     unsigned width = b;
 137     // here width == ceil(sqrt(size))
 138
 139     unsigned first_on_last_line = (size() - 1) - (size() - 1) % width;
 140     DEBUG4("torus size = %u ; width = %u ; height = %u ; foll = %u",
 141            (unsigned )size(), width,
 142            (unsigned )(size() / width + !!(size() % width)),
 143            first_on_last_line);
 144     for (unsigned i = 0; i < size(); i++) {
 145         unsigned next_line;
 146         unsigned prev_line;
 147         unsigned next_column;
 148         unsigned prev_column;
 149
 150         next_line = i + width;
 151         if (next_line >= size())
 152             next_line %= width; // rewind
 153
 154         if (i >= width) {
 155             prev_line = i - width;
 156         } else {
 157             prev_line = first_on_last_line + i; // rewind
 158             if (prev_line >= size())
 159                 prev_line -= width; // need to go at last but one line
 160         }
 161
 162         if (i != size() - 1) {
 163             next_column = i + 1;
 164             if (next_column % width == 0)
 165                 next_column -= width; // rewind
 166         } else {
 167             next_column = first_on_last_line; // special case for last cell
 168         }
 169
 170         if (i % width != 0) {
 171             prev_column = i - 1;
 172         } else if (i < first_on_last_line) {
 173             prev_column = i + width - 1; // rewind
 174         } else {
 175             prev_column = size() - 1; // special case for 1st cell of last line
 176         }
 177         if (next_line != i) {
 178             set_neighbor(i, next_line);
 179             if (prev_line != next_line)
 180                 set_neighbor(i, prev_line);
 181         }
 182         if (next_column != i) {
 183             set_neighbor(i, next_column);
 184             if (prev_column != next_column)
 185                 set_neighbor(i, prev_column);
 186         }
 187     }
 188 }