Salome HOME
Merge branch 'master' of salome:modules/shaper
[modules/shaper.git] / src / ModuleBase / ModuleBase_WidgetSelector.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:        ModuleBase_WidgetSelector.cpp
4 // Created:     19 June 2015
5 // Author:      Natalia ERMOLAEVA
6
7 #include <ModuleBase_WidgetSelector.h>
8
9 #include <ModuleBase_ISelection.h>
10 #include <ModuleBase_IWorkshop.h>
11 #include <ModuleBase_Tools.h>
12 #include <ModuleBase_Operation.h>
13 #include <ModuleBase_OperationDescription.h>
14 #include <ModuleBase_WidgetFactory.h>
15 #include <ModuleBase_IModule.h>
16 #include <ModuleBase_ResultPrs.h>
17 #include <ModuleBase_ViewerPrs.h>
18
19 #include <ModelAPI_ResultConstruction.h>
20
21 #include <TopoDS_Iterator.hxx>
22
23 ModuleBase_WidgetSelector::ModuleBase_WidgetSelector(QWidget* theParent,
24                                                      ModuleBase_IWorkshop* theWorkshop,
25                                                      const Config_WidgetAPI* theData)
26 : ModuleBase_WidgetValidated(theParent, theWorkshop, theData)
27 {
28 }
29
30 //********************************************************************
31 ModuleBase_WidgetSelector::~ModuleBase_WidgetSelector()
32 {
33 }
34
35 //********************************************************************
36 void ModuleBase_WidgetSelector::getGeomSelection(const ModuleBase_ViewerPrsPtr& thePrs,
37                                                       ObjectPtr& theObject,
38                                                       GeomShapePtr& theShape)
39 {
40   ModuleBase_ISelection* aSelection = myWorkshop->selection();
41   theObject = aSelection->getResult(thePrs);
42   theShape = aSelection->getShape(thePrs);
43 }
44
45 //********************************************************************
46 void ModuleBase_WidgetSelector::onSelectionChanged()
47 {
48   QList<ModuleBase_ViewerPrsPtr> aSelected = getFilteredSelected();
49   // equal vertices should not be used here
50   ModuleBase_ISelection::filterSelectionOnEqualPoints(aSelected);
51
52   bool isDone = setSelection(aSelected, true/*false*/);
53   updateOnSelectionChanged(isDone);
54 }
55
56 //********************************************************************
57 void ModuleBase_WidgetSelector::updateOnSelectionChanged(const bool theDone)
58 {
59   // "false" flag should be used here, it connects to the #26658 OCC bug, when the user click in 
60   // the same place repeatedly without mouse moved. In the case validation by filters is not
61   // perfromed, so an invalid object is selected. E.g. distance constraint, selection of a point.
62   // the 3rd click in the same point allow using this point.
63   emit valuesChanged();
64   // the updateObject method should be called to flush the updated sigal. The workshop listens it,
65   // calls validators for the feature and, as a result, updates the Apply button state.
66   updateObject(myFeature);
67
68   // we need to forget about previous validation result as the current selection can influence on it
69   clearValidatedCash();
70
71   if (theDone)
72     updateFocus();
73 }
74
75 //********************************************************************
76 QList<ModuleBase_ViewerPrsPtr> ModuleBase_WidgetSelector::getAttributeSelection() const
77 {
78   return QList<ModuleBase_ViewerPrsPtr>();
79 }
80
81 //********************************************************************
82 bool ModuleBase_WidgetSelector::acceptSubShape(const GeomShapePtr& theShape,
83                                                const ResultPtr& theResult) const
84 {
85   bool aValid = false;
86
87   GeomShapePtr aShape = theShape;
88
89   QIntList aShapeTypes = getShapeTypes();
90   if (aShapeTypes.empty()) {
91     aValid = true;
92     return aValid;
93   }
94   // when the SHAPE type is provided by XML, the hole result shape can be selected.
95   if (!aShape.get() && aShapeTypes.contains(TopAbs_SHAPE)) {
96     aValid = true;
97     return aValid;
98   }
99
100   if (!aShape.get() && theResult.get()) {
101     if (theResult.get())
102       aShape = theResult->shape();
103   }
104   TopAbs_ShapeEnum aShapeType = TopAbs_SHAPE;
105   if (aShape.get()) {
106     // Check that the selection corresponds to selection type
107     TopoDS_Shape aTopoShape = aShape->impl<TopoDS_Shape>();
108     aShapeType = aTopoShape.ShapeType();
109     // for compounds check sub-shapes: it may be compound of needed type:
110     // Booleans may produce compounds of Solids
111     if (aShapeType == TopAbs_COMPOUND) {
112       aShapeType = ModuleBase_Tools::getCompoundSubType(aTopoShape);
113     }
114   }
115
116   QIntList::const_iterator anIt = aShapeTypes.begin(), aLast = aShapeTypes.end();
117   for (; anIt != aLast; anIt++) {
118     TopAbs_ShapeEnum aCurrentShapeType = (TopAbs_ShapeEnum)*anIt;
119     if (aShapeType == aCurrentShapeType)
120       aValid = true;
121     else if (aCurrentShapeType == TopAbs_FACE) {
122       // try to process the construction shape only if there is no a selected shape in the viewer
123       if (!theShape.get() && theResult.get()) {
124         ResultConstructionPtr aCResult =
125                                 std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(theResult);
126         aValid = aCResult.get() && aCResult->facesNum() > 0;
127       }
128       aValid = ModuleBase_ResultPrs::isValidShapeType(aCurrentShapeType, aShapeType);
129     }
130   }
131   return aValid;
132 }
133
134 //********************************************************************
135 bool ModuleBase_WidgetSelector::activateSelectionAndFilters(bool toActivate)
136 {
137   updateSelectionName();
138
139   if (toActivate) {
140     myWorkshop->activateSubShapesSelection(getShapeTypes());
141   } else {
142     myWorkshop->deactivateSubShapesSelection();
143   }
144   return activateFilters(toActivate);
145 }
146
147 //********************************************************************
148 void ModuleBase_WidgetSelector::activateCustom()
149 {
150   connect(myWorkshop, SIGNAL(selectionChanged()), this,
151           SLOT(onSelectionChanged()), Qt::UniqueConnection);
152   
153   activateSelectionAndFilters(true);
154
155   // Restore selection in the viewer by the attribute selection list
156   myWorkshop->setSelected(getAttributeSelection());
157 }
158
159 //********************************************************************
160 bool ModuleBase_WidgetSelector::isValidSelectionCustom(const ModuleBase_ViewerPrsPtr& thePrs)
161 {
162   GeomShapePtr aShape = myWorkshop->selection()->getShape(thePrs);
163   ResultPtr aResult = myWorkshop->selection()->getResult(thePrs);
164   bool aValid = acceptSubShape(aShape, aResult);
165
166   if (aValid) {
167     // In order to avoid selection of the same object
168     ResultPtr aResult = myWorkshop->selection()->getResult(thePrs);
169     FeaturePtr aSelectedFeature = ModelAPI_Feature::feature(aResult);
170     aValid = aSelectedFeature != myFeature;
171   }
172   return aValid;
173 }
174
175 //********************************************************************
176 bool ModuleBase_WidgetSelector::setSelectionCustom(const ModuleBase_ViewerPrsPtr& thePrs)
177 {
178   ObjectPtr anObject;
179   GeomShapePtr aShape;
180   getGeomSelection(thePrs, anObject, aShape);
181
182   // the last flag is to be depending on hasObject is called before. To be corrected later
183   ModuleBase_Tools::setObject(attribute(), anObject, aShape, myWorkshop, myIsInValidate, true);
184   return true;
185 }
186
187 //********************************************************************
188 void ModuleBase_WidgetSelector::deactivate()
189 {
190   ModuleBase_ModelWidget::deactivate();
191   disconnect(myWorkshop, SIGNAL(selectionChanged()), this, SLOT(onSelectionChanged()));
192   activateSelectionAndFilters(false);
193   ModuleBase_ModelWidget::deactivate();
194 }
195
196 //********************************************************************
197 std::string ModuleBase_WidgetSelector::generateName(const AttributePtr& theAttribute,
198                                                     ModuleBase_IWorkshop* theWorkshop)
199 {
200   std::string aName;
201   if (theAttribute.get() != NULL) {
202     ModuleBase_Operation* anOperation = theWorkshop->currentOperation();
203
204     FeaturePtr aFeature = ModelAPI_Feature::feature(theAttribute->owner());
205     if (aFeature.get()) {
206       std::string aXmlCfg, aDescription;
207       theWorkshop->module()->getXMLRepresentation(aFeature->getKind(), aXmlCfg, aDescription);
208
209       ModuleBase_WidgetFactory aFactory(aXmlCfg, theWorkshop);
210       std::string anAttributeTitle;
211       aFactory.getAttributeTitle(theAttribute->id(), anAttributeTitle);
212
213       std::stringstream aStreamName;
214       aStreamName << theAttribute->owner()->data()->name() << "/"<< anAttributeTitle.c_str();
215       aName = aStreamName.str();
216     }
217   }
218   return aName;
219 }