QTableViewQTableWidget表头添加复选框

QTableView/QTableWidget表头添加复选框

核心代码如下 1、复选列属性定义

//可复选列的属性 struct stCheckbleColumnSettings{ bool canTristate; //是否支持三态 QString text; //显示的文本 Qt::Alignment al; //对齐方式 stCheckbleColumnSettings(){ canTristate = false; al = Qt::AlignCenter; } stCheckbleColumnSettings(bool canTristate, const QString& text, Qt::Alignment al = Qt::AlignLeft | Qt::AlignVCenter) : canTristate(canTristate), text(text), al(al){} };

2、重写QHeaderView的mouseRelease事件，用来判断点击

void CheckbleHeaderView::mouseReleaseEvent(QMouseEvent *e) { QHeaderView::mouseReleaseEvent(e); if(e->button() != Qt::LeftButton) return ; int nIdx = logicalIndexAt(e->pos()); Q_D(CheckbleHeaderView); if(!d->m_mapColumn2Data.contains(nIdx)) { return ; } auto& columnData = d->m_mapColumn2Data[nIdx]; //判断是否在checkbox的点击范围内 if(!d->contains(nIdx, e->pos())) { d->m_nCurrentPressColumn = -1; return; } if(columnData.canTristate) { if(columnData.state == Qt::Unchecked) { columnData.state = Qt::PartiallyChecked; } else { columnData.state = columnData.state == Qt::Checked ? Qt::Unchecked : Qt::Checked; } } else { columnData.state = columnData.state == Qt::Unchecked ? Qt::Checked : Qt::Unchecked; } d->m_nCurrentPressColumn = -1; //更新界面 updateSection(nIdx); emit sigStateChanged(nIdx, columnData.state); }

3、判断是否在checkbox点击范围内

bool contains(int nCol, const QPoint& pos) { if(!m_mapColumn2Data.contains(nCol)) { return false; } Q_Q(CheckbleHeaderView); int nX = q->sectionPosition(nCol); int size = q->sectionSize(nCol); QRect rc(nX, 0, size, q->height()); QCheckBox check; QStyleOptionButton opt = getStyleOption(&check, nCol, m_mapColumn2Data.value(nCol), rc); QRect rcIndicator = q->style()->subElementRect(QStyle::SE_CheckBoxIndicator, &opt, &check); return rcIndicator.contains(pos); }

4、获取绘制的styleoption

QStyleOptionButton getStyleOption(QCheckBox* check, int logicalIndex, const stColumnData& columnData, const QRect& rect)const { const Q_Q(CheckbleHeaderView); QStyleOptionButton opt; opt.initFrom(check); opt.rect = rect; opt.text = columnData.strText; if(m_nCurrentPressColumn == logicalIndex) { opt.state |= QStyle::State_Sunken; } if(columnData.state == Qt::PartiallyChecked) { opt.state |= QStyle::State_NoChange; } else { opt.state |= columnData.state == Qt::Checked ? QStyle::State_On : QStyle::State_Off; } if(q->testAttribute(Qt::WA_Hover) && q->underMouse()) { opt.state |= QStyle::State_MouseOver; } //根据对齐方式重新计算rect reCalcOptRect(check, columnData.al, opt); return opt; }

5、最后是绘制的代码

const Q_D(CheckbleHeaderView); painter->save(); QHeaderView::paintSection(painter, rect, logicalIndex); painter->restore(); //判断该列是否支持复选 if(!d->m_mapColumn2Data.contains(logicalIndex)) { return; } const auto& columnData = d->m_mapColumn2Data.value(logicalIndex); QCheckBox check((QWidget*)this); QStyleOptionButton opt = d->getStyleOption(&check, logicalIndex, columnData, rect); style()->drawControl(QStyle::CE_CheckBox, &opt, painter, &check);

使用方法

QTableWidget w; w.setColumnCount(2); w.setRowCount(2); CheckbleHeaderView view(Qt::Horizontal); view.setCheckbleColumn({{0, {true, "1", Qt::AlignCenter}}}); w.setHorizontalHeader(&view); w.show();

效果图