_filtering.py 文件源码

def _apply_method(x, fMeth, dtrd, method, wltCorr, wltWidth):
    npts, ntrial = x.shape
    nFce = len(fMeth)
    xf = np.zeros((nFce, npts, ntrial))

    # Detrend the signal :
    if dtrd:
        x = detrend(x, axis=0)

    # Apply methods :
    for k in range(nFce):
        xf[k, ...] = fMeth[k](x)

    # Correction for the wavelet (due to the wavelet width):
    if (method == 'wavelet') and (wltCorr is not None):
        w = 3*wltWidth
        xf[:, 0:w, :] = xf[:, w+1:2*w+1, :]
        xf[:, npts-w:npts, :] = xf[:, npts-2*w-1:npts-w-1, :]

    return xf