src/XGUI/XGUI_ColorDialog.cpp

   1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  18 //
  19
  20 #include <XGUI_ColorDialog.h>
  21
  22 #include <ModelAPI_Tools.h>
  23
  24 #include <QtxColorButton.h>
  25
  26 #include <QLabel>
  27 #include <QButtonGroup>
  28 #include <QGridLayout>
  29 #include <QRadioButton>
  30 #include <QDialogButtonBox>
  31
  32 XGUI_ColorDialog::XGUI_ColorDialog(QWidget* theParent)
  33   : QDialog(theParent)
  34 {
  35   setWindowTitle(tr("Color"));
  36   QGridLayout* aLay = new QGridLayout(this);
  37
  38   QRadioButton* aRandomChoiceBtn = new QRadioButton(this);
  39   QRadioButton* aColorChoiceBtn = new QRadioButton(this);
  40   aColorChoiceBtn->setChecked(true);
  41   myButtonGroup = new QButtonGroup(this);
  42   myButtonGroup->setExclusive(true);
  43   myButtonGroup->addButton(aColorChoiceBtn, 0);
  44   myButtonGroup->addButton(aRandomChoiceBtn, 1);
  45
  46   aLay->addWidget(aColorChoiceBtn, 0, 0);
  47   aLay->addWidget(aRandomChoiceBtn, 1, 0);
  48
  49   myColorButton = new QtxColorButton(this);
  50   myColorButton->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);
  51   aLay->addWidget(myColorButton, 0, 1);
  52
  53   QLabel* aRandomLabel = new QLabel(tr("Random"), this);
  54   aLay->addWidget(aRandomLabel, 1, 1);
  55
  56   QDialogButtonBox* aButtons = new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel,
  57                                                     Qt::Horizontal, this);
  58   connect(aButtons, SIGNAL(accepted()), this, SLOT(accept()));
  59   connect(aButtons, SIGNAL(rejected()), this, SLOT(reject()));
  60   aLay->addWidget(aButtons, 2, 0, 1, 2);
  61 }
  62
  63 bool XGUI_ColorDialog::isRandomColor() const
  64 {
  65   int anId = myButtonGroup->checkedId();
  66
  67   return myButtonGroup->checkedId() == 1;
  68 }
  69
  70 void XGUI_ColorDialog::setColor(const std::vector<int>& theValue)
  71 {
  72   if (theValue.size() != 3)
  73     return;
  74
  75   myColorButton->setColor(QColor(theValue[0], theValue[1], theValue[2]));
  76 }
  77
  78 std::vector<int> XGUI_ColorDialog::getColor() const
  79 {
  80   QColor aColorResult = myColorButton->color();
  81
  82   std::vector<int> aValues;
  83   aValues.push_back(aColorResult.red());
  84   aValues.push_back(aColorResult.green());
  85   aValues.push_back(aColorResult.blue());
  86
  87   return aValues;
  88 }
  89 // contains global cash for integer index of the color -> RGB of this color
  90 static std::map<int, std::vector<int> > myColorMap;
  91
  92 void appendValues(std::vector<int>& theRGB, const int theRed, const int theGreen, const int theBlue)
  93 {
  94   theRGB.push_back(theRed);
  95   theRGB.push_back(theGreen);
  96   theRGB.push_back(theBlue);
  97 }
  98
  99 bool containsValues(std::map<int, std::vector<int> >& theColorMap, std::vector<int>& theValues)
 100 {
 101   std::map<int, std::vector<int> >::const_iterator anIt = theColorMap.begin(),
 102                                                    aLast = theColorMap.end();
 103   bool isFound = false;
 104   for (; anIt != aLast && !isFound; anIt++) {
 105     std::vector<int> aValues = anIt->second;
 106     isFound = aValues[0] == theValues[0] &&
 107               aValues[1] == theValues[1] &&
 108               aValues[2] == theValues[2];
 109   }
 110   return isFound;
 111 }
 112
 113 std::vector<int> HSVtoRGB(int theH, int theS, int theV)
 114 {
 115   std::vector<int> aRGB;
 116   if (theH < 0 || theH > 360 ||
 117       theS < 0 || theS > 100 ||
 118       theV < 0 || theV > 100)
 119     return aRGB;
 120
 121   int aHi = (int)theH/60;
 122
 123   double aV = theV;
 124   double aVmin = (100 - theS)*theV/100;
 125
 126   double anA = (theV - aVmin)* (theH % 60) / 60;
 127
 128   double aVinc = aVmin + anA;
 129   double aVdec = theV - anA;
 130
 131   double aPercentToValue = 255./100;
 132   int aV_int    = (int)(aV*aPercentToValue);
 133   int aVinc_int = (int)(aVinc*aPercentToValue);
 134   int aVmin_int = (int)(aVmin*aPercentToValue);
 135   int aVdec_int = (int)(aVdec*aPercentToValue);
 136
 137   switch(aHi) {
 138     case 0: appendValues(aRGB, aV_int,    aVinc_int, aVmin_int); break;
 139     case 1: appendValues(aRGB, aVdec_int, aV_int,    aVmin_int); break;
 140     case 2: appendValues(aRGB, aVmin_int, aV_int,    aVinc_int); break;
 141     case 3: appendValues(aRGB, aVmin_int, aVdec_int, aV_int); break;
 142     case 4: appendValues(aRGB, aVinc_int, aVmin_int, aV_int); break;
 143     case 5: appendValues(aRGB, aV_int,    aVmin_int, aVdec_int); break;
 144     default: break;
 145   }
 146   return aRGB;
 147 }
 148
 149
 150 void fillColorMap()
 151 {
 152   if (!myColorMap.empty())
 153     return;
 154
 155   int i = 0;
 156   for (int s = 100; s > 0; s = s - 50)
 157   {
 158     for (int v = 100; v >= 40; v = v - 20)
 159     {
 160       for (int h = 0; h < 359 ; h = h + 60)
 161       {
 162         std::vector<int> aColor = HSVtoRGB(h, s, v);
 163         if (containsValues(myColorMap, aColor))
 164           continue;
 165         myColorMap[i] = aColor;
 166         i++;
 167       }
 168     }
 169   }
 170 }
 171
 172 void findRandomColor(std::vector<int>& theValues)
 173 {
 174   theValues.clear();
 175   if (myColorMap.empty()) {
 176     fillColorMap();
 177   }
 178
 179   size_t aSize = myColorMap.size();
 180   int anIndex = rand() % aSize;
 181   if (myColorMap.find(anIndex) != myColorMap.end()) {
 182     theValues = myColorMap.at(anIndex);
 183   }
 184 }
 185
 186 std::vector<int> XGUI_ColorDialog::getRandomColor() const
 187 {
 188   std::vector<int> aValues;
 189   if (isRandomColor()) {
 190     findRandomColor(aValues);
 191   }
 192   return aValues;
 193 }