src/kernel/resource/profile/Profile.cpp

   1 /* Copyright (c) 2004-2023. 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::kernel::profile {
  26
  27 /** @brief Register this profile for that resource onto that FES,
  28  * and get an iterator over the integrated trace  */
  29 Event* Profile::schedule(FutureEvtSet* fes, resource::Resource* resource)
  30 {
  31   auto* event     = new Event();
  32   event->profile  = this;
  33   event->idx      = 0;
  34   event->resource = resource;
  35   event->free_me  = false;
  36
  37   fes_ = fes;
  38
  39   if (get_enough_events(0)) {
  40     fes_->add_event(event_list[0].date_, event);
  41   } else {
  42     event->free_me  = true;
  43     tmgr_trace_event_unref(&event);
  44   }
  45   return event;
  46 }
  47
  48 /** @brief Gets the next event from a profile */
  49 DatedValue Profile::next(Event* event)
  50 {
  51   double event_date  = fes_->next_date();
  52
  53   DatedValue dateVal = event_list.at(event->idx);
  54
  55   event->idx++;
  56
  57   if (get_enough_events(event->idx)) {
  58     const DatedValue& nextDateVal = event_list[event->idx];
  59     xbt_assert(nextDateVal.date_>=0);
  60     xbt_assert(nextDateVal.value_>=0);
  61     fes_->add_event(event_date +nextDateVal.date_, event);
  62   } else {
  63     event->free_me = true;
  64   }
  65   return dateVal;
  66 }
  67
  68 Profile::Profile(const std::string& name, const std::function<ProfileBuilder::UpdateCb>& cb, double repeat_delay)
  69     : name(name), cb(cb), repeat_delay(repeat_delay)
  70 {
  71   xbt_assert(trace_list.find(name) == trace_list.end(), "Refusing to define trace %s twice", name.c_str());
  72   trace_list.try_emplace(name, this);
  73   get_enough_events(0);
  74 }
  75
  76 } // namespace simgrid::kernel::profile
  77
  78 void tmgr_finalize()
  79 {
  80   for (auto const& [_, trace] : trace_list)
  81     delete trace;
  82   trace_list.clear();
  83 }
  84
  85 void tmgr_trace_event_unref(simgrid::kernel::profile::Event** event)
  86 {
  87   if ((*event)->free_me) {
  88     delete *event;
  89     *event = nullptr;
  90   }
  91 }