Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
My god, the AIX compiler is rather picky
[simgrid.git] / src / surf / cpu.c
index 3f4c510..1aa1daa 100644 (file)
-/* Authors: Arnaud Legrand                                                  */
+/*     $Id$     */
+
+/* Copyright (c) 2004 Arnaud Legrand. All rights reserved.                  */
 
 /* This program is free software; you can redistribute it and/or modify it
  under the terms of the license (GNU LGPL) which comes with this package. */
* under the terms of the license (GNU LGPL) which comes with this package. */
 
 #include "cpu_private.h"
-#include "xbt/dict.h"
+
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(cpu, surf,
+                               "Logging specific to the SURF CPU module");
 
 surf_cpu_resource_t surf_cpu_resource = NULL;
-static xbt_dict_t cpu_set = NULL;
-static lmm_system_t sys = NULL;
-
-typedef struct cpu {
-  const char *name;
-  xbt_maxmin_float_t power_scale;
-  xbt_maxmin_float_t current_power;
-  tmgr_trace_t power_trace;
-  e_surf_action_state_t current_state;
-  tmgr_trace_t state_trace;
-  lmm_constraint_t constraint;
-} s_cpu_t, *cpu_t;
+
+xbt_dict_t cpu_set = NULL;
+
+static void cpu_free(void *CPU)
+{
+/*   cpu_t cpu = CPU; */
+/* lmm_constraint_free(maxmin_system,cpu->constraint); */
+/* Clean somewhere else ! */
+
+  xbt_free(CPU);
+}
 
 /* power_scale is the basic power of the cpu when the cpu is
-   completely available. initial_power is therefore expected to be
+   completely available. power_initial is therefore expected to be
    comprised between 0.0 and 1.0, just as the values of power_trace.
    state_trace values mean SURF_CPU_ON if >0 and SURF_CPU_OFF
    otherwise.
 */
-
-static void *new_cpu(const char *name, xbt_maxmin_float_t power_scale,
-                     xbt_maxmin_float_t initial_power, tmgr_trace_t power_trace,
-                     e_surf_cpu_state_t initial_state, tmgr_trace_t state_trace)
+static cpu_t cpu_new(const char *name, double power_scale,
+                    double power_initial,
+                    tmgr_trace_t power_trace,
+                    e_surf_cpu_state_t state_initial,
+                    tmgr_trace_t state_trace)
 {
-  cpu_t cpu = xbt_new0(s_cpu_t,1);
+  cpu_t cpu = xbt_new0(s_cpu_t, 1);
 
+  cpu->resource = (surf_resource_t) surf_cpu_resource;
   cpu->name = name;
   cpu->power_scale = power_scale;
-  cpu->current_power = initial_power;
-  cpu->power_trace = power_trace;
-  cpu->current_state = initial_state;
-  cpu->state_trace = state_trace;
-  cpu->constraint = lmm_constraint_new(sys, cpu, cpu->current_power);
+  cpu->power_current = power_initial;
+/*   cpu->power_trace = power_trace; */
+  if (power_trace)
+    cpu->power_event =
+       tmgr_history_add_trace(history, power_trace, 0.0, 0, cpu);
+
+  cpu->state_current = state_initial;
+/*   cpu->state_trace = state_trace; */
+  if (state_trace)
+    cpu->state_event =
+       tmgr_history_add_trace(history, state_trace, 0.0, 0, cpu);
 
-  xbt_dict_set(cpu_set, name, cpu, NULL);
+  cpu->constraint =
+      lmm_constraint_new(maxmin_system, cpu,
+                        cpu->power_current * cpu->power_scale);
+
+  xbt_dict_set(cpu_set, name, cpu, cpu_free);
 
   return cpu;
 }
 
+/*  
+   Semantic:  name       scale     initial     power     initial     state
+                                    power      trace      state      trace
+   
+   Token:   TOKEN_WORD TOKEN_WORD TOKEN_WORD TOKEN_WORD TOKEN_WORD TOKEN_WORD
+   Type:     string      double     double     string     ON/OFF     string
+*/
+
+static void parse_cpu(void)
+{
+  e_surf_token_t token;
+  char *name = NULL;
+  double power_scale = 0.0;
+  double power_initial = 0.0;
+  tmgr_trace_t power_trace = NULL;
+  e_surf_cpu_state_t state_initial = SURF_CPU_OFF;
+  tmgr_trace_t state_trace = NULL;
+
+  name = xbt_strdup(surf_parse_text);
+
+  surf_parse_double(&power_scale);
+  surf_parse_double(&power_initial);
+  surf_parse_trace(&power_trace);
+
+  token = surf_parse();                /* state_initial */
+  xbt_assert1((token == TOKEN_WORD), "Parse error line %d", line_pos);
+  if (strcmp(surf_parse_text, "ON") == 0)
+    state_initial = SURF_CPU_ON;
+  else if (strcmp(surf_parse_text, "OFF") == 0)
+    state_initial = SURF_CPU_OFF;
+  else {
+    CRITICAL2("Invalid cpu state (line %d): %s neq ON or OFF\n", line_pos,
+             surf_parse_text);
+    xbt_abort();
+  }
+
+  surf_parse_trace(&state_trace);
+
+  cpu_new(name, power_scale, power_initial, power_trace, state_initial,
+         state_trace);
+}
+
 static void parse_file(const char *file)
 {
-  new_cpu("Cpu A", 200.0, 1.0, NULL, SURF_CPU_ON, NULL);
-  new_cpu("Cpu B", 100.0, 1.0, NULL, SURF_CPU_ON, NULL);
+  e_surf_token_t token;
+
+  find_section(file, "CPU");
+
+  while (1) {
+    token = surf_parse();
+
+    if (token == TOKEN_END_SECTION)
+      break;
+    if (token == TOKEN_NEWLINE)
+      continue;
+
+    if (token == TOKEN_WORD)
+      parse_cpu();
+    else {
+      CRITICAL1("Parse error line %d\n", line_pos);
+      xbt_abort();
+    }
+  }
+
+  close_section("CPU");
 }
 
 static void *name_service(const char *name)
 {
-  void *cpu=NULL;
-  
+  void *cpu = NULL;
+
   xbt_dict_get(cpu_set, name, &cpu);
 
   return cpu;
@@ -66,18 +142,20 @@ static const char *get_resource_name(void *resource_id)
   return ((cpu_t) resource_id)->name;
 }
 
-static surf_action_t action_new(void *callback)
+static int resource_used(void *resource_id)
 {
-  return NULL;
+  return lmm_constraint_used(maxmin_system,
+                            ((cpu_t) resource_id)->constraint);
 }
 
-static e_surf_action_state_t action_get_state(surf_action_t action)
+static void action_free(surf_action_t action)
 {
-  return SURF_ACTION_NOT_IN_THE_SYSTEM;
-}
+  surf_action_cpu_t Action = (surf_action_cpu_t) action;
+
+  xbt_swag_remove(action, action->state_set);
+  lmm_variable_free(maxmin_system, Action->variable);
+  xbt_free(action);
 
-static void action_free(surf_action_t * action)
-{
   return;
 }
 
@@ -91,54 +169,230 @@ static void action_recycle(surf_action_t action)
   return;
 }
 
-static void action_change_state(surf_action_t action, e_surf_action_state_t state)
+static void action_change_state(surf_action_t action,
+                               e_surf_action_state_t state)
 {
   surf_action_change_state(action, state);
   return;
 }
 
-static xbt_heap_float_t share_resources(void)
+static double share_resources(double now)
 {
-  return -1.0;
+  s_surf_action_cpu_t action;
+  return generic_maxmin_share_resources(surf_cpu_resource->common_public->
+                                       states.running_action_set,
+                                       xbt_swag_offset(action, variable));
 }
 
-static void solve(xbt_heap_float_t date)
+static void update_actions_state(double now, double delta)
 {
+  surf_action_cpu_t action = NULL;
+  surf_action_cpu_t next_action = NULL;
+  xbt_swag_t running_actions =
+      surf_cpu_resource->common_public->states.running_action_set;
+  xbt_swag_t failed_actions =
+      surf_cpu_resource->common_public->states.failed_action_set;
+
+  xbt_swag_foreach_safe(action, next_action, running_actions) {
+    action->generic_action.remains -=
+       lmm_variable_getvalue(action->variable) * delta;
+    if (action->generic_action.max_duration != NO_MAX_DURATION)
+      action->generic_action.max_duration -= delta;
+/*     if(action->generic_action.remains<.00001) action->generic_action.remains=0; */
+    if (action->generic_action.remains <= 0) {
+      action->generic_action.finish = surf_get_clock();
+      action_change_state((surf_action_t) action, SURF_ACTION_DONE);
+    } else if ((action->generic_action.max_duration != NO_MAX_DURATION) &&
+              (action->generic_action.max_duration <= 0)) {
+      action->generic_action.finish = surf_get_clock();
+      action_change_state((surf_action_t) action, SURF_ACTION_DONE);
+    } else {                   /* Need to check that none of the resource has failed */
+      lmm_constraint_t cnst = NULL;
+      int i = 0;
+      cpu_t cpu = NULL;
+
+      while ((cnst =
+             lmm_get_cnst_from_var(maxmin_system, action->variable,
+                                   i++))) {
+       cpu = lmm_constraint_id(cnst);
+       if (cpu->state_current == SURF_CPU_OFF) {
+         action->generic_action.finish = surf_get_clock();
+         action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
+         break;
+       }
+      }
+    }
+  }
+
+  xbt_swag_foreach_safe(action, next_action, failed_actions) {
+    lmm_variable_disable(maxmin_system, action->variable);
+  }
+
   return;
 }
 
-static surf_action_t execute(void *cpu, xbt_maxmin_float_t size)
+static void update_resource_state(void *id,
+                                 tmgr_trace_event_t event_type,
+                                 double value)
 {
-  return NULL;
+  cpu_t cpu = id;
+
+/*   printf("[" "%lg" "] Asking to update CPU \"%s\" with value " */
+/*      "%lg" " for event %p\n", surf_get_clock(), cpu->name, */
+/*      value, event_type); */
+
+  if (event_type == cpu->power_event) {
+    cpu->power_current = value;
+    lmm_update_constraint_bound(maxmin_system, cpu->constraint,
+                               cpu->power_current * cpu->power_scale);
+  } else if (event_type == cpu->state_event) {
+    if (value > 0)
+      cpu->state_current = SURF_CPU_ON;
+    else
+      cpu->state_current = SURF_CPU_OFF;
+  } else {
+    CRITICAL0("Unknown event ! \n");
+    xbt_abort();
+  }
+
+  return;
+}
+
+static surf_action_t execute(void *cpu, double size)
+{
+  surf_action_cpu_t action = NULL;
+  cpu_t CPU = cpu;
+
+  action = xbt_new0(s_surf_action_cpu_t, 1);
+
+  action->generic_action.cost = size;
+  action->generic_action.remains = size;
+  action->generic_action.max_duration = NO_MAX_DURATION;
+  action->generic_action.start = surf_get_clock();
+  action->generic_action.finish = -1.0;
+/*   action->generic_action.callback = cpu; */
+  action->generic_action.callback = NULL;
+  action->generic_action.resource_type =
+      (surf_resource_t) surf_cpu_resource;
+
+  if (CPU->state_current == SURF_CPU_ON)
+    action->generic_action.state_set =
+       surf_cpu_resource->common_public->states.running_action_set;
+  else
+    action->generic_action.state_set =
+       surf_cpu_resource->common_public->states.failed_action_set;
+  xbt_swag_insert(action, action->generic_action.state_set);
+
+  action->variable = lmm_variable_new(maxmin_system, action, 1.0, -1.0, 1);
+  lmm_expand(maxmin_system, ((cpu_t) cpu)->constraint, action->variable,
+            1.0);
+
+  return (surf_action_t) action;
+}
+
+static surf_action_t action_sleep(void *cpu, double duration)
+{
+  surf_action_cpu_t action = NULL;
+
+  action = (surf_action_cpu_t) execute(cpu, 1.0);
+  action->generic_action.max_duration = duration;
+  lmm_update_variable_weight(maxmin_system, action->variable, 0.0);
+
+  return (surf_action_t) action;
+}
+
+static void action_suspend(surf_action_t action)
+{
+  lmm_update_variable_weight(maxmin_system,
+                            ((surf_action_cpu_t) action)->variable, 0.0);
+}
+
+static void action_resume(surf_action_t action)
+{
+  lmm_update_variable_weight(maxmin_system,
+                            ((surf_action_cpu_t) action)->variable, 1.0);
 }
 
 static e_surf_cpu_state_t get_state(void *cpu)
 {
-  return SURF_CPU_OFF;
+  return ((cpu_t) cpu)->state_current;
 }
 
+static void finalize(void)
+{
+  xbt_dict_free(&cpu_set);
+  xbt_swag_free(surf_cpu_resource->common_public->states.ready_action_set);
+  xbt_swag_free(surf_cpu_resource->common_public->states.
+               running_action_set);
+  xbt_swag_free(surf_cpu_resource->common_public->states.
+               failed_action_set);
+  xbt_swag_free(surf_cpu_resource->common_public->states.done_action_set);
+  xbt_free(surf_cpu_resource->common_public);
+  xbt_free(surf_cpu_resource->common_private);
+  xbt_free(surf_cpu_resource->extension_public);
+
+  xbt_free(surf_cpu_resource);
+  surf_cpu_resource = NULL;
+}
 
-surf_cpu_resource_t surf_cpu_resource_init(void)
+static void surf_cpu_resource_init_internal(void)
 {
-  surf_cpu_resource = xbt_new0(s_surf_cpu_resource_t,1);
+  s_surf_action_t action;
 
-  surf_cpu_resource->resource.parse_file = parse_file;
-  surf_cpu_resource->resource.name_service = name_service;
-  surf_cpu_resource->resource.get_resource_name = get_resource_name;
-  surf_cpu_resource->resource.action_get_state=surf_action_get_state;
-  surf_cpu_resource->resource.action_free = action_free;
-  surf_cpu_resource->resource.action_cancel = action_cancel;
-  surf_cpu_resource->resource.action_recycle = action_recycle;
-  surf_cpu_resource->resource.action_change_state = action_change_state;
-  surf_cpu_resource->resource.share_resources = share_resources;
-  surf_cpu_resource->resource.solve = solve;
+  surf_cpu_resource = xbt_new0(s_surf_cpu_resource_t, 1);
 
-  surf_cpu_resource->execute = execute;
-  surf_cpu_resource->get_state = get_state;
+  surf_cpu_resource->common_private =
+      xbt_new0(s_surf_resource_private_t, 1);
+  surf_cpu_resource->common_public = xbt_new0(s_surf_resource_public_t, 1);
+
+  surf_cpu_resource->extension_public =
+      xbt_new0(s_surf_cpu_resource_extension_public_t, 1);
+
+  surf_cpu_resource->common_public->states.ready_action_set =
+      xbt_swag_new(xbt_swag_offset(action, state_hookup));
+  surf_cpu_resource->common_public->states.running_action_set =
+      xbt_swag_new(xbt_swag_offset(action, state_hookup));
+  surf_cpu_resource->common_public->states.failed_action_set =
+      xbt_swag_new(xbt_swag_offset(action, state_hookup));
+  surf_cpu_resource->common_public->states.done_action_set =
+      xbt_swag_new(xbt_swag_offset(action, state_hookup));
+
+  surf_cpu_resource->common_public->name_service = name_service;
+  surf_cpu_resource->common_public->get_resource_name = get_resource_name;
+  surf_cpu_resource->common_public->action_get_state =
+      surf_action_get_state;
+  surf_cpu_resource->common_public->action_free = action_free;
+  surf_cpu_resource->common_public->action_cancel = action_cancel;
+  surf_cpu_resource->common_public->action_recycle = action_recycle;
+  surf_cpu_resource->common_public->action_change_state =
+      action_change_state;
+  surf_cpu_resource->common_public->name = "CPU";
+
+  surf_cpu_resource->common_private->resource_used = resource_used;
+  surf_cpu_resource->common_private->share_resources = share_resources;
+  surf_cpu_resource->common_private->update_actions_state =
+      update_actions_state;
+  surf_cpu_resource->common_private->update_resource_state =
+      update_resource_state;
+  surf_cpu_resource->common_private->finalize = finalize;
+
+  surf_cpu_resource->extension_public->execute = execute;
+  surf_cpu_resource->extension_public->sleep = action_sleep;
+  surf_cpu_resource->extension_public->suspend = action_suspend;
+  surf_cpu_resource->extension_public->resume = action_resume;
+
+  surf_cpu_resource->extension_public->get_state = get_state;
 
   cpu_set = xbt_dict_new();
 
-  sys = lmm_system_new();
+  xbt_assert0(maxmin_system, "surf_init has to be called first!");
+}
 
-  return surf_cpu_resource;
+void surf_cpu_resource_init(const char *filename)
+{
+  if (surf_cpu_resource)
+    return;
+  surf_cpu_resource_init_internal();
+  parse_file(filename);
+  xbt_dynar_push(resource_list, &surf_cpu_resource);
 }