mirror of
https://github.com/m-labs/artiq.git
synced 2024-12-11 12:46:37 +08:00
browser: own package, own datasets dock
This commit is contained in:
parent
1814f12152
commit
1f152bceb2
57
artiq/browser/datasets.py
Normal file
57
artiq/browser/datasets.py
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
from PyQt5 import QtCore, QtWidgets
|
||||||
|
|
||||||
|
from artiq.tools import short_format
|
||||||
|
from artiq.gui.tools import LayoutWidget
|
||||||
|
from artiq.gui.models import DictSyncTreeSepModel
|
||||||
|
|
||||||
|
# reduced read-only version of artiq.gui.datasets
|
||||||
|
|
||||||
|
|
||||||
|
class Model(DictSyncTreeSepModel):
|
||||||
|
def __init__(self, init):
|
||||||
|
DictSyncTreeSepModel.__init__(self, ".", ["Dataset", "Value"], init)
|
||||||
|
|
||||||
|
def convert(self, k, v, column):
|
||||||
|
return short_format(v[1])
|
||||||
|
|
||||||
|
|
||||||
|
class DatasetsDock(QtWidgets.QDockWidget):
|
||||||
|
def __init__(self, datasets_sub):
|
||||||
|
QtWidgets.QDockWidget.__init__(self, "Datasets")
|
||||||
|
self.setObjectName("Datasets")
|
||||||
|
self.setFeatures(QtWidgets.QDockWidget.DockWidgetMovable |
|
||||||
|
QtWidgets.QDockWidget.DockWidgetFloatable)
|
||||||
|
|
||||||
|
grid = LayoutWidget()
|
||||||
|
self.setWidget(grid)
|
||||||
|
|
||||||
|
self.search = QtWidgets.QLineEdit()
|
||||||
|
self.search.setPlaceholderText("search...")
|
||||||
|
self.search.editingFinished.connect(self._search_datasets)
|
||||||
|
grid.addWidget(self.search, 0, 0)
|
||||||
|
|
||||||
|
self.table = QtWidgets.QTreeView()
|
||||||
|
self.table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
|
||||||
|
self.table.setSelectionMode(
|
||||||
|
QtWidgets.QAbstractItemView.SingleSelection)
|
||||||
|
grid.addWidget(self.table, 1, 0)
|
||||||
|
|
||||||
|
self.table_model = Model(dict())
|
||||||
|
datasets_sub.add_setmodel_callback(self.set_model)
|
||||||
|
|
||||||
|
def _search_datasets(self):
|
||||||
|
if hasattr(self, "table_model_filter"):
|
||||||
|
self.table_model_filter.setFilterFixedString(
|
||||||
|
self.search.displayText())
|
||||||
|
|
||||||
|
def set_model(self, model):
|
||||||
|
self.table_model = model
|
||||||
|
self.table_model_filter = QtCore.QSortFilterProxyModel()
|
||||||
|
self.table_model_filter.setSourceModel(self.table_model)
|
||||||
|
self.table.setModel(self.table_model_filter)
|
||||||
|
|
||||||
|
def save_state(self):
|
||||||
|
return bytes(self.table.header().saveState())
|
||||||
|
|
||||||
|
def restore_state(self, state):
|
||||||
|
self.table.header().restoreState(QtCore.QByteArray(state))
|
@ -54,8 +54,9 @@ class ResultsBrowser(QtWidgets.QSplitter):
|
|||||||
try:
|
try:
|
||||||
with h5py.File(path, "r") as f:
|
with h5py.File(path, "r") as f:
|
||||||
rd = {}
|
rd = {}
|
||||||
for k in f: #["datasets"]:
|
group = f["datasets"]
|
||||||
rd[k] = False, f[k].value
|
for k in group:
|
||||||
|
rd[k] = True, group[k].value
|
||||||
self.datasets.init(rd)
|
self.datasets.init(rd)
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
@ -10,11 +10,12 @@ from quamash import QEventLoop
|
|||||||
|
|
||||||
from artiq import __artiq_dir__ as artiq_dir
|
from artiq import __artiq_dir__ as artiq_dir
|
||||||
from artiq.tools import verbosity_args, init_logger, atexit_register_coroutine
|
from artiq.tools import verbosity_args, init_logger, atexit_register_coroutine
|
||||||
from artiq.gui import state, results, datasets, applets, models
|
from artiq.gui import state, applets, models
|
||||||
|
from artiq.browser import datasets, results
|
||||||
|
|
||||||
|
|
||||||
def get_argparser():
|
def get_argparser():
|
||||||
parser = argparse.ArgumentParser(description="ARTIQ results browser")
|
parser = argparse.ArgumentParser(description="ARTIQ Browser")
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--db-file", default="artiq_browser.pyon",
|
"--db-file", default="artiq_browser.pyon",
|
||||||
help="database file for local browser settings")
|
help="database file for local browser settings")
|
||||||
@ -29,7 +30,7 @@ class MainWindow(QtWidgets.QMainWindow):
|
|||||||
|
|
||||||
icon = QtGui.QIcon(os.path.join(artiq_dir, "gui", "logo.svg"))
|
icon = QtGui.QIcon(os.path.join(artiq_dir, "gui", "logo.svg"))
|
||||||
self.setWindowIcon(icon)
|
self.setWindowIcon(icon)
|
||||||
self.setWindowTitle("ARTIQ - Browser")
|
self.setWindowTitle("ARTIQ Browser")
|
||||||
|
|
||||||
qfm = QtGui.QFontMetrics(self.font())
|
qfm = QtGui.QFontMetrics(self.font())
|
||||||
self.resize(140*qfm.averageCharWidth(), 38*qfm.lineSpacing())
|
self.resize(140*qfm.averageCharWidth(), 38*qfm.lineSpacing())
|
||||||
@ -94,8 +95,7 @@ def main():
|
|||||||
atexit_register_coroutine(d_applets.stop)
|
atexit_register_coroutine(d_applets.stop)
|
||||||
smgr.register(d_applets)
|
smgr.register(d_applets)
|
||||||
|
|
||||||
d_datasets = datasets.DatasetsDock(datasets_sub,
|
d_datasets = datasets.DatasetsDock(datasets_sub)
|
||||||
None) # TODO: datsets_ctl.delete()
|
|
||||||
smgr.register(d_datasets)
|
smgr.register(d_datasets)
|
||||||
|
|
||||||
main_window.setCentralWidget(d_results)
|
main_window.setCentralWidget(d_results)
|
||||||
|
Loading…
Reference in New Issue
Block a user