qgspointcloudattributemodel.h

Go to the documentation of this file.

/***************************************************************************
       qgspointcloudattributemodel.h
                         ---------------------
    begin                : November 2020
    copyright            : (C) 2020 by Nyall Dawson
    email                : nyall dot dawson 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 QGSPOINTCLOUDATTRIBUTEMODEL_H
#define QGSPOINTCLOUDATTRIBUTEMODEL_H
 
#include "qgspointcloudattribute.h"
 
#include <QAbstractItemModel>
#include <QPointer>
#include <QSortFilterProxyModel>
 
class QgsPointCloudLayer;
 
class CORE_EXPORT QgsPointCloudAttributeModel : public QAbstractItemModel
{
    Q_OBJECT
 
  public:
 
    enum FieldRoles
    {
      AttributeNameRole = Qt::UserRole + 1,  
      AttributeIndexRole, 
      AttributeSizeRole, 
      AttributeTypeRole, 
      IsEmptyRole, 
      IsNumericRole, 
    };
 
    explicit QgsPointCloudAttributeModel( QObject *parent SIP_TRANSFERTHIS = nullptr );
 
    void setLayer( QgsPointCloudLayer *layer );
 
    QgsPointCloudLayer *layer();
 
    void setAttributes( const QgsPointCloudAttributeCollection &attributes );
 
    QgsPointCloudAttributeCollection attributes() const { return mAttributes; }
 
    void setAllowEmptyAttributeName( bool allowEmpty );
 
    QModelIndex indexFromName( const QString &name );
 
    bool allowEmptyAttributeName() const { return mAllowEmpty; }
 
    // QAbstractItemModel interface
    QModelIndex index( int row, int column, const QModelIndex &parent = QModelIndex() ) const override;
    QModelIndex parent( const QModelIndex &child ) const override;
    int rowCount( const QModelIndex &parent = QModelIndex() ) const override;
    int columnCount( const QModelIndex &parent ) const override;
    QVariant data( const QModelIndex &index, int role ) const override;
 
    static QString attributeToolTip( const QgsPointCloudAttribute &attribute );
 
    static QIcon iconForAttributeType( QgsPointCloudAttribute::DataType type );
 
  private:
 
    QgsPointCloudAttributeCollection mAttributes;
    bool mAllowEmpty = false;
    QPointer< QgsPointCloudLayer > mLayer;
};
 
 
class CORE_EXPORT QgsPointCloudAttributeProxyModel : public QSortFilterProxyModel
{
    Q_OBJECT
  public:
 
    enum Filter
    {
      Char = 1 << 0, 
      Short = 1 << 1, 
      Int32 = 1 << 2, 
      Float = 1 << 3, 
      Double = 1 << 4, 
      Numeric = Short | Int32 | Float | Double, 
      AllTypes = Numeric | Char, 
    };
    Q_DECLARE_FLAGS( Filters, Filter )
    Q_FLAG( Filters )
 
    
    explicit QgsPointCloudAttributeProxyModel( QgsPointCloudAttributeModel *source, QObject *parent SIP_TRANSFERTHIS = nullptr );
 
    QgsPointCloudAttributeModel *sourceAttributeModel() { return mModel; }
 
    QgsPointCloudAttributeProxyModel *setFilters( QgsPointCloudAttributeProxyModel::Filters filters );
 
    Filters filters() const { return mFilters; }
 
  private:
 
    QgsPointCloudAttributeModel *mModel = nullptr;
    Filters mFilters = AllTypes;
 
    // QSortFilterProxyModel interface
  public:
    bool filterAcceptsRow( int source_row, const QModelIndex &source_parent ) const override;
    bool lessThan( const QModelIndex &left, const QModelIndex &right ) const override;
};
Q_DECLARE_OPERATORS_FOR_FLAGS( QgsPointCloudAttributeProxyModel::Filters )
 
 
#endif // QGSPOINTCLOUDATTRIBUTEMODEL_H