add core_count setter

diff --git a/include/simgrid/s4u/Host.hpp b/include/simgrid/s4u/Host.hpp
index 6988a6f..f50f148 100644 (file)
--- a/include/simgrid/s4u/Host.hpp
+++ b/include/simgrid/s4u/Host.hpp
@@ -128,8 +128,11 @@ public:
    * This accounts for external load (see @ref simgrid::surf::Cpu::set_speed_profile()).
    */
   double get_available_speed() const;
+
   /** Returns the number of core of the processor. */
   int get_core_count() const;
+  Host* set_core_count(int core_count);
+
   /** Returns the current computation load (in flops per second)
    *
    * The external load (coming from an availability trace) is not taken in account.

diff --git a/src/s4u/s4u_Host.cpp b/src/s4u/s4u_Host.cpp
index 39234b5..1036f39 100644 (file)
--- a/src/s4u/s4u_Host.cpp
+++ b/src/s4u/s4u_Host.cpp
@@ -251,6 +251,12 @@ int Host::get_core_count() const
   return this->pimpl_cpu->get_core_count();
 }
 
+Host* Host::set_core_count(int core_count)
+{
+  this->pimpl_cpu->set_core_count(core_count);
+  return this;
+}
+
 /** @brief Set the pstate at which the host should run */
 void Host::set_pstate(int pstate_index)
 {

diff --git a/src/surf/cpu_interface.cpp b/src/surf/cpu_interface.cpp
index 3687e16..cd1fa08 100644 (file)
--- a/src/surf/cpu_interface.cpp
+++ b/src/surf/cpu_interface.cpp
@@ -129,6 +129,13 @@ void Cpu::on_speed_change()
   s4u::Host::on_speed_change(*host_);
 }
 
+Cpu* Cpu::set_core_count(int core_count)
+{
+  xbt_assert(core_count > 0, "Host %s must have at least one core, not 0.", host_->get_cname());
+  core_count_ = core_count;
+  return this;
+}
+
 int Cpu::get_core_count()
 {
   return core_count_;

diff --git a/src/surf/cpu_interface.hpp b/src/surf/cpu_interface.hpp
index 37202d3..fa856e7 100644 (file)
--- a/src/surf/cpu_interface.hpp
+++ b/src/surf/cpu_interface.hpp
@@ -112,6 +112,7 @@ public:
    */
   virtual CpuAction* sleep(double duration) = 0;
 
+  Cpu* set_core_count(int core_count);
   /** @brief Get the amount of cores */
   virtual int get_core_count();