mirror of
https://github.com/PX4/PX4-Autopilot.git
synced 2026-06-02 20:28:37 +08:00
tight the uorb rtps classifier
This commit is contained in:
@@ -47,20 +47,12 @@ class Classifier():
|
|||||||
def __init__(self, yaml_file, msg_folder):
|
def __init__(self, yaml_file, msg_folder):
|
||||||
self.msg_id_map = self.parse_yaml_msg_id_file(yaml_file)
|
self.msg_id_map = self.parse_yaml_msg_id_file(yaml_file)
|
||||||
self.msg_folder = msg_folder
|
self.msg_folder = msg_folder
|
||||||
self.msgs_to_send = {}
|
self.msgs_to_send = self.set_msgs_to_send()
|
||||||
self.msgs_to_receive = {}
|
self.msgs_to_receive = self.set_msgs_to_receive()
|
||||||
self.msgs_to_ignore = {}
|
self.msgs_to_ignore = self.set_msgs_to_ignore()
|
||||||
self.msg_files_send = []
|
self.msg_files_send = self.set_msg_files_send()
|
||||||
self.msg_files_receive = []
|
self.msg_files_receive = self.set_msg_files_receive()
|
||||||
self.msg_files_ignore = []
|
self.msg_files_ignore = self.set_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()
|
|
||||||
|
|
||||||
# getters
|
# getters
|
||||||
@property
|
@property
|
||||||
@@ -97,27 +89,33 @@ class Classifier():
|
|||||||
|
|
||||||
# setters (for class init)
|
# setters (for class init)
|
||||||
def set_msgs_to_send(self):
|
def set_msgs_to_send(self):
|
||||||
|
send = {}
|
||||||
for dict in self.msg_id_map['rtps']['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):
|
def set_msgs_to_receive(self):
|
||||||
|
receive = {}
|
||||||
for dict in self.msg_id_map['rtps']['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):
|
def set_msgs_to_ignore(self):
|
||||||
|
ignore = {}
|
||||||
for dict in self.msg_id_map['rtps']['unclassified']:
|
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):
|
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()]
|
for msg in self.msgs_to_send.keys()]
|
||||||
|
|
||||||
def set_msg_files_receive(self):
|
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()]
|
for msg in self.msgs_to_receive.keys()]
|
||||||
|
|
||||||
def set_msg_files_ignore(self):
|
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()]
|
for msg in self.msgs_to_ignore.keys()]
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
|
|||||||
@@ -6,6 +6,8 @@ rtps:
|
|||||||
id: 6
|
id: 6
|
||||||
- msg: differential_pressure
|
- msg: differential_pressure
|
||||||
id: 15
|
id: 15
|
||||||
|
- msg: esc_report
|
||||||
|
id: 21
|
||||||
- msg: esc_status
|
- msg: esc_status
|
||||||
id: 22
|
id: 22
|
||||||
- msg: estimator_status
|
- msg: estimator_status
|
||||||
@@ -105,8 +107,6 @@ rtps:
|
|||||||
id: 19
|
id: 19
|
||||||
- msg: ekf_gps_position
|
- msg: ekf_gps_position
|
||||||
id: 20
|
id: 20
|
||||||
- msg: esc_report
|
|
||||||
id: 21
|
|
||||||
- msg: follow_target
|
- msg: follow_target
|
||||||
id: 24
|
id: 24
|
||||||
- msg: geofence_result
|
- msg: geofence_result
|
||||||
|
|||||||
Reference in New Issue
Block a user