qgsmemoryfeatureiterator.h

Go to the documentation of this file.

/***************************************************************************
    qgsmemoryfeatureiterator.h
    ---------------------
    begin                : Juli 2012
    copyright            : (C) 2012 by Martin Dobias
    email                : wonder dot 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 QGSMEMORYFEATUREITERATOR_H
#define QGSMEMORYFEATUREITERATOR_H
 
#define SIP_NO_FILE
 
#include "qgsfeatureiterator.h"
#include "qgsexpressioncontext.h"
#include "qgsfields.h"
#include "qgsgeometry.h"
 
 
class QgsMemoryProvider;
 
typedef QMap<QgsFeatureId, QgsFeature> QgsFeatureMap;
 
class QgsSpatialIndex;
 
 
class QgsMemoryFeatureSource final: public QgsAbstractFeatureSource
{
  public:
    explicit QgsMemoryFeatureSource( const QgsMemoryProvider *p );
 
    QgsFeatureIterator getFeatures( const QgsFeatureRequest &request ) override;
 
    QgsExpressionContext *expressionContext();
 
  private:
    QgsFields mFields;
    QgsFeatureMap mFeatures;
    std::unique_ptr< QgsSpatialIndex > mSpatialIndex;
    QString mSubsetString;
    std::unique_ptr< QgsExpressionContext > mExpressionContext;
    QgsCoordinateReferenceSystem mCrs;
 
    friend class QgsMemoryFeatureIterator;
};
 
 
class QgsMemoryFeatureIterator final: public QgsAbstractFeatureIteratorFromSource<QgsMemoryFeatureSource>
{
  public:
    QgsMemoryFeatureIterator( QgsMemoryFeatureSource *source, bool ownSource, const QgsFeatureRequest &request );
 
    ~QgsMemoryFeatureIterator() override;
 
    bool rewind() override;
    bool close() override;
 
  protected:
 
    bool fetchFeature( QgsFeature &feature ) override;
 
  private:
    bool nextFeatureUsingList( QgsFeature &feature );
    bool nextFeatureTraverseAll( QgsFeature &feature );
 
    QgsGeometry mSelectRectGeom;
    std::unique_ptr< QgsGeometryEngine > mSelectRectEngine;
    QgsRectangle mFilterRect;
    QgsFeatureMap::const_iterator mSelectIterator;
    bool mUsingFeatureIdList = false;
    QList<QgsFeatureId> mFeatureIdList;
    QList<QgsFeatureId>::const_iterator mFeatureIdListIterator;
    std::unique_ptr< QgsExpression > mSubsetExpression;
    QgsCoordinateTransform mTransform;
 
};
 
 
#endif // QGSMEMORYFEATUREITERATOR_H