configuring animation script for euler

plots/animation.py

@@ -22,7 +22,8 @@ def animate(i):
     plt.scatter(xi, yi, s=1)
     plt.xlim(-1, 1)
     plt.ylim(-1, 1)
+    plt.axis("equal")
 
 animation = FuncAnimation(fig, animate, frames=nSteps)
-animation.save("animation.mp4", writer="ffmpeg")
-plt.show()
\ No newline at end of file
+mp4writer = animation.FFMpegWriter(fps=30)
+animation.save("animation.mp4", writer=mp4writer)
\ No newline at end of file

run.sh

@@ -13,4 +13,6 @@ export OMP_NUM_THREADS=48
 cd build/
 make
 ./nbody
+cd ../plots/
+python3 animation.py
 cd ..