+def Plot_MeanErrors(control, program, field, sMin, sMax):
+ plot("set xlabel \"steps\"")
+ plot("set ylabel \"Error\"")
+ plot("set title \"Mean Position Error .vs. Steps\\n(Mean Error = Mean absolute error of all position components)\\nField: "+str(field)+"\"")
+ plot("set key outside right")
+ plot("set log y")
+ if (control == program):
+ return
+ d = []
+ for s in range(sMin, sMax, (sMax - sMin) / 20):
+ d.append([s, MeanError(control, program, field, s)])
+
+ plot.replot(Gnuplot.Data(d, title=str(program), with_="lp")) #linecolor "+str(fields.index(n))))
+
+def Plot_PerformanceN(d, steps, nMin, nMax, nStep):
+
+ plot("set xlabel \"N bodies\"")
+ plot("set ylabel \"Real Time (s)\"")
+ plot("set title \"Time taken to simulate N randomly generated bodies for 1000 steps\"")
+ plot("set key outside right")
+ for p in d.items():
+ data = []
+ for n in range(nMin, nMax, nStep):
+ print("Run for " + str(n) + " bodies; " + str(p[1]))
+ r = RunProgram(str(p[1]) + " -r " + str(n) + " -s " + str(steps))
+ data.append([n, r[len(r)-1][1]])
+ plot.replot(Gnuplot.Data(data, title=str(p[0]), with_="lp"))
+
+def MeanError(programA, programB, field, steps):
+ print("Calculate error for "+str(steps) + " steps: " + str(programB) + " .vs. " + str(programA))
+ RunProgram(str(programA) + " -s "+str(steps) + " " + str(field) + " " + "A.tmp")
+ RunProgram(str(programB) + " -s "+str(steps) + " " + str(field) + " " + "B.tmp")
+
+ A = open("A.tmp", "r", 0)
+ B = open("B.tmp", "r", 0)
+
+ nA = int(A.readline().strip("\r\n\t "))
+ nB = int(B.readline().strip("\r\n\t "))
+ if (nA != nB):
+ sys.stderr.write("Body number error; A: " + str(nA) + " vs B: " + str(nB) + "\n")
+ sys.exit(1)
+
+ total_error = 0.0
+ values = 0
+ for i in range(0, nA):
+ dataA = A.readline().strip("\r\n\t ").split(" ")
+ dataB = B.readline().strip("\r\n\t ").split(" ")
+ #print("A: "+str(dataA))
+ #print("B: "+str(dataB))
+ dataA = map(float, dataA)[1:4] # Only count positions
+ dataB = map(float, dataB)[1:4]
+ #print("Length " + str(len(dataA)))
+ if (len(dataB) != len(dataA)):
+ sys.stderr.write("Number of values in line are different\n")
+ sys.exit(1)
+
+
+
+ values += len(dataA)
+ for j in range(0, len(dataA)-1):
+ total_error += abs(dataB[j] - dataA[j])
+ os.system("rm A.tmp; rm B.tmp;")
+ return total_error / values
+