ADSBVehicleManager.cc

Go to the documentation of this file.

#include "ADSBVehicleManager.h"
#include "QGCApplication.h"
#include "SettingsManager.h"
#include "ADSBVehicleManagerSettings.h"
#include "ADSBTCPLink.h"
#include "ADSBVehicle.h"
#include "QmlObjectListModel.h"
#include "QGCLoggingCategory.h"
 
#include <QtCore/QApplicationStatic>
#include <QtCore/QTimer>
#include <qassert.h>
 
QGC_LOGGING_CATEGORY(ADSBVehicleManagerLog, "ADSB.ADSBVehicleManager")
 
Q_APPLICATION_STATIC(ADSBVehicleManager, _adsbVehicleManager, SettingsManager::instance()->adsbVehicleManagerSettings());
 
ADSBVehicleManager::ADSBVehicleManager(ADSBVehicleManagerSettings *settings, QObject *parent)
    : QObject(parent)
    , _adsbSettings(settings)
    , _adsbVehicleCleanupTimer(new QTimer(this))
    , _adsbVehicles(new QmlObjectListModel(this))
{
    // qCDebug(ADSBVehicleManagerLog) << Q_FUNC_INFO << this;
 
    (void) qRegisterMetaType<ADSB::VehicleInfo_t>("ADSB::VehicleInfo_t");
 
    _adsbVehicleCleanupTimer->setSingleShot(false);
    _adsbVehicleCleanupTimer->setInterval(1000);
    (void) connect(_adsbVehicleCleanupTimer, &QTimer::timeout, this, &ADSBVehicleManager::_cleanupStaleVehicles);
 
    Fact* const adsbEnabled = _adsbSettings->adsbServerConnectEnabled();
    Fact* const hostAddress = _adsbSettings->adsbServerHostAddress();
    Fact* const port = _adsbSettings->adsbServerPort();
 
    (void) connect(adsbEnabled, &Fact::rawValueChanged, this, [this, hostAddress, port](QVariant value) {
        if (value.toBool()) {
            _start(hostAddress->rawValue().toString(), port->rawValue().toUInt());
        } else {
            _stop();
        }
    });
 
    if (adsbEnabled->rawValue().toBool()) {
        _start(hostAddress->rawValue().toString(), port->rawValue().toUInt());
    }
}
 
ADSBVehicleManager::~ADSBVehicleManager()
{
    // qCDebug(ADSBVehicleManagerLog) << Q_FUNC_INFO << this;
}
 
ADSBVehicleManager *ADSBVehicleManager::instance()
{
    return _adsbVehicleManager();
}
 
void ADSBVehicleManager::mavlinkMessageReceived(const mavlink_message_t &message)
{
    if (message.msgid != MAVLINK_MSG_ID_ADSB_VEHICLE) {
        return;
    }
 
    _handleADSBVehicle(message);
}
 
void ADSBVehicleManager::_handleADSBVehicle(const mavlink_message_t &message)
{
    mavlink_adsb_vehicle_t adsbVehicleMsg{};
    mavlink_msg_adsb_vehicle_decode(&message, &adsbVehicleMsg);
 
    if (adsbVehicleMsg.tslc > kMaxTimeSinceLastSeen) {
        return;
    }
 
    ADSB::VehicleInfo_t vehicleInfo{};
 
    vehicleInfo.availableFlags = ADSB::AvailableInfoTypes::fromInt(0);
 
    vehicleInfo.icaoAddress = adsbVehicleMsg.ICAO_address;
    vehicleInfo.lastContact = adsbVehicleMsg.tslc;
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_VALID_COORDS) {
        vehicleInfo.availableFlags |= ADSB::LocationAvailable;
        vehicleInfo.location.setLatitude(adsbVehicleMsg.lat / 1e7);
        vehicleInfo.location.setLongitude(adsbVehicleMsg.lon / 1e7);
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_VALID_ALTITUDE) {
        vehicleInfo.availableFlags |= ADSB::AltitudeAvailable;
        vehicleInfo.location.setAltitude(adsbVehicleMsg.altitude / 1e3);
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_VALID_HEADING) {
        vehicleInfo.availableFlags |= ADSB::HeadingAvailable;
        vehicleInfo.heading = adsbVehicleMsg.heading / 1e2;
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_VALID_VELOCITY) {
        vehicleInfo.availableFlags |= ADSB::VelocityAvailable;
        vehicleInfo.velocity = adsbVehicleMsg.hor_velocity / 1e2;
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_VALID_CALLSIGN) {
        vehicleInfo.availableFlags |= ADSB::CallsignAvailable;
        vehicleInfo.callsign = QString::fromLatin1(adsbVehicleMsg.callsign, sizeof(adsbVehicleMsg.callsign));
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_VALID_SQUAWK) {
        vehicleInfo.availableFlags |= ADSB::SquawkAvailable;
        vehicleInfo.squawk = adsbVehicleMsg.squawk;
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_SIMULATED) {
        vehicleInfo.simulated = true;
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_VERTICAL_VELOCITY_VALID) {
        vehicleInfo.availableFlags |= ADSB::VerticalVelAvailable;
        vehicleInfo.verticalVel = adsbVehicleMsg.ver_velocity;
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_BARO_VALID) {
        vehicleInfo.baro = true;
    }
 
    if (adsbVehicleMsg.flags & ADSB_FLAGS_SOURCE_UAT) {
 
    }
 
    adsbVehicleMsg.altitude_type = adsbVehicleMsg.altitude_type;
    adsbVehicleMsg.emitter_type = adsbVehicleMsg.emitter_type;
 
    adsbVehicleUpdate(vehicleInfo);
}
 
void ADSBVehicleManager::adsbVehicleUpdate(const ADSB::VehicleInfo_t &vehicleInfo)
{
    const uint32_t icaoAddress = vehicleInfo.icaoAddress;
    if (_adsbICAOMap.contains(icaoAddress)) {
        _adsbICAOMap[icaoAddress]->update(vehicleInfo);
        return;
    }
 
    if (vehicleInfo.availableFlags & ADSB::LocationAvailable) {
        ADSBVehicle* const adsbVehicle = new ADSBVehicle(vehicleInfo, this);
        _adsbICAOMap[icaoAddress] = adsbVehicle;
        _adsbVehicles->append(adsbVehicle);
        qCDebug(ADSBVehicleManagerLog) << "Added" << QString::number(adsbVehicle->icaoAddress());
    }
}
 
void ADSBVehicleManager::_start(const QString &hostAddress, quint16 port)
{
    Q_ASSERT(!_adsbTcpLink);
 
    ADSBTCPLink *adsbTcpLink = new ADSBTCPLink(QHostAddress(hostAddress), port, this);
    if (!adsbTcpLink->init()) {
        delete adsbTcpLink;
        adsbTcpLink = nullptr;
        qCWarning(ADSBVehicleManagerLog) << "Failed to Initialize TCP Link at:" << hostAddress << port;
        return;
    }
 
    _adsbTcpLink = adsbTcpLink;
    (void) connect(_adsbTcpLink, &ADSBTCPLink::adsbVehicleUpdate, this, &ADSBVehicleManager::adsbVehicleUpdate, Qt::AutoConnection);
    (void) connect(_adsbTcpLink, &ADSBTCPLink::errorOccurred, this, &ADSBVehicleManager::_linkError, Qt::AutoConnection);
 
    _adsbVehicleCleanupTimer->start();
}
 
void ADSBVehicleManager::_stop()
{
    Q_CHECK_PTR(_adsbTcpLink);
    _adsbTcpLink->deleteLater();
    _adsbTcpLink = nullptr;
 
    _adsbVehicleCleanupTimer->stop();
 
    _adsbVehicles->clearAndDeleteContents();
    _adsbICAOMap.clear();
}
 
void ADSBVehicleManager::_cleanupStaleVehicles()
{
    for (qsizetype i = _adsbVehicles->count() - 1; i >= 0; i--) {
        ADSBVehicle* const adsbVehicle = _adsbVehicles->value<ADSBVehicle*>(i);
        if (adsbVehicle->expired()) {
            qCDebug(ADSBVehicleManagerLog) << "Expired" << QString::number(adsbVehicle->icaoAddress());
            (void) _adsbVehicles->removeAt(i);
            (void) _adsbICAOMap.remove(adsbVehicle->icaoAddress());
            adsbVehicle->deleteLater();
        }
    }
}
 
void ADSBVehicleManager::_linkError(const QString &errorMsg, bool stopped)
{
    qCDebug(ADSBVehicleManagerLog) << errorMsg;
 
    QString msg = QStringLiteral("ADSB Server Error: %1").arg(errorMsg);
 
    if (stopped) {
        (void) msg.append("\nADSB has been disabled");
        _adsbSettings->adsbServerConnectEnabled()->setRawValue(false);
    }
 
    qgcApp()->showAppMessage(msg);
}