]> AND Public Git Repository - simgrid.git/blobdiff - src/kernel/resource/DiskImpl.cpp
Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
De-obfuscation.
[simgrid.git] / src / kernel / resource / DiskImpl.cpp
index 5ae459329ed621dace13e2a85026675f112a22ff..8a45da63199d031bc5f71c06dda005416f91dffb 100644 (file)
@@ -1,4 +1,4 @@
-/* Copyright (c) 2019-2021. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2019-2022. The SimGrid Team. All rights reserved.          */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
@@ -6,10 +6,18 @@
 #include "DiskImpl.hpp"
 
 #include "simgrid/s4u/Engine.hpp"
+#include "simgrid/sg_config.hpp"
 #include "src/kernel/EngineImpl.hpp"
 #include "src/kernel/lmm/maxmin.hpp"
+#include "src/kernel/resource/profile/Profile.hpp"
 
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(res_disk, ker_resource, "Disk resources, that fuel I/O activities");
+/***********
+ * Options *
+ ***********/
+static simgrid::config::Flag<std::string> cfg_disk_solver("disk/solver",
+                                                          "Set linear equations solver used by disk model", "maxmin",
+                                                          &simgrid::kernel::lmm::System::validate_solver);
 
 namespace simgrid {
 namespace kernel {
@@ -23,28 +31,25 @@ xbt::signal<void(DiskAction const&, Action::State, Action::State)> DiskAction::o
 
 DiskModel::DiskModel(const std::string& name) : Model(name)
 {
-  set_maxmin_system(new lmm::System(true /* selective update */));
+  set_maxmin_system(lmm::System::build(cfg_disk_solver, true /* selective update */));
 }
 
 /************
  * Resource *
  ************/
-DiskImpl* DiskImpl::set_host(s4u::Host* host)
+DiskImpl::DiskImpl(const std::string& name, double read_bandwidth, double write_bandwidth)
+    : Resource_T(name), piface_(this)
 {
-  xbt_assert(host, "Cannot set host, none given");
-  host_ = host;
-  return this;
+  read_bw_.peak   = read_bandwidth;
+  read_bw_.scale  = 1.0;
+  write_bw_.peak  = write_bandwidth;
+  write_bw_.scale = 1.0;
 }
 
-DiskImpl* DiskImpl::set_read_bandwidth(double read_bw)
-{
-  read_bw_ = read_bw;
-  return this;
-}
-
-DiskImpl* DiskImpl::set_write_bandwidth(double write_bw)
+DiskImpl* DiskImpl::set_host(s4u::Host* host)
 {
-  write_bw_ = write_bw;
+  xbt_assert(host, "Cannot set host, none given");
+  host_ = host;
   return this;
 }
 
@@ -70,16 +75,6 @@ void DiskImpl::destroy()
   delete this;
 }
 
-bool DiskImpl::is_used() const
-{
-  return get_model()->get_maxmin_system()->constraint_used(get_constraint());
-}
-
-void DiskImpl::apply_event(kernel::profile::Event* /*event*/, double /*value*/)
-{
-  THROW_UNIMPLEMENTED;
-}
-
 void DiskImpl::turn_on()
 {
   if (not is_on()) {
@@ -95,18 +90,84 @@ void DiskImpl::turn_off()
   }
 }
 
+DiskImpl* DiskImpl::set_read_bandwidth_profile(profile::Profile* profile)
+{
+  if (profile) {
+    xbt_assert(read_bw_.event == nullptr, "Cannot set a second read bandwidth profile to Disk %s", get_cname());
+    read_bw_.event = profile->schedule(&profile::future_evt_set, this);
+  }
+  return this;
+}
+
+DiskImpl* DiskImpl::set_write_bandwidth_profile(profile::Profile* profile)
+{
+  if (profile) {
+    xbt_assert(write_bw_.event == nullptr, "Cannot set a second read bandwidth profile to Disk %s", get_cname());
+    write_bw_.event = profile->schedule(&profile::future_evt_set, this);
+  }
+  return this;
+}
+
 void DiskImpl::seal()
 {
+  if (is_sealed())
+    return;
+
   xbt_assert(this->get_model(), "Cannot seal Disk (%s) without setting the model first", get_cname());
   lmm::System* maxmin_system = get_model()->get_maxmin_system();
-  this->set_read_constraint(maxmin_system->constraint_new(this, read_bw_))
-      ->set_write_constraint(maxmin_system->constraint_new(this, write_bw_))
-      ->set_constraint(maxmin_system->constraint_new(this, std::max(read_bw_, write_bw_)));
-  XBT_DEBUG("Create resource with read_bw '%f' write_bw '%f'", read_bw_, write_bw_);
+  /* set readwrite constraint if not configured by user */
+  if (readwrite_bw_ == -1) {
+    readwrite_bw_ = std::max(read_bw_.peak, write_bw_.peak);
+  }
+  this->set_read_constraint(maxmin_system->constraint_new(this, read_bw_.peak * read_bw_.scale))
+      ->set_write_constraint(maxmin_system->constraint_new(this, write_bw_.peak * write_bw_.scale))
+      ->set_constraint(maxmin_system->constraint_new(this, readwrite_bw_));
+  apply_sharing_policy_cfg();
+  XBT_DEBUG("Create resource with read_bw '%f' write_bw '%f'", read_bw_.peak, write_bw_.peak);
   Resource::seal();
   turn_on();
 }
 
+constexpr kernel::lmm::Constraint::SharingPolicy to_maxmin_policy(s4u::Disk::SharingPolicy policy)
+{
+  kernel::lmm::Constraint::SharingPolicy lmm_policy = kernel::lmm::Constraint::SharingPolicy::SHARED;
+  if (policy == s4u::Disk::SharingPolicy::NONLINEAR)
+    lmm_policy = kernel::lmm::Constraint::SharingPolicy::NONLINEAR;
+  return lmm_policy;
+}
+
+void DiskImpl::set_sharing_policy(s4u::Disk::Operation op, s4u::Disk::SharingPolicy policy,
+                                  const s4u::NonLinearResourceCb& cb)
+{
+  sharing_policy_[op]    = policy;
+  sharing_policy_cb_[op] = cb;
+  apply_sharing_policy_cfg();
+}
+
+s4u::Disk::SharingPolicy DiskImpl::get_sharing_policy(s4u::Disk::Operation op) const
+{
+  return sharing_policy_.at(op);
+}
+
+void DiskImpl::apply_sharing_policy_cfg()
+{
+  if (get_constraint())
+    get_constraint()->set_sharing_policy(to_maxmin_policy(sharing_policy_[s4u::Disk::Operation::READWRITE]),
+                                         sharing_policy_cb_[s4u::Disk::Operation::READWRITE]);
+  if (constraint_read_)
+    constraint_read_->set_sharing_policy(to_maxmin_policy(sharing_policy_[s4u::Disk::Operation::READ]),
+                                         sharing_policy_cb_[s4u::Disk::Operation::READ]);
+  if (constraint_write_)
+    constraint_write_->set_sharing_policy(to_maxmin_policy(sharing_policy_[s4u::Disk::Operation::WRITE]),
+                                          sharing_policy_cb_[s4u::Disk::Operation::WRITE]);
+}
+
+void DiskImpl::set_factor_cb(const std::function<s4u::Disk::IoFactorCb>& cb)
+{
+  xbt_assert(not is_sealed(), "Cannot set I/O factor callback in an already sealed disk(%s)", get_cname());
+  factor_cb_ = cb;
+}
+
 /**********
  * Action *
  **********/