Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Rha!
[simgrid.git] / src / surf / cpu.c
1 /* Authors: Arnaud Legrand                                                  */
2
3 /* This program is free software; you can redistribute it and/or modify it
4    under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #include "cpu_private.h"
7 #include "xbt/dict.h"
8
9 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(cpu, surf ,"Logging specific to the SURF CPU module");
10
11 surf_cpu_resource_t surf_cpu_resource = NULL;
12
13 static xbt_dict_t cpu_set = NULL;
14
15 /* power_scale is the basic power of the cpu when the cpu is
16    completely available. initial_power is therefore expected to be
17    comprised between 0.0 and 1.0, just as the values of power_trace.
18    state_trace values mean SURF_CPU_ON if >0 and SURF_CPU_OFF
19    otherwise.
20 */
21
22 static void cpu_free(void *CPU){
23   cpu_t cpu = CPU;
24
25 /* lmm_constraint_free(maxmin_system,cpu->constraint); */
26 /* Clean somewhere else ! */
27
28   xbt_free(cpu);
29 }
30
31 static void *cpu_new(const char *name, xbt_maxmin_float_t power_scale,
32                       xbt_maxmin_float_t initial_power, tmgr_trace_t power_trace,
33                       e_surf_cpu_state_t initial_state, tmgr_trace_t state_trace)
34 {
35   cpu_t cpu = xbt_new0(s_cpu_t,1);
36
37   cpu->resource = (surf_resource_t) surf_cpu_resource;
38   cpu->name = name;
39   cpu->power_scale = power_scale;
40   cpu->current_power = initial_power;
41 /*   cpu->power_trace = power_trace; */
42   if(power_trace) 
43     cpu->power_event = tmgr_history_add_trace(history, power_trace, 0.0, 0, cpu);
44
45   cpu->current_state = initial_state;
46 /*   cpu->state_trace = state_trace; */
47   if(state_trace) 
48     cpu->state_event = tmgr_history_add_trace(history, state_trace, 0.0, 0, cpu);
49
50   cpu->constraint = lmm_constraint_new(maxmin_system, cpu, cpu->current_power * cpu->power_scale);
51
52   xbt_dict_set(cpu_set, name, cpu, cpu_free);
53
54   return cpu;
55 }
56
57 static void parse_file(const char *file)
58 {
59   tmgr_trace_t trace_A = tmgr_trace_new("trace_A.txt");
60   tmgr_trace_t trace_B = tmgr_trace_new("trace_B.txt");
61   tmgr_trace_t trace_A_failure = tmgr_trace_new("trace_A_failure.txt");
62
63   cpu_new("Cpu A", 100.0, 1.0, trace_A, SURF_CPU_ON, trace_A_failure);
64   cpu_new("Cpu B", 100.0, 1.0, trace_B, SURF_CPU_ON, NULL);
65 }
66
67 static void *name_service(const char *name)
68 {
69   void *cpu=NULL;
70   
71   xbt_dict_get(cpu_set, name, &cpu);
72
73   return cpu;
74 }
75
76 static const char *get_resource_name(void *resource_id)
77 {
78   return ((cpu_t) resource_id)->name;
79 }
80
81 static int resource_used(void *resource_id)
82 {
83   return lmm_constraint_used(maxmin_system, ((cpu_t)resource_id)->constraint);
84 }
85
86 static void action_free(surf_action_t action)
87 {
88   surf_action_cpu_t Action = (surf_action_cpu_t) action;
89
90   xbt_swag_remove(action,action->state_set);
91   lmm_variable_free(maxmin_system,Action->variable);
92   xbt_free(action);
93
94   return;
95 }
96
97 static void action_cancel(surf_action_t action)
98 {
99   return;
100 }
101
102 static void action_recycle(surf_action_t action)
103 {
104   return;
105 }
106
107 static void action_change_state(surf_action_t action, e_surf_action_state_t state)
108 {
109   surf_action_change_state(action, state);
110   return;
111 }
112
113 static xbt_heap_float_t share_resources()
114 {
115   surf_action_cpu_t action = NULL;
116   xbt_swag_t running_actions= surf_cpu_resource->common_public->states.running_action_set;
117   xbt_maxmin_float_t min = -1;
118   xbt_maxmin_float_t value = -1;
119   lmm_solve(maxmin_system);  
120
121   action = xbt_swag_getFirst(running_actions);
122   if(!action) return 0.0;
123   value = lmm_variable_getvalue(action->variable);
124   min = action->generic_action.remains / value ;
125
126   xbt_swag_foreach(action,running_actions) {
127     value = action->generic_action.remains / 
128       lmm_variable_getvalue(action->variable);
129     if(value<min) min=value;
130   }
131
132   return min;
133 }
134
135 static void update_actions_state(xbt_heap_float_t now,
136                                  xbt_heap_float_t delta)
137 {
138   surf_action_cpu_t action = NULL;
139   surf_action_cpu_t next_action = NULL;
140   xbt_swag_t running_actions= surf_cpu_resource->common_public->states.running_action_set;
141   xbt_swag_t failed_actions= surf_cpu_resource->common_public->states.failed_action_set;
142
143   xbt_swag_foreach_safe(action, next_action, running_actions) {
144     action->generic_action.remains -= 
145       lmm_variable_getvalue(action->variable)*delta;
146 /*     if(action->generic_action.remains<.00001) action->generic_action.remains=0; */
147     if(action->generic_action.remains<=0) {
148       action_change_state((surf_action_t)action, SURF_ACTION_DONE);
149     } else { /* Need to check that none of the resource has failed*/
150       lmm_constraint_t cnst = NULL;
151       int tab_size =  lmm_get_number_of_cnst_from_var(maxmin_system, action->variable);
152       int i=0;
153       cpu_t cpu = NULL;
154
155       while((cnst=lmm_get_cnst_from_var(maxmin_system, action->variable, i++))) {
156         cpu = lmm_constraint_id(cnst);
157         if(cpu->current_state==SURF_CPU_OFF) {
158           action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
159           break;
160         }
161       }
162     }
163   }
164
165   xbt_swag_foreach_safe(action, next_action, failed_actions) {
166     lmm_variable_disable(maxmin_system, action->variable);
167   }
168
169   return;
170 }
171
172 static void update_resource_state(void *id,
173                                   tmgr_trace_event_t event_type, 
174                                   xbt_maxmin_float_t value)
175 {
176   cpu_t cpu = id;
177   
178   printf("[" XBT_HEAP_FLOAT_T "] Asking to update \"%s\" with value " XBT_MAXMIN_FLOAT_T " for event %p\n",
179          surf_get_clock(), cpu->name, value, event_type);
180   
181   if(event_type==cpu->power_event) {
182     cpu->current_power = value;
183     lmm_update_constraint_bound(cpu->constraint,cpu->current_power * cpu->power_scale);
184   } else if (event_type==cpu->state_event) {
185     if(value>0) cpu->current_state = SURF_CPU_ON;
186     else cpu->current_state = SURF_CPU_OFF;
187   } else abort();
188
189   return;
190 }
191
192 static surf_action_t execute(void *cpu, xbt_maxmin_float_t size)
193 {
194   lmm_variable_t var;
195   surf_action_cpu_t action = NULL;
196   cpu_t CPU = cpu;
197   
198   action=xbt_new0(s_surf_action_cpu_t,1);
199
200   action->generic_action.cost=size;
201   action->generic_action.remains=size;  
202   action->generic_action.start=-1.0;
203   action->generic_action.finish=-1.0;
204   action->generic_action.callback=cpu;
205   action->generic_action.resource_type=(surf_resource_t)surf_cpu_resource;
206
207   if(CPU->current_state==SURF_CPU_ON) 
208     action->generic_action.state_set=surf_cpu_resource->common_public->states.running_action_set;
209   else 
210     action->generic_action.state_set=surf_cpu_resource->common_public->states.failed_action_set;
211   xbt_swag_insert(action,action->generic_action.state_set);
212
213   action->variable = lmm_variable_new(maxmin_system, action, 1.0, -1.0, 1);
214   lmm_expand(maxmin_system, ((cpu_t)cpu)->constraint, action->variable, 1.0);
215
216   return (surf_action_t) action;
217 }
218
219 static e_surf_cpu_state_t get_state(void *cpu)
220 {
221   return SURF_CPU_OFF;
222 }
223
224 static void finalize(void)
225 {
226   xbt_dict_free(&cpu_set);
227   xbt_swag_free(surf_cpu_resource->common_public->states.ready_action_set);
228   xbt_swag_free(surf_cpu_resource->common_public->states.running_action_set);
229   xbt_swag_free(surf_cpu_resource->common_public->states.failed_action_set);
230   xbt_swag_free(surf_cpu_resource->common_public->states.done_action_set);
231   xbt_free(surf_cpu_resource->common_public);
232   xbt_free(surf_cpu_resource->common_private);
233   xbt_free(surf_cpu_resource->extension_public);
234
235   xbt_free(surf_cpu_resource);
236   surf_cpu_resource = NULL;
237 }
238
239 static void surf_cpu_resource_init_internal(void)
240 {
241   s_surf_action_t action;
242
243   surf_cpu_resource = xbt_new0(s_surf_cpu_resource_t,1);
244   
245   surf_cpu_resource->common_private = xbt_new0(s_surf_resource_private_t,1);
246   surf_cpu_resource->common_public = xbt_new0(s_surf_resource_public_t,1);
247 /*   surf_cpu_resource->extension_private = xbt_new0(s_surf_cpu_resource_extension_private_t,1); */
248   surf_cpu_resource->extension_public = xbt_new0(s_surf_cpu_resource_extension_public_t,1);
249
250   surf_cpu_resource->common_public->states.ready_action_set=
251     xbt_swag_new(xbt_swag_offset(action,state_hookup));
252   surf_cpu_resource->common_public->states.running_action_set=
253     xbt_swag_new(xbt_swag_offset(action,state_hookup));
254   surf_cpu_resource->common_public->states.failed_action_set=
255     xbt_swag_new(xbt_swag_offset(action,state_hookup));
256   surf_cpu_resource->common_public->states.done_action_set=
257     xbt_swag_new(xbt_swag_offset(action,state_hookup));
258
259   surf_cpu_resource->common_public->name_service = name_service;
260   surf_cpu_resource->common_public->get_resource_name = get_resource_name;
261   surf_cpu_resource->common_public->resource_used = resource_used;
262   surf_cpu_resource->common_public->action_get_state=surf_action_get_state;
263   surf_cpu_resource->common_public->action_free = action_free;
264   surf_cpu_resource->common_public->action_cancel = action_cancel;
265   surf_cpu_resource->common_public->action_recycle = action_recycle;
266   surf_cpu_resource->common_public->action_change_state = action_change_state;
267
268   surf_cpu_resource->common_private->share_resources = share_resources;
269   surf_cpu_resource->common_private->update_actions_state = update_actions_state;
270   surf_cpu_resource->common_private->update_resource_state = update_resource_state;
271   surf_cpu_resource->common_private->finalize = finalize;
272
273   surf_cpu_resource->extension_public->execute = execute;
274   surf_cpu_resource->extension_public->get_state = get_state;
275
276   cpu_set = xbt_dict_new();
277
278   xbt_assert0(maxmin_system,"surf_init has to be called first!");
279 }
280
281 void surf_cpu_resource_init(const char* filename)
282 {
283   surf_cpu_resource_init_internal();
284   parse_file(filename);
285   xbt_dynar_push(resource_list, &surf_cpu_resource);
286 }