src/kernel/resource/profile/Profile.cpp

   1 /* Copyright (c) 2004-2022. The SimGrid Team. All rights reserved.          */
   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 "src/kernel/resource/profile/Profile.hpp"
   7 #include "xbt/asserts.h"
   8 #include "src/kernel/resource/profile/Event.hpp"
   9 #include "src/kernel/resource/profile/FutureEvtSet.hpp"
  10 #include "src/kernel/resource/profile/StochasticDatedValue.hpp"
  11 #include "src/surf/surf_interface.hpp"
  12
  13 #include <boost/algorithm/string.hpp>
  14 #include <fstream>
  15 #include <memory>
  16 #include <ostream>
  17 #include <sstream>
  18 #include <unordered_map>
  19 #include <utility>
  20 #include <vector>
  21 #include <string>
  22
  23 static std::unordered_map<std::string, simgrid::kernel::profile::Profile*> trace_list;
  24
  25 namespace simgrid {
  26 namespace kernel {
  27 namespace profile {
  28
  29 /** @brief Register this profile for that resource onto that FES,
  30  * and get an iterator over the integrated trace  */
  31 Event* Profile::schedule(FutureEvtSet* fes, resource::Resource* resource)
  32 {
  33   auto* event     = new Event();
  34   event->profile  = this;
  35   event->idx      = 0;
  36   event->resource = resource;
  37   event->free_me  = false;
  38
  39   fes_ = fes;
  40
  41   if (get_enough_events(0)) {
  42     fes_->add_event(event_list[0].date_, event);
  43   } else {
  44     event->free_me  = true;
  45     tmgr_trace_event_unref(&event);
  46   }
  47   return event;
  48 }
  49
  50 /** @brief Gets the next event from a profile */
  51 DatedValue Profile::next(Event* event)
  52 {
  53   double event_date  = fes_->next_date();
  54
  55   DatedValue dateVal = event_list.at(event->idx);
  56
  57   event->idx++;
  58
  59   if (get_enough_events(event->idx)) {
  60     const DatedValue& nextDateVal = event_list[event->idx];
  61     xbt_assert(nextDateVal.date_>=0);
  62     xbt_assert(nextDateVal.value_>=0);
  63     fes_->add_event(event_date +nextDateVal.date_, event);
  64   } else {
  65     event->free_me = true;
  66   }
  67   return dateVal;
  68 }
  69
  70 Profile::Profile(const std::string& name, const std::function<ProfileBuilder::UpdateCb>& cb, double repeat_delay)
  71     : name(name), cb(cb), repeat_delay(repeat_delay)
  72 {
  73   xbt_assert(trace_list.find(name) == trace_list.end(), "Refusing to define trace %s twice", name.c_str());
  74   trace_list.try_emplace(name, this);
  75   get_enough_events(0);
  76 }
  77
  78 } // namespace profile
  79 } // namespace kernel
  80 } // namespace simgrid
  81
  82 void tmgr_finalize()
  83 {
  84   for (auto const& [_, trace] : trace_list)
  85     delete trace;
  86   trace_list.clear();
  87 }
  88
  89 void tmgr_trace_event_unref(simgrid::kernel::profile::Event** event)
  90 {
  91   if ((*event)->free_me) {
  92     delete *event;
  93     *event = nullptr;
  94   }
  95 }