]> SALOME platform Git repositories - modules/shaper.git/blob - src/FeaturesPlugin/FeaturesPlugin_CompositeSketch.cpp
Salome HOME
La part
[modules/shaper.git] / src / FeaturesPlugin / FeaturesPlugin_CompositeSketch.cpp
1 // Copyright (C) 2014-2022  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 <FeaturesPlugin_CompositeSketch.h>
21 #include <FeaturesPlugin_Tools.h>
22
23 #include <ModelAPI_AttributeSelectionList.h>
24 #include <ModelAPI_AttributeReference.h>
25 #include <ModelAPI_BodyBuilder.h>
26 #include <ModelAPI_ResultConstruction.h>
27 #include <ModelAPI_Session.h>
28 #include <ModelAPI_Validator.h>
29
30 #include <GeomAlgoAPI_Revolution.h>
31
32 #include <GeomAPI_ShapeExplorer.h>
33
34
35 static const std::string COMPOSITESKETCH_VERSION_1("v9.6");
36
37 static void storeSubShape(const std::shared_ptr<GeomAlgoAPI_MakeShape> theMakeShape,
38                           ResultBodyPtr theResultBody,
39                           const GeomShapePtr theShape,
40                           const GeomAPI_Shape::ShapeType theType,
41                           const std::string& theName);
42
43 //=================================================================================================
44 void FeaturesPlugin_CompositeSketch::initCompositeSketchAttribtues(const int theInitFlags)
45 {
46   // Initialize sketch launcher.
47   if(theInitFlags & InitSketchLauncher) {
48     data()->addAttribute(SKETCH_ID(), ModelAPI_AttributeReference::typeId());
49     ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), SKETCH_ID());
50   }
51
52   // Initialize selection list.
53   if(theInitFlags & InitBaseObjectsList) {
54     AttributeSelectionListPtr anObjectsAttr =
55         std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(
56         data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId()));
57     myCurrentSelectionType = selectionList(BASE_OBJECTS_ID())->selectionType();
58     anObjectsAttr->setWholeResultAllowed(true);
59     if (!anObjectsAttr->isInitialized()) {
60       // new feature, specify the version
61       data()->setVersion(COMPOSITESKETCH_VERSION_1);
62     }
63   }
64 }
65
66 //=================================================================================================
67 std::shared_ptr<ModelAPI_Feature> FeaturesPlugin_CompositeSketch::addFeature(std::string theID)
68 {
69   FeaturePtr aNew = document()->addFeature(theID, false);
70   if(aNew) {
71     data()->reference(SKETCH_ID())->setValue(aNew);
72   }
73
74   // Set as current also after it becomes sub to set correctly enabled for other sketch subs.
75   document()->setCurrentFeature(aNew, false);
76   return aNew;
77 }
78
79 //=================================================================================================
80 int FeaturesPlugin_CompositeSketch::numberOfSubs(bool /*forTree*/) const
81 {
82   ObjectPtr aObj = data()->reference(SKETCH_ID())->value();
83   return aObj.get() ? 1 : 0;
84 }
85
86 //=================================================================================================
87 std::shared_ptr<ModelAPI_Feature> FeaturesPlugin_CompositeSketch::subFeature(const int theIndex,
88                                                                              bool /*forTree*/)
89 {
90   FeaturePtr aSubFeature;
91   if(theIndex == 0) {
92     aSubFeature =
93         std::dynamic_pointer_cast<ModelAPI_Feature>(data()->reference(SKETCH_ID())->value());
94   }
95   return aSubFeature;
96 }
97
98 //=================================================================================================
99 int FeaturesPlugin_CompositeSketch::subFeatureId(const int theIndex) const
100 {
101   if(theIndex == 0) {
102     FeaturePtr aFeature =
103       std::dynamic_pointer_cast<ModelAPI_Feature>(data()->reference(SKETCH_ID())->value());
104     if(aFeature.get()) {
105       return aFeature->data()->featureId();
106     }
107   }
108
109   return -1;
110 }
111
112 //=================================================================================================
113 bool FeaturesPlugin_CompositeSketch::isSub(ObjectPtr theObject) const
114 {
115   bool isSubFeature = false;
116   // Check is this feature of result
117   FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(theObject);
118   if (aFeature.get()) {
119     ObjectPtr aSub = data()->reference(SKETCH_ID())->value();
120     isSubFeature = aSub == theObject;
121   }
122   return isSubFeature;
123 }
124
125 //=================================================================================================
126 void FeaturesPlugin_CompositeSketch::removeFeature(std::shared_ptr<ModelAPI_Feature> theFeature)
127 {
128   AttributeSelectionListPtr aBaseObjectsSelectionList = selectionList(BASE_OBJECTS_ID());
129   if(aBaseObjectsSelectionList.get() && aBaseObjectsSelectionList->size() > 0) {
130     aBaseObjectsSelectionList->clear();
131   }
132
133   reference(SKETCH_ID())->setValue(ObjectPtr());
134 }
135
136 //=================================================================================================
137 void FeaturesPlugin_CompositeSketch::getBaseShapes(ListOfShape& theBaseShapesList,
138                                                    const bool theIsMakeShells)
139 {
140   AttributeSelectionListPtr aBaseObjectsSelectionList = selectionList(BASE_OBJECTS_ID());
141   std::string anError;
142   bool isOk = FeaturesPlugin_Tools::getShape(
143       aBaseObjectsSelectionList, theIsMakeShells, theBaseShapesList, anError);
144   if (!isOk)
145     setError(anError);
146 }
147
148 //=================================================================================================
149 void FeaturesPlugin_CompositeSketch::storeResult(const GeomShapePtr theBaseShape,
150                                         const std::shared_ptr<GeomAlgoAPI_MakeShape> theMakeShape,
151                                         const int theIndex)
152 {
153   // Create result body.
154   ResultBodyPtr aResultBody = document()->createBody(data(), theIndex);
155
156   // Store generated shape.
157   aResultBody->storeGenerated(theBaseShape, theMakeShape->shape());
158
159   // Store generated edges/faces.
160   storeGenerationHistory(aResultBody, theBaseShape, theMakeShape);
161
162   setResult(aResultBody, theIndex);
163 }
164
165 //=================================================================================================
166 void FeaturesPlugin_CompositeSketch::storeGenerationHistory(ResultBodyPtr theResultBody,
167                                         const GeomShapePtr theBaseShape,
168                                         const std::shared_ptr<GeomAlgoAPI_MakeShape> theMakeShape)
169 {
170   GeomAPI_Shape::ShapeType aBaseShapeType = theBaseShape->shapeType();
171   GeomAPI_Shape::ShapeType aShapeTypeToExplode = GeomAPI_Shape::SHAPE;
172
173   switch(aBaseShapeType) {
174     case GeomAPI_Shape::EDGE: {
175             aShapeTypeToExplode = GeomAPI_Shape::VERTEX;
176       break;
177     }
178     case GeomAPI_Shape::WIRE: {
179       aShapeTypeToExplode = GeomAPI_Shape::COMPOUND;
180       break;
181     }
182     case GeomAPI_Shape::FACE:
183     case GeomAPI_Shape::SHELL: {
184       aShapeTypeToExplode = GeomAPI_Shape::EDGE;
185       break;
186     }
187     case GeomAPI_Shape::COMPOUND: {
188       aShapeTypeToExplode = GeomAPI_Shape::COMPOUND;
189     }
190     default: // [to avoid compilation warnings]
191       break;
192   }
193
194   if(aShapeTypeToExplode == GeomAPI_Shape::VERTEX ||
195       aShapeTypeToExplode == GeomAPI_Shape::COMPOUND) {
196     theResultBody->loadGeneratedShapes(theMakeShape, theBaseShape, GeomAPI_Shape::VERTEX);
197   }
198   if(aShapeTypeToExplode == GeomAPI_Shape::EDGE ||
199       aShapeTypeToExplode == GeomAPI_Shape::COMPOUND) {
200     theResultBody->loadGeneratedShapes(theMakeShape, theBaseShape, GeomAPI_Shape::EDGE);
201   }
202   std::list<std::shared_ptr<GeomAlgoAPI_MakeSweep> > aSweeps; // all sweeps collected
203   std::shared_ptr<GeomAlgoAPI_MakeSweep> aMakeSweep =
204     std::dynamic_pointer_cast<GeomAlgoAPI_MakeSweep>(theMakeShape);
205   if(aMakeSweep.get()) {
206     aSweeps.push_back(aMakeSweep);
207   } else {
208     std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeList =
209       std::dynamic_pointer_cast<GeomAlgoAPI_MakeShapeList>(theMakeShape);
210     if (aMakeList.get()) {
211       ListOfMakeShape::const_iterator anIter = aMakeList->list().cbegin();
212       for(; anIter != aMakeList->list().cend(); anIter++) {
213         std::shared_ptr<GeomAlgoAPI_MakeSweep> aSweep =
214           std::dynamic_pointer_cast<GeomAlgoAPI_MakeSweep>(*anIter);
215         if (aSweep.get())
216           aSweeps.push_back(aSweep);
217       }
218     }
219   }
220   std::list<std::shared_ptr<GeomAlgoAPI_MakeSweep> >::iterator aSweep = aSweeps.begin();
221   for(; aSweep != aSweeps.end(); aSweep++) {
222     // Store from shapes.
223     storeShapes(theMakeShape, theResultBody, aBaseShapeType, (*aSweep)->fromShapes(), "From_");
224
225     // Store to shapes.
226     storeShapes(theMakeShape, theResultBody, aBaseShapeType, (*aSweep)->toShapes(), "To_");
227   }
228 }
229
230 //=================================================================================================
231 void FeaturesPlugin_CompositeSketch::storeShapes(
232   const std::shared_ptr<GeomAlgoAPI_MakeShape> theMakeShape,
233   ResultBodyPtr theResultBody,
234   const GeomAPI_Shape::ShapeType theBaseShapeType,
235   const ListOfShape& theShapes,
236   const std::string theName)
237 {
238   GeomAPI_Shape::ShapeType aShapeTypeToExplore = GeomAPI_Shape::FACE;
239   std::string aShapeTypeStr = "Face";
240   switch(theBaseShapeType) {
241     case GeomAPI_Shape::VERTEX: {
242       aShapeTypeToExplore = GeomAPI_Shape::VERTEX;
243       aShapeTypeStr = "Vertex";
244       break;
245     }
246     case GeomAPI_Shape::EDGE:
247     case GeomAPI_Shape::WIRE: {
248       aShapeTypeToExplore = GeomAPI_Shape::EDGE;
249       aShapeTypeStr = "Edge";
250       break;
251     }
252     case GeomAPI_Shape::FACE:
253     case GeomAPI_Shape::SHELL: {
254       aShapeTypeToExplore = GeomAPI_Shape::FACE;
255       aShapeTypeStr = "Face";
256       break;
257     }
258     case GeomAPI_Shape::COMPOUND: {
259       aShapeTypeToExplore = GeomAPI_Shape::COMPOUND;
260       break;
261     }
262     default: // [to avoid compilation warnings]
263       break;
264   }
265
266   // Store shapes.
267   for(ListOfShape::const_iterator anIt = theShapes.cbegin(); anIt != theShapes.cend(); ++anIt) {
268     GeomShapePtr aShape = *anIt;
269
270     if(aShapeTypeToExplore == GeomAPI_Shape::COMPOUND) {
271       std::string aName = theName + (aShape->shapeType() == GeomAPI_Shape::EDGE ? "Edge" : "Face");
272       storeSubShape(theMakeShape, theResultBody, aShape, aShape->shapeType(), aName);
273     } else {
274       std::string aName = theName + aShapeTypeStr;
275       storeSubShape(theMakeShape, theResultBody, aShape, aShapeTypeToExplore, aName);
276       if (theBaseShapeType == GeomAPI_Shape::WIRE) { // issue 2289: special names also for vertices
277         aName = theName + "Vertex";
278         storeSubShape(theMakeShape, theResultBody, aShape, GeomAPI_Shape::VERTEX, aName);
279       }
280     }
281   }
282 }
283
284 void storeSubShape(
285   const std::shared_ptr<GeomAlgoAPI_MakeShape> theMakeShape,
286   ResultBodyPtr theResultBody,
287   const GeomShapePtr theShape,
288   const GeomAPI_Shape::ShapeType theType,
289   const std::string& theName)
290 {
291   for(GeomAPI_ShapeExplorer anExp(theShape, theType); anExp.more(); anExp.next()) {
292     GeomShapePtr aSubShape = anExp.current();
293     if (!theResultBody->generated(aSubShape, theName)) {
294       int aNbSubs = theResultBody->numberOfSubs();
295       if (aNbSubs > 0) {
296         // check the modified shape is in the result body, don't store it if not
297         ListOfShape aNewShapes;
298         theMakeShape->modified(aSubShape, aNewShapes);
299         for (int i = 0; i < aNbSubs; ++i) {
300           ResultBodyPtr aSubRes = theResultBody->subResult(i);
301           GeomShapePtr aShape = aSubRes->shape();
302           ListOfShape::iterator aNewIt = aNewShapes.begin();
303           for (; aNewIt != aNewShapes.end(); ++aNewIt)
304             if (aShape->isSubShape(*aNewIt, false))
305               break;
306           if (aNewIt != aNewShapes.end()) {
307             // store from/to shapes as primitives and then store modification of them by the boolean
308             aSubRes->generated(aSubShape, theName, false);
309             aSubRes->loadModifiedShapes(theMakeShape, aSubShape, theType);
310           }
311         }
312       }
313       else {
314         // store from/to shapes as primitives and then store modification of them by the boolean
315         theResultBody->generated(aSubShape, theName, false);
316         theResultBody->loadModifiedShapes(theMakeShape, aSubShape, theType);
317       }
318     }
319   }
320 }