def setModel(self, model):
self.model = AddCheckboxToModel(model)
self.proxyModel = SortForeignFilterModel(self)
self.proxyModel.setSourceModel(self.model)
CompactTreeView.setModel(self, self.proxyModel)
self.selectedSpectra = SortForeignFilterModel(self)
self.selectedSpectra.setSourceModel(DB.Spectrum)
self.selectedSpectra.setForeignRelation(3, self.proxyModel, 1)
self.selectedSpectra.rowsAboutToBeRemoved.connect(self.removeSpectra)
self.selectedSpectra.rowsInserted.connect(self.addSpectra)
self.selectedSpectra.filterForForeignKeysList(self.checked)
self.selectedSpectra.sort(0)
self.model.rowsAboutToBeRemoved.connect(self.disableDataSet)
class SpectraSelector(QtGui.QWidget):
spectrumSelected = QtCore.pyqtSignal(int, bool)
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
hLayout = QtGui.QHBoxLayout(self)
splitter = QtGui.QSplitter(self)
splitter.setOrientation(QtCore.Qt.Horizontal)
splitter.setOpaqueResize(True)
splitter.setChildrenCollapsible(False)
hLayout.addWidget(splitter)
self.lView = CompactTreeView(splitter)
self.rView = CompactTreeView(splitter)
splitter.setStretchFactor(0,0)
splitter.setStretchFactor(1,1)
self.setModels(DB.DataSet, DB.Spectrum)
self.rProxyModel.setForeignRelation(4, self.lProxyModel, 1)
self.rProxyModel.filterForForeignKeysSelection()
self.lView.hideColumn(1)
self.lView.hideColumn(3)
self.lView.sortByColumn(3, QtCore.Qt.DescendingOrder)
self.lView.adjustColumnWidths(0, self.lProxyModel.columnCount())
self.rView.hideColumn(2)
self.rView.hideColumn(4)
self.rView.hideColumn(5)
self.rView.sortByColumn(1, QtCore.Qt.DescendingOrder)
self.rView.adjustColumnWidths(0, self.lProxyModel.columnCount())
def setModels(self, lModel, rModel):
self.lModel = AddCheckboxToModel(lModel)
self.lProxyModel = SortForeignFilterModel(self)
self.lProxyModel.setSourceModel(self.lModel)
self.lView.setModel(self.lProxyModel)
self.rModel = AddCheckboxToModel(rModel)
self.rProxyModel = SortForeignFilterModel(self)
self.rProxyModel.setSourceModel(self.rModel)
self.rView.setModel(self.rProxyModel)
self.lView.selectionModel().selectionChanged.connect(self.rProxyModel.filterForForeignKeysSelection)
self.rModel.dataChanged.connect(self.updateLeftCheckbox)
self.rModel.dataChanged.connect(self.updateSelectedSpectra)
self.lModel.dataChanged.connect(self.updateRightCheckboxes)
self.lModel.rowsAboutToBeRemoved.connect(self.disableDataSet)
@QtCore.pyqtSlot(QtCore.QModelIndex, int, int)
def disableDataSet(self, parent, startRow, endRow):
#self.lModel.dataChanged.disconnect(self.updateRightCheckboxes)
self.lModel.item(startRow, 0).setCheckState(QtCore.Qt.Unchecked)
#self.updateRightCheckboxes()
#self.lModel.dataChanged.connect(self.updateRightCheckboxes)
@QtCore.pyqtSlot(QtCore.QModelIndex, QtCore.QModelIndex)
def updateLeftCheckbox(self, currentPointer, previousPointer):
rModel_curentForeignKey = self.rModel.item(currentPointer.row(), self.rProxyModel.foreignKeyColumn).text()
lModel_currentCheckboxItem = self.lModel.item(int(rModel_curentForeignKey), 0)
self.lModel.dataChanged.disconnect(self.updateRightCheckboxes)
checkedCounter = 0
uncheckedCounter = 0
for row in range(self.rModel.rowCount()):
row_foreignKey = self.rModel.item(row, self.rProxyModel.foreignKeyColumn).text()
if row_foreignKey == rModel_curentForeignKey:
row_firstItem = self.rModel.item(row, 0)
if row_firstItem.checkState() == QtCore.Qt.Checked:
checkedCounter += 1
else:
uncheckedCounter += 1
if checkedCounter > 0 and uncheckedCounter > 0:
lModel_currentCheckboxItem.setCheckState(QtCore.Qt.PartiallyChecked)
self.lModel.dataChanged.connect(self.updateRightCheckboxes)
return
if checkedCounter == 0:
lModel_currentCheckboxItem.setCheckState(QtCore.Qt.Unchecked)
else:
lModel_currentCheckboxItem.setCheckState(QtCore.Qt.Checked)
self.lModel.dataChanged.connect(self.updateRightCheckboxes)
@QtCore.pyqtSlot(QtCore.QModelIndex, QtCore.QModelIndex)
def updateSelectedSpectra(self, currentPointer, previousPointer):
rModel_item = self.rModel.itemFromIndex(currentPointer)
if rModel_item.checkState() == QtCore.Qt.Checked:
self.spectrumSelected.emit(currentPointer.row(), True)
elif rModel_item.checkState() == QtCore.Qt.Unchecked:
self.spectrumSelected.emit(currentPointer.row(), False)
@QtCore.pyqtSlot(QtCore.QModelIndex, QtCore.QModelIndex)
def updateRightCheckboxes(self, currentPointer, previousPointer):
if currentPointer.column() == 0:
lModel_checkItem = self.lModel.itemFromIndex(currentPointer)
new_checkState = lModel_checkItem.checkState()
for row in range(self.rModel.rowCount()):
rModel_foreignKey = self.rModel.item(row, self.rProxyModel.foreignKeyColumn).text()
if int(rModel_foreignKey) == lModel_checkItem.row():
rModel_checkItem = self.rModel.item(row, 0)
rModel_checkItem.setCheckState(new_checkState)