From a9269caa46ca17847edd18561d78b956f6eb99a6 Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Thu, 28 Apr 2022 11:14:39 +0200 Subject: [PATCH] Changed number of iterations for animation; fixed bug in Displayer calass --- kalman_filter/display/display_2d.py | 1 + kalman_filter/examples/gps_two_sources_animated.py | 12 ++++++------ 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/kalman_filter/display/display_2d.py b/kalman_filter/display/display_2d.py index 82cfc62..6ac51a3 100644 --- a/kalman_filter/display/display_2d.py +++ b/kalman_filter/display/display_2d.py @@ -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 diff --git a/kalman_filter/examples/gps_two_sources_animated.py b/kalman_filter/examples/gps_two_sources_animated.py index e78ac7b..1179790 100644 --- a/kalman_filter/examples/gps_two_sources_animated.py +++ b/kalman_filter/examples/gps_two_sources_animated.py @@ -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)