Whitespace cleanup.

[loba.git] / process.cpp

diff --git a/process.cpp b/process.cpp
index 47dd94a19d1030db12c4e55c1cd9683ff1560176..193c87b45bece83a183d446c7fec2280c2f4e748 100644 (file)
--- a/process.cpp
+++ b/process.cpp
@@ -1,4 +1,5 @@
 #include <algorithm>
+#include <cmath>
 #include <functional>
 #include <iterator>
 #include <numeric>
@@ -303,26 +304,26 @@ void process::ctrl_send(neighbor& nb)
     }
 }
 
+double process::compute_load_to_send(double desired)
+{
+    if (opt::integer_transfer)
+        desired = std::floor(desired);
+    return desired >= opt::min_transfer_amount ? desired : 0.0;
+}
+
 void process::data_send(neighbor& nb)
 {
     double load_to_send;
     if (opt::bookkeeping) {
-        load_to_send = std::min(real_load, nb.get_debt());
-        if (load_to_send >= opt::min_transfer_amount) {
+        load_to_send = compute_load_to_send(std::min(real_load, nb.get_debt()));
+        if (load_to_send > 0.0)
             nb.set_debt(nb.get_debt() - load_to_send);
-            real_load -= load_to_send;
-        } else {
-            load_to_send = 0.0;
-        }
     } else {
-        load_to_send = nb.get_to_send();
-        if (load_to_send >= opt::min_transfer_amount) {
-            nb.set_to_send(0.0);
-            real_load -= load_to_send;
-        } else {
-            load_to_send = 0.0;
-        }
+        load_to_send = compute_load_to_send(nb.get_to_send());
+        if (load_to_send > 0.0)
+            nb.set_to_send(nb.get_to_send() - load_to_send);
     }
+    real_load -= load_to_send;
     while (load_to_send > 0.0) {
         double amount;
         if (opt::max_transfer_amount)