qgsvectorlayerdirector.h

Go to the documentation of this file.

/***************************************************************************
  qgsvectorlayerdirector.h
  --------------------------------------
  Date                 : 2010-10-20
  Copyright            : (C) 2010 by Yakushev Sergey
  Email                : YakushevS <at> list.ru
****************************************************************************
*                                                                          *
*   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 QGSVECTORLAYERDIRECTOR_H
#define QGSVECTORLAYERDIRECTOR_H
 
#include "qgis.h"
#include "qgis_analysis.h"
#include "qgsgraphdirector.h"
 
class QgsGraphBuilderInterface;
class QgsFeatureSource;

class ANALYSIS_EXPORT QgsVectorLayerDirector : public QgsGraphDirector
{
    Q_OBJECT
 
  public:
    enum Direction
    {
      DirectionForward,  
      DirectionBackward, 
      DirectionBoth,     
    };

    QgsVectorLayerDirector(
      QgsFeatureSource *source, int directionFieldId, const QString &directDirectionValue, const QString &reverseDirectionValue, const QString &bothDirectionValue, Direction defaultDirection
    );
 
    /*
     * MANDATORY DIRECTOR PROPERTY DECLARATION
     */
    void makeGraph( QgsGraphBuilderInterface *builder, const QVector<QgsPointXY> &additionalPoints, QVector<QgsPointXY> &snappedPoints SIP_OUT, QgsFeedback *feedback = nullptr ) const override;
 
    QString name() const override;
 
#ifndef SIP_RUN
    struct VertexSourceInfo
    {
        VertexSourceInfo( QgsFeatureId fid, int partId )
          : fid( fid )
          , partId( partId )
        {}

        QgsFeatureId fid = -1;
        int partId = 0;
 
        bool operator==( const VertexSourceInfo &other ) const = default;
    };

    const std::vector< VertexSourceInfo > &sourcesForVertex( std::size_t vertexIndex ) const { return mVertexSources[vertexIndex]; }
#endif
 
  private:
    QgsFeatureSource *mSource = nullptr;
    int mDirectionFieldId = -1;
    QString mDirectDirectionValue;
    QString mReverseDirectionValue;
    QString mBothDirectionValue;
    Direction mDefaultDirection = DirectionBoth;
 
    mutable std::vector<std::vector<VertexSourceInfo>> mVertexSources;
 
    QgsAttributeList requiredAttributes() const;
    Direction directionForFeature( const QgsFeature &feature ) const;
};
 
#endif // QGSVECTORLAYERDIRECTOR_H