subsurface/qt-models/divecomputermodel.cpp
Berthold Stoeger 4467477389 models: update DiveComputerModel when core data is reset
To implement undo-semantics, we want a longer-lived dive-computer-model
(currently, it is regenerated when the dialog is opened). Therefore, it
must be reloaded when the core data is reset. Do this like for other
models: listen to the dataReset() signal of DiveListNotifier.

Signed-off-by: Berthold Stoeger <bstoeger@mail.tuwien.ac.at>
2020-10-25 13:59:04 -07:00

133 lines
3.8 KiB
C++

// SPDX-License-Identifier: GPL-2.0
#include "qt-models/divecomputermodel.h"
#include "core/dive.h"
#include "core/divelist.h"
#include "core/subsurface-qt/divelistnotifier.h"
DiveComputerModel::DiveComputerModel(QObject *parent) : CleanerTableModel(parent),
dcs(device_table.devices)
{
connect(&diveListNotifier, &DiveListNotifier::dataReset, this, &DiveComputerModel::update);
setHeaderDataStrings(QStringList() << "" << tr("Model") << tr("Device ID") << tr("Nickname"));
update();
}
void DiveComputerModel::update()
{
beginResetModel();
dcs = device_table.devices;
endResetModel();
}
QVariant DiveComputerModel::data(const QModelIndex &index, int role) const
{
if (index.row() < 0 || index.row() >= (int)dcs.size())
return QVariant();
const device &node = dcs[index.row()];
if (role == Qt::DisplayRole || role == Qt::EditRole) {
switch (index.column()) {
case ID:
return QString("0x").append(QString::number(node.deviceId, 16));
case MODEL:
return QString::fromStdString(node.model);
case NICKNAME:
return QString::fromStdString(node.nickName);
}
}
if (index.column() == REMOVE) {
switch (role) {
case Qt::DecorationRole:
return trashIcon();
case Qt::SizeHintRole:
return trashIcon().size();
case Qt::ToolTipRole:
return tr("Clicking here will remove this dive computer.");
}
}
return QVariant();
}
int DiveComputerModel::rowCount(const QModelIndex&) const
{
return dcs.size();
}
Qt::ItemFlags DiveComputerModel::flags(const QModelIndex &index) const
{
Qt::ItemFlags flags = QAbstractItemModel::flags(index);
if (index.column() == NICKNAME)
flags |= Qt::ItemIsEditable;
return flags;
}
bool DiveComputerModel::setData(const QModelIndex &index, const QVariant &value, int)
{
// We should test if the role == Qt::EditRole
if (index.row() < 0 || index.row() >= (int)dcs.size())
return false;
device &node = dcs[index.row()];
node.nickName = value.toString().toStdString();
emit dataChanged(index, index);
return true;
}
void DiveComputerModel::remove(const QModelIndex &index)
{
if (index.row() < 0 || index.row() >= (int)dcs.size())
return;
beginRemoveRows(QModelIndex(), index.row(), index.row());
dcs.erase(dcs.begin() + index.row());
endRemoveRows();
}
void DiveComputerModel::keepWorkingList()
{
if (device_table.devices != dcs)
mark_divelist_changed(true);
device_table.devices = dcs;
}
// Convenience function to access alternative columns
static QString getData(const QModelIndex &idx, int col)
{
const QAbstractItemModel *model = idx.model();
QModelIndex idx2 = model->index(idx.row(), col, idx.parent());
return model->data(idx2).toString();
}
// Helper function: sort data pointed to by the given indexes.
// For equal data, sort by two alternative rows.
// All sorting is by case-insensitive string comparison.
static bool sortHelper(const QModelIndex &i1, const QModelIndex &i2, int altRow1, int altRow2)
{
if(int cmp = i1.data().toString().compare(i2.data().toString()))
return cmp < 0;
if(int cmp = getData(i1, altRow1).compare(getData(i2, altRow1)))
return cmp < 0;
return getData(i1, altRow2).compare(getData(i2, altRow2)) < 0;
}
bool DiveComputerSortedModel::lessThan(const QModelIndex &i1, const QModelIndex &i2) const
{
// We assume that i1.column() == i2.column()
switch (i1.column()) {
case DiveComputerModel::ID:
return sortHelper(i1, i2, DiveComputerModel::MODEL, DiveComputerModel::NICKNAME);
case DiveComputerModel::MODEL:
default:
return sortHelper(i1, i2, DiveComputerModel::ID, DiveComputerModel::NICKNAME);
case DiveComputerModel::NICKNAME:
return sortHelper(i1, i2, DiveComputerModel::MODEL, DiveComputerModel::ID);
}
}
void DiveComputerSortedModel::remove(const QModelIndex &index)
{
int row = mapToSource(index).row();
if (row < 0 || row >= (int)device_table.devices.size())
return;
device_table.devices.erase(device_table.devices.begin() + row);
}