Salome HOME
Merge branch 'master' of newgeom:newgeom.git
[modules/shaper.git] / src / ModuleBase / ModuleBase_WidgetSelector.cpp
1 // File:        ModuleBase_WidgetSelector.h
2 // Created:     2 June 2014
3 // Author:      Vitaly Smetannikov
4
5 #include "ModuleBase_WidgetSelector.h"
6 #include "ModuleBase_IWorkshop.h"
7
8 #include <Events_Loop.h>
9 #include <ModelAPI_Events.h>
10 #include <ModelAPI_Tools.h>
11
12 #include <ModelAPI_Data.h>
13 #include <ModelAPI_Object.h>
14 #include <ModelAPI_Result.h>
15 #include <ModelAPI_AttributeReference.h>
16 #include <Config_WidgetAPI.h>
17
18 #include <GeomAPI_Shape.h>
19
20 #include <TopoDS_Shape.hxx>
21 #include <TopExp_Explorer.hxx>
22
23 #include <QWidget>
24 #include <QLayout>
25 #include <QLabel>
26 #include <QLineEdit>
27 #include <QToolButton>
28 #include <QString>
29 #include <QEvent>
30 #include <QDockWidget>
31
32 #include <stdexcept>
33
34 typedef QMap<QString, TopAbs_ShapeEnum> ShapeTypes;
35 static ShapeTypes MyShapeTypes;
36
37 TopAbs_ShapeEnum ModuleBase_WidgetSelector::shapeType(const QString& theType)
38 {
39   if (MyShapeTypes.count() == 0) {
40     MyShapeTypes["face"] = TopAbs_FACE;
41     MyShapeTypes["vertex"] = TopAbs_VERTEX;
42     MyShapeTypes["wire"] = TopAbs_WIRE;
43     MyShapeTypes["edge"] = TopAbs_EDGE;
44     MyShapeTypes["shell"] = TopAbs_SHELL;
45     MyShapeTypes["solid"] = TopAbs_SOLID;
46   }
47   if (MyShapeTypes.contains(theType))
48     return MyShapeTypes[theType];
49   throw std::invalid_argument("Shape type defined in XML is not implemented!");
50 }
51
52 ModuleBase_WidgetSelector::ModuleBase_WidgetSelector(QWidget* theParent,
53                                                      ModuleBase_IWorkshop* theWorkshop,
54                                                      const Config_WidgetAPI* theData,
55                                                      const std::string& theParentId)
56     : ModuleBase_ModelWidget(theParent, theData, theParentId),
57       myWorkshop(theWorkshop)
58 {
59   myContainer = new QWidget(theParent);
60   QHBoxLayout* aLayout = new QHBoxLayout(myContainer);
61
62   aLayout->setContentsMargins(0, 0, 0, 0);
63   QString aLabelText = QString::fromStdString(theData->widgetLabel());
64   QString aLabelIcon = QString::fromStdString(theData->widgetIcon());
65   myLabel = new QLabel(aLabelText, myContainer);
66   if (!aLabelIcon.isEmpty())
67     myLabel->setPixmap(QPixmap(aLabelIcon));
68
69   aLayout->addWidget(myLabel);
70
71   QString aToolTip = QString::fromStdString(theData->widgetTooltip());
72   myTextLine = new QLineEdit(myContainer);
73   myTextLine->setReadOnly(true);
74   myTextLine->setToolTip(aToolTip);
75   myTextLine->installEventFilter(this);
76
77   myBasePalet = myTextLine->palette();
78   myInactivePalet = myBasePalet;
79   myInactivePalet.setBrush(QPalette::Base, QBrush(Qt::gray, Qt::Dense6Pattern));
80   myTextLine->setPalette(myInactivePalet);
81
82   aLayout->addWidget(myTextLine, 1);
83
84   myActivateBtn = new QToolButton(myContainer);
85   myActivateBtn->setIcon(QIcon(":icons/hand_point.png"));
86   myActivateBtn->setCheckable(true);
87   myActivateBtn->setToolTip(tr("Activate/Deactivate selection"));
88   connect(myActivateBtn, SIGNAL(toggled(bool)), this, SLOT(activateSelection(bool)));
89
90   aLayout->addWidget(myActivateBtn);
91
92   std::string aTypes = theData->getProperty("shape_types");
93   myShapeTypes = QString(aTypes.c_str()).split(' ');
94 }
95
96 //********************************************************************
97 ModuleBase_WidgetSelector::~ModuleBase_WidgetSelector()
98 {
99 }
100
101 //********************************************************************
102 bool ModuleBase_WidgetSelector::storeValue() const
103 {
104   FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(mySelectedObject);
105   if (aSelectedFeature == myFeature)  // In order to avoid selection of the same object
106     return false;
107
108   DataPtr aData = myFeature->data();
109   boost::shared_ptr<ModelAPI_AttributeReference> aRef = boost::dynamic_pointer_cast<
110       ModelAPI_AttributeReference>(aData->attribute(attributeID()));
111
112   ObjectPtr aObject = aRef->value();
113   if (!(aObject && aObject->isSame(mySelectedObject))) {
114     aRef->setValue(mySelectedObject);
115     updateObject(myFeature);
116   }
117   return true;
118 }
119
120 //********************************************************************
121 bool ModuleBase_WidgetSelector::restoreValue()
122 {
123   DataPtr aData = myFeature->data();
124   boost::shared_ptr<ModelAPI_AttributeReference> aRef = aData->reference(attributeID());
125
126   bool isBlocked = this->blockSignals(true);
127   mySelectedObject = aRef->value();
128   updateSelectionName();
129
130   this->blockSignals(isBlocked);
131   return true;
132 }
133
134 //********************************************************************
135 QList<QWidget*> ModuleBase_WidgetSelector::getControls() const
136 {
137   QList<QWidget*> aControls;
138   aControls.append(myLabel);
139   aControls.append(myTextLine);
140   aControls.append(myActivateBtn);
141   return aControls;
142 }
143
144 //********************************************************************
145 void ModuleBase_WidgetSelector::onSelectionChanged()
146 {
147   QList<ObjectPtr> aObjects = myWorkshop->selectedObjects();
148   if (aObjects.size() > 0) {
149     ObjectPtr aObject = aObjects.first();
150     if ((!mySelectedObject) && (!aObject))
151       return;
152     if (mySelectedObject && aObject && mySelectedObject->isSame(aObject))
153       return;
154
155     // Check that the selection corresponds to selection type
156     if (!isAccepted(aObject))
157       return;
158
159     mySelectedObject = aObject;
160     if (mySelectedObject) {
161       updateSelectionName();
162       myActivateBtn->setChecked(false);
163       raisePanel();
164     } else {
165       myTextLine->setText("");
166     }
167     emit valuesChanged();
168     emit focusOutWidget(this);
169   }
170 }
171
172 //********************************************************************
173 bool ModuleBase_WidgetSelector::isAccepted(const ObjectPtr theResult) const
174 {
175   ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(theResult);
176   boost::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
177   if (!aShapePtr)
178     return false;
179   TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
180   if (aShape.IsNull())
181     return false;
182
183   TopAbs_ShapeEnum aShapeType = aShape.ShapeType();
184   if (aShapeType == TopAbs_COMPOUND) {
185     foreach (QString aType, myShapeTypes) {
186       TopExp_Explorer aEx(aShape, shapeType(aType));
187       if (aEx.More())
188         return true;
189     }
190   } else {
191     foreach (QString aType, myShapeTypes) {
192       if (shapeType(aType) == aShapeType)
193         return true;
194     }
195   }
196   return false;
197 }
198
199 //********************************************************************
200 void ModuleBase_WidgetSelector::updateSelectionName()
201 {
202   if (mySelectedObject) {
203     std::string aName = mySelectedObject->data()->name();
204
205     myTextLine->setText(QString::fromStdString(aName));
206   } else
207     myTextLine->setText("");
208 }
209
210 //********************************************************************
211 void ModuleBase_WidgetSelector::enableOthersControls(bool toEnable) const
212 {
213   QWidget* aParent = myContainer->parentWidget();
214   QList<QWidget*> aChldList = aParent->findChildren<QWidget*>();
215   foreach(QWidget* aWgt, aChldList)
216   {
217     if ((aWgt != myLabel) && (aWgt != myActivateBtn) && (aWgt != myTextLine)
218         && (aWgt != myContainer))
219       aWgt->setEnabled(toEnable);
220   }
221 }
222
223 //********************************************************************
224 void ModuleBase_WidgetSelector::activateSelection(bool toActivate)
225 {
226   enableOthersControls(!toActivate);
227   //myTextLine->setEnabled(toActivate);
228   if (toActivate)
229     myTextLine->setPalette(myBasePalet);
230   else
231     myTextLine->setPalette(myInactivePalet);
232
233   if (toActivate)
234     connect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
235   else
236     disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
237
238   myActivateBtn->setDown(toActivate);
239 }
240
241 //********************************************************************
242 void ModuleBase_WidgetSelector::raisePanel() const
243 {
244   QWidget* aParent = myContainer->parentWidget();
245   QWidget* aLastPanel = 0;
246   while (!aParent->inherits("QDockWidget")) {
247     aLastPanel = aParent;
248     aParent = aParent->parentWidget();
249     if (!aParent)
250       return;
251   }
252   if (aParent->inherits("QDockWidget")) {
253     QDockWidget* aTabWgt = (QDockWidget*) aParent;
254     aTabWgt->raise();
255   }
256 }
257
258 //********************************************************************
259 bool ModuleBase_WidgetSelector::focusTo()
260 {
261   myActivateBtn->setChecked(true);
262   return true;
263 }