From d038f70eae4a2410b0446bafdfbf6a9905c7b573 Mon Sep 17 00:00:00 2001
From: Martin Quinson <martin.quinson@ens-rennes.fr>
Date: Fri, 10 Jul 2020 00:35:49 +0200
Subject: [PATCH] Add a WifiLink::get_host_count() function that is needed by
 the wifi energy model

---
 src/surf/network_wifi.cpp | 9 +++++++--
 src/surf/network_wifi.hpp | 1 +
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/surf/network_wifi.cpp b/src/surf/network_wifi.cpp
index 6d90be5044..838c1507fc 100644
--- a/src/surf/network_wifi.cpp
+++ b/src/surf/network_wifi.cpp
@@ -62,10 +62,15 @@ s4u::Link::SharingPolicy NetworkWifiLink::get_sharing_policy()
   return s4u::Link::SharingPolicy::WIFI;
 }
 
+int NetworkWifiLink::get_host_count()
+{
+  return host_rates_.size();
+}
+
 void NetworkWifiLink::refresh_decay_bandwidths(){
   // Compute number of STAtion on the Access Point
-  int nSTA=host_rates_.size();
-    
+  int nSTA = get_host_count();
+
   std::vector<Metric> new_bandwidths;
   for (auto bandwidth : bandwidths_){
     // Instantiate decay model relatively to the actual bandwidth
diff --git a/src/surf/network_wifi.hpp b/src/surf/network_wifi.hpp
index 08699dcee2..8ef62e877c 100644
--- a/src/surf/network_wifi.hpp
+++ b/src/surf/network_wifi.hpp
@@ -55,6 +55,7 @@ public:
   void set_latency(double) override { THROW_UNIMPLEMENTED; }
   void refresh_decay_bandwidths();
   bool toggle_decay_model();
+  int get_host_count();
 };
 
 class NetworkWifiAction : public NetworkCm02Action {
-- 
2.20.1