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