QSqlTableModel和QTableView设置复选框

//头文件

class SqlTableModel:public QSqlTableModel

{
Q_OBJECT  

public:  
SqlTableModel ( QObject * parent = 0, QSqlDatabase db = QSqlDatabase() );  
bool setData( const QModelIndex &index, const QVariant &value, int role );
QVariant data(const QModelIndex &index, int role) const;
Qt::ItemFlags flags(const QModelIndex &index) const;
};

//cpp

QMap<int, Qt::CheckState> check_state_map; 

// setData()方法主要用来设置是否被选中，然后将对应的状态保存到QMap中

SqlTableModel::SqlTableModel ( QObject * parent, QSqlDatabase db ):QSqlTableModel(parent, db)

{

}

bool SqlTableModel::setData( const QModelIndex &index, const QVariant &value, int role )  

{  
if(!index.isValid())  
return false;  

if (role == Qt::CheckStateRole && index.column() == 0)  
{  
check_state_map[index.row()] = (value == Qt::Checked ? Qt::Checked : Qt::Unchecked); 
return true;  
}
else 
return QSqlTableModel::setData(index, value,role);

} 

// data()方法主要用来显示，取出QMap中的值，返回对应的状态

QVariant SqlTableModel::data(const QModelIndex &index, int role) const

{
if (!index.isValid())
return QVariant();
if (index.column() != 0 )
{
return QSqlTableModel::data(index, role);
}
switch(role)
{
case Qt::CheckStateRole:
if(index.column() == 0)   
{   
if (check_state_map.contains(index.row()))   
return check_state_map[index.row()] == Qt::Checked ? Qt::Checked : Qt::Unchecked; 
return Qt::Unchecked;    
}
default:
return QVariant();
}
return QVariant();

}

//flag()方法主要设置用户可选角色，绘制出QCheckBox

Qt::ItemFlags SqlTableModel::flags(const QModelIndex &index) const

{
if (!index.isValid())  
return 0;  

if (index.column() == 0)  
return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsUserCheckable;  
else
return QSqlTableModel::flags(index);
return  Qt::ItemIsEnabled | Qt::ItemIsSelectable;  

}

效果图