Rewrite the energy plugin using the extension mechanism

[simgrid.git] / src / surf / plugins / energy.hpp

diff --git a/src/surf/plugins/energy.hpp b/src/surf/plugins/energy.hpp
index 076c16e..04a413c 100644 (file)
--- a/src/surf/plugins/energy.hpp
+++ b/src/surf/plugins/energy.hpp
@@ -19,27 +19,23 @@ namespace energy {
 
 class XBT_PRIVATE HostEnergy;
 
 
 class XBT_PRIVATE HostEnergy;
 

-extern XBT_PRIVATE std::map<simgrid::surf::Host*, HostEnergy*> *surf_energy;
-

 class HostEnergy {
 public:
 class HostEnergy {
 public:

+  static simgrid::xbt::Extension<simgrid::Host, HostEnergy> EXTENSION_ID;

   typedef std::pair<double,double> power_range;
 
   typedef std::pair<double,double> power_range;
 

-  HostEnergy(simgrid::surf::Host *ptr);
+  HostEnergy(simgrid::Host *ptr);

   ~HostEnergy();
 
   double getCurrentWattsValue(double cpu_load);
   double getConsumedEnergy();
   double getWattMinAt(int pstate);
   double getWattMaxAt(int pstate);
   ~HostEnergy();
 
   double getCurrentWattsValue(double cpu_load);
   double getConsumedEnergy();
   double getWattMinAt(int pstate);
   double getWattMaxAt(int pstate);

-
-  void unref() {if (--refcount == 0) delete this;}
-  void ref() {refcount++;}
+  void update();

 
 private:
   void initWattsRangeList();
 
 private:
   void initWattsRangeList();

-  int refcount = 1;
-  simgrid::surf::Host *host;
+  simgrid::Host *host = nullptr;

   std::vector<power_range> power_range_watts_list;   /*< List of (min_power,max_power) pairs corresponding to each cpu pstate */
 public:
   double watts_off = 0.0; /*< Consumption when the machine is turned off (shutdown) */
   std::vector<power_range> power_range_watts_list;   /*< List of (min_power,max_power) pairs corresponding to each cpu pstate */
 public:
   double watts_off = 0.0; /*< Consumption when the machine is turned off (shutdown) */