+void process::send_all()
+{
+ using std::tr1::bind;
+ using std::tr1::placeholders::_1;
+
+ if (opt::bookkeeping) {
+ std::for_each(neigh.begin(), neigh.end(),
+ bind(&process::send1_bookkeeping, this, _1));
+ prev_load_broadcast = expected_load;
+ } else {
+ std::for_each(neigh.begin(), neigh.end(),
+ bind(&process::send1_no_bookkeeping, this, _1));
+ prev_load_broadcast = real_load;
+ }
+ comm.flush(false);
+}
+
+void process::receive(double timeout)
+{
+ message* msg;
+ m_host_t from;
+
+ XBT_DEBUG("%sblocking receive (%g)", "\0non-" + !timeout, timeout);
+ while (may_receive() && comm.recv(msg, from, timeout)) {
+ switch (msg->get_type()) {
+ case message::INFO: {
+ neighbor* n = rev_neigh[from];
+ n->set_load(msg->get_amount());
+ break;
+ }
+ case message::CREDIT:
+ expected_load += msg->get_amount();
+ break;
+ case message::LOAD: {
+ double ld = msg->get_amount();
+ real_load += ld;
+ if (finalizing)
+ total_load_running -= ld;
+ break;
+ }
+ case message::CTRL_CLOSE:
+ ctrl_close_pending--;
+ close_received = true;
+ break;
+ case message::DATA_CLOSE:
+ data_close_pending--;
+ close_received = true;
+ break;
+ }
+ delete msg;
+ timeout = 0.0; // only wait on first recv
+ }
+ comm.flush(false);
+}
+
+void process::finalize1(neighbor& nb)
+{
+ comm.send(nb.get_ctrl_mbox(), new message(message::CTRL_CLOSE, 0.0));
+ comm.send(nb.get_data_mbox(), new message(message::DATA_CLOSE, 0.0));
+}
+
+void process::finalize()
+{
+ using std::tr1::bind;
+ using std::tr1::placeholders::_1;
+
+ finalizing = true;
+ total_load_running -= real_load;
+
+ XBT_DEBUG("send CLOSE to %zu neighbor%s", neigh.size(), ESSE(neigh.size()));
+ std::for_each(neigh.begin(), neigh.end(),
+ bind(&process::finalize1, this, _1));
+
+ while (may_receive()) {
+ comm.flush(false);
+ XBT_DEBUG("waiting for %d CTRL and %d DATA CLOSE",
+ ctrl_close_pending, data_close_pending);
+ receive(-1.0);
+ }
+
+ comm.flush(true);
+}
+
+#define print_loads_generic(vec, verbose, logp, cat) \
+ if (_XBT_LOG_ISENABLEDV((*cat), logp)) { \
+ using std::tr1::bind; \
+ using std::tr1::placeholders::_1; \
+ XBT_XCLOG(cat, logp, "Neighbor loads:"); \
+ std::for_each(vec.begin(), vec.end(), \
+ bind(&neighbor::print, _1, verbose, logp, cat)); \
+ } else ((void)0)
+
+void process::print_loads(bool verbose,
+ e_xbt_log_priority_t logp,
+ xbt_log_category_t cat) const
+{
+ print_loads_generic(neigh, verbose, logp, cat);
+}
+
+void process::print_loads_p(bool verbose,
+ e_xbt_log_priority_t logp,
+ xbt_log_category_t cat) const
+{
+ print_loads_generic(pneigh, verbose, logp, cat);
+}
+
+#undef print_loads_generic
+
