-#include "loba_fairstrategy.h"
-
+#include <algorithm>
#include <xbt/log.h>
XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(loba);
+#include "loba_fairstrategy.h"
+
/* simple version:
* load balance with a least-loaded neighbor,
* without breaking the ping-pong condition
*/
+
+class compare {
+public:
+ bool operator()(const neighbor*a, const neighbor*b) {
+ return a->get_load() > b->get_load();
+ }
+};
+
double loba_fairstrategy::load_balance(double my_load)
{
- int imin = -1;
- int imax = -1;
- double min = my_load;
- double max = -1.0;
- for (unsigned i = 0 ; i < pneigh.size() ; ++i) {
- double l = pneigh[i]->get_load();
- if (l >= my_load)
- continue;
- if (l < min) {
- imin = i;
- min = l;
- }
- if (l > max) {
- imax = i;
- max = l;
+ std::sort(pneigh.begin(), pneigh.end(), compare());
+
+ print_loads_p();
+
+ double sum_sent = 0;
+ bool found = true;
+
+ while (found) {
+ found = false;
+ for (unsigned i = 0 ; i < pneigh.size() ; ++i) {
+ double l = pneigh[i]->get_load();
+ if (l >= my_load)
+ continue;
+ if (l < my_load + 2) {
+ found = true;
+ pneigh[i]->add_load(1);
+ pneigh[i]->add_to_send(1);
+ INFO1("sent to %s", pneigh[i]->get_name());
+ my_load--;
+ sum_sent++;
+ }
}
}
- if (imin != -1) {
- // found someone
- double balance = (my_load - max) / 2;
- DEBUG6("%d:%g %d:%g %g %g", imin, min, imax, max, my_load, balance);
- pneigh[imin]->set_to_send(balance);
- return balance;
- } else {
- return 0.0;
- }
+
+ return sum_sent;
}
// Local variables:
