Exemplo n.º 1
0
def getTimeinCenterMulti(datasets, windowSize= None, tRange=None, stateRange=None):
    durations = []
    if tRange is not None: 
        endWindow = tRange[-1] 
    else: 
        for i in range(len(datasets)):
            tracking = datasets[i]['warpedTracking']
            if stateRange is not None: 
                _,et,_,_ = aba.state_to_time(datasets[i], stateRange[1])
                durations.append(et-tracking[0,0])
            else: 
                durations.append(datasets[i]['warpedTracking'][-1,0]-datasets[i]['warpedTracking'][0,0])
        endWindow = np.min(durations)

    windowsize = 30
    if windowSize is not None: 
        windowsize = windowSize
 
    ticks = np.arange(0,endWindow, windowsize)
    eTimeinCenter = np.zeros((len(datasets),len(ticks)-1))
    for i in range(len(ticks)-1):
        startT = ticks[i]
        endT = ticks[i+1]
        for d in range(len(datasets)):
            eTimeinCenter[d,i] = aba.getPercentTimeinCenter(datasets[d], tRange=[startT, endT])
    
    return eTimeinCenter