Branch data     Line data    Source code

       1                 :            : /***************************************************************************
       2                 :            :     qgsclassificationmethodregistry.h
       3                 :            :     ---------------------
       4                 :            :     begin                : September 2019
       5                 :            :     copyright            : (C) 2019 by Denis Rouzaud
       6                 :            :     email                : denis@opengis.ch
       7                 :            :  ***************************************************************************
       8                 :            :  *                                                                         *
       9                 :            :  *   This program is free software; you can redistribute it and/or modify  *
      10                 :            :  *   it under the terms of the GNU General Public License as published by  *
      11                 :            :  *   the Free Software Foundation; either version 2 of the License, or     *
      12                 :            :  *   (at your option) any later version.                                   *
      13                 :            :  *                                                                         *
      14                 :            :  ***************************************************************************/
      15                 :            : 
      16                 :            : #include <QIcon>
      17                 :            : 
      18                 :            : #include "qgsclassificationmethodregistry.h"
      19                 :            : 
      20                 :            : // classification methods
      21                 :            : #include "qgsclassificationcustom.h"
      22                 :            : #include "qgsclassificationequalinterval.h"
      23                 :            : #include "qgsclassificationquantile.h"
      24                 :            : #include "qgsclassificationjenks.h"
      25                 :            : #include "qgsclassificationstandarddeviation.h"
      26                 :            : #include "qgsclassificationprettybreaks.h"
      27                 :            : #include "qgsclassificationlogarithmic.h"
      28                 :            : 
      29                 :          5 : QgsClassificationMethodRegistry::QgsClassificationMethodRegistry()
      30                 :            : {
      31                 :          5 :   addMethod( new QgsClassificationEqualInterval() );
      32                 :          5 :   addMethod( new QgsClassificationQuantile() );
      33                 :          5 :   addMethod( new QgsClassificationJenks() );
      34                 :          5 :   addMethod( new QgsClassificationStandardDeviation() );
      35                 :          5 :   addMethod( new QgsClassificationPrettyBreaks() );
      36                 :          5 :   addMethod( new QgsClassificationLogarithmic() );
      37                 :          5 : }
      38                 :            : 
      39                 :          5 : QgsClassificationMethodRegistry::~QgsClassificationMethodRegistry()
      40                 :            : {
      41                 :          5 :   qDeleteAll( mMethods );
      42                 :          5 : }
      43                 :            : 
      44                 :         30 : bool QgsClassificationMethodRegistry::addMethod( QgsClassificationMethod *method )
      45                 :            : {
      46                 :         30 :   if ( mMethods.contains( method->id() ) )
      47                 :          0 :     return false;
      48                 :            : 
      49                 :         30 :   mMethods.insert( method->id(), method );
      50                 :         30 :   return true;
      51                 :         30 : }
      52                 :            : 
      53                 :          0 : QgsClassificationMethod *QgsClassificationMethodRegistry::method( const QString &id )
      54                 :            : {
      55                 :          0 :   QgsClassificationMethod *method = mMethods.value( id, new QgsClassificationCustom() );
      56                 :          0 :   return method->clone();
      57                 :          0 : }
      58                 :            : 
      59                 :          0 : QMap<QString, QString> QgsClassificationMethodRegistry::methodNames() const
      60                 :            : {
      61                 :          0 :   QMap<QString, QString> methods;
      62                 :          0 :   for ( const QgsClassificationMethod *method : std::as_const( mMethods ) )
      63                 :          0 :     methods.insert( method->name(), method->id() );
      64                 :          0 :   return methods;
      65                 :          0 : }
      66                 :            : 
      67                 :          0 : QIcon QgsClassificationMethodRegistry::icon( const QString &id ) const
      68                 :            : {
      69                 :          0 :   QgsClassificationMethod *method = mMethods.value( id, nullptr );
      70                 :          0 :   if ( method )
      71                 :          0 :     return method->icon();
      72                 :            :   else
      73                 :          0 :     return QIcon();
      74                 :          0 : }
      75                 :            :