MAVLinkSimulationLink.h 3.76 KB
Newer Older
pixhawk's avatar
pixhawk committed
1 2
/*=====================================================================

3
QGroundControl Open Source Ground Control Station
pixhawk's avatar
pixhawk committed
4

5
(c) 2009, 2010 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
pixhawk's avatar
pixhawk committed
6

7
This file is part of the QGROUNDCONTROL project
pixhawk's avatar
pixhawk committed
8

9
    QGROUNDCONTROL is free software: you can redistribute it and/or modify
pixhawk's avatar
pixhawk committed
10 11 12 13
    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.

14
    QGROUNDCONTROL is distributed in the hope that it will be useful,
pixhawk's avatar
pixhawk committed
15 16 17 18 19
    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
20
    along with QGROUNDCONTROL. If not, see <http://www.gnu.org/licenses/>.
pixhawk's avatar
pixhawk committed
21 22 23 24 25

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

/**
 * @file
26
 *   @brief Definition of MAVLinkSimulationLink
pixhawk's avatar
pixhawk committed
27 28 29 30 31 32 33 34 35 36 37 38 39
 *
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#ifndef MAVLINKSIMULATIONLINK_H
#define MAVLINKSIMULATIONLINK_H

#include <QFile>
#include <QTimer>
#include <QTextStream>
#include <QQueue>
#include <QMutex>
lm's avatar
lm committed
40
#include <QMap>
41
#include <qmath.h>
pixhawk's avatar
pixhawk committed
42
#include <inttypes.h>
pixhawk's avatar
pixhawk committed
43
#include "QGCMAVLink.h"
lm's avatar
lm committed
44

pixhawk's avatar
pixhawk committed
45
#include "LinkInterface.h"
pixhawk's avatar
pixhawk committed
46 47 48

class MAVLinkSimulationLink : public LinkInterface
{
pixhawk's avatar
pixhawk committed
49
    Q_OBJECT
pixhawk's avatar
pixhawk committed
50
public:
51
    MAVLinkSimulationLink(QString readFile="", QString writeFile="", int rate=5);
pixhawk's avatar
pixhawk committed
52
    ~MAVLinkSimulationLink();
53
    bool isConnected() const;
pixhawk's avatar
pixhawk committed
54 55 56
    qint64 bytesAvailable();

    void run();
57
    void requestReset() { }
pixhawk's avatar
pixhawk committed
58 59 60
    bool connect();
    bool disconnect();

61 62 63 64
    // Extensive statistics for scientific purposes
    qint64 getConnectionSpeed() const;
    qint64 getCurrentInDataRate() const;
    qint64 getCurrentOutDataRate() const;
65 66 67 68 69 70 71 72 73 74

    QString getName() const;
    int getId() const;
    int getBaudRate() const;
    int getBaudRateType() const;
    int getFlowType() const;
    int getParityType() const;
    int getDataBitsType() const;
    int getStopBitsType() const;

pixhawk's avatar
pixhawk committed
75 76
public slots:
    void writeBytes(const char* data, qint64 size);
77
    void readBytes();
78
    /** @brief Mainloop simulating the mainloop of the MAV */
pixhawk's avatar
pixhawk committed
79
    virtual void mainloop();
80
    bool connectLink(bool connect);
81
    void connectLink();
82
    void sendMAVLinkMessage(const mavlink_message_t* msg);
pixhawk's avatar
pixhawk committed
83 84 85 86 87 88


protected:

    // UAS properties
    float roll, pitch, yaw;
89 90
    double x, y, z;
    double spX, spY, spZ, spYaw;
pixhawk's avatar
pixhawk committed
91 92 93 94 95
    int battery;

    QTimer* timer;
    /** File which contains the input data (simulated robot messages) **/
    QFile* simulationFile;
96
    QFile* mavlinkLogFile;
pixhawk's avatar
pixhawk committed
97 98 99
    QString simulationHeader;
    /** File where the commands sent by the groundstation are stored **/
    QFile* receiveFile;
100
    QTextStream textStream;
pixhawk's avatar
pixhawk committed
101 102 103 104 105 106 107 108
    QTextStream* fileStream;
    QTextStream* outStream;
    /** Buffer which can be read from connected protocols through readBytes(). **/
    QMutex readyBufferMutex;
    bool _isConnected;
    quint64 rate;
    int maxTimeNoise;
    quint64 lastSent;
109 110 111 112
    static const int streamlength = 4096;
    unsigned int streampointer;
    //const int testoffset = 0;
    uint8_t stream[streamlength];
pixhawk's avatar
pixhawk committed
113 114 115 116 117 118

    int readyBytes;
    QQueue<uint8_t> readyBuffer;

    int id;
    QString name;
pixhawk's avatar
pixhawk committed
119
    qint64 timeOffset;
pixhawk's avatar
pixhawk committed
120
    mavlink_sys_status_t status;
121
    mavlink_heartbeat_t system;
lm's avatar
lm committed
122
    QMap<QString, float> onboardParams;
pixhawk's avatar
pixhawk committed
123

pixhawk's avatar
pixhawk committed
124 125
    void enqueue(uint8_t* stream, uint8_t* index, mavlink_message_t* msg);

126
    static const uint8_t systemId = 220;
127
    static const uint8_t componentId = 200;
128 129
    static const uint16_t version = 1000;

130
signals:
pixhawk's avatar
pixhawk committed
131
    void valueChanged(int uasId, QString curve, double value, quint64 usec);
132
    void messageReceived(const mavlink_message_t& message);
pixhawk's avatar
pixhawk committed
133 134 135 136

};

#endif // MAVLINKSIMULATIONLINK_H