1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 // File: FeaturesPlugin_Boolean.cpp
4 // Created: 02 Sept 2014
5 // Author: Vitaly SMETANNIKOV
7 #include "FeaturesPlugin_Boolean.h"
9 #include <ModelAPI_Data.h>
10 #include <ModelAPI_Document.h>
11 #include <ModelAPI_AttributeReference.h>
12 #include <ModelAPI_AttributeInteger.h>
13 #include <ModelAPI_ResultBody.h>
14 #include <ModelAPI_AttributeSelectionList.h>
16 #include <GeomAlgoAPI_Boolean.h>
17 #include <GeomAlgoAPI_MakeShapeList.h>
18 #include <GeomAlgoAPI_ShapeProps.h>
22 #define _DELETED_TAG 2
24 //=================================================================================================
25 FeaturesPlugin_Boolean::FeaturesPlugin_Boolean()
29 //=================================================================================================
30 void FeaturesPlugin_Boolean::initAttributes()
32 data()->addAttribute(FeaturesPlugin_Boolean::TYPE_ID(), ModelAPI_AttributeInteger::typeId());
34 AttributeSelectionListPtr aSelection =
35 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(data()->addAttribute(
36 FeaturesPlugin_Boolean::OBJECT_LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
37 // extrusion works with faces always
38 aSelection->setSelectionType("SOLID");
40 aSelection = std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(data()->addAttribute(
41 FeaturesPlugin_Boolean::TOOL_LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
42 // extrusion works with faces always
43 aSelection->setSelectionType("SOLID");
46 //=================================================================================================
47 std::shared_ptr<GeomAPI_Shape> FeaturesPlugin_Boolean::getShape(const std::string& theAttrName)
49 std::shared_ptr<ModelAPI_AttributeReference> aObjRef = std::dynamic_pointer_cast<
50 ModelAPI_AttributeReference>(data()->attribute(theAttrName));
52 std::shared_ptr<ModelAPI_ResultBody> aConstr = std::dynamic_pointer_cast<
53 ModelAPI_ResultBody>(aObjRef->value());
55 return aConstr->shape();
57 return std::shared_ptr<GeomAPI_Shape>();
60 //=================================================================================================
61 void FeaturesPlugin_Boolean::execute()
63 // Getting operation type.
64 std::shared_ptr<ModelAPI_AttributeInteger> aTypeAttr = std::dynamic_pointer_cast<
65 ModelAPI_AttributeInteger>(data()->attribute(FeaturesPlugin_Boolean::TYPE_ID()));
68 GeomAlgoAPI_Boolean::OperationType aType = (GeomAlgoAPI_Boolean::OperationType)aTypeAttr->value();
70 ListOfShape anObjects, aTools;
73 AttributeSelectionListPtr anObjectsSelList = selectionList(FeaturesPlugin_Boolean::OBJECT_LIST_ID());
74 if (anObjectsSelList->size() == 0) {
77 for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) {
78 std::shared_ptr<ModelAPI_AttributeSelection> anObjectAttr = anObjectsSelList->value(anObjectsIndex);
79 std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
83 anObjects.push_back(anObject);
87 AttributeSelectionListPtr aToolsSelList = selectionList(FeaturesPlugin_Boolean::TOOL_LIST_ID());
88 if (aToolsSelList->size() == 0) {
91 for(int aToolsIndex = 0; aToolsIndex < aToolsSelList->size(); aToolsIndex++) {
92 std::shared_ptr<ModelAPI_AttributeSelection> aToolAttr = aToolsSelList->value(aToolsIndex);
93 std::shared_ptr<GeomAPI_Shape> aTool = aToolAttr->value();
97 aTools.push_back(aTool);
100 int aResultIndex = 0;
101 ListOfMakeShape aListOfMakeShape;
102 std::shared_ptr<GeomAPI_DataMapOfShapeShape> aDataMapOfShapes;
105 case GeomAlgoAPI_Boolean::BOOL_CUT:
106 case GeomAlgoAPI_Boolean::BOOL_COMMON:{
107 // Cut each object with all tools
108 for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end(); anObjectsIt++) {
109 std::shared_ptr<GeomAPI_Shape> anObject = *anObjectsIt;
110 ListOfShape aListWithObject;
111 aListWithObject.push_back(anObject);
112 GeomAlgoAPI_Boolean aBoolAlgo(aListWithObject, aTools, aType);
114 // Checking that the algorithm worked properly.
115 if(!aBoolAlgo.isDone()) {
116 static const std::string aFeatureError = "Boolean algorithm failed";
117 setError(aFeatureError);
120 if(aBoolAlgo.shape()->isNull()) {
121 static const std::string aShapeError = "Resulting shape is Null";
122 setError(aShapeError);
125 if(!aBoolAlgo.isValid()) {
126 std::string aFeatureError = "Warning: resulting shape is not valid";
127 setError(aFeatureError);
131 if(GeomAlgoAPI_ShapeProps::volume(aBoolAlgo.shape()) > 1.e-7) {
132 std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data(), aResultIndex);
133 std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList = std::shared_ptr<GeomAlgoAPI_MakeShapeList>(
134 new GeomAlgoAPI_MakeShapeList(aListOfMakeShape));
136 LoadNamingDS(aResultBody, anObject, aTools, aBoolAlgo);
137 setResult(aResultBody, aResultIndex);
143 case GeomAlgoAPI_Boolean::BOOL_FUSE: {
144 // Fuse all objects and all tools.
145 GeomAlgoAPI_Boolean aBoolAlgo(anObjects, aTools, aType);
147 // Checking that the algorithm worked properly.
148 if(!aBoolAlgo.isDone()) {
149 static const std::string aFeatureError = "Boolean algorithm failed";
150 setError(aFeatureError);
153 if(aBoolAlgo.shape()->isNull()) {
154 static const std::string aShapeError = "Resulting shape is Null";
155 setError(aShapeError);
158 if(!aBoolAlgo.isValid()) {
159 std::string aFeatureError = "Warning: resulting shape is not valid";
160 setError(aFeatureError);
164 std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data(), aResultIndex);
165 std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList = std::shared_ptr<GeomAlgoAPI_MakeShapeList>(
166 new GeomAlgoAPI_MakeShapeList(aListOfMakeShape));
168 LoadNamingDS(aResultBody, anObjects.front(), aTools, aBoolAlgo);
169 setResult(aResultBody, aResultIndex);
174 std::string anOperationError = "Error: wrong type of operation";
175 setError(anOperationError);
179 // remove the rest results if there were produced in the previous pass
180 removeResults(aResultIndex);
183 //=================================================================================================
184 void FeaturesPlugin_Boolean::LoadNamingDS(std::shared_ptr<ModelAPI_ResultBody> theResultBody,
185 const std::shared_ptr<GeomAPI_Shape>& theBaseShape,
186 const ListOfShape& theTools,
187 const GeomAlgoAPI_Boolean& theAlgo)
190 if(theBaseShape->isEqual(theAlgo.shape())) {
191 theResultBody->store(theAlgo.shape());
193 theResultBody->storeModified(theBaseShape, theAlgo.shape());
195 GeomAPI_DataMapOfShapeShape* aSubShapes = new GeomAPI_DataMapOfShapeShape();
197 std::string aModName = "Modified";
198 theResultBody->loadAndOrientModifiedShapes(theAlgo.makeShape().get(), theBaseShape, FACE,
199 _MODIFY_TAG, aModName, *theAlgo.mapOfShapes().get());
200 theResultBody->loadDeletedShapes(theAlgo.makeShape().get(), theBaseShape, FACE, _DELETED_TAG);
202 for(ListOfShape::const_iterator anIter = theTools.begin(); anIter != theTools.end(); anIter++) {
203 theResultBody->loadAndOrientModifiedShapes(theAlgo.makeShape().get(), *anIter, FACE,
204 _MODIFY_TAG, aModName, *theAlgo.mapOfShapes().get());
205 theResultBody->loadDeletedShapes(theAlgo.makeShape().get(), *anIter, FACE, _DELETED_TAG);