qgsvectorlayerprofilegenerator.h

Go to the documentation of this file.

/***************************************************************************
                         qgsvectorlayerprofilegenerator.h
                         ---------------
    begin                : March 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 QGSVECTORLAYERPROFILEGENERATOR_H
#define QGSVECTORLAYERPROFILEGENERATOR_H
 
#include <memory>
 
#include "qgis_core.h"
#include "qgis_sip.h"
#include "qgsabstractprofilesurfacegenerator.h"
#include "qgscoordinatereferencesystem.h"
#include "qgscoordinatetransform.h"
#include "qgscoordinatetransformcontext.h"
#include "qgsfeatureid.h"
#include "qgsfillsymbol.h"
#include "qgslinesymbol.h"
#include "qgsmarkersymbol.h"
 
#define SIP_NO_FILE
 
class QgsProfileRequest;
class QgsCurve;
class QgsVectorLayer;
class QgsVectorLayerFeatureSource;
class QgsAbstractTerrainProvider;
class QgsGeos;
class QgsLineString;
class QgsPolygon;
class QgsProfileSnapContext;
 

class CORE_EXPORT QgsVectorLayerProfileResults : public QgsAbstractProfileSurfaceResults
{
  public:
    struct Feature
    {
        QgsFeatureId featureId;
        QgsGeometry geometry;
        QgsGeometry crossSectionGeometry;
    };
 
    QHash< QgsFeatureId, QVector< Feature > > features;
    QMap< double, QgsFeatureId > mDistanceToFeatureId;
    QPointer< QgsVectorLayer > mLayer;
 
    Qgis::VectorProfileType profileType = Qgis::VectorProfileType::IndividualFeatures;
    bool respectLayerSymbology = true;
    std::unique_ptr< QgsMarkerSymbol > mMarkerSymbol;
    bool mShowMarkerSymbolInSurfacePlots = false;
 
    QString type() const override;
    QVector< QgsGeometry > asGeometries() const override;
    QVector< QgsAbstractProfileResults::Feature > asFeatures( Qgis::ProfileExportType type, QgsFeedback *feedback = nullptr ) const override;
    QgsProfileSnapResult snapPoint( const QgsProfilePoint &point, const QgsProfileSnapContext &context ) override;
    QVector<QgsProfileIdentifyResults> identify( const QgsProfilePoint &point, const QgsProfileIdentifyContext &context ) override;
    QVector<QgsProfileIdentifyResults> identify( const QgsDoubleRange &distanceRange, const QgsDoubleRange &elevationRange, const QgsProfileIdentifyContext &context ) override;
    void renderResults( QgsProfileRenderContext &context ) override;
    void copyPropertiesFromGenerator( const QgsAbstractProfileGenerator *generator ) override;
 
  private:
    void renderResultsAsIndividualFeatures( QgsProfileRenderContext &context );
    void renderMarkersOverContinuousSurfacePlot( QgsProfileRenderContext &context );
    QVector< QgsAbstractProfileResults::Feature > asIndividualFeatures( Qgis::ProfileExportType type, QgsFeedback *feedback = nullptr ) const;
    QgsProfileSnapResult snapPointToIndividualFeatures( const QgsProfilePoint &point, const QgsProfileSnapContext &context );
 
    void visitFeaturesAtPoint(
      const QgsProfilePoint &point,
      double maximumPointDistanceDelta,
      double maximumPointElevationDelta,
      double maximumSurfaceElevationDelta,
      const std::function< void( QgsFeatureId, double delta, double distance, double elevation ) > &visitor,
      bool visitWithin
    ) const;
    void visitFeaturesInRange( const QgsDoubleRange &distanceRange, const QgsDoubleRange &elevationRange, const std::function<void( QgsFeatureId )> &visitor ) const;
};
 

class CORE_EXPORT QgsVectorLayerProfileGenerator : public QgsAbstractProfileSurfaceGenerator
{
  public:
    QgsVectorLayerProfileGenerator( QgsVectorLayer *layer, const QgsProfileRequest &request );
 
    ~QgsVectorLayerProfileGenerator() override;
 
    QString sourceId() const override;
    bool generateProfile( const QgsProfileGenerationContext &context = QgsProfileGenerationContext() ) override;
    QgsAbstractProfileResults *takeResults() override;
    QgsFeedback *feedback() const override;
    QString type() const override;
 
  private:
    // We may need to split mProfileCurve into multiple parts, this will be
    // called for each part.
    bool generateProfileInner( const QgsProfileGenerationContext &context = QgsProfileGenerationContext() );
 
    bool generateProfileForPoints();
    bool generateProfileForLines();
    bool generateProfileForPolygons();
 
    void processIntersectionPoint( const QgsPoint *intersectionPoint, const QgsFeature &feature );
    void processIntersectionCurve( const QgsLineString *intersectionCurve, const QgsFeature &feature );
 
    QgsPoint interpolatePointOnTriangle( const QgsPolygon *triangle, double x, double y ) const;
    void processTriangleIntersectForPoint(
      const QgsPolygon *triangle, const QgsPoint *intersect, QVector< QgsGeometry > &transformedParts, QVector< QgsGeometry > &crossSectionParts, const QgsFeatureId featureId
    );
    void processTriangleIntersectForLine(
      const QgsPolygon *triangle, const QgsLineString *intersect, QVector< QgsGeometry > &transformedParts, QVector< QgsGeometry > &crossSectionParts, const QgsFeatureId featureId
    );
    void processTriangleIntersectForPolygon(
      const QgsPolygon *triangle, const QgsPolygon *intersectionPolygon, QVector< QgsGeometry > &transformedParts, QVector< QgsGeometry > &crossSectionParts, const QgsFeatureId featureId
    );
 
    double tolerance() const;
 
    double terrainHeight( double x, double y ) const;
    double featureZToHeight( double x, double y, double z, double offset ) const;
 
    void clampAltitudes( QgsLineString *lineString, const QgsPoint &centroid, double offset ) const;
    bool clampAltitudes( QgsPolygon *polygon, double offset ) const;
 
    QString mId;
    std::unique_ptr<QgsFeedback> mFeedback = nullptr;
 
    std::unique_ptr< QgsCurve > mProfileCurve;
    std::unique_ptr< QgsGeos > mProfileCurveEngine;
 
    std::unique_ptr<QgsAbstractGeometry> mProfileBufferedCurve;
    std::unique_ptr< QgsGeos > mProfileBufferedCurveEngine;
 
    std::unique_ptr< QgsAbstractTerrainProvider > mTerrainProvider;
 
    std::unique_ptr< QgsCurve > mTransformedCurve;
    double mTolerance = 0;
 
 
    QgsCoordinateReferenceSystem mSourceCrs;
    QgsCoordinateReferenceSystem mTargetCrs;
    QgsCoordinateTransformContext mTransformContext;
    QgsRectangle mExtent;
 
    std::unique_ptr< QgsVectorLayerFeatureSource > mSource;
 
    double mOffset = 0;
    double mScale = 1;
    Qgis::VectorProfileType mType = Qgis::VectorProfileType::IndividualFeatures;
    Qgis::AltitudeClamping mClamping = Qgis::AltitudeClamping::Terrain;
    Qgis::AltitudeBinding mBinding = Qgis::AltitudeBinding::Centroid;
    bool mExtrusionEnabled = false;
    double mExtrusionHeight = 0;
    bool mCustomToleranceEnabled = false;
    double mCustomTolerance = 0;
 
    QgsExpressionContext mExpressionContext;
    QgsFields mFields;
    QgsPropertyCollection mDataDefinedProperties;
 
    Qgis::WkbType mWkbType = Qgis::WkbType::Unknown;
    QgsCoordinateTransform mLayerToTargetTransform;
    QgsCoordinateTransform mTargetToTerrainProviderTransform;
 
    std::unique_ptr< QgsVectorLayerProfileResults > mResults;
    std::unique_ptr< QgsMarkerSymbol > mMarkerSymbol;
 
    bool mRespectLayerSymbology = true;
    std::unique_ptr< QgsMarkerSymbol > mProfileMarkerSymbol;
    bool mShowMarkerSymbolInSurfacePlots = false;
 
    // NOT for use in the background thread!
    QPointer< QgsVectorLayer > mLayer;
 
    friend class QgsVectorLayerProfileResults;
};
 
#endif // QGSVECTORLAYERPROFILEGENERATOR_H