import pandas as pd
import os
import matplotlib
import matplotlib.pyplot as plt
import matplotlib.animation as anim
import random as rd

matplotlib.use('Agg')

files = os.listdir("./output")

df_ref = pd.read_csv("./output/"+files[0])

os.system("mkdir tmp")

for i in range(0,df_ref.shape[0]):
    fig = plt.figure()
    ax1 = fig.add_subplot(1,1,1)
    axes = plt.gca()
    axes.set_xlim([-20 ,20])
    axes.set_ylim([-10,10])
    for name in files:
        df = pd.read_csv("./output/"+name)
        ax1.scatter(df.iloc[i][1],df.iloc[i][2])
    fig.savefig("tmp/test_{}.png".format(i))
    plt.close(fig)

os.system("ffmpeg -r 30 -i tmp/test_%01d.png -vcodec mpeg4 -y visual.mp4")
os.system("rm -r tmp")