qgsmapthemecollection.h

Go to the documentation of this file.

/***************************************************************************
  qgsmapthemecollection.h
  --------------------------------------
  Date                 : September 2014
  Copyright            : (C) 2014 by Martin Dobias
  Email                : wonder dot sk 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 QGSMAPTHEMECOLLECTION_H
#define QGSMAPTHEMECOLLECTION_H

#include "qgis_core.h"
#include "qgis_sip.h"
#include <QMap>
#include <QObject>
#include <QPointer>
#include <QSet>
#include <QStringList>

#include "qgsmaplayer.h"

class QDomDocument;
class QgsLayerTreeModel;
class QgsLayerTreeNode;
class QgsLayerTreeGroup;
class QgsLayerTreeLayer;
class QgsProject;

class CORE_EXPORT QgsMapThemeCollection : public QObject
{
    Q_OBJECT

    Q_PROPERTY( QStringList mapThemes READ mapThemes NOTIFY mapThemesChanged )
    Q_PROPERTY( QgsProject *project READ project WRITE setProject NOTIFY projectChanged )

  public:

    class CORE_EXPORT MapThemeLayerRecord
    {
      public:
        MapThemeLayerRecord( QgsMapLayer *l = nullptr ): mLayer( l ) {}

        bool operator==( const QgsMapThemeCollection::MapThemeLayerRecord &other ) const
        {
          return mLayer == other.mLayer && isVisible == other.isVisible &&
                 usingCurrentStyle == other.usingCurrentStyle && currentStyle == other.currentStyle &&
                 usingLegendItems == other.usingLegendItems && checkedLegendItems == other.checkedLegendItems &&
                 expandedLegendItems == other.expandedLegendItems && expandedLayerNode == other.expandedLayerNode;
        }
        bool operator!=( const QgsMapThemeCollection::MapThemeLayerRecord &other ) const
        {
          return !( *this == other );
        }

        QgsMapLayer *layer() const { return mLayer; }

        void setLayer( QgsMapLayer *layer );

        bool isVisible = true;

        bool usingCurrentStyle = false;
        QString currentStyle;
        bool usingLegendItems = false;
        QSet<QString> checkedLegendItems;

        QSet<QString> expandedLegendItems;

        bool expandedLayerNode = false;
      private:
        QgsWeakMapLayerPointer mLayer;
    };

    class CORE_EXPORT MapThemeRecord
    {
      public:

        bool operator==( const QgsMapThemeCollection::MapThemeRecord &other ) const
        {
          return validLayerRecords() == other.validLayerRecords() &&
                 mHasExpandedStateInfo == other.mHasExpandedStateInfo &&
                 mExpandedGroupNodes == other.mExpandedGroupNodes && mCheckedGroupNodes == other.mCheckedGroupNodes;
        }
        bool operator!=( const QgsMapThemeCollection::MapThemeRecord &other ) const
        {
          return !( *this == other );
        }

        QList<QgsMapThemeCollection::MapThemeLayerRecord> layerRecords() const { return mLayerRecords; }

        void setLayerRecords( const QList<QgsMapThemeCollection::MapThemeLayerRecord> &records ) { mLayerRecords = records; }

        void removeLayerRecord( QgsMapLayer *layer );

        void addLayerRecord( const QgsMapThemeCollection::MapThemeLayerRecord &record );

        QHash<QgsMapLayer *, QgsMapThemeCollection::MapThemeLayerRecord> validLayerRecords() const SIP_SKIP;

        bool hasExpandedStateInfo() const { return mHasExpandedStateInfo; }

        bool hasCheckedStateInfo() const { return mHasCheckedStateInfo; } SIP_SKIP;

        void setHasExpandedStateInfo( bool hasInfo ) { mHasExpandedStateInfo = hasInfo; }

        void setHasCheckedStateInfo( bool hasInfo ) { mHasCheckedStateInfo = hasInfo; } SIP_SKIP;

        QSet<QString> expandedGroupNodes() const { return mExpandedGroupNodes; }

        QSet<QString> checkedGroupNodes() const { return mCheckedGroupNodes; }

        void setExpandedGroupNodes( const QSet<QString> &expandedGroupNodes ) { mExpandedGroupNodes = expandedGroupNodes; }

        void setCheckedGroupNodes( const QSet<QString> &checkedGroupNodes ) { mCheckedGroupNodes = checkedGroupNodes; }

      private:
        QList<MapThemeLayerRecord> mLayerRecords;

        bool mHasExpandedStateInfo = false;
        bool mHasCheckedStateInfo = false;

        QSet<QString> mExpandedGroupNodes;

        QSet<QString> mCheckedGroupNodes;

        friend class QgsMapThemeCollection;
    };

    QgsMapThemeCollection( QgsProject *project = nullptr );

    bool hasMapTheme( const QString &name ) const;

    void insert( const QString &name, const QgsMapThemeCollection::MapThemeRecord &state );

    void update( const QString &name, const QgsMapThemeCollection::MapThemeRecord &state );

    void removeMapTheme( const QString &name );

    void clear();

    QStringList mapThemes() const;

    QgsMapThemeCollection::MapThemeRecord mapThemeState( const QString &name ) const { return mMapThemes[name]; }

    QStringList mapThemeVisibleLayerIds( const QString &name ) const;

    QList<QgsMapLayer *> mapThemeVisibleLayers( const QString &name ) const;

    QMap<QString, QString> mapThemeStyleOverrides( const QString &name );

    void readXml( const QDomDocument &doc );

    void writeXml( QDomDocument &doc );

    static QgsMapThemeCollection::MapThemeRecord createThemeFromCurrentState( QgsLayerTreeGroup *root, QgsLayerTreeModel *model );

    void applyTheme( const QString &name, QgsLayerTreeGroup *root, QgsLayerTreeModel *model );

    QgsProject *project();

    void setProject( QgsProject *project );

    QList< QgsMapLayer * > masterLayerOrder() const;

    QList< QgsMapLayer * > masterVisibleLayers() const;

  signals:

    void mapThemesChanged();

    void mapThemeChanged( const QString &theme );

    void projectChanged();

  private slots:

    void registryLayersRemoved( const QStringList &layerIDs );

    void layerStyleRenamed( const QString &oldName, const QString &newName );

  private:

    void applyMapThemeCheckedLegendNodesToLayer( const MapThemeLayerRecord &layerRec, QgsMapLayer *layer );

    void reconnectToLayersStyleManager();

    static bool findRecordForLayer( QgsMapLayer *layer, const MapThemeRecord &rec, MapThemeLayerRecord &layerRec );
    static MapThemeLayerRecord createThemeLayerRecord( QgsLayerTreeLayer *nodeLayer, QgsLayerTreeModel *model );
    static void createThemeFromCurrentState( QgsLayerTreeGroup *parent, QgsLayerTreeModel *model, MapThemeRecord &rec );
    static void applyThemeToLayer( QgsLayerTreeLayer *nodeLayer, QgsLayerTreeModel *model, const MapThemeRecord &rec );
    static void applyThemeToGroup( QgsLayerTreeGroup *parent, QgsLayerTreeModel *model, const MapThemeRecord &rec );

    typedef QMap<QString, MapThemeRecord> MapThemeRecordMap;
    MapThemeRecordMap mMapThemes;
    QgsProject *mProject = nullptr;
};


#endif // QGSMAPTHEMECOLLECTION_H