Branch data     Line data    Source code

       1                 :            : /***************************************************************************
       2                 :            :                            qgscolorschemeregistry.cpp
       3                 :            :                              -------------------
       4                 :            :     begin                : July 2014
       5                 :            :     copyright            : (C) 2014 by Nyall Dawson
       6                 :            :     email                : nyall dot dawson at gmail dot com
       7                 :            :  ***************************************************************************/
       8                 :            : 
       9                 :            : /***************************************************************************
      10                 :            :  *                                                                         *
      11                 :            :  *   This program is free software; you can redistribute it and/or modify  *
      12                 :            :  *   it under the terms of the GNU General Public License as published by  *
      13                 :            :  *   the Free Software Foundation; either version 2 of the License, or     *
      14                 :            :  *   (at your option) any later version.                                   *
      15                 :            :  *                                                                         *
      16                 :            :  ***************************************************************************/
      17                 :            : 
      18                 :            : #include "qgscolorschemeregistry.h"
      19                 :            : #include "qgscolorscheme.h"
      20                 :            : #include "qgsapplication.h"
      21                 :            : #include <QDir>
      22                 :            : #include <QFileInfoList>
      23                 :            : #include <QMutex>
      24                 :            : #include <random>
      25                 :            : 
      26                 :         10 : QgsColorSchemeRegistry::~QgsColorSchemeRegistry()
      27                 :         10 : {
      28                 :          5 :   qDeleteAll( mColorSchemeList );
      29                 :          5 :   mColorSchemeList.clear();
      30                 :         10 : }
      31                 :            : 
      32                 :          0 : void QgsColorSchemeRegistry::populateFromInstance()
      33                 :            : {
      34                 :            :   //get schemes from global instance
      35                 :          0 :   QList< QgsColorScheme * > schemeList = QgsApplication::colorSchemeRegistry()->schemes();
      36                 :            : 
      37                 :            :   //add to this scheme registry
      38                 :          0 :   QList< QgsColorScheme * >::iterator it = schemeList.begin();
      39                 :          0 :   for ( ; it != schemeList.end(); ++it )
      40                 :            :   {
      41                 :          0 :     addColorScheme( ( *it )->clone() );
      42                 :          0 :   }
      43                 :          0 : }
      44                 :            : 
      45                 :          7 : void QgsColorSchemeRegistry::addDefaultSchemes()
      46                 :            : {
      47                 :            :   //default color schemes
      48                 :          7 :   addColorScheme( new QgsRecentColorScheme() );
      49                 :          7 :   addColorScheme( new QgsCustomColorScheme() );
      50                 :          7 :   addColorScheme( new QgsProjectColorScheme() );
      51                 :          7 :   addUserSchemes();
      52                 :          7 : }
      53                 :            : 
      54                 :          7 : void QgsColorSchemeRegistry::initStyleScheme()
      55                 :            : {
      56                 :         14 :   QString stylePalette = QgsApplication::pkgDataPath() + QStringLiteral( "/resources/palettes/new_layer_colors.gpl" );
      57                 :          7 :   if ( QFileInfo::exists( stylePalette ) )
      58                 :            :   {
      59                 :          0 :     QgsUserColorScheme *scheme = new QgsUserColorScheme( stylePalette );
      60                 :          0 :     addColorScheme( scheme );
      61                 :          0 :     setRandomStyleColorScheme( scheme );
      62                 :          0 :   }
      63                 :          7 : }
      64                 :            : 
      65                 :          7 : void QgsColorSchemeRegistry::addUserSchemes()
      66                 :            : {
      67                 :          7 :   QString palettesDir = QgsApplication::qgisSettingsDirPath() + "palettes";
      68                 :            : 
      69                 :          7 :   QDir localDir;
      70                 :          7 :   if ( !localDir.mkpath( palettesDir ) )
      71                 :            :   {
      72                 :          0 :     return;
      73                 :            :   }
      74                 :            : 
      75                 :         14 :   QFileInfoList fileInfoList = QDir( palettesDir ).entryInfoList( QStringList( QStringLiteral( "*.gpl" ) ), QDir::Files );
      76                 :          7 :   QFileInfoList::const_iterator infoIt = fileInfoList.constBegin();
      77                 :          7 :   for ( ; infoIt != fileInfoList.constEnd(); ++infoIt )
      78                 :            :   {
      79                 :          0 :     addColorScheme( new QgsUserColorScheme( infoIt->fileName() ) );
      80                 :          0 :   }
      81                 :          7 : }
      82                 :            : 
      83                 :         21 : void QgsColorSchemeRegistry::addColorScheme( QgsColorScheme *scheme )
      84                 :            : {
      85                 :         21 :   mColorSchemeList.append( scheme );
      86                 :         21 : }
      87                 :            : 
      88                 :          0 : QList<QgsColorScheme *> QgsColorSchemeRegistry::schemes() const
      89                 :            : {
      90                 :          0 :   QList< QgsColorScheme * > allSchemes;
      91                 :          0 :   QList<QgsColorScheme *>::const_iterator schemeIt;
      92                 :          0 :   for ( schemeIt = mColorSchemeList.constBegin(); schemeIt != mColorSchemeList.constEnd(); ++schemeIt )
      93                 :            :   {
      94                 :          0 :     allSchemes.append( ( *schemeIt ) );
      95                 :          0 :   }
      96                 :          0 :   return allSchemes;
      97                 :          0 : }
      98                 :            : 
      99                 :          0 : QList<QgsColorScheme *> QgsColorSchemeRegistry::schemes( const QgsColorScheme::SchemeFlag flag ) const
     100                 :            : {
     101                 :          0 :   QList< QgsColorScheme * > matchingSchemes;
     102                 :          0 :   QList<QgsColorScheme *>::const_iterator schemeIt;
     103                 :          0 :   for ( schemeIt = mColorSchemeList.constBegin(); schemeIt != mColorSchemeList.constEnd(); ++schemeIt )
     104                 :            :   {
     105                 :          0 :     if ( ( *schemeIt )->flags().testFlag( flag ) )
     106                 :            :     {
     107                 :          0 :       matchingSchemes.append( ( *schemeIt ) );
     108                 :          0 :     }
     109                 :          0 :   }
     110                 :          0 :   return matchingSchemes;
     111                 :          0 : }
     112                 :            : 
     113                 :          0 : void QgsColorSchemeRegistry::setRandomStyleColorScheme( QgsColorScheme *scheme )
     114                 :            : {
     115                 :          0 :   mRandomStyleColorScheme = scheme;
     116                 :          0 :   if ( scheme )
     117                 :            :   {
     118                 :          0 :     mRandomStyleColors = scheme->fetchColors();
     119                 :            : 
     120                 :          0 :     if ( mRandomStyleColors.count() > 0 )
     121                 :            :     {
     122                 :          0 :       std::random_device rd;
     123                 :          0 :       std::mt19937 mt( rd() );
     124                 :          0 :       std::uniform_int_distribution<int> colorDist( 0, mRandomStyleColors.count() - 1 );
     125                 :          0 :       mNextRandomStyleColorIndex = colorDist( mt );
     126                 :          0 :       std::uniform_int_distribution<int> colorDir( 0, 1 );
     127                 :          0 :       mNextRandomStyleColorDirection = colorDir( mt ) == 0 ? -1 : 1;
     128                 :          0 :     }
     129                 :          0 :   }
     130                 :            :   else
     131                 :            :   {
     132                 :          0 :     mRandomStyleColors.clear();
     133                 :            :   }
     134                 :          0 : }
     135                 :            : 
     136                 :          0 : QgsColorScheme *QgsColorSchemeRegistry::randomStyleColorScheme()
     137                 :            : {
     138                 :          0 :   return mRandomStyleColorScheme;
     139                 :            : }
     140                 :            : 
     141                 :         78 : QColor QgsColorSchemeRegistry::fetchRandomStyleColor() const
     142                 :            : {
     143                 :         78 :   if ( mRandomStyleColors.empty() )
     144                 :            :   {
     145                 :            :     // no random color scheme available - so just use totally random colors
     146                 :            : 
     147                 :            :     // Make sure we use get uniquely seeded random numbers, and not the same sequence of numbers
     148                 :         78 :     std::random_device rd;
     149                 :         78 :     std::mt19937 mt( rd() );
     150                 :         78 :     std::uniform_int_distribution<int> hueDist( 0, 359 );
     151                 :         78 :     std::uniform_int_distribution<int> satDist( 64, 255 );
     152                 :         78 :     std::uniform_int_distribution<int> valueDist( 128, 255 );
     153                 :         78 :     return QColor::fromHsv( hueDist( mt ), satDist( mt ), valueDist( mt ) );
     154                 :         78 :   }
     155                 :            :   else
     156                 :            :   {
     157                 :          0 :     static QMutex sMutex;
     158                 :          0 :     QMutexLocker locker( &sMutex );
     159                 :          0 :     QColor res = mRandomStyleColors.at( mNextRandomStyleColorIndex ).first;
     160                 :          0 :     mNextRandomStyleColorIndex += mNextRandomStyleColorDirection;
     161                 :          0 :     if ( mNextRandomStyleColorIndex < 0 )
     162                 :          0 :       mNextRandomStyleColorIndex = mRandomStyleColors.count() - 1;
     163                 :          0 :     if ( mNextRandomStyleColorIndex >= mRandomStyleColors.count() )
     164                 :          0 :       mNextRandomStyleColorIndex = 0;
     165                 :            :     return res;
     166                 :          0 :   }
     167                 :         78 : }
     168                 :            : 
     169                 :          0 : bool QgsColorSchemeRegistry::removeColorScheme( QgsColorScheme *scheme )
     170                 :            : {
     171                 :          0 :   if ( mRandomStyleColorScheme == scheme )
     172                 :            :   {
     173                 :          0 :     mRandomStyleColorScheme = nullptr;
     174                 :          0 :     mRandomStyleColors.clear();
     175                 :          0 :   }
     176                 :            : 
     177                 :          0 :   if ( mColorSchemeList.indexOf( scheme ) != -1 )
     178                 :            :   {
     179                 :          0 :     mColorSchemeList.removeAll( scheme );
     180                 :          0 :     return true;
     181                 :            :   }
     182                 :            : 
     183                 :            :   //not found
     184                 :          0 :   return false;
     185                 :          0 : }
     186                 :            : 
     187                 :            :