wirecell.gen.test.plot_g4tuple Namespace Reference

Functions
def	load_depos (filename)
def	plot_dedx (depos)
def	plot_dndx (depos)
def	plot_nxz (depos)

Variables
string	depo_keys = "xyzqtsn"
	depos = load_depos(sys.argv[1])

Function Documentation

def wirecell.gen.test.plot_g4tuple.load_depos

(

filename

)

Definition at line 12 of file plot_g4tuple.py.

def load_depos(filename):
    if filename.endswith(".bz2"):
        fp = bz2.BZ2File(filename)
    else:
        fp = open(filename);
    jdat = json.loads(fp.read())
    depos = list()
    for jdepo in jdat["depos"]:
        depo = tuple([jdepo[key] for key in depo_keys])
        if depo[5] == 0.0:
            continue
        depos.append(depo)
    return numpy.asarray(depos)

def wirecell.gen.test.plot_g4tuple.plot_dedx

(

depos

)

Definition at line 26 of file plot_g4tuple.py.

def plot_dedx(depos):
    dE = depos[:,3]
    dX = depos[:,5]
    dEdX = dE/dX
    h = numpy.histogram(dEdX, 1000, (0,10))
    plt.clf(); plt.plot(h[1][:-1], h[0])
    plt.savefig("g4tuple-dedx.pdf")
    pass

def wirecell.gen.test.plot_g4tuple.plot_dndx

(

depos

)

Definition at line 35 of file plot_g4tuple.py.

def plot_dndx(depos):
    dN = depos[:,4]
    dX = depos[:,5]
    dNdX = dN/dX
    print dNdX[0]
    h = numpy.histogram(dNdX, 150, (5.0e4,2.0e5))
    plt.clf(); plt.semilogy(h[1][:-1], h[0])
    plt.savefig("g4tuple-dndx.pdf")
    pass

def wirecell.gen.test.plot_g4tuple.plot_nxz

(

depos

)

Definition at line 45 of file plot_g4tuple.py.

def plot_nxz(depos):
    "Plot colz number as X vs Z (transverse)"
    x = depos[:,0]
    z = depos[:,2]
    n = depos[:,4]
    

    xedges = numpy.linspace(90, 120.0, 300)
    zedges = numpy.linspace(10, 70.0, 600)

    #H, xedg, zedg = numpy.histogram2d(z, x, bins=(xedges, zedges), weights=n)
    fig = plt.figure()
    ax = fig.add_subplot(111)
    ax.set_title("Electrons per mm$^2$")
    #X, Z = numpy.meshgrid(xedges, zedges)
    #cax = ax.pcolormesh(X, Z, H)
    #cax = ax.imshow(H)#, interpolation='nearest', origin='low',
    #                   # extent=[xedg[0], xedg[-1], zedg[0], zedg[-1]])
    h = ax.hist2d(x,z,bins=(xedges, zedges), weights=n)
    ax.set_xlabel("X [cm]")
    ax.set_ylabel("Z [cm]")
    plt.colorbar(h[3], ax=ax)
    fig.savefig("g4tuple-nxz.pdf")

Variable Documentation

string wirecell.gen.test.plot_g4tuple.depo_keys = "xyzqtsn"

Definition at line 10 of file plot_g4tuple.py.

wirecell.gen.test.plot_g4tuple.depos = load_depos(sys.argv[1])

Definition at line 72 of file plot_g4tuple.py.