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)