an.tsouchlos/kalman_filter
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added option to hide objects

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-04-25 09:46:21 +02:00
parent 0c992ec1e0
commit 17b07d9146
2 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
display/dislpay_2d.py
View File

@ -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

2
examples/gps_fusion.py
View File

@ -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])