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
|
# Authors: Alexandre Gramfort <alexandre.gramfort@inria.fr>
# Eric Larson <larson.eric.d@gmail.com>
# Guillaume Favelier <guillaume.favelier@gmail.com>
#
# License: Simplified BSD
import weakref
from ...utils import logger
class TimeCallBack(object):
"""Callback to update the time."""
def __init__(self, brain=None, callback=None):
self.brain = brain
self.callback = callback
self.widget = None
self.label = None
if self.brain is not None and callable(self.brain._data['time_label']):
self.time_label = self.brain._data['time_label']
else:
self.time_label = None
def __call__(self, value, update_widget=False, time_as_index=True):
"""Update the time slider."""
if not time_as_index:
value = self.brain._to_time_index(value)
self.brain.set_time_point(value)
if self.label is not None:
current_time = self.brain._current_time
self.label.set_value(f"{current_time: .3f}")
if self.callback is not None:
self.callback()
if self.widget is not None and update_widget:
self.widget.set_value(int(value))
class UpdateColorbarScale(object):
"""Class to update the values of the colorbar sliders."""
def __init__(self, brain, factor):
self.brain = brain
self.factor = factor
self.widget = None
self.widgets = {key: None for key in self.brain.keys}
def __call__(self):
"""Update the colorbar sliders."""
self.brain._update_fscale(self.factor)
for key in self.brain.keys:
if self.widgets[key] is not None:
self.widgets[key].set_value(self.brain._data[key])
class UpdateLUT(object):
"""Update the LUT."""
def __init__(self, brain=None):
self.brain = weakref.ref(brain)
self.widgets = {key: list() for key in brain.keys}
def __call__(self, fmin=None, fmid=None, fmax=None):
"""Update the colorbar sliders."""
brain = self.brain()
if brain is None:
return
brain.update_lut(fmin=fmin, fmid=fmid, fmax=fmax)
with brain._no_lut_update(f'UpdateLUT {fmin} {fmid} {fmax}'):
for key in ('fmin', 'fmid', 'fmax'):
value = brain._data[key]
logger.debug(f'Updating {key} = {value}')
for widget in self.widgets[key]:
widget.set_value(value)
class ShowView(object):
"""Class that selects the correct view."""
def __init__(self, brain=None, data=None):
self.brain = brain
self.data = data
self.widget = None
def __call__(self, value, update_widget=False):
"""Update the view."""
if "renderer" in self.brain.widgets:
idx = self.brain.widgets["renderer"].get_value()
else:
idx = 0
idx = int(idx)
if self.data[idx] is not None:
self.brain.show_view(
value,
row=self.data[idx]['row'],
col=self.data[idx]['col'],
hemi=self.data[idx]['hemi'],
)
if update_widget and self.widget is not None:
self.widget.set_value(value)
class SmartCallBack(object):
"""Class to manage smart slider.
It stores it's own slider representation for efficiency
and uses it when necessary.
"""
def __init__(self, callback=None):
self.callback = callback
self.widget = None
def __call__(self, value, update_widget=False):
"""Update the value."""
self.callback(value)
if self.widget is not None and update_widget:
self.widget.set_value(value)
|
