src/core/qgsvectorlayer.h

Go to the documentation of this file.

/***************************************************************************
                          qgsvectorlayer.h  -  description
                             -------------------
    begin                : Oct 29, 2003
    copyright            : (C) 2003 by Gary E.Sherman
    email                : sherman at mrcc.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 QGSVECTORLAYER_H
#define QGSVECTORLAYER_H

#include <QMap>
#include <QSet>
#include <QList>
#include <QStringList>

#include "qgis.h"
#include "qgsmaplayer.h"
#include "qgsfeature.h"
#include "qgsfeatureiterator.h"
#include "qgsfield.h"
#include "qgssnapper.h"
#include "qgsfield.h"

class QPainter;
class QImage;

class QgsAttributeAction;
class QgsCoordinateTransform;
class QgsFeatureRequest;
class QgsGeometry;
class QgsGeometryVertexIndex;
class QgsMapToPixel;
class QgsLabel;
class QgsRectangle;
class QgsVectorDataProvider;
class QgsVectorOverlay;
class QgsSingleSymbolRendererV2;
class QgsRectangle;
class QgsVectorLayerJoinBuffer;
class QgsFeatureRendererV2;
class QgsDiagramRendererV2;
class QgsDiagramLayerSettings;
class QgsGeometryCache;
class QgsVectorLayerEditBuffer;
class QgsSymbolV2;

typedef QList<int> QgsAttributeList;
typedef QSet<int> QgsAttributeIds;

class CORE_EXPORT QgsAttributeEditorElement : public QObject
{
    Q_OBJECT
  public:

    enum AttributeEditorType
    {
      AeTypeContainer,
      AeTypeField,
      AeTypeInvalid
    };

    QgsAttributeEditorElement( AttributeEditorType type, QString name, QObject *parent = NULL )
        : QObject( parent ), mType( type ), mName( name ) {}

    virtual ~QgsAttributeEditorElement() {}

    QString name() const { return mName; }
    AttributeEditorType type() const { return mType; }

    virtual QDomElement toDomElement( QDomDocument& doc ) const = 0;

  protected:
    AttributeEditorType mType;
    QString mName;
};

class CORE_EXPORT QgsAttributeEditorContainer : public QgsAttributeEditorElement
{
  public:
    QgsAttributeEditorContainer( QString name, QObject *parent )
        : QgsAttributeEditorElement( AeTypeContainer, name, parent ) {}

    ~QgsAttributeEditorContainer() {}

    virtual QDomElement toDomElement( QDomDocument& doc ) const;
    virtual void addChildElement( QgsAttributeEditorElement *widget );
    QList<QgsAttributeEditorElement*> children() const { return mChildren; }

  private:
    QList<QgsAttributeEditorElement*> mChildren;
};

class CORE_EXPORT QgsAttributeEditorField : public QgsAttributeEditorElement
{
  public:
    QgsAttributeEditorField( QString name , int idx, QObject *parent )
        : QgsAttributeEditorElement( AeTypeField, name, parent ), mIdx( idx ) {}

    ~QgsAttributeEditorField() {}

    virtual QDomElement toDomElement( QDomDocument& doc ) const;
    int idx() const { return mIdx; }

  private:
    int mIdx;
};

struct CORE_EXPORT QgsVectorJoinInfo
{
  QString targetFieldName;
  QString joinLayerId;
  QString joinFieldName;
  bool memoryCache;
  QHash< QString, QgsAttributes> cachedAttributes;

  int targetFieldIndex;
  int joinFieldIndex;
};


class CORE_EXPORT QgsVectorLayer : public QgsMapLayer
{
    Q_OBJECT

  public:
    enum EditorLayout
    {
      GeneratedLayout = 0,
      TabLayout = 1,
      UiFileLayout = 2
    };

    enum EditType
    {
      LineEdit,
      UniqueValues,
      UniqueValuesEditable,
      ValueMap,
      Classification,
      EditRange,
      SliderRange,
      CheckBox,      /* added in 1.4 */
      FileName,
      Enumeration,
      Immutable,     /* The attribute value should not be changed in the attribute form */
      Hidden,        /* The attribute value should not be shown in the attribute form @added in 1.4 */
      TextEdit,      /* multiline edit @added in 1.4*/
      Calendar,      /* calendar widget @added in 1.5 */
      DialRange,     /* dial range @added in 1.5 */
      ValueRelation, /* value map from an table @added in 1.8 */
      UuidGenerator, /* uuid generator - readonly and automatically intialized @added in 1.9 */
      Photo,         /* phote widget @added in 1.9 */
      WebView,       /* webview widget @added in 1.9 */
      Color,         /* color @added in 1.9 */
    };

    struct RangeData
    {
      RangeData() {}
      RangeData( QVariant theMin, QVariant theMax, QVariant theStep )
          : mMin( theMin ), mMax( theMax ), mStep( theStep ) {}

      QVariant mMin;
      QVariant mMax;
      QVariant mStep;
    };

    struct ValueRelationData
    {
      ValueRelationData() {}
      ValueRelationData( QString layer, QString key, QString value, bool allowNull, bool orderByValue,
                         bool allowMulti = false,
                         QString filterExpression = QString::null )
          : mLayer( layer )
          , mKey( key )
          , mValue( value )
          , mFilterExpression( filterExpression )
          , mAllowNull( allowNull )
          , mOrderByValue( orderByValue )
          , mAllowMulti( allowMulti )
      {}

      QString mLayer;
      QString mKey;
      QString mValue;
      QString mFilterExpression;
      bool mAllowNull;
      bool mOrderByValue;
      bool mAllowMulti;  /* allow selection of multiple keys @added in 1.9 */
    };

    struct GroupData
    {
      GroupData() {}
      GroupData( QString name , QList<QString> fields )
          : mName( name ), mFields( fields ) {}
      QString mName;
      QList<QString> mFields;
    };

    struct TabData
    {
      TabData() {}
      TabData( QString name , QList<QString> fields , QList<GroupData> groups )
          : mName( name ), mFields( fields ), mGroups( groups ) {}
      QString mName;
      QList<QString> mFields;
      QList<GroupData> mGroups;
    };

    QgsVectorLayer( QString path = QString::null, QString baseName = QString::null,
                    QString providerLib = QString::null, bool loadDefaultStyleFlag = true );

    virtual ~QgsVectorLayer();

    QString storageType() const;

    QString capabilitiesString() const;

    QString dataComment() const;

    void setDisplayField( QString fldName = "" );

    const QString displayField() const;

    void setDisplayExpression( const QString displayExpression );

    const QString displayExpression();

    QgsVectorDataProvider* dataProvider();

    const QgsVectorDataProvider* dataProvider() const;

    void setProviderEncoding( const QString& encoding );

    void setCoordinateSystem();

    void addJoin( const QgsVectorJoinInfo& joinInfo );

    void removeJoin( const QString& joinLayerId );

    const QList< QgsVectorJoinInfo >& vectorJoins() const;

    QgsLabel *label();

    const QgsLabel *label() const;

    QgsAttributeAction *actions() { return mActions; }

    int selectedFeatureCount();

    void select( QgsRectangle & rect, bool addToSelection );

    void modifySelection( QgsFeatureIds selectIds, QgsFeatureIds deselectIds );

    void invertSelection();

    void selectAll();

    void invertSelectionInRectangle( QgsRectangle & rect );

    QgsFeatureList selectedFeatures();

    const QgsFeatureIds &selectedFeaturesIds() const;

    void setSelectedFeatures( const QgsFeatureIds &ids );

    QgsRectangle boundingBoxOfSelected();

    void setDiagramRenderer( QgsDiagramRendererV2* r );
    const QgsDiagramRendererV2* diagramRenderer() const { return mDiagramRenderer; }

    void setDiagramLayerSettings( const QgsDiagramLayerSettings& s );
    const QgsDiagramLayerSettings *diagramLayerSettings() const { return mDiagramLayerSettings; }

    QgsFeatureRendererV2* rendererV2();
    void setRendererV2( QgsFeatureRendererV2* r );

    void drawRendererV2( QgsFeatureIterator &fit, QgsRenderContext& rendererContext, bool labeling );

    void drawRendererV2Levels( QgsFeatureIterator &fit, QgsRenderContext& rendererContext, bool labeling );

    QGis::GeometryType geometryType() const;

    bool hasGeometryType() const;

    QGis::WkbType wkbType() const;

    QString providerType() const;

    virtual bool readXml( const QDomNode& layer_node );

    virtual bool writeXml( QDomNode & layer_node, QDomDocument & doc );

    virtual void saveStyleToDatabase( QString name, QString description,
                                      bool useAsDefault, QString uiFileContent,
                                      QString &msgError );

    virtual int listStylesInDatabase( QStringList &ids, QStringList &names,
                                      QStringList &descriptions, QString &msgError );

    virtual QString getStyleFromDatabase( QString styleId, QString &msgError );

    virtual QString loadNamedStyle( const QString theURI, bool &theResultFlag, bool loadFromLocalDb );

    virtual QString loadNamedStyle( const QString theURI, bool &theResultFlag );

    virtual bool applyNamedStyle( QString namedStyle , QString errorMsg );

    QgsAttributeEditorElement* attributeEditorElementFromDomElement( QDomElement &elem, QObject* parent );

    bool readSymbology( const QDomNode& node, QString& errorMessage );

    bool writeSymbology( QDomNode& node, QDomDocument& doc, QString& errorMessage ) const;

    bool writeSld( QDomNode& node, QDomDocument& doc, QString& errorMessage ) const;
    bool readSld( const QDomNode& node, QString& errorMessage );

    virtual long featureCount() const;

    long featureCount( QgsSymbolV2* symbol );

    bool countSymbolFeatures( bool showProgress = true );

    virtual bool setSubsetString( QString subset );

    virtual QString subsetString();

    QgsFeatureIterator getFeatures( const QgsFeatureRequest& request = QgsFeatureRequest() );

    bool addFeature( QgsFeature& f, bool alsoUpdateExtent = true );

    bool updateFeature( QgsFeature &f );

    bool insertVertex( double x, double y, QgsFeatureId atFeatureId, int beforeVertex );

    bool moveVertex( double x, double y, QgsFeatureId atFeatureId, int atVertex );

    bool deleteVertex( QgsFeatureId atFeatureId, int atVertex );

    bool deleteSelectedFeatures();

    int addRing( const QList<QgsPoint>& ring );

    int addPart( const QList<QgsPoint>& ring );

    int translateFeature( QgsFeatureId featureId, double dx, double dy );

    int splitFeatures( const QList<QgsPoint>& splitLine, bool topologicalEditing = false );

    int removePolygonIntersections( QgsGeometry* geom, QgsFeatureIds ignoreFeatures = QgsFeatureIds() );

    int addTopologicalPoints( QgsGeometry* geom );

    int addTopologicalPoints( const QgsPoint& p );

    int insertSegmentVerticesForSnap( const QList<QgsSnappingResult>& snapResults );

    void enableLabels( bool on );

    bool hasLabelsEnabled() const;

    virtual bool isEditable() const;

    virtual bool isReadOnly() const;

    virtual bool isModified() const;

    bool snapPoint( QgsPoint& point, double tolerance );

    int snapWithContext( const QgsPoint& startPoint,
                         double snappingTolerance,
                         QMultiMap < double, QgsSnappingResult > &snappingResults,
                         QgsSnapper::SnappingType snap_to );

    virtual void reload();

    bool draw( QgsRenderContext& rendererContext );

    void drawLabels( QgsRenderContext& rendererContext );

    QgsRectangle extent();

    const QgsFields &pendingFields() const;

    QgsAttributeList pendingAllAttributesList();

    QgsAttributeList pendingPkAttributesList();

    int pendingFeatureCount();

    bool setReadOnly( bool readonly = true );

    bool startEditing();

    bool changeGeometry( QgsFeatureId fid, QgsGeometry* geom );

    bool changeAttributeValue( QgsFeatureId fid, int field, QVariant value, bool emitSignal = true );

    bool addAttribute( const QgsField &field );

    void addAttributeAlias( int attIndex, QString aliasString );

    void addAttributeEditorWidget( QgsAttributeEditorElement* data );
    QList< QgsAttributeEditorElement* > &attributeEditorElements();
    void clearAttributeEditorWidgets();

    QString attributeAlias( int attributeIndex ) const;

    QString attributeDisplayName( int attributeIndex ) const;

    const QMap< QString, QString >& attributeAliases() const { return mAttributeAliasMap; }

    const QSet<QString>& excludeAttributesWMS() const { return mExcludeAttributesWMS; }
    void setExcludeAttributesWMS( const QSet<QString>& att ) { mExcludeAttributesWMS = att; }

    const QSet<QString>& excludeAttributesWFS() const { return mExcludeAttributesWFS; }
    void setExcludeAttributesWFS( const QSet<QString>& att ) { mExcludeAttributesWFS = att; }

    bool deleteAttribute( int attr );

    bool deleteAttributes( QList<int> attrs );

    bool addFeatures( QgsFeatureList features, bool makeSelected = true );

    bool deleteFeature( QgsFeatureId fid );

    bool commitChanges();
    const QStringList &commitErrors();

    bool rollBack( bool deleteBuffer = true );

    EditType editType( int idx );

    void setEditType( int idx, EditType edit );

    EditorLayout editorLayout();

    void setEditorLayout( EditorLayout editorLayout );

    void setCheckedState( int idx, QString checked, QString notChecked );

    QPair<QString, QString> checkedState( int idx );

    QString editForm();

    void setEditForm( QString ui );

    QString annotationForm() const { return mAnnotationForm; }

    void setAnnotationForm( const QString& ui );

    QString editFormInit();

    void setEditFormInit( QString function );

    QMap<QString, QVariant> &valueMap( int idx );

    RangeData &range( int idx );

    ValueRelationData &valueRelation( int idx );

    QString &dateFormat( int idx );

    QSize &widgetSize( int idx );

    bool fieldEditable( int idx );

    bool labelOnTop( int idx );

    void setFieldEditable( int idx, bool editable );

    void setLabelOnTop( int idx, bool onTop );

    void addOverlay( QgsVectorOverlay* overlay );

    void removeOverlay( const QString& typeName );

    void vectorOverlays( QList<QgsVectorOverlay*>& overlayList );

    QgsVectorOverlay* findOverlayByType( const QString& typeName );

    QgsVectorLayerEditBuffer* editBuffer() { return mEditBuffer; }

    void beginEditCommand( QString text );

    void endEditCommand();

    void destroyEditCommand();

    int fieldNameIndex( const QString& fieldName ) const;

    enum VertexMarkerType
    {
      SemiTransparentCircle,
      Cross,
      NoMarker  /* added in version 1.1 */
    };

    static void drawVertexMarker( double x, double y, QPainter& p, QgsVectorLayer::VertexMarkerType type, int vertexSize );

    void updateFields();

    void createJoinCaches();

    void uniqueValues( int index, QList<QVariant> &uniqueValues, int limit = -1 );

    QVariant minimumValue( int index );

    QVariant maximumValue( int index );

    /* Set the blending mode used for rendering each feature
     * @note added in 2.0
     */
    void setFeatureBlendMode( const QPainter::CompositionMode blendMode );
    /* Returns the current blending mode for features
     * @note added in 2.0
     */
    QPainter::CompositionMode featureBlendMode() const;

    /* Set the transparency for the vector layer
     * @note added in 2.0
     */
    void setLayerTransparency( int layerTransparency );
    /* Returns the current transparency for the vector layer
     * @note added in 2.0
     */
    int layerTransparency() const;

  public slots:
    void select( const QgsFeatureId &featureId );

    void select( const QgsFeatureIds& featureIds );

    void deselect( const QgsFeatureId featureId );

    void deselect( const QgsFeatureIds& featureIds );

    void removeSelection();

    void triggerRepaint();

    virtual void updateExtents();

    void checkJoinLayerRemove( QString theLayerId );

    QString metadata();

    inline QgsGeometryCache* cache() { return mCache; }

    virtual void onCacheImageDelete();

  signals:

    void selectionChanged( const QgsFeatureIds selected, const QgsFeatureIds deselected, const bool clearAndSelect );

    void selectionChanged();

    void layerModified();

    void editingStarted();

    void editingStopped();

    void beforeCommitChanges();

    void beforeRollBack();

    void attributeAdded( int idx );
    void attributeDeleted( int idx );
    void featureAdded( QgsFeatureId fid );  // added in 1.7
    void featureDeleted( QgsFeatureId fid );
    void updatedFields();
    void layerDeleted();

    void attributeValueChanged( QgsFeatureId fid, int idx, const QVariant & );
    void geometryChanged( QgsFeatureId fid, QgsGeometry &geom ); // added in 1.9

    void committedAttributesDeleted( const QString& layerId, const QgsAttributeList& deletedAttributes );
    void committedAttributesAdded( const QString& layerId, const QList<QgsField>& addedAttributes );
    void committedFeaturesAdded( const QString& layerId, const QgsFeatureList& addedFeatures );
    void committedFeaturesRemoved( const QString& layerId, const QgsFeatureIds& deletedFeatureIds );
    void committedAttributeValuesChanges( const QString& layerId, const QgsChangedAttributesMap& changedAttributesValues );
    void committedGeometriesChanges( const QString& layerId, const QgsGeometryMap& changedGeometries );

    void labelingFontNotFound( QgsVectorLayer* layer, const QString& fontfamily );

  protected:
    void setExtent( const QgsRectangle &rect );

  private:                       // Private methods

    QgsVectorLayer( const QgsVectorLayer & rhs );

    QgsVectorLayer & operator=( QgsVectorLayer const & rhs );

    bool setDataProvider( QString const & provider );

    QgsFeatureId findFreeId();

    void snapToGeometry( const QgsPoint& startPoint,
                         QgsFeatureId featureId,
                         QgsGeometry* geom,
                         double sqrSnappingTolerance,
                         QMultiMap<double, QgsSnappingResult>& snappingResults,
                         QgsSnapper::SnappingType snap_to ) const;

    static QgsVectorLayer::VertexMarkerType currentVertexMarkerType();

    static int currentVertexMarkerSize();

    //void addJoinedAttributes( QgsFeature& f, bool all = false );

    void stopRendererV2( QgsRenderContext& rendererContext, QgsSingleSymbolRendererV2* selRenderer );

    void prepareLabelingAndDiagrams( QgsRenderContext& rendererContext, QgsAttributeList& attributes, bool& labeling );

  private:                       // Private attributes

    int mUpdateThreshold;

    bool mEnableBackbuffer;

    QgsVectorDataProvider *mDataProvider;

    QgsFeatureIterator mProviderIterator;

    QString mDisplayField;

    QString mDisplayExpression;

    QString mProviderKey;

    QgsAttributeAction* mActions;

    bool mReadOnly;

    QgsFeatureIds mSelectedFeatureIds;

    QgsFields mUpdatedFields;

    QMap< QString, QString > mAttributeAliasMap;

    QList< QgsAttributeEditorElement* > mAttributeEditorElements;

    QSet<QString> mExcludeAttributesWMS;
    QSet<QString> mExcludeAttributesWFS;

    QList< TabData > mTabs;

    int mWkbType;

    QgsFeatureRendererV2 *mRendererV2;

    QgsLabel *mLabel;

    bool mLabelOn;

    bool mLabelFontNotFoundNotified;

    QPainter::CompositionMode mFeatureBlendMode;

    int mLayerTransparency;

    QgsVectorLayer::VertexMarkerType mCurrentVertexMarkerType;

    int mCurrentVertexMarkerSize;

    bool mVertexMarkerOnlyForSelection;

    QList<QgsVectorOverlay*> mOverlays;

    QStringList mCommitErrors;

    QMap< QString, EditType > mEditTypes;
    QMap< QString, bool> mFieldEditables;
    QMap< QString, bool> mLabelOnTop;
    QMap< QString, QMap<QString, QVariant> > mValueMaps;
    QMap< QString, RangeData > mRanges;
    QMap< QString, QPair<QString, QString> > mCheckedStates;
    QMap< QString, ValueRelationData > mValueRelations;
    QMap< QString, QString> mDateFormats;
    QMap< QString, QSize> mWidgetSize;

    EditorLayout mEditorLayout;

    QString mEditForm, mEditFormInit;
    //annotation form for this layer
    QString mAnnotationForm;

    QgsGeometryCache* mCache;

    QgsVectorLayerEditBuffer* mEditBuffer;
    friend class QgsVectorLayerEditBuffer;

    //stores information about joined layers
    QgsVectorLayerJoinBuffer* mJoinBuffer;

    //diagram rendering object. 0 if diagram drawing is disabled
    QgsDiagramRendererV2* mDiagramRenderer;

    //stores infos about diagram placement (placement type, priority, position distance)
    QgsDiagramLayerSettings *mDiagramLayerSettings;

    bool mValidExtent;

    // Features in renderer classes counted
    bool mSymbolFeatureCounted;

    // Feature counts for each renderer symbol
    QMap<QgsSymbolV2*, long> mSymbolFeatureCountMap;

    QgsRenderContext* mCurrentRendererContext;

    friend class QgsVectorLayerFeatureIterator;
};

#endif