HSIDisplay.cc 22.3 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
/*=====================================================================

PIXHAWK Micro Air Vehicle Flying Robotics Toolkit

(c) 2009, 2010 PIXHAWK PROJECT  <http://pixhawk.ethz.ch>

This file is part of the PIXHAWK project

    PIXHAWK 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.

    PIXHAWK 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 PIXHAWK. If not, see <http://www.gnu.org/licenses/>.

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

/**
 * @file
 *   @brief Implementation of Horizontal Situation Indicator class
 *
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#include <QFile>
#include <QStringList>
34
#include <QPainter>
35 36 37
#include "UASManager.h"
#include "HSIDisplay.h"
#include "MG.h"
38
#include "QGC.h"
39 40 41

#include <QDebug>

42
HSIDisplay::HSIDisplay(QWidget *parent) :
lm's avatar
lm committed
43
        HDDisplay(NULL, parent),
pixhawk's avatar
pixhawk committed
44
        gpsSatellites(),
45 46 47 48 49 50 51 52
        satellitesUsed(0),
        attXSet(0),
        attYSet(0),
        attYawSet(0),
        altitudeSet(1.0),
        posXSet(0),
        posYSet(0),
        posZSet(0),
lm's avatar
lm committed
53 54 55 56 57
        attXSaturation(0.5f),
        attYSaturation(0.5f),
        attYawSaturation(0.5f),
        posXSaturation(0.05),
        posYSaturation(0.05),
58 59 60 61 62 63 64 65
        altitudeSaturation(1.0),
        lat(0),
        lon(0),
        alt(0),
        globalAvailable(0),
        x(0),
        y(0),
        z(0),
66 67 68 69 70 71 72
        vx(0),
        vy(0),
        vz(0),
        speed(0),
        localAvailable(0),
        roll(0),
        pitch(0),
pixhawk's avatar
pixhawk committed
73
        yaw(1.0f), // FIXME Should be 0
74 75 76 77 78 79 80 81
        bodyXSetCoordinate(0.0f),
        bodyYSetCoordinate(0.0f),
        bodyZSetCoordinate(0.0f),
        bodyYawSet(0.0f),
        uiXSetCoordinate(0.0f),
        uiYSetCoordinate(0.0f),
        uiZSetCoordinate(0.0f),
        uiYawSet(0.0f),
lm's avatar
lm committed
82 83 84 85
        metricWidth(2.0f),
        positionLock(false),
        attControlEnabled(false),
        xyControlEnabled(false),
pixhawk's avatar
pixhawk committed
86 87
        zControlEnabled(false),
        mavInitialized(false)
88
{
lm's avatar
lm committed
89
    connect(UASManager::instance(), SIGNAL(activeUASSet(UASInterface*)), this, SLOT(setActiveUAS(UASInterface*)));
lm's avatar
lm committed
90
    refreshTimer->setInterval(60);
91 92 93 94 95

    // FIXME
    float bottomMargin = 3.0f;
    xCenterPos = vwidth/2.0f;
    yCenterPos = vheight/2.0f - bottomMargin;
96 97
}

lm's avatar
lm committed
98 99 100 101 102
void HSIDisplay::paintEvent(QPaintEvent * event)
{
    Q_UNUSED(event);
    //paintGL();
    static quint64 interval = 0;
103
    //qDebug() << "INTERVAL:" << MG::TIME::getGroundTimeNow() - interval << __FILE__ << __LINE__;
lm's avatar
lm committed
104 105 106 107
    interval = MG::TIME::getGroundTimeNow();
    paintDisplay();
}

108 109
void HSIDisplay::paintDisplay()
{
110 111
    // Center location of the HSI gauge items

112
    float bottomMargin = 3.0f;
113 114 115

    // Size of the ring instrument
    const float margin = 0.1f;  // 10% margin of total width on each side
116 117
    float baseRadius = (vheight - vheight * 2.0f * margin) / 2.0f - bottomMargin / 2.0f;

118 119 120 121 122 123 124 125 126 127 128
    // Draw instruments
    // TESTING THIS SHOULD BE MOVED INTO A QGRAPHICSVIEW
    // Update scaling factor
    // adjust scaling to fit both horizontally and vertically
    scalingFactor = this->width()/vwidth;
    double scalingFactorH = this->height()/vheight;
    if (scalingFactorH < scalingFactor) scalingFactor = scalingFactorH;

    QPainter painter(this);
    painter.setRenderHint(QPainter::Antialiasing, true);
    painter.setRenderHint(QPainter::HighQualityAntialiasing, true);
129 130

    // Draw background
131
    painter.fillRect(QRect(0, 0, width(), height()), backgroundColor);
lm's avatar
lm committed
132

lm's avatar
lm committed
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
    // Draw status indicators
    QColor statusColor(255, 255, 255);
    QString lockStatus;
    QString xyContrStatus;
    QString zContrStatus;
    QString attContrStatus;

    QColor lockStatusColor;

    if (positionLock)
    {
        lockStatus = tr("LOCK");
        lockStatusColor = QColor(20, 255, 20);
    }
    else
    {
        lockStatus = tr("NO");
        lockStatusColor = QColor(255, 20, 20);
    }

153
    paintText(tr("POS"), QGC::colorCyan, 1.8f, 2.0f, 2.5f, &painter);
lm's avatar
lm committed
154 155 156 157 158
    painter.setBrush(lockStatusColor);
    painter.setPen(Qt::NoPen);
    painter.drawRect(QRect(refToScreenX(9.5f), refToScreenY(2.0f), refToScreenX(7.0f), refToScreenY(4.0f)));
    paintText(lockStatus, statusColor, 2.8f, 10.0f, 2.0f, &painter);

159 160
    // Draw base instrument
    // ----------------------
lm's avatar
lm committed
161
    painter.setBrush(Qt::NoBrush);
162
    const QColor ringColor = QColor(200, 250, 200);
lm's avatar
lm committed
163 164 165 166
    QPen pen;
    pen.setColor(ringColor);
    pen.setWidth(refLineWidthToPen(0.1f));
    painter.setPen(pen);
lm's avatar
lm committed
167 168 169
    const int ringCount = 2;
    for (int i = 0; i < ringCount; i++)
    {
170 171
        float radius = (vwidth - vwidth * 2.0f * margin) / (2.0f * i+1) / 2.0f - bottomMargin / 2.0f;
        drawCircle(xCenterPos, yCenterPos, radius, 0.1f, ringColor, &painter);
lm's avatar
lm committed
172 173
    }

lm's avatar
lm committed
174
    // Draw center indicator
175 176 177 178 179
    QPolygonF p(3);
    p.replace(0, QPointF(xCenterPos, yCenterPos-2.8484f));
    p.replace(1, QPointF(xCenterPos-2.0f, yCenterPos+2.0f));
    p.replace(2, QPointF(xCenterPos+2.0f, yCenterPos+2.0f));
    drawPolygon(p, &painter);
180 181

    // ----------------------
182

183 184 185
    // Draw satellites
    drawGPS(painter);

186 187 188
    // Draw state indicator

    // Draw position
189
    QColor positionColor(20, 20, 200);
190
    drawPositionDirection(xCenterPos, yCenterPos, baseRadius, positionColor, &painter);
191 192 193

    // Draw attitude
    QColor attitudeColor(200, 20, 20);
194
    drawAttitudeDirection(xCenterPos, yCenterPos, baseRadius, attitudeColor, &painter);
195 196


197
    // Draw position setpoints in body coordinates
lm's avatar
lm committed
198

199 200 201 202 203 204 205 206 207
    if (uiXSetCoordinate != 0 || uiYSetCoordinate != 0)
    {
        QColor spColor(150, 150, 150);
        drawSetpointXY(uiXSetCoordinate, uiYSetCoordinate, uiYawSet, spColor, painter);
    }

    if (bodyXSetCoordinate != 0 || bodyYSetCoordinate != 0)
    {
        // Draw setpoint
208
        drawSetpointXY(bodyXSetCoordinate, bodyYSetCoordinate, bodyYawSet, QGC::colorCyan, painter);
209 210
        // Draw travel direction line
        QPointF m(bodyXSetCoordinate, bodyYSetCoordinate);
lm's avatar
lm committed
211 212 213
        // Transform from body to world coordinates
        m = metricWorldToBody(m);
        // Scale from metric body to screen reference units
214
        QPointF s = metricBodyToRef(m);
215
        drawLine(s.x(), s.y(), xCenterPos, yCenterPos, 1.5f, QGC::colorCyan, &painter);
216 217 218
    }

    // Labels on outer part and bottom
219

220
    //if (localAvailable > 0)
221
    {
222
        // Position
223 224
        QString str;
        str.sprintf("%05.2f %05.2f %05.2f m", x, y, z);
225 226 227 228
        paintText(str, ringColor, 3.0f, xCenterPos + baseRadius - 30.75f, vheight - 5.0f, &painter);

        // Speed
        str.sprintf("%05.2f m/s", speed);
lm's avatar
lm committed
229
        paintText(str, ringColor, 3.0f, 10.0f, vheight - 5.0f, &painter);
230 231
    }

232 233 234
    // Draw orientation labels
    // Translate and rotate coordinate frame
    painter.translate((xCenterPos)*scalingFactor, (yCenterPos)*scalingFactor);
lm's avatar
lm committed
235
    painter.rotate((-yaw/(M_PI))*180.0f);
236 237 238 239 240
    paintText(tr("N"), ringColor, 3.5f, - 1.0f, - baseRadius - 5.5f, &painter);
    paintText(tr("S"), ringColor, 3.5f, - 1.0f, + baseRadius + 1.5f, &painter);
    paintText(tr("E"), ringColor, 3.5f, + baseRadius + 2.0f, - 1.75f, &painter);
    paintText(tr("W"), ringColor, 3.5f, - baseRadius - 5.5f, - 1.75f, &painter);

lm's avatar
lm committed
241 242 243 244 245
//    //  Just for testing
//    bodyXSetCoordinate = 0.95 * bodyXSetCoordinate + 0.05 * uiXSetCoordinate;
//    bodyYSetCoordinate = 0.95 * bodyYSetCoordinate + 0.05 * uiYSetCoordinate;
//    bodyZSetCoordinate = 0.95 * bodyZSetCoordinate + 0.05 * uiZSetCoordinate;
//    bodyYawSet = 0.95 * bodyYawSet + 0.05 * uiYawSet;
246
}
247

lm's avatar
lm committed
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271
void HSIDisplay::updatePositionLock(UASInterface* uas, bool lock)
{
    Q_UNUSED(uas);
    positionLock = lock;
}

void HSIDisplay::updateAttitudeControllerEnabled(UASInterface* uas, bool enabled)
{
    Q_UNUSED(uas);
    attControlEnabled = enabled;
}

void HSIDisplay::updatePositionXYControllerEnabled(UASInterface* uas, bool enabled)
{
    Q_UNUSED(uas);
    xyControlEnabled = enabled;
}

void HSIDisplay::updatePositionZControllerEnabled(UASInterface* uas, bool enabled)
{
    Q_UNUSED(uas);
    zControlEnabled = enabled;
}

pixhawk's avatar
pixhawk committed
272 273 274 275
QPointF HSIDisplay::metricWorldToBody(QPointF world)
{
    // First translate to body-centered coordinates
    // Rotate around -yaw
pixhawk's avatar
pixhawk committed
276
    QPointF result(cos(yaw) * (x - world.x()) + -sin(yaw) * (x - world.x()), sin(yaw) * (y - world.y()) + cos(yaw) * (y - world.y()));
pixhawk's avatar
pixhawk committed
277 278 279 280 281 282 283
    return result;
}

QPointF HSIDisplay::metricBodyToWorld(QPointF body)
{
    // First rotate into world coordinates
    // then translate to world position
lm's avatar
lm committed
284
    QPointF result((cos(yaw) * body.x()) + (sin(yaw) * body.x()) + x, (-sin(yaw) * body.y()) + (cos(yaw) * body.y()) + y);
pixhawk's avatar
pixhawk committed
285
    return result;
286
}
287

288 289
QPointF HSIDisplay::screenToMetricBody(QPointF ref)
{
lm's avatar
lm committed
290
    return QPointF(-((screenToRefY(ref.y()) - yCenterPos)/ vwidth) * metricWidth - x, ((screenToRefX(ref.x()) - xCenterPos) / vwidth) * metricWidth - y);
291 292 293 294
}

QPointF HSIDisplay::refToMetricBody(QPointF &ref)
{
lm's avatar
lm committed
295
    return QPointF(-((ref.y() - yCenterPos)/ vwidth) * metricWidth - x, ((ref.x() - xCenterPos) / vwidth) * metricWidth - y);
296 297 298 299 300
}

/**
 * @see refToScreenX()
 */
301
QPointF HSIDisplay::metricBodyToRef(QPointF &metric)
302 303 304 305
{
    return QPointF(((metric.y())/ metricWidth) * vwidth + xCenterPos, ((-metric.x()) / metricWidth) * vwidth + yCenterPos);
}

306 307 308 309 310 311 312 313
QPointF HSIDisplay::metricBodyToScreen(QPointF metric)
{
    QPointF ref = metricBodyToRef(metric);
    ref.setX(refToScreenX(ref.x()));
    ref.setY(refToScreenY(ref.y()));
    return ref;
}

314 315 316 317 318 319 320 321 322 323 324 325 326 327 328
void HSIDisplay::mouseDoubleClickEvent(QMouseEvent * event)
{
    static bool dragStarted;
    static float startX;

    if (event->MouseButtonDblClick)
    {
        //setBodySetpointCoordinateXY(-refToMetric(screenToRefY(event->y()) - yCenterPos), refToMetric(screenToRefX(event->x()) - xCenterPos));

        QPointF p = screenToMetricBody(event->posF());
        setBodySetpointCoordinateXY(p.x(), p.y());
        qDebug() << "Double click at x: " << screenToRefX(event->x()) - xCenterPos << "y:" << screenToRefY(event->y()) - yCenterPos;
    }
    else if (event->MouseButtonPress)
    {
329
        startX = event->globalX();
330 331
        if (event->button() == Qt::RightButton)
        {
332 333
            // Start tracking mouse move
            dragStarted = true;
334 335 336 337 338 339
        }
        else if (event->button() == Qt::LeftButton)
        {

        }
    }
340 341 342 343
    else if (event->MouseButtonRelease)
    {
        dragStarted = false;
    }
pixhawk's avatar
pixhawk committed
344 345
    else if (event->MouseMove)
    {
346
        if (dragStarted) uiYawSet += (startX - event->globalX()) / this->frameSize().width();
pixhawk's avatar
pixhawk committed
347
    }
348 349 350 351 352 353 354 355 356 357 358 359 360 361
}

/**
 *
 * @param uas the UAS/MAV to monitor/display with the HUD
 */
void HSIDisplay::setActiveUAS(UASInterface* uas)
{
    if (this->uas != NULL && this->uas != uas)
    {
        // Disconnect any previously connected active MAV
        //disconnect(uas, SIGNAL(valueChanged(UASInterface*,QString,double,quint64)), this, SLOT(updateValue(UASInterface*,QString,double,quint64)));
    }

362 363 364 365 366

    HDDisplay::setActiveUAS(uas);
    //qDebug() << "ATTEMPTING TO SET UAS";


lm's avatar
lm committed
367
    connect(uas, SIGNAL(gpsSatelliteStatusChanged(int,int,float,float,float,bool)), this, SLOT(updateSatellite(int,int,float,float,float,bool)));
368 369 370
    connect(uas, SIGNAL(localPositionChanged(UASInterface*,double,double,double,quint64)), this, SLOT(updateLocalPosition(UASInterface*,double,double,double,quint64)));
    connect(uas, SIGNAL(globalPositionChanged(UASInterface*,double,double,double,quint64)), this, SLOT(updateGlobalPosition(UASInterface*,double,double,double,quint64)));
    connect(uas, SIGNAL(attitudeThrustSetPointChanged(UASInterface*,double,double,double,double,quint64)), this, SLOT(updateAttitudeSetpoints(UASInterface*,double,double,double,double,quint64)));
lm's avatar
lm committed
371
    connect(uas, SIGNAL(positionSetPointsChanged(int,float,float,float,float,quint64)), this, SLOT(updatePositionSetpoints(int,float,float,float,float,quint64)));
372
    connect(uas, SIGNAL(speedChanged(UASInterface*,double,double,double,quint64)), this, SLOT(updateSpeed(UASInterface*,double,double,double,quint64)));
lm's avatar
lm committed
373
    connect(uas, SIGNAL(attitudeChanged(UASInterface*,double,double,double,quint64)), this, SLOT(updateAttitude(UASInterface*,double,double,double,quint64)));
lm's avatar
lm committed
374

375 376 377 378 379 380 381 382 383 384
    // Now connect the new UAS

    //if (this->uas != uas)
    // {
    //qDebug() << "UAS SET!" << "ID:" << uas->getUASID();
    // Setup communication
    //connect(uas, SIGNAL(valueChanged(UASInterface*,QString,double,quint64)), this, SLOT(updateValue(UASInterface*,QString,double,quint64)));
    //}
}

385 386 387 388 389 390 391 392 393 394 395 396 397 398
void HSIDisplay::updateSpeed(UASInterface* uas, double vx, double vy, double vz, quint64 time)
{
    Q_UNUSED(uas);
    Q_UNUSED(time);
    this->vx = vx;
    this->vy = vy;
    this->vz = vz;
    this->speed = sqrt(pow(vx, 2.0f) + pow(vy, 2.0f) + pow(vz, 2.0f));
}

void HSIDisplay::setBodySetpointCoordinateXY(double x, double y)
{
    // Set coordinates and send them out to MAV

lm's avatar
lm committed
399 400 401 402 403
    QPointF sp(x, y);
    sp = metricBodyToWorld(sp);
    uiXSetCoordinate = sp.x();
    uiYSetCoordinate = sp.y();

pixhawk's avatar
pixhawk committed
404 405 406
    qDebug() << "Attempting to set new setpoint at x: " << x << "metric y:" << y;

    if (uas && mavInitialized)
pixhawk's avatar
pixhawk committed
407 408 409 410
    {
        uas->setLocalPositionSetpoint(uiXSetCoordinate, uiYSetCoordinate, uiZSetCoordinate, uiYawSet);
        qDebug() << "Setting new setpoint at x: " << x << "metric y:" << y;
    }
411 412 413 414 415 416 417 418 419 420 421 422 423
}

void HSIDisplay::setBodySetpointCoordinateZ(double z)
{
    // Set coordinates and send them out to MAV
    uiZSetCoordinate = z;
}

void HSIDisplay::sendBodySetPointCoordinates()
{
    // Send the coordinates to the MAV
}

424
void HSIDisplay::updateAttitudeSetpoints(UASInterface* uas, double rollDesired, double pitchDesired, double yawDesired, double thrustDesired, quint64 usec)
425
{
426 427 428 429 430 431
    Q_UNUSED(uas);
    Q_UNUSED(usec);
    attXSet = pitchDesired;
    attYSet = rollDesired;
    attYawSet = yawDesired;
    altitudeSet = thrustDesired;
432 433
}

lm's avatar
lm committed
434 435 436 437 438 439 440 441 442
void HSIDisplay::updateAttitude(UASInterface* uas, double roll, double pitch, double yaw, quint64 time)
{
    Q_UNUSED(uas);
    Q_UNUSED(time);
    this->roll = roll;
    this->pitch = pitch;
    this->yaw = yaw;
}

lm's avatar
lm committed
443
void HSIDisplay::updatePositionSetpoints(int uasid, float xDesired, float yDesired, float zDesired, float yawDesired, quint64 usec)
444
{
445 446
    Q_UNUSED(usec);
    Q_UNUSED(uasid);
447 448 449 450
    bodyXSetCoordinate = xDesired;
    bodyYSetCoordinate = yDesired;
    bodyZSetCoordinate = zDesired;
    bodyYawSet = yawDesired;
pixhawk's avatar
pixhawk committed
451 452 453
    mavInitialized = true;

    qDebug() << "Received setpoint at x: " << x << "metric y:" << y;
454 455 456 457
//    posXSet = xDesired;
//    posYSet = yDesired;
//    posZSet = zDesired;
//    posYawSet = yawDesired;
458 459 460 461
}

void HSIDisplay::updateLocalPosition(UASInterface*, double x, double y, double z, quint64 usec)
{
462 463 464 465
    this->x = x;
    this->y = y;
    this->z = z;
    localAvailable = usec;
466 467 468 469
}

void HSIDisplay::updateGlobalPosition(UASInterface*, double lat, double lon, double alt, quint64 usec)
{
470 471 472 473
    this->lat = lat;
    this->lon = lon;
    this->alt = alt;
    globalAvailable = usec;
474 475
}

lm's avatar
lm committed
476
void HSIDisplay::updateSatellite(int uasid, int satid, float elevation, float azimuth, float snr, bool used)
lm's avatar
lm committed
477 478
{
    Q_UNUSED(uasid);
lm's avatar
lm committed
479
    //qDebug() << "UPDATED SATELLITE";
lm's avatar
lm committed
480
    // If slot is empty, insert object
lm's avatar
lm committed
481
    if (gpsSatellites.contains(satid))
lm's avatar
lm committed
482
    {
lm's avatar
lm committed
483
        gpsSatellites.value(satid)->update(satid, elevation, azimuth, snr, used);
lm's avatar
lm committed
484 485 486
    }
    else
    {
lm's avatar
lm committed
487
        gpsSatellites.insert(satid, new GPSSatellite(satid, elevation, azimuth, snr, used));
lm's avatar
lm committed
488 489 490 491 492
    }
}

QColor HSIDisplay::getColorForSNR(float snr)
{
lm's avatar
lm committed
493
    QColor color;
lm's avatar
lm committed
494 495 496 497 498
    if (snr > 0 && snr < 30)
    {
        color = QColor(250, 10, 10);
    }
    else if (snr >= 30 && snr < 35)
lm's avatar
lm committed
499
    {
lm's avatar
lm committed
500
        color = QColor(230, 230, 10);
lm's avatar
lm committed
501
    }
lm's avatar
lm committed
502
    else if (snr >= 35 && snr < 40)
lm's avatar
lm committed
503
    {
lm's avatar
lm committed
504
        color = QColor(90, 200, 90);
lm's avatar
lm committed
505
    }
lm's avatar
lm committed
506
    else if (snr >= 40)
lm's avatar
lm committed
507
    {
lm's avatar
lm committed
508
        color = QColor(20, 200, 20);
lm's avatar
lm committed
509 510 511 512 513 514
    }
    else
    {
        color = QColor(180, 180, 180);
    }
    return color;
lm's avatar
lm committed
515 516
}

517
void HSIDisplay::drawSetpointXY(float x, float y, float yaw, const QColor &color, QPainter &painter)
518
{
519 520 521 522 523 524 525
    float radius = vwidth / 20.0f;
    QPen pen(color);
    pen.setWidthF(refLineWidthToPen(0.4f));
    pen.setColor(color);
    painter.setPen(pen);
    painter.setBrush(Qt::NoBrush);
    QPointF in(x, y);
lm's avatar
lm committed
526 527 528
    // Transform from body to world coordinates
    in = metricWorldToBody(in);
    // Scale from metric to screen reference coordinates
529
    QPointF p = metricBodyToRef(in);
530 531 532 533 534 535 536 537 538 539 540 541 542
    drawCircle(p.x(), p.y(), radius, 0.4f, color, &painter);
    radius *= 0.8;
    drawLine(p.x(), p.y(), p.x()+sin(yaw) * radius, p.y()-cos(yaw) * radius, refLineWidthToPen(0.4f), color, &painter);
    painter.setBrush(color);
    drawCircle(p.x(), p.y(), radius * 0.1f, 0.1f, color, &painter);
}

void HSIDisplay::drawSafetyArea(const QPointF &topLeft, const QPointF &bottomRight, const QColor &color, QPainter &painter)
{
    QPen pen(color);
    pen.setWidthF(refLineWidthToPen(0.1f));
    pen.setColor(color);
    painter.setPen(pen);
543
    painter.drawRect(QRectF(metricBodyToScreen(metricWorldToBody(topLeft)), metricBodyToScreen(metricWorldToBody(bottomRight))));
544
}
lm's avatar
lm committed
545

546 547 548 549
void HSIDisplay::drawGPS(QPainter &painter)
{
    float xCenter = xCenterPos;
    float yCenter = xCenterPos;
lm's avatar
lm committed
550 551
    // Max satellite circle radius

lm's avatar
lm committed
552 553 554
    const float margin = 0.15f;  // 20% margin of total width on each side
    float radius = (vwidth - vwidth * 2.0f * margin) / 2.0f;
    quint64 currTime = MG::TIME::getGroundTimeNowUsecs();
lm's avatar
lm committed
555

pixhawk's avatar
pixhawk committed
556 557 558 559 560
    // Draw satellite labels
    //    QString label;
    //    label.sprintf("%05.1f", value);
    //    paintText(label, color, 4.5f, xRef-7.5f, yRef-2.0f, painter);

lm's avatar
lm committed
561 562
    QMapIterator<int, GPSSatellite*> i(gpsSatellites);
    while (i.hasNext())
lm's avatar
lm committed
563
    {
lm's avatar
lm committed
564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582
        i.next();
        GPSSatellite* sat = i.value();

        // Check if update is not older than 5 seconds, else delete satellite
        if (sat->lastUpdate + 1000000 < currTime)
        {
            // Delete and go to next satellite
            gpsSatellites.remove(i.key());
            if (i.hasNext())
            {
                i.next();
                sat = i.value();
            }
            else
            {
                continue;
            }
        }

lm's avatar
lm committed
583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600
        if (sat)
        {
            // Draw satellite
            QBrush brush;
            QColor color = getColorForSNR(sat->snr);
            brush.setColor(color);
            if (sat->used)
            {
                brush.setStyle(Qt::SolidPattern);
            }
            else
            {
                brush.setStyle(Qt::NoBrush);
            }
            painter.setPen(Qt::SolidLine);
            painter.setPen(color);
            painter.setBrush(brush);

lm's avatar
lm committed
601 602
            float xPos = xCenter + (sin(((sat->azimuth/255.0f)*360.0f)/180.0f * M_PI) * cos(sat->elevation/180.0f * M_PI)) * radius;
            float yPos = yCenter - (cos(((sat->azimuth/255.0f)*360.0f)/180.0f * M_PI) * cos(sat->elevation/180.0f * M_PI)) * radius;
lm's avatar
lm committed
603

604
            // Draw circle for satellite, filled for used satellites
lm's avatar
lm committed
605
            drawCircle(xPos, yPos, vwidth*0.02f, 1.0f, color, &painter);
606
            // Draw satellite PRN
lm's avatar
lm committed
607
            paintText(QString::number(sat->id), QColor(255, 255, 255), 2.9f, xPos+1.7f, yPos+2.0f, &painter);
lm's avatar
lm committed
608 609
        }
    }
610 611
}

612
void HSIDisplay::drawObjects(QPainter &painter)
613
{
614
    Q_UNUSED(painter);
615 616
}

617
void HSIDisplay::drawPositionDirection(float xRef, float yRef, float radius, const QColor& color, QPainter* painter)
618
{
619 620 621
    // Draw the needle
    const float maxWidth = radius / 10.0f;
    const float minWidth = maxWidth * 0.3f;
622

lm's avatar
lm committed
623 624
    float angle = atan2(posXSet, -posYSet);
    angle -= M_PI/2.0f;
625 626

    QPolygonF p(6);
627

lm's avatar
lm committed
628 629 630 631 632
    //radius *= ((posXSaturation + posYSaturation) - sqrt(pow(posXSet, 2), pow(posYSet, 2))) / (2*posXSaturation);

    radius *= sqrt(pow(posXSet, 2) + pow(posYSet, 2)) / sqrt(posXSaturation + posYSaturation);

    p.replace(0, QPointF(xRef-maxWidth/2.0f, yRef-radius * 0.4f));
633 634
    p.replace(1, QPointF(xRef-minWidth/2.0f, yRef-radius * 0.9f));
    p.replace(2, QPointF(xRef+minWidth/2.0f, yRef-radius * 0.9f));
lm's avatar
lm committed
635 636 637
    p.replace(3, QPointF(xRef+maxWidth/2.0f, yRef-radius * 0.4f));
    p.replace(4, QPointF(xRef,               yRef-radius * 0.36f));
    p.replace(5, QPointF(xRef-maxWidth/2.0f, yRef-radius * 0.4f));
638 639 640 641 642 643 644 645 646 647

    rotatePolygonClockWiseRad(p, angle, QPointF(xRef, yRef));

    QBrush indexBrush;
    indexBrush.setColor(color);
    indexBrush.setStyle(Qt::SolidPattern);
    painter->setPen(Qt::SolidLine);
    painter->setPen(color);
    painter->setBrush(indexBrush);
    drawPolygon(p, painter);
lm's avatar
lm committed
648

649
    //qDebug() << "DRAWING POS SETPOINT X:" << posXSet << "Y:" << posYSet << angle;
650 651
}

652
void HSIDisplay::drawAttitudeDirection(float xRef, float yRef, float radius, const QColor& color, QPainter* painter)
653
{
654 655 656 657
    // Draw the needle
    const float maxWidth = radius / 10.0f;
    const float minWidth = maxWidth * 0.3f;

lm's avatar
lm committed
658 659 660 661
    float angle = atan2(attXSet, attYSet);
    angle -= M_PI/2.0f;

    radius *= sqrt(pow(attXSet, 2) + pow(attYSet, 2)) / sqrt(attXSaturation + attYSaturation);
662

663 664
    QPolygonF p(6);

lm's avatar
lm committed
665
    p.replace(0, QPointF(xRef-maxWidth/2.0f, yRef-radius * 0.4f));
666 667
    p.replace(1, QPointF(xRef-minWidth/2.0f, yRef-radius * 0.9f));
    p.replace(2, QPointF(xRef+minWidth/2.0f, yRef-radius * 0.9f));
lm's avatar
lm committed
668 669 670
    p.replace(3, QPointF(xRef+maxWidth/2.0f, yRef-radius * 0.4f));
    p.replace(4, QPointF(xRef,               yRef-radius * 0.36f));
    p.replace(5, QPointF(xRef-maxWidth/2.0f, yRef-radius * 0.4f));
671

672
    rotatePolygonClockWiseRad(p, angle, QPointF(xRef, yRef));
673 674

    QBrush indexBrush;
675
    indexBrush.setColor(color);
676 677
    indexBrush.setStyle(Qt::SolidPattern);
    painter->setPen(Qt::SolidLine);
678
    painter->setPen(color);
679 680
    painter->setBrush(indexBrush);
    drawPolygon(p, painter);
681 682

    // TODO Draw Yaw indicator
lm's avatar
lm committed
683

684
    //qDebug() << "DRAWING ATT SETPOINT X:" << attXSet << "Y:" << attYSet << angle;
685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701
}

void HSIDisplay::drawAltitudeSetpoint(float xRef, float yRef, float radius, const QColor& color, QPainter* painter)
{
    // Draw the circle
    QPen circlePen(Qt::SolidLine);
    circlePen.setWidth(refLineWidthToPen(0.5f));
    circlePen.setColor(color);
    painter->setBrush(Qt::NoBrush);
    painter->setPen(circlePen);
    drawCircle(xRef, yRef, radius, 200.0f, color, painter);
    //drawCircle(xRef, yRef, radius, 200.0f, 170.0f, 1.0f, color, painter);

    //    // Draw the value
    //    QString label;
    //    label.sprintf("%05.1f", value);
    //    paintText(label, color, 4.5f, xRef-7.5f, yRef-2.0f, painter);
702
}
703 704 705 706 707 708 709 710 711 712

void HSIDisplay::updateJoystick(double roll, double pitch, double yaw, double thrust, int xHat, int yHat)
{

}

void HSIDisplay::pressKey(int key)
{

}