]> SALOME platform Git repositories - modules/shaper.git/blob - src/BuildPlugin/BuildPlugin_Shell.cpp
Salome HOME
Fix for the issue #2401
[modules/shaper.git] / src / BuildPlugin / BuildPlugin_Shell.cpp
1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
2 //
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.
7 //
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.
12 //
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
16 //
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "BuildPlugin_Shell.h"
22
23 #include <ModelAPI_AttributeSelectionList.h>
24 #include <ModelAPI_ResultBody.h>
25 #include <ModelAPI_ResultConstruction.h>
26
27 #include <GeomAlgoAPI_Sewing.h>
28 #include <GeomAPI_ShapeExplorer.h>
29
30 //=================================================================================================
31 BuildPlugin_Shell::BuildPlugin_Shell()
32 {
33 }
34
35 //=================================================================================================
36 void BuildPlugin_Shell::initAttributes()
37 {
38   data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
39 }
40
41 //=================================================================================================
42 void BuildPlugin_Shell::execute()
43 {
44   // Get base objects list.
45   AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
46
47   // Collect base shapes.
48   ListOfShape aShapes;
49   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
50     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
51     GeomShapePtr aShape = aSelection->value();
52     if(!aShape.get()) {
53       aShape = aSelection->context()->shape();
54     }
55     aShapes.push_back(aShape);
56   }
57
58   // Sew faces.
59   GeomAlgoAPI_Sewing aSewingAlgo(aShapes);
60
61   // Check that algo is done.
62   if(!aSewingAlgo.isDone()) {
63     setError("Error: " + getKind() + " algorithm failed.");
64     return;
65   }
66
67   // Check if shape is not null.
68   if(!aSewingAlgo.shape().get() || aSewingAlgo.shape()->isNull()) {
69     setError("Error: Resulting shape is null.");
70     return;
71   }
72
73   // Check that resulting shape is valid.
74   if(!aSewingAlgo.isValid()) {
75     setError("Error: Resulting shape is not valid.");
76     return;
77   }
78
79   // Store result.
80   GeomShapePtr aResult = aSewingAlgo.shape();
81   std::shared_ptr<GeomAPI_DataMapOfShapeShape> aMapOfShapes = aSewingAlgo.mapOfSubShapes();
82
83   int anIndex = 0;
84   for(GeomAPI_ShapeExplorer anExp(aResult, GeomAPI_Shape::SHELL); anExp.more(); anExp.next()) {
85     GeomShapePtr aShell = anExp.current();
86     ResultBodyPtr aResultBody = document()->createBody(data(), anIndex);
87     aResultBody->store(aShell);
88     for(ListOfShape::const_iterator anIt = aShapes.cbegin(); anIt != aShapes.cend(); ++anIt) {
89       for(GeomAPI_ShapeExplorer
90           aFaceExp(*anIt, GeomAPI_Shape::FACE); aFaceExp.more(); aFaceExp.next()) {
91         GeomShapePtr aFace = aFaceExp.current();
92         ListOfShape aHistory;
93         aSewingAlgo.modified(aFace, aHistory);
94         for(ListOfShape::const_iterator aHistoryIt = aHistory.cbegin();
95             aHistoryIt != aHistory.cend();
96             ++aHistoryIt) {
97           GeomShapePtr aHistoryShape = *aHistoryIt;
98           if(aMapOfShapes->isBound(aHistoryShape)) {
99             aHistoryShape = aMapOfShapes->find(aHistoryShape);
100           }
101           if(aShell->isSubShape(aHistoryShape)) {
102             aResultBody->loadAndOrientModifiedShapes(&aSewingAlgo, aFace, GeomAPI_Shape::EDGE,
103                                                      1, "Modified_Edge", *aMapOfShapes.get());
104             aResultBody->loadAndOrientModifiedShapes(&aSewingAlgo, aFace, GeomAPI_Shape::FACE,
105                                                      2, "Modified_Face", *aMapOfShapes.get());
106             break;
107           }
108         }
109       }
110     }
111     setResult(aResultBody, anIndex);
112     ++anIndex;
113   }
114
115   removeResults(anIndex);
116 }