Created basic outline of Displayer class

display/dislpay_2d.py

@@ -0,0 +1,35 @@
+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
+
+
+class Displayer:
+    def __init__(self, width, height):
+        self._fig, self._ax = plt.subplots()
+        self._fig.set_figwidth(width)
+        self._fig.set_figheight(height)
+        self._xdata, self._ydata = [], []
+        self._obj, = plt.plot([], [], 'ro')
+        self._circle = plt.Circle((5, 5), 1, color='r', fill=False)
+    
+        self._objects = [self._obj, self._circle]
+
+    def __init_animation(self):
+        self._ax.set_xlim(0, 20)
+        self._ax.set_ylim(0, 20)
+        self._ax.add_patch(self._circle)
+        return self._objects
+
+    def __update_animation(self, t):
+        self._obj.set_data([t, t])
+        self._circle.center = (t, t)
+        self._circle.radius = t*0.1
+        return self._objects
+
+    def animate(self, n_steps):
+        anim = FuncAnimation(self._fig, self.__update_animation, frames=np.linspace(0, n_steps-1, n_steps),
+                            init_func=self.__init_animation, blit=True)
+        plt.show()