Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
c23662912b59d600f92e7371431e9c7de2983479
[simgrid.git] / src / surf / workstation.c
1 /*      $Id$     */
2
3 /* Copyright (c) 2004 Arnaud Legrand. All rights reserved.                  */
4
5 /* This program is free software; you can redistribute it and/or modify it
6  * under the terms of the license (GNU LGPL) which comes with this package. */
7
8 #include "xbt/dict.h"
9 #include "workstation_private.h"
10 #include "cpu_private.h"
11 #include "network_private.h"
12
13 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(workstation, surf,
14                                 "Logging specific to the SURF workstation module");
15
16 surf_workstation_resource_t surf_workstation_resource = NULL;
17
18 xbt_dict_t workstation_set = NULL;
19
20 static workstation_CLM03_t workstation_new(const char *name,
21                                      void *cpu, void *card)
22 {
23   workstation_CLM03_t workstation = xbt_new0(s_workstation_CLM03_t, 1);
24
25   workstation->resource = (surf_resource_t) surf_workstation_resource;
26   workstation->name = xbt_strdup(name);
27   workstation->cpu = cpu;
28   workstation->network_card = card;
29
30   return workstation;
31 }
32
33 static void workstation_free(void *workstation)
34 {
35   free(((workstation_CLM03_t)workstation)->name);
36   free(workstation);
37 }
38
39 static void create_workstations(void)
40 {
41    xbt_dict_cursor_t cursor = NULL;
42   char *name = NULL;
43   void *cpu = NULL;
44   void *nw_card = NULL;
45
46   xbt_dict_foreach(cpu_set, cursor, name, cpu) {
47     nw_card = NULL;
48     xbt_dict_get(network_card_set, name, (void *) &nw_card);
49     xbt_assert1(nw_card, "No corresponding card found for %s",name);
50     xbt_dict_set(workstation_set, name,
51                  workstation_new(name, cpu, nw_card), workstation_free);
52   }
53 }
54
55 static void *name_service(const char *name)
56 {
57   void *workstation = NULL;
58
59   xbt_dict_get(workstation_set, name, &workstation);
60
61   return workstation;
62 }
63
64 static const char *get_resource_name(void *resource_id)
65 {
66   return ((workstation_CLM03_t) resource_id)->name;
67 }
68
69 static int resource_used(void *resource_id)
70 {
71   xbt_assert0(0,
72               "Workstation is a virtual resource. I should not be there!");
73   return 0;
74 }
75
76 static int parallel_action_free(surf_action_t action)
77 {
78   action->using--;
79   if(!action->using) {
80     xbt_swag_remove(action, action->state_set);
81     if(((surf_action_parallel_task_CSL05_t)action)->variable)
82       lmm_variable_free(maxmin_system, ((surf_action_parallel_task_CSL05_t)action)->variable);
83     free(action);
84     return 1;
85   }
86   return 0;
87 }
88
89 static void parallel_action_use(surf_action_t action)
90 {
91   action->using++;
92 }
93
94 static int action_free(surf_action_t action)
95 {
96   if(action->resource_type==(surf_resource_t)surf_network_resource) 
97     return surf_network_resource->common_public->action_free(action);
98   else if(action->resource_type==(surf_resource_t)surf_cpu_resource) 
99     return surf_cpu_resource->common_public->action_free(action);
100   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
101     return parallel_action_free(action);
102   else DIE_IMPOSSIBLE;
103   return 0;
104 }
105
106 static void action_use(surf_action_t action)
107 {
108   if(action->resource_type==(surf_resource_t)surf_network_resource) 
109     surf_network_resource->common_public->action_use(action);
110   else if(action->resource_type==(surf_resource_t)surf_cpu_resource) 
111     surf_cpu_resource->common_public->action_use(action);
112   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
113     return parallel_action_use(action);
114   else DIE_IMPOSSIBLE;
115   return;
116 }
117
118 static void action_cancel(surf_action_t action)
119 {
120   DIE_IMPOSSIBLE;
121   return;
122 }
123
124 static void action_recycle(surf_action_t action)
125 {
126   DIE_IMPOSSIBLE;
127   return;
128 }
129
130 static void action_change_state(surf_action_t action,
131                                 e_surf_action_state_t state)
132 {
133   if(action->resource_type==(surf_resource_t)surf_network_resource) 
134     surf_network_resource->common_public->action_change_state(action,state);
135   else if(action->resource_type==(surf_resource_t)surf_cpu_resource) 
136     surf_cpu_resource->common_public->action_change_state(action,state);
137   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
138     surf_action_change_state(action, state);
139   else DIE_IMPOSSIBLE;
140   return;
141 }
142
143 static double share_resources(double now)
144 {
145   return -1.0;
146 }
147
148 static void update_actions_state(double now, double delta)
149 {
150   surf_action_parallel_task_CSL05_t action = NULL;
151   surf_action_parallel_task_CSL05_t next_action = NULL;
152   xbt_swag_t running_actions =
153       surf_workstation_resource->common_public->states.running_action_set;
154   xbt_swag_t failed_actions =
155       surf_workstation_resource->common_public->states.failed_action_set;
156
157   xbt_swag_foreach_safe(action, next_action, running_actions) {
158     surf_double_update(&(action->generic_action.remains),
159         lmm_variable_getvalue(action->variable) * delta);
160     if (action->generic_action.max_duration != NO_MAX_DURATION)
161       surf_double_update(&(action->generic_action.max_duration), delta);
162     if ((action->generic_action.remains <= 0) && 
163         (lmm_get_variable_weight(action->variable)>0)) {
164       action->generic_action.finish = surf_get_clock();
165       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
166     } else if ((action->generic_action.max_duration != NO_MAX_DURATION) &&
167                (action->generic_action.max_duration <= 0)) {
168       action->generic_action.finish = surf_get_clock();
169       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
170     } else {                    /* Need to check that none of the resource has failed */
171       lmm_constraint_t cnst = NULL;
172       int i = 0;
173       surf_resource_t resource = NULL;
174
175       while ((cnst =
176               lmm_get_cnst_from_var(maxmin_system, action->variable,
177                                     i++))) {
178         resource = (surf_resource_t) lmm_constraint_id(cnst);
179         if(resource== (surf_resource_t) surf_cpu_resource) {
180           cpu_Cas01_t cpu = lmm_constraint_id(cnst);
181           if (cpu->state_current == SURF_CPU_OFF) {
182             action->generic_action.finish = surf_get_clock();
183             action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
184             break;
185           }
186         } else if (resource== (surf_resource_t) surf_network_resource) {
187           network_link_CM02_t nw_link = lmm_constraint_id(cnst);
188
189           if (nw_link->state_current == SURF_NETWORK_LINK_OFF) {
190             action->generic_action.finish = surf_get_clock();
191             action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
192             break;
193           }
194         } 
195       }
196     }
197   }
198
199   return;
200 }
201
202 static void update_resource_state(void *id,
203                                   tmgr_trace_event_t event_type,
204                                   double value)
205 {
206   return;
207 }
208
209 static surf_action_t execute(void *workstation, double size)
210 {
211   return surf_cpu_resource->extension_public->
212       execute(((workstation_CLM03_t) workstation)->cpu, size);
213 }
214
215 static surf_action_t action_sleep(void *workstation, double duration)
216 {
217   return surf_cpu_resource->extension_public->
218       sleep(((workstation_CLM03_t) workstation)->cpu, duration);
219 }
220
221 static void action_suspend(surf_action_t action)
222 {
223   if(action->resource_type==(surf_resource_t)surf_network_resource) 
224     surf_network_resource->common_public->suspend(action);
225   else if(action->resource_type==(surf_resource_t)surf_cpu_resource) 
226     surf_cpu_resource->common_public->suspend(action);
227   else DIE_IMPOSSIBLE;
228 }
229
230 static void action_resume(surf_action_t action)
231 {
232   if(action->resource_type==(surf_resource_t)surf_network_resource)
233     surf_network_resource->common_public->resume(action);
234   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
235     surf_cpu_resource->common_public->resume(action);
236   else DIE_IMPOSSIBLE;
237 }
238
239 static int action_is_suspended(surf_action_t action)
240 {
241   if(action->resource_type==(surf_resource_t)surf_network_resource) 
242     return surf_network_resource->common_public->is_suspended(action);
243   if(action->resource_type==(surf_resource_t)surf_cpu_resource) 
244     return surf_cpu_resource->common_public->is_suspended(action);
245   DIE_IMPOSSIBLE;
246 }
247
248 static void action_set_max_duration(surf_action_t action, double duration)
249 {
250   if(action->resource_type==(surf_resource_t)surf_network_resource)
251     surf_network_resource->common_public->set_max_duration(action,duration);
252   else if(action->resource_type==(surf_resource_t)surf_cpu_resource) 
253     surf_cpu_resource->common_public->set_max_duration(action,duration);
254   else  DIE_IMPOSSIBLE;
255 }
256
257 static surf_action_t communicate(void *workstation_src,
258                                  void *workstation_dst, double size,
259                                  double rate)
260 {
261   return surf_network_resource->extension_public->
262       communicate(((workstation_CLM03_t) workstation_src)->network_card,
263                   ((workstation_CLM03_t) workstation_dst)->network_card, size, rate);
264 }
265
266 static e_surf_cpu_state_t get_state(void *workstation)
267 {
268   return surf_cpu_resource->extension_public->
269       get_state(((workstation_CLM03_t) workstation)->cpu);
270 }
271
272 static surf_action_t execute_parallel_task (int workstation_nb,
273                                             void **workstation_list,
274                                             double *computation_amount,
275                                             double *communication_amount,
276                                             double amount,
277                                             double rate)
278 {
279   surf_action_parallel_task_CSL05_t action = NULL;
280   int i, j, k;
281   xbt_dict_t network_link_set = xbt_dict_new();
282   xbt_dict_cursor_t cursor = NULL;
283   char *name = NULL;
284   int nb_link = 0;
285   network_link_CM02_t link;
286
287   /* Compute the number of affected resources... */
288   for(i=0; i< workstation_nb; i++) {
289     for(j=0; j< workstation_nb; j++) {
290       network_card_CM02_t card_src = ((workstation_CLM03_t*)workstation_list)[i]->network_card;
291       network_card_CM02_t card_dst = ((workstation_CLM03_t*)workstation_list)[j]->network_card;
292       int route_size = ROUTE_SIZE(card_src->id, card_dst->id);
293       network_link_CM02_t *route = ROUTE(card_src->id, card_dst->id);
294       
295       if(communication_amount[i*workstation_nb+j]>=0)
296         for(k=0; k< route_size; k++) {
297           xbt_dict_set(network_link_set, route[k]->name, route[k], NULL);
298         }
299     }
300   }
301  
302   xbt_dict_foreach(network_link_set, cursor, name, link) {
303     nb_link++;
304   }
305   
306   action = xbt_new0(s_surf_action_parallel_task_CSL05_t, 1);
307   action->generic_action.using = 1;
308   action->generic_action.cost = amount;
309   action->generic_action.remains = amount;
310   action->generic_action.max_duration = NO_MAX_DURATION;
311   action->generic_action.start = -1.0;
312   action->generic_action.finish = -1.0;
313   action->generic_action.resource_type =
314       (surf_resource_t) surf_workstation_resource;
315   action->suspended = 0;  /* Should be useless because of the
316                              calloc but it seems to help valgrind... */
317   action->generic_action.state_set =
318       surf_network_resource->common_public->states.running_action_set;
319
320   xbt_swag_insert(action, action->generic_action.state_set);
321   action->rate = rate;
322
323   if(action->rate>0)
324     action->variable = lmm_variable_new(maxmin_system, action, 1.0, -1.0,
325                                         workstation_nb + nb_link);
326   else   
327     action->variable = lmm_variable_new(maxmin_system, action, 1.0, action->rate,
328                                         workstation_nb + nb_link);
329
330   if(nb_link + workstation_nb == 0)
331     action_change_state((surf_action_t) action, SURF_ACTION_DONE);
332
333   for (i = 0; i<workstation_nb; i++)
334     lmm_expand(maxmin_system, ((cpu_Cas01_t) ((workstation_CLM03_t) workstation_list[i])->cpu)->constraint, 
335                action->variable, computation_amount[i]);
336
337   for (i=0; i<workstation_nb; i++) {
338     for(j=0; j< workstation_nb; j++) {
339       network_card_CM02_t card_src = ((workstation_CLM03_t*)workstation_list)[i]->network_card;
340       network_card_CM02_t card_dst = ((workstation_CLM03_t*)workstation_list)[j]->network_card;
341       int route_size = ROUTE_SIZE(card_src->id, card_dst->id);
342       network_link_CM02_t *route = ROUTE(card_src->id, card_dst->id);
343       
344       for(k=0; k< route_size; k++) {
345         if(communication_amount[i*workstation_nb+j]>=0) {
346           lmm_expand_add(maxmin_system, route[k]->constraint, 
347                        action->variable, communication_amount[i*workstation_nb+j]);
348         }
349       }
350     }
351   }
352   
353   return (surf_action_t) action;
354 }
355
356 static void finalize(void)
357 {
358   xbt_dict_free(&workstation_set);
359   xbt_swag_free(surf_workstation_resource->common_public->states.ready_action_set);
360   xbt_swag_free(surf_workstation_resource->common_public->states.
361                 running_action_set);
362   xbt_swag_free(surf_workstation_resource->common_public->states.
363                 failed_action_set);
364   xbt_swag_free(surf_workstation_resource->common_public->states.done_action_set);
365
366   free(surf_workstation_resource->common_public);
367   free(surf_workstation_resource->common_private);
368   free(surf_workstation_resource->extension_public);
369
370   free(surf_workstation_resource);
371   surf_workstation_resource = NULL;
372 }
373
374 static void surf_workstation_resource_init_internal(void)
375 {
376   s_surf_action_t action;
377
378   surf_workstation_resource = xbt_new0(s_surf_workstation_resource_t, 1);
379
380   surf_workstation_resource->common_private =
381       xbt_new0(s_surf_resource_private_t, 1);
382   surf_workstation_resource->common_public =
383       xbt_new0(s_surf_resource_public_t, 1);
384 /*   surf_workstation_resource->extension_private = xbt_new0(s_surf_workstation_resource_extension_private_t,1); */
385   surf_workstation_resource->extension_public =
386       xbt_new0(s_surf_workstation_resource_extension_public_t, 1);
387
388   surf_workstation_resource->common_public->states.ready_action_set =
389       xbt_swag_new(xbt_swag_offset(action, state_hookup));
390   surf_workstation_resource->common_public->states.running_action_set =
391       xbt_swag_new(xbt_swag_offset(action, state_hookup));
392   surf_workstation_resource->common_public->states.failed_action_set =
393       xbt_swag_new(xbt_swag_offset(action, state_hookup));
394   surf_workstation_resource->common_public->states.done_action_set =
395       xbt_swag_new(xbt_swag_offset(action, state_hookup));
396
397   surf_workstation_resource->common_public->name_service = name_service;
398   surf_workstation_resource->common_public->get_resource_name =
399       get_resource_name;
400   surf_workstation_resource->common_public->action_get_state =
401       surf_action_get_state;
402   surf_workstation_resource->common_public->action_free = action_free;
403   surf_workstation_resource->common_public->action_use = action_use;
404   surf_workstation_resource->common_public->action_cancel = action_cancel;
405   surf_workstation_resource->common_public->action_recycle =
406       action_recycle;
407   surf_workstation_resource->common_public->action_change_state =
408       action_change_state;
409   surf_workstation_resource->common_public->action_set_data = surf_action_set_data;
410   surf_workstation_resource->common_public->name = "Workstation";
411
412   surf_workstation_resource->common_private->resource_used = resource_used;
413   surf_workstation_resource->common_private->share_resources =
414       share_resources;
415   surf_workstation_resource->common_private->update_actions_state =
416       update_actions_state;
417   surf_workstation_resource->common_private->update_resource_state =
418       update_resource_state;
419   surf_workstation_resource->common_private->finalize = finalize;
420
421   surf_workstation_resource->common_public->suspend = action_suspend;
422   surf_workstation_resource->common_public->resume = action_resume;
423   surf_workstation_resource->common_public->is_suspended = action_is_suspended;
424   surf_workstation_resource->common_public->set_max_duration = action_set_max_duration;
425
426   surf_workstation_resource->extension_public->execute = execute;
427   surf_workstation_resource->extension_public->sleep = action_sleep;
428   surf_workstation_resource->extension_public->get_state = get_state;
429   surf_workstation_resource->extension_public->communicate = communicate;
430   surf_workstation_resource->extension_public->execute_parallel_task = 
431     execute_parallel_task;
432
433   workstation_set = xbt_dict_new();
434
435   xbt_assert0(maxmin_system, "surf_init has to be called first!");
436 }
437
438 /********************************************************************/
439 /* The model used in MSG and presented at CCGrid03                  */
440 /********************************************************************/
441 /* @InProceedings{Casanova.CLM_03, */
442 /*   author = {Henri Casanova and Arnaud Legrand and Loris Marchal}, */
443 /*   title = {Scheduling Distributed Applications: the SimGrid Simulation Framework}, */
444 /*   booktitle = {Proceedings of the third IEEE International Symposium on Cluster Computing and the Grid (CCGrid'03)}, */
445 /*   publisher = {"IEEE Computer Society Press"}, */
446 /*   month = {may}, */
447 /*   year = {2003} */
448 /* } */
449 void surf_workstation_resource_init_CLM03(const char *filename)
450 {
451 /*   int i ; */
452 /*   surf_resource_t resource =  NULL; */
453
454   surf_workstation_resource_init_internal();
455   surf_cpu_resource_init_Cas01(filename);
456   surf_network_resource_init_CM02(filename);
457   create_workstations();
458   xbt_dynar_push(resource_list, &surf_workstation_resource);
459 /*   xbt_dynar_foreach(resource_list, i, resource) { */
460 /*     if(resource==surf_cpu_resource) { */
461 /*       xbt_dynar_remove_at(resource_list, i, NULL); */
462 /*       i--;  */
463 /*       continue; */
464 /*     } */
465 /*     if(resource==surf_network_resource) { */
466 /*       xbt_dynar_remove_at(resource_list, i, NULL); */
467 /*       i--;  */
468 /*       continue; */
469 /*     } */
470 /*   } */
471 }