qgsmeshforcebypolylines.h

Go to the documentation of this file.

/***************************************************************************
  qgsmeshforcebypolylines.h - QgsMeshForceByPolylines
 
 ---------------------
 begin                : 5.9.2021
 copyright            : (C) 2021 by Vincent Cloarec
 email                : vcloarec 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 QGSMESHFORCEBYPOLYLINES_H
#define QGSMESHFORCEBYPOLYLINES_H
 
 
#include "qgis_core.h"
#include "qgsabstractgeometry.h"
#include "qgsmeshadvancedediting.h"
#include "qgstopologicalmesh.h"
#include "qgstriangularmesh.h"

class CORE_EXPORT QgsMeshEditForceByLine : public QgsMeshAdvancedEditing
{
  public:
    QgsMeshEditForceByLine() = default;

    void setInputLine( const QgsPoint &pt1, const QgsPoint &pt2, double tolerance, bool newVertexOnIntersection );

    void setTolerance( double tolerance );

    void setAddVertexOnIntersection( bool addVertex );

    void setDefaultZValue( double defaultZValue );

    void setInterpolateZValueOnMesh( bool interpolateZValueOnMesh );
 
  private:
    QgsPoint mPoint1;
    QgsPoint mPoint2;
    bool mNewVertexOnIntersection = false;
    double mTolerance = 1e-8;
    double mDefaultZValue = 0;
    bool mInterpolateZValueOnMesh = false;
 
    QgsTopologicalMesh::Changes apply( QgsMeshEditor *meshEditor ) override;
 
    virtual void finish();
 
    //members and method used for the calculation
    QgsMeshEditor *mEditor = nullptr;
    QList<int> mRemovedFaces;
    QList<int> mHoleOnLeft;               // contains the border vertices of hole to fill on the right of the line (line go up)
    QList<int> mNeighborOnLeft;           //contains the neighbor face on the right of the line (line go up)
    QList<int> mHoleOnRight;              // contains the border vertices of hole to fill on the right of the line (line go up)
    QList<int> mNeighborOnRight;          // contains the neighbor face on the right of the line (line go up)
    QList<int> mNewVerticesIndexesOnLine; //the new vertices intersecting edges except
    bool mEndOnPoint2 = false;
    int mPoint2VertexIndex = -1;
    int mCurrentSnappedVertex = -1; // Last snapped point
    QgsPoint mCurrentPointPosition;
 
    bool mFirstPointChecked = false;
    bool mSecondPointChecked = false;
 
    void interpolateZValueOnMesh( QgsPoint &point ) const;
    void interpolateZValueOnMesh( int faceIndex, QgsPoint &point ) const;
    void interpolateZValue( QgsMeshVertex &point, const QgsPoint &otherPoint1, const QgsPoint &otherPoint2 );
 
 
    bool buildForcedElements();
 
    bool edgeIntersection( int vertex1, int vertex2, int &closestSnappedVertex, QgsPoint &intersectionPoint, bool outAllowed );
 
    bool searchIntersectionEdgeFromSnappedVertex(
      int &intersectionFaceIndex, int &previousSnappedVertex, int &currentSnappedVertexIndex, QgsPoint &intersectionPoint, int &edgePosition, QSet<int> &treatedFaces
    );
 
    // Insert a new vertex and returns its local index (0 is first index in th
    int insertNewVertex( const QgsMeshVertex &vertex );
 
    bool triangulateHoles( const QList<int> &holeOnLeft, const QList<int> &neighborOnLeft, bool isLeftHole, QList<std::array<int, 2> > &newFacesOnLine, std::array<int, 2> &extremeFaces );
 
    bool finishForcingLine();
 
    friend class TestQgsMeshEditor;
    friend class QgsMeshEditForceByPolylines;
};
 

class CORE_EXPORT QgsMeshEditForceByPolylines : public QgsMeshEditForceByLine
{
  public:
    QgsMeshEditForceByPolylines() = default;
 
    QString text() const override;
    bool isFinished() const override;

    void addLineFromGeometry( const QgsGeometry &geom );

    void addLinesFromGeometries( const QList<QgsGeometry> geometries );
 
  private:
    QList<QgsPointSequence> mPolylines;
    int mCurrentPolyline = 0;
    int mCurrentSegment = 0;
 
    void incrementSegment();
 
    using QgsMeshEditForceByLine::setInputLine;
    QgsTopologicalMesh::Changes apply( QgsMeshEditor *meshEditor ) override;
};
 
#endif // QGSMESHFORCEBYPOLYLINES_H