src/surf/trace_mgr.cpp

   1 /* Copyright (c) 2004-2005, 2007, 2009-2014. 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/sysdep.h"
   8 #include "xbt/log.h"
   9 #include "xbt/str.h"
  10 #include "xbt/dict.h"
  11 #include "src/surf/trace_mgr.hpp"
  12 #include "surf_private.h"
  13 #include "xbt/RngStream.h"
  14 #include <math.h>
  15 #include <unordered_map>
  16
  17 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_trace, surf, "Surf trace management");
  18
  19 static std::unordered_map<const char *, simgrid::trace_mgr::trace*> trace_list;
  20
  21 simgrid::trace_mgr::trace::trace()
  22 {
  23   event_list = xbt_dynar_new(sizeof(s_tmgr_event_t), NULL);
  24 }
  25
  26 simgrid::trace_mgr::trace::~trace()
  27 {
  28   xbt_dynar_free(&event_list);
  29 }
  30 simgrid::trace_mgr::future_evt_set::future_evt_set()
  31 {
  32 }
  33
  34 simgrid::trace_mgr::future_evt_set::~future_evt_set()
  35 {
  36   xbt_heap_free(p_heap);
  37 }
  38
  39 tmgr_trace_t tmgr_trace_new_from_string(const char *name, const char *input, double periodicity)
  40 {
  41   tmgr_trace_t trace = NULL;
  42   int linecount = 0;
  43   s_tmgr_event_t event;
  44   tmgr_event_t last_event = NULL;
  45   xbt_dynar_t list;
  46   unsigned int cpt;
  47   char *val;
  48
  49   if (trace_list.find(name) != trace_list.end()) {
  50     XBT_WARN("Ignoring redefinition of trace %s", name);
  51     return trace_list.at(name);
  52   }
  53
  54   xbt_assert(periodicity >= 0, "Invalid periodicity %g (must be positive)", periodicity);
  55
  56   trace = new simgrid::trace_mgr::trace();
  57
  58   list = xbt_str_split(input, "\n\r");
  59
  60   xbt_dynar_foreach(list, cpt, val) {
  61     linecount++;
  62     xbt_str_trim(val, " \t\n\r\x0B");
  63     if (val[0] == '#' || val[0] == '\0' || val[0] == '%')
  64       continue;
  65
  66     if (sscanf(val, "PERIODICITY " "%lg" "\n", &periodicity) == 1)
  67       continue;
  68
  69     if (sscanf(val, "%lg" " " "%lg" "\n", &event.delta, &event.value) != 2)
  70       xbt_die("%s:%d: Syntax error in trace\n%s", name, linecount, input);
  71
  72     if (last_event) {
  73       if (last_event->delta > event.delta) {
  74         xbt_die("%s:%d: Invalid trace: Events must be sorted, "
  75                 "but time %g > time %g.\n%s",
  76                 name, linecount, last_event->delta, event.delta, input);
  77       }
  78       last_event->delta = event.delta - last_event->delta;
  79     } else {
  80       if(event.delta > 0.0){
  81         s_tmgr_event_t first_event;
  82         first_event.delta=event.delta;
  83         first_event.value=-1.0;
  84         xbt_dynar_push(trace->event_list, &first_event);
  85       }
  86     }
  87     xbt_dynar_push(trace->event_list, &event);
  88     last_event = (tmgr_event_t)xbt_dynar_get_ptr(trace->event_list, xbt_dynar_length(trace->event_list) - 1);
  89   }
  90   if (last_event)
  91     last_event->delta = periodicity;
  92
  93   trace_list.insert({xbt_strdup(name), trace});
  94
  95   xbt_dynar_free(&list);
  96   return trace;
  97 }
  98
  99 tmgr_trace_t tmgr_trace_new_from_file(const char *filename)
 100 {
 101   tmgr_trace_t trace = NULL;
 102
 103   xbt_assert(filename && filename[0], "Cannot parse a trace from the null or empty filename");
 104
 105   if (trace_list.find(filename) != trace_list.end()) {
 106     XBT_WARN("Ignoring redefinition of trace file %s", filename);
 107     return trace_list.at(filename);
 108   }
 109
 110   FILE *f = surf_fopen(filename, "r");
 111   xbt_assert(f != NULL,
 112       "Cannot open file '%s' (path=%s)", filename, xbt_str_join(surf_path, ":"));
 113
 114   char *tstr = xbt_str_from_file(f);
 115   fclose(f);
 116   trace = tmgr_trace_new_from_string(filename, tstr, 0.);
 117   xbt_free(tstr);
 118
 119   return trace;
 120 }
 121
 122 tmgr_trace_t tmgr_empty_trace_new(void)
 123 {
 124   tmgr_trace_t trace = NULL;
 125   s_tmgr_event_t event;
 126
 127   trace = new simgrid::trace_mgr::trace();
 128
 129   event.delta = 0.0;
 130   event.value = 0.0;
 131   xbt_dynar_push(trace->event_list, &event);
 132
 133   return trace;
 134 }
 135
 136 void tmgr_trace_free(tmgr_trace_t trace)
 137 {
 138   delete trace;
 139 }
 140
 141 /** @brief Registers a new trace into the future event set, and get an iterator over the integrated trace  */
 142 tmgr_trace_iterator_t simgrid::trace_mgr::future_evt_set::add_trace(tmgr_trace_t trace, double start_time, surf::Resource *resource)
 143 {
 144   tmgr_trace_iterator_t trace_iterator = NULL;
 145
 146   trace_iterator = xbt_new0(s_tmgr_trace_event_t, 1);
 147   trace_iterator->trace = trace;
 148   trace_iterator->idx = 0;
 149   trace_iterator->resource = resource;
 150
 151   xbt_assert((trace_iterator->idx < xbt_dynar_length(trace->event_list)), "Your trace should have at least one event!");
 152
 153   xbt_heap_push(p_heap, trace_iterator, start_time);
 154
 155   return trace_iterator;
 156 }
 157
 158 /** @brief returns the date of the next occurring event (pure function) */
 159 double simgrid::trace_mgr::future_evt_set::next_date() const
 160 {
 161   if (xbt_heap_size(p_heap))
 162     return (xbt_heap_maxkey(p_heap));
 163   else
 164     return -1.0;
 165 }
 166
 167 /** @brief Retrieves the next occurring event, or NULL if none happens before #date */
 168 tmgr_trace_iterator_t simgrid::trace_mgr::future_evt_set::pop_leq(
 169     double date, double *value, simgrid::surf::Resource **resource)
 170 {
 171   double event_date = next_date();
 172   if (event_date > date)
 173     return NULL;
 174
 175   tmgr_trace_iterator_t trace_iterator = (tmgr_trace_iterator_t)xbt_heap_pop(p_heap);
 176   if (trace_iterator == NULL)
 177     return NULL;
 178
 179   tmgr_trace_t trace = trace_iterator->trace;
 180   *resource = trace_iterator->resource;
 181
 182   tmgr_event_t event = (tmgr_event_t)xbt_dynar_get_ptr(trace->event_list, trace_iterator->idx);
 183
 184   *value = event->value;
 185
 186   if (trace_iterator->idx < xbt_dynar_length(trace->event_list) - 1) {
 187     xbt_heap_push(p_heap, trace_iterator, event_date + event->delta);
 188     trace_iterator->idx++;
 189   } else if (event->delta > 0) {        /* Last element, checking for periodicity */
 190     xbt_heap_push(p_heap, trace_iterator, event_date + event->delta);
 191     trace_iterator->idx = 1; /* not 0 as the first event is a placeholder to handle when events really start */
 192   } else {                      /* We don't need this trace_event anymore */
 193     trace_iterator->free_me = 1;
 194   }
 195
 196   return trace_iterator;
 197 }
 198
 199 void tmgr_finalize(void)
 200 {
 201   for (auto kv : trace_list)
 202     delete kv.second;
 203 }
 204
 205 void tmgr_trace_event_unref(tmgr_trace_iterator_t *trace_event)
 206 {
 207   if ((*trace_event)->free_me) {
 208     xbt_free(*trace_event);
 209     *trace_event = nullptr;
 210   }
 211 }