allow to retrieve the source and destination of a host-to-host Comm

[simgrid.git] / include / simgrid / s4u / Comm.hpp

diff --git a/include/simgrid/s4u/Comm.hpp b/include/simgrid/s4u/Comm.hpp
index 36e04e8..6806ed9 100644 (file)
--- a/include/simgrid/s4u/Comm.hpp
+++ b/include/simgrid/s4u/Comm.hpp
@@ -107,8 +107,10 @@ public:
   }
 
   /** Set the source and destination of communications that bypass the mailbox mechanism */
-  CommPtr set_from(Host* from);
-  CommPtr set_to(Host* to);
+  CommPtr set_source(Host* from);
+  Host* get_source() { return from_; }
+  CommPtr set_destination(Host* to);
+  Host* get_destination() { return to_; }
 
   /** Sets the maximal communication rate (in byte/sec). Must be done before start */
   CommPtr set_rate(double rate);