PlotEnergies Namespace Reference

Functions
def	main ()

Function Documentation

main()

def PlotEnergies.main

(

)

def main():
    path = sys.argv[1]
    name = sys.argv[2]
    print(path)
    print(name)
 
    textfile = open(path + name + ".ene", "r")
    content_list = textfile.readlines()
 
 
    time = np.zeros(len(content_list))
    gra_ene = np.zeros(len(content_list))
 
    tra_kin_ene = np.zeros(len(content_list))
    rot_kin_ene = np.zeros(len(content_list))
    kin_ene = np.zeros(len(content_list))
 
    ela_ene = np.zeros(len(content_list))
    tot_ene = np.zeros(len(content_list))
 
    ratio = np.zeros(len(content_list))
 
    for i in range(1,len(content_list)):
        A = content_list[i].split(" ")
        B = []
        for j in range(len(A)):
            if A[j] != "": B.append(A[j])
        time[i] = float(B[0])
        gra_ene[i] = float(B[1])
        tra_kin_ene[i] = float(B[2])
        rot_kin_ene[i] = float(B[3])
        kin_ene[i] = float(B[2]) + float(B[3])
        ela_ene[i] = float(B[4])
        tot_ene[i] = float(B[1]) + float(B[2]) + float(B[3])  + float(B[4])
 
 
    tra_int = 0
    rot_int = 0
    for i in range(len(time)):
        tra_int += tra_kin_ene[i]
        rot_int += rot_kin_ene[i]
        if not tra_int == 0: ratio[i] = rot_int / tra_int
    
         
    np.savetxt(path + "paraview_" + name + "/time.txt", time, delimiter=",")
    np.savetxt(path + "paraview_" + name + "/gra_ene.txt", gra_ene, delimiter=",")
    np.savetxt(path + "paraview_" + name + "/tra_kin_ene.txt", tra_kin_ene, delimiter=",")
    np.savetxt(path + "paraview_" + name + "/rot_kin_ene.txt", rot_kin_ene, delimiter=",")
    np.savetxt(path + "paraview_" + name + "/kin_ene.txt", kin_ene, delimiter=",")
    np.savetxt(path + "paraview_" + name + "/ela_ene.txt", ela_ene, delimiter=",")
    np.savetxt(path + "paraview_" + name + "/tot_ene.txt", tot_ene, delimiter=",")
    np.savetxt(path + "paraview_" + name + "/ratio.txt", ratio, delimiter=",")
    print("Successfully written energy files to paraview dir:", path + "paraview_" + name)
 
 

References Eigen::internal.print(), and oomph::DoubleVectorHelpers.split().