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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
# Authors: Alexandre Gramfort <alexandre.gramfort@inria.fr>
# Eric Larson <larson.eric.d@gmail.com>
# Guillaume Favelier <guillaume.favelier@gmail.com>
#
# License: Simplified BSD
import numpy as np
from ...utils import warn
class _LinkViewer(object):
"""Class to link multiple Brain objects."""
def __init__(self, brains, time=True, camera=False, colorbar=True,
picking=False):
self.brains = brains
self.leader = self.brains[0] # select a brain as leader
# check time infos
times = [brain._times for brain in brains]
if time and not all(np.allclose(x, times[0]) for x in times):
warn('stc.times do not match, not linking time')
time = False
if camera:
self.link_cameras()
if time:
# link time sliders
self.link_widgets(
name="time",
callback=self.set_time_point,
signal_type="floatValueChanged",
)
# link playback speed sliders
self.link_widgets(
name="playback_speed",
callback=self.set_playback_speed,
signal_type="valueChanged",
)
# link toggle to start/pause playback
self.link_widgets(
name="play",
callback=self.toggle_playback,
signal_type="triggered",
actions=True,
)
# link time course canvas
def _time_func(*args, **kwargs):
for brain in self.brains:
brain.callbacks["time"](*args, **kwargs)
for brain in self.brains:
if brain.show_traces:
brain.mpl_canvas.time_func = _time_func
if picking:
def _func_add(*args, **kwargs):
for brain in self.brains:
brain._add_vertex_glyph2(*args, **kwargs)
brain.plotter.update()
def _func_remove(*args, **kwargs):
for brain in self.brains:
brain._remove_vertex_glyph2(*args, **kwargs)
# save initial picked points
initial_points = dict()
for hemi in ('lh', 'rh'):
initial_points[hemi] = set()
for brain in self.brains:
initial_points[hemi] |= \
set(brain.picked_points[hemi])
# link the viewers
for brain in self.brains:
brain.clear_glyphs()
brain._add_vertex_glyph2 = brain._add_vertex_glyph
brain._add_vertex_glyph = _func_add
brain._remove_vertex_glyph2 = brain._remove_vertex_glyph
brain._remove_vertex_glyph = _func_remove
# link the initial points
for hemi in initial_points.keys():
if hemi in brain._layered_meshes:
mesh = brain._layered_meshes[hemi]._polydata
for vertex_id in initial_points[hemi]:
self.leader._add_vertex_glyph(hemi, mesh, vertex_id)
if colorbar:
fmin = self.leader._data["fmin"]
fmid = self.leader._data["fmid"]
fmax = self.leader._data["fmax"]
for brain in self.brains:
brain.callbacks["fmin"](fmin)
brain.callbacks["fmid"](fmid)
brain.callbacks["fmax"](fmax)
for name in ('fmin', 'fmid', 'fmax'):
func = getattr(self, "set_" + name)
self.link_widgets(
name=name,
callback=func,
signal_type="floatValueChanged"
)
def set_fmin(self, value):
for brain in self.brains:
brain.callbacks["fmin"](value)
def set_fmid(self, value):
for brain in self.brains:
brain.callbacks["fmid"](value)
def set_fmax(self, value):
for brain in self.brains:
brain.callbacks["fmax"](value)
def set_time_point(self, value):
for brain in self.brains:
brain.callbacks["time"](value, update_widget=True)
def set_playback_speed(self, value):
for brain in self.brains:
brain.callbacks["playback_speed"](value, update_widget=True)
def toggle_playback(self):
value = self.leader.callbacks["time"].widget.get_value()
# synchronize starting points before playback
self.set_time_point(value)
for brain in self.brains:
brain.toggle_playback()
def link_widgets(self, name, callback, signal_type, actions=False):
for brain in self.brains:
if actions:
widget = brain._renderer.actions[name]._action
else:
widget = brain.widgets[name].widget
if widget is not None:
signal = getattr(widget, signal_type)
signal.disconnect()
signal.connect(callback)
def link_cameras(self):
from ..backends._pyvista import _add_camera_callback
def _update_camera(vtk_picker, event):
for brain in self.brains:
brain.plotter.update()
camera = self.leader.plotter.camera
_add_camera_callback(camera, _update_camera)
for brain in self.brains:
for renderer in brain.plotter.renderers:
renderer.camera = camera
|
