src/surf/trace_mgr.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/sysdep.h"
   9 #include "xbt/log.h"
  10 #include "xbt/str.h"
  11 #include "xbt/dict.h"
  12 #include "trace_mgr_private.h"
  13 #include "surf_private.h"
  14
  15 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_trace, surf, "Surf trace management");
  16
  17 static xbt_dict_t trace_list = NULL;
  18
  19 tmgr_history_t tmgr_history_new(void)
  20 {
  21   tmgr_history_t h;
  22
  23   h = xbt_new0(s_tmgr_history_t, 1);
  24
  25   h->heap = xbt_heap_new(8, xbt_free_f);        /* Why 8 ? Well, why not... */
  26
  27   return h;
  28 }
  29
  30 void tmgr_history_free(tmgr_history_t h)
  31 {
  32   xbt_heap_free(h->heap);
  33   free(h);
  34 }
  35
  36 tmgr_trace_t tmgr_trace_new_from_string(const char *id, const char *input,
  37                                         double periodicity, double timestep)
  38 {
  39   tmgr_trace_t trace = NULL;
  40   int linecount = 0;
  41   s_tmgr_event_t event;
  42   tmgr_event_t last_event = NULL;
  43   xbt_dynar_t list;
  44   unsigned int cpt;
  45   char *val;
  46
  47   if (trace_list) {
  48     trace = xbt_dict_get_or_null(trace_list, id);
  49     if (trace) {
  50       WARN1("Ignoring redefinition of trace %s", id);
  51       return trace;
  52     }
  53   }
  54
  55   xbt_assert1(periodicity >= 0,
  56               "Invalid periodicity %lg (must be positive)", periodicity);
  57
  58   trace = xbt_new0(s_tmgr_trace_t, 1);
  59   trace->event_list = xbt_dynar_new(sizeof(s_tmgr_event_t), NULL);
  60
  61   list = xbt_str_split(input, "\n\r");
  62
  63   xbt_dynar_foreach(list, cpt, val) {
  64     linecount++;
  65     xbt_str_trim(val, " \t\n\r\x0B");
  66     if (val[0] == '#' || val[0] == '\0' || val[0] == '%')
  67       continue;
  68
  69     if (sscanf(val, "PERIODICITY " "%lg" "\n", &periodicity) == 1)
  70       continue;
  71
  72     if (sscanf(val, "TIMESTEP " "%lg" "\n", &timestep) == 1)
  73       continue;
  74
  75     if (sscanf(val, "%lg" " " "%lg" "\n", &event.delta, &event.value) != 2)
  76       xbt_die(bprintf
  77               ("%s:%d: Syntax error in trace\n%s", id, linecount, input));
  78
  79     if (last_event) {
  80       if (last_event->delta > event.delta) {
  81         xbt_die(bprintf
  82                 ("%s:%d: Invalid trace: Events must be sorted, but time %lg > time %lg.\n%s",
  83                  id, linecount, last_event->delta, event.delta, input));
  84       }
  85       last_event->delta = event.delta - last_event->delta;
  86     }
  87     xbt_dynar_push(trace->event_list, &event);
  88     last_event =
  89       xbt_dynar_get_ptr(trace->event_list,
  90                         xbt_dynar_length(trace->event_list) - 1);
  91   }
  92   if (last_event)
  93     last_event->delta = periodicity;
  94
  95   trace->timestep = timestep;
  96
  97   if (!trace_list)
  98     trace_list = xbt_dict_new();
  99
 100   xbt_dict_set(trace_list, id, (void *) trace,
 101                (void (*)(void *)) tmgr_trace_free);
 102
 103   xbt_dynar_free(&list);
 104   return trace;
 105 }
 106
 107 tmgr_trace_t tmgr_trace_new(const char *filename)
 108 {
 109   FILE *f = NULL;
 110   tmgr_trace_t trace = NULL;
 111
 112   if ((!filename) || (strcmp(filename, "") == 0))
 113     return NULL;
 114
 115   if (trace_list) {
 116     trace = xbt_dict_get_or_null(trace_list, filename);
 117     if (trace) {
 118       WARN1("Ignoring redefinition of trace %s", filename);
 119       return trace;
 120     }
 121   }
 122
 123   f = surf_fopen(filename, "r");
 124   xbt_assert2(f!=NULL, "Cannot open file '%s' (path=%s)", filename,
 125        xbt_str_join(surf_path,":"));
 126
 127   char *tstr = xbt_str_from_file(f);
 128   fclose(f);
 129   trace = tmgr_trace_new_from_string(filename, tstr, 0., 10.);
 130   xbt_free(tstr);
 131
 132   return trace;
 133 }
 134
 135 tmgr_trace_t tmgr_empty_trace_new(void)
 136 {
 137   tmgr_trace_t trace = NULL;
 138   s_tmgr_event_t event;
 139
 140   trace = xbt_new0(s_tmgr_trace_t, 1);
 141   trace->event_list = xbt_dynar_new(sizeof(s_tmgr_event_t), NULL);
 142
 143   event.delta = 0.0;
 144   event.value = 0.0;
 145   xbt_dynar_push(trace->event_list, &event);
 146
 147   return trace;
 148 }
 149
 150 void tmgr_trace_free(tmgr_trace_t trace)
 151 {
 152   if (!trace)
 153     return;
 154   xbt_dynar_free(&(trace->event_list));
 155   free(trace);
 156 }
 157
 158 tmgr_trace_event_t tmgr_history_add_trace(tmgr_history_t h,
 159                                           tmgr_trace_t trace,
 160                                           double start_time,
 161                                           unsigned int offset, void *model)
 162 {
 163   tmgr_trace_event_t trace_event = NULL;
 164
 165   trace_event = xbt_new0(s_tmgr_trace_event_t, 1);
 166   trace_event->trace = trace;
 167   trace_event->idx = offset;
 168   trace_event->model = model;
 169
 170   xbt_assert0((trace_event->idx < xbt_dynar_length(trace->event_list)),
 171               "You're referring to an event that does not exist!");
 172
 173   xbt_heap_push(h->heap, trace_event, start_time);
 174
 175   return trace_event;
 176 }
 177
 178 double tmgr_history_next_date(tmgr_history_t h)
 179 {
 180   if (xbt_heap_size(h->heap))
 181     return (xbt_heap_maxkey(h->heap));
 182   else
 183     return -1.0;
 184 }
 185
 186 tmgr_trace_event_t tmgr_history_get_next_event_leq(tmgr_history_t h,
 187                                                    double date,
 188                                                    double *value,
 189                                                    void **model)
 190 {
 191   double event_date = tmgr_history_next_date(h);
 192   tmgr_trace_event_t trace_event = NULL;
 193   tmgr_event_t event = NULL;
 194   tmgr_trace_t trace = NULL;
 195
 196   if (event_date > date)
 197     return NULL;
 198
 199   if (!(trace_event = xbt_heap_pop(h->heap)))
 200     return NULL;
 201
 202   trace = trace_event->trace;
 203   event = xbt_dynar_get_ptr(trace->event_list, trace_event->idx);
 204
 205   *value = event->value;
 206   *model = trace_event->model;
 207
 208   if (trace_event->idx < xbt_dynar_length(trace->event_list) - 1) {
 209     xbt_heap_push(h->heap, trace_event, event_date + event->delta);
 210     trace_event->idx++;
 211   } else if (event->delta > 0) {        /* Last element, checking for periodicity */
 212     xbt_heap_push(h->heap, trace_event, event_date + event->delta);
 213     trace_event->idx = 0;
 214   } else {                      /* We don't need this trace_event anymore */
 215     trace_event->free_me = 1;
 216   }
 217
 218   return trace_event;
 219 }
 220
 221 void tmgr_finalize(void)
 222 {
 223   xbt_dict_free(&trace_list);
 224 }
 225
 226 int tmgr_trace_event_free(tmgr_trace_event_t trace_event)
 227 {
 228   if (trace_event->free_me) {
 229     xbt_free(trace_event);
 230     return 1;
 231   }
 232   return 0;
 233 }