class DendroWidget(DataViewer):
"""
An interactive dendrogram display
"""
LABEL = 'Dendrogram'
_property_set = DataViewer._property_set + \
'ylog height parent order'.split()
ylog = ButtonProperty('ui.ylog', 'log scaling on y axis?')
height = CurrentComboProperty('ui.heightCombo', 'height attribute')
parent = CurrentComboProperty('ui.parentCombo', 'parent attribute')
order = CurrentComboProperty('ui.orderCombo', 'layout sorter attribute')
_toolbar_cls = MatplotlibViewerToolbar
tools = ['Pick']
def __init__(self, session, parent=None):
super(DendroWidget, self).__init__(session, parent)
self.central_widget = MplWidget()
self.option_widget = QtWidgets.QWidget()
self.setCentralWidget(self.central_widget)
self.ui = load_ui('options_widget.ui',
self.option_widget,
directory=os.path.dirname(__file__))
self.client = DendroClient(
self._data,
self.central_widget.canvas.fig,
layer_artist_container=self._layer_artist_container)
self._connect()
self.initialize_toolbar()
self.statusBar().setSizeGripEnabled(False)
def _connect(self):
ui = self.ui
cl = self.client
connect_bool_button(cl, 'ylog', ui.ylog)
connect_current_combo(cl, 'parent_attr', ui.parentCombo)
connect_current_combo(cl, 'height_attr', ui.heightCombo)
connect_current_combo(cl, 'order_attr', ui.orderCombo)
def initialize_toolbar(self):
super(DendroWidget, self).initialize_toolbar()
def on_move(mode):
if mode._drag:
self.client.apply_roi(mode.roi())
self.toolbar.tools['Pick']._move_callback = on_move
def apply_roi(self, roi):
self.client.apply_roi(roi)
def _update_combos(self, data=None):
data = data or self.client.display_data
if data is None:
return
for combo in [
self.ui.heightCombo, self.ui.parentCombo, self.ui.orderCombo
]:
combo.blockSignals(True)
ids = []
idx = combo.currentIndex()
old = combo.itemData(idx) if idx > 0 else None
combo.clear()
for cid in data.components:
if cid.hidden and cid is not data.pixel_component_ids[0]:
continue
combo.addItem(cid.label, userData=cid)
ids.append(cid)
try:
combo.setCurrentIndex(ids.index(old))
except ValueError:
combo.setCurrentIndex(0)
combo.blockSignals(False)
def add_data(self, data):
"""Add a new data set to the widget
:returns: True if the addition was expected, False otherwise
"""
if data in self.client:
return
self._update_combos(data)
self.client.add_layer(data)
return True
def add_subset(self, subset):
"""Add a subset to the widget
:returns: True if the addition was accepted, False otherwise
"""
self.add_data(subset.data)
if subset.data in self.client:
self.client.add_layer(subset)
return True
def register_to_hub(self, hub):
super(DendroWidget, self).register_to_hub(hub)
self.client.register_to_hub(hub)
hub.subscribe(self, core.message.ComponentsChangedMessage,
nonpartial(self._update_combos()))
def unregister(self, hub):
super(DendroWidget, self).unregister(hub)
hub.unsubscribe_all(self.client)
hub.unsubscribe_all(self)
def options_widget(self):
return self.option_widget
@defer_draw
def restore_layers(self, rec, context):
from glue.core.callback_property import delay_callback
with delay_callback(self.client, 'height_attr', 'parent_attr',
'order_attr'):
self.client.restore_layers(rec, context)
self._update_combos()
class DendroWidget(DataViewer):
"""
An interactive dendrogram display
"""
LABEL = 'Dendrogram'
_property_set = DataViewer._property_set + \
'ylog height parent order'.split()
ylog = ButtonProperty('ui.ylog', 'log scaling on y axis?')
height = CurrentComboProperty('ui.heightCombo', 'height attribute')
parent = CurrentComboProperty('ui.parentCombo', 'parent attribute')
order = CurrentComboProperty('ui.orderCombo', 'layout sorter attribute')
_toolbar_cls = MatplotlibViewerToolbar
tools = ['Pick']
def __init__(self, session, parent=None):
super(DendroWidget, self).__init__(session, parent)
self.central_widget = MplWidget()
self.option_widget = QtWidgets.QWidget()
self.setCentralWidget(self.central_widget)
self.ui = load_ui('options_widget.ui', self.option_widget,
directory=os.path.dirname(__file__))
self.client = DendroClient(self._data,
self.central_widget.canvas.fig,
layer_artist_container=self._layer_artist_container)
self._connect()
self.statusBar().setSizeGripEnabled(False)
def _connect(self):
ui = self.ui
cl = self.client
connect_bool_button(cl, 'ylog', ui.ylog)
connect_current_combo(cl, 'parent_attr', ui.parentCombo)
connect_current_combo(cl, 'height_attr', ui.heightCombo)
connect_current_combo(cl, 'order_attr', ui.orderCombo)
def initialize_toolbar(self):
super(DendroWidget, self).initialize_toolbar()
def on_move(mode):
if mode._drag:
self.client.apply_roi(mode.roi())
self.toolbar.tools['Pick']._move_callback = on_move
def apply_roi(self, roi):
self.client.apply_roi(roi)
def _update_combos(self, data=None):
data = data or self.client.display_data
if data is None:
return
for combo in [self.ui.heightCombo,
self.ui.parentCombo,
self.ui.orderCombo]:
combo.blockSignals(True)
ids = []
idx = combo.currentIndex()
old = combo.itemData(idx) if idx > 0 else None
combo.clear()
for cid in data.components:
if cid.hidden and cid is not data.pixel_component_ids[0]:
continue
combo.addItem(cid.label, userData=cid)
ids.append(cid)
try:
combo.setCurrentIndex(ids.index(old))
except ValueError:
combo.setCurrentIndex(0)
combo.blockSignals(False)
def add_data(self, data):
"""Add a new data set to the widget
:returns: True if the addition was expected, False otherwise
"""
if data in self.client:
return
self._update_combos(data)
self.client.add_layer(data)
return True
def add_subset(self, subset):
"""Add a subset to the widget
:returns: True if the addition was accepted, False otherwise
"""
self.add_data(subset.data)
if subset.data in self.client:
self.client.add_layer(subset)
return True
def register_to_hub(self, hub):
super(DendroWidget, self).register_to_hub(hub)
self.client.register_to_hub(hub)
hub.subscribe(self, core.message.ComponentsChangedMessage,
nonpartial(self._update_combos()))
def unregister(self, hub):
super(DendroWidget, self).unregister(hub)
hub.unsubscribe_all(self.client)
hub.unsubscribe_all(self)
def options_widget(self):
return self.option_widget
@defer_draw
def restore_layers(self, rec, context):
from glue.core.callback_property import delay_callback
with delay_callback(self.client, 'height_attr',
'parent_attr',
'order_attr'):
self.client.restore_layers(rec, context)
self._update_combos()