1 files changed, 8 insertions, 6 deletions
diff --git a/ns3-simulations/ns3-simulator/modules/energy.cc b/ns3-simulations/ns3-simulator/modules/energy.cc
index 64d30e1..11f1c7a 100644
--- a/ns3-simulations/ns3-simulator/modules/energy.cc
+++ b/ns3-simulations/ns3-simulator/modules/energy.cc
@@ -22,15 +22,17 @@ void setupCellEnergy(Cell cell){
 
   
   // Trace
-  DeviceEnergyModelContainer energyModels(edgeApDeviceModels, edgeDeviceModels);
-  DeviceEnergyModelContainer::Iterator it=energyModels.Begin();
-
-  int i=0;
-  while(it!=energyModels.End()){
-    (*it)->TraceConnect ("TotalEnergyConsumption", std::to_string(i),MakeCallback (&EnergyUpdated));
+  DeviceEnergyModelContainer::Iterator it=edgeDeviceModels.Begin();
+  int i=1; // Node 0 will be AP, other node will have negative id (cf following while)
+           // This is usefull in logs, in fact ECOFEN nodes will have positive ID and WIFI energy nodes negative id
+           // AP will have id 0 in ECOFEN and WIFI (in order to combine their energy value when parsing logs
+  while(it!=edgeDeviceModels.End()){
+    (*it)->TraceConnect ("TotalEnergyConsumption", std::to_string(0-i),MakeCallback (&EnergyUpdated));
     it++;
     i++;
   }
+  // AP will have id 0
+  (*edgeApDeviceModels.Begin())->TraceConnect ("TotalEnergyConsumption", std::to_string(0),MakeCallback (&EnergyUpdated));
 
  //   Ptr<BasicEnergySource> basicSourcePtr0 = DynamicCast<BasicEnergySource> (wifiEdgeNodesSources.Get (0));
   // //basicSourcePtr0->TraceConnectWithoutContext ("RemainingEnergy", MakeCallback (&RemainingEnergy));