Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
b37d9ff045bd1ef6aa96381a3c70bd9b4583f12c
[simgrid.git] / src / s4u / s4u_mailbox.cpp
1 /* Copyright (c) 2006-2017. The SimGrid Team. All rights reserved.          */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #include "simgrid/s4u/Comm.hpp"
7 #include "simgrid/s4u/Mailbox.hpp"
8 #include "src/msg/msg_private.h"
9 #include "src/simix/ActorImpl.hpp"
10 #include "src/simix/smx_network_private.h"
11 #include "xbt/log.h"
12
13 XBT_LOG_EXTERNAL_CATEGORY(s4u);
14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(s4u_channel,s4u,"S4U Communication Mailboxes");
15
16 namespace simgrid {
17 namespace s4u {
18
19 const char* Mailbox::getName()
20 {
21   return pimpl_->name_;
22 }
23
24 MailboxPtr Mailbox::byName(const char*name)
25 {
26   kernel::activity::MailboxImpl* mbox = kernel::activity::MailboxImpl::byNameOrNull(name);
27   if (mbox == nullptr) {
28     mbox = simix::kernelImmediate([name] {
29       return kernel::activity::MailboxImpl::byNameOrCreate(name);
30     });
31   }
32   return MailboxPtr(&mbox->piface_, true);
33 }
34
35 MailboxPtr Mailbox::byName(std::string name)
36 {
37   return byName(name.c_str());
38 }
39
40 bool Mailbox::empty()
41 {
42   return pimpl_->comm_queue.empty();
43 }
44
45 bool Mailbox::listen()
46 {
47   return not this->empty() || (pimpl_->permanent_receiver && not pimpl_->done_comm_queue.empty());
48 }
49
50 smx_activity_t Mailbox::front()
51 {
52   return pimpl_->comm_queue.empty() ? nullptr : pimpl_->comm_queue.front();
53 }
54
55 void Mailbox::setReceiver(ActorPtr actor) {
56   simix::kernelImmediate([this, actor]() {
57     this->pimpl_->setReceiver(actor);
58   });
59 }
60
61 /** @brief get the receiver (process associated to the mailbox) */
62 ActorPtr Mailbox::getReceiver()
63 {
64   if (pimpl_->permanent_receiver == nullptr)
65     return ActorPtr();
66   return pimpl_->permanent_receiver->iface();
67 }
68
69 CommPtr Mailbox::put_init()
70 {
71   CommPtr res   = CommPtr(new s4u::Comm());
72   res->sender_  = SIMIX_process_self();
73   res->mailbox_ = this;
74   return res;
75 }
76 s4u::CommPtr Mailbox::put_init(void* data, uint64_t simulatedSize)
77 {
78   s4u::CommPtr res = put_init();
79   res->setRemains(simulatedSize);
80   res->srcBuff_     = data;
81   res->srcBuffSize_ = sizeof(void*);
82   return res;
83 }
84 s4u::CommPtr Mailbox::put_async(void* data, uint64_t simulatedSize)
85 {
86   s4u::CommPtr res = put_init(data, simulatedSize);
87   res->start();
88   return res;
89 }
90 void Mailbox::put(void* payload, uint64_t simulatedSize)
91 {
92   CommPtr c = put_init();
93   c->setRemains(simulatedSize);
94   c->setSrcData(payload);
95   c->wait();
96 }
97 /** Blocking send with timeout */
98 void Mailbox::put(void* payload, uint64_t simulatedSize, double timeout)
99 {
100   CommPtr c = put_init();
101   c->setRemains(simulatedSize);
102   c->setSrcData(payload);
103   // c->start() is optional.
104   c->wait(timeout);
105 }
106
107 s4u::CommPtr Mailbox::get_init()
108 {
109   CommPtr res    = CommPtr(new s4u::Comm());
110   res->receiver_ = SIMIX_process_self();
111   res->mailbox_  = this;
112   return res;
113 }
114 s4u::CommPtr Mailbox::get_async(void** data)
115 {
116   s4u::CommPtr res = get_init();
117   res->setDstData(data, sizeof(*data));
118   res->start();
119   return res;
120 }
121
122 void* Mailbox::get()
123 {
124   void* res = nullptr;
125   CommPtr c = get_init();
126   c->setDstData(&res, sizeof(res));
127   c->wait();
128   return res;
129 }
130 void* Mailbox::get(double timeout)
131 {
132   void* res = nullptr;
133   CommPtr c = get_init();
134   c->setDstData(&res, sizeof(res));
135   c->wait(timeout);
136   return res;
137 }
138 }
139 }