]> SALOME platform Git repositories - modules/shaper.git/blob - src/BuildPlugin/BuildPlugin_Wire.cpp
Salome HOME
14d5158945d7b90c37d8caf0ad884013228bad24
[modules/shaper.git] / src / BuildPlugin / BuildPlugin_Wire.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:        BuildPlugin_Wire.cpp
4 // Created:     14 April 2016
5 // Author:      Dmitry Bobylev
6
7 #include "BuildPlugin_Wire.h"
8
9 #include <ModelAPI_AttributeSelectionList.h>
10 #include <ModelAPI_ResultBody.h>
11 #include <ModelAPI_ResultConstruction.h>
12
13 #include <Events_Error.h>
14
15 #include <GeomAPI_PlanarEdges.h>
16 #include <GeomAPI_ShapeExplorer.h>
17
18 #include <GeomAlgoAPI_ShapeTools.h>
19 #include <GeomAlgoAPI_WireBuilder.h>
20
21 #include <algorithm>
22
23 //=================================================================================================
24 BuildPlugin_Wire::BuildPlugin_Wire()
25 {
26 }
27
28 //=================================================================================================
29 void BuildPlugin_Wire::initAttributes()
30 {
31   data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
32 }
33
34 //=================================================================================================
35 void BuildPlugin_Wire::execute()
36 {
37   // Get base objects list.
38   AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
39   if(!aSelectionList.get()) {
40     setError("Error: Could not get selection list.");
41     return;
42   }
43   if(aSelectionList->size() == 0) {
44     setError("Error: Empty selection list.");
45     return;
46   }
47
48   // Collect base shapes.
49   ListOfShape anEdges;
50   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
51     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
52     GeomShapePtr aShape = aSelection->value();
53     if(!aShape.get()) {
54       aShape = aSelection->context()->shape();
55     }
56     for(GeomAPI_ShapeExplorer anExp(aShape, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
57       GeomShapePtr anEdge = anExp.current();
58       anEdges.push_back(anEdge);
59     }
60   }
61
62   // Create wire.
63   GeomShapePtr aWire = GeomAlgoAPI_WireBuilder::wire(anEdges);
64   if(!aWire.get()) {
65     setError("Error: Result wire is empty. Probably it has disconnected edges or non-manifold.");
66     return;
67   }
68
69   // Store result.
70   ResultBodyPtr aResultBody = document()->createBody(data());
71   aResultBody->store(aWire);
72   for(GeomAPI_ShapeExplorer anExp(aWire, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
73     GeomShapePtr anEdgeInResult = anExp.current();
74     for(ListOfShape::const_iterator anIt = anEdges.cbegin(); anIt != anEdges.cend(); ++anIt) {
75       std::shared_ptr<GeomAPI_Edge> anEdgeInList(new GeomAPI_Edge(*anIt));
76       if(anEdgeInList->isEqual(anEdgeInResult)) {
77         aResultBody->modified(anEdgeInList, anEdgeInResult, "Edge");
78         break;
79       }
80     }
81   }
82   setResult(aResultBody);
83 }
84
85 //=================================================================================================
86 bool BuildPlugin_Wire::customAction(const std::string& theActionId)
87 {
88   if(theActionId == "add_contour") {
89     return addContour();
90   } else {
91     Events_Error::send("Error: Feature \"" + getKind() + "\" does not support action \"" + theActionId + "\".");
92   }
93
94   return false;
95 }
96
97 //=================================================================================================
98 bool BuildPlugin_Wire::addContour()
99 {
100   // Get base objects list.
101   AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
102   if(aSelectionList->size() == 0) {
103     Events_Error::send("Error: Empty selection list.");
104     return false;
105   }
106
107   // Collect attributes to check.
108   ListOfShape anAddedEdges;
109   std::list<AttributeSelectionPtr> anAttributesToCheck;
110   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
111     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
112     GeomShapePtr anEdgeInList = aSelection->value();
113     if(!anEdgeInList.get()) {
114       continue;
115     }
116
117     // Check that it is edge.
118     if(anEdgeInList->shapeType() != GeomAPI_Shape::EDGE) {
119       continue;
120     }
121
122     // Check that it is edge on sketch.
123     ResultPtr aContext = aSelection->context();
124     ResultConstructionPtr aConstruction = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aContext);
125     if(!aConstruction.get()) {
126       continue;
127     }
128     GeomShapePtr aContextShape = aConstruction->shape();
129     std::shared_ptr<GeomAPI_PlanarEdges> aPlanarEdges = std::dynamic_pointer_cast<GeomAPI_PlanarEdges>(aContextShape);
130     if(!aPlanarEdges.get()) {
131       continue;
132     }
133
134     // Check that sketch have faces.
135     if(aConstruction->facesNum() == 0) {
136       continue;
137     }
138
139     anAddedEdges.push_back(anEdgeInList);
140     anAttributesToCheck.push_back(aSelection);
141   }
142
143   // Check if edges have contours.
144   bool isAnyContourAdded = false;
145   for(std::list<AttributeSelectionPtr>::const_iterator aListIt = anAttributesToCheck.cbegin();
146       aListIt != anAttributesToCheck.cend();
147       ++aListIt) {
148     AttributeSelectionPtr aSelection = *aListIt;
149     std::shared_ptr<GeomAPI_Edge> anEdgeInList(new GeomAPI_Edge(aSelection->value()));
150
151     ResultConstructionPtr aConstruction = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aSelection->context());
152
153     // Iterate on wires and add wire with this edge.
154     std::shared_ptr<GeomAPI_Shape> aFoundWire;
155     for(int anIndex = 0; anIndex < aConstruction->facesNum(); ++anIndex) {
156       std::shared_ptr<GeomAPI_Face> aFace = aConstruction->face(anIndex);
157       for(GeomAPI_ShapeExplorer aWireExp(aFace, GeomAPI_Shape::WIRE); aWireExp.more(); aWireExp.next()) {
158         GeomShapePtr aWireOnFace = aWireExp.current();
159         for(GeomAPI_ShapeExplorer anExp(aWireOnFace, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
160           std::shared_ptr<GeomAPI_Edge> anEdgeOnFace(new GeomAPI_Edge(anExp.current()));
161           if(anEdgeInList->isEqual(anEdgeOnFace)) {
162             aFoundWire = aWireOnFace;
163             break;
164           }
165         }
166       }
167       if(aFoundWire.get()) {
168         break;
169       }
170     }
171
172     // If wire with the same edge found. Add all other edges to list.
173     if(aFoundWire.get()) {
174       for(GeomAPI_ShapeExplorer anExp(aFoundWire, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
175         std::shared_ptr<GeomAPI_Edge> anEdgeOnFace(new GeomAPI_Edge(anExp.current()));
176         ListOfShape::const_iterator anEdgesIt = anAddedEdges.cbegin();
177         for(; anEdgesIt != anAddedEdges.cend(); ++anEdgesIt) {
178           if(anEdgeOnFace->isEqual(*anEdgesIt)) {
179             break;
180           }
181         }
182         if(anEdgesIt == anAddedEdges.cend()) {
183           isAnyContourAdded = true;
184           anAddedEdges.push_back(anEdgeOnFace);
185           aSelectionList->append(aConstruction, anEdgeOnFace);
186         }
187       }
188     }
189   }
190
191   if(!isAnyContourAdded) {
192     Events_Error::send("Error: Contours already closed or no contours found for selected edges.");
193     return false;
194   }
195
196   return true;
197 }