QGCXPlaneLink.h 6.17 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
 *
 * QGroundControl is licensed according to the terms in the file
 * COPYING.md in the root of the source code directory.
 *
 ****************************************************************************/
9 10 11 12 13 14 15 16 17


/**
 * @file QGCXPlaneLink.h
 *   @brief X-Plane simulation link
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

18
#pragma once
19 20 21 22 23 24 25 26 27

#include <QString>
#include <QList>
#include <QMap>
#include <QMutex>
#include <QUdpSocket>
#include <QTimer>
#include <QProcess>
#include <LinkInterface.h>
28
#include "QGCConfig.h"
29
#include "QGCHilLink.h"
30
#include "Vehicle.h"
31 32 33 34 35 36 37

class QGCXPlaneLink : public QGCHilLink
{
    Q_OBJECT
    //Q_INTERFACES(QGCXPlaneLinkInterface:LinkInterface)

public:
Bart Slinger's avatar
Bart Slinger committed
38
    QGCXPlaneLink(Vehicle* vehicle, QString remoteHost=QString("127.0.0.1:49000"), QHostAddress localHost = QHostAddress::Any, quint16 localPort = 49005);
39 40
    ~QGCXPlaneLink();

Lorenz Meier's avatar
Lorenz Meier committed
41 42 43 44 45 46 47 48 49 50
    /**
     * @brief Load X-Plane HIL settings
     */
    void loadSettings();

    /**
     * @brief Store X-Plane HIL settings
     */
    void storeSettings();

51 52
    bool isConnected();
    qint64 bytesAvailable();
Bart Slinger's avatar
Bart Slinger committed
53
    int getPort() const {
54
        return localPort;
55 56 57 58 59 60 61 62 63
    }

    /**
     * @brief The human readable port name
     */
    QString getName();

    void run();

Lorenz Meier's avatar
Lorenz Meier committed
64 65 66 67 68 69
    /**
     * @brief Get remote host and port
     * @return string in format <host>:<port>
     */
    QString getRemoteHost();

70 71 72 73 74 75 76 77 78 79
    enum AIRFRAME
    {
        AIRFRAME_UNKNOWN = 0,
        AIRFRAME_QUAD_DJI_F450_PWM,
        AIRFRAME_QUAD_X_MK_10INCH_I2C,
        AIRFRAME_QUAD_X_ARDRONE,
        AIRFRAME_FIXED_WING_BIXLER_II,
        AIRFRAME_FIXED_WING_BIXLER_II_AILERONS
    };

80 81 82 83 84 85 86 87 88 89
    QString getVersion()
    {
        return QString("X-Plane %1").arg(xPlaneVersion);
    }

    int getAirFrameIndex()
    {
        return (int)airframeID;
    }

Bart Slinger's avatar
Bart Slinger committed
90
    bool sensorHilEnabled() {
91 92 93
        return _sensorHilEnabled;
    }

Bart Slinger's avatar
Bart Slinger committed
94
    bool useHilActuatorControls() {
95 96 97 98 99 100 101
        return _useHilActuatorControls;
    }

signals:
    /** @brief Sensor leve HIL state changed */
    void useHilActuatorControlsChanged(bool enabled);

102 103 104 105
public slots:
//    void setAddress(QString address);
    void setPort(int port);
    /** @brief Add a new host to broadcast messages to */
Bart Slinger's avatar
Bart Slinger committed
106
    void setRemoteHost(const QString& host);
107
    /** @brief Send new control states to the simulation */
108
    void updateControls(quint64 time, float rollAilerons, float pitchElevator, float yawRudder, float throttle, quint8 systemMode, quint8 navMode);
109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
    /** @brief Send new control commands to the simulation */
    void updateActuatorControls(quint64 time, quint64 flags,
                                float ctl_0,
                                float ctl_1,
                                float ctl_2,
                                float ctl_3,
                                float ctl_4,
                                float ctl_5,
                                float ctl_6,
                                float ctl_7,
                                float ctl_8,
                                float ctl_9,
                                float ctl_10,
                                float ctl_11,
                                float ctl_12,
                                float ctl_13,
                                float ctl_14,
                                float ctl_15,
                                quint8 mode);
Lorenz Meier's avatar
Lorenz Meier committed
128
    /** @brief Set the simulator version as text string */
Bart Slinger's avatar
Bart Slinger committed
129
    void setVersion(const QString& version);
Lorenz Meier's avatar
Lorenz Meier committed
130 131
    /** @brief Set the simulator version as integer */
    void setVersion(unsigned int version);
Lorenz Meier's avatar
Lorenz Meier committed
132

Bart Slinger's avatar
Bart Slinger committed
133
    void enableSensorHIL(bool enable) {
134 135
        if (enable != _sensorHilEnabled)
            _sensorHilEnabled = enable;
Bart Slinger's avatar
Bart Slinger committed
136
            emit sensorHilChanged(enable);
Lorenz Meier's avatar
Lorenz Meier committed
137 138
    }

139 140
    void enableHilActuatorControls(bool enable);

141 142 143
    void processError(QProcess::ProcessError err);

    void readBytes();
144 145

private slots:
146 147 148 149 150 151
    /**
     * @brief Write a number of bytes to the interface.
     *
     * @param data Pointer to the data byte array
     * @param size The size of the bytes array
     **/
152
    void _writeBytes(const QByteArray data);
153 154

public slots:
155 156
    bool connectSimulation();
    bool disconnectSimulation();
157 158 159 160
    /**
     * @brief Select airplane model
     * @param plane the name of the airplane
     */
Bart Slinger's avatar
Bart Slinger committed
161
    void selectAirframe(const QString& airframe);
162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181
    /**
     * @brief Set the airplane position and attitude
     * @param lat
     * @param lon
     * @param alt
     * @param roll
     * @param pitch
     * @param yaw
     */
    void setPositionAttitude(double lat, double lon, double alt, double roll, double pitch, double yaw);

    /**
     * @brief Set a random position
     */
    void setRandomPosition();

    /**
     * @brief Set a random attitude
     */
    void setRandomAttitude();
182 183

protected:
Bart Slinger's avatar
Bart Slinger committed
184
    Vehicle* _vehicle;
185
    QString name;
186 187 188 189
    QHostAddress localHost;
    quint16 localPort;
    QHostAddress remoteHost;
    quint16 remotePort;
190
    int id;
Bart Slinger's avatar
Bart Slinger committed
191
    QUdpSocket* socket;
192 193 194 195 196 197 198 199 200 201 202 203
    bool connectState;

    quint64 bitsSentTotal;
    quint64 bitsSentCurrent;
    quint64 bitsSentMax;
    quint64 bitsReceivedTotal;
    quint64 bitsReceivedCurrent;
    quint64 bitsReceivedMax;
    quint64 connectionStartTime;
    QMutex statisticsMutex;
    QMutex dataMutex;
    QTimer refreshTimer;
Bart Slinger's avatar
Bart Slinger committed
204 205
    QProcess* process;
    QProcess* terraSync;
206

207 208 209
    bool gpsReceived;
    bool attitudeReceived;

210
    float roll, pitch, yaw, rollspeed, pitchspeed, yawspeed;
211
    double lat, lon, alt, alt_agl;
212
    float vx, vy, vz, xacc, yacc, zacc;
213 214
    float ind_airspeed;
    float true_airspeed;
215
    float groundspeed;
Lorenz Meier's avatar
Lorenz Meier committed
216
    float xmag, ymag, zmag, abs_pressure, diff_pressure, pressure_alt, temperature;
217
    float barometerOffsetkPa;
218 219

    float man_roll, man_pitch, man_yaw;
220 221 222
    QString airframeName;
    enum AIRFRAME airframeID;
    bool xPlaneConnected;
Lorenz Meier's avatar
Lorenz Meier committed
223
    unsigned int xPlaneVersion;
224
    quint64 simUpdateLast;
225
    quint64 simUpdateFirst;
226
    quint64 simUpdateLastText;
227
    quint64 simUpdateLastGroundTruth;
228
    float simUpdateHz;
229
    bool _sensorHilEnabled;
230
    bool _useHilActuatorControls;
231
    bool _should_exit;
232

233
    void setName(QString name);
234
    void sendDataRef(QString ref, float value);
235 236
};