wirecell.gen.sim Namespace Reference

Classes
class	Depos
class	Frame
class	NumpySaver

Functions
def	baseline_subtract (frame)
def	parse_channel_boundaries (cb)
def	group_channel_indices (channels, boundaries=())

Detailed Description

Work with sim data.

fixme: this module is poorly named.

Function Documentation

def wirecell.gen.sim.baseline_subtract

(

frame

)

Return a new frame array (chan x tick) were each channel row has
its baseline subtracted.

Definition at line 15 of file sim.py.

def baseline_subtract(frame):
    '''
    Return a new frame array (chan x tick) were each channel row has
    its baseline subtracted.
    '''
    # gotta transpose around to get the right shapes to trigger broadcasting.
    return (frame.T - numpy.median(frame, axis=1)).T

def wirecell.gen.sim.group_channel_indices	(		channels,
			boundaries = ()
	)

Given a list of channels, return a list of lists where each
sublist is a sequential set.  If a sequence of boundaries are
given, then force a grouping on that channel even if there's not
jump.

Definition at line 30 of file sim.py.

def group_channel_indices(channels, boundaries=()):
    '''
    Given a list of channels, return a list of lists where each
    sublist is a sequential set.  If a sequence of boundaries are
    given, then force a grouping on that channel even if there's not
    jump.
    '''

    print ("Channel boundaries: %s" % str(boundaries))

    channels = numpy.asarray(channels)
    chan_list = channels.tolist()
    binds = list()
    for b in boundaries:
        try:
            i = chan_list.index(b)
        except ValueError:
            continue
        binds.append(i)

    chd = channels[1:] - channels[:-1]
    chjumps = [-1] + list(numpy.where(chd>1)[0]) + [channels.size-1]
    ret = list()
    for a,b in zip(chjumps[:-1], chjumps[1:]):
        #print a,b,channels[a+1:b+1]
        a += 1
        b += 1

        gotsome = 0
        for bind in binds:
            if a < bind and bind < b:
                ret.append((a,bind))
                ret.append((bind,b))
                gotsome += 1
        if gotsome == 0:
            ret.append((a,b))
    return ret

def wirecell.gen.sim.parse_channel_boundaries

(

cb

)

Definition at line 24 of file sim.py.

def parse_channel_boundaries(cb):
    if type(cb) != type(""):
        return cb               # assume already parsed.
    return tuple(map(int, cb.split(',')))