M-Labs
/
artiq
mirror of https://github.com/m-labs/artiq.git
2
0
Fork 1
Code Issues Projects Releases Wiki Activity

gui/explorer: tree view

This commit is contained in:
Sebastien Bourdeauducq 2015-11-17 19:46:26 +08:00
parent 9c5db288a5
commit 250ab6b489
1 changed files with 14 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
artiq/gui/explorer.py
View File

@ -6,26 +6,21 @@ from pyqtgraph import dockarea
from pyqtgraph import LayoutWidget from pyqtgraph import LayoutWidget
from artiq.protocols import pyon from artiq.protocols import pyon
from artiq.gui.models import DictSyncModel from artiq.gui.models import DictSyncTreeSepModel
from artiq.gui.scan import ScanController from artiq.gui.scan import ScanController
from artiq.gui.shortcuts import ShortcutManager from artiq.gui.shortcuts import ShortcutManager
class Model(DictSyncModel): class Model(DictSyncTreeSepModel):
def __init__(self, init): def __init__(self, init):
DictSyncModel.__init__(self, self.explorer = None
DictSyncTreeSepModel.__init__(self,
"/",
["Experiment"], ["Experiment"],
init) init)
self.explorer = None
def sort_key(self, k, v):
return k
def convert(self, k, v, column):
return k
def __setitem__(self, k, v): def __setitem__(self, k, v):
DictSyncModel.__setitem__(self, k, v) DictSyncTreeSepModel.__setitem__(self, k, v)
if self.explorer is not None: if self.explorer is not None:
if k == self.explorer.selected_key: if k == self.explorer.selected_key:
self.explorer.update_selection(k, k) self.explorer.update_selection(k, k)
@ -232,7 +227,8 @@ class ExplorerDock(dockarea.Dock):
grid = LayoutWidget() grid = LayoutWidget()
self.splitter.addWidget(grid) self.splitter.addWidget(grid)
self.el = QtGui.QListView() self.el = QtGui.QTreeView()
self.el.setHeaderHidden(True)
self.el.selectionChanged = self._selection_changed self.el.selectionChanged = self._selection_changed
self.selected_key = None self.selected_key = None
grid.addWidget(self.el, 0, 0, colspan=4) grid.addWidget(self.el, 0, 0, colspan=4)
@ -310,10 +306,10 @@ class ExplorerDock(dockarea.Dock):
self.el.setModel(model) self.el.setModel(model)
def update_selection(self, selected, deselected): def update_selection(self, selected, deselected):
if deselected: if deselected is not None:
self.argeditor_states[deselected] = self.argeditor.save_state() self.argeditor_states[deselected] = self.argeditor.save_state()
if selected: if selected is not None:
expinfo = self.explist_model.backing_store[selected] expinfo = self.explist_model.backing_store[selected]
self.argeditor.set_arguments(expinfo["arguments"]) self.argeditor.set_arguments(expinfo["arguments"])
if selected in self.argeditor_states: if selected in self.argeditor_states:
@ -324,8 +320,7 @@ class ExplorerDock(dockarea.Dock):
def _sel_to_key(self, selection): def _sel_to_key(self, selection):
selection = selection.indexes() selection = selection.indexes()
if selection: if selection:
row = selection[0].row() return self.explist_model.index_to_key(selection[0])
return self.explist_model.row_to_key[row]
else: else:
return None return None
@ -336,9 +331,9 @@ class ExplorerDock(dockarea.Dock):
def save_state(self): def save_state(self):
idx = self.el.selectedIndexes() idx = self.el.selectedIndexes()
if idx: if idx:
row = idx[0].row() key = self.explist_model.index_to_key(idx[0])
key = self.explist_model.row_to_key[row] if key is not None:
self.argeditor_states[key] = self.argeditor.save_state() self.argeditor_states[key] = self.argeditor.save_state()
return { return {
"argeditor": self.argeditor_states, "argeditor": self.argeditor_states,
"shortcuts": self.shortcuts.save_state() "shortcuts": self.shortcuts.save_state()