pathmodel.cpp

/*
    SPDX-FileCopyrightText: 2021 Volker Krause <vkrause@kde.org>
 
    SPDX-License-Identifier: LGPL-2.0-or-later
*/
 
#include "pathmodel.h"
 
using namespace KPublicTransport;
 
PathModel::PathModel(QObject *parent)
    : QAbstractListModel(parent)
{
}
 
PathModel::~PathModel() = default;
 
Path PathModel::path() const
{
    return m_path;
}
 
void PathModel::setPath(const Path &path)
{
    beginResetModel();
    m_path = path;
    endResetModel();
}
 
int PathModel::rowCount(const QModelIndex &parent) const
{
    if (parent.isValid()) {
        return 0;
    }
    return m_path.sections().size();
}
 
QVariant PathModel::data(const QModelIndex &index, int role) const
{
    switch (role) {
        case PathSectionRole:
            return m_path.sections()[index.row()];
        case TurnDirectionRole:
        {
            const auto curDir = m_path.sections()[index.row()].direction();
            if (index.row() == 0) {
                return curDir;
            }
            const auto prevDir = m_path.sections()[index.row() - 1].direction();
            if (prevDir >= 0 && curDir >= 0) {
                return (360 + curDir - prevDir) % 360;
            }
            return curDir;
        }
    }
 
    return {};
}
 
QHash<int, QByteArray> PathModel::roleNames() const
{
    auto r = QAbstractListModel::roleNames();
    r.insert(PathSectionRole, "section");
    r.insert(TurnDirectionRole, "turnDirection");
    return r;
}