directly store the ns3Node, don't rely on a num ID to find it later on

[simgrid.git] / src / surf / ns3 / ns3_interface.hpp

diff --git a/src/surf/ns3/ns3_interface.hpp b/src/surf/ns3/ns3_interface.hpp
index 8b1bc35..dfee129 100644 (file)
--- a/src/surf/ns3/ns3_interface.hpp
+++ b/src/surf/ns3/ns3_interface.hpp
@@ -6,6 +6,7 @@
 #ifndef NS3_INTERFACE_HPP
 #define NS3_INTERFACE_HPP
 
 #ifndef NS3_INTERFACE_HPP
 #define NS3_INTERFACE_HPP
 

+#include "ns3/ptr.h"

 #include "simgrid/s4u/Host.hpp"
 
 namespace simgrid {
 #include "simgrid/s4u/Host.hpp"
 
 namespace simgrid {


@@ -13,6 +14,9 @@ namespace surf {
 class NetworkNS3Action;
 }
 }
 class NetworkNS3Action;
 }
 }

+namespace ns3 {
+class Node;
+}

 
 class NetPointNs3 {
 public:
 
 class NetPointNs3 {
 public:


@@ -20,6 +24,7 @@ public:
 
   explicit NetPointNs3();
   int node_num;
 
   explicit NetPointNs3();
   int node_num;

+  ns3::Ptr<ns3::Node> ns3Node_;

 };
 
 extern "C" {
 };
 
 extern "C" {