observe/planetmodel.cpp at main · Buffes/observe · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "planetmodel.h"
#include "calculate_positions.h"
#include <QFile>
#include <QThread>

PlanetModel::PlanetModel(QObject *parent):
    QAbstractListModel(parent)
{
    this->data_manager = DataManager::getInstance();

    distance_from_center = 25.0;
    m_workerThread = new WorkerThread(data_manager->m_planets, QDateTime::currentDateTime());
    QObject::connect(m_workerThread, &WorkerThread::new_positions,
                     this, &PlanetModel::updatePositions);
    QObject::connect(this, &PlanetModel::new_date_input,
                     m_workerThread, &WorkerThread::set_date);
}

QHash<int, QByteArray> PlanetModel::roleNames() const {
    QHash<int, QByteArray> roles;
    roles[XRole] = "x";
    roles[YRole] = "y";
    roles[ZRole] = "z";
    roles[ColorRole] = "p_color";
    roles[RadiusRole] = "p_radius";
    return roles;
}

int PlanetModel::rowCount(const QModelIndex& parent) const {
    return data_manager->m_planet_positions.size();
}

QVariant PlanetModel::data(const QModelIndex& index, int role) const {
    if (!index.isValid() || index.row() >= data_manager->m_planet_positions.size())
        return QVariant();

    switch (role) {
        // NOTE: QML seems to only support +Y as up. The math uses +Z. So we convert systems here (swap Y and Z, and negate Z).
        case XRole:
            return data_manager->m_planet_positions[index.row()].x;
        case YRole:
            return data_manager->m_planet_positions[index.row()].z;
        case ZRole:
            return -data_manager->m_planet_positions[index.row()].y;
        case ColorRole:
            return data_manager->m_planets[index.row()].color;
        case RadiusRole:
            return data_manager->m_planets[index.row()].radius;
        default:
            return QVariant();
    }
}

void PlanetModel::updatePositions(QList<dVector3D> positions) {
    // scale positions for visualization purposes. units in are AU
    for (dVector3D &pos : positions) {
            pos.x *= distance_from_center;
            pos.y *= distance_from_center;
            pos.z *= distance_from_center;
    }

    if (data_manager->m_planet_positions.size() < positions.size()) {
            beginInsertRows(QModelIndex(), data_manager->m_planet_positions.size(), data_manager->m_planet_positions.size() + positions.size() - 1);
            data_manager->m_planet_positions = positions;
            endInsertRows();
    }
    else {
            data_manager->m_planet_positions = positions;
            emit dataChanged(createIndex(0, 0), createIndex(data_manager->m_planet_positions.size()-1, 0));
    }
}

void PlanetModel::calculatePositions(QDateTime datetime) {
    emit new_date_input(datetime);
    if (!m_workerThread->active) {
            QList<dVector3D> positions = calc::calculatePositions(data_manager->m_planets, datetime);
        updatePositions(positions);
    }
}

void PlanetModel::calculatePositionsRepeatedly() {
    if (!m_workerThread->active)
        m_workerThread->start();
    else
        m_workerThread->disable();
}

void PlanetModel::setAnimationSpeed(double value) {
    m_workerThread->set_speed(value);
}