class TableDemo(QWidget):
def __init__(self, parent=None):
super(TableDemo, self).__init__(parent)
self.setWindowTitle('TableView Demo')
self.resize(600, 500)
modelVerticalTitleLabelList = ['A', 'B', 'C', 'D', 'E']
modelHorizontalTitleLabelList = ['甲', '乙', '丙', '丁', '戊', '已']
self.model = QStandardItemModel(5, 5)
self.model.setHorizontalHeaderLabels(modelVerticalTitleLabelList)
self.model.setVerticalHeaderLabels(modelHorizontalTitleLabelList)
for i in range(5):
for j in range(5):
item = QStandardItem(str(i + j))
self.model.setItem(i, j, item)
self.tabletView = QTableView()
self.tabletView.setModel(self.model)
self.tabletView.horizontalHeader().setStretchLastSection(True)
mainLayout = QVBoxLayout()
mainLayout.addWidget(self.tabletView)
self.testRowColumn()
self.setLayout(mainLayout)
def testRowColumn(self):
# 用rowAt()和columnAt(),返回一个坐标在table中的行和列的序号
x, y = 100, 200
row_at_x = self.tabletView.rowAt(x)
column_at_y = self.tabletView.columnAt(y)
print('坐标(%d,%d)在表格中是第%d行,第%d列' %
(x, y, row_at_x + 1, column_at_y + 1))
# 如果(x,y)不在表格内,函数返回值是-1
# setRowHeight,setColumnWidth:可以设置行高'列宽
self.tabletView.setRowHeight(1, 50)
self.tabletView.setColumnWidth(1, 150)
# <int> columnWidth():返回指定列的宽度
# <int> rowHeight():返回指定行的高度
print('第(2,2)单元格的列宽,行高为为:%d,%d(像素)' %
(self.tabletView.columnWidth(1), self.tabletView.rowHeight(1)))
# setSpan():设定指定行和列的行跨度和列跨度
self.tabletView.setSpan(2, 2, 2, 2)
# <int> rowSpan():返回指定行的位置的行跨度
# <int> columnSpan:返回指定(row,column)的列跨度
print(
'第(3,3)单元格的的行和列跨度为(%dx%d)' %
(self.tabletView.columnSpan(2, 2), self.tabletView.rowSpan(2, 2)))
# setCornerButtonEnable():设置是否启用左上角的按钮
# 此按钮(用来全选整个表格),默认是启用的
self.tabletView.setCornerButtonEnabled(False) # 此时,左上角的按钮将不再起作用
def contextMenuEvent(self, QContextMenuEvent):
# 设置tableView的右键弹出菜单和菜单信号槽
# 如何只在单元格内弹出右键菜单呢?现在在空白地方点击右键也会弹出菜单????????
menu = QMenu(self)
hideMenu = menu.addAction('&Hide')
hideMenu.triggered.connect(self.hideCurrentColumn)
# 设置显示所有被隐藏的列
showhiddenColumnMenu = menu.addAction('显示隐藏列')
showhiddenColumnMenu.triggered.connect(self.showAllHiddenColumns)
menu.addSeparator()
# 设置当前点击的列按照内容自适应列宽度
resizeColumnToCtnsMenu = menu.addAction('宽度适应')
resizeColumnToCtnsMenu.triggered.connect(
lambda: self.tabletView.resizeColumnToContents(
self.tabletView.currentIndex().column()))
# 排序当前选择的列
orderCurentColumnMenu = menu.addAction('排序')
orderCurentColumnMenu.triggered.connect(self.orderCurrentColum)
menu.exec_(QContextMenuEvent.globalPos())
def hideCurrentColumn(self):
print('第%d列被隐藏了!' % self.tabletView.currentIndex().column())
self.tabletView.setColumnHidden(
self.tabletView.currentIndex().column(), True)
def showAllHiddenColumns(self):
print('显示所有被隐藏的列')
# 遍历所有的列,找到隐藏的列,设置其隐藏为False
for i in range(self.model.columnCount()):
if self.tabletView.isColumnHidden(i):
self.tabletView.setColumnHidden(i, False)
print('列%d已被重新显示' % (i + 1))
def orderCurrentColum(self):
self.tabletView.setSortingEnabled(True)
self.tabletView.sortByColumn(self.tabletView.currentIndex().column(),
Qt.AscendingOrder)
class QtComparePanel(QWidget):
filterChanged = pyqtSignal(str)
def __init__(self, parent=None):
super(QtComparePanel, self).__init__(parent)
self.visibility_flags = []
self.visibility_buttons = []
self.labels_projects = []
self.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.setMinimumWidth(400)
self.setMinimumHeight(100)
self.data_table = QTableView()
self.data_table.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.MinimumExpanding)
self.data_table.setSelectionMode(QAbstractItemView.MultiSelection)
self.data_table.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.data_table.setSelectionBehavior(QAbstractItemView.SelectRows)
self.model = None
self.data = None
self.combodelegate1 = ComboBoxItemDelegate(self.data_table)
self.combodelegate2 = ComboBoxItemDelegate(self.data_table)
lblFilter = QLabel("Filter: ")
self.comboboxFilter = QComboBox()
self.comboboxFilter.setMinimumWidth(80)
self.comboboxFilter.addItem("All")
self.comboboxFilter.addItem("Same")
self.comboboxFilter.addItem("Born")
self.comboboxFilter.addItem("Dead")
self.comboboxFilter.addItem("Grow")
self.comboboxFilter.addItem("Shrink")
filter_layout = QHBoxLayout()
filter_layout.addWidget(lblFilter)
filter_layout.addWidget(self.comboboxFilter)
filter_layout.addStretch()
layout = QVBoxLayout()
layout.addLayout(filter_layout)
layout.addWidget(self.data_table)
self.setLayout(layout)
self.correspondences = None
self.data = None
self.comboboxFilter.currentTextChanged.connect(self.changeFilter)
def setTable(self, project, img1idx, img2idx):
self.correspondences = project.getImagePairCorrespondences(
img1idx, img2idx)
self.data = self.correspondences.data
self.model = TableModel(self.data)
self.sortfilter = QSortFilterProxyModel(self)
self.sortfilter.setSourceModel(self.model)
self.data_table.setModel(self.sortfilter)
self.data_table.setVisible(False)
self.data_table.verticalHeader().hide()
self.data_table.resizeColumnsToContents()
self.data_table.setVisible(True)
self.data_table.setItemDelegateForColumn(5, self.combodelegate1)
self.data_table.setItemDelegateForColumn(6, self.combodelegate2)
self.data_table.setEditTriggers(QAbstractItemView.DoubleClicked)
self.data_table.update()
self.data_table.setStyleSheet(
"QHeaderView::section { background-color: rgb(40,40,40) }")
def updateData(self, corr):
if corr is None:
return
self.correspondences = corr
self.sortfilter.beginResetModel()
self.model.beginResetModel()
self.model._data = corr.data
self.sortfilter.endResetModel()
self.model.endResetModel()
self.data_table.update()
def selectRows(self, rows):
self.data_table.clearSelection()
indexes = [self.model.index(r, 0) for r in rows]
mode = QItemSelectionModel.Select | QItemSelectionModel.Rows
[
self.data_table.selectionModel().select(index, mode)
for index in indexes
]
if len(rows) > 0:
value = self.data_table.horizontalScrollBar().value()
column = self.data_table.columnAt(value)
self.data_table.scrollTo(self.data_table.model().index(
rows[0], column))
@pyqtSlot(QModelIndex)
def getData(self, index):
pass
#column = index.column()
#row = index.row()
#self.data_table.model().index(row, column).data()
@pyqtSlot(str)
def changeFilter(self, txt):
if self.data is None:
return
if txt == 'All':
self.sortfilter.setFilterRegExp(QRegExp())
else:
self.sortfilter.setFilterKeyColumn(5)
self.sortfilter.setFilterRegExp(txt.lower())
self.sortfilter.setFilterRole(Qt.DisplayRole)
self.filterChanged.emit(txt.lower())
