if __name__=='__main__': dirlist=sys.argv[1:] if len(dirlist)==0: sys.exit("Warning: No directories specified, will exit.") normlist="norms.namelist" species="d,N,e" species=species.split(',') #print species simulList=perfect_simulations_from_dirs(dirlist,normlist,species) #print simulList plotObjList=[] plotObjList.append(plot_object("normed_particle_flux",len(species),1,"particle flux")) plotObjList.append(plot_object("normed_conductive_heat_flux",len(species),1,"heat flux")) plotObjList.append(plot_object("T",len(species),1,"T")) plotObjList.append(plot_object("n",len(species),1,"n")) for plotObj in plotObjList: for simul in simulList: #print "simul:" #print simul plotObj.plot(simul) #plotObjList[0].show_figure() #plotObjList[1].show_figure() #fig0=plotObjList[0].fig #fig1=plotObjList[1].fig plt.show(block=True)
normlist = "norms.namelist" species = "d,N,e" species = species.split(",") # print species simulList = perfect_simulations_from_dirs(dirlist, normlist, species) # print simulList gl = ["", " local"] i = 0 for simul in simulList: simul.description = pretty_parse(simul.description) + gl[i % 2] i = i + 1 plotObjList = [] ncolors = len(simulList) / 2 plotObjList.append(plot_object("normed_particle_flux", len(species), 1, "particle flux", ncolors)) plotObjList.append(plot_object("normed_conductive_heat_flux", len(species), 1, "heat flux", ncolors)) plotObjList.append(plot_object("T", len(species), 1, "T", ncolors)) plotObjList.append(plot_object("n", len(species), 1, "n", ncolors)) plotObjList.append(plot_object("normed_heat_source", len(species), 1, "normed heat source", ncolors)) plotObjList.append(plot_object("normed_particle_source", len(species), 1, "normed particle source", ncolors)) i = 0 for plotObj in plotObjList: for simul in simulList: # print "simul:" # print simul # print "to call plotObj.plot()" plotObj.plot(simul, i % 2) # alternates between holding colors or not i = i + 1 # not python3 safe