srenblad
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

browser: link list and tree

This commit is contained in:
Robert Jördens 2016-04-06 18:16:40 +08:00
parent d84bcfddbe
commit 9685771275
1 changed files with 86 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

137
artiq/browser/results.py
View File

@ -6,27 +6,50 @@ from PyQt5 import QtCore, QtWidgets, QtGui
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class ResultError(Exception):
pass
class ResultIconProvider(QtWidgets.QFileIconProvider): class ResultIconProvider(QtWidgets.QFileIconProvider):
def icon(self, info): def icon(self, info):
try: icon = self.hdf5_thumbnail(info)
if not (info.isFile() and info.isReadable() and if icon is None:
info.suffix() == "h5"): icon = QtWidgets.QFileIconProvider.icon(self, info)
raise ResultError return icon
with h5py.File(info.filePath(), "r") as f:
if "thumbnail" not in f: def hdf5_thumbnail(self, info):
raise ResultError if not (info.isFile() and info.isReadable() and
d = f["thumbnail"] info.suffix() == "h5"):
if "suffix" not in d.attrs: return
raise ResultError with h5py.File(info.filePath(), "r") as f:
img = QtGui.QImage.fromData(d.value, d.attrs["suffix"]) if "thumbnail" not in f:
pix = QtGui.QPixmap.fromImage(img) return
return QtGui.QIcon(pix) img = QtGui.QImage.fromData(f["thumbnail"].value)
except ResultError: pix = QtGui.QPixmap.fromImage(img)
return QtWidgets.QFileIconProvider.icon(self, info) return QtGui.QIcon(pix)
class DirsOnly(QtCore.QSortFilterProxyModel):
def filterAcceptsRow(self, row, parent):
m = self.sourceModel()
if not m.isDir(m.index(row, 0, parent)):
return False
return QtCore.QSortFilterProxyModel.filterAcceptsRow(self, row, parent)
class FilesOnly(QtCore.QSortFilterProxyModel):
_root_idx = None
def filterAcceptsRow(self, row, parent):
if self._root_idx is not None:
model = self.sourceModel()
idx = model.index(row, 0, parent)
if idx == self._root_idx:
return True
if model.isDir(idx):
print("false", model.filePath(idx),
model.filePath(self._root_idx))
return False
return QtCore.QSortFilterProxyModel.filterAcceptsRow(self, row, parent)
def setRootIndex(self, idx):
self._root_idx = idx
class ResultsBrowser(QtWidgets.QSplitter): class ResultsBrowser(QtWidgets.QSplitter):
@ -35,31 +58,36 @@ class ResultsBrowser(QtWidgets.QSplitter):
self.datasets = datasets self.datasets = datasets
self.rt_model = QtWidgets.QFileSystemModel() self.model = QtWidgets.QFileSystemModel()
self.rt_model.setRootPath(QtCore.QDir.currentPath()) self.model.setRootPath(QtCore.QDir.currentPath())
self.rt_model.setNameFilters(["*.h5"]) self.model.setNameFilters(["*.h5"])
self.rt_model.setNameFilterDisables(False) self.model.setNameFilterDisables(False)
self.rt_model.setIconProvider(ResultIconProvider()) self.model.setIconProvider(ResultIconProvider())
self.rt = QtWidgets.QTreeView() self.rt = QtWidgets.QTreeView()
self.rt_model = DirsOnly()
self.rt_model.setSourceModel(self.model)
self.rt.setModel(self.rt_model) self.rt.setModel(self.rt_model)
self.rt.setRootIndex(self.rt_model.index(QtCore.QDir.currentPath())) self.rt.setRootIndex(self.rt_model.mapFromSource(self.model.index(
QtCore.QDir.currentPath())))
self.rt.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows) self.rt.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
self.rt.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection) self.rt.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection)
self.rt.selectionModel().currentChanged.connect( self.rt.selectionModel().currentChanged.connect(
self.current_changed) self.tree_current_changed)
self.rt.setRootIsDecorated(False) self.rt.setRootIsDecorated(False)
self.addWidget(self.rt) self.addWidget(self.rt)
self.rl = QtWidgets.QListView() self.rl = QtWidgets.QListView()
self.rl_model = FilesOnly()
self.rl_model.setSourceModel(self.model)
self.rl.setModel(self.rl_model)
self.rl.setViewMode(QtWidgets.QListView.IconMode) self.rl.setViewMode(QtWidgets.QListView.IconMode)
self.rl.setModel(self.rt_model)
self.rl.setSelectionModel(self.rt.selectionModel())
self.rl.setRootIndex(self.rt.rootIndex())
l = QtGui.QFontMetrics(self.font()).lineSpacing() l = QtGui.QFontMetrics(self.font()).lineSpacing()
self.rl.setIconSize(QtCore.QSize(20*l, 20*l)) self.rl.setIconSize(QtCore.QSize(20*l, 15*l))
self.rl.setWrapping(True)
self.rl.setFlow(QtWidgets.QListView.LeftToRight) self.rl.setFlow(QtWidgets.QListView.LeftToRight)
self.rl.setWrapping(True)
self.rl.selectionModel().currentChanged.connect(
self.list_current_changed)
self.addWidget(self.rl) self.addWidget(self.rl)
def showEvent(self, ev): def showEvent(self, ev):
@ -71,33 +99,40 @@ class ResultsBrowser(QtWidgets.QSplitter):
self.rt.hideColumn(3) self.rt.hideColumn(3)
self.rt.scrollTo(self.rt.selectionModel().currentIndex()) self.rt.scrollTo(self.rt.selectionModel().currentIndex())
def current_changed(self, current, previous): def tree_current_changed(self, current, previous):
info = self.rt_model.fileInfo(current) i = self.rt_model.mapToSource(current)
self.rl_model.setRootIndex(i)
j = self.rl_model.mapFromSource(i)
print("root", self.model.filePath(i),
i.isValid())
self.rl.setRootIndex(j)
def list_current_changed(self, current, previous):
info = self.model.fileInfo(self.rl_model.mapToSource(current))
logger.info("opening %s", info.filePath()) logger.info("opening %s", info.filePath())
try: if not (info.isFile() and info.isReadable() and
if not (info.isFile() and info.isReadable() and info.suffix() == "h5"):
info.suffix() == "h5"): return
raise ResultError with h5py.File(info.filePath(), "r") as f:
with h5py.File(info.filePath(), "r") as f: rd = {}
rd = {} if "datasets" not in f:
if "datasets" not in f: return
raise ResultError group = f["datasets"]
group = f["datasets"] for k in group:
for k in group: rd[k] = True, group[k].value
rd[k] = True, group[k].value self.datasets.init(rd)
self.datasets.init(rd)
except ResultError:
pass
def select(self, path): def select(self, path):
idx = self.rt_model.mapFromSource(self.model.index(path))
self.rt.scrollTo(idx)
self.rt.expand(idx)
self.rt.selectionModel().setCurrentIndex( self.rt.selectionModel().setCurrentIndex(
self.rt_model.index(path), idx, QtCore.QItemSelectionModel.ClearAndSelect)
QtCore.QItemSelectionModel.ClearAndSelect)
def save_state(self): def save_state(self):
return { return {
"selected": self.rt_model.filePath( "selected": self.model.filePath(self.rt_model.mapToSource(
self.rt.selectionModel().currentIndex()), self.rt.selectionModel().currentIndex())),
"header": bytes(self.rt.header().saveState()), "header": bytes(self.rt.header().saveState()),
"splitter": bytes(self.saveState()), "splitter": bytes(self.saveState()),
} }