138 lines
4.1 KiB
C++
138 lines
4.1 KiB
C++
#include "tablemodel.h"
|
|
|
|
#include "../formats/jsonparser.h"
|
|
#include "modelitem.h"
|
|
|
|
QHash<int, QByteArray> TableModel::ROLE_NAMES = {{NameRole, "Name"},
|
|
{DescriptionRole, "Description"},
|
|
{InfoRole, "Info"},
|
|
{AmountRole, "Amount"},
|
|
{FactorRole, "Factor"}};
|
|
|
|
TableModel::TableModel(QObject* parent)
|
|
: QAbstractTableModel{parent} {
|
|
for (int row = 0; row < 5; ++row) {
|
|
QHash<int, QVariant> values;
|
|
values[NameRole] = QString("Item %1").arg(row);
|
|
values[DescriptionRole] = QString("This is item %1").arg(row);
|
|
values[InfoRole] = QString("Info of item %1").arg(row);
|
|
values[AmountRole] = row;
|
|
values[FactorRole] = row * 1.1;
|
|
|
|
shared_ptr<ModelItem> item = make_unique<ModelItem>(values);
|
|
m_items.append(std::move(item));
|
|
}
|
|
}
|
|
|
|
Qt::ItemFlags TableModel::flags(const QModelIndex& index) const {
|
|
return Qt::ItemIsEditable | QAbstractTableModel::flags(index);
|
|
}
|
|
|
|
QHash<int, QByteArray> TableModel::roleNames() const { return ROLE_NAMES; }
|
|
|
|
int TableModel::rowCount(const QModelIndex& parent) const {
|
|
Q_UNUSED(parent);
|
|
return m_items.size();
|
|
}
|
|
|
|
int TableModel::columnCount(const QModelIndex& parent) const {
|
|
Q_UNUSED(parent);
|
|
return ROLE_NAMES.size();
|
|
}
|
|
|
|
QVariant TableModel::data(const QModelIndex& index, int role) const {
|
|
const int row = index.row();
|
|
const int column = index.column();
|
|
|
|
if (!index.isValid()) {
|
|
return QVariant();
|
|
}
|
|
if (row >= rowCount(QModelIndex()) || column >= columnCount(QModelIndex())) {
|
|
return QVariant();
|
|
}
|
|
|
|
int roleForColumn = getRoleForColumn(column);
|
|
switch (role) {
|
|
case Qt::DisplayRole:
|
|
case Qt::EditRole:
|
|
return m_items.at(row)->data(roleForColumn);
|
|
}
|
|
|
|
return QVariant();
|
|
}
|
|
|
|
QVariant TableModel::headerData(int section, Qt::Orientation orientation, int role) const {
|
|
if (role == Qt::DisplayRole) {
|
|
if (orientation == Qt::Horizontal) {
|
|
const int columnRole = getRoleForColumn(section);
|
|
const QString headerName = ROLE_NAMES.value(columnRole);
|
|
return QString("%1").arg(headerName);
|
|
} else {
|
|
return QString("%1").arg(section);
|
|
}
|
|
}
|
|
return QVariant();
|
|
}
|
|
|
|
bool TableModel::setData(const QModelIndex& index, const QVariant& value, int role) {
|
|
if (role == Qt::EditRole) {
|
|
if (!checkIndex(index)) {
|
|
return false;
|
|
}
|
|
int columnRole = getRoleForColumn(index.column());
|
|
shared_ptr<ModelItem> item = m_items.at(index.row());
|
|
return item->setData(value, columnRole);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// bool TableModel::setItemData(const QModelIndex& index, const QMap<int, QVariant>& roles) {}
|
|
|
|
void TableModel::appendItems(const QByteArray& jsonDoc) { insertItems(-1, jsonDoc, QModelIndex()); }
|
|
|
|
void TableModel::insertItems(int startPosition,
|
|
const QByteArray& jsonDoc,
|
|
const QModelIndex& parentIndex) {
|
|
qInfo() << "Inserting item(s) into model...";
|
|
if (parentIndex != QModelIndex()) {
|
|
qWarning() << "Using invalid parent index (no child support for now)";
|
|
}
|
|
if (startPosition == -1 || startPosition > m_items.size()) {
|
|
/// Appending item(s)
|
|
startPosition = m_items.size();
|
|
}
|
|
|
|
QList<QHash<int, QVariant>> valueList = JsonParser::toItemValuesList(jsonDoc);
|
|
const int nRows = valueList.size();
|
|
beginInsertRows(QModelIndex(), startPosition, startPosition + nRows - 1);
|
|
for (int row = 0; row < nRows; ++row) {
|
|
const int rowPosition = startPosition + row;
|
|
shared_ptr<ModelItem> item = make_unique<ModelItem>(valueList.at(row));
|
|
m_items.insert(rowPosition, std::move(item));
|
|
}
|
|
endInsertRows();
|
|
}
|
|
|
|
int TableModel::getRoleForColumn(const int column) const {
|
|
switch (column) {
|
|
case 0:
|
|
return NameRole;
|
|
break;
|
|
case 1:
|
|
return DescriptionRole;
|
|
break;
|
|
case 2:
|
|
return InfoRole;
|
|
break;
|
|
case 3:
|
|
return AmountRole;
|
|
break;
|
|
case 4:
|
|
return FactorRole;
|
|
break;
|
|
default:
|
|
return NameRole;
|
|
break;
|
|
}
|
|
}
|