1.自定义类继承QAbstractTableModel
头文件如下：
#ifndef TESTMOUDLE_H
#define TESTMOUDLE_H
#include "StructTest.h"
#include <QAbstractTableModel>
class TestMoudle : public QAbstractTableModel
{
public:
TestMoudle(QStringList& list, QObject *parent = nullptr);
~TestMoudle();
void AddData(StructTest test);
void DelData(int index );
public:
virtual Qt::ItemFlags flags(const QModelIndex &index) const override;
virtual int rowCount(const QModelIndex &parent = QModelIndex()) const override;
virtual int columnCount(const QModelIndex &parent = QModelIndex()) const override;
virtual bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
virtual QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
public:
QList<StructTest> m_list;
QStringList m_headList;
};
#endif // TESTMOUDLE_H
源文件
#include "TestMoudle.h"
TestMoudle::TestMoudle(QStringList& list, QObject *parent )
: QAbstractTableModel(parent), m_headList(list)
{
}
TestMoudle::~TestMoudle()
{
}
Qt::ItemFlags TestMoudle::flags(const QModelIndex &index) const
{
Qt::ItemFlags flags = QAbstractTableModel::flags(index);
return flags|Qt::ItemIsEditable; //需要双击编辑 添加 Qt::ItemIsEditable
}
int TestMoudle::rowCount(const QModelIndex &parent) const
{
Q_UNUSED(parent)
return m_list.size();
}
int TestMoudle::columnCount(const QModelIndex &parent ) const
{
Q_UNUSED(parent)
return 2;
}
bool TestMoudle::setData(const QModelIndex &index, const QVariant &value, int role )
{
if (!index.isValid())
return false;
if(role == Qt::EditRole)
{
int iRow = index.row(), iCol = index.column();
if( iRow >= 0 && iRow < m_list.size() )
{
if(iCol == 0)
{
m_list[iRow].ID = value.toString();
}
else if(iCol == 1)
{
m_list[iRow].name = value.toString();
}
}
}
return true;
}
QVariant TestMoudle::data(const QModelIndex &index, int role ) const
{
if (!index.isValid())
{
return QVariant();
}
switch (role)
{
case Qt::DisplayRole:
case Qt::EditRole://不添加 双击编辑 原来的内容为空了
{
int row = index.row();
if(row < 0 || row > m_list.size() )
return QVariant();
if(index.column()==0)
return m_list[row].ID;
else
return m_list[row].name;
}
break;
case Qt::TextAlignmentRole:
{
return QVariant(Qt::AlignCenter);
}
break;
default:
return QVariant();
}
return QVariant();
}
QVariant TestMoudle::headerData(int section, Qt::Orientation orientation, int role ) const
{
if (role==Qt::TextAlignmentRole)
{
return QVariant(Qt::AlignCenter);
if (section==0)
{
return QVariant(Qt::AlignCenter);
}
else
{
return QVariant(Qt::AlignVCenter|Qt::AlignLeft);
}
}
else if (role==Qt::DisplayRole)
{
if( orientation==Qt::Horizontal )
{
if( section>=0 && section < m_headList.size() )
return m_headList.at(section);
}
else
return QString("%1").arg(section + 1);
}
return QAbstractTableModel::headerData(section,orientation,role);
}
void TestMoudle::AddData(StructTest test)
{
beginResetModel();
m_list << test;
endResetModel();
}
void TestMoudle::DelData(int index )
{
beginResetModel();
m_list.removeAt(index);
endResetModel();
}
//mainwidow.cpp
bool InitTableWidget( QTableView *pTableView, QStringList& list, bool bStretch, QList<int>& width, bool bVStretch, int hight)
{
if( !bStretch && list.size() != width.size() )
return false;
pTableView->horizontalHeader()->setMinimumSectionSize(1);
if( bStretch )
{
pTableView->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);//宽度自适应
//pTableView->verticalHeader()->setSectionResizeMode(QHeaderView::Stretch);
}
else
{
for (int i=0;i< width.size();i++ ) {
pTableView->setColumnWidth(i, width.at(i));
}
}
pTableView->horizontalHeader()->setDefaultAlignment (Qt::AlignLeft | Qt::AlignVCenter);
pTableView->setAlternatingRowColors(true);/* 行颜色交替显示 */
if(bVStretch)
pTableView->verticalHeader()->setSectionResizeMode(QHeaderView::Stretch);
else
pTableView->verticalHeader()->setDefaultSectionSize(hight);
pTableView->setSelectionBehavior ( QAbstractItemView::SelectItems);
//pTableView->setEditTriggers(QAbstractItemView::DoubleClicked);
pTableView->verticalHeader()->setHidden( true );
pTableView->setShowGrid(true);
pTableView->setFocusPolicy(Qt::NoFocus);//隐藏选中虚线
pTableView->horizontalHeader()->setHighlightSections(true);
pTableView->setHorizontalScrollBarPolicy(Qt::ScrollBarAsNeeded);
return true;
}
void MainWindow::InitTable()
{
QList<int> width;
QStringList list;
list << "ID" << "Name";
m_pMoude = new TestMoudle(list);
ui->tableView->setModel(m_pMoude);
QAbstractItemView::EditTriggers xx = ui->tableView->editTriggers();
InitTableWidget( ui->tableView, list, true, width, false, 30);
//ui->tableView->setEditTriggers(QAbstractItemView::CurrentChanged);
//ui->tableView->setEditTriggers(QAbstractItemView::DoubleClicked);
// connect(ui->tableView, &QTableView::doubleClicked, this, &MainWindow::slotRowDoubleClicked);
}
//添加
void MainWindow::on_pushButton_clicked()
{
int row = m_pMoude->rowCount();
StructTest test;
test.ID = QString("ID %1").arg(row+1);
test.name = QString("name %1").arg(row+1);
m_pMoude->AddData(test);
}
//删除
void MainWindow::on_pushButton_2_clicked()
{
QModelIndex index = ui->tableView->selectionModel()->currentIndex();
int currentRow = index.row();
if( currentRow < 0)
{
return;
}
m_pMoude->DelData(currentRow);
}
