forked from M-Labs/artiq
1
0
Fork 0

browser: expid from h5

This commit is contained in:
Robert Jördens 2016-04-18 17:10:40 +02:00
parent 686f670a3c
commit ba24bbc8ed
1 changed files with 29 additions and 14 deletions

View File

@ -4,9 +4,24 @@ import os
import h5py import h5py
from PyQt5 import QtCore, QtWidgets, QtGui from PyQt5 import QtCore, QtWidgets, QtGui
from artiq.protocols import pyon
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def open_h5(info):
if not (info.isFile() and info.isReadable() and
info.suffix() == "h5"):
return
try:
f = h5py.File(info.filePath(), "r")
except:
logger.warning("unable to read HDF5 file %s", info.filePath(),
exc_info=True)
return
return f
class ThumbnailIconProvider(QtWidgets.QFileIconProvider): class ThumbnailIconProvider(QtWidgets.QFileIconProvider):
def icon(self, info): def icon(self, info):
icon = self.hdf5_thumbnail(info) icon = self.hdf5_thumbnail(info)
@ -15,12 +30,8 @@ class ThumbnailIconProvider(QtWidgets.QFileIconProvider):
return icon return icon
def hdf5_thumbnail(self, info): def hdf5_thumbnail(self, info):
if not (info.isFile() and info.isReadable() and f = open_h5(info)
info.suffix() == "h5"): if not f:
return
try:
f = h5py.File(info.filePath(), "r")
except:
return return
with f: with f:
try: try:
@ -92,6 +103,7 @@ class FilesDock(QtWidgets.QDockWidget):
self.rl.setModel(self.model) self.rl.setModel(self.model)
self.rl.selectionModel().currentChanged.connect( self.rl.selectionModel().currentChanged.connect(
self.list_current_changed) self.list_current_changed)
self.rl.doubleClicked.connect(self.open_experiment)
self.splitter.addWidget(self.rl) self.splitter.addWidget(self.rl)
def tree_current_changed(self, current, previous): def tree_current_changed(self, current, previous):
@ -101,14 +113,8 @@ class FilesDock(QtWidgets.QDockWidget):
def list_current_changed(self, current, previous): def list_current_changed(self, current, previous):
info = self.model.fileInfo(current) info = self.model.fileInfo(current)
logger.info("opening %s", info.filePath()) logger.info("opening %s", info.filePath())
if not (info.isFile() and info.isReadable() and f = open_h5(info)
info.suffix() == "h5"): if not f:
return
try:
f = h5py.File(info.filePath(), "r")
except:
logger.warning("unable to read HDF5 file %s", info.filePath(),
exc_info=True)
return return
with f: with f:
rd = {} rd = {}
@ -120,6 +126,15 @@ class FilesDock(QtWidgets.QDockWidget):
rd[k] = True, group[k].value rd[k] = True, group[k].value
self.datasets.init(rd) self.datasets.init(rd)
def open_experiment(self, current):
info = self.model.fileInfo(current)
logger.info("loading experiment for %s", info.filePath())
f = open_h5(info)
if not f:
return
with f:
expid = pyon.decode(f["expid"].value)
def select_dir(self, path): def select_dir(self, path):
if not os.path.exists(path): if not os.path.exists(path):
return return