X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FFeaturesPlugin%2FFeaturesPlugin_Boolean.cpp;h=9ab80ccfe545f2bc959017efead61a5a284e6093;hb=020e7187ee88afbd18286f149536cdef61d9f61e;hp=471e5c7c08d0ba2eadb823da7e6a483e1e8469cf;hpb=5e90be5a37588dd936e6d5f1a6426141bfda7170;p=modules%2Fshaper.git diff --git a/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp b/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp index 471e5c7c0..9ab80ccfe 100644 --- a/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp +++ b/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp @@ -10,16 +10,20 @@ #include #include #include +#include #include #include +#include +#include +#include #include #include -#include +#include +#include +#include -#define FACE 4 -#define _MODIFY_TAG 1 -#define _DELETED_TAG 2 +#include //================================================================================================= FeaturesPlugin_Boolean::FeaturesPlugin_Boolean() @@ -41,6 +45,9 @@ void FeaturesPlugin_Boolean::initAttributes() 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()); } //================================================================================================= @@ -65,31 +72,42 @@ void FeaturesPlugin_Boolean::execute() 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; + std::map, ListOfShape> aCompSolidsObjects; // 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 anObjectAttr = anObjectsSelList->value(anObjectsIndex); + AttributeSelectionPtr anObjectAttr = anObjectsSelList->value(anObjectsIndex); std::shared_ptr anObject = anObjectAttr->value(); if(!anObject.get()) { return; } - anObjects.push_back(anObject); + ResultPtr aContext = anObjectAttr->context(); + ResultCompSolidPtr aResCompSolidPtr = ModelAPI_Tools::compSolidOwner(aContext); + if(aResCompSolidPtr.get()) { + std::shared_ptr aContextShape = aResCompSolidPtr->shape(); + std::map, ListOfShape>::iterator anIt = aCompSolidsObjects.begin(); + for(; anIt != aCompSolidsObjects.end(); anIt++) { + if(anIt->first->isEqual(aContextShape)) { + aCompSolidsObjects[anIt->first].push_back(anObject); + break; + } + } + if(anIt == aCompSolidsObjects.end()) { + aCompSolidsObjects[aContextShape].push_back(anObject); + } + } else { + anObjects.push_back(anObject); + } } // 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 aToolAttr = aToolsSelList->value(aToolsIndex); + AttributeSelectionPtr aToolAttr = aToolsSelList->value(aToolsIndex); std::shared_ptr aTool = aToolAttr->value(); if(!aTool.get()) { return; @@ -98,155 +116,268 @@ void FeaturesPlugin_Boolean::execute() } 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 + case GeomAlgoAPI_Boolean::BOOL_CUT: + case GeomAlgoAPI_Boolean::BOOL_COMMON:{ + if((anObjects.empty() && aCompSolidsObjects.empty()) || aTools.empty()) { + std::string aFeatureError = "Not enough objects for boolean operation"; + setError(aFeatureError); + return; + } + + // For solids 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); + 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_ShapeTools::volume(aBoolAlgo.shape()) > 1.e-7) { 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); + loadNamingDS(aResultBody, anObject, aTools, aBoolAlgo.shape(), *aBoolAlgo.makeShape(), *aBoolAlgo.mapOfShapes()); 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); + + // Compsolids handling + for(std::map, ListOfShape>::iterator anIt = aCompSolidsObjects.begin(); + anIt != aCompSolidsObjects.end(); anIt++) { + std::shared_ptr aCompSolid = anIt->first; + ListOfShape& aUsedInOperationSolids = anIt->second; + + // Collecting solids from compsolids which will not be modified in boolean operation. + ListOfShape aNotUsedSolids; + for(GeomAPI_ShapeExplorer anExp(aCompSolid, GeomAPI_Shape::SOLID); anExp.more(); anExp.next()) { + std::shared_ptr aSolidInCompSolid = anExp.current(); + ListOfShape::iterator anIt = aUsedInOperationSolids.begin(); + for(; anIt != aUsedInOperationSolids.end(); anIt++) { + if(aSolidInCompSolid->isEqual(*anIt)) { + break; + } + } + if(anIt == aUsedInOperationSolids.end()) { + aNotUsedSolids.push_back(aSolidInCompSolid); + } + } + + GeomAlgoAPI_Boolean aBoolAlgo(aUsedInOperationSolids, 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; } - 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); + GeomAlgoAPI_MakeShapeList aMakeShapeList; + aMakeShapeList.append(aBoolAlgo.makeShape()); + GeomAPI_DataMapOfShapeShape aMapOfShapes; + aMapOfShapes.merge(aBoolAlgo.mapOfShapes()); + + // Add result to not used solids from compsolid. + ListOfShape aShapesToAdd = aNotUsedSolids; + aShapesToAdd.push_back(aBoolAlgo.shape()); + GeomAlgoAPI_PaveFiller aFillerAlgo(aShapesToAdd, true); + if(!aFillerAlgo.isDone()) { + std::string aFeatureError = "PaveFiller algorithm failed"; + setError(aFeatureError); 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)); + aMakeShapeList.append(aFillerAlgo.makeShape()); + aMapOfShapes.merge(aFillerAlgo.mapOfShapes()); - LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, anObjects, aTools); - setResult(aResultBody); - aResultIndex++; + if(GeomAlgoAPI_ShapeTools::volume(aFillerAlgo.shape()) > 1.e-7) { + std::shared_ptr aResultBody = document()->createBody(data(), aResultIndex); + loadNamingDS(aResultBody, aCompSolid, aTools, aFillerAlgo.shape(), aMakeShapeList, aMapOfShapes); + 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 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()); - - 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; + case GeomAlgoAPI_Boolean::BOOL_FUSE: { + if((anObjects.size() + aTools.size() + aCompSolidsObjects.size()) < 2) { + std::string aFeatureError = "Not enough objects for boolean operation"; + setError(aFeatureError); + return; + } + + // Collecting all solids which will be fused. + ListOfShape aSolidsToFuse; + aSolidsToFuse.insert(aSolidsToFuse.end(), anObjects.begin(), anObjects.end()); + aSolidsToFuse.insert(aSolidsToFuse.end(), aTools.begin(), aTools.end()); + + // Collecting solids from compsolids which will not be modified in boolean operation. + ListOfShape aNotUsedSolids; + for(std::map, ListOfShape>::iterator anIt = aCompSolidsObjects.begin(); + anIt != aCompSolidsObjects.end(); anIt++) { + std::shared_ptr aCompSolid = anIt->first; + ListOfShape& aUsedInOperationSolids = anIt->second; + aSolidsToFuse.insert(aSolidsToFuse.end(), aUsedInOperationSolids.begin(), aUsedInOperationSolids.end()); + + // Collect solids from compsolid which will not be modified in boolean operation. + for(GeomAPI_ShapeExplorer anExp(aCompSolid, GeomAPI_Shape::SOLID); anExp.more(); anExp.next()) { + std::shared_ptr aSolidInCompSolid = anExp.current(); + ListOfShape::iterator anIt = aUsedInOperationSolids.begin(); + for(; anIt != aUsedInOperationSolids.end(); anIt++) { + if(aSolidInCompSolid->isEqual(*anIt)) { + break; } - aListOfMakeShape.push_back(aBoolAlgo2->makeShape()); - aTempRes = aBoolAlgo2->shape(); - aBoolAlgo2->mapOfShapes(aDataMapOfShapes); + } + if(anIt == aUsedInOperationSolids.end()) { + aNotUsedSolids.push_back(aSolidInCompSolid); } } - 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)); + ListOfShape anOriginalSolids = aSolidsToFuse; + anOriginalSolids.insert(anOriginalSolids.end(), aNotUsedSolids.begin(), aNotUsedSolids.end()); + GeomAlgoAPI_MakeShapeList aMakeShapeList; + GeomAPI_DataMapOfShapeShape aMapOfShapes; - LoadNamingDS(aMakeShapeList, aResultBody, aResShape, aDataMapOfShapes, aSingleObjectList, aTools); - setResult(aResultBody, aResultIndex); - aResultIndex++; + // If we have compsolids then cut with not used solids all others. + if(!aNotUsedSolids.empty()) { + aSolidsToFuse.clear(); + for(ListOfShape::iterator anIt = anOriginalSolids.begin(); anIt != anOriginalSolids.end(); anIt++) { + ListOfShape aOneObjectList; + aOneObjectList.push_back(*anIt); + GeomAlgoAPI_Boolean aCutAlgo(aOneObjectList, aNotUsedSolids, GeomAlgoAPI_Boolean::BOOL_CUT); + + if(GeomAlgoAPI_ShapeTools::volume(aCutAlgo.shape()) > 1.e-7) { + aSolidsToFuse.push_back(aCutAlgo.shape()); + aMakeShapeList.append(aCutAlgo.makeShape()); + aMapOfShapes.merge(aCutAlgo.mapOfShapes()); + } + } + } + + anObjects.clear(); + anObjects.push_back(aSolidsToFuse.back()); + aSolidsToFuse.pop_back(); + aTools = aSolidsToFuse; + + // Fuse all objects and all tools. + GeomAlgoAPI_Boolean aFuseAlgo(anObjects, aTools, aType); + + // Checking that the algorithm worked properly. + if(!aFuseAlgo.isDone()) { + static const std::string aFeatureError = "Boolean algorithm failed"; + setError(aFeatureError); + return; + } + if(aFuseAlgo.shape()->isNull()) { + static const std::string aShapeError = "Resulting shape is Null"; + setError(aShapeError); + return; + } + if(!aFuseAlgo.isValid()) { + std::string aFeatureError = "Warning: resulting shape is not valid"; + setError(aFeatureError); + return; + } + + std::shared_ptr aShape = aFuseAlgo.shape(); + aMakeShapeList.append(aFuseAlgo.makeShape()); + aMapOfShapes.merge(aFuseAlgo.mapOfShapes()); + + // Add result to not used solids from compsolid (if we have any). + if(!aNotUsedSolids.empty()) { + aNotUsedSolids.push_back(aShape); + GeomAlgoAPI_PaveFiller aFillerAlgo(aNotUsedSolids, true); + if(!aFillerAlgo.isDone()) { + std::string aFeatureError = "PaveFiller algorithm failed"; + setError(aFeatureError); + return; + } + if(aFillerAlgo.shape()->isNull()) { + static const std::string aShapeError = "Resulting shape is Null"; + setError(aShapeError); + return; + } + if(!aFillerAlgo.isValid()) { + std::string aFeatureError = "Warning: resulting shape is not valid"; + setError(aFeatureError); + return; } + + aShape = aFillerAlgo.shape(); + aMakeShapeList.append(aFillerAlgo.makeShape()); + aMapOfShapes.merge(aFillerAlgo.mapOfShapes()); } + + std::shared_ptr aResultBody = document()->createBody(data(), aResultIndex); + loadNamingDS(aResultBody, anOriginalSolids.front(), anOriginalSolids, aShape, aMakeShapeList, aMapOfShapes); + 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 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) +void FeaturesPlugin_Boolean::loadNamingDS(std::shared_ptr theResultBody, + const std::shared_ptr theBaseShape, + const ListOfShape& theTools, + const std::shared_ptr theResultShape, + GeomAlgoAPI_MakeShape& theMakeShape, + GeomAPI_DataMapOfShapeShape& theMapOfShapes) { //load result - theResultBody->storeModified(theObjects.front(), theResult); + if(theBaseShape->isEqual(theResultShape)) { + theResultBody->store(theResultShape); + } else { + const int aModifyTag = 1; + const int aDeletedTag = 2; + const int aSubsolidsTag = 3; /// sub solids will be placed at labels 3, 4, etc. if result is compound of solids - GeomAPI_DataMapOfShapeShape* aSubShapes = new GeomAPI_DataMapOfShapeShape(); + theResultBody->storeModified(theBaseShape, theResultShape, aSubsolidsTag); - 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); - } + std::string aModName = "Modified"; + theResultBody->loadAndOrientModifiedShapes(&theMakeShape, theBaseShape, GeomAPI_Shape::FACE, + aModifyTag, aModName, theMapOfShapes); + theResultBody->loadDeletedShapes(&theMakeShape, theBaseShape, GeomAPI_Shape::FACE, aDeletedTag); - 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); + for(ListOfShape::const_iterator anIter = theTools.begin(); anIter != theTools.end(); anIter++) { + theResultBody->loadAndOrientModifiedShapes(&theMakeShape, *anIter, GeomAPI_Shape::FACE, + aModifyTag, aModName, theMapOfShapes); + theResultBody->loadDeletedShapes(&theMakeShape, *anIter, GeomAPI_Shape::FACE, aDeletedTag); + } } }