xarray-graph 0.2.0__tar.gz → 0.2.1__tar.gz

{xarray_graph-0.2.0 → xarray_graph-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xarray-graph
-Version: 0.2.0
+Version: 0.2.1
 Summary: PyQt/PySide UI for graphing (x,y) slices of Xarray datasets.
 Keywords: PyQt,PySide,xarray,graph
 Home-page: https://github.com/marcel-goldschen-ohm/xarray-graph
@@ -20,11 +20,11 @@ Project-URL: Repository, https://github.com/marcel-goldschen-ohm/xarray-graph
 Requires-Python: >=3.9
 Requires-Dist: numpy>=1.26.2
 Requires-Dist: xarray>=2023.12.0
-Requires-Dist: xarray-tree>=0.1.0
 Requires-Dist: qtpy>=2.4.1
 Requires-Dist: qtawesome>=1.3.0
-Requires-Dist: pyqt-ext>=1.1.0
-Requires-Dist: pyqtgraph-ext>=1.0.0
+Requires-Dist: pyqt-ext>=1.2.0
+Requires-Dist: pyqtgraph-ext>=1.2.0
+Requires-Dist: xarray-treeview>=1.2.0
 Requires-Dist: scipy>=1.11.4
 Requires-Dist: lmfit>=1.2.2
 Description-Content-Type: text/markdown
@@ -38,6 +38,11 @@ Should work with PySide6, PyQt6, or PyQt5.
 pip install PySide6 xarray-graph
 ```
 
+# Run
+```shell
+xarray-graph
+```
+
 # TODO
 - i/o
 - tests

{xarray_graph-0.2.0 → xarray_graph-0.2.1}/README.md

@@ -7,6 +7,11 @@ Should work with PySide6, PyQt6, or PyQt5.
 pip install PySide6 xarray-graph
 ```
 
+# Run
+```shell
+xarray-graph
+```
+
 # TODO
 - i/o
 - tests

{xarray_graph-0.2.0 → xarray_graph-0.2.1}/pyproject.toml

@@ -9,11 +9,11 @@ requires-python = ">=3.9"
 dependencies = [
     "numpy>=1.26.2",
     "xarray>=2023.12.0",
-    "xarray-tree>=0.1.0",
     "qtpy>=2.4.1",
     "qtawesome>=1.3.0",
-    "pyqt-ext>=1.1.0",
-    "pyqtgraph-ext>=1.0.0",
+    "pyqt-ext>=1.2.0",
+    "pyqtgraph-ext>=1.2.0",
+    "xarray-treeview>=1.2.0",
     "scipy>=1.11.4",
     "lmfit>=1.2.2",
 ]
@@ -35,7 +35,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
 ]
-version = "0.2.0"
+version = "0.2.1"
 
 [project.license]
 text = "MIT"
@@ -45,7 +45,7 @@ homepage = "https://github.com/marcel-goldschen-ohm/xarray-graph"
 repository = "https://github.com/marcel-goldschen-ohm/xarray-graph"
 
 [project.scripts]
-xrg = "xarray_graph.__main__:main"
+xarray-graph = "xarray_graph.__main__:main"
 
 [build-system]
 requires = [

{xarray_graph-0.2.0 → xarray_graph-0.2.1}/src/xarray_graph/XarrayGraph.py

@@ -1034,6 +1034,17 @@ class XarrayGraph(QMainWindow):
             if i != index:
                 button.setChecked(False)
     
+    def _show_control_panel_at(self, index: int) -> None:
+        actions = self._control_panel_toolbar.actions()
+        widgets = [self._control_panel_toolbar.widgetForAction(action) for action in actions]
+        buttons = [widget for widget in widgets if isinstance(widget, QToolButton)]
+        buttons[index].setChecked(True)
+        self._control_panel.setCurrentIndex(index)
+        self._control_panel.setVisible(True)
+        for i, button in enumerate(buttons):
+            if i != index:
+                button.setChecked(False)
+    
     def _setup_data_control_panel(self) -> None:
         button = QToolButton()
         button.setIcon(qta.icon('ph.eye', options=[{'opacity': 0.5}]))

xarray_graph-0.2.1/src/xarray_graph/__main__.py

@@ -0,0 +1,76 @@
+from xarray_graph.XarrayGraph import XarrayGraph
+from qtpy.QtCore import QTimer
+from qtpy.QtWidgets import QApplication
+
+
+def main():
+    app = QApplication()
+    ui = XarrayGraph()
+    # ui.setWindowTitle(ui.__class__.__name__)
+    ui.setWindowTitle('xarray-graph')
+    ui.show()
+    QTimer.singleShot(100, lambda: ask_for_example(ui))
+    app.exec()
+
+
+def ask_for_example(ui: XarrayGraph):
+    from qtpy.QtWidgets import QMessageBox
+
+    example = QMessageBox.question(ui, 'Example?', 'Load example data?')
+    if example == QMessageBox.StandardButton.Yes:
+        load_example(ui)
+
+
+def load_example(ui: XarrayGraph):
+    import numpy as np
+    import xarray as xr
+    from datatree import DataTree
+
+    n = 100
+    raw_ds = xr.Dataset(
+        data_vars={
+            'current': (['series', 'sweep', 'time'], np.random.rand(3, 10, n) * 1e-9, {'units': 'A'}),
+            'voltage': (['series', 'sweep', 'time'], np.random.rand(3, 10, n) * 10000, {'units': 'V'}),
+        },
+        coords={
+            'series': ('series', np.arange(3)),
+            'sweep': ('sweep', np.arange(10)),
+            'time': ('time', np.arange(n) * 0.01, {'units': 's'}),
+        },
+    )
+
+    baselined_ds = xr.Dataset(
+        data_vars={
+            'current': (['series', 'sweep', 'time'], np.random.rand(3, 10, n) * 1e-9, {'units': 'A'}),
+        },
+        coords={
+            'series': ('series', np.arange(3)),
+            'sweep': ('sweep', np.arange(10)),
+            'time': ('time', np.arange(n) * 0.01, {'units': 's'}),
+        },
+    )
+
+    scaled_ds = xr.Dataset(
+        data_vars={
+            'current': (['series', 'sweep', 'time'], np.random.rand(1, 2, n) * 1e-9, {'units': 'A'}),
+        },
+        coords={
+            'series': ('series', [1]),
+            'sweep': ('sweep', [5,8]),
+            'time': ('time', np.arange(n) * 0.01, {'units': 's'}),
+        },
+    )
+    
+    root_node = DataTree()
+    raw_node = DataTree(name='raw', data=raw_ds, parent=root_node)
+    baselined_node = DataTree(name='baselined', data=baselined_ds, parent=raw_node)
+    scaled_node = DataTree(name='scaled', data=scaled_ds, parent=baselined_node)
+
+    ui.data = root_node
+
+    ui._show_control_panel_at(0)
+    ui._data_treeview.expandAll()
+
+
+if __name__ == '__main__':
+    main()

xarray_graph-0.2.0/src/xarray_graph/__main__.py

@@ -1,14 +0,0 @@
-from xarray_graph.XarrayGraph import XarrayGraph
-from qtpy.QtWidgets import QApplication
-
-
-def main():
-    app = QApplication()
-    ui = XarrayGraph()
-    ui.setWindowTitle(ui.__class__.__name__)
-    ui.show()
-    app.exec()
-
-
-if __name__ == '__main__':
-    main()