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

Tidying up

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-04-21 17:00:01 +02:00
parent 2e92185afd
commit b95561011b
2 changed files with 42 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
display/dislpay_2d.py
View File

@ -1,11 +1,6 @@
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
import numpy as np
import matplotlib.gridspec as gridspec
from mpl_toolkits.axes_grid1.inset_locator import zoomed_inset_axes
import functools
from time import sleep
import itertools
#
@ -22,6 +17,29 @@ class Displayer:
self._circle_map = {}
self._objects = []
self._xlim = [0, 20]
self._ylim = [0, 20]
def __init_animation(self):
self._ax.set_xlim(self._xlim)
self._ax.set_ylim(self._ylim)
for key in self._circle_map:
self._ax.add_patch(self._circle_map[key])
return self._objects
#
# Functions for initalization
#
def set_axis_limits(self, min, max):
self._xlim = (min, max)
self._ylim = (min, max)
def register_object(self, obj_name, obj_color='r'):
self._object_map[obj_name], = plt.plot([], [], color=obj_color, marker='o')
@ -31,19 +49,22 @@ class Displayer:
self._objects.append(self._circle_map[obj_name])
#
# Functions to be used in the animation callback
#
def move_object(self, obj_name, x, y):
self._object_map[obj_name].set_data([x, y])
self._circle_map[obj_name].center = (x, y)
def set_circle_radius(self, obj_name, r):
self._circle_map[obj_name].radius = r
def __init_animation(self):
self._ax.set_xlim(0, 20)
self._ax.set_ylim(0, 20)
for key in self._circle_map:
self._ax.add_patch(self._circle_map[key])
return self._objects
#
# Start the animation
#
def animate(self, n_steps, anim_callback):

14
examples/gps_fusion.py
View File

@ -8,11 +8,15 @@ from display.dislpay_2d import Displayer
def update(disp, t):
disp.move_object("test", t, t)
disp.move_object("test2", 20-t, 20-t)
disp.move_object("test2", 20-t, t)
disp.set_circle_radius("test", t*0.2)
def run():
disp = Displayer(width=6, height=6)
#disp.show_object([1, 1], 4, "green")
disp.register_object("test")
disp.register_object("test2", 'g')
disp.animate(n_steps = 100, anim_callback=update)
disp.set_axis_limits(min=0, max=20)
disp.register_object("test", "red")
disp.register_object("test2", "black")
disp.animate(n_steps = 20, anim_callback=update)