Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
d0412eae2828ba4aa44e1f6f3a7bc7ef6242c883
[simgrid.git] / src / surf / gtnets / gtnets_interface.cc
1 /*      $Id$     */
2 /* Copyright (c) 2007 Kayo Fujiwara. All rights reserved.                  */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7
8 #include "gtnets_simulator.h"
9 #include "gtnets_interface.h"
10 #ifdef XBT_DEBUG
11         #undef XBT_DEBUG
12 #endif
13 #include "xbt/log.h"
14 #include "xbt/asserts.h"
15
16 static GTSim* gtnets_sim = 0;
17
18
19 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_network_gtnets_interface, surf_network_gtnets,
20                                 "Logging specific to the SURF network GTNetS interface");
21
22
23
24 // initialize the GTNetS interface and environment
25 int gtnets_initialize(int wsize){
26   XBT_DEBUG("Using logging.");
27   xbt_assert(!gtnets_sim, "gtnets already initialized");
28
29   if(wsize > 0){
30         XBT_INFO("TCP window maximum size : %d", wsize);
31         gtnets_sim = new GTSim(wsize);
32   }else{
33         gtnets_sim = new GTSim(wsize);
34   }
35
36   return 0;
37 }
38
39 // add a link (argument link is just an index starting at 0... 
40 // add link 0, add link 1, etc.)
41 int gtnets_add_link(int id, double bandwidth, double latency){
42   return gtnets_sim->add_link(id, bandwidth, latency);
43 }
44
45 // add a route between a source and a destination as an array of link indices
46 // (note that there is no gtnets_add_network_card(), as we discover them
47 // on the fly via calls to gtnets_add_route()
48 int gtnets_add_route(int src, int dst, int* links, int nlink){
49   return gtnets_sim->add_route(src, dst, links, nlink);
50 }
51
52 // add a router
53 int gtnets_add_router(int id){
54   return gtnets_sim->add_router(id);
55 }
56
57 // add a route between a source and a destination as an array of link indices
58 // (note that there is no gtnets_add_network_card(), as we discover them
59 // on the fly via calls to gtnets_add_route()
60 int gtnets_add_onehop_route(int src, int dst, int link){
61   return gtnets_sim->add_onehop_route(src, dst, link);
62 }
63
64 // create a new flow on a route
65 // one can attach arbitrary metadata to a flow
66 int gtnets_create_flow(int src, int dst, long datasize, void* metadata){
67   return gtnets_sim->create_flow(src, dst, datasize, metadata);
68 }
69
70 // get the time (double) until a flow completes (the first such flow)
71 // if no flows exist, return -1.0
72 double gtnets_get_time_to_next_flow_completion(){
73   ofstream file;
74   streambuf* sbuf;
75   double value;
76
77   if (!XBT_LOG_ISENABLED(surf_network_gtnets_interface, xbt_log_priority_debug)) {
78           file.open ("/dev/null");
79           sbuf = cout.rdbuf();
80           cout.rdbuf(file.rdbuf());
81           XBT_DEBUG("Enable GTNetS library quite mode");
82   }else {
83           XBT_DEBUG("Disable GTNetS library quite mode");
84   }
85
86   value = gtnets_sim->get_time_to_next_flow_completion();
87
88   if (!XBT_LOG_ISENABLED(surf_network_gtnets_interface, xbt_log_priority_debug)) {
89           cout.rdbuf(sbuf);
90           file.close();
91   }
92   return value;
93 }
94
95 // run until a flow completes (returns that flow's metadata)
96 double gtnets_run_until_next_flow_completion(void ***metadata, int *number_of_flows){
97   ofstream file;
98   streambuf* sbuf;
99   double value;
100
101   if (!XBT_LOG_ISENABLED(surf_network_gtnets_interface, xbt_log_priority_debug)) {
102           file.open ("/dev/null");
103           sbuf = cout.rdbuf();
104           cout.rdbuf(file.rdbuf());
105           XBT_DEBUG("Enable GTNetS library quite mode");
106   }else {
107           XBT_DEBUG("Disable GTNetS library quite mode");
108   }
109
110   value = gtnets_sim->run_until_next_flow_completion(metadata, number_of_flows);
111
112   if (!XBT_LOG_ISENABLED(surf_network_gtnets_interface, xbt_log_priority_debug)) {
113           cout.rdbuf(sbuf);
114           file.close();
115   }
116   return (double) value;
117 }
118
119 // get the total received in bytes using the TCPServer object totRx field
120 double gtnets_get_flow_rx(void *metadata){
121   return gtnets_sim->gtnets_get_flow_rx(metadata);
122 }
123
124
125 // run for a given time (double)
126 int gtnets_run(Time_t deltat){
127   gtnets_sim->run(deltat);
128   return 0;
129 }
130
131 // clean up
132 int gtnets_finalize(){
133   if (!gtnets_sim) return -1;
134   delete gtnets_sim;
135   gtnets_sim = 0;
136   return 0;
137 }
138
139 // print topology
140 void gtnets_print_topology(void){
141   gtnets_sim->print_topology();
142 }
143
144 // set jitter
145 void gtnets_set_jitter(double d){
146   gtnets_sim->set_jitter(d);
147 }
148
149 // set jitter seed
150 void gtnets_set_jitter_seed(int seed){
151   gtnets_sim->set_jitter_seed(seed);
152 }
153
154