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