Skip to content
SnakeWorker.cc 2 KiB
Newer Older
Valentin Platzgummer's avatar
Valentin Platzgummer committed
#include <QGeoCoordinate>


SnakeWorker::SnakeWorker(QObject *parent) : QThread(parent)
  , _lineDistance       (3) // meter
  , _minTransectLength  (2) // meter
{
Valentin Platzgummer's avatar
Valentin Platzgummer committed
}

void SnakeWorker::setScenario(const Scenario &scenario)
{
    _scenario = scenario;
}

Valentin Platzgummer's avatar
Valentin Platzgummer committed
void SnakeWorker::setProgress(const QVector<int> &progress)
Valentin Platzgummer's avatar
Valentin Platzgummer committed
    for (auto p : progress) {
Valentin Platzgummer's avatar
Valentin Platzgummer committed
        _progress.push_back(p);
    }
}

void SnakeWorker::setLineDistance(double lineDistance)
{
    assert(lineDistance > 0);
    _lineDistance = lineDistance;
}

void SnakeWorker::setMinTransectLength(double minTransectLength)
{
    assert(minTransectLength > 0);
    _minTransectLength = minTransectLength;
}

const WorkerResult_t &SnakeWorker::getResult() const
{
    return _result;
}

Valentin Platzgummer's avatar
Valentin Platzgummer committed
void SnakeWorker::run()
{
    // Reset _result struct.
    _result.success = false;
    _result.errorMessage = "";
    _result.waypoints.clear();
    _result.returnPathIdx.clear();
    _result.arrivalPathIdx.clear();
Valentin Platzgummer's avatar
Valentin Platzgummer committed

    FlightPlan flightplan;
    flightplan.setScenario(_scenario);
    flightplan.setProgress(_progress);

    // Trying to generate flight plan.
    if ( !flightplan.generate(_lineDistance, _minTransectLength) ){
        // error
        for (auto c : flightplan.errorString){
            _result.errorMessage.push_back(QChar(c));
Valentin Platzgummer's avatar
Valentin Platzgummer committed
        }
    } else {
        //success!!!
Valentin Platzgummer's avatar
Valentin Platzgummer committed

        // Fill result struct.
        auto waypoints = flightplan.getWaypoints();
        for (auto vertex : waypoints){
            _result.waypoints.append(QVariant::fromValue(QGeoCoordinate(vertex[0], vertex[1], 0)));
Valentin Platzgummer's avatar
Valentin Platzgummer committed
        }

        auto arrivalPathIdx = flightplan.getArrivalPathIdx();
        for (auto idx : arrivalPathIdx){
            _result.arrivalPathIdx.append(idx);
Valentin Platzgummer's avatar
Valentin Platzgummer committed
        }

        auto returnPathIdx = flightplan.getReturnPathIdx();
        for (auto idx : returnPathIdx){
            _result.returnPathIdx.append(idx);