1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# Authors: Daniel Strohmeier <daniel.strohmeier@tu-ilmenau.de>
#
# License: BSD (3-clause)
import os.path as op
import numpy as np
from numpy.testing import assert_array_almost_equal
from nose.tools import assert_true
from mne.io import Raw
from mne.event import read_events
from mne.preprocessing.stim import eliminate_stim_artifact
data_path = op.join(op.dirname(__file__), '..', '..', 'io', 'tests', 'data')
raw_fname = op.join(data_path, 'test_raw.fif')
event_fname = op.join(data_path, 'test-eve.fif')
def test_stim_elim():
"""Test eliminate stim artifact"""
raw = Raw(raw_fname, preload=True)
events = read_events(event_fname)
event_idx = np.where(events[:, 2] == 1)[0][0]
tidx = int(events[event_idx, 0] - raw.first_samp)
# use window around stimulus
tmin = -0.02
tmax = 0.02
test_tminidx = int(-0.01 * raw.info['sfreq'])
test_tmaxidx = int(0.01 * raw.info['sfreq'])
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='linear')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
diff_data0 = np.diff(data[0])
diff_data0 -= np.mean(diff_data0)
assert_array_almost_equal(diff_data0, np.zeros(len(diff_data0)))
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='window')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
assert_true(np.all(data) == 0.)
# use window before stimulus
tmin = -0.045
tmax = 0.015
test_tminidx = int(-0.035 * raw.info['sfreq'])
test_tmaxidx = int(-0.015 * raw.info['sfreq'])
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='linear')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
diff_data0 = np.diff(data[0])
diff_data0 -= np.mean(diff_data0)
assert_array_almost_equal(diff_data0, np.zeros(len(diff_data0)))
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='window')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
assert_true(np.all(data) == 0.)
# use window after stimulus
tmin = 0.005
tmax = 0.045
test_tminidx = int(0.015 * raw.info['sfreq'])
test_tmaxidx = int(0.035 * raw.info['sfreq'])
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='linear')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
diff_data0 = np.diff(data[0])
diff_data0 -= np.mean(diff_data0)
assert_array_almost_equal(diff_data0, np.zeros(len(diff_data0)))
raw = eliminate_stim_artifact(raw, events, event_id=1, tmin=tmin,
tmax=tmax, mode='window')
data, times = raw[:, (tidx + test_tminidx):(tidx + test_tmaxidx)]
assert_true(np.all(data) == 0.)
|
