From 17b07d91462fc94326cb09c7dfe45dc6551928a2 Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Mon, 25 Apr 2022 09:46:21 +0200 Subject: [PATCH] Added option to hide objects --- display/dislpay_2d.py | 16 ++++++++++++---- examples/gps_fusion.py | 2 -- 2 files changed, 12 insertions(+), 6 deletions(-) diff --git a/display/dislpay_2d.py b/display/dislpay_2d.py index 747af87..cad35cf 100644 --- a/display/dislpay_2d.py +++ b/display/dislpay_2d.py @@ -10,9 +10,9 @@ class Displayer: self._fig.set_figheight(height) self._frame_interval = frame_interval - self._object_map = {} - self._circle_map = {} - self._objects = [] + self._object_map = {} + self._circle_map = {} + self._objects = [] self._xlim = [0, 20] self._ylim = [0, 20] @@ -38,7 +38,7 @@ class Displayer: def register_object(self, obj_name, obj_color='r'): self._object_map[obj_name], = plt.plot([], [], color=obj_color, marker='o', label=obj_name) - self._circle_map[obj_name] = plt.Circle((0, 0), 1, color=obj_color, fill=False) + self._circle_map[obj_name] = plt.Circle((0, 0), 1, color=obj_color, fill=False) self._objects.append(self._object_map[obj_name]) self._objects.append(self._circle_map[obj_name]) @@ -54,6 +54,14 @@ class Displayer: def set_circle_radius(self, obj_name, r): self._circle_map[obj_name].radius = r + def hide_object(self, obj_name): + self._object_map[obj_name].set_alpha(0) + self._circle_map[obj_name].set_alpha(0) + + def show_object(self, obj_name): + self._object_map[obj_name].set_alpha(1) + self._circle_map[obj_name].set_alpha(1) + # Start the animation diff --git a/examples/gps_fusion.py b/examples/gps_fusion.py index 15d611a..26bce2d 100644 --- a/examples/gps_fusion.py +++ b/examples/gps_fusion.py @@ -51,8 +51,6 @@ def run(kalman_filter_t): # Show results - #print(P_kalman[:, 0]) - def update(disp, t): disp.move_object("measurement", [x_measurement[:, 0][t], 0]) disp.set_circle_radius("measurement", std_dist[0])