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(event_list.empty())
  42     cb(event_list);
  43
  44   if(event_list.empty()) {
  45     event->free_me  = false;
  46   } else {
  47     //FIXME: This is a bug, but keep old behaviour for now
  48     //fes_->add_event(0, event);
  49     fes_->add_event(event_list[0].date_, event);
  50   }
  51   return event;
  52 }
  53
  54 /** @brief Gets the next event from a profile */
  55 DatedValue Profile::next(Event* event)
  56 {
  57   double event_date  = fes_->next_date();
  58
  59   DatedValue dateVal = event_list.at(event->idx);
  60
  61   event->idx++;
  62
  63   if (event->idx == event_list.size())
  64     cb(event_list);
  65   if(event->idx>=event_list.size())
  66     event->free_me = true;
  67   else {
  68     DatedValue& nextDateVal=event_list.at(event->idx);
  69     xbt_assert(nextDateVal.date_>=0);
  70     xbt_assert(nextDateVal.value_>=0);
  71     fes_->add_event(event_date +nextDateVal.date_, event);
  72   }
  73   return dateVal;
  74 }
  75
  76 Profile::Profile(const std::string& name, const std::function<ProfileBuilder::UpdateCb>& cb, double repeat_delay): name(name),cb(std::move(cb)),repeat_delay(repeat_delay) {
  77   xbt_assert(trace_list.find(name) == trace_list.end(), "Refusing to define trace %s twice", name.c_str());
  78   trace_list.insert({name,this});
  79   cb(event_list);
  80 }
  81
  82 } // namespace profile
  83 } // namespace kernel
  84 } // namespace simgrid
  85
  86 void tmgr_finalize()
  87 {
  88   for (auto const& kv : trace_list)
  89     delete kv.second;
  90   trace_list.clear();
  91 }
  92
  93 void tmgr_trace_event_unref(simgrid::kernel::profile::Event** event)
  94 {
  95   if ((*event)->free_me) {
  96     delete *event;
  97     *event = nullptr;
  98   }
  99 }