forked from M-Labs/artiq
1
0
Fork 0
artiq/artiq/gui/results.py

81 lines
2.6 KiB
Python
Raw Normal View History

2016-04-04 23:37:51 +08:00
import logging
2016-04-05 15:51:04 +08:00
import h5py
2016-04-04 23:37:51 +08:00
from PyQt5 import QtCore, QtWidgets
from artiq.gui.tools import LayoutWidget
logger = logging.getLogger(__name__)
class ResultsDock(QtWidgets.QDockWidget):
2016-04-05 15:51:04 +08:00
def __init__(self, datasets):
2016-04-04 23:37:51 +08:00
QtWidgets.QDockWidget.__init__(self, "Results")
2016-04-05 15:51:04 +08:00
self.datasets = datasets
2016-04-04 23:37:51 +08:00
self.setObjectName("Results")
self.setFeatures(QtWidgets.QDockWidget.DockWidgetMovable |
QtWidgets.QDockWidget.DockWidgetFloatable)
top_widget = LayoutWidget()
self.setWidget(top_widget)
self.stack = QtWidgets.QStackedWidget()
top_widget.addWidget(self.stack, 1, 0, colspan=2)
self.rt_buttons = LayoutWidget()
self.rt_buttons.layout.setContentsMargins(0, 0, 0, 0)
self.stack.addWidget(self.rt_buttons)
self.rt_model = QtWidgets.QFileSystemModel()
self.rt_model.setRootPath(QtCore.QDir.currentPath())
2016-04-05 15:51:04 +08:00
self.rt_model.setNameFilters(["*.h5"])
2016-04-04 23:37:51 +08:00
self.rt_model.setNameFilterDisables(False)
self.rt = QtWidgets.QTreeView()
self.rt.setModel(self.rt_model)
self.rt.setRootIndex(self.rt_model.index(QtCore.QDir.currentPath()))
2016-04-05 15:51:04 +08:00
self.rt.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
self.rt.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection)
2016-04-04 23:37:51 +08:00
self.rt.selectionModel().selectionChanged.connect(
self.selection_changed)
self.rt_buttons.addWidget(self.rt, 0, 0, colspan=2)
def selection_changed(self, selected, deselected):
indexes = selected.indexes()
2016-04-05 15:51:04 +08:00
if not indexes:
return
path = self.rt_model.filePath(indexes[0])
logger.info("opening %s", path)
try:
with h5py.File(path, "r") as f:
rd = {}
for k in f["datasets"]:
rd[k] = False, f[k].value
self.datasets.init(rd)
except:
pass
2016-04-04 23:37:51 +08:00
def select(self, path):
s = self.rt_model.index(path)
self.rt.selectionModel().setCurrentIndex(
s,
QtCore.QItemSelectionModel.ClearAndSelect)
self.rt.scrollTo(s) # TODO: call_soon?
def save_state(self):
return {
"selected": self.rt_model.filePath(
self.rt.selectionModel().currentIndex()),
"header": bytes(self.rt.header().saveState()),
}
def restore_state(self, state):
selected = state.get("selected")
if selected:
self.select(selected)
header = state.get("header")
if header:
self.rt.header().restoreState(QtCore.QByteArray(header))