qgsremappingproxyfeaturesink.h

Go to the documentation of this file.

/***************************************************************************
                         qgsremappingproxyfeaturesink.h
                         ----------------------
    begin                : April 2020
    copyright            : (C) 2020 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 QGSREMAPPINGPROXYFEATURESINK_H
#define QGSREMAPPINGPROXYFEATURESINK_H
 
#include "qgis.h"
#include "qgis_core.h"
#include "qgscoordinatetransform.h"
#include "qgsfeaturesink.h"
#include "qgsproperty.h"

class CORE_EXPORT QgsRemappingSinkDefinition
{
  public:
    QMap< QString, QgsProperty > fieldMap() const { return mFieldMap; }

    void setFieldMap( const QMap< QString, QgsProperty > &map ) { mFieldMap = map; }

    void addMappedField( const QString &destinationField, const QgsProperty &property ) { mFieldMap.insert( destinationField, property ); }

    QgsCoordinateReferenceSystem sourceCrs() const { return mSourceCrs; }

    void setSourceCrs( const QgsCoordinateReferenceSystem &source ) { mSourceCrs = source; }

    QgsCoordinateReferenceSystem destinationCrs() const { return mDestinationCrs; }

    void setDestinationCrs( const QgsCoordinateReferenceSystem &destination ) { mDestinationCrs = destination; }

    Qgis::WkbType destinationWkbType() const { return mDestinationWkbType; }

    void setDestinationWkbType( Qgis::WkbType type ) { mDestinationWkbType = type; }

    QgsFields destinationFields() const { return mDestinationFields; }

    void setDestinationFields( const QgsFields &fields ) { mDestinationFields = fields; }

    QVariant toVariant() const;

    bool loadVariant( const QVariantMap &map );
 
    // TODO c++20 - replace with = default
    bool operator==( const QgsRemappingSinkDefinition &other ) const;
    bool operator!=( const QgsRemappingSinkDefinition &other ) const;
 
  private:
    QMap< QString, QgsProperty > mFieldMap;
 
    QgsCoordinateReferenceSystem mSourceCrs;
    QgsCoordinateReferenceSystem mDestinationCrs;
 
    Qgis::WkbType mDestinationWkbType = Qgis::WkbType::Unknown;
 
    QgsFields mDestinationFields;
};
 
Q_DECLARE_METATYPE( QgsRemappingSinkDefinition )
 
 

class CORE_EXPORT QgsRemappingProxyFeatureSink : public QgsFeatureSink
{
  public:
#ifndef SIP_RUN

    QgsRemappingProxyFeatureSink( const QgsRemappingSinkDefinition &mappingDefinition, QgsFeatureSink *sink, bool ownsSink = false );
#else

    QgsRemappingProxyFeatureSink( const QgsRemappingSinkDefinition &mappingDefinition, QgsFeatureSink *sink );
#endif
 
    ~QgsRemappingProxyFeatureSink() override;

    void setExpressionContext( const QgsExpressionContext &context ) const;

    void setTransformContext( const QgsCoordinateTransformContext &context );

    QgsFeatureList remapFeature( const QgsFeature &feature ) const;
 
    bool addFeature( QgsFeature &feature, QgsFeatureSink::Flags flags = QgsFeatureSink::Flags() ) override;
    bool addFeatures( QgsFeatureList &features, QgsFeatureSink::Flags flags = QgsFeatureSink::Flags() ) override;
    bool addFeatures( QgsFeatureIterator &iterator, QgsFeatureSink::Flags flags = QgsFeatureSink::Flags() ) override;
    QString lastError() const override;

    QgsFeatureSink *destinationSink() { return mSink; }
 
  private:
    QgsRemappingSinkDefinition mDefinition;
    QgsCoordinateTransform mTransform;
    QgsFeatureSink *mSink = nullptr;
    mutable QgsExpressionContext mContext;
    bool mOwnsSink = false;
};
 
#endif // QGSREMAPPINGPROXYFEATURESINK_H