qgsspatialindex.h

Go to the documentation of this file.

/***************************************************************************
    qgsspatialindex.h  - wrapper class for spatial index library
    ----------------------
    begin                : December 2006
    copyright            : (C) 2006 by Martin Dobias
    email                : wonder.sk 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 QGSSPATIALINDEX_H
#define QGSSPATIALINDEX_H
 
 
#include "qgis_sip.h"
 
#include <QString>
 
using namespace Qt::StringLiterals;
 
// forward declaration
namespace SpatialIndex SIP_SKIP
{
  class IStorageManager;
  class ISpatialIndex;
  class Region;
  class Point;
 
  namespace StorageManager
  {
    class IBuffer;
  }
} //namespace SpatialIndex
 
class QgsFeedback;
class QgsFeature;
class QgsRectangle;
class QgsPointXY;
 
#include "qgis_core.h"
#include "qgsfeaturesink.h"
#include <QList>
#include <QSharedDataPointer>
 
#include "qgsfeature.h"
 
class QgsSpatialIndexData;
class QgsFeatureIterator;
class QgsFeatureSource;

class CORE_EXPORT QgsSpatialIndex : public QgsFeatureSink
{
  public:
    /* creation of spatial index */

    enum Flag SIP_ENUM_BASETYPE( IntFlag )
    {
      FlagStoreFeatureGeometries
      = 1 << 0, 
    };
    Q_DECLARE_FLAGS( Flags, Flag )
 
    
    QgsSpatialIndex( QgsSpatialIndex::Flags flags = QgsSpatialIndex::Flags() );

    explicit QgsSpatialIndex( const QgsFeatureIterator &fi, QgsFeedback *feedback = nullptr, QgsSpatialIndex::Flags flags = QgsSpatialIndex::Flags() );
 
#ifndef SIP_RUN

    explicit QgsSpatialIndex( const QgsFeatureIterator &fi, const std::function< bool( const QgsFeature & ) > &callback, QgsSpatialIndex::Flags flags = QgsSpatialIndex::Flags() );
#endif

    explicit QgsSpatialIndex( const QgsFeatureSource &source, QgsFeedback *feedback = nullptr, QgsSpatialIndex::Flags flags = QgsSpatialIndex::Flags() );
 
    QgsSpatialIndex( const QgsSpatialIndex &other );

    ~QgsSpatialIndex() override;
 
    QgsSpatialIndex &operator=( const QgsSpatialIndex &other );
 
    /* operations */

    Q_DECL_DEPRECATED bool insertFeature( const QgsFeature &feature ) SIP_DEPRECATED;

    bool addFeature( QgsFeature &feature, QgsFeatureSink::Flags flags = QgsFeatureSink::Flags() ) override;
 
    using QgsFeatureSink::addFeatures;

    bool addFeatures( QgsFeatureList &features, QgsFeatureSink::Flags flags = QgsFeatureSink::Flags() ) override;

    Q_DECL_DEPRECATED bool insertFeature( QgsFeatureId id, const QgsRectangle &bounds ) SIP_DEPRECATED;

    bool addFeature( QgsFeatureId id, const QgsRectangle &bounds );

    bool deleteFeature( const QgsFeature &feature );

    bool deleteFeature( QgsFeatureId id, const QgsRectangle &bounds );
 
    /* queries */

    QList<QgsFeatureId> intersects( const QgsRectangle &rectangle ) const;

    QList<QgsFeatureId> nearestNeighbor( const QgsPointXY &point, int neighbors = 1, double maxDistance = 0 ) const;

    QList<QgsFeatureId> nearestNeighbor( const QgsGeometry &geometry, int neighbors = 1, double maxDistance = 0 ) const;
 
#ifndef SIP_RUN

    QgsGeometry geometry( QgsFeatureId id ) const;
 
#else
    // clang-format off

    SIP_PYOBJECT geometry( QgsFeatureId id ) const SIP_TYPEHINT( QgsGeometry );
    % MethodCode
    auto g = std::make_unique< QgsGeometry >( sipCpp->geometry( a0 ) );
    if ( g->isNull() )
    {
      PyErr_SetString( PyExc_KeyError, u"No geometry with feature id %1 exists in the index."_s.arg( a0 ).toUtf8().constData() );
      sipIsErr = 1;
    }
    else
    {
      sipRes = sipConvertFromType( g.release(), sipType_QgsGeometry, Py_None );
    }
    % End
// clang-format on
#endif
 
    /* debugging */

    QAtomicInt SIP_PYALTERNATIVETYPE( int ) refs() const;
 
  private:
    static bool featureInfo( const QgsFeature &f, SpatialIndex::Region &r, QgsFeatureId &id ) SIP_SKIP;

    static bool featureInfo( const QgsFeature &f, QgsRectangle &rect, QgsFeatureId &id );
 
    friend class QgsFeatureIteratorDataStream; // for access to featureInfo()
 
  private:
    QSharedDataPointer<QgsSpatialIndexData> d;
};
 
Q_DECLARE_OPERATORS_FOR_FLAGS( QgsSpatialIndex::Flags )
 
#endif //QGSSPATIALINDEX_H