diff --git a/model/tablemodel.cpp b/model/tablemodel.cpp index cb415cf..40b9bb9 100644 --- a/model/tablemodel.cpp +++ b/model/tablemodel.cpp @@ -10,6 +10,10 @@ enum UserRoles { FactorRole, FactoredAmountRole }; +static const QHash ROLE_NAMES = { + {NameRole, "Name"}, {DescriptionRole, "Description"}, + {InfoRole, "Info"}, {AmountRole, "Amount"}, + {FactorRole, "Factor"}, {FactoredAmountRole, "FactoredAmount"}}; TableModel::TableModel(QObject* parent) : QAbstractTableModel{parent} { @@ -61,7 +65,9 @@ QVariant TableModel::data(const QModelIndex& index, int role) const { QVariant TableModel::headerData(int section, Qt::Orientation orientation, int role) const { if (role == Qt::DisplayRole) { if (orientation == Qt::Horizontal) { - return QString("Section %1").arg(section); + const int columnRole = getRoleForColumn(section); + const QString headerName = ROLE_NAMES.value(columnRole); + return QString("%1").arg(headerName); } else { return QString("%1").arg(section); }