Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Support for ns3-3.12
[simgrid.git] / src / surf / ns3 / ns3_simulator.h
index dce821b..1335f02 100644 (file)
@@ -9,15 +9,61 @@
 
 #ifdef __cplusplus
 
+#include "ns3/core-module.h"
+
+#ifdef _NS3_3_10
+       /*NS3 3.10*/
+       #include "ns3/helper-module.h"
+       #include "ns3/simulator-module.h"
+       #include "ns3/node-module.h"
+       #include "ns3/helper-module.h"
+       #include "ns3/global-routing-module.h"
+       #include "ns3/tcp-socket-factory.h"
+#else
+       /*NS3 3.12*/
+       #include "ns3/node.h"
+       #include "ns3/global-route-manager.h"
+       #include "ns3/csma-helper.h"
+       #include "ns3/internet-stack-helper.h"
+       #include "ns3/ipv4-address-helper.h"
+       #include "ns3/point-to-point-helper.h"
+       #include "ns3/packet-sink-helper.h"
+       #include "ns3/inet-socket-address.h"
+#endif
+
+using namespace ns3;
+using namespace std;
+
+struct MySocket{
+  uint32_t bufferedBytes;
+  uint32_t sentBytes;
+  uint32_t remaining;
+  uint32_t totalBytes;
+  char finished;
+  void* action;
+};
+
 //Simulator s;
 class NS3Sim {
-       NS3Sim();
-       ~NS3Sim();
 
 private:
 
 public:
-
+       NS3Sim();
+       ~NS3Sim();
+       void create_flow_NS3(Ptr<Node> src,
+                                               Ptr<Node> dst,
+                                               uint16_t port_number,
+                                               double start,
+                                               const char *addr,
+                                               uint32_t TotalBytes,
+                                               void * action);
+       void simulator_stop(double min);
+       void simulator_start(void);
+       void* get_action_from_socket(void *socket);
+       double get_remains_from_socket(void *socket);
+       double get_sent_from_socket(void *socket);
+       char get_finished(void *socket);
 };
 
 #endif                          /* __cplusplus */