1 // Copyright (C) 2017 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
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
21 #include "FeaturesPlugin_Fillet.h"
23 #include <ModelAPI_Data.h>
24 #include <ModelAPI_AttributeDouble.h>
25 #include <ModelAPI_AttributeSelectionList.h>
26 #include <ModelAPI_AttributeString.h>
27 #include <ModelAPI_ResultBody.h>
28 #include <ModelAPI_Session.h>
29 #include <ModelAPI_Tools.h>
30 #include <ModelAPI_Validator.h>
32 #include <GeomAlgoAPI_Fillet.h>
33 #include <GeomAlgoAPI_MakeShapeList.h>
35 #include <GeomAPI_DataMapOfShapeMapOfShapes.h>
36 #include <GeomAPI_ShapeExplorer.h>
37 #include <GeomAPI_ShapeIterator.h>
40 // Obtain all sub-shapes from the shape and append them to the list
41 static void collectSubs(const GeomShapePtr& theShape,
43 const GeomAPI_Shape::ShapeType theShapeType)
45 GeomAPI_ShapeExplorer anExp(theShape, theShapeType);
46 for (; anExp.more(); anExp.next()) {
47 GeomShapePtr aShape = anExp.current();
48 // Store all shapes with FORWARD orientation to avoid duplication of shared edges/vertices
49 aShape->setOrientation(GeomAPI_Shape::FORWARD);
50 theSubs.push_back(aShape);
54 // Extract edges from the list
55 static ListOfShape selectEdges(const ListOfShape& theShapes)
58 for (ListOfShape::const_iterator anIt = theShapes.begin(); anIt != theShapes.end(); ++anIt)
59 if ((*anIt)->isEdge())
60 anEdges.push_back(*anIt);
64 // If theShape is a compound of single shape, return it
65 static GeomShapePtr unwrapCompound(const GeomShapePtr& theShape)
67 GeomShapePtr aShape = theShape;
68 if(aShape->shapeType() == GeomAPI_Shape::COMPOUND) {
69 int aSubResultsNb = 0;
70 GeomAPI_ShapeIterator anIt(aShape);
71 for(; anIt.more(); anIt.next())
74 if(aSubResultsNb == 1) {
76 aShape = anIt.current();
83 FeaturesPlugin_Fillet::FeaturesPlugin_Fillet()
87 void FeaturesPlugin_Fillet::initAttributes()
89 data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId());
90 data()->addAttribute(OBJECT_LIST_ID(), ModelAPI_AttributeSelectionList::typeId());
91 data()->addAttribute(START_RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
92 data()->addAttribute(END_RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
94 ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), END_RADIUS_ID());
98 void FeaturesPlugin_Fillet::execute()
100 AttributeStringPtr aCreationMethod = string(CREATION_METHOD());
101 if (!aCreationMethod)
104 GeomAPI_DataMapOfShapeMapOfShapes aSolidsAndSubs;
106 // getting objects and sort them accroding to parent solids
107 AttributeSelectionListPtr anObjectsSelList = selectionList(OBJECT_LIST_ID());
108 for (int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); ++anObjectsIndex) {
109 AttributeSelectionPtr anObjectAttr = anObjectsSelList->value(anObjectsIndex);
110 GeomShapePtr anObject = anObjectAttr->value();
114 ResultPtr aContext = anObjectAttr->context();
115 ResultBodyPtr aCtxOwner = ModelAPI_Tools::bodyOwner(aContext);
116 GeomShapePtr aParent = aCtxOwner ? aCtxOwner->shape() : aContext->shape();
120 ListOfShape anEdgesAndVertices;
121 collectSubs(anObject, anEdgesAndVertices, GeomAPI_Shape::EDGE);
122 collectSubs(anObject, anEdgesAndVertices, GeomAPI_Shape::VERTEX);
123 for (ListOfShape::iterator aEIt = anEdgesAndVertices.begin();
124 aEIt != anEdgesAndVertices.end(); ++aEIt)
125 aSolidsAndSubs.add(aParent, *aEIt);
128 bool isFixedRadius = true;
129 double aRadius1 = 0.0, aRadius2 = 0.0;
130 if (aCreationMethod->value() == CREATION_METHOD_SINGLE_RADIUS())
131 aRadius1 = real(RADIUS_ID())->value();
133 aRadius1 = real(START_RADIUS_ID())->value();
134 aRadius2 = real(END_RADIUS_ID())->value();
135 isFixedRadius = false;
138 // Perform fillet operation
139 GeomAlgoAPI_MakeShapeList aMakeShapeList;
140 std::shared_ptr<GeomAlgoAPI_Fillet> aFilletBuilder;
141 int aResultIndex = 0;
143 GeomAPI_DataMapOfShapeMapOfShapes::iterator anIt = aSolidsAndSubs.begin();
144 for (; anIt != aSolidsAndSubs.end(); ++anIt) {
145 GeomShapePtr aSolid = anIt.first();
146 ListOfShape aFilletEdgesAndVertices = anIt.second();
148 ListOfShape aFilletEdges = selectEdges(aFilletEdgesAndVertices);
150 aFilletBuilder.reset(new GeomAlgoAPI_Fillet(aSolid, aFilletEdges, aRadius1));
152 aFilletBuilder.reset(new GeomAlgoAPI_Fillet(aSolid, aFilletEdges, aRadius1, aRadius2));
153 if (isFailed(aFilletBuilder))
156 GeomShapePtr aResult = unwrapCompound(aFilletBuilder->shape());
157 std::shared_ptr<ModelAPI_ResultBody> aResultBody =
158 document()->createBody(data(), aResultIndex);
160 loadNamingDS(aResultBody, aSolid, aFilletEdgesAndVertices, aResult, aFilletBuilder);
161 setResult(aResultBody, aResultIndex);
164 removeResults(aResultIndex);
167 bool FeaturesPlugin_Fillet::isFailed(
168 const std::shared_ptr<GeomAlgoAPI_MakeShape>& theAlgorithm)
170 if (!theAlgorithm->isDone()) {
171 static const std::string aFeatureError = "Error: fillet algorithm failed.";
172 setError(aFeatureError);
175 if (theAlgorithm->shape()->isNull()) {
176 static const std::string aShapeError = "Error: Resulting shape of fillet is Null.";
177 setError(aShapeError);
180 if (!theAlgorithm->isValid()) {
181 std::string aFeatureError = "Error: Resulting shape of fillet is not valid.";
182 setError(aFeatureError);
188 void FeaturesPlugin_Fillet::loadNamingDS(
189 std::shared_ptr<ModelAPI_ResultBody> theResultBody,
190 const std::shared_ptr<GeomAPI_Shape> theBaseShape,
191 const ListOfShape& theFilletShapes,
192 const std::shared_ptr<GeomAPI_Shape> theResultShape,
193 const std::shared_ptr<GeomAlgoAPI_MakeShape>& theMakeShape)
196 if(theBaseShape->isEqual(theResultShape)) {
197 theResultBody->store(theResultShape, false);
201 std::shared_ptr<GeomAPI_DataMapOfShapeShape> aMapOfShapes = theMakeShape->mapOfSubShapes();
203 const int aDeletedTag = 1;
204 const int aModifyTag = 2;
205 const int aGeneratedTag = 3;
206 /// sub solids will be placed at labels 4, 5, etc. if result is compound of solids
207 const int aSubsolidsTag = 4;
209 theResultBody->storeModified(theBaseShape, theResultShape, aSubsolidsTag);
211 const std::string aModFaceName = "Modified_Face";
212 const std::string aFilletFaceName = "Fillet_Face";
214 // Store modified faces
215 theResultBody->loadAndOrientModifiedShapes(theMakeShape.get(), theBaseShape,
216 GeomAPI_Shape::FACE, aModifyTag, aModFaceName, *aMapOfShapes);
218 // Store new faces generated from edges and vertices
219 theResultBody->loadAndOrientGeneratedShapes(theMakeShape.get(), theBaseShape,
220 GeomAPI_Shape::EDGE, aGeneratedTag, aFilletFaceName, *aMapOfShapes);
221 theResultBody->loadAndOrientGeneratedShapes(theMakeShape.get(), theBaseShape,
222 GeomAPI_Shape::VERTEX, aGeneratedTag, aFilletFaceName, *aMapOfShapes);
225 theResultBody->loadDeletedShapes(theMakeShape.get(), theBaseShape,
226 GeomAPI_Shape::EDGE, aDeletedTag);
227 theResultBody->loadDeletedShapes(theMakeShape.get(), theBaseShape,
228 GeomAPI_Shape::FACE, aDeletedTag);