qgsattributetablemodel.h

Go to the documentation of this file.

/***************************************************************************
  QgsAttributeTableModel.h - Models for attribute table
  -------------------
         date                 : Feb 2009
         copyright            : Vita Cizek
         email                : weetya (at) gmail.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 QGSATTRIBUTETABLEMODEL_H
#define QGSATTRIBUTETABLEMODEL_H

#include <QAbstractTableModel>
#include <QModelIndex>
#include <QObject>
#include <QHash>
#include <QQueue>
#include <QMap>

#include "qgsvectorlayer.h" // QgsAttributeList
#include "qgsvectorlayercache.h"
#include "qgsconditionalstyle.h"
#include "qgsattributeeditorcontext.h"

class QgsMapCanvas;
class QgsMapLayerAction;
class QgsEditorWidgetFactory;


class GUI_EXPORT QgsAttributeTableModel: public QAbstractTableModel
{
    Q_OBJECT

  public:
    enum Role
    {
      SortRole = Qt::UserRole + 1, 
      FeatureIdRole,               
      FieldIndexRole,              
      UserRole                     
    };

  public:
    QgsAttributeTableModel( QgsVectorLayerCache *layerCache, QObject *parent = nullptr );

    virtual int rowCount( const QModelIndex &parent = QModelIndex() ) const override;

    int columnCount( const QModelIndex &parent = QModelIndex() ) const override;

    QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const override;

    virtual QVariant data( const QModelIndex &index, int role ) const override;

    virtual bool setData( const QModelIndex &index, const QVariant &value, int role = Qt::EditRole ) override;

    Qt::ItemFlags flags( const QModelIndex &index ) const override;

    void reload( const QModelIndex &index1, const QModelIndex &index2 );

    bool removeRows( int row, int count, const QModelIndex &parent = QModelIndex() ) override;

    inline void resetModel() { loadLayer(); }

    int idToRow( QgsFeatureId id ) const;

    QModelIndex idToIndex( QgsFeatureId id ) const;

    QModelIndexList idToIndexList( QgsFeatureId id ) const;

    int fieldIdx( int col ) const;

    int fieldCol( int idx ) const;

    QgsFeatureId rowToId( int row ) const;

    void swapRows( QgsFeatureId a, QgsFeatureId b );

    inline QgsVectorLayer* layer() const { return mLayerCache ? mLayerCache->layer() : nullptr; }

    inline QgsVectorLayerCache* layerCache() const { return mLayerCache; }

    void executeAction( int action, const QModelIndex &idx ) const;

    void executeMapLayerAction( QgsMapLayerAction* action, const QModelIndex &idx ) const;

    QgsFeature feature( const QModelIndex &idx ) const;

    void prefetchColumnData( int column );

    void prefetchSortData( const QString& expression );

    QString sortCacheExpression() const;

    void setRequest( const QgsFeatureRequest& request );

    // TODO QGIS 3: return copy instead of reference
    const QgsFeatureRequest& request() const;

    void setEditorContext( const QgsAttributeEditorContext& context ) { mEditorContext = context; }

    const QgsAttributeEditorContext& editorContext() const { return mEditorContext; }

    int extraColumns() const;

    void setExtraColumns( int extraColumns );

  public slots:
    virtual void loadLayer();

    void fieldConditionalStyleChanged( const QString& fieldName );

  signals:
    void modelChanged();

    void progress( int i, bool &cancel );
    void finished();

  private slots:
    virtual void updatedFields();

    virtual void editCommandEnded();

    virtual void attributeDeleted( int idx );

  protected slots:
    virtual void attributeValueChanged( QgsFeatureId fid, int idx, const QVariant &value );
    virtual void featuresDeleted( const QgsFeatureIds& fids );
    virtual void featureAdded( QgsFeatureId fid, bool resettingModel = false );

    virtual void layerDeleted();

  protected:
    QgsVectorLayerCache *mLayerCache;
    int mFieldCount;

    mutable QgsFeature mFeat;

    QgsAttributeList mAttributes;
    QVector<QgsEditorWidgetFactory*> mWidgetFactories;
    QVector<QVariant> mAttributeWidgetCaches;
    QVector<QgsEditorWidgetConfig> mWidgetConfigs;

    QHash<QgsFeatureId, int> mIdRowMap;
    QHash<int, QgsFeatureId> mRowIdMap;
    mutable QHash<int, QList<QgsConditionalStyle> > mRowStylesMap;

    mutable QgsExpressionContext mExpressionContext;

    virtual void loadAttributes();

  private:
    virtual bool loadFeatureAtId( QgsFeatureId fid ) const;

    QgsFeatureRequest mFeatureRequest;

    QgsExpression mSortCacheExpression;
    QgsAttributeList mSortCacheAttributes;
    int mSortFieldIndex;
    QHash<QgsFeatureId, QVariant> mSortCache;

    QRect mChangedCellBounds;

    QgsAttributeEditorContext mEditorContext;

    int mExtraColumns;
};


#endif