CorridorScanComplexItem.cc 21.7 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
/****************************************************************************
 *
 *   (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 "CorridorScanComplexItem.h"
#include "JsonHelper.h"
#include "MissionController.h"
#include "QGCGeo.h"
#include "QGroundControlQmlGlobal.h"
#include "QGCQGeoCoordinate.h"
#include "SettingsManager.h"
#include "AppSettings.h"
#include "QGCQGeoCoordinate.h"

#include <QPolygonF>

QGC_LOGGING_CATEGORY(CorridorScanComplexItemLog, "CorridorScanComplexItemLog")

24 25
const char* CorridorScanComplexItem::settingsGroup =            "CorridorScan";
const char* CorridorScanComplexItem::corridorWidthName =        "CorridorWidth";
26
const char* CorridorScanComplexItem::_jsonEntryPointKey =       "EntryPoint";
27

28
const char* CorridorScanComplexItem::jsonComplexItemTypeValue = "CorridorScan";
29

30
CorridorScanComplexItem::CorridorScanComplexItem(Vehicle* vehicle, bool flyView, const QString& kmlFile, QObject* parent)
31
    : TransectStyleComplexItem  (vehicle, flyView, settingsGroup, parent)
32 33 34
    , _entryPoint               (0)
    , _metaDataMap              (FactMetaData::createMapFromJsonFile(QStringLiteral(":/json/CorridorScan.SettingsGroup.json"), this))
    , _corridorWidthFact        (settingsGroup, _metaDataMap[corridorWidthName])
35 36 37
{
    _editorQml = "qrc:/qml/CorridorScanEditor.qml";

DonLakeFlyer's avatar
DonLakeFlyer committed
38 39 40 41 42
    // We override the altitude to the mission default
    if (_cameraCalc.isManualCamera() || !_cameraCalc.valueSetIsDistance()->rawValue().toBool()) {
        _cameraCalc.distanceToSurface()->setRawValue(qgcApp()->toolbox()->settingsManager()->appSettings()->defaultMissionItemAltitude()->rawValue());
    }

43 44 45
    connect(&_corridorWidthFact,    &Fact::valueChanged,                                this, &CorridorScanComplexItem::_setDirty);
    connect(&_corridorPolyline,     &QGCMapPolyline::pathChanged,                       this, &CorridorScanComplexItem::_setDirty);

46 47
    connect(&_cameraCalc,           &CameraCalc::distanceToSurfaceRelativeChanged, this, &CorridorScanComplexItem::coordinateHasRelativeAltitudeChanged);
    connect(&_cameraCalc,           &CameraCalc::distanceToSurfaceRelativeChanged, this, &CorridorScanComplexItem::exitCoordinateHasRelativeAltitudeChanged);
48

49
    connect(&_corridorPolyline,     &QGCMapPolyline::dirtyChanged,  this, &CorridorScanComplexItem::_polylineDirtyChanged);
50

51 52
    connect(&_corridorPolyline,     &QGCMapPolyline::pathChanged,   this, &CorridorScanComplexItem::_rebuildCorridorPolygon);
    connect(&_corridorWidthFact,    &Fact::valueChanged,            this, &CorridorScanComplexItem::_rebuildCorridorPolygon);
53 54 55 56 57 58

    if (!kmlFile.isEmpty()) {
        _corridorPolyline.loadKMLFile(kmlFile);
        _corridorPolyline.setDirty(false);
    }
    setDirty(false);
59 60
}

61
void CorridorScanComplexItem::save(QJsonArray&  planItems)
62 63 64
{
    QJsonObject saveObject;

Don Gagne's avatar
Don Gagne committed
65
    TransectStyleComplexItem::_save(saveObject);
66 67

    saveObject[JsonHelper::jsonVersionKey] =                    2;
68 69
    saveObject[VisualMissionItem::jsonTypeKey] =                VisualMissionItem::jsonTypeComplexItemValue;
    saveObject[ComplexMissionItem::jsonComplexItemTypeKey] =    jsonComplexItemTypeValue;
70
    saveObject[corridorWidthName] =                             _corridorWidthFact.rawValue().toDouble();
71
    saveObject[_jsonEntryPointKey] =                            _entryPoint;
72 73 74

    _corridorPolyline.saveToJson(saveObject);

75
    planItems.append(saveObject);
76 77 78 79
}

bool CorridorScanComplexItem::load(const QJsonObject& complexObject, int sequenceNumber, QString& errorString)
{
80 81 82
    // We don't recalc while loading since all the information we need is specified in the file
    _ignoreRecalc = true;

83 84 85 86
    QList<JsonHelper::KeyValidateInfo> keyInfoList = {
        { JsonHelper::jsonVersionKey,                   QJsonValue::Double, true },
        { VisualMissionItem::jsonTypeKey,               QJsonValue::String, true },
        { ComplexMissionItem::jsonComplexItemTypeKey,   QJsonValue::String, true },
87
        { corridorWidthName,                            QJsonValue::Double, true },
88
        { _jsonEntryPointKey,                           QJsonValue::Double, true },
89 90 91
        { QGCMapPolyline::jsonPolylineKey,              QJsonValue::Array,  true },
    };
    if (!JsonHelper::validateKeys(complexObject, keyInfoList, errorString)) {
92
        _ignoreRecalc = false;
93 94 95
        return false;
    }

DonLakeFlyer's avatar
DonLakeFlyer committed
96
    if (!_corridorPolyline.loadFromJson(complexObject, true, errorString)) {
97
        _ignoreRecalc = false;
DonLakeFlyer's avatar
DonLakeFlyer committed
98 99
        return false;
    }
100 101 102 103 104

    QString itemType = complexObject[VisualMissionItem::jsonTypeKey].toString();
    QString complexType = complexObject[ComplexMissionItem::jsonComplexItemTypeKey].toString();
    if (itemType != VisualMissionItem::jsonTypeComplexItemValue || complexType != jsonComplexItemTypeValue) {
        errorString = tr("%1 does not support loading this complex mission item type: %2:%3").arg(qgcApp()->applicationName()).arg(itemType).arg(complexType);
105
        _ignoreRecalc = false;
106 107 108 109
        return false;
    }

    int version = complexObject[JsonHelper::jsonVersionKey].toInt();
110
    if (version != 2) {
111
        errorString = tr("%1 complex item version %2 not supported").arg(jsonComplexItemTypeValue).arg(version);
112
        _ignoreRecalc = false;
113 114 115 116 117
        return false;
    }

    setSequenceNumber(sequenceNumber);

Don Gagne's avatar
Don Gagne committed
118
    if (!_load(complexObject, false /* forPresets */, errorString)) {
119
        _ignoreRecalc = false;
120 121 122
        return false;
    }

123 124
    _corridorWidthFact.setRawValue      (complexObject[corridorWidthName].toDouble());

125
    _entryPoint = complexObject[_jsonEntryPointKey].toInt();
126

127 128
    _ignoreRecalc = false;

129 130 131 132 133 134
    _recalcComplexDistance();
    if (_cameraShots == 0) {
        // Shot count was possibly not available from plan file
        _recalcCameraShots();
    }

135 136 137 138 139 140 141 142 143 144 145 146 147 148 149
    return true;
}

bool CorridorScanComplexItem::specifiesCoordinate(void) const
{
    return _corridorPolyline.count() > 1;
}

int CorridorScanComplexItem::_transectCount(void) const
{
    double transectSpacing = _cameraCalc.adjustedFootprintSide()->rawValue().toDouble();
    double fullWidth = _corridorWidthFact.rawValue().toDouble();
    return fullWidth > 0.0 ? qCeil(fullWidth / transectSpacing) : 1;
}

150 151 152 153 154 155
void CorridorScanComplexItem::_appendLoadedMissionItems(QList<MissionItem*>& items, QObject* missionItemParent)
{
    qCDebug(CorridorScanComplexItemLog) << "_appendLoadedMissionItems";

    int seqNum = _sequenceNumber;

156
    for (const MissionItem* loadedMissionItem: _loadedMissionItems) {
157 158 159 160 161 162 163
        MissionItem* item = new MissionItem(*loadedMissionItem, missionItemParent);
        item->setSequenceNumber(seqNum++);
        items.append(item);
    }
}

void CorridorScanComplexItem::_buildAndAppendMissionItems(QList<MissionItem*>& items, QObject* missionItemParent)
164
{
165 166
    qCDebug(CorridorScanComplexItemLog) << "_buildAndAppendMissionItems";

167 168 169 170 171 172
    // Now build the mission items from the transect points

    MissionItem* item;
    int seqNum =                    _sequenceNumber;
    bool imagesEverywhere =         _cameraTriggerInTurnAroundFact.rawValue().toBool();
    bool addTriggerAtBeginning =    imagesEverywhere;
173
    bool firstOverallPoint =        true;
174 175 176

    MAV_FRAME mavFrame = followTerrain() || !_cameraCalc.distanceToSurfaceRelative() ? MAV_FRAME_GLOBAL : MAV_FRAME_GLOBAL_RELATIVE_ALT;

177
    //qDebug() << "_buildAndAppendMissionItems";
178
    for (const QList<TransectStyleComplexItem::CoordInfo_t>& transect: _transects) {
179
        bool transectEntry = true;
180 181

        //qDebug() << "start transect";
182
        for (const CoordInfo_t& transectCoordInfo: transect) {
183
            //qDebug() << transectCoordInfo.coordType;
184

185
            item = new MissionItem(seqNum++,
186 187 188 189 190 191 192 193 194 195 196
                                   MAV_CMD_NAV_WAYPOINT,
                                   mavFrame,
                                   0,                                          // No hold time
                                   0.0,                                        // No acceptance radius specified
                                   0.0,                                        // Pass through waypoint
                                   std::numeric_limits<double>::quiet_NaN(),   // Yaw unchanged
                                   transectCoordInfo.coord.latitude(),
                                   transectCoordInfo.coord.longitude(),
                                   transectCoordInfo.coord.altitude(),
                                   true,                                       // autoContinue
                                   false,                                      // isCurrentItem
197
                                   missionItemParent);
198 199
            items.append(item);

200
            if (triggerCamera() && firstOverallPoint && addTriggerAtBeginning) {
201
                // Start triggering
202
                addTriggerAtBeginning = false;
203 204 205 206 207 208 209 210 211 212 213 214
                item = new MissionItem(seqNum++,
                                       MAV_CMD_DO_SET_CAM_TRIGG_DIST,
                                       MAV_FRAME_MISSION,
                                       _cameraCalc.adjustedFootprintFrontal()->rawValue().toDouble(),   // trigger distance
                                       0,                                                               // shutter integration (ignore)
                                       1,                                                               // trigger immediately when starting
                                       0, 0, 0, 0,                                                      // param 4-7 unused
                                       true,                                                            // autoContinue
                                       false,                                                           // isCurrentItem
                                       missionItemParent);
                items.append(item);
            }
215 216
            firstOverallPoint = false;

217
            // Possibly add trigger start/stop to survey area entrance/exit
Don Gagne's avatar
Don Gagne committed
218 219
            if (triggerCamera() && transectCoordInfo.coordType == TransectStyleComplexItem::CoordTypeSurveyEdge) {
                if (transectEntry) {
220 221 222
                    // Start of transect, always start triggering. We do this even if we are taking images everywhere.
                    // This allows a restart of the mission in mid-air without losing images from the entire mission.
                    // At most you may lose part of a transect.
223 224 225 226 227 228 229 230 231 232 233
                    item = new MissionItem(seqNum++,
                                           MAV_CMD_DO_SET_CAM_TRIGG_DIST,
                                           MAV_FRAME_MISSION,
                                           _cameraCalc.adjustedFootprintFrontal()->rawValue().toDouble(),   // trigger distance
                                           0,                                                               // shutter integration (ignore)
                                           1,                                                               // trigger immediately when starting
                                           0, 0, 0, 0,                                                      // param 4-7 unused
                                           true,                                                            // autoContinue
                                           false,                                                           // isCurrentItem
                                           missionItemParent);
                    items.append(item);
234 235
                    transectEntry = false;
                } else if (!imagesEverywhere && !transectEntry){
236 237 238 239 240 241 242 243 244 245 246 247 248 249
                    // End of transect, stop triggering
                    item = new MissionItem(seqNum++,
                                           MAV_CMD_DO_SET_CAM_TRIGG_DIST,
                                           MAV_FRAME_MISSION,
                                           0,           // stop triggering
                                           0,           // shutter integration (ignore)
                                           0,           // trigger immediately when starting
                                           0, 0, 0, 0,  // param 4-7 unused
                                           true,        // autoContinue
                                           false,       // isCurrentItem
                                           missionItemParent);
                    items.append(item);
                }
            }
250
        }
251
    }
252 253

    if (imagesEverywhere) {
254
        // Stop triggering
255 256 257 258 259 260 261 262 263 264 265 266
        MissionItem* item = new MissionItem(seqNum++,
                                            MAV_CMD_DO_SET_CAM_TRIGG_DIST,
                                            MAV_FRAME_MISSION,
                                            0,           // stop triggering
                                            0,           // shutter integration (ignore)
                                            0,           // trigger immediately when starting
                                            0, 0, 0, 0,  // param 4-7 unused
                                            true,        // autoContinue
                                            false,       // isCurrentItem
                                            missionItemParent);
        items.append(item);
    }
267 268
}

269 270 271 272 273 274 275 276 277 278 279
void CorridorScanComplexItem::appendMissionItems(QList<MissionItem*>& items, QObject* missionItemParent)
{
    if (_loadedMissionItems.count()) {
        // We have mission items from the loaded plan, use those
        _appendLoadedMissionItems(items, missionItemParent);
    } else {
        // Build the mission items on the fly
        _buildAndAppendMissionItems(items, missionItemParent);
    }
}

280 281
void CorridorScanComplexItem::applyNewAltitude(double newAltitude)
{
DonLakeFlyer's avatar
DonLakeFlyer committed
282 283 284
    _cameraCalc.valueSetIsDistance()->setRawValue(true);
    _cameraCalc.distanceToSurface()->setRawValue(newAltitude);
    _cameraCalc.setDistanceToSurfaceRelative(true);
285 286 287 288 289 290 291 292 293 294 295
}

void CorridorScanComplexItem::_polylineDirtyChanged(bool dirty)
{
    if (dirty) {
        setDirty(true);
    }
}

void CorridorScanComplexItem::rotateEntryPoint(void)
{
296 297 298
    _entryPoint++;
    if (_entryPoint > 3) {
        _entryPoint = 0;
299
    }
300

301
    _rebuildTransects();
302 303 304 305 306
}

void CorridorScanComplexItem::_rebuildCorridorPolygon(void)
{
    if (_corridorPolyline.count() < 2) {
307
        _surveyAreaPolygon.clear();
308 309 310 311 312 313 314 315
        return;
    }

    double halfWidth = _corridorWidthFact.rawValue().toDouble() / 2.0;

    QList<QGeoCoordinate> firstSideVertices = _corridorPolyline.offsetPolyline(halfWidth);
    QList<QGeoCoordinate> secondSideVertices = _corridorPolyline.offsetPolyline(-halfWidth);

316
    _surveyAreaPolygon.clear();
DonLakeFlyer's avatar
DonLakeFlyer committed
317 318

    QList<QGeoCoordinate> rgCoord;
319
    for (const QGeoCoordinate& vertex: firstSideVertices) {
DonLakeFlyer's avatar
DonLakeFlyer committed
320
        rgCoord.append(vertex);
321 322
    }
    for (int i=secondSideVertices.count() - 1; i >= 0; i--) {
DonLakeFlyer's avatar
DonLakeFlyer committed
323
        rgCoord.append(secondSideVertices[i]);
324
    }
DonLakeFlyer's avatar
DonLakeFlyer committed
325
    _surveyAreaPolygon.appendVertices(rgCoord);
326 327
}

328
void CorridorScanComplexItem::_rebuildTransectsPhase1(void)
329
{
330 331 332 333 334 335 336 337 338 339 340
    if (_ignoreRecalc) {
        return;
    }

    // If the transects are getting rebuilt then any previsouly loaded mission items are now invalid
    if (_loadedMissionItemsParent) {
        _loadedMissionItems.clear();
        _loadedMissionItemsParent->deleteLater();
        _loadedMissionItemsParent = NULL;
    }

341
    _transects.clear();
342
    _transectsPathHeightInfo.clear();
343 344 345 346 347 348 349

    double transectSpacing = _cameraCalc.adjustedFootprintSide()->rawValue().toDouble();
    double fullWidth = _corridorWidthFact.rawValue().toDouble();
    double halfWidth = fullWidth / 2.0;
    int transectCount = _transectCount();
    double normalizedTransectPosition = transectSpacing / 2.0;

350 351
    if (_corridorPolyline.count() >= 2) {
        // First build up the transects all going the same direction
352
        //qDebug() << "_rebuildTransectsPhase1";
353
        for (int i=0; i<transectCount; i++) {
354
            //qDebug() << "start transect";
355 356 357 358 359 360 361 362 363
            double offsetDistance;
            if (transectCount == 1) {
                // Single transect is flown over scan line
                offsetDistance = 0;
            } else {
                // Convert from normalized to absolute transect offset distance
                offsetDistance = halfWidth - normalizedTransectPosition;
            }

364 365 366 367 368 369 370 371 372 373 374 375 376
            // Turn transect into CoordInfo transect
            QList<TransectStyleComplexItem::CoordInfo_t> transect;
            QList<QGeoCoordinate> transectCoords = _corridorPolyline.offsetPolyline(offsetDistance);
            for (int j=1; j<transectCoords.count() - 1; j++) {
                TransectStyleComplexItem::CoordInfo_t coordInfo = { transectCoords[j], CoordTypeInterior };
                transect.append(coordInfo);
            }
            TransectStyleComplexItem::CoordInfo_t coordInfo = { transectCoords.first(), CoordTypeSurveyEdge };
            transect.prepend(coordInfo);
            coordInfo = { transectCoords.last(), CoordTypeSurveyEdge };
            transect.append(coordInfo);

            // Extend the transect ends for turnaround
377
            if (_hasTurnaround()) {
378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395
                 QGeoCoordinate turnaroundCoord;
                 double turnAroundDistance = _turnAroundDistanceFact.rawValue().toDouble();

                 double azimuth = transectCoords[0].azimuthTo(transectCoords[1]);
                 turnaroundCoord = transectCoords[0].atDistanceAndAzimuth(-turnAroundDistance, azimuth);
                 turnaroundCoord.setAltitude(qQNaN());
                 TransectStyleComplexItem::CoordInfo_t coordInfo = { turnaroundCoord, CoordTypeTurnaround };
                 transect.prepend(coordInfo);

                 azimuth = transectCoords.last().azimuthTo(transectCoords[transectCoords.count() - 2]);
                 turnaroundCoord = transectCoords.last().atDistanceAndAzimuth(-turnAroundDistance, azimuth);
                 turnaroundCoord.setAltitude(qQNaN());
                 coordInfo = { turnaroundCoord, CoordTypeTurnaround };
                 transect.append(coordInfo);
            }

#if 0
            qDebug() << "transect debug";
396
            for (const TransectStyleComplexItem::CoordInfo_t& coordInfo: transect) {
397
                qDebug() << coordInfo.coordType;
398
            }
399
#endif
400

401
            _transects.append(transect);
402
            normalizedTransectPosition += transectSpacing;
403 404
        }

405 406 407 408 409 410 411 412 413 414 415
        // Now deal with fixing up the entry point:
        //  0: Leave alone
        //  1: Start at same end, opposite side of center
        //  2: Start at opposite end, same side
        //  3: Start at opposite end, opposite side

        bool reverseTransects = false;
        bool reverseVertices = false;
        switch (_entryPoint) {
        case 0:
            reverseTransects = false;
416
            reverseVertices = false;
417 418 419 420 421 422 423 424 425 426 427
            break;
        case 1:
            reverseTransects = true;
            reverseVertices = false;
            break;
        case 2:
            reverseTransects = false;
            reverseVertices = true;
            break;
        case 3:
            reverseTransects = true;
428
            reverseVertices = true;
429
            break;
430
        }
431
        if (reverseTransects) {
432
            QList<QList<TransectStyleComplexItem::CoordInfo_t>> reversedTransects;
433
            for (const QList<TransectStyleComplexItem::CoordInfo_t>& transect: _transects) {
434 435
                reversedTransects.prepend(transect);
            }
436
            _transects = reversedTransects;
437 438
        }
        if (reverseVertices) {
439 440
            for (int i=0; i<_transects.count(); i++) {
                QList<TransectStyleComplexItem::CoordInfo_t> reversedVertices;
441
                for (const TransectStyleComplexItem::CoordInfo_t& vertex: _transects[i]) {
442 443
                    reversedVertices.prepend(vertex);
                }
444
                _transects[i] = reversedVertices;
445
            }
446 447
        }

448
        // Adjust to lawnmower pattern
449
        reverseVertices = false;
450
        for (int i=0; i<_transects.count(); i++) {
451
            // We must reverse the vertices for every other transect in order to make a lawnmower pattern
452
            QList<TransectStyleComplexItem::CoordInfo_t> transectVertices = _transects[i];
453 454
            if (reverseVertices) {
                reverseVertices = false;
455
                QList<TransectStyleComplexItem::CoordInfo_t> reversedVertices;
456 457 458 459 460 461 462
                for (int j=transectVertices.count()-1; j>=0; j--) {
                    reversedVertices.append(transectVertices[j]);
                }
                transectVertices = reversedVertices;
            } else {
                reverseVertices = true;
            }
463
            _transects[i] = transectVertices;
464
        }
465
    }
466
}
467

468
void CorridorScanComplexItem::_recalcComplexDistance(void)
469
{
470
    _complexDistance = 0;
471
    for (int i=0; i<_visualTransectPoints.count() - 1; i++) {
472
        _complexDistance += _visualTransectPoints[i].value<QGeoCoordinate>().distanceTo(_visualTransectPoints[i+1].value<QGeoCoordinate>());
473
    }
474 475
    emit complexDistanceChanged();
}
476

477 478
void CorridorScanComplexItem::_recalcCameraShots(void)
{
479 480 481
    double triggerDistance = _cameraCalc.adjustedFootprintFrontal()->rawValue().toDouble();
    if (triggerDistance == 0) {
        _cameraShots = 0;
482
    } else {
483 484 485 486 487 488
        if (_cameraTriggerInTurnAroundFact.rawValue().toBool()) {
            _cameraShots = qCeil(_complexDistance / triggerDistance);
        } else {
            int singleTransectImageCount = qCeil(_corridorPolyline.length() / triggerDistance);
            _cameraShots = singleTransectImageCount * _transectCount();
        }
489 490 491 492
    }
    emit cameraShotsChanged();
}

493 494 495 496
bool CorridorScanComplexItem::readyForSave(void) const
{
    return TransectStyleComplexItem::readyForSave();
}
497 498 499

double CorridorScanComplexItem::timeBetweenShots(void)
{
500
    return _cruiseSpeed == 0 ? 0 : _cameraCalc.adjustedFootprintFrontal()->rawValue().toDouble() / _cruiseSpeed;
501
}