def extract_wave_simple(IndList, FilteredArr): IndArr = np.array(IndList, dtype=np.int32) SampArr = IndArr[:, 0] ChArr = IndArr[:, 1] PeakSample = SampArr[FilteredArr[SampArr, ChArr].argmin()] Wave = get_padded(FilteredArr, PeakSample - S_BEFORE, PeakSample + S_AFTER) return Wave, PeakSample, bincount(ChArr, N_CH).astype(np.bool8)
def extract_wave_simple(IndList,FilteredArr): IndArr = np.array(IndList,dtype=np.int32) SampArr = IndArr[:,0] ChArr = IndArr[:,1] PeakSample = SampArr[FilteredArr[SampArr,ChArr].argmin()] Wave = get_padded(FilteredArr,PeakSample-S_BEFORE,PeakSample+S_AFTER) return Wave,PeakSample,bincount(ChArr,N_CH).astype(np.bool8)
def extract_wave_interp(IndList,FilteredArr): IndArr = np.array(IndList,dtype=np.int32) SampArr = IndArr[:,0] ChArr = IndArr[:,1] PeakInd = FilteredArr[SampArr,ChArr].argmin() PeakSample,PeakChannel = SampArr[PeakInd],ChArr[PeakInd] WavePlus = get_padded(FilteredArr,PeakSample-S_BEFORE-1,PeakSample+S_AFTER+1) Wave = interp_around_peak(WavePlus,S_BEFORE+1,PeakChannel,S_BEFORE,S_AFTER,kind=INTERP_METHOD) return Wave,PeakSample,bincount(ChArr,N_CH).astype(np.bool8)
def extract_wave_interp(IndList, FilteredArr): IndArr = np.array(IndList, dtype=np.int32) SampArr = IndArr[:, 0] ChArr = IndArr[:, 1] PeakInd = FilteredArr[SampArr, ChArr].argmin() PeakSample, PeakChannel = SampArr[PeakInd], ChArr[PeakInd] WavePlus = get_padded(FilteredArr, PeakSample - S_BEFORE - 1, PeakSample + S_AFTER + 1) Wave = interp_around_peak(WavePlus, S_BEFORE + 1, PeakChannel, S_BEFORE, S_AFTER, kind=INTERP_METHOD) return Wave, PeakSample, bincount(ChArr, N_CH).astype(np.bool8)