-// Copyright (C) 2014-2019 CEA/DEN, EDF R&D
+// Copyright (C) 2014-2022 CEA/DEN, EDF R&D
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
#include <GeomAlgoAPI_MakeShapeCustom.h>
#include <GeomAlgoAPI_MakeShapeList.h>
#include <GeomAlgoAPI_Partition.h>
+#include <GeomAlgoAPI_ShapeBuilder.h>
#include <GeomAlgoAPI_ShapeTools.h>
#include <GeomAlgoAPI_Tools.h>
#include <list>
#include <sstream>
-typedef std::list<std::pair<GeomShapePtr, ListOfShape> > CompsolidSubs;
-
-static void pullObjectsAndPlanes(const AttributeSelectionListPtr& theSelectedList,
- CompsolidSubs& theObjects, ListOfShape& thePlanes);
-
-static void resizePlanes(const CompsolidSubs& theObjects, ListOfShape& thePlanes,
- std::shared_ptr<GeomAlgoAPI_MakeShapeList>& theMakeShapeList);
-
-static void unusedSubsOfComposolid(const CompsolidSubs& theObjects, CompsolidSubs& theNotUsed);
-
-static bool cutUnusedSubs(CompsolidSubs& theObjects, CompsolidSubs& theNotUsed,
- std::shared_ptr<GeomAlgoAPI_MakeShapeList>& theMakeShapeList,
- std::string& theError);
-
//=================================================================================================
FeaturesPlugin_Partition::FeaturesPlugin_Partition()
void FeaturesPlugin_Partition::initAttributes()
{
data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
+ initVersion(BOP_VERSION_9_4(), selectionList(BASE_OBJECTS_ID()));
}
//=================================================================================================
void FeaturesPlugin_Partition::execute()
{
- CompsolidSubs anObjects;
+ GeomAPI_ShapeHierarchy anObjects;
ListOfShape aPlanes;
// Getting objects.
- pullObjectsAndPlanes(selectionList(BASE_OBJECTS_ID()), anObjects, aPlanes);
+ processAttribute(BASE_OBJECTS_ID(), anObjects, aPlanes);
if(anObjects.empty()) {
static const std::string aFeatureError = "Error: No objects for partition.";
setError(aFeatureError);
return;
}
- ListOfShape aBaseObjects;
- for (CompsolidSubs::iterator anIt = anObjects.begin(); anIt != anObjects.end(); ++anIt)
- aBaseObjects.insert(aBaseObjects.end(), anIt->second.begin(), anIt->second.end());
+ ListOfShape aBaseObjects = anObjects.objects();
aBaseObjects.insert(aBaseObjects.end(), aPlanes.begin(), aPlanes.end());
// resize planes to the bounding box of operated shapes
std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList(new GeomAlgoAPI_MakeShapeList());
- resizePlanes(anObjects, aPlanes, aMakeShapeList);
+ resizePlanes(anObjects.objects(), aPlanes, aMakeShapeList);
// cut unused solids of composolids from the objects of partition
- CompsolidSubs anUnusedSubs;
- unusedSubsOfComposolid(anObjects, anUnusedSubs);
- for (CompsolidSubs::iterator anIt = anUnusedSubs.begin(); anIt != anUnusedSubs.end(); ++anIt)
- aBaseObjects.insert(aBaseObjects.end(), anIt->second.begin(), anIt->second.end());
-
+ ListOfShape aTargetObjects, anUnusedSubs;
std::string aError;
- if (!cutUnusedSubs(anObjects, anUnusedSubs, aMakeShapeList, aError)) {
+ if (!cutSubs(anObjects, aTargetObjects, anUnusedSubs, aMakeShapeList, aError)) {
setError(aError);
return;
}
+ aBaseObjects.insert(aBaseObjects.end(), anUnusedSubs.begin(), anUnusedSubs.end());
- // perform partition first time to split target solids
- ListOfShape aTargetObjects;
- for (CompsolidSubs::iterator anIt = anObjects.begin(); anIt != anObjects.end(); ++anIt)
- aTargetObjects.insert(aTargetObjects.end(), anIt->second.begin(), anIt->second.end());
-
+ // perform partition first time to split target solids by planes
std::shared_ptr<GeomAlgoAPI_Partition> aPartitionAlgo(
- new GeomAlgoAPI_Partition(aTargetObjects, aPlanes));
+ new GeomAlgoAPI_Partition(aTargetObjects, aPlanes));
// Checking that the algorithm worked properly.
if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aPartitionAlgo, getKind(), aError)) {
// second pass of a partition to split shared faces of compsolids
aTargetObjects.clear();
aTargetObjects.push_back(aResultShape);
- for (CompsolidSubs::iterator anIt = anUnusedSubs.begin(); anIt != anUnusedSubs.end(); ++anIt)
- aTargetObjects.insert(aTargetObjects.end(), anIt->second.begin(), anIt->second.end());
+ aTargetObjects.insert(aTargetObjects.end(), anUnusedSubs.begin(), anUnusedSubs.end());
aPartitionAlgo.reset(new GeomAlgoAPI_Partition(aTargetObjects, ListOfShape()));
}
int aResultIndex = 0;
- if(aResultShape->shapeType() == GeomAPI_Shape::COMPOUND) {
- for(GeomAPI_ShapeIterator anIt(aResultShape); anIt.more(); anIt.next()) {
- storeResult(aBaseObjects, aPlanes, anIt.current(), aMakeShapeList, aResultIndex);
+
+ if (data()->version().empty()) {
+ // default behaviors of Partition
+ if(aResultShape->shapeType() == GeomAPI_Shape::COMPOUND) {
+ for(GeomAPI_ShapeIterator anIt(aResultShape); anIt.more(); anIt.next()) {
+ storeResult(aBaseObjects, aPlanes, anIt.current(), aMakeShapeList, aResultIndex);
+ ++aResultIndex;
+ }
+ } else {
+ storeResult(aBaseObjects, aPlanes, aResultShape, aMakeShapeList, aResultIndex);
++aResultIndex;
}
- } else {
- storeResult(aBaseObjects, aPlanes, aResultShape, aMakeShapeList, aResultIndex);
+ }
+ else {
+ // merge hierarchies of compounds containing objects and tools
+ GeomShapePtr aFirstShape = aResultShape;
+ GeomAPI_ShapeIterator anIt;
+ if (aResultShape->shapeType() == GeomAPI_Shape::COMPOUND) {
+ anIt = GeomAPI_ShapeIterator(aResultShape);
+ aFirstShape = anIt.current();
+ anIt.next();
+ }
+
+ GeomShapePtr aResultCompound =
+ keepUnusedSubsOfCompound(aFirstShape, anObjects, GeomAPI_ShapeHierarchy(), aMakeShapeList);
+
+ if (anIt.more()) {
+ if (aResultCompound->shapeType() != GeomAPI_Shape::COMPOUND) {
+ // put the shape into compound
+ ListOfShape aShapes;
+ aShapes.push_back(aResultCompound);
+ aResultCompound = GeomAlgoAPI_CompoundBuilder::compound(aShapes);
+ }
+ std::shared_ptr<GeomAlgoAPI_ShapeBuilder> aBuilder(new GeomAlgoAPI_ShapeBuilder);
+ for (; anIt.more(); anIt.next())
+ aBuilder->add(aResultCompound, anIt.current());
+ aMakeShapeList->appendAlgo(aBuilder);
+ }
+
+ storeResult(aBaseObjects, aPlanes, aResultCompound, aMakeShapeList, aResultIndex);
++aResultIndex;
}
}
-//================= Auxiliary functions ===================================================
-
-static CompsolidSubs::iterator findOrAdd(CompsolidSubs& theList, const GeomShapePtr& theCompsolid)
-{
- CompsolidSubs::iterator aFound = theList.begin();
- for (; aFound != theList.end(); ++aFound)
- if (aFound->first == theCompsolid)
- break;
- if (aFound == theList.end()) {
- theList.push_back(std::pair<GeomShapePtr, ListOfShape>(theCompsolid, ListOfShape()));
- aFound = --theList.end();
- }
- return aFound;
-}
-
-void pullObjectsAndPlanes(const AttributeSelectionListPtr& theSelectedList,
- CompsolidSubs& theObjects, ListOfShape& thePlanes)
-{
- std::map<ResultBodyPtr, GeomShapePtr> aMapCompsolidShape;
-
- int aSize = theSelectedList->size();
- for (int anIndex = 0; anIndex < aSize; ++anIndex) {
- AttributeSelectionPtr anObjectAttr = theSelectedList->value(anIndex);
- ResultPtr aContext = anObjectAttr->context();
- GeomShapePtr anObject = anObjectAttr->value();
- if (anObject) {
- GeomShapePtr anOwnerShape = anObject;
- // check the result is a compsolid and store all used subs in a single list
- ResultBodyPtr aResCompSolidPtr = ModelAPI_Tools::bodyOwner(aContext);
- if (aResCompSolidPtr && aResCompSolidPtr->shape()->shapeType() == GeomAPI_Shape::COMPSOLID) {
- std::map<ResultBodyPtr, GeomShapePtr>::const_iterator
- aFound = aMapCompsolidShape.find(aResCompSolidPtr);
- if (aFound != aMapCompsolidShape.end())
- anOwnerShape = aFound->second;
- else {
- anOwnerShape = aResCompSolidPtr->shape();
- aMapCompsolidShape[aResCompSolidPtr] = anOwnerShape;
- }
- }
-
- CompsolidSubs::iterator aFound = findOrAdd(theObjects, anOwnerShape);
- aFound->second.push_back(anObject);
- }
- else {
- // It could be a construction plane.
- thePlanes.push_back(anObjectAttr->context()->shape());
- }
- }
-}
-
-void resizePlanes(const CompsolidSubs& theObjects, ListOfShape& thePlanes,
- std::shared_ptr<GeomAlgoAPI_MakeShapeList>& theMakeShapeList)
-{
- ListOfShape aSolidsInOperation;
- for (CompsolidSubs::const_iterator anIt = theObjects.begin(); anIt != theObjects.end(); ++anIt)
- aSolidsInOperation.insert(aSolidsInOperation.end(), anIt->second.begin(), anIt->second.end());
-
- std::list<std::shared_ptr<GeomAPI_Pnt> > aBoundingPoints =
- GeomAlgoAPI_ShapeTools::getBoundingBox(aSolidsInOperation, 1.0);
-
- ListOfShape aPlanesCopy = thePlanes;
- thePlanes.clear();
-
- // Resize planes to fit in bounding box
- for (ListOfShape::const_iterator anIt = aPlanesCopy.begin(); anIt != aPlanesCopy.end(); ++anIt) {
- GeomShapePtr aPlane = *anIt;
- GeomShapePtr aTool = GeomAlgoAPI_ShapeTools::fitPlaneToBox(aPlane, aBoundingPoints);
- std::shared_ptr<GeomAlgoAPI_MakeShapeCustom> aMkShCustom(new GeomAlgoAPI_MakeShapeCustom);
- aMkShCustom->addModified(aPlane, aTool);
- theMakeShapeList->appendAlgo(aMkShCustom);
- thePlanes.push_back(aTool);
- }
-}
-
-void unusedSubsOfComposolid(const CompsolidSubs& theObjects, CompsolidSubs& theNotUsed)
-{
- for (CompsolidSubs::const_iterator aCSIt = theObjects.begin();
- aCSIt != theObjects.end(); ++aCSIt) {
- if (aCSIt->first->shapeType() != GeomAPI_Shape::COMPSOLID)
- continue;
-
- // check the compsolid is selected
- if (aCSIt->second.size() == 1 && aCSIt->first->isEqual(aCSIt->second.front()))
- continue;
-
- // process all sub-solids of compsolid
- ListOfShape aNotUsedSolids;
- for (GeomAPI_ShapeExplorer anExp(aCSIt->first, GeomAPI_Shape::SOLID);
- anExp.more(); anExp.next()) {
- GeomShapePtr aSolidInCompSolid = anExp.current();
- ListOfShape::const_iterator anIt = aCSIt->second.begin();
- for (; anIt != aCSIt->second.end(); ++anIt)
- if (aSolidInCompSolid->isEqual(*anIt))
- break;
-
- if (anIt == aCSIt->second.end())
- aNotUsedSolids.push_back(aSolidInCompSolid);
- }
-
- if (!aNotUsedSolids.empty())
- theNotUsed.push_back(std::pair<GeomShapePtr, ListOfShape>(aCSIt->first, aNotUsedSolids));
- }
-}
+//=================================================================================================
-static bool cutSubs(const GeomShapePtr& theFirstArgument,
- CompsolidSubs& theSubsToCut,
+static bool cutSubs(ListOfShape& theSubsToCut,
const ListOfShape& theTools,
std::shared_ptr<GeomAlgoAPI_MakeShapeList>& theMakeShapeList,
std::string& theError)
{
- if (theTools.empty())
+ if (theSubsToCut.empty() || theTools.empty())
return true;
- std::shared_ptr<GeomAlgoAPI_MakeShape> aCutAlgo;
-
- for (CompsolidSubs::iterator aUIt= theSubsToCut.begin(); aUIt != theSubsToCut.end(); ++aUIt) {
- if (aUIt->first == theFirstArgument)
- continue; // no need to split unused subs of the first compsolid
-
- // cut from current list of solids
- aCutAlgo.reset(
- new GeomAlgoAPI_Boolean(aUIt->second, theTools, GeomAlgoAPI_Tools::BOOL_CUT));
- if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aCutAlgo, "", theError))
- return false;
- theMakeShapeList->appendAlgo(aCutAlgo);
-
- // update list of un-selected objects of the partition
- GeomAPI_Shape::ShapeType aType = aUIt->second.front()->shapeType();
- aUIt->second.clear();
- for (GeomAPI_ShapeExplorer anExp(aCutAlgo->shape(), aType); anExp.more(); anExp.next())
- aUIt->second.push_back(anExp.current());
- }
+ // cut from current list of solids
+ std::shared_ptr<GeomAlgoAPI_MakeShape> aCutAlgo(
+ new GeomAlgoAPI_Boolean(theSubsToCut, theTools, GeomAlgoAPI_Tools::BOOL_CUT));
+ if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aCutAlgo, "", theError))
+ return false;
+ theMakeShapeList->appendAlgo(aCutAlgo);
+
+ // update list of un-selected objects of the partition
+ GeomAPI_Shape::ShapeType aType = theSubsToCut.front()->shapeType();
+ theSubsToCut.clear();
+ for (GeomAPI_ShapeExplorer anExp(aCutAlgo->shape(), aType); anExp.more(); anExp.next())
+ theSubsToCut.push_back(anExp.current());
return true;
}
-bool cutUnusedSubs(CompsolidSubs& theObjects, CompsolidSubs& theNotUsed,
- std::shared_ptr<GeomAlgoAPI_MakeShapeList>& theMakeShapeList,
- std::string& theError)
+bool FeaturesPlugin_Partition::cutSubs(
+ GeomAPI_ShapeHierarchy& theHierarchy,
+ ListOfShape& theUsed,
+ ListOfShape& theNotUsed,
+ std::shared_ptr<GeomAlgoAPI_MakeShapeList>& theMakeShapeList,
+ std::string& theError)
{
- GeomShapePtr aFirstArgument = theObjects.front().first;
+ theUsed.clear();
+ theNotUsed.clear();
+
+ GeomAPI_ShapeHierarchy::iterator anIt = theHierarchy.begin();
// compose a set of tools for the CUT operation:
// find the list of unused subs of the first argument or use itself
- ListOfShape aToolsForUsed;
- CompsolidSubs::iterator aUIt = theNotUsed.begin();
- for (; aUIt != theNotUsed.end(); ++aUIt)
- if (aUIt->first == aFirstArgument) {
- aToolsForUsed.insert(aToolsForUsed.end(), aUIt->second.begin(), aUIt->second.end());
- break;
- }
- ListOfShape aToolsForUnused;
+ ListOfShape aToolsForUsed, aToolsForUnused;
+ GeomShapePtr aFirstArgument = theHierarchy.parent(*anIt, false);
+ if (aFirstArgument && aFirstArgument->shapeType() == GeomAPI_Shape::COMPSOLID) {
+ theHierarchy.splitCompound(aFirstArgument, theUsed, aToolsForUsed);
+ theNotUsed = aToolsForUsed;
+ }
+ else {
+ aFirstArgument = *anIt;
+ theUsed.push_back(aFirstArgument);
+ }
aToolsForUnused.push_back(aFirstArgument);
// cut subs
- return cutSubs(aFirstArgument, theObjects, aToolsForUsed, theMakeShapeList, theError)
- && cutSubs(aFirstArgument, theNotUsed, aToolsForUnused, theMakeShapeList, theError);
+ bool isOk = true;
+ for (++anIt; anIt != theHierarchy.end() && isOk; ++anIt) {
+ ListOfShape aUsed, aNotUsed;
+
+ GeomShapePtr aParent = theHierarchy.parent(*anIt, false);
+ if (aParent && aParent->shapeType() == GeomAPI_Shape::COMPSOLID) {
+ aParent = theHierarchy.parent(*anIt); // get parent once again to mark its subs as processed
+ theHierarchy.splitCompound(aParent, aUsed, aNotUsed);
+ }
+ else
+ aUsed.push_back(*anIt);
+
+ isOk = ::cutSubs(aUsed, aToolsForUsed, theMakeShapeList, theError)
+ && ::cutSubs(aNotUsed, aToolsForUnused, theMakeShapeList, theError);
+ if (isOk) {
+ theUsed.insert(theUsed.end(), aUsed.begin(), aUsed.end());
+ theNotUsed.insert(theNotUsed.end(), aNotUsed.begin(), aNotUsed.end());
+ }
+ }
+
+ return isOk;
}