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:
 
    // *INDENT-OFF*

    enum class CustomRole SIP_MONKEYPATCH_SCOPEENUM_UNNEST( QgsPointCloudAttributeModel, FieldRoles ) : int
    {
      AttributeName SIP_MONKEYPATCH_COMPAT_NAME(AttributeNameRole)= Qt::UserRole + 1,  
      AttributeIndex SIP_MONKEYPATCH_COMPAT_NAME(AttributeIndexRole), 
      AttributeSize SIP_MONKEYPATCH_COMPAT_NAME(AttributeSizeRole), 
      AttributeType SIP_MONKEYPATCH_COMPAT_NAME(AttributeTypeRole), 
      IsEmpty SIP_MONKEYPATCH_COMPAT_NAME(IsEmptyRole), 
      IsNumeric SIP_MONKEYPATCH_COMPAT_NAME(IsNumericRole), 
    };
    Q_ENUM( CustomRole )
    // *INDENT-ON*
 
    
    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 SIP_ENUM_BASETYPE( IntFlag )
    {
      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