1 // Copyright (C) 2014-2019 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include "FeaturesPlugin_Tools.h"
22 #include <ModelAPI_AttributeSelectionList.h>
23 #include <ModelAPI_ResultBody.h>
24 #include <ModelAPI_ResultConstruction.h>
25 #include <ModelAPI_ResultPart.h>
26 #include <ModelAPI_Tools.h>
28 #include <GeomAlgoAPI_CompoundBuilder.h>
29 #include <GeomAlgoAPI_ShapeTools.h>
31 #include <GeomAPI_PlanarEdges.h>
32 #include <GeomAPI_ShapeIterator.h>
34 //==================================================================================================
35 void FeaturesPlugin_Tools::loadModifiedShapes(ResultBodyPtr theResultBody,
36 const ListOfShape& theBaseShapes,
37 const ListOfShape& theTools,
38 const GeomMakeShapePtr& theMakeShape,
39 const GeomShapePtr theResultShape,
40 const std::string& theNamePrefix)
42 theResultBody->storeModified(theBaseShapes, theResultShape, theMakeShape);
44 ListOfShape aShapes = theBaseShapes;
45 ListOfShape::const_iterator aToolIter = theTools.cbegin();
46 for(; aToolIter != theTools.cend(); aToolIter++)
47 aShapes.push_back(*aToolIter);
49 for (ListOfShape::const_iterator anIter = aShapes.begin(); anIter != aShapes.end(); ++anIter)
51 theResultBody->loadModifiedShapes(theMakeShape, *anIter, GeomAPI_Shape::VERTEX, theNamePrefix);
52 theResultBody->loadModifiedShapes(theMakeShape, *anIter, GeomAPI_Shape::EDGE, theNamePrefix);
53 theResultBody->loadModifiedShapes(theMakeShape, *anIter, GeomAPI_Shape::FACE, theNamePrefix);
57 //==================================================================================================
58 void FeaturesPlugin_Tools::loadModifiedShapes(ResultBodyPtr theResultBody,
59 const GeomShapePtr& theBaseShape,
60 const GeomMakeShapePtr& theMakeShape,
61 const std::string theName)
63 switch(theBaseShape->shapeType()) {
64 case GeomAPI_Shape::COMPOUND: {
65 for(GeomAPI_ShapeIterator anIt(theBaseShape); anIt.more(); anIt.next())
67 loadModifiedShapes(theResultBody,
74 case GeomAPI_Shape::COMPSOLID:
75 case GeomAPI_Shape::SOLID:
76 case GeomAPI_Shape::SHELL: {
77 theResultBody->loadModifiedShapes(theMakeShape,
82 case GeomAPI_Shape::FACE:
83 case GeomAPI_Shape::WIRE: {
84 theResultBody->loadModifiedShapes(theMakeShape,
89 case GeomAPI_Shape::EDGE: {
90 theResultBody->loadModifiedShapes(theMakeShape,
92 GeomAPI_Shape::VERTEX,
98 //==================================================================================================
99 void FeaturesPlugin_Tools::loadDeletedShapes(ResultBodyPtr theResultBody,
100 const GeomShapePtr theBaseShape,
101 const ListOfShape& theTools,
102 const GeomMakeShapePtr& theMakeShape,
103 const GeomShapePtr theResultShapesCompound)
105 ListOfShape aShapes = theTools;
106 if (theBaseShape.get())
107 aShapes.push_front(theBaseShape);
109 for (ListOfShape::const_iterator anIter = aShapes.begin(); anIter != aShapes.end(); anIter++)
111 theResultBody->loadDeletedShapes(theMakeShape,
113 GeomAPI_Shape::VERTEX,
114 theResultShapesCompound);
115 theResultBody->loadDeletedShapes(theMakeShape,
118 theResultShapesCompound);
119 theResultBody->loadDeletedShapes(theMakeShape,
122 theResultShapesCompound);
126 //==================================================================================================
127 void FeaturesPlugin_Tools::loadDeletedShapes(
128 std::vector<ResultBaseAlgo>& theResultBaseAlgoList,
129 const ListOfShape& theTools,
130 const GeomShapePtr theResultShapesCompound)
132 for (std::vector<ResultBaseAlgo>::iterator anIt = theResultBaseAlgoList.begin();
133 anIt != theResultBaseAlgoList.end();
136 ResultBaseAlgo& aRCA = *anIt;
137 loadDeletedShapes(aRCA.resultBody,
141 theResultShapesCompound);
145 //==================================================================================================
146 bool FeaturesPlugin_Tools::getShape(const AttributeSelectionListPtr theSelectionList,
147 const bool theShareTopology,
148 ListOfShape& theShapesList,
149 std::string& theError)
151 theShapesList.clear();
153 ListOfShape aBaseFacesList;
154 std::map<ResultConstructionPtr, ListOfShape> aSketchWiresMap;
155 if(!theSelectionList.get()) {
156 theError = "Error: Could not get base objects selection list.";
159 if(theSelectionList->size() == 0) {
160 theError = "Error: Base objects list is empty.";
163 for(int anIndex = 0; anIndex < theSelectionList->size(); anIndex++) {
164 AttributeSelectionPtr aBaseObjectSelection = theSelectionList->value(anIndex);
165 if(!aBaseObjectSelection.get()) {
166 theError = "Error: Selected base object is empty.";
169 GeomShapePtr aBaseShape = aBaseObjectSelection->value();
170 if(aBaseShape.get() && !aBaseShape->isNull()) {
171 GeomAPI_Shape::ShapeType aST = aBaseShape->shapeType();
172 if(aST == GeomAPI_Shape::SOLID || aST == GeomAPI_Shape::COMPSOLID) {
173 theError = "Error: Selected shapes has unsupported type.";
176 ResultConstructionPtr aConstruction =
177 std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aBaseObjectSelection->context());
178 if(aConstruction.get() && !aBaseShape->isEqual(aConstruction->shape()) &&
179 aST == GeomAPI_Shape::WIRE) {
180 // It is a wire on the sketch, store it to make face later.
181 aSketchWiresMap[aConstruction].push_back(aBaseShape);
184 aST == GeomAPI_Shape::FACE ? aBaseFacesList.push_back(aBaseShape) :
185 theShapesList.push_back(aBaseShape);
188 // This may be the whole sketch result selected, check and get faces.
189 ResultConstructionPtr aConstruction =
190 std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aBaseObjectSelection->context());
191 if(!aConstruction.get()) {
192 theError = "Error: Selected sketches does not have results.";
195 int aFacesNum = aConstruction->facesNum();
197 // Probably it can be construction.
198 aBaseShape = aConstruction->shape();
199 if(aBaseShape.get() && !aBaseShape->isNull()) {
200 GeomAPI_Shape::ShapeType aST = aBaseShape->shapeType();
201 if(aST != GeomAPI_Shape::VERTEX && aST != GeomAPI_Shape::EDGE &&
202 aST != GeomAPI_Shape::WIRE &&
203 aST != GeomAPI_Shape::FACE && aST != GeomAPI_Shape::SHELL) {
204 theError = "Error: Selected shapes has unsupported type.";
207 aST == GeomAPI_Shape::FACE ? aBaseFacesList.push_back(aBaseShape) :
208 theShapesList.push_back(aBaseShape);
211 for(int aFaceIndex = 0; aFaceIndex < aFacesNum; aFaceIndex++) {
212 GeomShapePtr aBaseFace = aConstruction->face(aFaceIndex);
213 if(!aBaseFace.get() || aBaseFace->isNull()) {
214 theError = "Error: One of the faces on selected sketch is null.";
217 aBaseFacesList.push_back(aBaseFace);
223 // Make faces from sketch wires.
224 for(std::map<ResultConstructionPtr, ListOfShape>::const_iterator anIt = aSketchWiresMap.cbegin();
225 anIt != aSketchWiresMap.cend(); ++anIt) {
226 const std::shared_ptr<GeomAPI_PlanarEdges> aSketchPlanarEdges =
227 std::dynamic_pointer_cast<GeomAPI_PlanarEdges>((*anIt).first->shape());
228 const ListOfShape& aWiresList = (*anIt).second;
230 GeomAlgoAPI_ShapeTools::makeFacesWithHoles(aSketchPlanarEdges->origin(),
231 aSketchPlanarEdges->norm(),
234 aBaseFacesList.insert(aBaseFacesList.end(), aFaces.begin(), aFaces.end());
237 // Searching faces with common edges.
238 if(theShareTopology && aBaseFacesList.size() > 1) {
239 GeomShapePtr aFacesCompound = GeomAlgoAPI_CompoundBuilder::compound(aBaseFacesList);
240 GeomAlgoAPI_ShapeTools::combineShapes(aFacesCompound, GeomAPI_Shape::SHELL, theShapesList);
242 theShapesList.insert(theShapesList.end(), aBaseFacesList.begin(), aBaseFacesList.end());
247 //==================================================================================================
248 bool FeaturesPlugin_Tools::shapesFromSelectionList(
249 const std::shared_ptr<ModelAPI_AttributeSelectionList> theSelectionList,
250 const bool theStoreFullHierarchy,
251 GeomAPI_ShapeHierarchy& theHierarchy,
252 std::list<ResultPtr>& theParts)
254 int aSize = theSelectionList->size();
255 for (int anObjectsIndex = 0; anObjectsIndex < aSize; anObjectsIndex++) {
256 AttributeSelectionPtr anObjectAttr = theSelectionList->value(anObjectsIndex);
257 std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
258 if (!anObject.get()) { // may be for not-activated parts
261 ResultPtr aContext = anObjectAttr->context();
262 if (aContext->groupName() == ModelAPI_ResultPart::group())
263 theParts.push_back(aContext);
265 // store full shape hierarchy for the corresponding version only
266 theHierarchy.addObject(anObject);
267 if (theStoreFullHierarchy)
268 ModelAPI_Tools::fillShapeHierarchy(anObject, aContext, theHierarchy);