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
|
import biosig
import numpy as S
def load(fname):
HDR = biosig.constructHDR(0, 0)
HDR = biosig.sopen(fname, "r", HDR);
# turn off all channels
# for i in range(HDR.NS):
# HDR.CHANNEL[i].OnOff = 0
# turn on specific channels
# HDR.CHANNEL[0].OnOff = 1
# HDR.CHANNEL[1].OnOff = 1
# HDR.CHANNEL[HDR.NS-1].OnOff = 1
data = biosig.sread(0, HDR.NRec, HDR)
biosig.sclose(HDR)
biosig.destructHDR(HDR)
return data
def testsin(sig, sr, freq):
quartper = int(sr / freq * .25)
x = S.sqrt(sig[:-quartper] ** 2 + sig[quartper:] ** 2)
ampl = x.mean()
err = x.std()
if err > ampl/3:
return 0
return ampl
import sys
if len(sys.argv) < 3:
print "usage:\n%s filename.bdf freq" % sys.argv[0]
sys.exit(1)
fname = sys.argv[1]
sr = int(sys.argv[2])
print "\nexample.py\nchecking file %s" % fname
sig = load(fname)
freq = 3
print "Looking for sinusoidal signals at %f Hz" % freq
sig = sig - S.mean(sig, axis=1)[:, S.newaxis]
a1 = testsin(sig[0], sr, freq)
print "channel A1, peak to peak amplitude of sin at %f Hz: %f" % (freq, 2*a1)
a2 = testsin(sig[1], sr, freq)
print "channel A2, peak to peak amplitude of sin at %f Hz: %f" % (freq, 2*a2)
a3 = testsin(sig[0] + 2*sig[1], sr, freq)
print "channel A1 + 2*A2, peak to peak amplitude of sin at %f Hz: %f" % (freq, 2*a3)
a4 = testsin(sig[2], sr, freq)
print "status channel, peak to peak amplitude of sin at %f Hz: %f" % (freq, 2*a4)
|
