Qt學習之路(46): 自定義model之二

原創作品，允許轉載，轉載時請務必以超鏈接形式標明文章 原始出處 、作者信息和本聲明。否則將追究法律責任。http://devbean.blog.51cto.com/448512/267972

前面的例子已經比較清楚的給出了自定義model的方法，就是要覆蓋我們所需要的那幾個函數就可以了。但是，前面的例子僅僅是簡單的展示數據，也就是說數據時只讀的。那麼，如何能做到讀寫數據呢？那就要來看進來的例子了。這個例子也是來自C++GUI Programming with Qt 4, 2nd Edition這本書的。

 

還是先來看代碼吧：

 

citymodel.h

 

class CityModel : public QAbstractTableModel 
{ 
        Q_OBJECT 
 
public: 
        CityModel(QObject *parent = 0); 
 
        void setCities(const QStringList &cityNames); 
        int rowCount(const QModelIndex &parent) const; 
        int columnCount(const QModelIndex &parent) const; 
        QVariant data(const QModelIndex &index, int role) const; 
        bool setData(const QModelIndex &index, const QVariant &value, int role); 
        QVariant headerData(int section, Qt::Orientation orientation, int role) const; 
        Qt::ItemFlags flags(const QModelIndex &index) const; 
 
private: 
        int offsetOf(int row, int column) const; 
 
        QStringList cities; 
        QVector<int> distances; 
};

 

 

citymodel.cpp

CityModel::CityModel(QObject *parent) 
        : QAbstractTableModel(parent) 
{ 
} 
 
int CityModel::rowCount(const QModelIndex & parent) const 
{ 
        return cities.count(); 
} 
 
int CityModel::columnCount(const QModelIndex & parent) const 
{ 
        return cities.count(); 
} 
 
QVariant CityModel::data(const QModelIndex &index, int role) const 
{ 
        if (!index.isValid()) { 
                return QVariant(); 
        } 
 
        if (role == Qt::TextAlignmentRole) { 
                return int(Qt::AlignRight | Qt::AlignVCenter); 
        } else if (role == Qt::DisplayRole) { 
                if (index.row() == index.column()) { 
                        return 0; 
                } 
                int offset = offsetOf(index.row(), index.column()); 
                return distances[offset]; 
        } 
        return QVariant(); 
} 
 
QVariant CityModel::headerData(int section, Qt::Orientation orientation, int role) const 
{ 
        if (role == Qt::DisplayRole) { 
                return cities[section]; 
        } 
        return QVariant(); 
} 
 
bool CityModel::setData(const QModelIndex &index, const QVariant &value, int role) 
{ 
        if (index.isValid() && index.row() != index.column() && role == Qt::EditRole) { 
                int offset = offsetOf(index.row(), index.column()); 
                distances[offset] = value.toInt(); 
 
                QModelIndex transposedIndex = createIndex(index.column(), index.row()); 
                emit dataChanged(index, index); 
                emit dataChanged(transposedIndex, transposedIndex); 
                return true; 
        } 
        return false; 
} 
 
Qt::ItemFlags CityModel::flags(const QModelIndex &index) const 
{ 
        Qt::ItemFlags flags = QAbstractItemModel::flags(index); 
        if (index.row() != index.column()) { 
                flags |= Qt::ItemIsEditable; 
        } 
        return flags; 
} 
 
void CityModel::setCities(const QStringList &cityNames) 
{ 
        cities = cityNames; 
        distances.resize(cities.count() * (cities.count() - 1) / 2); 
        distances.fill(0); 
        reset(); 
} 
 
int CityModel::offsetOf(int row, int column) const 
{ 
        if (row < column) { 
                qSwap(row, column); 
        } 
        return (row * (row - 1) / 2) + column; 
} 

 

代碼很長，但實際上和前面我們的那個例子非常相似。這個model也是用於table的，因此還是繼承了QAbstractTableModel。CityModel內部有兩個數據源：一個QStringList類型的對象，一個QVector&lt;int>類型的對象。前者用於保存城市的名字，需要用戶顯示的給出；後者是model內部維護的一個存放int的向量。這個CityModel就是要在table中顯示兩個城市之間的距離。同前面的例子一樣，如果我們要把所有的數據都保存下來，顯然會造成數據的冗餘：城市A到城市B的距離同城市B到城市A的距離是一樣的！因此我們還是自定義一個model。同樣這個CityModel有個簡單的空構造函數，rowCount()和columnCount()函數也是返回list的長度。data()函數根據role的不同返回不同的值。由於在table中座標是由row和column給出的，因此需要有一個二維座標到一維座標的轉換，這就是offsetOf()函數的作用。我們把主要精力放在setData()函數上面。

 

bool CityModel::setData(const QModelIndex &index, const QVariant &value, int role) 
{ 
        if (index.isValid() && index.row() != index.column() && role == Qt::EditRole) { 
                int offset = offsetOf(index.row(), index.column()); 
                distances[offset] = value.toInt(); 
 
                QModelIndex transposedIndex = createIndex(index.column(), index.row()); 
                emit dataChanged(index, index); 
                emit dataChanged(transposedIndex, transposedIndex); 
                return true; 
        } 
        return false; 
}

 

這個函數在用戶編輯數據時會自動調用。也就是說，這時我們的數據已經不是隻讀的了。函數開始是一個長長的判斷：index要是合法的；index的row和column不相等，也就是說兩個城市是不同的；數據想的role是Qt::EditRole。如果滿足了這三個條件，纔會執行下面的操作。首先，由row和column座標定位到表中的數據項在vector中的位置。然後用戶新修改的數據被作爲參數value傳入，所以我們要把這個參數賦值給distances。createIndex()函數根據column和row值生成一個QModelIndex對象。請注意這裏的順序：row和column是顛倒的！這就把座標爲(row, column)的點關於主對角線對稱的那個點(column, row)的index找到了。還記得我們的需求嗎？當我們修改了一個數據時，對應的數據也要被修改，這就是這個功能的實現。我們需要emit dataChanged()信號，這個信號接收兩個參數：一個是被修改的數據的左上角的座標，一個是被修改的數據的右下角的座標。爲什麼會有兩個座標呢？因此我們修改的數據不一定只是一個。像這裏，我們只修改了一個數據，因此這兩個值是相同的。數據更新了，我們用這個信號通知view刷新，這樣就可以顯示新的數據了。最後，如果函數數據修改成功就返回true，否則返回false。

 

最後，我們在main()函數中顯示出來這個model：

 

int main(int argc, char *argv[]) 
{ 
        QApplication a(argc, argv); 
        QStringList cities; 
        cities << "Arvika" << "Boden" << "Eskilstuna" << "Falun"; 
 
        CityModel cityModel; 
        cityModel.setCities(cities); 
 
        QTableView tableView; 
        tableView.setModel(&cityModel); 
        tableView.setAlternatingRowColors(true); 
        tableView.setWindowTitle(QObject::tr("Cities")); 
        tableView.show(); 
        return a.exec(); 
}

 

這樣，我們就把這個model做完了。最後來看看效果吧！

 

本文出自 “豆子空間” 博客，請務必保留此出處http://devbean.blog.51cto.com/448512/267972