communicator.h

   1 // Asynchronous communicator
   2
   3 #ifndef COMMUNICATOR_H
   4 #define COMMUNICATOR_H
   5
   6 #include <list>
   7 #include <string>
   8 #include <msg/msg.h>
   9
  10 class message {
  11 public:
  12     enum message_type { INFO, CREDIT, LOAD, CTRL_CLOSE, DATA_CLOSE };
  13
  14     message(message_type t, double a): type(t), amount(a) { }
  15
  16     message_type get_type() const       { return type;   }
  17     double get_amount() const           { return amount; }
  18
  19     std::string to_string();
  20
  21 private:
  22     message_type type;
  23     double amount;
  24 };
  25
  26 class communicator {
  27 public:
  28     communicator();
  29     ~communicator();
  30
  31     void listen();
  32
  33     void send(const char* dest, message* msg);
  34     bool recv(message*& msg, m_host_t& from, bool wait);
  35     void flush(bool wait);
  36
  37     void next_close_on_ctrl_is_last();
  38     void next_close_on_data_is_last();
  39
  40 private:
  41     // List of pending send communications
  42     std::list<msg_comm_t> sent_comm;
  43
  44     // Control channel for receiving
  45     std::string ctrl_mbox;
  46     msg_comm_t  ctrl_comm;
  47     m_task_t    ctrl_task;
  48     bool        ctrl_close_is_last;
  49
  50     // Data channel for receiving
  51     std::string data_mbox;
  52     msg_comm_t  data_comm;
  53     m_task_t    data_task;
  54     bool        data_close_is_last;
  55
  56     const char* get_ctrl_mbox() const   { return ctrl_mbox.c_str(); }
  57     const char* get_data_mbox() const   { return data_mbox.c_str(); }
  58
  59     static void comm_push_in_dynar(xbt_dynar_t dynar, msg_comm_t comm);
  60     static bool comm_test_n_destroy(msg_comm_t comm);
  61 };
  62
  63 #endif // !COMMUNICATOR_H
  64
  65 // Local variables:
  66 // mode: c++
  67 // End: