APMFirmwarePlugin.h 7.54 KB
Newer Older
1 2
/****************************************************************************
 *
Gus Grubba's avatar
Gus Grubba committed
3
 * (c) 2009-2020 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
4 5 6 7 8 9
 *
 * QGroundControl is licensed according to the terms in the file
 * COPYING.md in the root of the source code directory.
 *
 ****************************************************************************/

Don Gagne's avatar
Don Gagne committed
10 11 12 13 14 15 16 17

/// @file
///     @author Don Gagne <don@thegagnes.com>

#ifndef APMFirmwarePlugin_H
#define APMFirmwarePlugin_H

#include "FirmwarePlugin.h"
18
#include "QGCLoggingCategory.h"
19
#include "APMParameterMetaData.h"
20
#include "FollowMe.h"
Don Gagne's avatar
Don Gagne committed
21

Don Gagne's avatar
Don Gagne committed
22 23
#include <QAbstractSocket>

24
Q_DECLARE_LOGGING_CATEGORY(APMFirmwarePluginLog)
Pritam Ghanghas's avatar
Pritam Ghanghas committed
25

26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
class APMFirmwareVersion
{
public:
    APMFirmwareVersion(const QString &versionText = "");
    bool isValid() const;
    bool isBeta() const;
    bool isDev() const;
    bool operator<(const APMFirmwareVersion& other) const;
    QString versionString() const { return _versionString; }
    QString vehicleType() const { return _vehicleType; }
    int majorNumber() const { return _major; }
    int minorNumber() const { return _minor; }
    int patchNumber() const { return _patch; }

private:
    void _parseVersion(const QString &versionText);
    QString _versionString;
    QString _vehicleType;
    int     _major;
    int     _minor;
    int     _patch;
};

49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
class APMCustomMode
{
public:
    APMCustomMode(uint32_t mode, bool settable);
    uint32_t modeAsInt() const { return _mode; }
    bool canBeSet() const { return _settable; }
    QString modeString() const;

protected:
    void setEnumToStringMapping(const QMap<uint32_t,QString>& enumToString);

private:
    uint32_t               _mode;
    bool                   _settable;
    QMap<uint32_t,QString> _enumToString;
};

66
/// This is the base class for all stack specific APM firmware plugins
Don Gagne's avatar
Don Gagne committed
67 68 69
class APMFirmwarePlugin : public FirmwarePlugin
{
    Q_OBJECT
70

Don Gagne's avatar
Don Gagne committed
71 72
public:
    // Overrides from FirmwarePlugin
73

74 75
    QList<VehicleComponent*> componentsForVehicle(AutoPilotPlugin* vehicle) override;
    QList<MAV_CMD> supportedMissionCommands(void) override;
76

77
    AutoPilotPlugin*    autopilotPlugin                 (Vehicle* vehicle) override;
78
    bool                isCapable                       (const Vehicle *vehicle, FirmwareCapabilities capabilities) override;
79
    void                setGuidedMode                   (Vehicle* vehicle, bool guidedMode) override;
80
    void                guidedModeTakeoff               (Vehicle* vehicle, double altitudeRel) override;
81
    void                guidedModeGotoLocation          (Vehicle* vehicle, const QGeoCoordinate& gotoCoord) override;
82
    double              minimumTakeoffAltitude          (Vehicle* vehicle) override;
83 84 85 86 87
    void                startMission                    (Vehicle* vehicle) override;
    QStringList         flightModes                     (Vehicle* vehicle) override;
    QString             flightMode                      (uint8_t base_mode, uint32_t custom_mode) const override;
    bool                setFlightMode                   (const QString& flightMode, uint8_t* base_mode, uint32_t* custom_mode) override;
    bool                isGuidedMode                    (const Vehicle* vehicle) const override;
Don Gagne's avatar
Don Gagne committed
88
    QString             gotoFlightMode                  (void) const override { return QStringLiteral("Guided"); }
89
    QString             rtlFlightMode                   (void) const override { return QString("RTL"); }
DonLakeFlyer's avatar
DonLakeFlyer committed
90
    QString             smartRTLFlightMode              (void) const override { return QString("Smart RTL"); }
91 92
    QString             missionFlightMode               (void) const override { return QString("Auto"); }
    void                pauseVehicle                    (Vehicle* vehicle) override;
DonLakeFlyer's avatar
DonLakeFlyer committed
93
    void                guidedModeRTL                   (Vehicle* vehicle, bool smartRTL) override;
94
    void                guidedModeChangeAltitude        (Vehicle* vehicle, double altitudeChange) override;
95
    bool                adjustIncomingMavlinkMessage    (Vehicle* vehicle, mavlink_message_t* message) override;
96
    void                adjustOutgoingMavlinkMessage    (Vehicle* vehicle, LinkInterface* outgoingLink, mavlink_message_t* message) override;
97
    virtual void        initializeStreamRates           (Vehicle* vehicle);
98 99 100
    void                initializeVehicle               (Vehicle* vehicle) override;
    bool                sendHomePositionToVehicle       (void) override;
    void                addMetaDataToFact               (QObject* parameterMetaData, Fact* fact, MAV_TYPE vehicleType) override;
101
    QString             missionCommandOverrides         (MAV_TYPE vehicleType) const override;
102 103 104 105
    QString             getVersionParam                 (void) override { return QStringLiteral("SYSID_SW_MREV"); }
    QString             internalParameterMetaDataFile   (Vehicle* vehicle) override;
    void                getParameterMetaDataVersionInfo (const QString& metaDataFile, int& majorVersion, int& minorVersion) override { APMParameterMetaData::getParameterMetaDataVersionInfo(metaDataFile, majorVersion, minorVersion); }
    QObject*            loadParameterMetaData           (const QString& metaDataFile) override;
106 107
    QString             brandImageIndoor                (const Vehicle* vehicle) const override { Q_UNUSED(vehicle); return QStringLiteral("/qmlimages/APM/BrandImage"); }
    QString             brandImageOutdoor               (const Vehicle* vehicle) const override { Q_UNUSED(vehicle); return QStringLiteral("/qmlimages/APM/BrandImage"); }
Don Gagne's avatar
Don Gagne committed
108

109 110
protected:
    /// All access to singleton is through stack specific implementation
111
    APMFirmwarePlugin(void);
112 113 114

    void setSupportedModes  (QList<APMCustomMode> supportedModes);
    void _sendGCSMotionReport(Vehicle* vehicle, FollowMe::GCSMotionReport& motionReport, uint8_t estimatationCapabilities);
Don Gagne's avatar
Don Gagne committed
115

116
    bool                _coaxialMotors;
Don Gagne's avatar
Don Gagne committed
117

Don Gagne's avatar
Don Gagne committed
118 119
private slots:
    void _artooSocketError(QAbstractSocket::SocketError socketError);
120

121
private:
122
    void _adjustSeverity(mavlink_message_t* message) const;
Don Gagne's avatar
Don Gagne committed
123
    void _adjustCalibrationMessageSeverity(mavlink_message_t* message) const;
Pritam Ghanghas's avatar
Pritam Ghanghas committed
124
    static bool _isTextSeverityAdjustmentNeeded(const APMFirmwareVersion& firmwareVersion);
125 126
    void _setInfoSeverity(mavlink_message_t* message) const;
    QString _getMessageText(mavlink_message_t* message) const;
127
    void _handleIncomingParamValue(Vehicle* vehicle, mavlink_message_t* message);
128
    bool _handleIncomingStatusText(Vehicle* vehicle, mavlink_message_t* message);
129
    void _handleIncomingHeartbeat(Vehicle* vehicle, mavlink_message_t* message);
130
    void _handleOutgoingParamSet(Vehicle* vehicle, LinkInterface* outgoingLink, mavlink_message_t* message);
DonLakeFlyer's avatar
DonLakeFlyer committed
131
    void _soloVideoHandshake(Vehicle* vehicle, bool originalSoloFirmware);
132
    bool _guidedModeTakeoff(Vehicle* vehicle, double altitudeRel);
133 134
    void _handleRCChannels(Vehicle* vehicle, mavlink_message_t* message);
    void _handleRCChannelsRaw(Vehicle* vehicle, mavlink_message_t* message);
135 136
    QString _getLatestVersionFileUrl(Vehicle* vehicle) override;
    QString _versionRegex() override;
137

138 139 140
    // Any instance data here must be global to all vehicles
    // Vehicle specific data should go into APMFirmwarePluginInstanceData

141
    QList<APMCustomMode>    _supportedModes;
142
    QMap<int /* vehicle id */, QMap<int /* componentId */, bool /* true: component is part of ArduPilot stack */>> _ardupilotComponentMap;
Don Gagne's avatar
Don Gagne committed
143

144 145
    static const char*      _artooIP;
    static const int        _artooVideoHandshakePort;
146 147 148 149 150 151 152
};

class APMFirmwarePluginInstanceData : public QObject
{
    Q_OBJECT

public:
153
    APMFirmwarePluginInstanceData(QObject* parent = nullptr);
Don Gagne's avatar
Don Gagne committed
154

155
    bool                    textSeverityAdjustmentNeeded;
Don Gagne's avatar
Don Gagne committed
156 157 158
};

#endif