4 # @purpose Extract statistics from repeated runs of N-Body simulators
5 # @author Sam Moore (20503628) - 2012
11 import Gnuplot, Gnuplot.funcutils
12 plot = Gnuplot.Gnuplot(persist=0)
15 "single-thread" : "./single-thread/nbody -v 5"
18 programs.update({"mthread"+str(n) : "./mthread/nbody -v 5 --pedantic-graphics -n "+str(n)})
19 programs.update({"mthread"+str(n) : "./mthread/nbody -v 5 -n "+str(n)})
20 #programs.update({"slow-mthread"+str(n) : "./mthread/nbody-slow --pedantic-graphics -v 5 -n "+str(n)})
21 #programs.update({"openmp"+str(n) : "./openmp/nbody -v 5 --pedantic-graphics -n " +str(n)})
25 def RunProgram(string):
26 p = subprocess.Popen(string.split(" "), stdout=subprocess.PIPE)
29 results.append(map(float, line.strip(" \r\n\t").split("\t")))
30 results.sort(key = lambda e : e[0])
33 def RunForSteps(program, field, steps):
35 print(str(program) + " -s "+str(steps) + " " + str(field))
36 results = RunProgram(str(program) + " -s "+str(steps) + " " + str(field))
39 def VaryField(program, steps, fields):
42 results.append(RunProgram(program + " -s " +str(steps) + " " + str(field)))
45 def VaryTime(program, field, time_min, time_max, time_inc=10):
47 for t in range(time_min, time_max, time_inc):
48 results.append(RunProgram(program + " -t "+str(t) + " " + str(field)))
52 plot("set xlabel \"steps\"")
53 plot("set ylabel \"real time (s)\"")
54 plot("set title \"Time .vs. Steps Computed\"")
55 plot("set key outside right")
59 fields = map(int, os.listdir("fields"))
60 fields.sort(key = lambda e : e)
67 data[p] = RunForSteps(programs[p], "fields/"+str(n), 200)
70 #print(str(data.items()));
71 for d in sorted(data.items(), key = lambda e : e[1][len(e[1])-1][1]):
73 plot.replot(Gnuplot.Data(d[1], title=str(d[0])+":"+str(n), with_="lp")) #linecolor "+str(fields.index(n))))
75 #print(str(data.items()))
78 #for p in sorted(data.items(), key = lambda e : e[1][0][len(e[1][0])-1][1]):
79 # print(str(p[0])+":"+str(n)+"\t"+str(p[1][0][len(p[1][0])-1][1]))
84 if __name__ == "__main__":
85 sys.exit(main(sys.argv))