Factor common code.

diff --git a/src/surf/disk_s19.cpp b/src/surf/disk_s19.cpp
index ede1c45..2ccb39e 100644 (file)
--- a/src/surf/disk_s19.cpp
+++ b/src/surf/disk_s19.cpp
@@ -71,15 +71,8 @@ DiskAction* DiskS19Model::io_start(const DiskImpl* disk, sg_size_t size, s4u::Io
 /************
  * Resource *
  ************/
-void DiskS19::set_read_bandwidth(double value)
+void DiskS19::update_penalties(double delta)
 {
-  read_bw_.peak = value;
-
-  get_model()->get_maxmin_system()->update_constraint_bound(get_constraint(),
-                                                            sg_bandwidth_factor * (read_bw_.peak * read_bw_.scale));
-
-  double delta = 1.0 / value - 1.0 / (read_bw_.peak * read_bw_.scale);
-
   const kernel::lmm::Element* elem     = nullptr;
   const kernel::lmm::Element* nextelem = nullptr;
   int numelem                          = 0;
@@ -91,6 +84,17 @@ void DiskS19::set_read_bandwidth(double value)
   }
 }
 
+void DiskS19::set_read_bandwidth(double value)
+{
+  read_bw_.peak = value;
+
+  get_model()->get_maxmin_system()->update_constraint_bound(get_constraint(),
+                                                            sg_bandwidth_factor * (read_bw_.peak * read_bw_.scale));
+
+  double delta = 1.0 / value - 1.0 / (read_bw_.peak * read_bw_.scale);
+  update_penalties(delta);
+}
+
 void DiskS19::set_write_bandwidth(double value)
 {
   write_bw_.peak = value;
@@ -99,16 +103,7 @@ void DiskS19::set_write_bandwidth(double value)
                                                             sg_bandwidth_factor * (write_bw_.peak * write_bw_.scale));
 
   double delta = 1.0 / value - 1.0 / (write_bw_.peak * write_bw_.scale);
-
-  const kernel::lmm::Element* elem     = nullptr;
-  const kernel::lmm::Element* nextelem = nullptr;
-  int numelem                          = 0;
-  while (const auto* var = get_constraint()->get_variable_safe(&elem, &nextelem, &numelem)) {
-    auto* action = static_cast<DiskS19Action*>(var->get_id());
-    action->sharing_penalty_ += delta;
-    if (not action->is_suspended())
-      get_model()->get_maxmin_system()->update_variable_penalty(action->get_variable(), action->sharing_penalty_);
-  }
+  update_penalties(delta);
 }
 
 void DiskS19::apply_event(kernel::profile::Event* triggered, double value)

diff --git a/src/surf/disk_s19.hpp b/src/surf/disk_s19.hpp
index 609c7d9..09c0c86 100644 (file)
--- a/src/surf/disk_s19.hpp
+++ b/src/surf/disk_s19.hpp
@@ -41,6 +41,8 @@ public:
  ************/
 
 class DiskS19 : public DiskImpl {
+  void update_penalties(double delta);
+
 public:
   using DiskImpl::DiskImpl;
   void set_read_bandwidth(double value) override;