X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/7f6d6d56876d2e286d0f11bfb390af6de6c9863e..29cedcb0af274d0dd82224451767e8d01162d734:/src/surf/ns3/ns3_interface.h

diff --git a/src/surf/ns3/ns3_interface.h b/src/surf/ns3/ns3_interface.h
index e6d2187751..0537b104e4 100644
--- a/src/surf/ns3/ns3_interface.h
+++ b/src/surf/ns3/ns3_interface.h
@@ -1,4 +1,4 @@
-/* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
+/* Copyright (c) 2007-2014. The SimGrid Team.
  * All rights reserved.                                                     */
 
 /* This program is free software; you can redistribute it and/or modify it
@@ -11,41 +11,60 @@
 #include "xbt/dynar.h"
 #include "xbt/misc.h"
 #include "xbt/sysdep.h"
+#include <xbt/Extendable.hpp>
 
+#include <simgrid/s4u/host.hpp>
+#include <surf/surf_routing.h>
+
+namespace simgrid{
+  namespace surf{
+    class NetworkNS3Action;
+  }
+}
 typedef enum {
-	NS3_NETWORK_ELEMENT_NULL = 0,    /* NULL */
-	NS3_NETWORK_ELEMENT_HOST,    	/* host type */
-	NS3_NETWORK_ELEMENT_ROUTER,   	/* router type */
-	NS3_NETWORK_ELEMENT_AS,      	/* AS type */
+  NS3_NETWORK_ELEMENT_NULL = 0,    /* NULL */
+  NS3_NETWORK_ELEMENT_HOST,      /* host type */
+  NS3_NETWORK_ELEMENT_ROUTER,     /* router type */
+  NS3_NETWORK_ELEMENT_AS,        /* AS type */
 } e_ns3_network_element_type_t;
 
-typedef struct ns3_nodes{
-	int node_num;
-	e_ns3_network_element_type_t type;
-	void * data;
-}s_ns3_nodes_t, *ns3_nodes_t;
 
-#ifdef __cplusplus
-extern "C" {
-#endif
+typedef struct ns3_node {
+  int node_num;
+  e_ns3_network_element_type_t type;
+} s_ns3_node_t, *ns3_node_t;
+
+XBT_PUBLIC_DATA(int) NS3_EXTENSION_ID;
 
-XBT_PUBLIC(int)    ns3_finalize(void);
-XBT_PUBLIC(int)    ns3_initialize(void);
-XBT_PUBLIC(int)    ns3_create_flow(const char* a,const char *b,double start,u_int32_t TotalBytes,void * action);
+SG_BEGIN_DECL()
+
+XBT_PUBLIC(void)   ns3_finalize();
+XBT_PUBLIC(int)    ns3_initialize(const char* TcpProtocol);
+XBT_PUBLIC(int)    ns3_create_flow(const char* a,const char *b,double start,u_int32_t TotalBytes,simgrid::surf::NetworkNS3Action * action);
 XBT_PUBLIC(void)   ns3_simulator(double min);
-XBT_PUBLIC(double) ns3_time(void);
-XBT_PUBLIC(void*)  ns3_get_socket_action(void *socket);
+XBT_PUBLIC(simgrid::surf::NetworkNS3Action*)  ns3_get_socket_action(void *socket);
 XBT_PUBLIC(double) ns3_get_socket_remains(void *socket);
-XBT_PUBLIC(char)   ns3_get_socket_is_finished(void *socket);
-XBT_PUBLIC(void *) ns3_add_host(char * id);
-XBT_PUBLIC(void *) ns3_add_host_cluster(char * id);
-XBT_PUBLIC(void *) ns3_add_router(char * id);
-XBT_PUBLIC(void *) ns3_add_AS(char * id);
-XBT_PUBLIC(void *) ns3_add_link(int src,int dst,char * bw,char * lat);
-XBT_PUBLIC(void *) ns3_end_platform(void);
-XBT_PUBLIC(void *) ns3_add_cluster(char * bw,char * lat,char *id);
-#ifdef __cplusplus
+XBT_PUBLIC(double) ns3_get_socket_sent(void *socket);
+XBT_PUBLIC(bool)   ns3_get_socket_is_finished(void *socket);
+XBT_PUBLIC(void *) ns3_add_host_cluster(const char * id);
+XBT_PUBLIC(void *) ns3_add_router(const char * id);
+XBT_PUBLIC(void *) ns3_add_AS(const char * id);
+XBT_PUBLIC(void) ns3_add_link(int src, e_ns3_network_element_type_t type_src,
+                int dst, e_ns3_network_element_type_t type_dst,
+                char * bw,char * lat);
+XBT_PUBLIC(void) ns3_end_platform(void);
+XBT_PUBLIC(void) ns3_add_cluster(char * bw,char * lat,const char *id);
+
+inline
+ns3_node_t ns3_find_host(const char* id)
+{
+  sg_host_t host = sg_host_by_name(id);
+  if (host == nullptr)
+    return nullptr;
+  else
+    return (ns3_node_t) host->extension(NS3_EXTENSION_ID);
 }
-#endif
+
+SG_END_DECL()
 
 #endif