#include <ModelAPI_Document.h>
#include <ModelAPI_AttributeReference.h>
#include <ModelAPI_AttributeInteger.h>
+#include <ModelAPI_BodyBuilder.h>
#include <ModelAPI_ResultBody.h>
#include <ModelAPI_AttributeSelectionList.h>
+#include <ModelAPI_Session.h>
+#include <ModelAPI_Validator.h>
#include <GeomAlgoAPI_Boolean.h>
#include <GeomAlgoAPI_MakeShapeList.h>
#define FACE 4
#define _MODIFY_TAG 1
#define _DELETED_TAG 2
+#define _SUBSOLIDS_TAG 3 /// sub solids will be placed at labels 3, 4, etc. if result is compound of solids
//=================================================================================================
FeaturesPlugin_Boolean::FeaturesPlugin_Boolean()
FeaturesPlugin_Boolean::TOOL_LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
// extrusion works with faces always
aSelection->setSelectionType("SOLID");
+
+ ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), OBJECT_LIST_ID());
+ ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), TOOL_LIST_ID());
}
//=================================================================================================
ModelAPI_AttributeInteger>(data()->attribute(FeaturesPlugin_Boolean::TYPE_ID()));
if (!aTypeAttr)
return;
- int aType = aTypeAttr->value();
+ GeomAlgoAPI_Boolean::OperationType aType = (GeomAlgoAPI_Boolean::OperationType)aTypeAttr->value();
ListOfShape anObjects, aTools;
// Getting objects.
AttributeSelectionListPtr anObjectsSelList = selectionList(FeaturesPlugin_Boolean::OBJECT_LIST_ID());
- if (anObjectsSelList->size() == 0) {
- return;
- }
for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) {
std::shared_ptr<ModelAPI_AttributeSelection> anObjectAttr = anObjectsSelList->value(anObjectsIndex);
std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
// Getting tools.
AttributeSelectionListPtr aToolsSelList = selectionList(FeaturesPlugin_Boolean::TOOL_LIST_ID());
- if (aToolsSelList->size() == 0) {
- return;
- }
for(int aToolsIndex = 0; aToolsIndex < aToolsSelList->size(); aToolsIndex++) {
std::shared_ptr<ModelAPI_AttributeSelection> aToolAttr = aToolsSelList->value(aToolsIndex);
std::shared_ptr<GeomAPI_Shape> aTool = aToolAttr->value();
}
int aResultIndex = 0;
- ListOfMakeShape aListOfMakeShape;
- std::shared_ptr<GeomAPI_Shape> aResShape;
- std::shared_ptr<GeomAPI_DataMapOfShapeShape> aDataMapOfShapes;
switch(aType) {
- case GeomAlgoAPI_Boolean::BOOL_CUT: {
+ case GeomAlgoAPI_Boolean::BOOL_CUT:
+ case GeomAlgoAPI_Boolean::BOOL_COMMON:{
+ if(anObjects.empty() || aTools.empty()) {
+ std::string aFeatureError = "Not enough objects for boolean operation";
+ setError(aFeatureError);
+ return;
+ }
+
// Cut each object with all tools
for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end(); anObjectsIt++) {
std::shared_ptr<GeomAPI_Shape> anObject = *anObjectsIt;
- aResShape = anObject;
- ListOfShape aSingleObjectList; aSingleObjectList.push_back(anObject);
- for(ListOfShape::iterator aToolsIt = aTools.begin(); aToolsIt != aTools.end(); aToolsIt++) {
- std::shared_ptr<GeomAPI_Shape> aTool = *aToolsIt;
- GeomAlgoAPI_Boolean* aBoolAlgo = new GeomAlgoAPI_Boolean(aResShape, aTool, aType);
- if(!aBoolAlgo->isDone() || aBoolAlgo->shape()->isNull() || !aBoolAlgo->isValid()) {
- static const std::string aBoolAlgoError = "Boolean feature: algorithm failed";
- setError(aBoolAlgoError);
- return;
- }
- aListOfMakeShape.push_back(aBoolAlgo->makeShape());
- aResShape = aBoolAlgo->shape();
- aBoolAlgo->mapOfShapes(aDataMapOfShapes);
+ ListOfShape aListWithObject;
+ aListWithObject.push_back(anObject);
+ GeomAlgoAPI_Boolean aBoolAlgo(aListWithObject, aTools, aType);
+
+ // Checking that the algorithm worked properly.
+ if(!aBoolAlgo.isDone()) {
+ static const std::string aFeatureError = "Boolean algorithm failed";
+ setError(aFeatureError);
+ return;
+ }
+ if(aBoolAlgo.shape()->isNull()) {
+ static const std::string aShapeError = "Resulting shape is Null";
+ setError(aShapeError);
+ return;
+ }
+ if(!aBoolAlgo.isValid()) {
+ std::string aFeatureError = "Warning: resulting shape is not valid";
+ setError(aFeatureError);
+ return;
}
- if(GeomAlgoAPI_ShapeProps::volume(aResShape) > 10.e-5) {
+ if(GeomAlgoAPI_ShapeProps::volume(aBoolAlgo.shape()) > 1.e-7) {
std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data(), aResultIndex);
- std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList = std::shared_ptr<GeomAlgoAPI_MakeShapeList>(
- new GeomAlgoAPI_MakeShapeList(aListOfMakeShape));
-
- LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, aSingleObjectList, aTools);
+ LoadNamingDS(aResultBody, anObject, aTools, aBoolAlgo);
setResult(aResultBody, aResultIndex);
aResultIndex++;
}
break;
}
case GeomAlgoAPI_Boolean::BOOL_FUSE: {
- // Fuse all objects.
- std::shared_ptr<GeomAPI_Shape> aTempRes = anObjects.front();
- for(ListOfShape::iterator anObjectsIt = ++anObjects.begin(); anObjectsIt != anObjects.end(); anObjectsIt++) {
- std::shared_ptr<GeomAPI_Shape> anObject = *anObjectsIt;
- GeomAlgoAPI_Boolean* aBoolAlgo = new GeomAlgoAPI_Boolean(aTempRes, anObject, aType);
- if(!aBoolAlgo->isDone() || aBoolAlgo->shape()->isNull() || !aBoolAlgo->isValid()) {
- static const std::string aBoolAlgoError = "Boolean feature: algorithm failed";
- setError(aBoolAlgoError);
- return;
- }
- aListOfMakeShape.push_back(aBoolAlgo->makeShape());
- aTempRes = aBoolAlgo->shape();
+ if(anObjects.empty() && aTools.size() > 1) {
+ anObjects.push_back(aTools.back());
+ aTools.pop_back();
+ }else if(aTools.empty() && anObjects.size() > 1) {
+ aTools.push_back(anObjects.back());
+ anObjects.pop_back();
}
- // Fuse all tools with result of objects fuse.
- for(ListOfShape::iterator aToolsIt = aTools.begin(); aToolsIt != aTools.end(); aToolsIt++) {
- std::shared_ptr<GeomAPI_Shape> aTool = *aToolsIt;
- GeomAlgoAPI_Boolean* aBoolAlgo = new GeomAlgoAPI_Boolean(aTempRes, aTool, aType);
- if(!aBoolAlgo->isDone() || aBoolAlgo->shape()->isNull() || !aBoolAlgo->isValid()) {
- static const std::string aBoolAlgoError = "Boolean feature: algorithm failed";
- setError(aBoolAlgoError);
- return;
- }
- aListOfMakeShape.push_back(aBoolAlgo->makeShape());
- aTempRes = aBoolAlgo->shape();
- aBoolAlgo->mapOfShapes(aDataMapOfShapes);
+ if(anObjects.empty() || aTools.empty()) {
+ std::string aFeatureError = "Not enough objects for boolean operation";
+ setError(aFeatureError);
+ return;
}
- aResShape = aTempRes;
- std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data());
- std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList = std::shared_ptr<GeomAlgoAPI_MakeShapeList>(
- new GeomAlgoAPI_MakeShapeList(aListOfMakeShape));
+ // Fuse all objects and all tools.
+ GeomAlgoAPI_Boolean aBoolAlgo(anObjects, aTools, aType);
- LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, anObjects, aTools);
- setResult(aResultBody);
+ // Checking that the algorithm worked properly.
+ if(!aBoolAlgo.isDone()) {
+ static const std::string aFeatureError = "Boolean algorithm failed";
+ setError(aFeatureError);
+ return;
+ }
+ if(aBoolAlgo.shape()->isNull()) {
+ static const std::string aShapeError = "Resulting shape is Null";
+ setError(aShapeError);
+ return;
+ }
+ if(!aBoolAlgo.isValid()) {
+ std::string aFeatureError = "Warning: resulting shape is not valid";
+ setError(aFeatureError);
+ return;
+ }
+
+ std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data(), aResultIndex);
+ LoadNamingDS(aResultBody, anObjects.front(), aTools, aBoolAlgo);
+ setResult(aResultBody, aResultIndex);
aResultIndex++;
break;
}
- case GeomAlgoAPI_Boolean::BOOL_COMMON: {
- // Search common between object and tool and fuse them.
- for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end(); anObjectsIt++) {
- std::shared_ptr<GeomAPI_Shape> anObject = *anObjectsIt;
- std::shared_ptr<GeomAPI_Shape> aTempRes;
- ListOfShape aSingleObjectList; aSingleObjectList.push_back(anObject);
- for(ListOfShape::iterator aToolsIt = aTools.begin(); aToolsIt != aTools.end(); aToolsIt++) {
- std::shared_ptr<GeomAPI_Shape> aTool = *aToolsIt;
- GeomAlgoAPI_Boolean* aBoolAlgo1 = new GeomAlgoAPI_Boolean(anObject, aTool, aType);
- if(!aBoolAlgo1->isDone() || aBoolAlgo1->shape()->isNull() || !aBoolAlgo1->isValid()) {
- static const std::string aBoolAlgoError = "Boolean feature: algorithm failed";
- setError(aBoolAlgoError);
- return;
- }
- aListOfMakeShape.push_back(aBoolAlgo1->makeShape());
-
- if(!aTempRes) {
- aTempRes = aBoolAlgo1->shape();
- aBoolAlgo1->mapOfShapes(aDataMapOfShapes);
- } else {
- // Fuse common result with previous common results.
- GeomAlgoAPI_Boolean* aBoolAlgo2 = new GeomAlgoAPI_Boolean(aTempRes,
- aBoolAlgo1->shape(),
- GeomAlgoAPI_Boolean::BOOL_FUSE);
- if(!aBoolAlgo1->isDone() || aBoolAlgo1->shape()->isNull() || !aBoolAlgo1->isValid()) {
- static const std::string aBoolAlgoError = "Boolean feature: algorithm failed";
- setError(aBoolAlgoError);
- return;
- }
- aListOfMakeShape.push_back(aBoolAlgo2->makeShape());
- aTempRes = aBoolAlgo2->shape();
- aBoolAlgo2->mapOfShapes(aDataMapOfShapes);
- }
- }
- aResShape = aTempRes;
-
- if(GeomAlgoAPI_ShapeProps::volume(aResShape) > 10.e-5) {
- std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data(), aResultIndex);
- std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList = std::shared_ptr<GeomAlgoAPI_MakeShapeList>(
- new GeomAlgoAPI_MakeShapeList(aListOfMakeShape));
-
- LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, aSingleObjectList, aTools);
- setResult(aResultBody, aResultIndex);
- aResultIndex++;
- }
- }
- break;
+ default: {
+ std::string anOperationError = "Error: wrong type of operation";
+ setError(anOperationError);
+ return;
}
}
// remove the rest results if there were produced in the previous pass
}
//=================================================================================================
-void FeaturesPlugin_Boolean::LoadNamingDS(std::shared_ptr<GeomAlgoAPI_MakeShapeList> theMakeShapeList,
- std::shared_ptr<ModelAPI_ResultBody> theResultBody,
- std::shared_ptr<GeomAPI_Shape> theResult,
- std::shared_ptr<GeomAPI_DataMapOfShapeShape> theDataMapOfShapes,
- const ListOfShape& theObjects,
- const ListOfShape& theTools)
+void FeaturesPlugin_Boolean::LoadNamingDS(std::shared_ptr<ModelAPI_ResultBody> theResultBody,
+ const std::shared_ptr<GeomAPI_Shape>& theBaseShape,
+ const ListOfShape& theTools,
+ const GeomAlgoAPI_Boolean& theAlgo)
{
+ ModelAPI_BodyBuilder* aResultBuilder = theResultBody->getBodyBuilder();
//load result
- theResultBody->storeModified(theObjects.front(), theResult);
-
- GeomAPI_DataMapOfShapeShape* aSubShapes = new GeomAPI_DataMapOfShapeShape();
-
- std::string aModName = "Modified";
- for(ListOfShape::const_iterator anIter = theObjects.begin(); anIter != theObjects.end(); anIter++) {
- theResultBody->loadAndOrientModifiedShapes(theMakeShapeList.get(), *anIter, FACE, _MODIFY_TAG, aModName, *theDataMapOfShapes.get());
- theResultBody->loadDeletedShapes(theMakeShapeList.get(), *anIter, FACE, _DELETED_TAG);
- }
-
- for(ListOfShape::const_iterator anIter = theTools.begin(); anIter != theTools.end(); anIter++) {
- theResultBody->loadAndOrientModifiedShapes(theMakeShapeList.get(), *anIter, FACE, _MODIFY_TAG, aModName, *theDataMapOfShapes.get());
- theResultBody->loadDeletedShapes(theMakeShapeList.get(), *anIter, FACE, _DELETED_TAG);
+ if(theBaseShape->isEqual(theAlgo.shape())) {
+ aResultBuilder->store(theAlgo.shape());
+ } else {
+ aResultBuilder->storeModified(theBaseShape, theAlgo.shape(), _SUBSOLIDS_TAG);
+
+ GeomAPI_DataMapOfShapeShape* aSubShapes = new GeomAPI_DataMapOfShapeShape();
+
+ std::string aModName = "Modified";
+ aResultBuilder->loadAndOrientModifiedShapes(theAlgo.makeShape().get(), theBaseShape, FACE,
+ _MODIFY_TAG, aModName, *theAlgo.mapOfShapes().get());
+ aResultBuilder->loadDeletedShapes(theAlgo.makeShape().get(), theBaseShape, FACE, _DELETED_TAG);
+
+ for(ListOfShape::const_iterator anIter = theTools.begin(); anIter != theTools.end(); anIter++) {
+ aResultBuilder->loadAndOrientModifiedShapes(theAlgo.makeShape().get(), *anIter, FACE,
+ _MODIFY_TAG, aModName, *theAlgo.mapOfShapes().get());
+ aResultBuilder->loadDeletedShapes(theAlgo.makeShape().get(), *anIter, FACE, _DELETED_TAG);
+ }
}
}
