#ifndef COMMUNICATOR_H
#define COMMUNICATOR_H
-#include <list>
-#include <string>
-#include <msg/msg.h>
+#include <vector>
+#include <simgrid/msg.h>
+#include "hostdata.h"
+#include "messages.h"
+#include "msg_thread.h"
+#include "simgrid_features.h"
class communicator {
public:
communicator();
~communicator();
- void send(m_task_t task, const char *dest);
- void send(m_task_t task, const std::string& dest);
- m_task_t recv();
- int sent_count();
+ // Send a message to the "dest" mailbox
+ void ctrl_send(const char* dest, message* msg) {
+ ctrl_sent.push_back(real_send(dest, msg));
+ }
+ void data_send(const char* dest, message* msg) {
+ data_sent.push_back(real_send(dest, msg));
+ }
+
+ // Try to flush pending sending communications.
+ // If "wait" is true, blocks until success.
+ void ctrl_flush(bool wait) {
+ real_flush(ctrl_sent, wait);
+ }
+ void data_flush(bool wait) {
+ real_flush(data_sent, wait);
+ }
+
+ // Flush all sending_communications. Blocking.
+ void flush_all();
+
+ // Try to get a message. Returns true on success.
+ // Parameter "timeout" may be 0 for non-blocking operation, -1 for
+ // infinite waiting, or any positive timeout.
+ bool ctrl_recv(message*& msg, msg_host_t& from, double timeout) {
+ return ctrl_received.pop(msg, from, timeout);
+ }
+ bool data_recv(message*& msg, msg_host_t& from, double timeout) {
+ return data_received.pop(msg, from, timeout);
+ }
private:
- typedef std::list<msg_comm_t> comm_list;
- comm_list sent_comm;
- char* recv_mbox;
- msg_comm_t recv_comm;
- m_task_t recv_task;
+ // Myself
+ const hostdata* host;
+
+ // List of pending send communications
+ typedef std::vector<msg_comm_t> sent_comm_type;
+ sent_comm_type ctrl_sent;
+ sent_comm_type data_sent;
+
+ msg_comm_t real_send(const char* dest, message* msg);
+ void real_flush(sent_comm_type& sent_comm, bool wait);
+
+ // Queues of received messages
+ message_queue ctrl_received;
+ message_queue data_received;
+
+ // Handling of receiving thread
+ msg_thread* receiver_thread;
+ void receiver();
- void send_acknowledge();
+ // Used to chek if a communication is successfull before destroying it
+ static void comm_check_n_destroy(msg_comm_t comm);
+ // If comm is over, call comm_check_n_destroy(comm), and return true
+ static bool comm_test_n_destroy(msg_comm_t comm);
};
#endif // !COMMUNICATOR_H
