xarray-graph 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

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/__main__.py

@@ -1,14 +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(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.dist-info → xarray_graph-0.2.1.dist-info}/METADATA

@@ -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.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+xarray_graph-0.2.1.dist-info/METADATA,sha256=C5TY1bM_h-A7wYuh3bC-vjrL2XZSW1jeyZCIGqn6PTw,1488
+xarray_graph-0.2.1.dist-info/WHEEL,sha256=vnE8JVcI2Wz7GRKorsPArnBdnW2SWKWGow5gu5tHlRU,90
+xarray_graph-0.2.1.dist-info/entry_points.txt,sha256=aW9CKuCZREtN_3Mtg4dKrysPMmHc-tQdUCNmzQYE_vA,61
+xarray_graph-0.2.1.dist-info/licenses/LICENSE,sha256=OE7VkhwkLeFeY8LXi86O-nMXC9B3WsBSILLfPmYGd-8,1077
+xarray_graph/XarrayGraph.py,sha256=KyiQ2oOgvffe1DpQw5fIUVhAtLP0gj2s99eVGVBg6FU,101127
+xarray_graph/__init__.py,sha256=1aAM-Fcno60-ou3AnSqt5HKm6UmZrtULGzg2LF34IJE,48
+xarray_graph/__main__.py,sha256=8c74k8lpaqOuMeKlcBahOKfgo08O7xmJs_htvKKKpck,2274
+xarray_graph/tmp/RegionsManager.py,sha256=0nnrO1e1mqvrs88doO51SJ3JvMNW5wXBKOp2nLQcUFs,1841
+xarray_graph-0.2.1.dist-info/RECORD,,

xarray_graph-0.2.1.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+xarray-graph = xarray_graph.__main__:main
+

xarray_graph-0.2.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-xarray_graph-0.2.0.dist-info/METADATA,sha256=PlV0f9GGnKYWLqFqs13wQBL1mZsBHs93TztzG4tRhPo,1451
-xarray_graph-0.2.0.dist-info/WHEEL,sha256=vnE8JVcI2Wz7GRKorsPArnBdnW2SWKWGow5gu5tHlRU,90
-xarray_graph-0.2.0.dist-info/entry_points.txt,sha256=8qLdEjfVtm86FytoAItpsjvgDEUDj2e7Z2ekiTkUsQs,52
-xarray_graph-0.2.0.dist-info/licenses/LICENSE,sha256=OE7VkhwkLeFeY8LXi86O-nMXC9B3WsBSILLfPmYGd-8,1077
-xarray_graph/RegionsManager.py,sha256=0nnrO1e1mqvrs88doO51SJ3JvMNW5wXBKOp2nLQcUFs,1841
-xarray_graph/XarrayGraph.py,sha256=7hAiKNx0UcLT9rQfGX-vvfeVz0FcjCMVw5o7PL2fRvU,100581
-xarray_graph/__init__.py,sha256=1aAM-Fcno60-ou3AnSqt5HKm6UmZrtULGzg2LF34IJE,48
-xarray_graph/__main__.py,sha256=TMeudufLyynSMYS3WuLpHxUjr0ijfb_-a3vnZgEomgQ,265
-xarray_graph-0.2.0.dist-info/RECORD,,

xarray_graph-0.2.0.dist-info/entry_points.txt

@@ -1,3 +0,0 @@
-[console_scripts]
-xrg = xarray_graph.__main__:main
-