From d9c9e8dd601d4a7e8318ba7b5dc71bc89457fec7 Mon Sep 17 00:00:00 2001 From: kaatrasa Date: Fri, 22 Dec 2023 12:49:21 +0200 Subject: [PATCH] Serialize tracking status in MockVioOutput --- python/cli/visualization/serialization.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/python/cli/visualization/serialization.py b/python/cli/visualization/serialization.py index eaf3148..7598798 100644 --- a/python/cli/visualization/serialization.py +++ b/python/cli/visualization/serialization.py @@ -92,6 +92,20 @@ def getPosition(self): class MockVioOutput: def __init__(self, data): self.data = data + if "trackingStatus" in data: + status = data["trackingStatus"] + if status == 0: + self.status = spectacularAI.TrackingStatus.INIT + elif status == 1: + self.status = spectacularAI.TrackingStatus.TRACKING + elif status == 2: + self.status = spectacularAI.TrackingStatus.LOST_TRACKING + else: + raise ValueError("Unknown tracking status: {0}".format(status)) + else: + # Support older versions of cpp serialization + self.status = spectacularAI.TrackingStatus.TRACKING + def getCameraPose(self, index): return MockCameraPose(self.data["cameraPoses"][index])