bci-repl/topology.py at master · julfy/bci-repl · GitHub

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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from typing import List

from math import ceil

class Electrode:
    def __init__(self, name, x, y):
        self.name = name
        # Positions for gui
        self.x = x
        self.y = y

electrodes = {}

##########################
# Electrode Declarations #
##########################

# dy : <1|-1>=inverted y direction * (<offset> + <y step>)
# dx : pow(-1, i)=x direction * ceil(i/2.0)=number of steps * <step size>

def def_points(prefix, dx, dy, range):
    for i in range:
        name = '{}{}'.format(prefix, 'Z' if i == 0 else i)
        electrodes.update({name: Electrode(name, dx(i), dy(i))})

# C, T
c_dx = lambda i: pow(-1, i) * ceil(i/2.0) * 0.077
c_dy = lambda i: 0
def_points('C', c_dx, c_dy, range(7))
def_points('T', c_dx, c_dy, range(7,7+4))
# def_points('A', ) # ear refs
# FC, FT
fc_dx = lambda i: pow(-1, i) * ceil(i/2.0) * 0.075
fc_dy = lambda i: -1 * (0.075 + 0.002* pow(ceil(i/2.0), 2) )
def_points('FC', fc_dx, fc_dy, range(7))
def_points('FT', fc_dx, fc_dy, range(7,7+4))
# F
f_dx = lambda i: pow(-1, i) * ceil(i/2.0) * 0.065
f_dy = lambda i: -1 * (0.15 + 0.0035* pow(ceil(i/2.0), 2) )
def_points('F', f_dx, f_dy, range(11))
# AF
af_dx = lambda i: pow(-1, i) * ceil(i/2.0) * 0.06
af_dy = lambda i: -1 * (0.23 + 0.005* pow(ceil(i/2.0), 2) )
def_points('AF', af_dx, af_dy, range(9))
# Fp
fp_dx = lambda i: pow(-1, i) * ceil(i/2.0) * 0.09
fp_dy = lambda i: -1 * (0.32 - 0.005* pow(ceil(i/2.0), 2) )
def_points('Fp', fp_dx, fp_dy, range(3))
# N
def_points('N', lambda i: 0, lambda i: -0.4, range(1))
# CP, TP
cp_dx = fc_dx
cp_dy = lambda i: -1 * fc_dy(i)
def_points('CP', cp_dx, cp_dy, range(7))
def_points('TP', cp_dx, cp_dy, range(7,7+4))
# P
p_dx = f_dx
p_dy = lambda i: -1 * f_dy(i)
def_points('P', p_dx, p_dy, range(11))
# PO
p_dx = af_dx
p_dy = lambda i: -1 * af_dy(i)
def_points('PO', p_dx, p_dy, range(9))
# O
p_dx = fp_dx
p_dy = lambda i: -1 * fp_dy(i)
def_points('O', p_dx, p_dy, range(3))
# I
def_points('I', lambda i: 0, lambda i: 0.4, range(1))

##############
# Topologies #
##############

top_8c_10_20 = ['FZ', 'CZ', 'P3', 'PZ', 'P4', 'PO7', 'PO8', 'OZ' ]
top_16c_10_10 = [ # occipital
    'FZ',
    'C5', 'CZ', 'C6',
    'P7', 'P3', 'PZ', 'P4', 'P8',
    'PO7', 'PO3', 'POZ', 'PO4', 'PO8',
    'O1', 'O2']
top_16c_obci_10_20 = [
    # 'Fp1', 'Fp2',
    'F7', 'F3', 'F4', 'F8', # 'FZ',
    'T7', 'C3', 'CZ', 'C4', 'T8',
    'P7', 'P3', 'PZ', 'P4', 'P8',
    'O1', 'O2'
]

topologies = {
    'all': [k for k in electrodes.keys()],
    'top_8c_10_20': top_8c_10_20,
    'top_16c_10_10': top_16c_10_10,
    'top_16c_obci_10_20': top_16c_obci_10_20,
}


def get_topology(name: str) -> List[str]:
    return topologies[name]