1 // Copyright (C) 2014-2017 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
21 #include "FeaturesPlugin_RemoveSubShapes.h"
23 #include <ModelAPI_AttributeSelectionList.h>
24 #include <ModelAPI_AttributeString.h>
25 #include <ModelAPI_ResultBody.h>
26 #include <ModelAPI_ResultConstruction.h>
27 #include <ModelAPI_Session.h>
28 #include <ModelAPI_Tools.h>
29 #include <ModelAPI_Validator.h>
31 #include <GeomAPI_ShapeIterator.h>
32 #include <GeomAPI_ShapeExplorer.h>
34 #include <GeomAlgoAPI_Copy.h>
35 #include <GeomAlgoAPI_ShapeBuilder.h>
36 #include <GeomAlgoAPI_ShapeTools.h>
37 #include <GeomAlgoAPI_MakeShapeCustom.h>
40 //==================================================================================================
41 FeaturesPlugin_RemoveSubShapes::FeaturesPlugin_RemoveSubShapes()
42 : myChangedInCode(false)
46 //==================================================================================================
47 void FeaturesPlugin_RemoveSubShapes::initAttributes()
49 data()->addAttribute(BASE_SHAPE_ID(), ModelAPI_AttributeSelection::typeId());
51 data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId());
53 data()->addAttribute(SUBSHAPES_TO_KEEP_ID(), ModelAPI_AttributeSelectionList::typeId());
55 data()->addAttribute(SUBSHAPES_TO_REMOVE_ID(), ModelAPI_AttributeSelectionList::typeId());
58 void FeaturesPlugin_RemoveSubShapes::attributeChanged(const std::string& theID)
60 ModelAPI_Feature::attributeChanged(theID);
62 if (myChangedInCode) return;
64 AttributeSelectionPtr aShapeAttrSelection = selection(BASE_SHAPE_ID());
65 AttributeSelectionListPtr aSubShapesToKeepAttrList = selectionList(SUBSHAPES_TO_KEEP_ID());
66 AttributeSelectionListPtr aSubShapesToRemoveAttrList = selectionList(SUBSHAPES_TO_REMOVE_ID());
67 if (!aShapeAttrSelection.get()
68 || !aSubShapesToKeepAttrList.get()
69 || !aSubShapesToRemoveAttrList.get())
74 ResultPtr aContext = aShapeAttrSelection->context();
75 ResultBodyPtr aResultBody =
76 std::dynamic_pointer_cast<ModelAPI_ResultBody>(aContext);
77 if (!aResultBody.get()) {
78 aSubShapesToKeepAttrList->clear();
79 aSubShapesToRemoveAttrList->clear();
82 const bool isHasSubs = ModelAPI_Tools::hasSubResults(aResultBody);
84 GeomShapePtr aBaseShape = aShapeAttrSelection->value();
85 if(!aBaseShape.get()) {
86 aBaseShape = aContext->shape();
89 myChangedInCode = true;
91 if(theID == BASE_SHAPE_ID()) {
92 aSubShapesToKeepAttrList->clear();
93 aSubShapesToRemoveAttrList->clear();
95 if (!aBaseShape.get()) {
99 std::list<GeomShapePtr> aSubShapes = GeomAlgoAPI_ShapeTools::getLowLevelSubShapes(aBaseShape);
100 for (ListOfShape::const_iterator anIt = aSubShapes.cbegin(); anIt != aSubShapes.cend(); ++anIt) {
101 GeomShapePtr aSubShape = *anIt;
103 aSubShapesToKeepAttrList->append(aContext, aSubShape);
105 std::list<ResultPtr> anAllSubs;
106 ModelAPI_Tools::allSubs(aResultBody, anAllSubs);
107 std::list<ResultPtr>::iterator aSubsIt = anAllSubs.begin();
108 for(; aSubsIt != anAllSubs.end(); aSubsIt++) {
109 ResultBodyPtr aSub = std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aSubsIt);
110 if (aSub && aSub->shape().get() && aSub->shape()->isSubShape(aSubShape)) {
111 aSubShapesToKeepAttrList->append(aSub, aSubShape);
118 else if (theID == SUBSHAPES_TO_KEEP_ID())
120 aSubShapesToRemoveAttrList->clear();
122 if (!aBaseShape.get()) {
127 const int aSubsToKeepNb = aSubShapesToKeepAttrList->size();
128 std::list<GeomShapePtr> aSubShapes = GeomAlgoAPI_ShapeTools::getLowLevelSubShapes(aBaseShape);
129 for (ListOfShape::const_iterator anIt = aSubShapes.cbegin(); anIt != aSubShapes.cend(); ++anIt) {
130 GeomShapePtr aSubShape = *anIt;
131 for(anIndex = 0; anIndex < aSubsToKeepNb; ++anIndex) {
132 AttributeSelectionPtr anAttrSelectionInList = aSubShapesToKeepAttrList->value(anIndex);
133 GeomShapePtr aSubShapeToKeep = anAttrSelectionInList->value();
134 if (aSubShapeToKeep.get() && aSubShapeToKeep->isEqual(aSubShape)) {
139 if (anIndex == aSubsToKeepNb) {
141 aSubShapesToRemoveAttrList->append(aContext, aSubShape);
143 std::list<ResultPtr> anAllSubs;
144 ModelAPI_Tools::allSubs(aResultBody, anAllSubs);
145 std::list<ResultPtr>::iterator aSubsIt = anAllSubs.begin();
146 for(; aSubsIt != anAllSubs.end(); aSubsIt++) {
147 ResultBodyPtr aSub = std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aSubsIt);
148 if (aSub && aSub->shape().get() && aSub->shape()->isSubShape(aSubShape)) {
149 aSubShapesToRemoveAttrList->append(aSub, aSubShape);
157 else if (theID == SUBSHAPES_TO_REMOVE_ID())
159 aSubShapesToKeepAttrList->clear();
161 if (!aBaseShape.get()) {
166 const int aSubsToRemoveNb = aSubShapesToRemoveAttrList->size();
167 std::list<GeomShapePtr> aSubShapes = GeomAlgoAPI_ShapeTools::getLowLevelSubShapes(aBaseShape);
168 for (ListOfShape::const_iterator anIt = aSubShapes.cbegin(); anIt != aSubShapes.cend(); ++anIt) {
169 GeomShapePtr aSubShape = *anIt;
170 for(anIndex = 0; anIndex < aSubsToRemoveNb; ++anIndex) {
171 AttributeSelectionPtr anAttrSelectionInList = aSubShapesToRemoveAttrList->value(anIndex);
172 GeomShapePtr aSubShapeToRemove = anAttrSelectionInList->value();
173 if (aSubShapeToRemove.get() && aSubShapeToRemove->isEqual(aSubShape)) {
178 if (anIndex == aSubsToRemoveNb) {
180 aSubShapesToKeepAttrList->append(aContext, aSubShape);
182 std::list<ResultPtr> anAllSubs;
183 ModelAPI_Tools::allSubs(aResultBody, anAllSubs);
184 std::list<ResultPtr>::iterator aSubsIt = anAllSubs.begin();
185 for(; aSubsIt != anAllSubs.end(); aSubsIt++) {
186 ResultBodyPtr aSub = std::dynamic_pointer_cast<ModelAPI_ResultBody>(*aSubsIt);
187 if (aSub && aSub->shape().get() && aSub->shape()->isSubShape(aSubShape)) {
188 aSubShapesToKeepAttrList->append(aSub, aSubShape);
197 myChangedInCode = false;
200 //==================================================================================================
201 void FeaturesPlugin_RemoveSubShapes::execute()
203 // Get base shape and sub-shapes list.
204 AttributeSelectionPtr aShapeAttrSelection = selection(BASE_SHAPE_ID());
205 AttributeSelectionListPtr aSubShapesAttrList = selectionList(SUBSHAPES_TO_KEEP_ID());
206 if(!aShapeAttrSelection.get() || !aSubShapesAttrList.get()) {
211 GeomShapePtr aBaseShape = aShapeAttrSelection->value();
213 GeomShapePtr aResultShape;
214 int aSubsNb = aSubShapesAttrList->size();
216 if(!aBaseShape.get()) {
219 aResultShape = aBaseShape->emptyCopied();
221 // Copy sub-shapes from list to new shape.
222 for(int anIndex = 0; anIndex < aSubsNb; ++anIndex) {
223 AttributeSelectionPtr anAttrSelectionInList = aSubShapesAttrList->value(anIndex);
224 GeomShapePtr aShapeToAdd = anAttrSelectionInList->value();
225 GeomAlgoAPI_ShapeBuilder::add(aResultShape, aShapeToAdd);
227 } else if(aSubsNb == 1) {
228 AttributeSelectionPtr anAttrSelectionInList = aSubShapesAttrList->value(0);
229 aResultShape = anAttrSelectionInList->value();
231 // deleted and copied must be jointed to one list which keeps all the history
232 GeomAlgoAPI_MakeShapeList aMakeShapeList;
234 // find all removed shapes
235 std::shared_ptr<GeomAlgoAPI_MakeShapeCustom> aDeletedSubs(new GeomAlgoAPI_MakeShapeCustom);
236 std::set<GeomAPI_Shape::ShapeType> aTypes; // types that where removed
237 aTypes.insert(GeomAPI_Shape::FACE);
239 std::list<GeomShapePtr> aSubShapes = GeomAlgoAPI_ShapeTools::getLowLevelSubShapes(aBaseShape);
240 for (ListOfShape::const_iterator anIt = aSubShapes.cbegin(); anIt != aSubShapes.cend(); ++anIt) {
241 GeomShapePtr aSubShape = *anIt;
242 if (!aSubShape.get() || aSubShape->isNull())
246 for(anIndex = 0; anIndex < aSubsNb; ++anIndex) {
247 AttributeSelectionPtr anAttrSelectionInList = aSubShapesAttrList->value(anIndex);
248 GeomShapePtr aLeftShape = anAttrSelectionInList->value();
249 if (aSubShape->isEqual(aLeftShape)) {
250 break; // found in a left-list
253 if (anIndex == aSubsNb) { // not found in left
254 aDeletedSubs->addDeleted(aSubShape);
255 aTypes.insert(aSubShape->shapeType());
256 if (aSubShape->shapeType() != GeomAPI_Shape::FACE) {
257 GeomAPI_ShapeExplorer aFaces(aSubShape, GeomAPI_Shape::FACE);
258 for(; aFaces.more(); aFaces.next())
259 aDeletedSubs->addDeleted(aFaces.current());
263 aMakeShapeList.appendAlgo(aDeletedSubs);
265 std::shared_ptr<GeomAlgoAPI_Copy> aCopy(new GeomAlgoAPI_Copy(aResultShape));
266 aResultShape = aCopy->shape();
267 aMakeShapeList.appendAlgo(aCopy);
270 ResultBodyPtr aResultBody = document()->createBody(data());
271 aResultBody->storeModified(aBaseShape, aResultShape, 1);
272 std::set<GeomAPI_Shape::ShapeType>::iterator aTypeIter = aTypes.begin();
273 for(; aTypeIter != aTypes.end(); aTypeIter++)
274 aResultBody->loadDeletedShapes(&aMakeShapeList, aBaseShape, *aTypeIter, 1);
275 aResultBody->loadAndOrientModifiedShapes(&aMakeShapeList, aBaseShape, GeomAPI_Shape::FACE,
276 2, "Modified_Face", *aMakeShapeList.mapOfSubShapes().get(), true, false, true);
277 aResultBody->loadAndOrientModifiedShapes(&aMakeShapeList, aBaseShape, GeomAPI_Shape::EDGE,
278 3, "Modified_Edge", *aMakeShapeList.mapOfSubShapes().get(), false, false, true);
279 aResultBody->loadAndOrientModifiedShapes(&aMakeShapeList, aBaseShape, GeomAPI_Shape::VERTEX,
280 4, "Modified_Vertex", *aMakeShapeList.mapOfSubShapes().get());
281 setResult(aResultBody);