diff --git a/wetb/signal_tools/filters/cy_filters.py b/wetb/signal_tools/filters/cy_filters.py index 4df4a89e5c11202428b955ab2f73485999e058ec..35fe75dba9ac05fe3ebbbdea7c6757160670e55f 100644 --- a/wetb/signal_tools/filters/cy_filters.py +++ b/wetb/signal_tools/filters/cy_filters.py @@ -23,7 +23,7 @@ def cy_low_pass_filter(inp, delta_t, tau): #cpdef cy_low_pass_filter(np.ndarray return output -def cy_dynamic_low_pass_filter(inp, delta_t, tau, method): #cpdef cy_dynamic_low_pass_filter(np.ndarray[double,ndim=1] inp, double delta_t, np.ndarray[double,ndim=1] tau, int method): +def cy_dynamic_low_pass_filter(inp, delta_t, tau, method=1): #cpdef cy_dynamic_low_pass_filter(np.ndarray[double,ndim=1] inp, double delta_t, np.ndarray[double,ndim=1] tau, int method=1): #cdef np.ndarray[double,ndim=1] output, alpha #cdef int i diff --git a/wetb/signal_tools/filters/first_order.py b/wetb/signal_tools/filters/first_order.py index e41927634eb818fab6671f2b9d69c7db057a8f33..6bcfd04655633e423127d80a4537654aad9385cc 100644 --- a/wetb/signal_tools/filters/first_order.py +++ b/wetb/signal_tools/filters/first_order.py @@ -6,11 +6,11 @@ Created on 10/01/2015 import numpy as np from wetb.signal_tools.filters import cy_filters -def low_pass(input, delta_t, tau): +def low_pass(input, delta_t, tau, method=1): if isinstance(tau, (int, float)): return cy_filters.cy_low_pass_filter(input.astype(np.float64), delta_t, tau) else: - return cy_filters.cy_dynamic_low_pass_filter(input.astype(np.float64), delta_t, tau) + return cy_filters.cy_dynamic_low_pass_filter(input.astype(np.float64), delta_t, tau, method) def high_pass(input, delta_t, tau): return cy_filters.cy_high_pass_filter(input.astype(np.float64), delta_t, tau)