model.cpp

Go to the documentation of this file.

// Copyright (C) 2016 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
 
/*
    model.cpp
 
    Provides a table model for use in various examples.
*/
 
#include <QtGui>
 
#include "model.h"
 
TableModel::TableModel(int rows, int columns, QObject *parent)
    : QAbstractTableModel(parent)
{
    QStringList newList;
 
    for (int column = 0; column < qMax(1, columns); ++column) {
        newList.append(QString());
    }
 
    for (int row = 0; row < qMax(1, rows); ++row) {
        rowList.append(newList);
    }
}
 
 
int TableModel::rowCount(const QModelIndex &/*parent*/) const
{
    return rowList.size();
}
 
int TableModel::columnCount(const QModelIndex &/*parent*/) const
{
    return rowList[0].size();
}
 
QVariant TableModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid())
        return QVariant();
 
    if (role == Qt::DisplayRole)
        return rowList[index.row()][index.column()];
    else
        return QVariant();
}
 
QVariant TableModel::headerData(int section, Qt::Orientation orientation,
                                int role) const
{
    if (role != Qt::DisplayRole)
        return QVariant();
 
    if (orientation == Qt::Horizontal)
        return QStringLiteral("Column %1").arg(section);
    else
        return QStringLiteral("Row %1").arg(section);
}
 
Qt::ItemFlags TableModel::flags(const QModelIndex &index) const
{
    if (!index.isValid())
        return Qt::ItemIsEnabled;
 
    return QAbstractTableModel::flags(index) | Qt::ItemIsEditable;
}
 
bool TableModel::setData(const QModelIndex &index,
                         const QVariant &value, int role)
{
    if (!index.isValid() || role != Qt::EditRole)
        return false;
 
    rowList[index.row()][index.column()] = value.toString();
    emit dataChanged(index, index, {role});
    return true;
}
 
bool TableModel::insertRows(int position, int rows, const QModelIndex &parent)
{
    int columns = columnCount();
    beginInsertRows(parent, position, position + rows - 1);
 
    for (int row = 0; row < rows; ++row) {
        QStringList items;
        for (int column = 0; column < columns; ++column)
            items.append(QString());
        rowList.insert(position, items);
    }
 
    endInsertRows();
    return true;
}
 
bool TableModel::insertColumns(int position, int columns,
                               const QModelIndex &parent)
{
    int rows = rowCount();
    beginInsertColumns(parent, position, position + columns - 1);
 
    for (int row = 0; row < rows; ++row) {
        for (int column = position; column < columns; ++column) {
            rowList[row].insert(position, QString());
        }
    }
 
    endInsertColumns();
    return true;
}
 
bool TableModel::removeRows(int position, int rows, const QModelIndex &parent)
{
    beginRemoveRows(parent, position, position + rows - 1);
 
    for (int row = 0; row < rows; ++row) {
        rowList.removeAt(position);
    }
 
    endRemoveRows();
    return true;
}
 
bool TableModel::removeColumns(int position, int columns,
                               const QModelIndex &parent)
{
    int rows = rowCount();
    beginRemoveColumns(parent, position, position + columns - 1);
 
    for (int row = 0; row < rows; ++row) {
        for (int column = 0; column < columns; ++column) {
            rowList[row].removeAt(position);
        }
    }
 
    endRemoveColumns();
    return true;
}