From b581df32aff459e913adc5a9bae9e9bf8bb51340 Mon Sep 17 00:00:00 2001 From: Capri2014 Date: Wed, 26 Aug 2020 19:44:11 -0700 Subject: [PATCH] Tools: update message_manager --- modules/tools/common/BUILD | 1 + modules/tools/common/message_manager.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/modules/tools/common/BUILD b/modules/tools/common/BUILD index 6fb989bc60..9dc5f1a12a 100644 --- a/modules/tools/common/BUILD +++ b/modules/tools/common/BUILD @@ -12,6 +12,7 @@ py_library( srcs = ["message_manager.py"], deps = [ ":proto_utils", + "//modules/audio/proto:audio_event_py_pb2", "//modules/canbus/proto:chassis_py_pb2", "//modules/common/proto:drive_event_py_pb2", "//modules/control/proto:control_cmd_py_pb2", diff --git a/modules/tools/common/message_manager.py b/modules/tools/common/message_manager.py index 03c0e45d00..af01d8326d 100644 --- a/modules/tools/common/message_manager.py +++ b/modules/tools/common/message_manager.py @@ -16,6 +16,7 @@ # limitations under the License. ############################################################################### +from modules.audio.proto import audio_event_pb2 from modules.localization.proto import localization_pb2 from modules.perception.proto import perception_obstacle_pb2 from modules.perception.proto import traffic_light_detection_pb2 @@ -31,6 +32,7 @@ from modules.guardian.proto import guardian_pb2 from modules.tools.common import proto_utils + class MessageType: def __init__(self, name, topic, msg_type): self.name = name @@ -50,6 +52,8 @@ class MessageType: topic_pb_list = [ + MessageType("audio_event", "/apollo/audio_event", + audio_event_pb2.AudioEvent), MessageType("planning", "/apollo/planning", planning_pb2.ADCTrajectory), MessageType("control", "/apollo/control", control_cmd_pb2.ControlCommand), MessageType("chassis", "/apollo/canbus/chassis", chassis_pb2.Chassis), -- GitLab