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