1 // Copyright (C) 2014-2020 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include <XGUI_ColorDialog.h>
22 #include <ModelAPI_Tools.h>
24 #include <QtxColorButton.h>
27 #include <QButtonGroup>
28 #include <QGridLayout>
29 #include <QRadioButton>
30 #include <QDialogButtonBox>
32 XGUI_ColorDialog::XGUI_ColorDialog(QWidget* theParent)
33 : QDialog(theParent, Qt::Dialog | Qt::CustomizeWindowHint | Qt::WindowTitleHint)
35 setWindowTitle(tr("Color"));
36 QGridLayout* aLay = new QGridLayout(this);
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);
46 aLay->addWidget(aColorChoiceBtn, 0, 0);
47 aLay->addWidget(aRandomChoiceBtn, 1, 0);
49 myColorButton = new QtxColorButton(this);
50 myColorButton->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);
51 aLay->addWidget(myColorButton, 0, 1);
53 QLabel* aRandomLabel = new QLabel(tr("Random"), this);
54 aLay->addWidget(aRandomLabel, 1, 1);
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);
63 bool XGUI_ColorDialog::isRandomColor() const
65 int anId = myButtonGroup->checkedId();
67 return myButtonGroup->checkedId() == 1;
70 void XGUI_ColorDialog::setColor(const std::vector<int>& theValue)
72 if (theValue.size() != 3)
75 myColorButton->setColor(QColor(theValue[0], theValue[1], theValue[2]));
78 std::vector<int> XGUI_ColorDialog::getColor() const
80 QColor aColorResult = myColorButton->color();
82 std::vector<int> aValues;
83 aValues.push_back(aColorResult.red());
84 aValues.push_back(aColorResult.green());
85 aValues.push_back(aColorResult.blue());
89 // contains global cash for integer index of the color -> RGB of this color
90 static std::map<int, std::vector<int> > myColorMap;
92 void appendValues(std::vector<int>& theRGB, const int theRed, const int theGreen, const int theBlue)
94 theRGB.push_back(theRed);
95 theRGB.push_back(theGreen);
96 theRGB.push_back(theBlue);
99 bool containsValues(std::map<int, std::vector<int> >& theColorMap, std::vector<int>& theValues)
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];
113 std::vector<int> HSVtoRGB(int theH, int theS, int theV)
115 std::vector<int> aRGB;
116 if (theH < 0 || theH > 360 ||
117 theS < 0 || theS > 100 ||
118 theV < 0 || theV > 100)
121 int aHi = (int)theH/60;
124 double aVmin = (100 - theS)*theV/100;
126 double anA = (theV - aVmin)* (theH % 60) / 60;
128 double aVinc = aVmin + anA;
129 double aVdec = theV - anA;
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);
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;
152 if (!myColorMap.empty())
156 for (int s = 100; s > 0; s = s - 50)
158 for (int v = 100; v >= 40; v = v - 20)
160 for (int h = 0; h < 359 ; h = h + 60)
162 std::vector<int> aColor = HSVtoRGB(h, s, v);
163 if (containsValues(myColorMap, aColor))
165 myColorMap[i] = aColor;
172 void findRandomColor(std::vector<int>& theValues)
175 if (myColorMap.empty()) {
179 size_t aSize = myColorMap.size();
180 int anIndex = rand() % aSize;
181 if (myColorMap.find(anIndex) != myColorMap.end()) {
182 theValues = myColorMap.at(anIndex);
186 std::vector<int> XGUI_ColorDialog::getRandomColor() const
188 std::vector<int> aValues;
189 if (isRandomColor()) {
190 findRandomColor(aValues);