qgsmaplayerstylecategoriesmodel.h

Go to the documentation of this file.

/***************************************************************************
  qgsmaplayerstylecategoriesmodel.h
  --------------------------------------
  Date                 : September 2018
  Copyright            : (C) 2018 by Denis Rouzaud
  Email                : [email protected]
 ***************************************************************************
 *                                                                         *
 *   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 QGSMAPLAYERSTYLECATEGORIESMODEL_H
#define QGSMAPLAYERSTYLECATEGORIESMODEL_H
 
#include "qgis.h"
#include "qgis_gui.h"
#include "qgsmaplayer.h"
 
#include <QAbstractListModel>
#include <QItemDelegate>
#include <QLabel>

class GUI_EXPORT QgsMapLayerStyleCategoriesModel : public QAbstractListModel
{
    Q_OBJECT
 
  public:
    enum class Role : int
    {
      NameRole = Qt::UserRole + 1,
    };

    explicit QgsMapLayerStyleCategoriesModel( Qgis::LayerType type, QObject *parent = nullptr );

    void setCategories( QgsMapLayer::StyleCategories categories );

    QgsMapLayer::StyleCategories categories() const;

    void setShowAllCategories( bool showAll );
 
    int rowCount( const QModelIndex & = QModelIndex() ) const override;
    int columnCount( const QModelIndex & = QModelIndex() ) const override;
    QVariant data( const QModelIndex &index, int role ) const override;
    bool setData( const QModelIndex &index, const QVariant &value, int role ) override;
    Qt::ItemFlags flags( const QModelIndex & ) const override;
 
  private:
    Qgis::LayerType mLayerType = Qgis::LayerType::Vector;
    QgsMapLayer::StyleCategories mCategories;
    QList<QgsMapLayer::StyleCategory> mCategoryList;
    bool mShowAllCategories = false;
};

class GUI_EXPORT QgsCategoryDisplayLabelDelegate : public QItemDelegate
{
    Q_OBJECT
 
  public:
    explicit QgsCategoryDisplayLabelDelegate( QObject *parent = nullptr );
 
  protected:
    void drawDisplay( QPainter *painter, const QStyleOptionViewItem &option, const QRect &rect, const QString &text ) const override;
    QSize sizeHint( const QStyleOptionViewItem &option, const QModelIndex &index ) const override;
};
 
#endif // QGSMAPLAYERSTYLECATEGORIESMODEL_H