]> SALOME platform Git repositories - modules/shaper.git/blob - src/ModuleBase/ModuleBase_WidgetShapeSelector.cpp
Salome HOME
The validator shape_type is not used for the multi-selector and shape-selector control.
[modules/shaper.git] / src / ModuleBase / ModuleBase_WidgetShapeSelector.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:        ModuleBase_WidgetShapeSelector.h
4 // Created:     2 June 2014
5 // Author:      Vitaly Smetannikov
6
7 #include <ModuleBase_WidgetShapeSelector.h>
8 #include <ModuleBase_Definitions.h>
9 #include <ModuleBase_ISelection.h>
10 #include <ModuleBase_IWorkshop.h>
11 #include <ModuleBase_IViewer.h>
12 #include <ModuleBase_Tools.h>
13 #include <ModuleBase_FilterFactory.h>
14 #include <ModuleBase_Filter.h>
15
16 #include <GeomValidators_ShapeType.h>
17
18 #include <Config_WidgetAPI.h>
19 #include <Events_Loop.h>
20 #include <Events_Message.h>
21 #include <GeomAPI_Interface.h>
22 #include <GeomAPI_Shape.h>
23 #include <GeomValidators_Tools.h>
24
25 #include <ModelAPI_AttributeReference.h>
26 #include <ModelAPI_Data.h>
27 #include <ModelAPI_Document.h>
28 #include <ModelAPI_Events.h>
29 #include <ModelAPI_Feature.h>
30 #include <ModelAPI_Result.h>
31 #include <ModelAPI_ResultConstruction.h>
32 #include <ModelAPI_AttributeReference.h>
33 #include <ModelAPI_AttributeSelection.h>
34 #include <ModelAPI_Session.h>
35 #include <ModelAPI_Tools.h>
36 #include <ModelAPI_ResultBody.h>
37 #include <ModelAPI_AttributeRefAttr.h>
38 #include <ModelAPI_Validator.h>
39 #include <ModelAPI_AttributeValidator.h>
40 #include <ModelAPI_ShapeValidator.h>
41
42 #include <Config_WidgetAPI.h>
43 #include <Events_Error.h>
44
45 #include <GeomAPI_Shape.h>
46
47 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
48 #include <TopoDS_Shape.hxx>
49 #include <TopExp_Explorer.hxx>
50
51 #include <QWidget>
52 #include <QLayout>
53 #include <QLabel>
54 #include <QLineEdit>
55 #include <QToolButton>
56 #include <QString>
57 #include <QEvent>
58 #include <QDockWidget>
59 #include <QApplication>
60 #include <QFormLayout>
61
62 #include <TopExp_Explorer.hxx>
63 #include <TopoDS_Shape.hxx>
64
65 #include <memory>
66
67 #include <list>
68 #include <string>
69
70
71 ModuleBase_WidgetShapeSelector::ModuleBase_WidgetShapeSelector(QWidget* theParent,
72                                                      ModuleBase_IWorkshop* theWorkshop,
73                                                      const Config_WidgetAPI* theData,
74                                                      const std::string& theParentId)
75     : ModuleBase_WidgetValidated(theParent, theData, theParentId),
76       myWorkshop(theWorkshop)
77 {
78   QFormLayout* aLayout = new QFormLayout(this);
79   ModuleBase_Tools::adjustMargins(aLayout);
80
81   QString aLabelText = QString::fromStdString(theData->widgetLabel());
82   QString aLabelIcon = QString::fromStdString(theData->widgetIcon());
83   myLabel = new QLabel(aLabelText, this);
84   if (!aLabelIcon.isEmpty())
85     myLabel->setPixmap(QPixmap(aLabelIcon));
86
87
88   QString aToolTip = QString::fromStdString(theData->widgetTooltip());
89   myTextLine = new QLineEdit(this);
90   myTextLine->setReadOnly(true);
91   myTextLine->setToolTip(aToolTip);
92   myTextLine->installEventFilter(this);
93
94   aLayout->addRow(myLabel, myTextLine);
95
96   std::string aTypes = theData->getProperty("shape_types");
97   myShapeTypes = QString(aTypes.c_str()).split(' ', QString::SkipEmptyParts);
98
99   myShapeValidator = new GeomValidators_ShapeType();
100 }
101
102 //********************************************************************
103 ModuleBase_WidgetShapeSelector::~ModuleBase_WidgetShapeSelector()
104 {
105   activateSelection(false);
106   activateFilters(myWorkshop, false);
107
108   delete myShapeValidator;
109 }
110
111 //********************************************************************
112 bool ModuleBase_WidgetShapeSelector::storeValueCustom() const
113 {
114   // the value is stored on the selection changed signal processing 
115   return true;
116 }
117
118 //********************************************************************
119 bool ModuleBase_WidgetShapeSelector::setObject(ObjectPtr theSelectedObject,
120                                                GeomShapePtr theShape)
121 {
122   bool isChanged = false;
123   FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(theSelectedObject);
124   if (aSelectedFeature == myFeature)  // In order to avoid selection of the same object
125     return isChanged;
126
127   DataPtr aData = myFeature->data();
128   AttributeReferencePtr aRef = aData->reference(attributeID());
129   if (aRef) {
130     ObjectPtr aObject = aRef->value();
131     if (!(aObject && aObject->isSame(theSelectedObject))) {
132       aRef->setValue(theSelectedObject);
133       isChanged = true;
134     }
135   } else {
136     AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
137     if (aRefAttr) {
138       ObjectPtr aObject = aRefAttr->object();
139       if (!(aObject && aObject->isSame(theSelectedObject))) {
140         aRefAttr->setObject(theSelectedObject);
141         isChanged = true;
142       }
143     } else {
144       AttributeSelectionPtr aSelectAttr = aData->selection(attributeID());
145       ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theSelectedObject);
146       if (aSelectAttr.get() != NULL) {
147         aSelectAttr->setValue(aResult, theShape);
148         isChanged = true;
149       }
150     }
151   }
152   return isChanged;
153 }
154
155 //********************************************************************
156 QList<ModuleBase_ViewerPrs> ModuleBase_WidgetShapeSelector::getAttributeSelection() const
157 {
158   QList<ModuleBase_ViewerPrs> aSelected;
159   if(myFeature) {
160     DataPtr aData = myFeature->data();
161     AttributePtr anAttribute = myFeature->attribute(attributeID());
162
163     ObjectPtr anObject = GeomValidators_Tools::getObject(anAttribute);
164     TopoDS_Shape aShape;
165     std::shared_ptr<GeomAPI_Shape> aShapePtr = getShape();
166     if (aShapePtr.get()) {
167       aShape = aShapePtr->impl<TopoDS_Shape>();
168     }
169     ModuleBase_ViewerPrs aPrs(anObject, aShape, NULL);
170     aSelected.append(aPrs);
171   }
172   return aSelected;
173 }
174
175 //********************************************************************
176 void ModuleBase_WidgetShapeSelector::clearAttribute()
177 {
178   DataPtr aData = myFeature->data();
179   AttributeSelectionPtr aSelect = aData->selection(attributeID());
180   if (aSelect) {
181     aSelect->setValue(ResultPtr(), std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape()));
182     return;
183   }
184   AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
185   if (aRefAttr) {
186     aRefAttr->setObject(ObjectPtr());
187     return;
188   }
189   AttributeReferencePtr aRef = aData->reference(attributeID());
190   if (aRef) {
191     aRef->setObject(ObjectPtr());
192   }
193 }
194
195 //********************************************************************
196 bool ModuleBase_WidgetShapeSelector::restoreValue()
197 {
198   bool isBlocked = this->blockSignals(true);
199   updateSelectionName();
200   this->blockSignals(isBlocked);
201
202   return true;
203 }
204
205 //********************************************************************
206 QList<QWidget*> ModuleBase_WidgetShapeSelector::getControls() const
207 {
208   QList<QWidget*> aControls;
209   aControls.append(myTextLine);
210   return aControls;
211 }
212
213 //********************************************************************
214 void ModuleBase_WidgetShapeSelector::onSelectionChanged()
215 {
216   // In order to make reselection possible, set empty object and shape should be done
217   setObject(ObjectPtr(), std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape()));
218
219   bool aHasObject = false;
220   QList<ModuleBase_ViewerPrs> aSelectedPrs = getSelectedEntitiesOrObjects(myWorkshop->selection());
221   if (!aSelectedPrs.empty()) {
222     ModuleBase_ViewerPrs aPrs = aSelectedPrs.first();
223     if (!aPrs.isEmpty() && isValidSelection(aPrs)) {
224       setSelectionCustom(aPrs);
225       aHasObject = true;
226     }
227   }
228   // the updateObject method should be called to flush the updated sigal. The workshop listens it,
229   // calls validators for the feature and, as a result, updates the Apply button state.
230   updateObject(myFeature);
231   // the widget loses the focus only if the selected object is set
232   if (aHasObject)
233     emit focusOutWidget(this);
234 }
235
236 //********************************************************************
237 bool ModuleBase_WidgetShapeSelector::acceptSubShape(const TopoDS_Shape& theShape) const
238 {
239   foreach (QString aType, myShapeTypes) {
240     if (theShape.ShapeType() == ModuleBase_Tools::shapeType(aType))
241       return true;
242   }
243   return false;
244 }
245
246 //********************************************************************
247 GeomShapePtr ModuleBase_WidgetShapeSelector::getShape() const
248 {
249   GeomShapePtr aShape;
250   DataPtr aData = myFeature->data();
251   if (!aData->isValid())
252     return aShape;
253
254   AttributeSelectionPtr aSelect = aData->selection(attributeID());
255   if (aSelect)
256     aShape = aSelect->value();
257
258   return aShape;
259 }
260
261 //********************************************************************
262 void ModuleBase_WidgetShapeSelector::updateSelectionName()
263 {
264   DataPtr aData = myFeature->data();
265   if (!aData->isValid())
266     return;
267
268   bool isNameUpdated = false;
269   AttributeSelectionPtr aSelect = aData->selection(attributeID());
270   if (aSelect) {
271     myTextLine->setText(QString::fromStdString(aSelect->namingName(getDefaultValue())));
272     isNameUpdated = true;
273   }
274   if (!isNameUpdated) {
275     ObjectPtr anObject = GeomValidators_Tools::getObject(myFeature->attribute(attributeID()));
276     if (anObject.get() != NULL) {
277       std::string aName = anObject->data()->name();
278       myTextLine->setText(QString::fromStdString(aName));
279     } else {
280       AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
281       if (aRefAttr && aRefAttr->attr().get() != NULL) {
282         //myIsObject = aRefAttr->isObject();
283         AttributePtr anAttr = aRefAttr->attr();
284         if (anAttr.get() != NULL) {
285           std::stringstream aName;
286           aName <<anAttr->owner()->data()->name()<<"/"<<anAttr->id();
287           myTextLine->setText(QString::fromStdString(aName.str()));
288         }
289       }
290       else {
291         myTextLine->setText(getDefaultValue().c_str());
292       }
293     }
294   }
295 }
296
297
298 //********************************************************************
299 void ModuleBase_WidgetShapeSelector::activateSelection(bool toActivate)
300 {
301   updateSelectionName();
302
303   if (toActivate) {
304     QIntList aList;
305     foreach (QString aType, myShapeTypes) {
306       aList.append(ModuleBase_Tools::shapeType(aType));
307     }
308     myWorkshop->activateSubShapesSelection(aList);
309   } else {
310     myWorkshop->deactivateSubShapesSelection();
311   }
312 }
313
314 //********************************************************************
315 void ModuleBase_WidgetShapeSelector::raisePanel() const
316 {
317   QWidget* aParent = this->parentWidget();
318   QWidget* aLastPanel = 0;
319   while (!aParent->inherits("QDockWidget")) {
320     aLastPanel = aParent;
321     aParent = aParent->parentWidget();
322     if (!aParent)
323       return;
324   }
325   if (aParent->inherits("QDockWidget")) {
326     QDockWidget* aTabWgt = (QDockWidget*) aParent;
327     aTabWgt->raise();
328   }
329 }
330
331 //********************************************************************
332 void ModuleBase_WidgetShapeSelector::activateCustom()
333 {
334   connect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
335   activateSelection(true);
336
337   // Restore selection in the viewer by the attribute selection list
338   myWorkshop->setSelected(getAttributeSelection());
339
340   activateFilters(myWorkshop, true);
341 }
342
343 //********************************************************************
344 void ModuleBase_WidgetShapeSelector::storeAttributeValue()
345 {
346   DataPtr aData = myFeature->data();
347   AttributePtr anAttribute = myFeature->attribute(attributeID());
348
349   myObject = GeomValidators_Tools::getObject(anAttribute);
350   myShape = getShape();
351   myRefAttribute = AttributePtr();
352   myIsObject = false;
353   AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
354   if (aRefAttr) {
355     myIsObject = aRefAttr->isObject();
356     myRefAttribute = aRefAttr->attr();
357   }
358 }
359
360 //********************************************************************
361 void ModuleBase_WidgetShapeSelector::restoreAttributeValue(bool theValid)
362 {
363   DataPtr aData = myFeature->data();
364   AttributePtr anAttribute = myFeature->attribute(attributeID());
365
366   setObject(myObject, myShape);
367   AttributeRefAttrPtr aRefAttr = aData->refattr(attributeID());
368   if (aRefAttr) {
369     if (!myIsObject)
370       aRefAttr->setAttr(myRefAttribute);
371   }
372 }
373
374 //********************************************************************
375 void ModuleBase_WidgetShapeSelector::customValidators(
376                                     std::list<ModelAPI_Validator*>& theValidators,
377                                     std::list<std::list<std::string> >& theArguments) const
378 {
379   return;
380   theValidators.push_back(myShapeValidator);
381
382   std::list<std::string> anArguments;
383   foreach(QString aType, myShapeTypes) {
384     anArguments.push_back(aType.toStdString().c_str());
385   }
386   theArguments.push_back(anArguments);
387 }
388
389 //********************************************************************
390 bool ModuleBase_WidgetShapeSelector::setSelectionCustom(const ModuleBase_ViewerPrs& thePrs)
391 {
392   bool isDone = false;
393
394   // It should be checked by corresponded validator
395   ObjectPtr aObject = thePrs.object();
396   ObjectPtr aCurrentObject = GeomValidators_Tools::getObject(myFeature->attribute(attributeID()));
397   /*
398   if ((!aCurrentObject) && (!aObject))
399     return false;*/
400
401   // It should be checked by corresponded validator
402   // Check that the selected object is result (others can not be accepted)
403   ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aObject);
404   if (!aRes)
405     return false;
406   /*if (myFeature) {
407     // We can not select a result of our feature
408     const std::list<std::shared_ptr<ModelAPI_Result>>& aResList = myFeature->results();
409     std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aIt;
410     for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
411       if ((*aIt) == aRes)
412         return false;
413     }
414   }
415   */
416   // It should be checked by corresponded validator
417   /*
418   // Check that object belongs to active document or PartSet
419   DocumentPtr aDoc = aRes->document();
420   SessionPtr aMgr = ModelAPI_Session::get();
421   if (!(aDoc == aMgr->activeDocument()) && !(aDoc == aMgr->moduleDocument()))
422     return false;*/
423
424   // It should be checked by corresponded validator
425   // Check that the result has a shape
426   GeomShapePtr aShape = ModelAPI_Tools::shape(aRes);
427   if (!aShape)
428     return false;
429
430   // Get sub-shapes from local selection
431   if (!thePrs.shape().IsNull()) {
432     aShape = std::shared_ptr<GeomAPI_Shape>(new GeomAPI_Shape());
433     aShape->setImpl(new TopoDS_Shape(thePrs.shape()));
434   }
435   // Check that the selection corresponds to selection type
436   TopoDS_Shape aTopoShape = aShape->impl<TopoDS_Shape>();
437   if (!acceptSubShape(aTopoShape))
438     return false;
439
440   setObject(aObject, aShape);
441   return true;
442 }
443
444 //********************************************************************
445 void ModuleBase_WidgetShapeSelector::deactivate()
446 {
447   activateSelection(false);
448   activateFilters(myWorkshop, false);
449   disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
450 }