qgstriangularmesh.h

Go to the documentation of this file.

/***************************************************************************
                         qgstriangularmesh.h
                         -------------------
    begin                : April 2018
    copyright            : (C) 2018 by Peter Petrik
    email                : zilolv 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 QGSTRIANGULARMESH_H
#define QGSTRIANGULARMESH_H


#define SIP_NO_FILE

#include <QVector>
#include "qgis_core.h"
#include "qgsmeshdataprovider.h"
#include "qgsgeometry.h"
#include "qgsfeatureid.h"
#include "qgsspatialindex.h"
#include "qgsfeatureiterator.h"

class QgsRenderContext;
class QgsCoordinateTransform;
class QgsRectangle;

struct CORE_EXPORT QgsMesh
{
  QVector<QgsMeshVertex> vertices;
  QVector<QgsMeshFace> faces;
};


class CORE_NO_EXPORT QgsMeshFeatureIterator : public QgsAbstractFeatureIterator
{
  public:

    QgsMeshFeatureIterator( QgsMesh *mesh );
    ~QgsMeshFeatureIterator() override;

    bool rewind() override;
    bool close() override;

  protected:
    bool fetchFeature( QgsFeature &f ) override;

  private:
    QgsMesh *mMesh = nullptr;
    int it = 0;
};


class CORE_EXPORT QgsTriangularMesh
{
  public:
    QgsTriangularMesh() = default;
    ~QgsTriangularMesh() = default;

    void update( QgsMesh *nativeMesh, QgsRenderContext *context );

    const QVector<QgsMeshVertex> &vertices() const ;
    const QVector<QgsMeshFace> &triangles() const ;

    const QVector<QgsMeshVertex> &centroids() const ;

    const QVector<int> &trianglesToNativeFaces() const ;

    int faceIndexForPoint( const QgsPointXY &point ) const ;

    QList<int> faceIndexesForRectangle( const QgsRectangle &rectangle ) const ;

  private:
    // vertices: map CRS; 0-N ... native vertices, N+1 - len ... extra vertices
    // faces are derived triangles
    QgsMesh mTriangularMesh;
    QVector<int> mTrianglesToNativeFaces; //len(mTrianglesToNativeFaces) == len(mTriangles). Mapping derived -> native

    // centroids of the native faces in map CRS
    QVector<QgsMeshVertex> mNativeMeshFaceCentroids;

    QgsSpatialIndex mSpatialIndex;
    QgsCoordinateTransform mCoordinateTransform; //coordinate transform used to convert native mesh vertices to map vertices
};

namespace QgsMeshUtils
{
  QgsGeometry toGeometry( const QgsMeshFace &face, const QVector<QgsMeshVertex> &vertices );

  QList<int> nativeFacesFromTriangles( const QList<int> &triangleIndexes, const QVector<int> &trianglesToNativeFaces );
};

#endif // QGSTRIANGULARMESH_H