wirecell.gen.plots Namespace Reference

Functions
def	numpy_saver (filename, outfile)

Detailed Description

Make some plots from simulation output

Function Documentation

def wirecell.gen.plots.numpy_saver	(		filename,
			outfile
	)

plot data from Numpy*Saver files

Definition at line 10 of file plots.py.

def numpy_saver(filename, outfile):
    '''
    plot data from Numpy*Saver files
    '''
    tick1=0
    tick2=500

    fp = numpy.load(filename)
    frame = fp['frame__0'].T
    channels = fp['channels__0']
    chd = channels[1:] - channels[:-1]
    chjumps = [0] + list(numpy.where(chd>1)[0]) + [channels.size-1]
    indjumps = zip(chjumps[:-1], chjumps[1:])

    njumps = len(indjumps)
    fig, axes = plt.subplots(nrows=njumps, ncols=1)
    if njumps == 1:
        axes = [axes]

    for ax, (ind1, ind2) in zip(axes, indjumps):
        ch1 = channels[ind1+1]
        ch2 = channels[ind2]

        extent = (tick1, tick2, ch2, ch1)

        print ( "array ind: [%4d,%4d] channels: [%4d,%4d] %d" % (ind1,ind2, ch1, ch2, ind2-ind1+1))
        im = ax.imshow(frame[ind1+1:ind2,tick1:tick2],aspect='auto')#,extent=extent)
        plt.colorbar(im, ax=ax)

    plt.savefig(outfile)
    return fig,axes