Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Update copyright notices
[simgrid.git] / examples / msg / masterslave / masterslave_platfgen.c
1 /* Copyright (c) 2012-2015. The SimGrid Team.
2  * 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 #include "simgrid/platf_generator.h"
8 #include "xbt.h"
9 #include "simgrid/msg.h"
10
11 #include "xbt/log.h"
12 XBT_LOG_NEW_DEFAULT_CATEGORY(test_generation,
13                              "Messages specific for this msg example");
14
15
16 #define FINALIZE ((void*)221297)        /* a magic number to tell people to stop working */
17
18 void promoter_1(context_node_t node);
19 void labeler_1(context_edge_t edge);
20 int master(int argc, char *argv[]);
21 int slave(int argc, char *argv[]);
22
23 /** Promoter function
24  * Just promote each node into a host, with fixed power
25  */
26 void promoter_1(context_node_t node) {
27
28   s_sg_platf_host_cbarg_t host_parameters;
29
30   host_parameters.id = NULL;
31   host_parameters.power_peak = xbt_dynar_new(sizeof(double), NULL);
32   xbt_dynar_push_as(host_parameters.power_peak, double, 1000000.0);
33   host_parameters.core_amount = 1;
34   host_parameters.power_scale = 1;
35   host_parameters.power_trace = NULL;
36   host_parameters.initial_state = SURF_RESOURCE_ON;
37   host_parameters.state_trace = NULL;
38   host_parameters.coord = NULL;
39   host_parameters.properties = NULL;
40
41   platf_graph_promote_to_host(node, &host_parameters);
42
43 }
44
45 /** Labeler function
46  * Set all links with the same bandwidth and latency
47  */
48 void labeler_1(context_edge_t edge) {
49   s_sg_platf_link_cbarg_t link_parameters;
50   link_parameters.id = NULL;
51   link_parameters.bandwidth = 1000000;
52   link_parameters.bandwidth_trace = NULL;
53   link_parameters.latency = 0.01;
54   link_parameters.latency_trace = NULL;
55   link_parameters.state = SURF_RESOURCE_ON;
56   link_parameters.state_trace = NULL;
57   link_parameters.policy = SURF_LINK_SHARED;
58   link_parameters.properties = NULL;
59
60   platf_graph_link_label(edge, &link_parameters);
61 }
62
63 /** Emitter function  */
64 int master(int argc, char *argv[])
65 {
66   int slaves_count = 0;
67   msg_host_t *slaves = NULL;
68   int number_of_tasks = 0;
69   double task_comp_size = 0;
70   double task_comm_size = 0;
71   int i;
72
73   number_of_tasks = 3*argc;
74   task_comp_size = 2400000*argc;
75   task_comm_size = 1000000;
76
77   {                             /* Process organisation */
78     slaves_count = argc;
79     slaves = xbt_new0(msg_host_t, slaves_count);
80
81     for (i = 0; i < argc; i++) {
82       slaves[i] = MSG_host_by_name(argv[i]);
83       if (slaves[i] == NULL) {
84         XBT_INFO("Unknown host %s. Stopping Now! ", argv[i]);
85         abort();
86       }
87     }
88   }
89
90   XBT_INFO("Got %d slave(s) :", slaves_count);
91   for (i = 0; i < slaves_count; i++)
92     XBT_INFO("%s", MSG_host_get_name(slaves[i]));
93
94   XBT_INFO("Got %d task to process :", number_of_tasks);
95
96   for (i = 0; i < number_of_tasks; i++) {
97     msg_task_t task = MSG_task_create("Task", task_comp_size, task_comm_size,
98                                     xbt_new0(double, 1));
99     int a;
100     *((double *) task->data) = MSG_get_clock();
101
102     a = MSG_task_send(task,MSG_host_get_name(slaves[i % slaves_count]));
103
104     if (a == MSG_OK) {
105       XBT_INFO("Send completed");
106     } else {
107       XBT_INFO("Hey ?! What's up ? ");
108       xbt_die( "Unexpected behavior");
109     }
110   }
111
112   XBT_INFO
113       ("All tasks have been dispatched. Let's tell everybody the computation is over.");
114   for (i = 0; i < slaves_count; i++) {
115     msg_task_t task = MSG_task_create("finalize", 0, 0, FINALIZE);
116     int a = MSG_task_send(task,MSG_host_get_name(slaves[i]));
117     if (a != MSG_OK) {
118       XBT_INFO("Hey ?! What's up ? ");
119       xbt_die("Unexpected behavior with '%s': %d", MSG_host_get_name(slaves[i]), a);
120     }
121   }
122
123   XBT_INFO("Goodbye now!");
124   free(slaves);
125   return 0;
126 }                               /* end_of_master */
127
128 /** Receiver function  */
129 int slave(int argc, char *argv[])
130 {
131   while (1) {
132     msg_task_t task = NULL;
133     int a;
134     double time1, time2;
135
136     time1 = MSG_get_clock();
137     a = MSG_task_receive( &(task), MSG_host_get_name(MSG_host_self()) );
138     time2 = MSG_get_clock();
139     if (a == MSG_OK) {
140       XBT_INFO("Received \"%s\"", MSG_task_get_name(task));
141       if (MSG_task_get_data(task) == FINALIZE) {
142         MSG_task_destroy(task);
143         break;
144       }
145       if (time1 < *((double *) task->data))
146         time1 = *((double *) task->data);
147       XBT_INFO("Communication time : \"%f\"", time2 - time1);
148       XBT_INFO("Processing \"%s\"", MSG_task_get_name(task));
149       a = MSG_task_execute(task);
150       if (a == MSG_OK) {
151         XBT_INFO("\"%s\" done", MSG_task_get_name(task));
152         free(task->data);
153         MSG_task_destroy(task);
154       } else {
155         XBT_INFO("Hey ?! What's up ? ");
156         xbt_die("Unexpected behavior");
157       }
158     } else {
159       XBT_INFO("Hey ?! What's up ? error %d", a);
160       xbt_die("Unexpected behavior");
161     }
162   }
163   XBT_INFO("I'm done. See you!");
164   return 0;
165 }
166
167 /**
168  * Main function
169  * Create the platform, list the available hosts and give them some work
170  */
171 int main(int argc, char **argv) {
172
173   unsigned long seed[] = {134, 233445, 865, 2634, 424242, 876543};
174   int connected;
175   int max_tries = 10;
176
177   //MSG initialisation
178   MSG_init(&argc, argv);
179
180   //Set up the seed for the platform generation
181   platf_random_seed(seed);
182
183   XBT_INFO("creating nodes...");
184   platf_graph_uniform(50);
185   do {
186     max_tries--;
187     XBT_INFO("creating links...");
188     platf_graph_clear_links();
189     platf_graph_interconnect_uniform(0.07); //Unrealistic, but simple
190     XBT_INFO("done. Check connectedness...");
191     connected = platf_graph_is_connected();
192     XBT_INFO("Is it connected : %s", connected ? "yes" : (max_tries ? "no, retrying" : "no"));
193   } while(!connected && max_tries);
194
195   if(!connected && !max_tries) {
196     xbt_die("Impossible to connect the graph, aborting.");
197   }
198
199   XBT_INFO("registering callbacks...");
200   platf_graph_promoter(promoter_1);
201   platf_graph_labeler(labeler_1);
202
203   XBT_INFO("protmoting...");
204   platf_do_promote();
205
206   XBT_INFO("labeling...");
207   platf_do_label();
208
209   XBT_INFO("Putting it in surf...");
210   platf_generate();
211
212   XBT_INFO("Let's get the available hosts and dispatch work:");
213
214   unsigned int i;
215   msg_host_t host = NULL;
216   msg_host_t host_master = NULL;
217   xbt_dynar_t host_dynar = MSG_hosts_as_dynar();
218   char** hostname_list = xbt_malloc(sizeof(char*) * xbt_dynar_length(host_dynar));
219
220   xbt_dynar_foreach(host_dynar, i, host) {
221     MSG_process_create("slave", slave, NULL, host);
222     if(i==0) {
223       //The first node will also be the master
224       XBT_INFO("%s will be the master", MSG_host_get_name(host));
225       host_master = host;
226     }
227     hostname_list[i] = (char*) MSG_host_get_name(host);
228   }
229
230   MSG_process_create_with_arguments("master", master, NULL, host_master, xbt_dynar_length(host_dynar), hostname_list);
231
232   XBT_INFO("Let's rock!");
233   MSG_main();
234   XBT_INFO("Simulation time %g", MSG_get_clock());
235   return 0;
236 }