class YCrossSection(XCrossSection):
"""
Y-axis cross section widget
parent (QWidget): parent widget
position (string): "left" or "right"
"""
PANEL_ID = ID_YCS
OTHER_PANEL_ID = ID_XCS
CrossSectionPlotKlass = YCrossSectionPlot
def __init__(self, parent=None, position="right", xsection_pos="top"):
self.xsection_pos = xsection_pos
self.spacer = QSpacerItem(0, 0)
super(YCrossSection, self).__init__(parent)
self.cs_plot.set_axis_direction("bottom", reverse=position == "left")
def setup_widget(self):
toolbar = self.toolbar
toolbar.setOrientation(Qt.Horizontal)
layout = QVBoxLayout()
if self.xsection_pos == "top":
layout.addSpacerItem(self.spacer)
layout.addWidget(toolbar)
layout.addWidget(self.cs_plot)
if self.xsection_pos == "bottom":
layout.addSpacerItem(self.spacer)
layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(layout)
def adjust_height(self, height):
self.spacer.changeSize(0, height, QSizePolicy.Fixed, QSizePolicy.Fixed)
self.layout().invalidate()
class QtSearch(QWidget):
"""
A Qt view for a Search model.
"""
def __init__(self, model, *args, **kwargs):
super().__init__(*args, **kwargs)
self.model = model
self.setLayout(QVBoxLayout())
self.layout().setContentsMargins(0, 0, 0, 0)
# A "Back" button
self._back_button = QPushButton("Back")
self._back_button.setEnabled(False)
self.layout().addWidget(self._back_button)
self._back_button.clicked.connect(model.go_back)
# Hook up model Events to Qt Slots.
self.model.events.enter.connect(self.on_enter)
self.model.events.go_back.connect(self.on_go_back)
self.model.events.run_search_ready.connect(self.on_run_search_ready)
self.model.events.run_search_cleared.connect(
self.on_run_search_cleared)
self._selector_widgets = [] # QComboBoxes
self._run_search_widgets = [
] # The SearchInput and SearchOutput widgets
self._vspacer = QSpacerItem(0, 0, QSizePolicy.Minimum,
QSizePolicy.Minimum)
run_search = model.run_search
if run_search:
# The root catalog contains Runs, so immediately display Run Search
# input and output.
self._initialize_run_search(run_search.search_input,
run_search.search_results)
# No need to have a "Back" button in this case
self._back_button.setVisible(False)
else:
# The root catalog is a catalog-of-catalogs, so display a combo
# box.
self._initialize_selector(list(model.current_catalog))
def on_enter(self, event=None):
"We are entering a subcatalog."
names = list(event.catalog)
self._initialize_selector(names)
self._back_button.setEnabled(True)
def _initialize_selector(self, names):
"Create a combobox to select from subcatalogs."
selector = QtSubcatalogSelector(names)
self._selector_widgets.append(selector)
def on_selection(index):
name = names[index]
try:
self.model.enter(name)
except Exception:
logger.exception("Failed to select %r", name)
selector.setCurrentIndex(-1)
else:
selector.setEnabled(False)
selector.activated.connect(on_selection)
self.layout().addWidget(selector)
self._vspacer.changeSize(0, 0, QSizePolicy.Expanding,
QSizePolicy.Expanding)
self.layout().addItem(self._vspacer)
def on_go_back(self, event):
"Move up the tree of subcatalogs by one step."
breadcrumbs = self.model.breadcrumbs
while len(self._selector_widgets) > len(breadcrumbs) + 1:
widget = self._selector_widgets.pop()
widget.close()
self._selector_widgets[-1].setEnabled(True)
self._selector_widgets[-1].setCurrentIndex(-1)
if not breadcrumbs:
# This is the last widget. Disable back button.
self._back_button.setEnabled(False)
def on_run_search_ready(self, event):
"We have a catalog of Runs."
self._initialize_run_search(event.search_input, event.search_results)
self._back_button.setEnabled(True)
def _initialize_run_search(self, search_input, search_results):
"Create search input and output for a catalog of Runs."
# Create run search widgets and stash them as state for removal later.
self._run_search_widgets.extend(
[QtSearchInput(search_input),
QtSearchResults(search_results)])
for w in self._run_search_widgets:
self.layout().addWidget(w)
self._vspacer.changeSize(0, 0, QSizePolicy.Minimum,
QSizePolicy.Minimum)
def on_run_search_cleared(self, event):
"Clear search input and output."
for w in self._run_search_widgets:
w.setParent(None)
self._run_search_widgets.clear()
self._vspacer.changeSize(0, 0, QSizePolicy.Expanding,
QSizePolicy.Expanding)