Improve debug

1 files changed, 5 insertions, 1 deletions

diff --git a/esds/simulator.py b/esds/simulator.py
index 2ea77b9..fb52fec 100644
--- a/esds/simulator.py
+++ b/esds/simulator.py
@@ -50,6 +50,7 @@ class Simulator:
         self.interferences=True
         self.wait_end_nodes=list() # Keep track of nodes that wait for the end of the simulation
         self.time_truncated=format(self.time,self.precision) # Truncated version is used in log print
+        self.debug=None # No debug by default
 
     def update_network(self,netmat):
         for event in self.events:
@@ -85,7 +86,10 @@ class Simulator:
 
     def log(self,msg,node=None):
         src = "esds" if node is None else "n"+str(node)
-        print("[t="+str(self.time_truncated)+",src="+src+"] "+msg)
+        logline="[t="+str(self.time_truncated)+",src="+src+"] "+msg
+        if self.debug is not None:
+            self.debug.append_log(logline)
+        print(logline)
 
     def sort_events(self):
         """