-static void update_consumption(simgrid::surf::Host *host, HostEnergy *host_energy) {
- double start_time = host_energy->last_updated;
- double finish_time = surf_get_clock();
- double cpu_load;
- if (host->p_cpu->m_speedPeak == 0)
- // Some users declare a pstate of speed 0 flops (eg to model boot time).
- // We consider that the machine is then fully loaded. That's arbitrary but it avoids a NaN
- cpu_load = 1;
- else
- cpu_load = lmm_constraint_get_usage(host->p_cpu->getConstraint()) / host->p_cpu->m_speedPeak;
-
- double previous_energy = host_energy->total_energy;
-
- double instantaneous_consumption;
- if (host->getState() == SURF_RESOURCE_OFF)
- instantaneous_consumption = host_energy->watts_off;
- else
- instantaneous_consumption = host_energy->getCurrentWattsValue(cpu_load);
-
- double energy_this_step = instantaneous_consumption*(finish_time-start_time);
-
- host_energy->total_energy = previous_energy + energy_this_step;
- host_energy->last_updated = finish_time;
-
- XBT_DEBUG("[cpu_update_energy] period=[%.2f-%.2f]; current power peak=%.0E flop/s; consumption change: %.2f J -> %.2f J",
- start_time, finish_time, host->p_cpu->m_speedPeak, previous_energy, energy_this_step);
-}
-
-/** \ingroup SURF_plugin_energy
- * \brief Enable energy plugin
- * \details Enable energy plugin to get joules consumption of each cpu. You should call this function before #MSG_init().
- */
-void sg_energy_plugin_init() {
- if (simgrid::energy::surf_energy == NULL) {
-
- simgrid::energy::surf_energy = new std::map<simgrid::surf::Host*, simgrid::energy::HostEnergy*>();
-
- /* The following attaches an anonymous function to the Host::onCreation signal */
- /* Search for "C++ lambda" for more information on the syntax used here */
- simgrid::surf::Host::onCreation.connect([](simgrid::surf::Host *host) {
- (*surf_energy)[host] = new HostEnergy(host);
- });
-
- simgrid::surf::VMCreatedCallbacks.connect([](simgrid::surf::VirtualMachine* vm) {
- std::map<simgrid::surf::Host*, HostEnergy*>::iterator host_energy_it =
- surf_energy->find(vm->p_hostPM->extension(simgrid::surf::Host::EXTENSION_ID));
- xbt_assert(host_energy_it != surf_energy->end(), "The host is not in surf_energy.");
- (*surf_energy)[vm] = host_energy_it->second;
- host_energy_it->second->ref(); // protect the HostEnergy from getting deleted too early
- });
-
- simgrid::surf::Host::onDestruction.connect([](simgrid::surf::Host *host) {
- std::map<simgrid::surf::Host*, HostEnergy*>::iterator host_energy_it = surf_energy->find(host);
- xbt_assert(host_energy_it != surf_energy->end(), "The host is not in surf_energy.");
-
- HostEnergy *host_energy = host_energy_it->second;
- update_consumption(host, host_energy);
-
- if (host_energy_it->second->refcount == 1) // Don't display anything for virtual CPUs
- XBT_INFO("Total energy of host %s: %f Joules", host->getName(), host_energy->getConsumedEnergy());
- host_energy_it->second->unref();
- surf_energy->erase(host_energy_it);
- });
- simgrid::surf::CpuAction::onStateChange.connect([](simgrid::surf::CpuAction *action,
- e_surf_action_state_t old,
- e_surf_action_state_t cur) {
- const char *name = getActionCpu(action)->getName();
- simgrid::surf::Host *host = static_cast<simgrid::surf::Host*>(surf_host_resource_priv(sg_host_by_name(name)));
-
- HostEnergy *host_energy = (*surf_energy)[host];
-
- if(host_energy->last_updated < surf_get_clock())
- update_consumption(host, host_energy);
-
- });
-
- simgrid::surf::Host::onStateChange.connect([]
- (simgrid::surf::Host *host,
- e_surf_resource_state_t oldState,
- e_surf_resource_state_t newState) {
- HostEnergy *host_energy = (*surf_energy)[host];
-
- if(host_energy->last_updated < surf_get_clock())
- update_consumption(host, host_energy);
- });
-
- simgrid::surf::surfExitCallbacks.connect([]() {
- delete surf_energy;
- surf_energy = NULL;
- });
- }
+void HostEnergy::update()
+{
+ simgrid::surf::HostImpl* surf_host = host->extension<simgrid::surf::HostImpl>();
+ double start_time = this->last_updated;
+ double finish_time = surf_get_clock();
+ double cpu_load;
+ if (surf_host->p_cpu->speed_.peak == 0)
+ // Some users declare a pstate of speed 0 flops (eg to model boot time).
+ // We consider that the machine is then fully loaded. That's arbitrary but it avoids a NaN
+ cpu_load = 1;
+ else
+ cpu_load = lmm_constraint_get_usage(surf_host->p_cpu->getConstraint())
+ / surf_host->p_cpu->speed_.peak;
+
+ if (cpu_load > 1) // A machine with a load > 1 consumes as much as a fully loaded machine, not mores
+ cpu_load = 1;
+
+ double previous_energy = this->total_energy;
+
+ double instantaneous_consumption;
+ if (host->isOff())
+ instantaneous_consumption = this->watts_off;
+ else
+ instantaneous_consumption = this->getCurrentWattsValue(cpu_load);
+
+ double energy_this_step = instantaneous_consumption*(finish_time-start_time);
+
+ this->total_energy = previous_energy + energy_this_step;
+ this->last_updated = finish_time;
+
+ XBT_DEBUG("[update_energy of %s] period=[%.2f-%.2f]; current power peak=%.0E flop/s; consumption change: %.2f J -> %.2f J",
+ surf_host->getName(), start_time, finish_time, surf_host->p_cpu->speed_.peak, previous_energy, energy_this_step);