diff --git a/msg/tools/uorb_rtps_classifier.py b/msg/tools/uorb_rtps_classifier.py index c588c5ad22..52ab3d5d35 100644 --- a/msg/tools/uorb_rtps_classifier.py +++ b/msg/tools/uorb_rtps_classifier.py @@ -47,20 +47,12 @@ class Classifier(): def __init__(self, yaml_file, msg_folder): self.msg_id_map = self.parse_yaml_msg_id_file(yaml_file) self.msg_folder = msg_folder - self.msgs_to_send = {} - self.msgs_to_receive = {} - self.msgs_to_ignore = {} - self.msg_files_send = [] - self.msg_files_receive = [] - self.msg_files_ignore = [] - - # init class attributtes - self.set_msgs_to_send() - self.set_msgs_to_receive() - self.set_msgs_to_ignore() - self.set_msg_files_send() - self.set_msg_files_receive() - self.set_msg_files_ignore() + self.msgs_to_send = self.set_msgs_to_send() + self.msgs_to_receive = self.set_msgs_to_receive() + self.msgs_to_ignore = self.set_msgs_to_ignore() + self.msg_files_send = self.set_msg_files_send() + self.msg_files_receive = self.set_msg_files_receive() + self.msg_files_ignore = self.set_msg_files_ignore() # getters @property @@ -97,27 +89,33 @@ class Classifier(): # setters (for class init) def set_msgs_to_send(self): + send = {} for dict in self.msg_id_map['rtps']['send']: - self.msgs_to_send.update({dict['msg']: dict['id']}) + send.update({dict['msg']: dict['id']}) + return send def set_msgs_to_receive(self): + receive = {} for dict in self.msg_id_map['rtps']['receive']: - self.msgs_to_receive.update({dict['msg']: dict['id']}) + receive.update({dict['msg']: dict['id']}) + return receive def set_msgs_to_ignore(self): + ignore = {} for dict in self.msg_id_map['rtps']['unclassified']: - self.msgs_to_ignore.update({dict['msg']: dict['id']}) + ignore.update({dict['msg']: dict['id']}) + return ignore def set_msg_files_send(self): - self.msgs_files_send = [os.path.dirname(os.path.dirname(os.path.join(self.msg_folder, msg + ".msg"))) + return [os.path.join(self.msg_folder, msg + ".msg") for msg in self.msgs_to_send.keys()] def set_msg_files_receive(self): - self.msgs_files_receive = [os.path.dirname(os.path.dirname(os.path.join(self.msg_folder, msg + ".msg"))) + return [os.path.join(self.msg_folder, msg + ".msg") for msg in self.msgs_to_receive.keys()] def set_msg_files_ignore(self): - self.msgs_files_ignore = [os.path.dirname(os.path.dirname(os.path.join(self.msg_folder, msg + ".msg"))) + return [os.path.join(self.msg_folder, msg + ".msg") for msg in self.msgs_to_ignore.keys()] @staticmethod diff --git a/msg/tools/uorb_rtps_message_ids.yaml b/msg/tools/uorb_rtps_message_ids.yaml index 722e61bac4..dcc3209c46 100644 --- a/msg/tools/uorb_rtps_message_ids.yaml +++ b/msg/tools/uorb_rtps_message_ids.yaml @@ -6,6 +6,8 @@ rtps: id: 6 - msg: differential_pressure id: 15 + - msg: esc_report + id: 21 - msg: esc_status id: 22 - msg: estimator_status @@ -105,8 +107,6 @@ rtps: id: 19 - msg: ekf_gps_position id: 20 - - msg: esc_report - id: 21 - msg: follow_target id: 24 - msg: geofence_result