Logo AND Algorithmique Numérique Distribuée

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