src/core/symbology-ng/qgsstylev2.h

Go to the documentation of this file.

/***************************************************************************
    qgsstylev2.h
    ---------------------
    begin                : November 2009
    copyright            : (C) 2009 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 QGSSTYLEV2_H
#define QGSSTYLEV2_H

#include <QMap>
#include <QMultiMap>
#include <QString>

#include <sqlite3.h>

#include "qgssymbollayerv2utils.h" // QgsStringMap

class QgsSymbolV2;
class QgsSymbolLayerV2;
class QgsVectorColorRampV2;

class QDomDocument;
class QDomElement;

typedef QMap<QString, QgsVectorColorRampV2* > QgsVectorColorRampV2Map;
typedef QMap<int, QString> QgsSymbolGroupMap;

typedef QMultiMap<QString, QString> QgsSmartConditionMap;

// enumerators representing sqlite DB columns
enum SymbolTable { SymbolId, SymbolName, SymbolXML, SymbolGroupId };
enum SymgroupTable { SymgroupId, SymgroupName, SymgroupParent };
enum TagTable { TagId, TagName };
enum TagmapTable { TagmapTagId, TagmapSymbolId };
enum ColorrampTable { ColorrampId, ColorrampName, ColorrampXML, ColorrampGroupId };
enum SmartgroupTable { SmartgroupId, SmartgroupName, SmartgroupXML };

class CORE_EXPORT QgsStyleV2
{
  public:
    QgsStyleV2();
    ~QgsStyleV2();


    enum StyleEntity { SymbolEntity, GroupEntity, TagEntity, ColorrampEntity, SmartgroupEntity };


    bool addColorRamp( QString name, QgsVectorColorRampV2* colorRamp, bool update = false );


    int addGroup( QString groupName, int parent = 0 );


    int addSmartgroup( QString name, QString op, QgsSmartConditionMap conditions );


    bool addSymbol( QString name, QgsSymbolV2* symbol, bool update = false );


    int addTag( QString tagName );

    QgsSymbolGroupMap childGroupNames( QString parent = "" );

    void clear();

    QgsVectorColorRampV2* colorRamp( QString name );

    int colorRampCount();

    QStringList colorRampNames();

    const QgsVectorColorRampV2* colorRampRef( QString name ) const;

    int colorrampId( QString name );

    static QgsStyleV2* defaultStyle();


    bool tagSymbol( StyleEntity type, QString symbol, QStringList tags );


    bool detagSymbol( StyleEntity type, QString symbol, QStringList tags );

    bool removeSymbol( QString name );

    bool renameSymbol( QString oldName, QString newName );

    QgsSymbolV2* symbol( QString name );

    const QgsSymbolV2* symbolRef( QString name ) const;

    int symbolCount();

    QStringList symbolNames();

    int symbolId( QString name );
    int groupId( QString group );
    int tagId( QString tag );
    int smartgroupId( QString smartgroup );

    QStringList groupNames();


    QStringList symbolsOfGroup( StyleEntity type, int groupid );


    QStringList symbolsWithTag( StyleEntity type, int tagid );


    bool group( StyleEntity type, QString name, int groupid );


    void rename( StyleEntity type, int id, QString newName );


    void remove( StyleEntity type, int id );


    bool saveSymbol( QString name, QgsSymbolV2* symbol, int groupid, QStringList tags );


    bool saveColorRamp( QString name, QgsVectorColorRampV2* ramp, int groupid, QStringList tags );

    bool removeColorRamp( QString name );

    bool renameColorRamp( QString oldName, QString newName );


    bool load( QString filename );

    bool save( QString filename = QString() );

    QString errorString() { return mErrorString; }

    QString fileName() { return mFileName; }

    QStringList findSymbols( QString qword );


    QStringList tagsOfSymbol( StyleEntity type, QString symbol );

    QgsSymbolGroupMap smartgroupsListMap();

    QStringList smartgroupNames();

    QgsSmartConditionMap smartgroup( int id );

    //clumsy implementation TODO create a class for smartgroups
    QString smartgroupOperator( int id );

    QStringList symbolsOfSmartgroup( StyleEntity type, int id );

    bool exportXML( QString filename );

    bool importXML( QString filename );

  protected:

    QgsSymbolV2Map mSymbols;
    QgsVectorColorRampV2Map mColorRamps;

    QString mErrorString;
    QString mFileName;

    sqlite3* mCurrentDB;

    static QgsStyleV2* mDefaultStyle;

    bool openDB( QString filename );

    bool runEmptyQuery( char* query, bool freeQuery = true );

    char* getGroupRemoveQuery( int id );

    int getId( QString table, QString name );


    bool updateSymbol( StyleEntity type, QString name );
};


#endif