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)
Exemplo n.º 2
0
    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