SurveyComplexItemTest.h 2.84 KB
Newer Older
1 2
/****************************************************************************
 *
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
#include "TransectStyleComplexItemTestBase.h"
13
#include "MultiSignalSpy.h"
14
#include "SurveyComplexItem.h"
15 16
#include "PlanMasterController.h"
#include "PlanViewSettings.h"
17 18 19

#include <QGeoCoordinate>

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

protected:
    void init(void) final;
    void cleanup(void) final;
    
32
#if 1
33 34
private slots:
    void _testDirty(void);
35 36
    void _testGridAngle(void);
    void _testEntryLocation(void);
37 38 39 40 41 42
    void _testItemGeneration(void);
    void _testItemCount(void);
    void _testHoverCaptureItemGeneration(void);
#else
    // Handy mechanism to to a single test
private slots:
43
    void _testItemCount(void);
44
private:
45 46 47 48 49 50
    void _testDirty(void);
    void _testGridAngle(void);
    void _testEntryLocation(void);
    void _testItemGeneration(void);
    void _testHoverCaptureItemGeneration(void);
#endif
51

52 53 54 55
private:
    double          _clampGridAngle180(double gridAngle);
    QList<MAV_CMD>  _createExpectedCommands(bool hasTurnaround, bool useConditionGate);
    void            _testItemGenerationWorker(bool imagesInTurnaround, bool hasTurnaround, bool useConditionGate, const QList<MAV_CMD>& expectedCommands);
56

57 58
    // SurveyComplexItem signals

59
    enum {
60 61 62 63 64 65 66
        surveyVisualTransectPointsChangedIndex = 0,
        surveyCameraShotsChangedIndex,
        surveyCoveredAreaChangedIndex,
        surveyTimeBetweenShotsChangedIndex,
        surveyRefly90DegreesChangedIndex,
        surveyDirtyChangedIndex,
        surveyMaxSignalIndex
67 68 69
    };

    enum {
70 71 72 73 74 75
        surveyVisualTransectPointsChangedMask = 1 << surveyVisualTransectPointsChangedIndex,
        surveyCameraShotsChangedMask =          1 << surveyCameraShotsChangedIndex,
        surveyCoveredAreaChangedMask =          1 << surveyCoveredAreaChangedIndex,
        surveyTimeBetweenShotsChangedMask =     1 << surveyTimeBetweenShotsChangedIndex,
        surveyRefly90DegreesChangedMask =       1 << surveyRefly90DegreesChangedIndex,
        surveyDirtyChangedMask =                1 << surveyDirtyChangedIndex
76 77
    };

78
    static const size_t _cSurveySignals = surveyMaxSignalIndex;
Don Gagne's avatar
Don Gagne committed
79
    const char*         _rgSurveySignals[_cSurveySignals];
80

81 82 83 84 85 86
    MultiSignalSpy*         _multiSpy =             nullptr;
    SurveyComplexItem*      _surveyItem =           nullptr;
    QGCMapPolygon*          _mapPolygon =           nullptr;
    QList<QGeoCoordinate>   _polyVertices;

    static const int _expectedTransectCount = 2;
87
};