From e597303e2fe7f273a967abf45b3d18a7c91159fc Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Fri, 20 Dec 2024 20:30:40 +0900
Subject: [PATCH] add steps to Latency

---
 simlib/blendnet-sims/scripts/latency.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/simlib/blendnet-sims/scripts/latency.py b/simlib/blendnet-sims/scripts/latency.py
index e56f179..5d603ab 100755
--- a/simlib/blendnet-sims/scripts/latency.py
+++ b/simlib/blendnet-sims/scripts/latency.py
@@ -21,21 +21,17 @@ class Event:
 class Latency:
     start_event: Event
     end_event: Optional[Event] = None
+    steps: Optional[int] = None
 
     def finish(self, event: Event):
         assert self.end_event is None
         assert event.step_id >= self.start_event.step_id
         self.end_event = event
+        self.steps = self.end_event.step_id - self.start_event.step_id
 
     def finished(self) -> bool:
         return self.end_event is not None
 
-    @property
-    def value(self) -> Optional[int]:
-        if self.end_event is None:
-            return None
-        return self.end_event.step_id - self.start_event.step_id
-
 
 @dataclass
 class Message:
@@ -77,7 +73,7 @@ class Message:
 
     @property
     def latency(self) -> Optional[int]:
-        return self.total_latency.value
+        return self.total_latency.steps
 
     def __eq__(self, other):
         if not isinstance(other, Message):