wirecell.gen.sim.Frame Class Reference

Inheritance diagram for wirecell.gen.sim.Frame:

Public Member Functions
def	__init__ (self, fp, ident=0, tag='')
def	plot_ticks (self, tick0=0, tickf=-1, raw=True, chinds=())
def	plot (self, t0=None, tf=None, raw=True, chinds=())

Public Attributes
	channel_boundaries

Detailed Description

Definition at line 68 of file sim.py.

Constructor & Destructor Documentation

def wirecell.gen.sim.Frame.__init__	(		self,
			fp,
			ident = 0,
			tag = ''
	)

Definition at line 69 of file sim.py.

    def __init__(self, fp, ident=0, tag=''):
        def thing(n):
            arr = fp['%s_%s_%d' % (n, tag, ident)]
            if n == 'frame':
                return arr.T
            return arr

        for n in 'frame channels tickinfo'.split():
            setattr(self, n, thing(n))
        
        self.channel_boundaries = ()

Member Function Documentation

def wirecell.gen.sim.Frame.plot	(		self,
			t0 = None,
			tf = None,
			raw = True,
			chinds = ()
	)

Definition at line 121 of file sim.py.

    def plot(self, t0=None, tf=None, raw=True, chinds = ()):

        frame = self.frame
        if not raw:
            frame = baseline_subtract(frame)

        tstart, tick = self.tickinfo[:2]
        nticks = frame.shape[1]
        tend = tstart + nticks*tick

        if t0 is None or t0 < tstart or t0 > tend:
            t0 = tstart

        if tf is None or tf < t0 or tf > tend:
            tf = tend

        tick0 = int((t0-tstart)/tick)
        tickf = int((tf-tstart)/tick)
        
        #print ("trange=[%.2f %.2f]ms ticks=[%d,%d]" % (t0/units.ms,tf/units.ms,tick0,tickf))

        if not chinds:
            chinds = group_channel_indices(self.channels)
        ngroups = len(chinds)
        fig, axes = plt.subplots(nrows=ngroups, ncols=1, sharex = True)
        if ngroups == 1:
            axes = [axes]       # always list 

        for ax, chind in zip(axes, chinds):
            ngroups -= 1
            chind0 = chind[0]
            chind1 = chind[1]

            ch1 = self.channels[chind0]
            ch2 = self.channels[chind1-1]

            extent = (t0/units.ms, tf/units.ms, ch2, ch1)
            print ("exent: %s, chind=(%d,%d)" % (str(extent), chind0, chind1) )


            #print (chind,extent, chind, tick0, tickf, frame.shape)
            im = ax.imshow(frame[chind0:chind1,tick0:tickf],
                           aspect='auto', extent=extent, interpolation='none')
            plt.colorbar(im, ax=ax)
            if not ngroups:
                ax.set_xlabel('time [ms]')

        #plt.savefig(outfile)
        return fig,axes
        
        #plt.imshow(fr[ch[0]:ch[1], ti[0]:ti[1]], extent=[ti[0], ti[1], ch[1], ch[0]], aspect='auto', interpolation='none'); plt.colorbar()


# f = numpy.load("test-boundaries.npz"); reload(sim);
# fo = sim.Frame(f)
# fo.plot()
# dd = f['depo_data_0']
# plt.scatter(dd[2]/units.m, dd[4]/units.m)

def wirecell.gen.sim.Frame.plot_ticks	(		self,
			tick0 = 0,
			tickf = -1,
			raw = True,
			chinds = ()
	)

Plot in terms of ticks.  Here, the frame is assumed to be
dense and chinds are taken as channel ranges.

Definition at line 81 of file sim.py.

    def plot_ticks(self, tick0=0, tickf=-1, raw=True, chinds = ()):
        '''
        Plot in terms of ticks.  Here, the frame is assumed to be
        dense and chinds are taken as channel ranges.
        '''
        frame = self.frame
        print ("Frame shape: %s" % str(frame.shape))

        if not raw:
            frame = baseline_subtract(frame)


        if not chinds:
            chinds = group_channel_indices(self.channels, self.channel_boundaries)

        tick = self.tickinfo[1]
        if tickf < 0:
            tickf += frame.shape[1]

        ngroups = len(chinds)
        fig, axes = plt.subplots(nrows=ngroups, ncols=1, sharex = True)
        if ngroups == 1:
            axes = [axes]       # always list 

        for ax, chind in zip(axes, chinds):
            ngroups -= 1
            ch1 = self.channels[chind[0]]
            ch2 = self.channels[chind[1]-1]

            extent = (tick0, tickf, ch2, ch1)
            print ("exent: %s" % str(extent))

            im = ax.imshow(frame[chind[0]:chind[1],tick0:tickf],
                           aspect='auto', extent=extent, interpolation='none')
            plt.colorbar(im, ax=ax)
            if not ngroups:
                ax.set_xlabel('ticks (%.2f us)' % (tick/units.us),)

        return fig,axes

Member Data Documentation

wirecell.gen.sim.Frame.channel_boundaries

Definition at line 79 of file sim.py.

---

The documentation for this class was generated from the following file:

• wire-cell-build/python/wirecell/gen/sim.py