1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 // File: FeaturesPlugin_Partition.cpp
4 // Created: 31 Jul 2015
5 // Author: Natalia ERMOLAEVA
7 #include "FeaturesPlugin_Partition.h"
9 #include <ModelAPI_Data.h>
10 #include <ModelAPI_Document.h>
11 #include <ModelAPI_AttributeBoolean.h>
12 #include <ModelAPI_AttributeReference.h>
13 #include <ModelAPI_AttributeInteger.h>
14 #include <ModelAPI_BodyBuilder.h>
15 #include <ModelAPI_ResultBody.h>
16 #include <ModelAPI_AttributeSelectionList.h>
17 #include <ModelAPI_Session.h>
18 #include <ModelAPI_Validator.h>
20 #include <GeomAlgoAPI_CompoundBuilder.h>
21 #include <GeomAlgoAPI_Partition.h>
22 #include <GeomAlgoAPI_MakeShapeCustom.h>
23 #include <GeomAlgoAPI_MakeShapeList.h>
24 #include <GeomAlgoAPI_ShapeTools.h>
28 //=================================================================================================
29 FeaturesPlugin_Partition::FeaturesPlugin_Partition()
33 //=================================================================================================
34 void FeaturesPlugin_Partition::initAttributes()
36 AttributeSelectionListPtr aSelection =
37 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(data()->addAttribute(
38 FeaturesPlugin_Partition::OBJECT_LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
39 aSelection->setSelectionType("SOLID");
42 std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(data()->addAttribute(
43 FeaturesPlugin_Partition::TOOL_LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
44 aSelection->setSelectionType("SOLID");
46 ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), TOOL_LIST_ID());
48 data()->addAttribute(COMBINE_ID(), ModelAPI_AttributeBoolean::typeId());
51 //=================================================================================================
52 std::shared_ptr<GeomAPI_Shape> FeaturesPlugin_Partition::getShape(const std::string& theAttrName)
54 std::shared_ptr<ModelAPI_AttributeReference> aObjRef =
55 std::dynamic_pointer_cast<ModelAPI_AttributeReference>(data()->attribute(theAttrName));
57 std::shared_ptr<ModelAPI_ResultBody> aConstr =
58 std::dynamic_pointer_cast<ModelAPI_ResultBody>(aObjRef->value());
60 return aConstr->shape();
62 return std::shared_ptr<GeomAPI_Shape>();
65 //=================================================================================================
66 void FeaturesPlugin_Partition::execute()
68 ListOfShape anObjects, aTools, aToolsForNaming;
71 AttributeSelectionListPtr anObjectsSelList = selectionList(FeaturesPlugin_Partition::OBJECT_LIST_ID());
72 for (int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) {
73 std::shared_ptr<ModelAPI_AttributeSelection> anObjectAttr = anObjectsSelList->value(anObjectsIndex);
74 std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
75 if (!anObject.get()) {
78 anObjects.push_back(anObject);
81 GeomAlgoAPI_MakeShapeList aMakeShapeList;
82 std::list<std::shared_ptr<GeomAPI_Pnt> > aBoundingPoints = GeomAlgoAPI_ShapeTools::getBoundingBox(anObjects, 1.0);
85 AttributeSelectionListPtr aToolsSelList = selectionList(FeaturesPlugin_Partition::TOOL_LIST_ID());
86 for (int aToolsIndex = 0; aToolsIndex < aToolsSelList->size(); aToolsIndex++) {
87 std::shared_ptr<ModelAPI_AttributeSelection> aToolAttr = aToolsSelList->value(aToolsIndex);
88 std::shared_ptr<GeomAPI_Shape> aTool = aToolAttr->value();
90 // it could be a construction plane
91 ResultPtr aContext = aToolAttr->context();
93 aTool = GeomAlgoAPI_ShapeTools::fitPlaneToBox(aContext->shape(), aBoundingPoints);
94 std::shared_ptr<GeomAlgoAPI_MakeShapeCustom> aMkShCustom(new GeomAlgoAPI_MakeShapeCustom);
95 aMkShCustom->addModified(aContext->shape(), aTool);
96 aMakeShapeList.append(aMkShCustom);
97 aTools.push_back(aTool);
98 aToolsForNaming.push_back(aContext->shape());
101 aTools.push_back(aTool);
102 aToolsForNaming.push_back(aTool);
106 // Getting combine flag.
107 bool isCombine = boolean(COMBINE_ID())->value();
109 if(anObjects.empty()/* || aTools.empty()*/) {
110 std::string aFeatureError = "Not enough objects for partition operation";
111 setError(aFeatureError);
115 int aResultIndex = 0;
118 // Create single result.
119 if(!aTools.empty()) {
120 // This is a workaround for naming. Passing compound of objects as argument instead each object separately.
121 std::shared_ptr<GeomAPI_Shape> aCompoud = GeomAlgoAPI_CompoundBuilder::compound(anObjects);
123 anObjects.push_back(aCompoud);
125 GeomAlgoAPI_Partition aPartitionAlgo(anObjects, aTools);
127 // Checking that the algorithm worked properly.
128 if (!aPartitionAlgo.isDone()) {
129 static const std::string aFeatureError = "Partition algorithm failed";
130 setError(aFeatureError);
133 if (aPartitionAlgo.shape()->isNull()) {
134 static const std::string aShapeError = "Resulting shape is Null";
135 setError(aShapeError);
138 if (!aPartitionAlgo.isValid()) {
139 std::string aFeatureError = "Warning: resulting shape is not valid";
140 setError(aFeatureError);
144 if (GeomAlgoAPI_ShapeTools::volume(aPartitionAlgo.shape()) > 1.e-7) {
145 std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data(), aResultIndex);
146 aMakeShapeList.append(aPartitionAlgo.makeShape());
147 GeomAPI_DataMapOfShapeShape aMapOfShapes = *aPartitionAlgo.mapOfShapes().get();
148 loadNamingDS(aResultBody, anObjects.front(), aToolsForNaming, aPartitionAlgo.shape(), aMakeShapeList, aMapOfShapes);
149 setResult(aResultBody, aResultIndex);
153 // Create result for each object.
154 for (ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end(); anObjectsIt++) {
155 std::shared_ptr<GeomAPI_Shape> anObject = *anObjectsIt;
156 ListOfShape aListWithObject; aListWithObject.push_back(anObject);
157 GeomAlgoAPI_Partition aPartitionAlgo(aListWithObject, aTools);
159 // Checking that the algorithm worked properly.
160 if (!aPartitionAlgo.isDone()) {
161 static const std::string aFeatureError = "Partition algorithm failed";
162 setError(aFeatureError);
165 if (aPartitionAlgo.shape()->isNull()) {
166 static const std::string aShapeError = "Resulting shape is Null";
167 setError(aShapeError);
170 if (!aPartitionAlgo.isValid()) {
171 std::string aFeatureError = "Warning: resulting shape is not valid";
172 setError(aFeatureError);
176 if (GeomAlgoAPI_ShapeTools::volume(aPartitionAlgo.shape()) > 1.e-7) {
177 std::shared_ptr<ModelAPI_ResultBody> aResultBody = document()->createBody(data(), aResultIndex);
178 aMakeShapeList.append(aPartitionAlgo.makeShape());
179 GeomAPI_DataMapOfShapeShape aMapOfShapes = *aPartitionAlgo.mapOfShapes().get();
180 loadNamingDS(aResultBody, anObject, aToolsForNaming, aPartitionAlgo.shape(), aMakeShapeList, aMapOfShapes);
181 setResult(aResultBody, aResultIndex);
187 // remove the rest results if there were produced in the previous pass
188 removeResults(aResultIndex);
191 //=================================================================================================
192 void FeaturesPlugin_Partition::loadNamingDS(std::shared_ptr<ModelAPI_ResultBody> theResultBody,
193 const std::shared_ptr<GeomAPI_Shape> theBaseShape,
194 const ListOfShape& theTools,
195 const std::shared_ptr<GeomAPI_Shape> theResultShape,
196 GeomAlgoAPI_MakeShape& theMakeShape,
197 GeomAPI_DataMapOfShapeShape& theMapOfShapes)
200 if(theBaseShape->isEqual(theResultShape)) {
201 theResultBody->store(theResultShape);
203 const int aDeletedTag = 1;
204 const int aSubsolidsTag = 2; /// sub solids will be placed at labels 3, 4, etc. if result is compound of solids
205 const int aModifyTag = 100000;
206 int aModifyToolsTag = 200000;
207 std::ostringstream aStream;
209 theResultBody->storeModified(theBaseShape, theResultShape, aSubsolidsTag);
211 std::string aModName = "Modified";
212 theResultBody->loadAndOrientModifiedShapes(&theMakeShape, theBaseShape, GeomAPI_Shape::FACE,
213 aModifyTag, aModName, theMapOfShapes, true);
214 theResultBody->loadDeletedShapes(&theMakeShape, theBaseShape, GeomAPI_Shape::FACE, aDeletedTag);
217 for(ListOfShape::const_iterator anIter = theTools.begin(); anIter != theTools.end(); anIter++) {
218 aStream.str(std::string());
220 aStream << aModName << "_" << anIndex++;
221 theResultBody->loadAndOrientModifiedShapes(&theMakeShape, *anIter, GeomAPI_Shape::FACE,
222 aModifyToolsTag, aStream.str(), theMapOfShapes, true);
223 theResultBody->loadDeletedShapes(&theMakeShape, *anIter, GeomAPI_Shape::FACE, aDeletedTag);
224 aModifyToolsTag += 10000;