Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Fixed full routing support for routers, no routing complexity is added routers are...
[simgrid.git] / examples / msg / gtnets / gtnets.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include "msg/msg.h"
4 #include "xbt/log.h"
5 #include "xbt/asserts.h"
6
7 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
8                              "Messages specific for this msg example");
9
10 int master(int argc, char *argv[]);
11 int slave(int argc, char *argv[]);
12 MSG_error_t test_all(const char *platform_file, const char *application_file);
13
14 typedef enum {
15   PORT_22 = 0,
16   MAX_CHANNEL
17 } channel_t;
18
19 //keep a pointer to all surf running tasks.
20 #define NTASKS 1500
21 int bool_printed = 0;
22 double start_time, end_time, elapsed_time;
23 double gl_data_size[NTASKS];
24 m_task_t gl_task_array[NTASKS];
25 const char *slavenames[NTASKS];
26 const char *masternames[NTASKS];
27 int gl_task_array_id = 0;
28
29 #define FINALIZE ((void*)221297)        /* a magic number to tell people to stop working */
30
31 /** master */
32 int master(int argc, char *argv[])
33 {
34   char *slavename = NULL;
35   double task_comm_size = 0;
36   m_task_t todo;
37   m_host_t slave;
38   char id_alias[10];
39   //unique id to control statistics
40   int id = -1;
41
42   if (argc != 4) {
43     INFO1("Strange number of arguments expected 3 got %d", argc - 1);
44   }
45
46   /* data size */
47   xbt_assert1(sscanf(argv[1], "%lg", &task_comm_size),
48               "Invalid argument %s\n", argv[1]);
49
50   /* slave name */
51   slavename = argv[2];
52   id = atoi(argv[3]);
53   sprintf(id_alias, "%d", id);
54   slavenames[id] = slavename;
55
56   masternames[id] = MSG_host_get_name(MSG_host_self());
57
58   {                             /*  Task creation.  */
59     char sprintf_buffer[64] = "Task_0";
60     todo = MSG_task_create(sprintf_buffer, 0, task_comm_size, NULL);
61     //keep track of running tasks
62     gl_task_array[id] = todo;
63     gl_data_size[id] = task_comm_size;
64   }
65
66   {                             /* Process organisation */
67     slave = MSG_get_host_by_name(slavename);
68   }
69
70   /* time measurement */
71   start_time = MSG_get_clock();
72   MSG_task_send(todo, id_alias);
73
74   end_time = MSG_get_clock();
75   INFO3("Send completed (to %s). Transfer time: %f\t Agregate bandwidth: %f",
76         slave->name, (end_time - start_time),
77         task_comm_size / (end_time - start_time));
78   INFO2("Completed peer: %s time: %f", slave->name, (end_time - start_time));
79
80   return 0;
81 }                               /* end_of_master */
82
83 /** Receiver function  */
84 int slave(int argc, char *argv[])
85 {
86
87   m_task_t task = NULL;
88   int a;
89   int id = 0;
90   double remaining = 0;
91   char id_alias[10];
92
93   if (argc != 2) {
94     INFO1("Strange number of arguments expected 1 got %d", argc - 1);
95   }
96
97   id = atoi(argv[1]);
98   sprintf(id_alias, "%d", id);
99
100   a = MSG_task_receive(&(task), id_alias);
101
102   if (a != MSG_OK) {
103     INFO0("Hey?! What's up?");
104     xbt_assert0(0, "Unexpected behavior.");
105   }
106
107   elapsed_time = MSG_get_clock() - start_time;
108
109   if (!bool_printed) {
110     bool_printed = 1;
111     for (id = 0; id < NTASKS; id++) {
112       if (gl_task_array[id] == NULL) {
113       } else if (gl_task_array[id] == task) {
114         INFO5
115           ("===> Estimated Bw of FLOW[%d] : %f ;  message from %s to %s  with remaining : %f",
116            id, gl_data_size[id] / elapsed_time, masternames[id],
117            slavenames[id], 0.0);
118       } else {
119         remaining = MSG_task_get_remaining_communication(gl_task_array[id]);
120         INFO5
121           ("===> Estimated Bw of FLOW[%d] : %f ;  message from %s to %s  with remaining : %f",
122            id, (gl_data_size[id] - remaining) / elapsed_time, masternames[id],
123            slavenames[id], remaining);
124       }
125     }
126     exit(0);
127   }
128
129   for (id = 0; id < NTASKS; id++) {
130     if (gl_task_array[id] == task) {
131       MSG_task_destroy(task);
132       gl_task_array[id] = NULL;
133       return 0;
134     }
135   }
136
137   return 0;
138 }                               /* end_of_slave */
139
140 /** Test function */
141 MSG_error_t test_all(const char *platform_file, const char *application_file)
142 {
143   MSG_error_t res = MSG_OK;
144
145   /* MSG_config("workstation_model", "GTNETS"); */
146   /* MSG_config("workstation_model","KCCFLN05"); */
147   {                             /*  Simulation setting */
148     MSG_set_channel_number(MAX_CHANNEL);
149     //MSG_paje_output("msg_test.trace");
150     MSG_create_environment(platform_file);
151   }
152   {                             /*   Application deployment */
153     MSG_function_register("master", master);
154     MSG_function_register("slave", slave);
155     MSG_launch_application(application_file);
156   }
157   res = MSG_main();
158   return res;
159 }                               /* end_of_test_all */
160
161 /** Main function */
162 int main(int argc, char *argv[])
163 {
164   MSG_error_t res = MSG_OK;
165   bool_printed = 0;
166
167   MSG_global_init(&argc, argv);
168   if (argc < 3) {
169     printf("Usage: %s platform_file deployment_file\n", argv[0]);
170     exit(1);
171   }
172   res = test_all(argv[1], argv[2]);
173
174   MSG_clean();
175
176   if (res == MSG_OK)
177     return 0;
178   else
179     return 1;
180 }                               /* end_of_main */