SurveyComplexItemTest.h 2.71 KB
Newer Older
1 2
/****************************************************************************
 *
Gus Grubba's avatar
Gus Grubba committed
3
 * (c) 2009-2020 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
4 5 6 7 8 9
 *
 * QGroundControl is licensed according to the terms in the file
 * COPYING.md in the root of the source code directory.
 *
 ****************************************************************************/

10
#pragma once
11 12 13 14

#include "UnitTest.h"
#include "TCPLink.h"
#include "MultiSignalSpy.h"
15
#include "SurveyComplexItem.h"
16
#include "PlanMasterController.h"
17
#include "PlanViewSettings.h"
18 19 20

#include <QGeoCoordinate>

21 22
/// Unit test for SurveyComplexItem
class SurveyComplexItemTest : public UnitTest
23 24 25 26
{
    Q_OBJECT
    
public:
27
    SurveyComplexItemTest(void);
28 29 30 31 32 33

protected:
    void init(void) final;
    void cleanup(void) final;
    
private slots:
34 35 36
    void _testHoverCaptureItemGeneration(void);

private:
37
    void _testDirty(void);
38 39
    void _testGridAngle(void);
    void _testEntryLocation(void);
40
    void _testItemCount(void);
41
    void _testItemGeneration(void);
42
    double _clampGridAngle180(double gridAngle);
43
    void _setPolygon(void);
44 45
    QList<MAV_CMD> _createExpectedCommands(bool hasTurnaround, bool useConditionGate);
    void _testItemGenerationWorker(bool imagesInTurnaround, bool hasTurnaround, bool useConditionGate, const QList<MAV_CMD>& expectedCommands);
46

47 48
    // SurveyComplexItem signals

49
    enum {
50 51 52 53 54 55 56
        surveyVisualTransectPointsChangedIndex = 0,
        surveyCameraShotsChangedIndex,
        surveyCoveredAreaChangedIndex,
        surveyTimeBetweenShotsChangedIndex,
        surveyRefly90DegreesChangedIndex,
        surveyDirtyChangedIndex,
        surveyMaxSignalIndex
57 58 59
    };

    enum {
60 61 62 63 64 65
        surveyVisualTransectPointsChangedMask = 1 << surveyVisualTransectPointsChangedIndex,
        surveyCameraShotsChangedMask =          1 << surveyCameraShotsChangedIndex,
        surveyCoveredAreaChangedMask =          1 << surveyCoveredAreaChangedIndex,
        surveyTimeBetweenShotsChangedMask =     1 << surveyTimeBetweenShotsChangedIndex,
        surveyRefly90DegreesChangedMask =       1 << surveyRefly90DegreesChangedIndex,
        surveyDirtyChangedMask =                1 << surveyDirtyChangedIndex
66 67
    };

68
    static const size_t _cSurveySignals = surveyMaxSignalIndex;
Don Gagne's avatar
Don Gagne committed
69
    const char*         _rgSurveySignals[_cSurveySignals];
70

71 72 73 74 75
    PlanMasterController*   _masterController =     nullptr;
    Vehicle*                _controllerVehicle =    nullptr;
    MultiSignalSpy*         _multiSpy =             nullptr;
    SurveyComplexItem*      _surveyItem =           nullptr;
    QGCMapPolygon*          _mapPolygon =           nullptr;
76 77 78 79
    PlanViewSettings*       _planViewSettings =     nullptr;
    QList<QGeoCoordinate>   _polyVertices;

    static const int _expectedTransectCount = 2;
80
};