export.py

from time import ctime,localtime
from layout import box_h, box_w
import os
from setup import arg_names

def export_red_collisions(collisions, args):
    cutime=localtime()
    filename="_".join([str(x) for x in cutime[:6]])
    make_directory_if_not_exists("results/")
    file=open("results/"+filename+".txt", "w+") #export dir
    file.write("Atom box simulation results from "+ctime()+"\n")
    for i in range(len(args)):
        file.write(arg_names[i]+": "+str(args[i])+"\n")
    for i in collisions:
        file.write(str(i)+"\n")
    export_distances(file,collisions)

def export_distances(file, collisions):
    file.write("Distances\n")
    distances=[]
    for i in range(len(collisions)-1):
        distances.append(calculate_distance(collisions[i], collisions[i+1]))
        file.write(str(distances[i])+"\n")
    file.write("Average\n")
    avg=sum(distances)/max(collisions)[0]
    file.write(str(avg)+"\n")

def make_directory_if_not_exists(path):
    if not os.path.isdir(path):
        os.makedirs(path)

def calculate_distance(first, second): 
    a, firstx, firsty=first
    a, secondx, secondy=second
    distance=((firstx-secondx)**2+(firsty-secondy)**2)**0.5
    return(distance)