import argon

def main():

    nat=int(input("Numero di atomi:\t"))
    density=float(input("Densità (kg/m^3):\t"))
    temperature=float(input("Temperatura (K):\t"))
    dt=float(input("Time step (ps):\t"))
    dt*=1e-12
    nn=int(input("Numero di passi temporali:\t"))

    ar=argon.argon(nat,density)
    ar.initialize(temperature)
    #print(ar.energy())
    print(ar.energy2())

    ar.force2()

    for i in range(nn):
        ar_next=ar.step_velocity_verlet(dt)
        ar=ar_next
        print(ar.energy2())


if __name__ == "__main__":
    main()