Add three more top-level messages

Signed-off-by: ClemensLinnhoff <[email protected]>
OpenSimulationInterface · May 27, 2024 · 27df0e4 · 27df0e4
1 parent 2c3953b
commit 27df0e4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/osi3trace/osi2read.py b/osi3trace/osi2read.py
@@ -28,7 +28,7 @@ def command_line_arguments():
         "--type",
         "-t",
         help="Name of the type used to serialize data.",
-        choices=["SensorView", "GroundTruth", "SensorData"],
+        choices=["SensorView", "GroundTruth", "SensorData", "TrafficUpdate", "TrafficCommand", "TrafficCommandUpdate"],
         default="SensorView",
         type=str,
         required=False,

diff --git a/osi3trace/osi_trace.py b/osi3trace/osi_trace.py
@@ -8,12 +8,18 @@
 from osi3.osi_sensorview_pb2 import SensorView
 from osi3.osi_groundtruth_pb2 import GroundTruth
 from osi3.osi_sensordata_pb2 import SensorData
+from osi3.osi_trafficupdate_pb2 import TrafficUpdate
+from osi3.osi_trafficcommand_pb2 import TrafficCommand
+from osi3.osi_trafficcommandupdate_pb2 import TrafficCommandUpdate
 
 
 MESSAGES_TYPE = {
     "SensorView": SensorView,
     "GroundTruth": GroundTruth,
     "SensorData": SensorData,
+    "TrafficUpdate": TrafficUpdate,
+    "TrafficCommand": TrafficCommand,
+    "TrafficCommandUpdate": TrafficCommandUpdate
 }