Add three more top-level messages

Signed-off-by: ClemensLinnhoff <clemens.linnhoff@partner.bmw.de>

Author: ClemensLinnhoff

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,

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
 }