class ChannelPropertiesDialog(QDialog):
def __init__(self, parent, info, title="Channel Properties"):
super().__init__(parent)
self.setWindowTitle(title)
self.model = QStandardItemModel(info["nchan"], 4)
self.model.setHorizontalHeaderLabels(["#", "Label", "Type", "Bad"])
for index, ch in enumerate(info["chs"]):
item = QStandardItem()
item.setData(index, Qt.DisplayRole)
item.setFlags(item.flags() & ~Qt.ItemIsEditable)
self.model.setItem(index, 0, item)
self.model.setItem(index, 1, QStandardItem(ch["ch_name"]))
kind = channel_type(info, index).upper()
self.model.setItem(index, 2, QStandardItem(str(kind)))
bad = QStandardItem()
bad.setData(ch["ch_name"] in info["bads"], Qt.UserRole)
bad.setCheckable(True)
bad.setEditable(False)
checked = ch["ch_name"] in info["bads"]
bad.setCheckState(Qt.Checked if checked else Qt.Unchecked)
self.model.setItem(index, 3, bad)
self.model.itemChanged.connect(bad_changed)
self.proxymodel = MySortFilterProxyModel()
self.proxymodel.setDynamicSortFilter(False)
self.proxymodel.setSourceModel(self.model)
self.view = QTableView()
self.view.setModel(self.proxymodel)
self.view.setItemDelegateForColumn(2, ComboBoxDelegate(self.view))
self.view.setEditTriggers(QAbstractItemView.AllEditTriggers)
self.view.verticalHeader().setVisible(False)
self.view.horizontalHeader().setStretchLastSection(True)
self.view.setShowGrid(False)
self.view.setSelectionMode(QAbstractItemView.NoSelection)
self.view.setSortingEnabled(True)
self.view.sortByColumn(0, Qt.AscendingOrder)
vbox = QVBoxLayout(self)
vbox.addWidget(self.view)
self.buttonbox = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
vbox.addWidget(self.buttonbox)
self.buttonbox.accepted.connect(self.accept)
self.buttonbox.rejected.connect(self.reject)
self.resize(475, 650)
self.view.setColumnWidth(0, 70)
self.view.setColumnWidth(1, 155)
self.view.setColumnWidth(2, 90)
class XDFStreamsDialog(QDialog):
def __init__(self, parent, rows, selected=None, disabled=None):
super().__init__(parent)
self.setWindowTitle("Select XDF Stream")
self.model = QStandardItemModel()
self.model.setHorizontalHeaderLabels(
["ID", "Name", "Type", "Channels", "Format", "Sampling Rate"])
for index, stream in enumerate(rows):
items = []
for item in stream:
tmp = QStandardItem()
tmp.setData(item, Qt.DisplayRole)
items.append(tmp)
for item in items:
item.setEditable(False)
if disabled is not None and index in disabled:
item.setFlags(Qt.NoItemFlags)
self.model.appendRow(items)
self.view = QTableView()
self.view.setModel(self.model)
self.view.verticalHeader().setVisible(False)
self.view.horizontalHeader().setStretchLastSection(True)
self.view.setShowGrid(False)
self.view.setSelectionMode(QAbstractItemView.SingleSelection)
self.view.setSelectionBehavior(QAbstractItemView.SelectRows)
if selected is not None:
self.view.selectRow(selected)
self.view.setSortingEnabled(True)
self.view.sortByColumn(0, Qt.AscendingOrder)
vbox = QVBoxLayout(self)
vbox.addWidget(self.view)
hbox = QHBoxLayout()
self._effective_srate = QCheckBox("Use effective sampling rate")
self._effective_srate.setChecked(True)
hbox.addWidget(self._effective_srate)
self._prefix_markers = QCheckBox("Prefix markers with stream ID")
self._prefix_markers.setChecked(False)
if not disabled:
self._prefix_markers.setEnabled(False)
hbox.addWidget(self._prefix_markers)
vbox.addLayout(hbox)
self.buttonbox = QDialogButtonBox(QDialogButtonBox.Ok
| QDialogButtonBox.Cancel)
vbox.addWidget(self.buttonbox)
self.buttonbox.accepted.connect(self.accept)
self.buttonbox.rejected.connect(self.reject)
self.resize(775, 650)
self.view.setColumnWidth(0, 90)
self.view.setColumnWidth(1, 200)
self.view.setColumnWidth(2, 140)
@property
def effective_srate(self):
return self._effective_srate.isChecked()
@property
def prefix_markers(self):
return self._prefix_markers.isChecked()
class XDFChunksDialog(QDialog):
def __init__(self, parent, chunks, fname):
super().__init__(parent)
self.setWindowTitle(f"XDF Chunks ({fname})")
self.chunks = chunks
TAGS = {
1: "FileHeader",
2: "StreamHeader",
3: "Samples",
4: "ClockOffset",
5: "Boundary",
6: "StreamFooter"
}
self.model = QStandardItemModel()
self.model.setHorizontalHeaderLabels(
["#", "Bytes", "Tag", "Stream ID"])
for i, chunk in enumerate(chunks, start=1):
row = []
row.append(_add_item(i))
row.append(_add_item(chunk["nbytes"]))
row.append(_add_item(f"{chunk['tag']} ({TAGS[chunk['tag']]})"))
row.append(_add_item(chunk.get("stream_id", "")))
self.model.appendRow(row)
self.view = QTableView()
self.view.setModel(self.model)
self.view.verticalHeader().setVisible(False)
self.view.horizontalHeader().setStretchLastSection(True)
self.view.setShowGrid(False)
self.view.setSelectionMode(QAbstractItemView.SingleSelection)
self.view.setSelectionBehavior(QAbstractItemView.SelectRows)
self.view.setSortingEnabled(True)
self.view.sortByColumn(0, Qt.AscendingOrder)
self.view.setEditTriggers(QTableView.NoEditTriggers)
self.view.setFixedWidth(450)
self.details = QPlainTextEdit("")
self.details.setFixedWidth(450)
self.details.setReadOnly(True)
self.details.setTabStopDistance(30)
font = QFont()
font.setFamily("monospace")
font.setStyleHint(QFont.Monospace)
self.details.setFont(font)
self.details.setLineWrapMode(QPlainTextEdit.NoWrap)
hbox = QHBoxLayout()
hbox.addWidget(self.view)
hbox.addWidget(self.details)
vbox = QVBoxLayout(self)
vbox.addLayout(hbox)
self.buttonbox = QDialogButtonBox(QDialogButtonBox.Ok)
vbox.addWidget(self.buttonbox)
self.buttonbox.accepted.connect(self.accept)
self.view.clicked.connect(self._update_details)
self.setFixedSize(980, 650)
self.view.setColumnWidth(0, 70)
self.view.setColumnWidth(1, 80)
self.view.setColumnWidth(2, 150)
self.view.setColumnWidth(3, 70)
@Slot()
def _update_details(self):
selection = self.view.selectionModel()
if selection.hasSelection():
n = int(selection.selectedIndexes()[0].data())
self.details.setPlainText(self.chunks[n - 1].get("content", ""))
