X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FFeaturesPlugin%2FFeaturesPlugin_Boolean.cpp;h=96b7de83fde0c3401e1df443d948b78ef4aece15;hb=2fbe8125820a68dc99915086b03843f037ae22b6;hp=2932d5e5cce410fa963f068e5579554a3dcfc13c;hpb=8babac25037f2666dcb422dd68a66dd2ed1383d8;p=modules%2Fshaper.git diff --git a/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp b/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp index 2932d5e5c..96b7de83f 100644 --- a/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp +++ b/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp @@ -11,23 +11,39 @@ #include #include #include +#include + #include -using namespace std; +#include +#include #define FACE 4 #define _MODIFY_TAG 1 #define _DELETED_TAG 2 + +//================================================================================================= FeaturesPlugin_Boolean::FeaturesPlugin_Boolean() { } +//================================================================================================= void FeaturesPlugin_Boolean::initAttributes() { data()->addAttribute(FeaturesPlugin_Boolean::TYPE_ID(), ModelAPI_AttributeInteger::typeId()); - data()->addAttribute(FeaturesPlugin_Boolean::OBJECT_ID(), ModelAPI_AttributeReference::typeId()); - data()->addAttribute(FeaturesPlugin_Boolean::TOOL_ID(), ModelAPI_AttributeReference::typeId()); + + AttributeSelectionListPtr aSelection = + std::dynamic_pointer_cast(data()->addAttribute( + FeaturesPlugin_Boolean::OBJECT_LIST_ID(), ModelAPI_AttributeSelectionList::typeId())); + // extrusion works with faces always + aSelection->setSelectionType("SOLID"); + + aSelection = std::dynamic_pointer_cast(data()->addAttribute( + FeaturesPlugin_Boolean::TOOL_LIST_ID(), ModelAPI_AttributeSelectionList::typeId())); + // extrusion works with faces always + aSelection->setSelectionType("SOLID"); } +//================================================================================================= std::shared_ptr FeaturesPlugin_Boolean::getShape(const std::string& theAttrName) { std::shared_ptr aObjRef = std::dynamic_pointer_cast< @@ -41,75 +57,200 @@ std::shared_ptr FeaturesPlugin_Boolean::getShape(const std::strin return std::shared_ptr(); } - +//================================================================================================= void FeaturesPlugin_Boolean::execute() { + // Getting operation type. std::shared_ptr aTypeAttr = std::dynamic_pointer_cast< ModelAPI_AttributeInteger>(data()->attribute(FeaturesPlugin_Boolean::TYPE_ID())); if (!aTypeAttr) return; int aType = aTypeAttr->value(); - std::shared_ptr anObject = this->getShape(FeaturesPlugin_Boolean::OBJECT_ID()); - if (!anObject) - return; + ListOfShape anObjects, aTools; - std::shared_ptr aTool = this->getShape(FeaturesPlugin_Boolean::TOOL_ID()); - if (!aTool) - return; - - std::shared_ptr aResultBody = document()->createBody(data()); - - GeomAlgoAPI_Boolean* aFeature = new GeomAlgoAPI_Boolean(anObject, aTool, aType); - if(aFeature && !aFeature->isDone()) { - static const std::string aFeatureError = "Boolean feature: algorithm failed"; - setError(aFeatureError); + // Getting objects. + AttributeSelectionListPtr anObjectsSelList = selectionList(FeaturesPlugin_Boolean::OBJECT_LIST_ID()); + if (anObjectsSelList->size() == 0) { return; } - // Check if shape is valid - if (aFeature->shape()->isNull()) { - static const std::string aShapeError = "Boolean feature: resulting shape is Null"; - setError(aShapeError); - return; + for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) { + std::shared_ptr anObjectAttr = anObjectsSelList->value(anObjectsIndex); + std::shared_ptr anObject = anObjectAttr->value(); + if(!anObject.get()) { + return; + } + anObjects.push_back(anObject); } - if(!aFeature->isValid()) { - static const std::string aFeatureError = "Boolean feature: resulting shape is not valid"; - setError(aFeatureError); - return; - } - // if result of Boolean operation is same as was before it means that Boolean operation has no sence - // and naming provides no result, so, generate an error in this case - if (anObject->isEqual(aFeature->shape())) { - static const std::string aFeatureError = "Boolean feature: operation was not performed"; - setError(aFeatureError); + + // Getting tools. + AttributeSelectionListPtr aToolsSelList = selectionList(FeaturesPlugin_Boolean::TOOL_LIST_ID()); + if (aToolsSelList->size() == 0) { return; } - //LoadNamingDS - LoadNamingDS(aFeature, aResultBody, anObject, aTool, aType); + for(int aToolsIndex = 0; aToolsIndex < aToolsSelList->size(); aToolsIndex++) { + std::shared_ptr aToolAttr = aToolsSelList->value(aToolsIndex); + std::shared_ptr aTool = aToolAttr->value(); + if(!aTool.get()) { + return; + } + aTools.push_back(aTool); + } - setResult(aResultBody); -} + int aResultIndex = 0; + ListOfMakeShape aListOfMakeShape; + std::shared_ptr aResShape; + std::shared_ptr aDataMapOfShapes; + + switch(aType) { + case GeomAlgoAPI_Boolean::BOOL_CUT: { + // Cut each object with all tools + for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end(); anObjectsIt++) { + std::shared_ptr anObject = *anObjectsIt; + aResShape = anObject; + ListOfShape aSingleObjectList; aSingleObjectList.push_back(anObject); + for(ListOfShape::iterator aToolsIt = aTools.begin(); aToolsIt != aTools.end(); aToolsIt++) { + std::shared_ptr 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); + } + + if(GeomAlgoAPI_ShapeProps::volume(aResShape) > 10.e-5) { + std::shared_ptr aResultBody = document()->createBody(data(), aResultIndex); + std::shared_ptr aMakeShapeList = std::shared_ptr( + new GeomAlgoAPI_MakeShapeList(aListOfMakeShape)); + + LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, aSingleObjectList, aTools); + setResult(aResultBody, aResultIndex); + aResultIndex++; + } + } + break; + } + case GeomAlgoAPI_Boolean::BOOL_FUSE: { + // Fuse all objects. + std::shared_ptr aTempRes = anObjects.front(); + for(ListOfShape::iterator anObjectsIt = ++anObjects.begin(); anObjectsIt != anObjects.end(); anObjectsIt++) { + std::shared_ptr 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(); + } + + // Fuse all tools with result of objects fuse. + for(ListOfShape::iterator aToolsIt = aTools.begin(); aToolsIt != aTools.end(); aToolsIt++) { + std::shared_ptr 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); + } + aResShape = aTempRes; + + std::shared_ptr aResultBody = document()->createBody(data()); + std::shared_ptr aMakeShapeList = std::shared_ptr( + new GeomAlgoAPI_MakeShapeList(aListOfMakeShape)); + + LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, anObjects, aTools); + setResult(aResultBody); + 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 anObject = *anObjectsIt; + std::shared_ptr aTempRes; + ListOfShape aSingleObjectList; aSingleObjectList.push_back(anObject); + for(ListOfShape::iterator aToolsIt = aTools.begin(); aToolsIt != aTools.end(); aToolsIt++) { + std::shared_ptr 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()); -//============================================================================ -void FeaturesPlugin_Boolean::LoadNamingDS(GeomAlgoAPI_Boolean* theFeature, - std::shared_ptr theResultBody, - std::shared_ptr theObject, - std::shared_ptr theTool, - int theType) -{ + 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 aResultBody = document()->createBody(data(), aResultIndex); + std::shared_ptr aMakeShapeList = std::shared_ptr( + new GeomAlgoAPI_MakeShapeList(aListOfMakeShape)); + + LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, aSingleObjectList, aTools); + setResult(aResultBody, aResultIndex); + aResultIndex++; + } + } + break; + } + } + // remove the rest results if there were produced in the previous pass + removeResults(aResultIndex); +} + +//================================================================================================= +void FeaturesPlugin_Boolean::LoadNamingDS(std::shared_ptr theMakeShapeList, + std::shared_ptr theResultBody, + std::shared_ptr theResult, + std::shared_ptr theDataMapOfShapes, + const ListOfShape& theObjects, + const ListOfShape& theTools) +{ //load result - theResultBody->storeModified(theObject, theFeature->shape()); + if(theObjects.front()->isEqual(theResult)) { + theResultBody->store(theObjects.front()); + } else { + theResultBody->storeModified(theObjects.front(), theResult); - GeomAPI_DataMapOfShapeShape* aSubShapes = new GeomAPI_DataMapOfShapeShape(); - theFeature->mapOfShapes(*aSubShapes); + GeomAPI_DataMapOfShapeShape* aSubShapes = new GeomAPI_DataMapOfShapeShape(); - // Put in DF modified faces - std::string aModName = "Modified"; - theResultBody->loadAndOrientModifiedShapes(theFeature->makeShape(), theObject, FACE, _MODIFY_TAG, aModName, *aSubShapes); - theResultBody->loadAndOrientModifiedShapes(theFeature->makeShape(), theTool, FACE, _MODIFY_TAG, aModName, *aSubShapes); + 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); + } - //Put in DF deleted faces - theResultBody->loadDeletedShapes(theFeature->makeShape(), theObject, FACE, _DELETED_TAG); - theResultBody->loadDeletedShapes(theFeature->makeShape(), theTool, 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); + } + } }