diff --git a/plots/animation.py b/plots/animation.py index 95f626493761333419422aa1f2cbd84d4a54ab4d..582c80e95990a27a76802423eb0107e8ea661e92 100644 --- a/plots/animation.py +++ b/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 diff --git a/run.sh b/run.sh index 34de9f49d040ba7f7e008c68050a8947405e1764..c9898f6905e8cdff91324d1b7d1e6cb10f4af9c5 100644 --- a/run.sh +++ b/run.sh @@ -13,4 +13,6 @@ export OMP_NUM_THREADS=48 cd build/ make ./nbody +cd ../plots/ +python3 animation.py cd ..