Skip to content

Commit dcc93b6

Browse files
committed
Example module: generate separate filter coefficients for each channel
1 parent 0fd7aca commit dcc93b6

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

Modules/examples/bandpass_filter.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,16 @@ class PyProcessor:
3535

3636
def __init__(self, num_channels, sample_rate):
3737
""" A new bandpass filter is initialized whenever the plugin settings are updated """
38+
3839
self.num_chans = num_channels
3940
self.sample_rate = sample_rate
40-
self.sos = butter_bandpass(500, 2000, sample_rate)
41+
42+
self.sos = []
43+
sos_t = butter_bandpass(500, 2000, sample_rate)
44+
for i in range(self.num_chans):
45+
self.sos.append(sos_t)
46+
47+
self.sos = np.asarray(self.sos)
4148

4249
def process(self, data):
4350
"""
@@ -48,7 +55,7 @@ def process(self, data):
4855
"""
4956
try:
5057
for i in range(self.num_chans):
51-
data[i] = butter_bandpass_filter(self.sos, data[i])
58+
data[i] = butter_bandpass_filter(self.sos[i], data[i])
5259
except:
5360
pass
5461

0 commit comments

Comments
 (0)