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))
|
