]> SALOME platform Git repositories - modules/shaper.git/blob - src/BuildPlugin/BuildPlugin_Face.cpp
Salome HOME
bos #20661 EDF 22847 - Move to the end
[modules/shaper.git] / src / BuildPlugin / BuildPlugin_Face.cpp
1 // Copyright (C) 2014-2020  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
18 //
19
20 #include "BuildPlugin_Face.h"
21
22 #include <ModelAPI_AttributeSelectionList.h>
23 #include <ModelAPI_ResultBody.h>
24 #include <ModelAPI_ResultConstruction.h>
25
26 #include <GeomAPI_Edge.h>
27 #include <GeomAPI_PlanarEdges.h>
28 #include <GeomAPI_Pln.h>
29 #include <GeomAPI_ShapeExplorer.h>
30
31 #include <GeomAlgoAPI_MakeShapeList.h>
32 #include <GeomAlgoAPI_ShapeTools.h>
33 #include <GeomAlgoAPI_SketchBuilder.h>
34 #include <GeomAlgoAPI_Copy.h>
35
36 //=================================================================================================
37 BuildPlugin_Face::BuildPlugin_Face()
38 {
39 }
40
41 //=================================================================================================
42 void BuildPlugin_Face::initAttributes()
43 {
44   data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
45 }
46
47 //=================================================================================================
48 void BuildPlugin_Face::execute()
49 {
50   // Get base objects list.
51   AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
52   if(!aSelectionList.get()) {
53     setError("Error: Could not get selection list.");
54     return;
55   }
56   if(aSelectionList->size() == 0) {
57     setError("Error: Empty selection list.");
58     return;
59   }
60
61   // Collect base shapes.
62   ListOfShape anEdges;
63   ListOfShape anOriginalFaces;
64   ListOfShape aContexts;
65   getOriginalShapesAndContexts(BASE_OBJECTS_ID(), anOriginalFaces, aContexts);
66   anOriginalFaces.clear();
67   std::list< std::shared_ptr<GeomAPI_Dir> > aListOfNormals;
68   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
69     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
70     GeomShapePtr aShape = aSelection->value();
71     GeomShapePtr aContext = aSelection->context()->shape();
72     if(!aShape.get()) {
73       aShape = aContext;
74     }
75     if (aShape->shapeType() == GeomAPI_Shape::FACE) {
76       // keep selected faces "as is"
77       anOriginalFaces.push_back(aShape);
78       continue;
79     }
80     else if (!aSelection->value() && aShape->shapeType() == GeomAPI_Shape::COMPOUND) {
81       // collect faces from the sketch
82       ResultConstructionPtr aSketch =
83           std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aSelection->context());
84       if (aSketch && aSketch->facesNum() > 0) {
85         for (int i = 0; i < aSketch->facesNum(); ++i)
86           anOriginalFaces.push_back(aSketch->face(i));
87         continue;
88       }
89     }
90
91     for(GeomAPI_ShapeExplorer anExp(aShape, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
92       GeomShapePtr anEdge = anExp.current();
93       anEdges.push_back(anEdge);
94     }
95
96     // check whether the context is a sketch, in this case store its normal for further needs
97     std::shared_ptr<GeomAPI_PlanarEdges> aSketch =
98         std::dynamic_pointer_cast<GeomAPI_PlanarEdges>(aContext);
99     if (aSketch)
100       aListOfNormals.push_back(aSketch->norm());
101   }
102
103   // Build faces by edges.
104   ListOfShape aFaces;
105   GeomMakeShapePtr aFaceBuilder;
106   if (!anEdges.empty())
107     buildFacesByEdges(anEdges, aListOfNormals, aFaces, aFaceBuilder);
108   int aNbFacesFromEdges = (int)aFaces.size();
109
110   // Add faces selected by user.
111   aFaces.insert(aFaces.end(), anOriginalFaces.begin(), anOriginalFaces.end());
112
113   // Store result.
114   int anIndex = 0;
115   for(ListOfShape::const_iterator anIt = aFaces.cbegin(); anIt != aFaces.cend(); ++anIt) {
116     std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList(new GeomAlgoAPI_MakeShapeList);
117     if (anIndex < aNbFacesFromEdges)
118       aMakeShapeList->appendAlgo(aFaceBuilder);
119
120     GeomShapePtr aShape = *anIt;
121     GeomMakeShapePtr aCopy(new GeomAlgoAPI_Copy(aShape));
122     aMakeShapeList->appendAlgo(aCopy);
123
124     ListOfShape aBaseShapes;
125     if (anIndex < aNbFacesFromEdges)
126       aBaseShapes = anEdges;
127     else
128       aBaseShapes.push_back(aShape);
129     storeResult(aMakeShapeList, aBaseShapes, aContexts, aCopy->shape(), anIndex++);
130   }
131
132   removeResults(anIndex);
133 }
134
135 void BuildPlugin_Face::buildFacesByEdges(
136     const ListOfShape& theEdges,
137     const std::list< std::shared_ptr<GeomAPI_Dir> >& theNormals,
138     ListOfShape& theFaces,
139     std::shared_ptr<GeomAlgoAPI_MakeShape>& theBuilderAlgo) const
140 {
141   // Get plane.
142   std::shared_ptr<GeomAPI_Pln> aPln = GeomAlgoAPI_ShapeTools::findPlane(theEdges);
143   std::shared_ptr<GeomAPI_Dir> aNormal = aPln->direction();
144   bool isReverse = !theNormals.empty();
145   std::list< std::shared_ptr<GeomAPI_Dir> >::const_iterator aNormIt = theNormals.begin();
146   for (; aNormIt != theNormals.end() && isReverse; ++aNormIt)
147     if ((*aNormIt)->dot(aNormal) > 1.e-7)
148       isReverse = false;
149   if (isReverse) {
150     aNormal->reverse();
151     aPln = std::shared_ptr<GeomAPI_Pln>(new GeomAPI_Pln(aPln->location(), aNormal));
152   }
153
154   // Get faces.
155   std::shared_ptr<GeomAlgoAPI_SketchBuilder> aSketchBuilder(
156       new GeomAlgoAPI_SketchBuilder(aPln, theEdges));
157   theFaces = aSketchBuilder->faces();
158   theBuilderAlgo = aSketchBuilder;
159
160   // Get wires from faces.
161   ListOfShape aWires;
162   for(ListOfShape::const_iterator anIt = theFaces.cbegin(); anIt != theFaces.cend(); ++anIt)
163     aWires.push_back(GeomAlgoAPI_ShapeTools::getFaceOuterWire(*anIt));
164
165   // Make faces with holes.
166   theFaces.clear();
167   GeomAlgoAPI_ShapeTools::makeFacesWithHoles(aPln->location(), aPln->direction(),
168                                              aWires, theFaces);
169 }