X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FFeaturesPlugin%2FFeaturesPlugin_Boolean.cpp;h=96b7de83fde0c3401e1df443d948b78ef4aece15;hb=2fbe8125820a68dc99915086b03843f037ae22b6;hp=220b4144c6e05edfdfe87eb12124309be7ea5e4f;hpb=b801d5f85aa3dc5786510acf87c8003f1d59eadd;p=modules%2Fshaper.git diff --git a/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp b/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp index 220b4144c..96b7de83f 100644 --- a/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp +++ b/src/FeaturesPlugin/FeaturesPlugin_Boolean.cpp @@ -1,3 +1,5 @@ +// Copyright (C) 2014-20xx CEA/DEN, EDF R&D --> + // File: FeaturesPlugin_Boolean.cpp // Created: 02 Sept 2014 // Author: Vitaly SMETANNIKOV @@ -7,47 +9,248 @@ #include #include #include +#include #include +#include + #include +#include +#include -using namespace std; +#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_AttributeReference::type()); - data()->addAttribute(FeaturesPlugin_Boolean::OBJECT_ID(), ModelAPI_AttributeReference::type()); - data()->addAttribute(FeaturesPlugin_Boolean::TOOL_ID(), ModelAPI_AttributeReference::type()); + data()->addAttribute(FeaturesPlugin_Boolean::TYPE_ID(), ModelAPI_AttributeInteger::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"); } -boost::shared_ptr FeaturesPlugin_Boolean::getShape(const std::string& theAttrName) +//================================================================================================= +std::shared_ptr FeaturesPlugin_Boolean::getShape(const std::string& theAttrName) { - boost::shared_ptr aObjRef = boost::dynamic_pointer_cast< + std::shared_ptr aObjRef = std::dynamic_pointer_cast< ModelAPI_AttributeReference>(data()->attribute(theAttrName)); if (aObjRef) { - boost::shared_ptr aConstr = boost::dynamic_pointer_cast< + std::shared_ptr aConstr = std::dynamic_pointer_cast< ModelAPI_ResultBody>(aObjRef->value()); if (aConstr) return aConstr->shape(); } - return boost::shared_ptr(); + return std::shared_ptr(); } - +//================================================================================================= void FeaturesPlugin_Boolean::execute() { - boost::shared_ptr aObject = this->getShape(FeaturesPlugin_Boolean::OBJECT_ID()); - if (!aObject) + // 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(); + + ListOfShape anObjects, aTools; - boost::shared_ptr aTool = this->getShape(FeaturesPlugin_Boolean::TOOL_ID()); - if (!aTool) + // 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); + std::shared_ptr anObject = anObjectAttr->value(); + if(!anObject.get()) { + return; + } + 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); + std::shared_ptr aTool = aToolAttr->value(); + if(!aTool.get()) { + return; + } + aTools.push_back(aTool); + } + + 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)); - boost::shared_ptr aResult = document()->createBody(data()); - aResult->store(GeomAlgoAPI_Boolean::makeCut(aObject, aTool)); - setResult(aResult); -} \ No newline at end of file + 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()); + + 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 + if(theObjects.front()->isEqual(theResult)) { + theResultBody->store(theObjects.front()); + } else { + 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); + } + } +}