QGCFlightGearLink.h 4.87 KB
Newer Older
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
/*=====================================================================

QGroundControl Open Source Ground Control Station

(c) 2009 - 2011 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>

This file is part of the QGROUNDCONTROL project

    QGROUNDCONTROL is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    QGROUNDCONTROL is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.

======================================================================*/

/**
 * @file
 *   @brief UDP connection (server) for unmanned vehicles
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#ifndef QGCFLIGHTGEARLINK_H
#define QGCFLIGHTGEARLINK_H

#include <QString>
#include <QList>
#include <QMap>
#include <QMutex>
#include <QUdpSocket>
lm's avatar
lm committed
39
#include <QTimer>
lm's avatar
lm committed
40
#include <QProcess>
41
#include <LinkInterface.h>
42
#include "QGCConfig.h"
lm's avatar
lm committed
43
#include "UASInterface.h"
44
#include "QGCHilLink.h"
45
#include <QGCHilFlightGearConfiguration.h>
46

47
class QGCFlightGearLink : public QGCHilLink
48 49 50 51
{
    Q_OBJECT

public:
52
    QGCFlightGearLink(UASInterface* mav, QString startupArguments, QString remoteHost=QString("127.0.0.1:49000"), QHostAddress host = QHostAddress::Any, quint16 port = 49005);
53 54 55 56 57 58 59 60 61 62 63 64 65
    ~QGCFlightGearLink();

    bool isConnected();
    qint64 bytesAvailable();
    int getPort() const {
        return port;
    }

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

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

Lorenz Meier's avatar
Lorenz Meier committed
72 73 74 75 76 77 78 79 80 81
    QString getVersion()
    {
        return QString("FlightGear %1").arg(flightGearVersion);
    }

    int getAirFrameIndex()
    {
        return -1;
    }

82 83 84 85
    bool sensorHilEnabled() {
        return _sensorHilEnabled;
    }

Thomas Gubler's avatar
Thomas Gubler committed
86 87 88
    void sensorHilEnabled(bool sensorHilEnabled) {
        _sensorHilEnabled = sensorHilEnabled;
    }
Don Gagne's avatar
Don Gagne committed
89 90
    
    static bool parseUIArguments(QString uiArgs, QStringList& argList);
Thomas Gubler's avatar
Thomas Gubler committed
91

92
    void run();
Don Gagne's avatar
Don Gagne committed
93 94 95
    
signals:
    void showCriticalMessageFromThread(const QString& title, const QString& message);
96 97

public slots:
lm's avatar
lm committed
98
//    void setAddress(QString address);
99 100
    void setPort(int port);
    /** @brief Add a new host to broadcast messages to */
lm's avatar
lm committed
101
    void setRemoteHost(const QString& host);
lm's avatar
lm committed
102
    /** @brief Send new control states to the simulation */
103 104
    void updateControls(quint64 time, float rollAilerons, float pitchElevator, float yawRudder, float throttle, quint8 systemMode, quint8 navMode);
    void updateActuators(quint64 time, float act1, float act2, float act3, float act4, float act5, float act6, float act7, float act8);
Lorenz Meier's avatar
Lorenz Meier committed
105 106 107 108 109
    /** @brief Set the simulator version as text string */
    void setVersion(const QString& version)
    {
        Q_UNUSED(version);
    }
110

Lorenz Meier's avatar
Lorenz Meier committed
111 112 113 114 115
    void selectAirframe(const QString& airframe)
    {
        Q_UNUSED(airframe);
    }

116 117 118 119 120 121
    void enableSensorHIL(bool enable) {
        if (enable != _sensorHilEnabled)
            _sensorHilEnabled = enable;
            emit sensorHilChanged(enable);
    }

122 123 124 125 126 127 128 129
    void readBytes();
    /**
     * @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
     **/
    void writeBytes(const char* data, qint64 length);
lm's avatar
lm committed
130 131
    bool connectSimulation();
    bool disconnectSimulation();
132

133
    void setStartupArguments(QString startupArguments);
134
    void setBarometerOffset(float barometerOffsetkPa);
Don Gagne's avatar
Don Gagne committed
135
    void processError(QProcess::ProcessError err);
136

137 138 139
protected:
    QString name;
    QHostAddress host;
lm's avatar
lm committed
140 141
    QHostAddress currentHost;
    quint16 currentPort;
142 143 144 145
    quint16 port;
    int id;
    bool connectState;

lm's avatar
lm committed
146
    UASInterface* mav;
Lorenz Meier's avatar
Lorenz Meier committed
147
    unsigned int flightGearVersion;
148
    QString startupArguments;
149
    bool _sensorHilEnabled;
150
    float barometerOffsetkPa;
151 152

    void setName(QString name);
Don Gagne's avatar
Don Gagne committed
153
    
Don Gagne's avatar
Don Gagne committed
154 155 156 157
private slots:
    void _printFgfsOutput(void);
    void _printFgfsError(void);
    
Don Gagne's avatar
Don Gagne committed
158
private:
Don Gagne's avatar
Don Gagne committed
159 160
    static bool _findUIArgument(const QStringList& uiArgList, const QString& argLabel, QString& argValue);

Don Gagne's avatar
Don Gagne committed
161 162 163
    QString     _fgProcessName;             ///< FlightGear process to start
    QString     _fgProcessWorkingDirPath;   ///< Working directory to start FG process in, empty for none
    QStringList _fgArgList;                 ///< Arguments passed to FlightGear process
Don Gagne's avatar
Don Gagne committed
164 165 166

    QUdpSocket* _udpCommSocket;             ///< UDP communication sockect between FG and QGC
    QProcess*   _fgProcess;                 ///< FlightGear process
167 168 169
};

#endif // QGCFLIGHTGEARLINK_H