qgsvectorwarper.h

Go to the documentation of this file.

/***************************************************************************
    qgsvectorwarper.h
     --------------------------------------
    Date                 : February 2022
    Copyright            : (C) 2022 by Nyall Dawson
    Email                : nyall dot dawson at gmail dot com
 ***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#ifndef QGSVECTORWARPER_H
#define QGSVECTORWARPER_H
 
#include "qgspoint.h"
#include "qgis_analysis.h"
#include "qgis_sip.h"
#include "qgsgcptransformer.h"
#include "qgsgcppoint.h"
#include "qgsfeatureiterator.h"
#include "qgsfeaturesource.h"
#include "qgstaskmanager.h"
#include "qgsvectorfilewriter.h"
#include "qgsvectorlayerfeatureiterator.h"
 
class QgsFeatureSink;
 
class ANALYSIS_EXPORT QgsVectorWarper
{
  public:
 
    explicit QgsVectorWarper( QgsGcpTransformerInterface::TransformMethod method, const QList < QgsGcpPoint > &points,
                              const QgsCoordinateReferenceSystem &destinationCrs );
 
    bool transformFeatures( QgsFeatureIterator &iterator, QgsFeatureSink *sink,
                            const QgsCoordinateTransformContext &context,
                            QgsFeedback *feedback = nullptr ) const;
 
    QString error() const { return mError; }
 
  private:
    QgsGcpTransformerInterface::TransformMethod mMethod = QgsGcpTransformerInterface::TransformMethod::Linear;
    QList < QgsGcpPoint > mPoints;
    QgsCoordinateReferenceSystem mDestinationCrs;
 
    mutable QString mError;
 
};
 
 
class ANALYSIS_EXPORT QgsVectorWarperTask : public QgsTask
{
    Q_OBJECT
 
  public:
 
    QgsVectorWarperTask( QgsGcpTransformerInterface::TransformMethod method, const QList < QgsGcpPoint > &points,
                         const QgsCoordinateReferenceSystem &destinationCrs,
                         QgsVectorLayer *layer,
                         const QString &fileName );
 
    void cancel() override;
 
    enum class Result
    {
      Success, 
      Canceled, 
      Error, 
    };
 
    Result result() const { return mResult; }
 
    QString errorMessage() const { return mErrorMessage; }
 
  protected:
 
    bool run() override;
 
  private:
 
    QgsGcpTransformerInterface::TransformMethod mMethod = QgsGcpTransformerInterface::TransformMethod::Linear;
    long long mFeatureCount = 0;
    std::unique_ptr< QgsVectorLayerFeatureSource > mSource;
    QList < QgsGcpPoint > mPoints;
    QgsCoordinateReferenceSystem mDestinationCrs;
 
    QString mDestFileName;
 
    std::unique_ptr< QgsFeedback > mFeedback;
 
    QgsCoordinateTransformContext mTransformContext;
    QgsFields mFields;
    QgsWkbTypes::Type mWkbType = QgsWkbTypes::Unknown;
 
    Result mResult = Result::Success;
    QString mErrorMessage;
    double mLastProgress = 0;
};
 
 
#endif //QGSVECTORWARPER_H