Salome HOME
CEA : Lot 2 - Auto Color for Groups
[modules/shaper.git] / src / XGUI / XGUI_ColorDialog.cpp
1 // Copyright (C) 2014-2021  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
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, Qt::Dialog | Qt::CustomizeWindowHint | Qt::WindowTitleHint)
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   return myButtonGroup->checkedId() == 1;
66 }
67
68 void XGUI_ColorDialog::setColor(const std::vector<int>& theValue)
69 {
70   if (theValue.size() != 3)
71     return;
72
73   myColorButton->setColor(QColor(theValue[0], theValue[1], theValue[2]));
74 }
75
76 std::vector<int> XGUI_ColorDialog::getColor() const
77 {
78   QColor aColorResult = myColorButton->color();
79
80   std::vector<int> aValues;
81   aValues.push_back(aColorResult.red());
82   aValues.push_back(aColorResult.green());
83   aValues.push_back(aColorResult.blue());
84
85   return aValues;
86 }
87
88 std::vector<int> XGUI_ColorDialog::getRandomColor() const
89 {
90   std::vector<int> aValues;
91   if (isRandomColor()) {
92     ModelAPI_Tools::findRandomColor(aValues);
93   }
94   return aValues;
95 }
96