src/kernel/resource/models/disk_s19.cpp

   1 /* Copyright (c) 2013-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/simgrid/sg_config.hpp"
   7 #include <simgrid/kernel/routing/NetPoint.hpp>
   8 #include <simgrid/kernel/routing/NetZoneImpl.hpp>
   9 #include <simgrid/s4u/Engine.hpp>
  10 #include <simgrid/s4u/Host.hpp>
  11
  12 #include "src/kernel/EngineImpl.hpp"
  13 #include "src/kernel/lmm/maxmin.hpp"
  14 #include "src/kernel/resource/models/disk_s19.hpp"
  15 #include "src/kernel/resource/profile/Event.hpp"
  16 #include "src/simgrid/module.hpp"
  17
  18 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(res_disk);
  19 /***********
  20  * Options *
  21  ***********/
  22 static simgrid::config::Flag<std::string> cfg_disk_solver("disk/solver",
  23                                                           "Set linear equations solver used by disk model", "maxmin",
  24                                                           &simgrid::kernel::lmm::System::validate_solver);
  25
  26 /*********
  27  * Model *
  28  *********/
  29
  30 SIMGRID_REGISTER_DISK_MODEL(S19, "Simplistic disk model.", []() {
  31   auto disk_model = std::make_shared<simgrid::kernel::resource::DiskS19Model>("Disk");
  32   auto* engine    = simgrid::kernel::EngineImpl::get_instance();
  33   engine->add_model(disk_model);
  34   engine->get_netzone_root()->set_disk_model(disk_model);
  35 });
  36
  37 namespace simgrid::kernel::resource {
  38 /*********
  39  * Model *
  40  *********/
  41
  42 DiskS19Model::DiskS19Model(const std::string& name) : DiskModel(name)
  43 {
  44   set_maxmin_system(lmm::System::build(cfg_disk_solver.get(), true /* selective update */));
  45 }
  46
  47 DiskImpl* DiskS19Model::create_disk(const std::string& name, double read_bandwidth, double write_bandwidth)
  48 {
  49   return (new DiskS19(name, read_bandwidth, write_bandwidth))->set_model(this);
  50 }
  51
  52 void DiskS19Model::update_actions_state(double /*now*/, double delta)
  53 {
  54   for (auto it = std::begin(*get_started_action_set()); it != std::end(*get_started_action_set());) {
  55     auto& action = *it;
  56     ++it; // increment iterator here since the following calls to action.finish() may invalidate it
  57     action.update_remains(rint(action.get_rate() * delta));
  58     action.update_max_duration(delta);
  59
  60     if (((action.get_remains_no_update() <= 0) && (action.get_variable()->get_penalty() > 0)) ||
  61         ((action.get_max_duration() != NO_MAX_DURATION) && (action.get_max_duration() <= 0))) {
  62       action.finish(Action::State::FINISHED);
  63     }
  64   }
  65 }
  66
  67 /************
  68  * Resource *
  69  ************/
  70
  71 void DiskS19::apply_event(kernel::profile::Event* triggered, double value)
  72 {
  73   /* Find out which of my iterators was triggered, and react accordingly */
  74   if (triggered == get_read_event()) {
  75     set_read_bandwidth(value);
  76     unref_read_event();
  77   } else if (triggered == get_write_event()) {
  78     set_write_bandwidth(value);
  79     unref_write_event();
  80   } else if (triggered == get_state_event()) {
  81     if (value > 0)
  82       turn_on();
  83     else
  84       turn_off();
  85     unref_state_event();
  86   } else {
  87     xbt_die("Unknown event!\n");
  88   }
  89
  90   XBT_DEBUG("There was a resource state event, need to update actions related to the constraint (%p)",
  91             get_constraint());
  92 }
  93
  94 /**********
  95  * Action *
  96  **********/
  97
  98 DiskS19Action::DiskS19Action(Model* model, double cost, bool failed)
  99     : DiskAction(model, cost, failed, model->get_maxmin_system()->variable_new(this, 1.0, -1.0, 3))
 100 {
 101 }
 102
 103 DiskAction* DiskS19::io_start(sg_size_t size, s4u::Io::OpType type)
 104 {
 105   auto* action = new DiskS19Action(get_model(), static_cast<double>(size), not is_on());
 106   get_model()->get_maxmin_system()->expand(get_constraint(), action->get_variable(), 1.0);
 107   switch (type) {
 108     case s4u::Io::OpType::READ:
 109       get_model()->get_maxmin_system()->expand(get_read_constraint(), action->get_variable(), 1.0);
 110       break;
 111     case s4u::Io::OpType::WRITE:
 112       get_model()->get_maxmin_system()->expand(get_write_constraint(), action->get_variable(), 1.0);
 113       break;
 114     default:
 115       THROW_UNIMPLEMENTED;
 116   }
 117   if (const auto& factor_cb = get_factor_cb()) { // handling disk variability
 118     action->set_rate_factor(factor_cb(size, type));
 119   }
 120   return action;
 121 }
 122
 123 } // namespace simgrid::kernel::resource