Fix loba_simple.

[loba.git] / loba_simple.cpp

diff --git a/loba_simple.cpp b/loba_simple.cpp
index d44a1de27ad855bf4599de5ec291a1d2fa1e073f..5a0a153a706e46b7769d37eb8b06b5f93757c776 100644 (file)
--- a/loba_simple.cpp
+++ b/loba_simple.cpp
@@ -1,22 +1,22 @@
-#include "loba_simple.h"
-

 #include <xbt/log.h>
 
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(loba);
 
 #include <xbt/log.h>
 
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(loba);
 

+#include "loba_simple.h"
+

 /* simple version:
  *   load balance with a least-loaded neighbor,
  *   without breaking the ping-pong condition
  */
 /* simple version:
  *   load balance with a least-loaded neighbor,
  *   without breaking the ping-pong condition
  */

-double loba_simple::load_balance(double my_load)
+void loba_simple::load_balance()

 {
     int imin = -1;
     int imax = -1;
 {
     int imin = -1;
     int imax = -1;

-    double min = my_load;
+    double min = get_load();

     double max = -1.0;
     for (unsigned i = 0 ; i < pneigh.size() ; ++i) {
         double l = pneigh[i]->get_load();
     double max = -1.0;
     for (unsigned i = 0 ; i < pneigh.size() ; ++i) {
         double l = pneigh[i]->get_load();

-        if (l >= my_load)
+        if (l >= get_load())

             continue;
         if (l < min) {
             imin = i;
             continue;
         if (l < min) {
             imin = i;


@@ -27,15 +27,12 @@ double loba_simple::load_balance(double my_load)
             max = l;
         }
     }
             max = l;
         }
     }

-    if (imin != -1) {
-        // found someone
-        double balance = (my_load - min) / 2;
-        DEBUG6("%d:%g %d:%g %g %g", imin, min, imax, max, my_load, balance);
-        pneigh[imin]->set_to_send(balance);
-                               pneigh[imin]->set_load(min+balance);
-        return balance;
-    } else {
-        return 0.0;
+    if (imin != -1) {           // found someone
+        double balance = std::min((get_load() - min) / 2.0,
+                                  (get_load() - max));
+        XBT_DEBUG("%d:%g %d:%g %g %g",
+                  imin, min, imax, max, get_load(), balance);
+        send(pneigh[imin], balance);

     }
 }
 
     }
 }