Changed number of iterations for animation; fixed bug in Displayer calass

This commit is contained in:
Andreas Tsouchlos 2022-04-28 11:14:39 +02:00
parent 3f4c8b2a51
commit a9269caa46
2 changed files with 7 additions and 6 deletions

View File

@ -23,6 +23,7 @@ class Displayer:
for key in self._circle_map:
self._ax.add_patch(self._circle_map[key])
self.show_object(key)
return self._objects

View File

@ -6,8 +6,8 @@ from kalman_filter.display.display_2d import Displayer
def run(kalman_filter_t):
# Simulation parameter definitions
n_iterations = 500
std_dev1 = 0.2
n_iterations = 60
std_dev1 = 1
std_dev2 = 10
x_offset2 = 6
x_ground_truth = 50
@ -70,10 +70,10 @@ def run(kalman_filter_t):
disp = Displayer(width=8, height=8, frame_interval=150)
disp.set_axis_limits(xlim=[35, 75], ylim=[-15, 25])
disp.register_object("measurement1", "purple")
disp.register_object("measurement2", "pink")
disp.register_object("filtered", "blue")
disp.register_object("ground_truth", "green")
disp.register_object("measurement1", "purple")
disp.register_object("measurement2", "pink")
#disp.animate(n_steps=n_iterations, anim_callback=update)
disp.save_frames([1, 20, 70], update)
disp.animate(n_steps=n_iterations, anim_callback=update)
#disp.save_frames([1, 20, 70], update)