QGCMapPolygon.cc 15.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
/****************************************************************************
 *
 *   (c) 2009-2016 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
 *
 * QGroundControl is licensed according to the terms in the file
 * COPYING.md in the root of the source code directory.
 *
 ****************************************************************************/

#include "QGCMapPolygon.h"
#include "QGCGeo.h"
12
#include "JsonHelper.h"
13
#include "QGCQGeoCoordinate.h"
DonLakeFlyer's avatar
DonLakeFlyer committed
14
#include "QGCApplication.h"
15
#include "ShapeFileHelper.h"
16 17 18

#include <QGeoRectangle>
#include <QDebug>
19
#include <QJsonArray>
20
#include <QLineF>
DonLakeFlyer's avatar
DonLakeFlyer committed
21 22
#include <QFile>
#include <QDomDocument>
23

24
const char* QGCMapPolygon::jsonPolygonKey = "polygon";
25

26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
QGCMapPolygon::QGCMapPolygon(QObject* parent)
    : QObject               (parent)
    , _dirty                (false)
    , _centerDrag           (false)
    , _ignoreCenterUpdates  (false)
    , _interactive          (false)
{
    _init();
}

QGCMapPolygon::QGCMapPolygon(const QGCMapPolygon& other, QObject* parent)
    : QObject               (parent)
    , _dirty                (false)
    , _centerDrag           (false)
    , _ignoreCenterUpdates  (false)
    , _interactive          (false)
{
    *this = other;

    _init();
}

void QGCMapPolygon::_init(void)
49
{
50 51
    connect(&_polygonModel, &QmlObjectListModel::dirtyChanged, this, &QGCMapPolygon::_polygonModelDirtyChanged);
    connect(&_polygonModel, &QmlObjectListModel::countChanged, this, &QGCMapPolygon::_polygonModelCountChanged);
52
    connect(this, &QGCMapPolygon::pathChanged, this, &QGCMapPolygon::_updateCenter);
53 54
}

55 56 57 58 59
const QGCMapPolygon& QGCMapPolygon::operator=(const QGCMapPolygon& other)
{
    clear();

    QVariantList vertices = other.path();
DonLakeFlyer's avatar
DonLakeFlyer committed
60
    QList<QGeoCoordinate> rgCoord;
61
    for (const QVariant& vertexVar: vertices) {
DonLakeFlyer's avatar
DonLakeFlyer committed
62
        rgCoord.append(vertexVar.value<QGeoCoordinate>());
63
    }
DonLakeFlyer's avatar
DonLakeFlyer committed
64
    appendVertices(rgCoord);
65 66 67 68 69 70

    setDirty(true);

    return *this;
}

71 72 73 74 75 76 77 78 79 80 81 82 83 84
void QGCMapPolygon::clear(void)
{
    // Bug workaround, see below
    while (_polygonPath.count() > 1) {
        _polygonPath.takeLast();
    }
    emit pathChanged();

    // Although this code should remove the polygon from the map it doesn't. There appears
    // to be a bug in QGCMapPolygon which causes it to not be redrawn if the list is empty. So
    // we work around it by using the code above to remove all but the last point which in turn
    // will cause the polygon to go away.
    _polygonPath.clear();

85 86
    _polygonModel.clearAndDeleteContents();

87 88
    emit cleared();

89 90 91
    setDirty(true);
}

92
void QGCMapPolygon::adjustVertex(int vertexIndex, const QGeoCoordinate coordinate)
93 94
{
    _polygonPath[vertexIndex] = QVariant::fromValue(coordinate);
95
    _polygonModel.value<QGCQGeoCoordinate*>(vertexIndex)->setCoordinate(coordinate);
DonLakeFlyer's avatar
DonLakeFlyer committed
96
    if (!_centerDrag) {
97
        // When dragging center we don't signal path changed until all vertices are updated
DonLakeFlyer's avatar
DonLakeFlyer committed
98 99
        emit pathChanged();
    }
100 101 102 103 104 105 106
    setDirty(true);
}

void QGCMapPolygon::setDirty(bool dirty)
{
    if (_dirty != dirty) {
        _dirty = dirty;
107 108 109
        if (!dirty) {
            _polygonModel.setDirty(false);
        }
110 111 112 113
        emit dirtyChanged(dirty);
    }
}

114
QGeoCoordinate QGCMapPolygon::_coordFromPointF(const QPointF& point) const
115
{
116 117 118 119 120 121
    QGeoCoordinate coord;

    if (_polygonPath.count() > 0) {
        QGeoCoordinate tangentOrigin = _polygonPath[0].value<QGeoCoordinate>();
        convertNedToGeo(-point.y(), point.x(), 0, tangentOrigin, &coord);
    }
122

123 124
    return coord;
}
125

126 127 128
QPointF QGCMapPolygon::_pointFFromCoord(const QGeoCoordinate& coordinate) const
{
    if (_polygonPath.count() > 0) {
129
        double y, x, down;
130
        QGeoCoordinate tangentOrigin = _polygonPath[0].value<QGeoCoordinate>();
131

132 133
        convertGeoToNed(coordinate, tangentOrigin, &y, &x, &down);
        return QPointF(x, -y);
134 135
    }

136 137
    return QPointF();
}
138

139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
QPolygonF QGCMapPolygon::_toPolygonF(void) const
{
    QPolygonF polygon;

    if (_polygonPath.count() > 2) {
        for (int i=0; i<_polygonPath.count(); i++) {
            polygon.append(_pointFFromCoord(_polygonPath[i].value<QGeoCoordinate>()));
        }
    }

    return polygon;
}

bool QGCMapPolygon::containsCoordinate(const QGeoCoordinate& coordinate) const
{
    if (_polygonPath.count() > 2) {
        return _toPolygonF().containsPoint(_pointFFromCoord(coordinate), Qt::OddEvenFill);
    } else {
        return false;
    }
}

161 162 163
void QGCMapPolygon::setPath(const QList<QGeoCoordinate>& path)
{
    _polygonPath.clear();
164
    _polygonModel.clearAndDeleteContents();
165
    for(const QGeoCoordinate& coord: path) {
166
        _polygonPath.append(QVariant::fromValue(coord));
167
        _polygonModel.append(new QGCQGeoCoordinate(coord, this));
168
    }
169

170 171 172 173 174 175 176
    setDirty(true);
    emit pathChanged();
}

void QGCMapPolygon::setPath(const QVariantList& path)
{
    _polygonPath = path;
177 178 179

    _polygonModel.clearAndDeleteContents();
    for (int i=0; i<_polygonPath.count(); i++) {
180
        _polygonModel.append(new QGCQGeoCoordinate(_polygonPath[i].value<QGeoCoordinate>(), this));
181 182
    }

183 184 185
    setDirty(true);
    emit pathChanged();
}
186 187 188

void QGCMapPolygon::saveToJson(QJsonObject& json)
{
189
    QJsonValue jsonValue;
190

191
    JsonHelper::saveGeoCoordinateArray(_polygonPath, false /* writeAltitude*/, jsonValue);
192
    json.insert(jsonPolygonKey, jsonValue);
193 194 195 196 197 198 199 200 201
    setDirty(false);
}

bool QGCMapPolygon::loadFromJson(const QJsonObject& json, bool required, QString& errorString)
{
    errorString.clear();
    clear();

    if (required) {
202
        if (!JsonHelper::validateRequiredKeys(json, QStringList(jsonPolygonKey), errorString)) {
203 204
            return false;
        }
205
    } else if (!json.contains(jsonPolygonKey)) {
206 207 208
        return true;
    }

209
    if (!JsonHelper::loadGeoCoordinateArray(json[jsonPolygonKey], false /* altitudeRequired */, _polygonPath, errorString)) {
210 211
        return false;
    }
212 213

    for (int i=0; i<_polygonPath.count(); i++) {
214
        _polygonModel.append(new QGCQGeoCoordinate(_polygonPath[i].value<QGeoCoordinate>(), this));
215 216
    }

217
    setDirty(false);
218
    emit pathChanged();
219 220 221 222

    return true;
}

223 224 225 226
QList<QGeoCoordinate> QGCMapPolygon::coordinateList(void) const
{
    QList<QGeoCoordinate> coords;

227 228
    for (int i=0; i<_polygonPath.count(); i++) {
        coords.append(_polygonPath[i].value<QGeoCoordinate>());
229 230 231 232
    }

    return coords;
}
233 234 235 236

void QGCMapPolygon::splitPolygonSegment(int vertexIndex)
{
    int nextIndex = vertexIndex + 1;
237 238


239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261
    if (nextIndex > _polygonPath.length() - 1) {
        nextIndex = 0;
    }

    QGeoCoordinate firstVertex = _polygonPath[vertexIndex].value<QGeoCoordinate>();
    QGeoCoordinate nextVertex = _polygonPath[nextIndex].value<QGeoCoordinate>();

    double distance = firstVertex.distanceTo(nextVertex);
    double azimuth = firstVertex.azimuthTo(nextVertex);
    QGeoCoordinate newVertex = firstVertex.atDistanceAndAzimuth(distance / 2, azimuth);

    if (nextIndex == 0) {
        appendVertex(newVertex);
    } else {
        _polygonModel.insert(nextIndex, new QGCQGeoCoordinate(newVertex, this));
        _polygonPath.insert(nextIndex, QVariant::fromValue(newVertex));
        emit pathChanged();
    }
}

void QGCMapPolygon::appendVertex(const QGeoCoordinate& coordinate)
{
    _polygonPath.append(QVariant::fromValue(coordinate));
262
    _polygonModel.append(new QGCQGeoCoordinate(coordinate, this));
263 264 265
    emit pathChanged();
}

266 267 268 269
void QGCMapPolygon::appendVertices(const QList<QGeoCoordinate>& coordinates)
{
    QList<QObject*> objects;

270
    for (const QGeoCoordinate& coordinate: coordinates) {
271 272 273 274 275 276 277
        objects.append(new QGCQGeoCoordinate(coordinate, this));
        _polygonPath.append(QVariant::fromValue(coordinate));
    }
    _polygonModel.append(objects);
    emit pathChanged();
}

278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307
void QGCMapPolygon::_polygonModelDirtyChanged(bool dirty)
{
    if (dirty) {
        setDirty(true);
    }
}

void QGCMapPolygon::removeVertex(int vertexIndex)
{
    if (vertexIndex < 0 && vertexIndex > _polygonPath.length() - 1) {
        qWarning() << "Call to removePolygonCoordinate with bad vertexIndex:count" << vertexIndex << _polygonPath.length();
        return;
    }

    if (_polygonPath.length() <= 3) {
        // Don't allow the user to trash the polygon
        return;
    }

    QObject* coordObj = _polygonModel.removeAt(vertexIndex);
    coordObj->deleteLater();

    _polygonPath.removeAt(vertexIndex);
    emit pathChanged();
}

void QGCMapPolygon::_polygonModelCountChanged(int count)
{
    emit countChanged(count);
}
308 309 310 311 312 313

void QGCMapPolygon::_updateCenter(void)
{
    if (!_ignoreCenterUpdates) {
        QGeoCoordinate center;

314
        if (_polygonPath.count() > 2) {
Don Gagne's avatar
Don Gagne committed
315 316 317 318 319 320
            QPointF centroid(0, 0);
            QPolygonF polygonF = _toPolygonF();
            for (int i=0; i<polygonF.count(); i++) {
                centroid += polygonF[i];
            }
            center = _coordFromPointF(QPointF(centroid.x() / polygonF.count(), centroid.y() / polygonF.count()));
321
        }
Don Gagne's avatar
Don Gagne committed
322 323 324 325
        if (_center != center) {
            _center = center;
            emit centerChanged(center);
        }
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343
    }
}

void QGCMapPolygon::setCenter(QGeoCoordinate newCenter)
{
    if (newCenter != _center) {
        _ignoreCenterUpdates = true;

        // Adjust polygon vertices to new center
        double distance = _center.distanceTo(newCenter);
        double azimuth = _center.azimuthTo(newCenter);

        for (int i=0; i<count(); i++) {
            QGeoCoordinate oldVertex = _polygonPath[i].value<QGeoCoordinate>();
            QGeoCoordinate newVertex = oldVertex.atDistanceAndAzimuth(distance, azimuth);
            adjustVertex(i, newVertex);
        }

DonLakeFlyer's avatar
DonLakeFlyer committed
344
        if (_centerDrag) {
345
            // When center dragging, signals from adjustVertext are not sent. So we need to signal here when all adjusting is complete.
DonLakeFlyer's avatar
DonLakeFlyer committed
346 347 348
            emit pathChanged();
        }

349
        _ignoreCenterUpdates = false;
DonLakeFlyer's avatar
DonLakeFlyer committed
350 351 352 353 354 355 356 357 358 359 360

        _center = newCenter;
        emit centerChanged(newCenter);
    }
}

void QGCMapPolygon::setCenterDrag(bool centerDrag)
{
    if (centerDrag != _centerDrag) {
        _centerDrag = centerDrag;
        emit centerDragChanged(centerDrag);
361 362
    }
}
363 364 365 366 367 368 369 370

void QGCMapPolygon::setInteractive(bool interactive)
{
    if (_interactive != interactive) {
        _interactive = interactive;
        emit interactiveChanged(interactive);
    }
}
371

372 373 374 375 376 377 378 379 380 381
void print(const QGCMapPolygon &poly)
{
    QString message;
    print(poly, message);

    qWarning() << message;
}

void print(const QGCMapPolygon &poly, QString &outputString)
{
382
    outputString.append(QString("Coordinates:\r\n"));
383 384 385

    for (int i = 0; i < poly.count(); i++) {
        QGeoCoordinate coordinate = poly.vertexCoordinate(i);
386
        outputString.append(QString("%1\n").arg(coordinate.toString(QGeoCoordinate::Degrees)));
387 388
    }

389 390 391 392 393
    outputString.append(QString("Dirty: %1\n").arg(QVariant(poly._dirty).toString()));
    outputString.append(QString("Center: %1\n").arg(poly._center.toString(QGeoCoordinate::Degrees)));
    outputString.append(QString("Center Drag: %1\n").arg(QVariant(poly._centerDrag).toString()));
    outputString.append(QString("Ignore Center Updates: %1\n").arg(QVariant(poly._centerDrag).toString()));
    outputString.append(QString("Interactive: %1\n").arg(QVariant(poly._interactive).toString()));
394 395
}

396 397 398 399 400
QGeoCoordinate QGCMapPolygon::vertexCoordinate(int vertex) const
{
    if (vertex >= 0 && vertex < _polygonPath.count()) {
        return _polygonPath[vertex].value<QGeoCoordinate>();
    } else {
DonLakeFlyer's avatar
DonLakeFlyer committed
401
        qWarning() << "QGCMapPolygon::vertexCoordinate bad vertex requested:count" << vertex << _polygonPath.count();
402 403 404
        return QGeoCoordinate();
    }
}
405

406
QList<QPointF> QGCMapPolygon::nedPolygon(void) const
407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477
{
    QList<QPointF>  nedPolygon;

    if (count() > 0) {
        QGeoCoordinate  tangentOrigin = vertexCoordinate(0);

        for (int i=0; i<_polygonModel.count(); i++) {
            double y, x, down;
            QGeoCoordinate vertex = vertexCoordinate(i);
            if (i == 0) {
                // This avoids a nan calculation that comes out of convertGeoToNed
                x = y = 0;
            } else {
                convertGeoToNed(vertex, tangentOrigin, &y, &x, &down);
            }
            nedPolygon += QPointF(x, y);
        }
    }

    return nedPolygon;
}


void QGCMapPolygon::offset(double distance)
{
    QList<QGeoCoordinate> rgNewPolygon;

    // I'm sure there is some beautiful famous algorithm to do this, but here is a brute force method

    if (count() > 2) {
        // Convert the polygon to NED
        QList<QPointF> rgNedVertices = nedPolygon();

        // Walk the edges, offsetting by the specified distance
        QList<QLineF> rgOffsetEdges;
        for (int i=0; i<rgNedVertices.count(); i++) {
            int     lastIndex = i == rgNedVertices.count() - 1 ? 0 : i + 1;
            QLineF  offsetEdge;
            QLineF  originalEdge(rgNedVertices[i], rgNedVertices[lastIndex]);

            QLineF workerLine = originalEdge;
            workerLine.setLength(distance);
            workerLine.setAngle(workerLine.angle() - 90.0);
            offsetEdge.setP1(workerLine.p2());

            workerLine.setPoints(originalEdge.p2(), originalEdge.p1());
            workerLine.setLength(distance);
            workerLine.setAngle(workerLine.angle() + 90.0);
            offsetEdge.setP2(workerLine.p2());

            rgOffsetEdges.append(offsetEdge);
        }

        // Intersect the offset edges to generate new vertices
        QPointF         newVertex;
        QGeoCoordinate  tangentOrigin = vertexCoordinate(0);
        for (int i=0; i<rgOffsetEdges.count(); i++) {
            int prevIndex = i == 0 ? rgOffsetEdges.count() - 1 : i - 1;
            if (rgOffsetEdges[prevIndex].intersect(rgOffsetEdges[i], &newVertex) == QLineF::NoIntersection) {
                // FIXME: Better error handling?
                qWarning("Intersection failed");
                return;
            }
            QGeoCoordinate coord;
            convertNedToGeo(newVertex.y(), newVertex.x(), 0, tangentOrigin, &coord);
            rgNewPolygon.append(coord);
        }
    }

    // Update internals
    clear();
DonLakeFlyer's avatar
DonLakeFlyer committed
478
    appendVertices(rgNewPolygon);
479
}
DonLakeFlyer's avatar
DonLakeFlyer committed
480

481
bool QGCMapPolygon::loadKMLOrSHPFile(const QString& file)
DonLakeFlyer's avatar
DonLakeFlyer committed
482
{
483
    QString errorString;
DonLakeFlyer's avatar
DonLakeFlyer committed
484
    QList<QGeoCoordinate> rgCoords;
485
    if (!ShapeFileHelper::loadPolygonFromFile(file, rgCoords, errorString)) {
486 487
        qgcApp()->showMessage(errorString);
        return false;
DonLakeFlyer's avatar
DonLakeFlyer committed
488 489 490
    }

    clear();
491
    appendVertices(rgCoords);
DonLakeFlyer's avatar
DonLakeFlyer committed
492 493 494

    return true;
}
495 496 497 498 499

double QGCMapPolygon::area(void) const
{
    // https://www.mathopenref.com/coordpolygonarea2.html

DonLakeFlyer's avatar
DonLakeFlyer committed
500 501 502 503
    if (_polygonPath.count() < 3) {
        return 0;
    }

504 505 506 507 508 509 510 511 512 513 514
    double coveredArea = 0.0;
    QList<QPointF> nedVertices = nedPolygon();
    for (int i=0; i<nedVertices.count(); i++) {
        if (i != 0) {
            coveredArea += nedVertices[i - 1].x() * nedVertices[i].y() - nedVertices[i].x() * nedVertices[i -1].y();
        } else {
            coveredArea += nedVertices.last().x() * nedVertices[i].y() - nedVertices[i].x() * nedVertices.last().y();
        }
    }
    return 0.5 * fabs(coveredArea);
}
515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541

void QGCMapPolygon::verifyClockwiseWinding(void)
{
    if (_polygonPath.count() <= 2) {
        return;
    }

    double sum = 0;
    for (int i=0; i<_polygonPath.count(); i++) {
        QGeoCoordinate coord1 = _polygonPath[i].value<QGeoCoordinate>();
        QGeoCoordinate coord2 = (i == _polygonPath.count() - 1) ? _polygonPath[0].value<QGeoCoordinate>() : _polygonPath[i+1].value<QGeoCoordinate>();

        sum += (coord2.longitude() - coord1.longitude()) * (coord2.latitude() + coord1.latitude());
    }

    if (sum < 0.0) {
        // Winding is counter-clockwise and needs reversal

        QList<QGeoCoordinate> rgReversed;
        for (const QVariant& varCoord: _polygonPath) {
            rgReversed.prepend(varCoord.value<QGeoCoordinate>());
        }

        clear();
        appendVertices(rgReversed);
    }
}