subsurface/qt-models/divesiteimportmodel.cpp
Berthold Stoeger 858a0aecba import: initialize DiveSiteImportModel in constructor
The old code would construct and then initialize the object
in a separate function, which added lots of complication.

Just initialize the thing in the constructor, store a
reference, not a pointer to the table. And do a few other
code cleanups. The result is distinctly more pleasing.

Signed-off-by: Berthold Stoeger <bstoeger@mail.tuwien.ac.at>
2024-08-13 19:28:30 +02:00

128 lines
3.7 KiB
C++

#include "divesiteimportmodel.h"
#include "core/divelog.h"
#include "core/qthelper.h"
#include "core/range.h"
#include "core/taxonomy.h"
DivesiteImportedModel::DivesiteImportedModel(dive_site_table &table, QObject *o) : QAbstractTableModel(o),
firstIndex(0),
lastIndex(-1),
importedSitesTable(table)
{
checkStates.resize(importedSitesTable.size());
for (const auto &[row, item]: enumerated_range(importedSitesTable))
checkStates[row] = !divelog.sites->get_by_gps(&item->location);
}
int DivesiteImportedModel::columnCount(const QModelIndex &) const
{
return 5;
}
int DivesiteImportedModel::rowCount(const QModelIndex &) const
{
return static_cast<int>(importedSitesTable.size());
}
QVariant DivesiteImportedModel::headerData(int section, Qt::Orientation orientation, int role) const
{
if (orientation == Qt::Vertical)
return QVariant();
if (role == Qt::DisplayRole) {
switch (section) {
case NAME:
return tr("Name");
case LOCATION:
return tr("Location");
case COUNTRY:
return tr("Country");
case NEAREST:
return tr("Nearest\nExisting Site");
case DISTANCE:
return tr("Distance");
}
}
return QVariant();
}
QVariant DivesiteImportedModel::data(const QModelIndex &index, int role) const
{
if (!index.isValid())
return QVariant();
if (index.row() < 0 || index.row() >= (int)importedSitesTable.size())
return QVariant();
struct dive_site *ds = importedSitesTable[index.row()].get();
// widgets access the model via index.column()
// Not supporting QML access via roles
if (role == Qt::DisplayRole) {
switch (index.column()) {
case NAME:
return QString::fromStdString(ds->name);
case LOCATION:
return printGPSCoords(&ds->location);
case COUNTRY:
return QString::fromStdString(taxonomy_get_country(ds->taxonomy));
case NEAREST: {
// 40075000 is circumference of the earth in meters
struct dive_site *nearest_ds =
divelog.sites->get_by_gps_proximity(ds->location, 40075000);
if (nearest_ds)
return QString::fromStdString(nearest_ds->name);
else
return QString();
}
case DISTANCE: {
unsigned int distance = 0;
struct dive_site *nearest_ds =
divelog.sites->get_by_gps_proximity(ds->location, 40075000);
if (nearest_ds)
distance = get_distance(ds->location, nearest_ds->location);
return distance_string(distance);
}
case SELECTED:
return checkStates[index.row()];
}
}
if (role == Qt::CheckStateRole) {
if (index.column() == 0)
return checkStates[index.row()] ? Qt::Checked : Qt::Unchecked;
}
return QVariant();
}
void DivesiteImportedModel::changeSelected(QModelIndex clickedIndex)
{
checkStates[clickedIndex.row()] = !checkStates[clickedIndex.row()];
dataChanged(index(clickedIndex.row(), 0), index(clickedIndex.row(), 0), QVector<int> { Qt::CheckStateRole, SELECTED });
}
void DivesiteImportedModel::selectAll()
{
std::fill(checkStates.begin(), checkStates.end(), true);
// Qt is mad: for empty lists, last index would be -1, but that makes it crash.
dataChanged(index(0, 0), index(rowCount() - 1, 0), QVector<int> { Qt::CheckStateRole, SELECTED });
}
void DivesiteImportedModel::selectRow(int row)
{
checkStates[row] = !checkStates[row];
dataChanged(index(row, 0), index(row, 0), QVector<int> { Qt::CheckStateRole, SELECTED });
}
void DivesiteImportedModel::selectNone()
{
std::fill(checkStates.begin(), checkStates.end(), false);
// Qt is mad: for empty lists, last index would be -1, but that makes it crash.
dataChanged(index(0, 0), index(rowCount() - 1, 0), QVector<int> { Qt::CheckStateRole, SELECTED });
}
Qt::ItemFlags DivesiteImportedModel::flags(const QModelIndex &index) const
{
if (index.column() != 0)
return QAbstractTableModel::flags(index);
return QAbstractTableModel::flags(index) | Qt::ItemIsUserCheckable;
}