Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
add new callbacks to the Workstation's models, and define new lua methods to extend...
[simgrid.git] / src / surf / workstation.c
1 /* Copyright (c) 2004, 2005, 2006, 2007, 2008, 2009, 2010. 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 "xbt/ex.h"
8 #include "xbt/dict.h"
9 #include "portable.h"
10 #include "surf_private.h"
11
12 typedef struct workstation_CLM03 {
13   s_surf_resource_t generic_resource;   /* Must remain first to add this to a trace */
14   void *cpu;
15   int id;
16 } s_workstation_CLM03_t, *workstation_CLM03_t;
17
18 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_workstation, surf,
19                                 "Logging specific to the SURF workstation module");
20
21 surf_model_t surf_workstation_model = NULL;
22
23 static workstation_CLM03_t workstation_new(const char *name,
24                                            void *cpu, int id)
25 {
26   workstation_CLM03_t workstation = xbt_new0(s_workstation_CLM03_t, 1);
27
28   workstation->generic_resource.model = surf_workstation_model;
29   workstation->generic_resource.name = xbt_strdup(name);
30   workstation->cpu = cpu;
31   workstation->id = id;
32
33   xbt_dict_set(surf_model_resource_set(surf_workstation_model), name,
34                workstation, surf_resource_free);
35
36   return workstation;
37 }
38
39 void create_workstations(void)
40 {
41   xbt_dict_cursor_t cursor = NULL;
42   char *name = NULL;
43   void *cpu = NULL;
44
45   xbt_dict_foreach(surf_model_resource_set(surf_cpu_model), cursor, name, cpu) {
46     int *id = NULL;
47     if (used_routing && used_routing->host_id)
48       id = xbt_dict_get_or_null(used_routing->host_id,name);
49
50     workstation_new(name, cpu, id ? *id : 0);
51   }
52 }
53
54 static void ws_cpu_create_resource(char *name, double power_peak,
55         double power_scale,
56         tmgr_trace_t power_trace,
57         e_surf_resource_state_t state_initial,
58         tmgr_trace_t state_trace,
59         xbt_dict_t cpu_properties)
60 {
61         THROW_UNIMPLEMENTED;
62 }
63
64 static void ws_link_create_resource(char *name,double bw_initial,double lat_initial)
65 {
66         THROW_UNIMPLEMENTED;
67 }
68
69 static int ws_resource_used(void *resource_id)
70 {
71   THROW_IMPOSSIBLE;             /* This model does not implement parallel tasks */
72 }
73
74 static void ws_parallel_action_cancel(surf_action_t action)
75 {
76   THROW_UNIMPLEMENTED;          /* This model does not implement parallel tasks */
77 }
78
79 static int ws_parallel_action_free(surf_action_t action)
80 {
81   THROW_UNIMPLEMENTED;          /* This model does not implement parallel tasks */
82 }
83
84 static int ws_action_unref(surf_action_t action)
85 {
86   if (action->model_type == surf_network_model)
87     return surf_network_model->action_unref(action);
88   else if (action->model_type == surf_cpu_model)
89     return surf_cpu_model->action_unref(action);
90   else if (action->model_type == surf_workstation_model)
91     return ws_parallel_action_free(action);
92   else
93     DIE_IMPOSSIBLE;
94   return 0;
95 }
96
97 static void ws_action_cancel(surf_action_t action)
98 {
99   if (action->model_type == surf_network_model)
100     surf_network_model->action_cancel(action);
101   else if (action->model_type == surf_cpu_model)
102     surf_cpu_model->action_cancel(action);
103   else if (action->model_type == surf_workstation_model)
104     ws_parallel_action_cancel(action);
105   else
106     DIE_IMPOSSIBLE;
107   return;
108 }
109
110 static void ws_action_state_set(surf_action_t action,
111                                 e_surf_action_state_t state)
112 {
113   if (action->model_type == surf_network_model)
114     surf_network_model->action_state_set(action, state);
115   else if (action->model_type == surf_cpu_model)
116     surf_cpu_model->action_state_set(action, state);
117   else if (action->model_type == surf_workstation_model)
118     surf_action_state_set(action, state);
119   else
120     DIE_IMPOSSIBLE;
121   return;
122 }
123
124 static double ws_share_resources(double now)
125 {
126   return -1.0;
127 }
128
129 static void ws_update_actions_state(double now, double delta)
130 {
131   return;
132 }
133
134 static void ws_update_resource_state(void *id,
135                                   tmgr_trace_event_t event_type,
136                                   double value, double date)
137 {
138   THROW_IMPOSSIBLE;             /* This model does not implement parallel tasks */
139 }
140
141 static surf_action_t ws_execute(void *workstation, double size)
142 {
143   return surf_cpu_model->extension.cpu.
144     execute(((workstation_CLM03_t) workstation)->cpu, size);
145 }
146
147 static surf_action_t ws_action_sleep(void *workstation, double duration)
148 {
149   return surf_cpu_model->extension.cpu.
150     sleep(((workstation_CLM03_t) workstation)->cpu, duration);
151 }
152
153 static void ws_action_suspend(surf_action_t action)
154 {
155   if (action->model_type == surf_network_model)
156     surf_network_model->suspend(action);
157   else if (action->model_type == surf_cpu_model)
158     surf_cpu_model->suspend(action);
159   else
160     DIE_IMPOSSIBLE;
161 }
162
163 static void ws_action_resume(surf_action_t action)
164 {
165   if (action->model_type == surf_network_model)
166     surf_network_model->resume(action);
167   else if (action->model_type == surf_cpu_model)
168     surf_cpu_model->resume(action);
169   else
170     DIE_IMPOSSIBLE;
171 }
172
173 static int ws_action_is_suspended(surf_action_t action)
174 {
175   if (action->model_type == surf_network_model)
176     return surf_network_model->is_suspended(action);
177   if (action->model_type == surf_cpu_model)
178     return surf_cpu_model->is_suspended(action);
179   DIE_IMPOSSIBLE;
180 }
181
182 static void ws_action_set_max_duration(surf_action_t action, double duration)
183 {
184   if (action->model_type == surf_network_model)
185     surf_network_model->set_max_duration(action, duration);
186   else if (action->model_type == surf_cpu_model)
187     surf_cpu_model->set_max_duration(action, duration);
188   else
189     DIE_IMPOSSIBLE;
190 }
191
192 static void ws_action_set_priority(surf_action_t action, double priority)
193 {
194   if (action->model_type == surf_network_model)
195     surf_network_model->set_priority(action, priority);
196   else if (action->model_type == surf_cpu_model)
197     surf_cpu_model->set_priority(action, priority);
198   else
199     DIE_IMPOSSIBLE;
200 }
201
202 static double ws_action_get_remains(surf_action_t action)
203 {
204   if (action->model_type == surf_network_model)
205     return surf_network_model->get_remains(action);
206   if (action->model_type == surf_cpu_model)
207     return surf_cpu_model->get_remains(action);
208   DIE_IMPOSSIBLE;
209 }
210
211 static surf_action_t ws_communicate(void *workstation_src,
212                                  void *workstation_dst, double size,
213                                  double rate)
214 {
215   workstation_CLM03_t src = (workstation_CLM03_t) workstation_src;
216   workstation_CLM03_t dst = (workstation_CLM03_t) workstation_dst;
217   return surf_network_model->extension.network.
218     communicate(surf_resource_name(src->cpu), surf_resource_name(dst->cpu),
219                 src->id, dst->id, size, rate);
220 }
221
222 static e_surf_resource_state_t ws_get_state(void *workstation)
223 {
224   return surf_cpu_model->extension.cpu.
225     get_state(((workstation_CLM03_t) workstation)->cpu);
226 }
227
228 static double ws_get_speed(void *workstation, double load)
229 {
230   return surf_cpu_model->extension.cpu.
231     get_speed(((workstation_CLM03_t) workstation)->cpu, load);
232 }
233
234 static double ws_get_available_speed(void *workstation)
235 {
236   return surf_cpu_model->extension.cpu.
237     get_available_speed(((workstation_CLM03_t)
238                          workstation)->cpu);
239 }
240
241 static surf_action_t ws_execute_parallel_task(int workstation_nb,
242                                            void **workstation_list,
243                                            double *computation_amount,
244                                            double *communication_amount,
245                                            double amount, double rate)
246 {
247   THROW_UNIMPLEMENTED;          /* This model does not implement parallel tasks */
248 }
249
250
251 /* returns an array of network_link_CM02_t */
252 static xbt_dynar_t ws_get_route(void *src, void *dst)
253 {
254   workstation_CLM03_t workstation_src = (workstation_CLM03_t) src;
255   workstation_CLM03_t workstation_dst = (workstation_CLM03_t) dst;
256   return surf_network_model->extension.network.get_route(workstation_src->id,
257                                                          workstation_dst->id);
258 }
259
260 static double ws_get_link_bandwidth(const void *link)
261 {
262   return surf_network_model->extension.network.get_link_bandwidth(link);
263 }
264
265 static double ws_get_link_latency(const void *link)
266 {
267   return surf_network_model->extension.network.get_link_latency(link);
268 }
269
270 static int ws_link_shared(const void *link)
271 {
272   return surf_network_model->extension.network.get_link_latency(link);
273 }
274
275 static void ws_finalize(void)
276 {
277   surf_model_exit(surf_workstation_model);
278   surf_workstation_model = NULL;
279 }
280
281 static xbt_dict_t ws_get_properties(const void *ws)
282 {
283   return surf_resource_properties(((workstation_CLM03_t) ws)->cpu);
284 }
285
286 static void surf_workstation_model_init_internal(void)
287 {
288   surf_workstation_model = surf_model_init();
289
290   surf_workstation_model->name = "Workstation";
291   surf_workstation_model->action_unref = ws_action_unref;
292   surf_workstation_model->action_cancel = ws_action_cancel;
293   surf_workstation_model->action_state_set = ws_action_state_set;
294
295   surf_workstation_model->model_private->resource_used = ws_resource_used;
296   surf_workstation_model->model_private->share_resources = ws_share_resources;
297   surf_workstation_model->model_private->update_actions_state =
298     ws_update_actions_state;
299   surf_workstation_model->model_private->update_resource_state =
300     ws_update_resource_state;
301   surf_workstation_model->model_private->finalize = ws_finalize;
302
303   surf_workstation_model->suspend = ws_action_suspend;
304   surf_workstation_model->resume = ws_action_resume;
305   surf_workstation_model->is_suspended = ws_action_is_suspended;
306   surf_workstation_model->set_max_duration = ws_action_set_max_duration;
307   surf_workstation_model->set_priority = ws_action_set_priority;
308   surf_workstation_model->get_remains = ws_action_get_remains;
309
310   surf_workstation_model->extension.workstation.execute = ws_execute;
311   surf_workstation_model->extension.workstation.sleep = ws_action_sleep;
312   surf_workstation_model->extension.workstation.get_state = ws_get_state;
313   surf_workstation_model->extension.workstation.get_speed = ws_get_speed;
314   surf_workstation_model->extension.workstation.get_available_speed =
315     ws_get_available_speed;
316
317   surf_workstation_model->extension.workstation.communicate = ws_communicate;
318   surf_workstation_model->extension.workstation.get_route = ws_get_route;
319   surf_workstation_model->extension.workstation.execute_parallel_task =
320     ws_execute_parallel_task;
321   surf_workstation_model->extension.workstation.get_link_bandwidth =
322     ws_get_link_bandwidth;
323   surf_workstation_model->extension.workstation.get_link_latency =
324     ws_get_link_latency;
325   surf_workstation_model->extension.workstation.link_shared = ws_link_shared;
326   surf_workstation_model->extension.workstation.get_properties =
327     ws_get_properties;
328   surf_workstation_model->extension.workstation.link_create_resource =
329     ws_link_create_resource;
330   surf_workstation_model->extension.workstation.cpu_create_resource =
331         ws_cpu_create_resource;
332 }
333
334 /********************************************************************/
335 /* The model used in MSG and presented at CCGrid03                  */
336 /********************************************************************/
337 /* @InProceedings{Casanova.CLM_03, */
338 /*   author = {Henri Casanova and Arnaud Legrand and Loris Marchal}, */
339 /*   title = {Scheduling Distributed Applications: the SimGrid Simulation Framework}, */
340 /*   booktitle = {Proceedings of the third IEEE International Symposium on Cluster Computing and the Grid (CCGrid'03)}, */
341 /*   publisher = {"IEEE Computer Society Press"}, */
342 /*   month = {may}, */
343 /*   year = {2003} */
344 /* } */
345 void surf_workstation_model_init_CLM03(const char *filename)
346 {
347   surf_workstation_model_init_internal();
348   surf_cpu_model_init_Cas01_im(filename);
349   surf_network_model_init_LegrandVelho(filename);
350   update_model_description(surf_workstation_model_description,
351                            "CLM03", surf_workstation_model);
352   xbt_dynar_push(model_list, &surf_workstation_model);
353 }
354
355 void surf_workstation_model_init_compound(const char *filename)
356 {
357
358   xbt_assert0(surf_cpu_model, "No CPU model defined yet!");
359   xbt_assert0(surf_network_model, "No network model defined yet!");
360   surf_workstation_model_init_internal();
361
362   update_model_description(surf_workstation_model_description,
363                            "compound", surf_workstation_model);
364
365   xbt_dynar_push(model_list, &surf_workstation_model);
366 }