datalab-platform 0.0.1.dev0__py3-none-any.whl → 1.0.1__py3-none-any.whl

datalab/gui/docks.py

@@ -0,0 +1,473 @@
+# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
+
+"""
+Docks
+=====
+
+The :mod:`datalab.gui.docks` module provides the dockable widgets for the
+DataLab main window.
+
+Plot widget
+-----------
+
+.. autoclass:: DataLabPlotWidget
+
+Dockable plot widget
+--------------------
+
+.. autoclass:: DockablePlotWidget
+"""
+
+from __future__ import annotations
+
+import warnings
+from typing import TYPE_CHECKING
+
+import numpy as np
+import scipy.integrate as spt
+from guidata.configtools import get_icon, get_image_file_path
+from guidata.qthelpers import create_action, is_dark_theme
+from guidata.widgets.dockable import DockableWidget
+from plotpy.constants import PlotType
+from plotpy.items import CurveItem
+from plotpy.panels import XCrossSection, YCrossSection
+from plotpy.plot import PlotOptions, PlotWidget
+from plotpy.tools import (
+    BasePlotMenuTool,
+    CurveStatsTool,
+    DeleteItemTool,
+    DisplayCoordsTool,
+    DoAutoscaleTool,
+    EditItemDataTool,
+    ExportItemDataTool,
+    ImageStatsTool,
+    ItemCenterTool,
+    RectangularSelectionTool,
+    RectZoomTool,
+    SelectTool,
+    YRangeCursorTool,
+)
+from plotpy.tools.image import get_stats as get_image_stats
+from qtpy import QtCore as QC
+from qtpy import QtGui as QG
+from qtpy import QtWidgets as QW
+from qtpy.QtWidgets import QApplication, QMainWindow
+from sigima.objects import create_signal
+from sigima.tools.signal import pulse
+from skimage import measure
+
+from datalab.config import APP_NAME, Conf, _
+
+if TYPE_CHECKING:
+    from plotpy.items.image.base import BaseImageItem
+    from plotpy.plot import BasePlot
+    from plotpy.styles import BaseImageParam
+
+
+class CurveStatsToolFunctions:
+    """Statistical functions for `CurveStatsTool` and `YRangeCursorTool`"""
+
+    @classmethod
+    def set_labelfuncs(cls, statstool: CurveStatsTool) -> None:
+        """Set label functions for the statistics tool"""
+        if isinstance(statstool, CurveStatsTool):
+            labelfuncs = (
+                ("%g < x < %g", lambda *args: cls.nan_min_max(args[0])),
+                ("%g < y < %g", lambda *args: cls.nan_min_max(args[1])),
+                ("∆x=%g", lambda *args: cls.nan_delta(args[0])),
+                ("∆y=%g", lambda *args: cls.nan_delta(args[1])),
+                ("<y>=%g", lambda *args: cls.nan_mean(args[1])),
+                ("σ(y)=%g", lambda *args: cls.nan_std(args[1])),
+                ("∑(y)=%g", lambda *args: spt.trapezoid(args[1])),
+                ("∫ydx=%g<br>", lambda *args: spt.trapezoid(args[1], args[0])),
+                ("FWHM=%s", cls.fwhm_info),
+                ("∆x<sub>RISE 10-90</sub>=%s", cls.rise_time_info),
+                (
+                    "∆x<sub>RISE 20-80</sub>=%s",
+                    lambda x, y: cls.rise_time_info(x, y, 0.2, 0.8),
+                ),
+                ("∆x<sub>FALL 90-10</sub>=%s", cls.fall_time_info),
+                (
+                    "∆x<sub>FALL 80-20</sub>=%s",
+                    lambda x, y: cls.fall_time_info(x, y, 0.8, 0.2),
+                ),
+            )
+        else:  # YRangeCursorTool
+            labelfuncs = (
+                ("%g &lt; y &lt; %g", lambda ymin, ymax: (ymin, ymax)),
+                ("∆y=%g", lambda ymin, ymax: ymax - ymin),
+            )
+        statstool.set_labelfuncs(labelfuncs)
+
+    @staticmethod
+    def nan_min_max(arr: np.ndarray) -> tuple[float, float]:
+        """Return min/max tuple"""
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", RuntimeWarning)
+            min_val = np.nanmin(arr)
+            max_val = np.nanmax(arr)
+        return (min_val, max_val)
+
+    @staticmethod
+    def nan_delta(arr: np.ndarray) -> float:
+        """Return delta value, ignoring NaNs"""
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", RuntimeWarning)
+            delta_val = np.nanmax(arr) - np.nanmin(arr)
+        return delta_val
+
+    @staticmethod
+    def nan_mean(arr: np.ndarray) -> float:
+        """Return mean value, ignoring NaNs"""
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", RuntimeWarning)
+            mean_val = np.nanmean(arr)
+        return mean_val
+
+    @staticmethod
+    def nan_std(arr: np.ndarray) -> float:
+        """Return standard deviation, ignoring NaNs"""
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", RuntimeWarning)
+            std_val = np.nanstd(arr)
+        return std_val
+
+    @staticmethod
+    def fwhm_info(x, y):
+        """Return FWHM information string"""
+        try:
+            with warnings.catch_warnings(record=True) as w:
+                x0, _y0, x1, _y1 = pulse.fwhm(x, y, "zero-crossing")
+                wstr = " ⚠️" if w else ""
+        except (ValueError, ZeroDivisionError, pulse.InvalidSignalError):
+            return "🛑"
+        return f"{x1 - x0:g}{wstr}"
+
+    @staticmethod
+    def rise_time_info(x, y, start_ratio=0.1, end_ratio=0.9):
+        """Return rise time information string"""
+        try:
+            with warnings.catch_warnings(record=True) as w:
+                dt = pulse.get_rise_time(x, y, start_ratio, end_ratio)
+                wstr = " ⚠️" if w else ""
+            if dt is None:
+                return "🛑"
+        except (ValueError, ZeroDivisionError, pulse.InvalidSignalError):
+            return "🛑"
+        return f"{dt:g}{wstr}"
+
+    @staticmethod
+    def fall_time_info(x, y, start_ratio=0.9, end_ratio=0.1):
+        """Return fall time information string"""
+        try:
+            with warnings.catch_warnings(record=True) as w:
+                dt = pulse.get_fall_time(x, y, start_ratio, end_ratio)
+                wstr = " ⚠️" if w else ""
+            if dt is None:
+                return "🛑"
+        except (ValueError, ZeroDivisionError, pulse.InvalidSignalError):
+            return "🛑"
+        return f"{dt:g}{wstr}"
+
+
+def get_more_image_stats(
+    item: BaseImageItem,
+    x0: float,
+    y0: float,
+    x1: float,
+    y1: float,
+) -> str:
+    """Return formatted string with stats on image rectangular area
+    (output should be compatible with AnnotatedShape.get_info)
+
+    Args:
+        item: image item
+        x0: X0
+        y0: Y0
+        x1: X1
+        y1: Y1
+    """
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore", UserWarning)
+        info = get_image_stats(item, x0, y0, x1, y1)
+
+    ix0, iy0, ix1, iy1 = item.get_closest_index_rect(x0, y0, x1, y1)
+    data = item.data[iy0:iy1, ix0:ix1]
+    p: BaseImageParam = item.param
+    xunit, yunit, zunit = p.get_units()
+
+    integral = np.nansum(data)
+    integral_fmt = r"%.3e " + zunit
+    info += f"<br>∑ = {integral_fmt % integral}"
+
+    if xunit == yunit:
+        surfacefmt = p.xformat.split()[0] + " " + xunit
+        if xunit != "":
+            surfacefmt = surfacefmt + "²"
+        surface = abs((x1 - x0) * (y1 - y0))
+        info += f"<br>A = {surfacefmt % surface}"
+        if xunit is not None and zunit is not None:
+            if surface != 0:
+                density = integral / surface
+                densityfmt = r"%.3e"
+                if xunit and zunit:
+                    densityfmt += " " + zunit + "/" + xunit + "²"
+                info = info + f"<br>ρ = {densityfmt % density}"
+
+    c_i, c_j = measure.centroid(data)
+    c_x, c_y = item.get_plot_coordinates(c_j + ix0, c_i + iy0)
+    info += "<br>" + "<br>".join(
+        [
+            "C|x = " + p.xformat % c_x,
+            "C|y = " + p.yformat % c_y,
+        ]
+    )
+
+    return info
+
+
+def profile_to_signal(plot: BasePlot, panel: XCrossSection | YCrossSection) -> None:
+    """Send cross section curve to DataLab's signal list
+
+    Args:
+        panel: Cross section panel
+    """
+    win = None
+    for win in QApplication.topLevelWidgets():
+        if isinstance(win, QMainWindow):
+            break
+    if win is None or win.objectName() != APP_NAME:
+        # pylint: disable=import-outside-toplevel
+        # pylint: disable=cyclic-import
+        from datalab.gui import main
+
+        # Note : this is the only way to retrieve the DataLab main window instance
+        # when the CrossSectionItem object is embedded into an image widget
+        # parented to another main window.
+        win = main.DLMainWindow.get_instance()
+        assert win is not None  # Should never happen
+
+    for item in panel.cs_plot.get_items():
+        if not isinstance(item, CurveItem):
+            continue
+        x, y, _dx, _dy = item.get_data()
+        if x is None or y is None or x.size == 0 or y.size == 0:
+            continue
+
+        signal = create_signal(item.param.label)
+
+        if isinstance(panel, YCrossSection):
+            signal.set_xydata(y, x)
+            xaxis_name = "left"
+            xunit = plot.get_axis_unit("bottom")
+            if xunit:
+                signal.title += " " + xunit
+        else:
+            signal.set_xydata(x, y)
+            xaxis_name = "bottom"
+            yunit = plot.get_axis_unit("left")
+            if yunit:
+                signal.title += " " + yunit
+
+        signal.ylabel = plot.get_axis_title("right")
+        signal.yunit = plot.get_axis_unit("right")
+        signal.xlabel = plot.get_axis_title(xaxis_name)
+        signal.xunit = plot.get_axis_unit(xaxis_name)
+
+        win.signalpanel.add_object(signal)
+
+    # Show DataLab main window on top, if not already visible
+    win.show()
+    win.raise_()
+
+
+class DataLabPlotWidget(PlotWidget):
+    """DataLab PlotWidget
+
+    This class is a subclass of `plotpy.plot.PlotWidget` that provides a
+    customized widget for DataLab, with a specific set of tools and a
+    customized appearance.
+
+    Args:
+        plot_type: Plot type
+    """
+
+    def __init__(self, plot_type: PlotType) -> None:
+        # Get autoscale margin from configuration based on plot type
+        if plot_type == PlotType.CURVE:
+            autoscale_margin = Conf.view.sig_autoscale_margin_percent.get()
+        elif plot_type == PlotType.IMAGE:
+            autoscale_margin = Conf.view.ima_autoscale_margin_percent.get()
+        else:
+            # For AUTO or MANUAL types, use signal margin as default
+            autoscale_margin = Conf.view.sig_autoscale_margin_percent.get()
+
+        super().__init__(
+            options=PlotOptions(
+                type=plot_type,
+                show_axes_tab=False,
+                autoscale_margin_percent=autoscale_margin,
+            ),
+            toolbar=True,
+        )
+
+    def __register_standard_tools(self) -> None:
+        """Register standard tools
+
+        The only differences with the `manager.register_standard_tools` method are
+        the following:
+
+        1. We don't register the `BasePlotMenuTool, "axes"` tool, because it is not
+        compatible with DataLab's approach to axes management.
+        2. We don't register the `ItemListPanelTool` tool (this intends to prevent
+        the user from accessing the item list panel, and thus, the parameters of all
+        the items - some of them are read-only and should not be modified, like the
+        annotations for example).
+        """
+        mgr = self.manager
+        select_tool = mgr.add_tool(SelectTool)
+        mgr.set_default_tool(select_tool)
+        mgr.add_tool(RectangularSelectionTool, intersect=False)
+        mgr.add_tool(RectZoomTool)
+        mgr.add_tool(DoAutoscaleTool)
+        mgr.add_tool(BasePlotMenuTool, "item")
+        mgr.add_tool(ExportItemDataTool)
+        mgr.add_tool(EditItemDataTool)
+        mgr.add_tool(ItemCenterTool)
+        mgr.add_tool(DeleteItemTool)
+        mgr.add_separator_tool()
+        mgr.add_tool(BasePlotMenuTool, "grid")
+        mgr.add_tool(DisplayCoordsTool)
+
+    def __register_other_tools(self) -> None:
+        """Register other tools"""
+        mgr = self.manager
+        mgr.add_separator_tool()
+        if self.options.type == PlotType.CURVE:
+            mgr.register_curve_tools()
+            xstatstool = mgr.get_tool(CurveStatsTool)
+            CurveStatsToolFunctions.set_labelfuncs(xstatstool)
+            ystatstool = mgr.get_tool(YRangeCursorTool)
+            CurveStatsToolFunctions.set_labelfuncs(ystatstool)
+        else:
+            mgr.register_image_tools()
+            # Customizing the ImageStatsTool
+            statstool = mgr.get_tool(ImageStatsTool)
+            statstool.set_stats_func(get_more_image_stats, replace=True)
+            # Customizing the X and Y cross section panels
+            plot = mgr.get_plot()
+            for panel in (mgr.get_xcs_panel(), mgr.get_ycs_panel()):
+                to_signal_action = create_action(
+                    panel,
+                    _("Process signal"),
+                    icon=get_icon("to_signal.svg"),
+                    triggered=lambda panel=panel: profile_to_signal(plot, panel),
+                )
+                tb = panel.toolbar
+                tb.insertSeparator(tb.actions()[0])
+                tb.insertAction(tb.actions()[0], to_signal_action)
+
+        mgr.add_separator_tool()
+        mgr.register_other_tools()
+        mgr.add_separator_tool()
+        mgr.update_tools_status()
+        mgr.get_default_tool().activate()
+
+    def register_tools(self) -> None:
+        """Register the plotting tools according to the plot type"""
+        self.__register_standard_tools()
+        self.__register_other_tools()
+
+
+class DockablePlotWidget(DockableWidget):
+    """Docked plotting widget
+
+    Args:
+        parent: Parent widget
+        plot_type: Plot type
+    """
+
+    LOCATION = QC.Qt.RightDockWidgetArea
+
+    def __init__(
+        self,
+        parent: QW.QWidget,
+        plot_type: PlotType,
+    ) -> None:
+        super().__init__(parent)
+        self.plotwidget = DataLabPlotWidget(plot_type)
+        self.toolbar = self.plotwidget.get_toolbar()
+        self.watermark = QW.QLabel()
+        original_image = QG.QPixmap(get_image_file_path("DataLab-watermark.png"))
+        self.watermark.setPixmap(original_image)
+        self.setup_layout()
+        self.setup_plotwidget()
+
+    def __get_toolbar_row_col(self) -> tuple[int, int]:
+        """Return toolbar row and column"""
+        tb_pos = Conf.view.plot_toolbar_position.get()
+        tb_col, tb_row = 1, 1
+        if tb_pos in ("left", "right"):
+            self.toolbar.setOrientation(QC.Qt.Vertical)
+            tb_col = 0 if tb_pos == "left" else 2
+        else:
+            self.toolbar.setOrientation(QC.Qt.Horizontal)
+            tb_row = 0 if tb_pos == "top" else 2
+        return tb_row, tb_col
+
+    def setup_layout(self) -> None:
+        """Setup layout"""
+        tb_row, tb_col = self.__get_toolbar_row_col()
+        layout = QW.QGridLayout()
+        layout.addWidget(self.toolbar, tb_row, tb_col)
+        layout.addWidget(self.plotwidget, 1, 1)
+        layout.addWidget(self.watermark, 1, 1, QC.Qt.AlignCenter)
+        self.setLayout(layout)
+
+    def update_toolbar_position(self) -> None:
+        """Update toolbar position"""
+        tb_row, tb_col = self.__get_toolbar_row_col()
+        layout = self.layout()
+        layout.removeWidget(self.toolbar)
+        layout.addWidget(self.toolbar, tb_row, tb_col)
+
+    def setup_plotwidget(self) -> None:
+        """Setup plotting widget"""
+        title = self.toolbar.windowTitle()
+        self.plotwidget.get_manager().add_toolbar(self.toolbar, title)
+        #  Customizing widget appearances
+        self.update_color_mode()
+        plot = self.plotwidget.get_plot()
+        canvas = plot.canvas()
+        canvas.setFrameStyle(canvas.Plain | canvas.NoFrame)
+        plot.SIG_ITEMS_CHANGED.connect(self.update_watermark)
+
+    def update_color_mode(self) -> None:
+        """Update plot widget styles according to application color mode"""
+        if is_dark_theme():
+            palette = QApplication.instance().palette()
+        else:
+            palette = QG.QPalette(QC.Qt.white)
+        for widget in (self.plotwidget, self.plotwidget.get_plot(), self):
+            widget.setBackgroundRole(QG.QPalette.Window)
+            widget.setAutoFillBackground(True)
+            widget.setPalette(palette)
+
+    def get_plot(self) -> BasePlot:
+        """Return plot instance"""
+        return self.plotwidget.get_plot()
+
+    def update_watermark(self, plot: BasePlot) -> None:
+        """Update watermark visibility"""
+        items = plot.get_items()
+        if self.plotwidget.options.type == PlotType.IMAGE:
+            enabled = len(items) <= 1
+        else:
+            enabled = len(items) <= 2
+        self.watermark.setVisible(enabled)
+
+    # ------DockableWidget API
+    def visibility_changed(self, enable: bool) -> None:
+        """DockWidget visibility has changed"""
+        DockableWidget.visibility_changed(self, enable)
+        self.toolbar.setVisible(enable)

datalab/gui/h5io.py

@@ -0,0 +1,150 @@
+# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
+
+"""
+HDF5 I/O
+========
+
+The :mod:`datalab.gui.h5io` module provides the HDF5 file open/save into/from
+DataLab data model/main window.
+
+.. autoclass:: H5InputOutput
+"""
+
+from __future__ import annotations
+
+import os.path as osp
+from typing import TYPE_CHECKING
+
+from guidata.qthelpers import exec_dialog
+from qtpy import QtWidgets as QW
+from sigima.objects import SignalObj
+
+from datalab.config import _
+from datalab.env import execenv
+from datalab.h5 import H5Importer
+from datalab.h5.native import NativeH5Reader, NativeH5Writer
+from datalab.utils.qthelpers import create_progress_bar, qt_try_loadsave_file
+from datalab.widgets.h5browser import H5BrowserDialog
+
+if TYPE_CHECKING:
+    from datalab.gui.main import DLMainWindow
+    from datalab.h5.common import BaseNode
+
+
+class H5InputOutput:
+    """Object handling HDF5 file open/save into/from DataLab data model/main window
+
+    Args:
+        mainwindow: Main window
+    """
+
+    def __init__(self, mainwindow: DLMainWindow) -> None:
+        self.mainwindow = mainwindow
+        self.uint32_wng: bool = None
+
+    @staticmethod
+    def __progbartitle(fname: str) -> str:
+        """Return progress bar title"""
+        return _("Loading data from %s...") % osp.basename(fname)
+
+    def save_file(self, filename: str) -> None:
+        """Save all signals and images from DataLab model into a HDF5 file"""
+        writer = NativeH5Writer(filename)
+        for panel in self.mainwindow.panels:
+            panel.serialize_to_hdf5(writer)
+        writer.close()
+
+    def open_file(self, filename: str, import_all: bool, reset_all: bool) -> None:
+        """Open HDF5 file"""
+        progress = None
+        try:
+            reader = NativeH5Reader(filename)
+            if reset_all:
+                self.mainwindow.reset_all()
+            with create_progress_bar(
+                self.mainwindow, self.__progbartitle(filename), 2
+            ) as progress:
+                for idx, panel in enumerate(self.mainwindow.panels):
+                    progress.setValue(idx + 1)
+                    QW.QApplication.processEvents()
+                    panel.deserialize_from_hdf5(reader, reset_all)
+                    if progress.wasCanceled():
+                        break
+            reader.close()
+        except KeyError:
+            if progress is not None:
+                # KeyError was encoutered when deserializing datasets (DataLab data
+                # model is not compatible with this version)
+                progress.close()
+            self.import_files([filename], import_all, reset_all)
+
+    def __add_object_from_node(self, node: BaseNode) -> None:
+        """Add DataLab object from h5 node"""
+        obj = node.get_native_object()
+        if obj is None:
+            return
+        self.uint32_wng = self.uint32_wng or node.uint32_wng
+        if isinstance(obj, SignalObj):
+            self.mainwindow.signalpanel.add_object(obj)
+        else:
+            self.mainwindow.imagepanel.add_object(obj)
+
+    def __eventually_show_warnings(self) -> None:
+        """Eventually show warnings after everything is imported"""
+        if self.uint32_wng:
+            QW.QMessageBox.warning(
+                self.mainwindow, _("Warning"), _("Clipping uint32 data to int32.")
+            )
+
+    def import_files(
+        self, filenames: list[str], import_all: bool, reset_all: bool
+    ) -> None:
+        """Import HDF5 files"""
+        h5browser = H5BrowserDialog(self.mainwindow)
+        for filename in filenames:
+            with qt_try_loadsave_file(self.mainwindow, filename, "load"):
+                h5browser.open_file(filename)
+        if h5browser.is_empty():
+            h5browser.cleanup()
+            if not execenv.unattended:
+                QW.QMessageBox.warning(
+                    self.mainwindow,
+                    _("Warning"),
+                    _("No supported data available in HDF5 file(s)."),
+                )
+            return
+        if execenv.unattended:
+            # Unattended mode: import all datasets (for testing)
+            import_all = True
+        if import_all or exec_dialog(h5browser):
+            if import_all:
+                nodes = h5browser.get_all_nodes()
+            else:
+                nodes = h5browser.get_nodes()
+            if nodes is not None:
+                if reset_all:
+                    self.mainwindow.reset_all()
+                with qt_try_loadsave_file(self.mainwindow, "*.h5", "load"):
+                    with create_progress_bar(self.mainwindow, "", len(nodes)) as prog:
+                        self.uint32_wng = False
+                        for idx, node in enumerate(nodes):
+                            prog.setLabelText(self.__progbartitle(node.h5file.filename))
+                            prog.setValue(idx + 1)
+                            QW.QApplication.processEvents()
+                            if prog.wasCanceled():
+                                break
+                            self.__add_object_from_node(node)
+                self.__eventually_show_warnings()
+        h5browser.cleanup()
+
+    def import_dataset_from_file(self, filename: str, dsetname: str) -> None:
+        """Import dataset from HDF5 file"""
+        h5importer = H5Importer(filename)
+        try:
+            node = h5importer.get(dsetname)
+            self.uint32_wng = False
+            self.__add_object_from_node(node)
+            self.__eventually_show_warnings()
+        except KeyError as exc:
+            raise KeyError(f"Dataset not found: {dsetname}") from exc
+        h5importer.close()