Cosmetics...

[loba.git] / communicator.cpp

diff --git a/communicator.cpp b/communicator.cpp
index 0c64b5b9e22869026db3043083ed27f5deec0b32..83935155dfce1308c380ab567eb80c929698c0a7 100644 (file)
--- a/communicator.cpp
+++ b/communicator.cpp
@@ -1,15 +1,14 @@
+#include "communicator.h"
+
 #include <algorithm>
 #include <tr1/functional>
 #include <sstream>
 #include <msg/msg.h>
 #include <xbt/log.h>
-#include "communicator.h"
-#include "simgrid_features.h"
 #include "misc.h"
+#include "options.h"
 
-// XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simu);
-XBT_LOG_NEW_DEFAULT_SUBCATEGORY(comm, simu,
-                                "Messages from asynchronous pipes");
+XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(comm);
 
 std::string message::to_string()
 {
@@ -20,21 +19,18 @@ std::string message::to_string()
     return oss.str();
 }
 
+const int communicator::send_count_before_flush = 128;
+
 communicator::communicator()
+    : host((hostdata* )MSG_host_get_data(MSG_host_self()))
+    , send_counter(0)
+    , ctrl_task(NULL)
+    , ctrl_comm(NULL)
+    , ctrl_close_is_last(false)
+    , data_task(NULL)
+    , data_comm(NULL)
+    , data_close_is_last(false)
 {
-    const char* hostname = MSG_host_get_name(MSG_host_self());
-
-    ctrl_mbox = hostname;
-    ctrl_mbox += "_ctrl";
-    ctrl_task = NULL;
-    ctrl_comm = NULL;
-    ctrl_close_is_last = false;
-
-    data_mbox = hostname;
-    data_mbox += "_data";
-    data_task = NULL;
-    data_comm = NULL;
-    data_close_is_last = false;
 }
 
 communicator::~communicator()
@@ -56,12 +52,18 @@ void communicator::listen()
 
 void communicator::send(const char* dest, message* msg)
 {
+    DEBUG2("send %s to %s", msg->to_string().c_str(), dest);
     double msg_size = sizeof *msg;
     if (msg->get_type() == message::LOAD)
-        msg_size += msg->get_amount();
+        msg_size += opt::comm_cost(msg->get_amount());
     m_task_t task = MSG_task_create("message", 0.0, msg_size, msg);    
     msg_comm_t comm = MSG_task_isend(task, dest);
     sent_comm.push_back(comm);
+
+    if (++send_counter >= send_count_before_flush) {
+        flush(false);
+        send_counter = 0;
+    }
 }
 
 bool communicator::recv(message*& msg, m_host_t& from, bool wait)
@@ -103,18 +105,25 @@ bool communicator::recv(message*& msg, m_host_t& from, bool wait)
         }
     } while (restart);
 
+    if (msg)
+        DEBUG2("received %s from %s",
+               msg->to_string().c_str(), MSG_host_get_name(from));
+
     return msg != NULL;
 }
 
 void communicator::flush(bool wait)
 {
+    using namespace std::tr1;
+    using namespace std::tr1::placeholders;
+
     sent_comm.remove_if(comm_test_n_destroy);
     if (wait && !sent_comm.empty()) {
         xbt_dynar_t comms = xbt_dynar_new(sizeof(msg_comm_t), NULL);
         while (!sent_comm.empty()) {
             std::for_each(sent_comm.begin(), sent_comm.end(),
-                          std::tr1::bind(comm_push_in_dynar,
-                                         comms, std::tr1::placeholders::_1));
+                          bind(xbt_dynar_push,
+                               comms, bind(misc::address<msg_comm_t>(), _1)));
             MSG_comm_waitany(comms);
             xbt_dynar_reset(comms);
             sent_comm.remove_if(comm_test_n_destroy);
@@ -133,11 +142,6 @@ void communicator::next_close_on_data_is_last()
     data_close_is_last = true;
 }
 
-void communicator::comm_push_in_dynar(xbt_dynar_t dynar, msg_comm_t comm)
-{
-    xbt_dynar_push(dynar, &comm);
-}
-
 bool communicator::comm_test_n_destroy(msg_comm_t comm)
 {
     if (MSG_comm_test(comm)) {