diff --git a/src/sk8plotlib/camera.py b/src/sk8plotlib/camera.py new file mode 100644 index 0000000..ff8f5d5 --- /dev/null +++ b/src/sk8plotlib/camera.py @@ -0,0 +1,19 @@ +import numpy as np + + +class Camera: + def __init__(self, fig, ax, skater): + self.fig, self.ax, self.skater = fig, ax, skater + + def move_camera(self): + SCALE = 10 + + x, y, v_x, v_y = self.skater.x, self.skater.y, self.skater.v_x, self.skater.v_y + location, velocity = np.asarray([x, y]), np.asarray([v_x, v_y]) + + location += velocity * 0.05 + + self.ax.set( + xlim=(location[0] - SCALE, location[0] + SCALE), + ylim=(location[1] - SCALE, location[1] + SCALE), + )