| 126 | return self.position |
| 127 | |
| 128 | class MockVioOutput: |
| 129 | def __init__(self, data): |
| 130 | self.data = data |
| 131 | if "trackingStatus" in data: |
| 132 | status = data["trackingStatus"] |
| 133 | if status == 0: |
| 134 | self.status = spectacularAI.TrackingStatus.INIT |
| 135 | elif status == 1: |
| 136 | self.status = spectacularAI.TrackingStatus.TRACKING |
| 137 | elif status == 2: |
| 138 | self.status = spectacularAI.TrackingStatus.LOST_TRACKING |
| 139 | else: |
| 140 | raise ValueError("Unknown tracking status: {0}".format(status)) |
| 141 | else: |
| 142 | # Support older versions of cpp serialization |
| 143 | self.status = spectacularAI.TrackingStatus.TRACKING |
| 144 | |
| 145 | def getCameraPose(self, index): |
| 146 | return MockCameraPose(self.data["cameraPoses"][index]) |
| 147 | |
| 148 | class MockFrame: |
| 149 | def __init__(self, data): |