def test_laser_view_space_apply_dialogs(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
qtbot.addWidget(viewspace)
viewspace.views[0].addLaser(Laser(rand_data("A1")))
viewspace.refresh()
dlg = viewspace.configDialog()
dlg.applyPressed.emit(dlg)
dlg.close()
dlg = viewspace.colortableRangeDialog()
dlg.applyPressed.emit(dlg)
dlg.close()
dlg = viewspace.fontsizeDialog()
dlg.intValueSelected.emit(5)
dlg.close()
assert viewspace.options.font.pointSize() == 5
class MainWindow(QtWidgets.QMainWindow):
"""Pewpew mainwindow, holding a LaserViewSpace.
Actions for the menu and status bars are created and stored here.
"""
def __init__(self, parent: QtWidgets.QWidget = None):
super().__init__(parent)
self.resize(1280, 800)
self.log = LoggingDialog()
self.help = HelpDialog()
self.viewspace = LaserViewSpace()
self.viewspace.numTabsChanged.connect(self.updateActionAvailablity)
self.viewspace.activeViewChanged.connect(self.updateActionAvailablity)
self.setCentralWidget(self.viewspace)
self.createActions()
self.createMenus()
self.statusBar().showMessage(f"Welcome to pew² version {__version__}.")
self.button_status_um = QtWidgets.QRadioButton("μ")
self.button_status_index = QtWidgets.QRadioButton("i")
self.button_status_um.setChecked(True)
self.button_status_um.toggled.connect(self.buttonStatusUnit)
self.button_status_index.toggled.connect(self.buttonStatusUnit)
self.statusBar().addPermanentWidget(self.button_status_um)
self.statusBar().addPermanentWidget(self.button_status_index)
self.updateActionAvailablity()
def createActions(self) -> None:
self.action_about = qAction("help-about", "&About", "About pew².",
self.actionAbout)
self.action_help = qAction("help-contents", "&Help",
"Show the help contents.", self.actionHelp)
self.action_colortable_range = qAction(
"",
"Set &Range",
"Set the range of the colortable.",
self.viewspace.colortableRangeDialog,
)
self.action_colortable_range.setShortcut("Ctrl+R")
self.action_config = qAction(
"document-edit",
"Default Config",
"Edit the default config.",
self.viewspace.configDialog,
)
self.action_config.setShortcut("Ctrl+K")
self.action_exit = qAction("application-exit", "Quit",
"Exit the program.", self.close)
self.action_exit.setShortcut("Ctrl+Shift+Q")
self.action_export_all = qAction(
"document-save-all",
"E&xport All",
"Export all open documents.",
self.actionExportAll,
)
self.action_export_all.setShortcut("Ctrl+Shift+X")
self.action_fontsize = qAction(
"insert-text",
"Fontsize",
"Set the font size in points.",
self.viewspace.fontsizeDialog,
)
self.action_group_colortable = qActionGroup(
self,
list(self.viewspace.options.colortables.keys()),
self.actionGroupColortable,
checked=self.viewspace.options.colortable,
statuses=list(self.viewspace.options.colortables.values()),
)
self.action_smooth = qAction(
"smooth",
"&Smooth",
"Smooth images with bilinear interpolation.",
self.viewspace.toggleSmooth,
)
self.action_smooth.setCheckable(True)
self.action_smooth.setChecked(self.viewspace.options.smoothing)
self.action_wizard_import = qAction(
"",
"Import Wizard",
"Start the line-wise import wizard. .",
self.actionWizardImport,
)
self.action_wizard_spot = qAction(
"",
"Spotwise Wizard",
"Start the import wizard for data collected spot-wise.",
self.actionWizardSpot,
)
self.action_wizard_srr = qAction(
"",
"Kriss Kross Wizard",
"Start the Super-Resolution-Reconstruction import wizard.",
self.actionWizardSRR,
)
self.action_log = qAction("clock", "&Show Log",
"Show the pew² event and error log.",
self.actionLog)
self.action_open = qAction("document-open", "&Open",
"Open new document(s).", self.actionOpen)
self.action_open.setShortcut("Ctrl+O")
self.action_toggle_calibrate = qAction(
"go-top",
"Ca&librate",
"Toggle calibration.",
self.viewspace.toggleCalibrate,
)
self.action_toggle_calibrate.setShortcut("Ctrl+L")
self.action_toggle_calibrate.setCheckable(True)
self.action_toggle_calibrate.setChecked(
self.viewspace.options.calibrate)
self.action_toggle_colorbar = qAction(
"", "Show Colorbar", "Toggle colorbars.",
self.viewspace.setColorbarVisible)
self.action_toggle_colorbar.setCheckable(True)
self.action_toggle_colorbar.setChecked(
self.viewspace.options.items["colorbar"])
self.action_toggle_label = qAction("", "Show Labels",
"Toggle element labels.",
self.viewspace.setLabelVisible)
self.action_toggle_label.setCheckable(True)
self.action_toggle_label.setChecked(
self.viewspace.options.items["label"])
self.action_toggle_scalebar = qAction(
"", "Show Scalebar", "Toggle scalebar.",
self.viewspace.setScalebarVisible)
self.action_toggle_scalebar.setCheckable(True)
self.action_toggle_scalebar.setChecked(
self.viewspace.options.items["scalebar"])
self.action_tool_calculator = qAction(
"document-properties",
"Calculator",
"Open the calculator.",
self.actionToolCalculator,
)
self.action_tool_drift = qAction(
"document-properties",
"Drift Compensation",
"Open the drift compensation tool.",
self.actionToolDrift,
)
self.action_tool_filter = qAction(
"document-properties",
"Filtering",
"Open the filtering tool.",
self.actionToolFilter,
)
self.action_tool_merge = qAction(
"align-vertical-top",
"Merge Tool",
"Open tool for merging multiple images.",
self.actionToolMerge,
)
self.action_tool_standards = qAction(
"document-properties",
"Calibration Standards",
"Open the standards calibration tool.",
self.actionToolStandards,
)
self.action_tool_overlay = qAction(
"document-properties",
"Image Overlay",
"Open the overlay tool.",
self.actionToolOverlay,
)
self.action_transform_flip_horizontal = qAction(
"object-flip-horizontal",
"Flip Horizontal",
"Flip the image about vertical axis.",
self.actionTransformFlipHorz,
)
self.action_transform_flip_vertical = qAction(
"object-flip-vertical",
"Flip Vertical",
"Flip the image about horizontal axis.",
self.actionTransformFlipVert,
)
self.action_transform_rotate_left = qAction(
"object-rotate-left",
"Rotate Left",
"Rotate the image 90° counter clockwise.",
self.actionTransformRotateLeft,
)
self.action_transform_rotate_right = qAction(
"object-rotate-right",
"Rotate Right",
"Rotate the image 90° clockwise.",
self.actionTransformRotateRight,
)
self.action_refresh = qAction("view-refresh", "Refresh",
"Redraw documents.",
self.viewspace.refresh)
self.action_refresh.setShortcut("F5")
def actionAbout(self) -> QtWidgets.QDialog:
dlg = QtWidgets.QMessageBox(
QtWidgets.QMessageBox.Information,
"About pew²",
("Import, process and export of LA-ICP-MS data.\n"
f"Version {__version__}\n"
"Developed by the Atomic Medicine Initiative.\n"
"https://github.com/djdt/pewpew"),
parent=self,
)
if self.windowIcon() is not None:
dlg.setIconPixmap(self.windowIcon().pixmap(64, 64))
dlg.open()
return dlg
def actionExportAll(self) -> QtWidgets.QDialog:
widgets = [
w for v in self.viewspace.views for w in v.widgets()
if isinstance(w, LaserWidget)
]
dlg = ExportAllDialog(widgets, self)
dlg.open()
return dlg
def actionGroupColortable(self, action: QtWidgets.QAction) -> None:
text = action.text().replace("&", "")
self.viewspace.options.colortable = text
self.viewspace.refresh()
def actionHelp(self) -> None:
self.help.show()
def actionLog(self) -> None:
self.log.show()
def actionOpen(self) -> QtWidgets.QDialog:
view = self.viewspace.activeView()
return view.actionOpen()
def openTool(self, tool: ToolWidget, name: str) -> None:
widget = self.viewspace.activeWidget()
if widget is None:
return
index = widget.index
if isinstance(widget, ToolWidget):
widget = widget.widget
tool = tool(widget)
name = f"{name}: {widget.laserName()}"
widget.view.removeTab(index)
widget.view.insertTab(index, name, tool)
tool.activate()
def actionToolCalculator(self) -> None:
self.openTool(CalculatorTool, "Calculator")
def actionToolDrift(self) -> None:
self.openTool(DriftTool, "Drift")
def actionToolFilter(self) -> None:
self.openTool(FilteringTool, "Filter")
def actionToolMerge(self) -> None:
self.openTool(MergeTool, "Merge")
def actionToolStandards(self) -> None:
self.openTool(StandardsTool, "Standards")
def actionToolOverlay(self) -> None:
self.openTool(OverlayTool, "Overlay")
def actionTransformFlipHorz(self) -> None:
widget = self.viewspace.activeWidget()
if widget is None:
return
widget.transform(flip="horizontal")
def actionTransformFlipVert(self) -> None:
widget = self.viewspace.activeWidget()
if widget is None:
return
widget.transform(flip="vertical")
def actionTransformRotateLeft(self) -> None:
widget = self.viewspace.activeWidget()
if widget is None:
return
widget.transform(rotate="left")
def actionTransformRotateRight(self) -> None:
widget = self.viewspace.activeWidget()
if widget is None:
return
widget.transform(rotate="right")
def actionWizardImport(self) -> QtWidgets.QWizard:
wiz = ImportWizard(config=self.viewspace.config, parent=self)
wiz.laserImported.connect(self.viewspace.activeView().addLaser)
wiz.open()
return wiz
def actionWizardSpot(self) -> QtWidgets.QWizard:
config = SpotConfig(self.viewspace.config.spotsize,
self.viewspace.config.spotsize)
wiz = SpotImportWizard(config=config,
options=self.viewspace.options,
parent=self)
wiz.laserImported.connect(self.viewspace.activeView().addLaser)
wiz.open()
return wiz
def actionWizardSRR(self) -> QtWidgets.QWizard:
wiz = SRRImportWizard(config=self.viewspace.config, parent=self)
wiz.laserImported.connect(self.viewspace.activeView().addLaser)
wiz.open()
return wiz
def createMenus(self) -> None:
# File
menu_file = self.menuBar().addMenu("&File")
menu_file.addAction(self.action_open)
# File -> Import
menu_import = menu_file.addMenu("&Import")
menu_import.addAction(self.action_wizard_import)
menu_import.addAction(self.action_wizard_spot)
menu_import.addAction(self.action_wizard_srr)
menu_file.addSeparator()
menu_file.addAction(self.action_export_all)
menu_file.addSeparator()
menu_file.addAction(self.action_exit)
# Edit
menu_edit = self.menuBar().addMenu("&Edit")
menu_edit.addAction(self.action_config)
menu_edit.addAction(self.action_toggle_calibrate)
menu_edit.addSeparator()
menu_edit.addAction(self.action_transform_flip_horizontal)
menu_edit.addAction(self.action_transform_flip_vertical)
menu_edit.addAction(self.action_transform_rotate_left)
menu_edit.addAction(self.action_transform_rotate_right)
menu_tools = self.menuBar().addMenu("&Tools")
menu_tools.addAction(self.action_tool_calculator)
menu_tools.addAction(self.action_tool_drift)
menu_tools.addAction(self.action_tool_filter)
menu_tools.addAction(self.action_tool_merge)
menu_tools.addAction(self.action_tool_standards)
menu_tools.addAction(self.action_tool_overlay)
# View
menu_view = self.menuBar().addMenu("&View")
menu_cmap = menu_view.addMenu("&Colortable")
menu_cmap.setIcon(QtGui.QIcon.fromTheme("color-management"))
menu_cmap.setStatusTip("Colortable of displayed images.")
menu_cmap.addActions(self.action_group_colortable.actions())
menu_cmap.addAction(self.action_colortable_range)
# View - interpolation
menu_view.addAction(self.action_smooth)
menu_view.addAction(self.action_fontsize)
menu_view.addSeparator()
menu_view.addAction(self.action_toggle_colorbar)
menu_view.addAction(self.action_toggle_label)
menu_view.addAction(self.action_toggle_scalebar)
menu_view.addSeparator()
menu_view.addAction(self.action_refresh)
# Help
menu_help = self.menuBar().addMenu("&Help")
menu_help.addAction(self.action_log)
menu_help.addAction(self.action_help)
menu_help.addAction(self.action_about)
def buttonStatusUnit(self, toggled: bool) -> None:
"""Callback for 'button_status_um'."""
if self.button_status_um.isChecked():
self.viewspace.options.units = "μm"
elif self.button_status_index.isChecked():
self.viewspace.options.units = "index"
def updateActionAvailablity(self) -> None:
"""Enables tools if at least one view is present."""
enabled = self.viewspace.countViewTabs() > 0
self.action_export_all.setEnabled(enabled)
# Tools require an active view
enabled = enabled and self.viewspace.activeView().tabs.count() > 0
self.action_tool_calculator.setEnabled(enabled)
self.action_tool_drift.setEnabled(enabled)
self.action_tool_filter.setEnabled(enabled)
self.action_tool_standards.setEnabled(enabled)
self.action_tool_overlay.setEnabled(enabled)
def exceptHook(self, etype: type, value: BaseException,
tb: TracebackType) -> None: # pragma: no cover
"""Redirect errors to the log."""
if etype == KeyboardInterrupt:
logger.info("Keyboard interrupt, exiting.")
sys.exit(1)
logger.exception("Uncaught exception", exc_info=(etype, value, tb))
QtWidgets.QMessageBox.critical(self, "Uncaught Exception", str(value))