from matplotlib import pyplot as plt
from matplotlib import animation
from boids import Boids
# Deliberately terrible code for teaching purposes
repulsion = 100
attraction = 0.01
speed = 10000
acceleration = 0.125
model = Boids(50, repulsion, attraction, speed, acceleration)
model.initialise_random()
figure = plt.figure()
axes = plt.axes(xlim=(-500,1500), ylim=(-500,1500))
scatter = axes.scatter( [boid.x for boid in model.boids],\
[boid.y for boid in model.boids] )
def animate(frame):
model.update_boids()
scatter.set_offsets(zip([boid.x for boid in model.boids], \
[boid.y for boid in model.boids]))
anim = animation.FuncAnimation(figure, animate, frames=50, interval=50)
if __name__ == "__main__":
plt.show()
