1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: BuildPlugin_Wire.cpp
4 // Created: 14 April 2016
5 // Author: Dmitry Bobylev
7 #include "BuildPlugin_Wire.h"
9 #include <ModelAPI_AttributeSelectionList.h>
10 #include <ModelAPI_ResultBody.h>
11 #include <ModelAPI_ResultConstruction.h>
13 #include <Events_InfoMessage.h>
15 #include <GeomAPI_PlanarEdges.h>
16 #include <GeomAPI_ShapeExplorer.h>
18 #include <GeomAlgoAPI_ShapeTools.h>
19 #include <GeomAlgoAPI_WireBuilder.h>
23 //=================================================================================================
24 BuildPlugin_Wire::BuildPlugin_Wire()
28 //=================================================================================================
29 void BuildPlugin_Wire::initAttributes()
31 data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
34 //=================================================================================================
35 void BuildPlugin_Wire::execute()
37 // Get base objects list.
38 AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
39 if(!aSelectionList.get()) {
40 setError("Error: Could not get selection list.");
43 if(aSelectionList->size() == 0) {
44 setError("Error: Empty selection list.");
48 // Collect base shapes.
50 for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
51 AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
52 GeomShapePtr aShape = aSelection->value();
54 aShape = aSelection->context()->shape();
56 for(GeomAPI_ShapeExplorer anExp(aShape, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
57 GeomShapePtr anEdge = anExp.current();
58 anEdges.push_back(anEdge);
63 GeomShapePtr aWire = GeomAlgoAPI_WireBuilder::wire(anEdges);
65 setError("Error: Result wire is empty. Probably it has disconnected edges or non-manifold.");
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");
82 setResult(aResultBody);
85 //=================================================================================================
86 bool BuildPlugin_Wire::customAction(const std::string& theActionId)
88 if(theActionId == ADD_CONTOUR_ACTION_ID()) {
91 std::string aMsg = "Error: Feature \"%1\" does not support action \"%2\".";
92 Events_InfoMessage("BuildPlugin_Wire", aMsg).arg(getKind()).arg(theActionId).send();
98 //=================================================================================================
99 bool BuildPlugin_Wire::addContour()
101 // Get base objects list.
102 AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
103 if(aSelectionList->size() == 0) {
104 Events_InfoMessage("BuildPlugin_Wire", "Error: Empty selection list.").send();
108 // Collect attributes to check.
109 ListOfShape anAddedEdges;
110 std::list<AttributeSelectionPtr> anAttributesToCheck;
111 for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
112 AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
113 GeomShapePtr anEdgeInList = aSelection->value();
114 if(!anEdgeInList.get()) {
118 // Check that it is edge.
119 if(anEdgeInList->shapeType() != GeomAPI_Shape::EDGE) {
123 // Check that it is edge on sketch.
124 ResultPtr aContext = aSelection->context();
125 ResultConstructionPtr aConstruction = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aContext);
126 if(!aConstruction.get()) {
129 GeomShapePtr aContextShape = aConstruction->shape();
130 std::shared_ptr<GeomAPI_PlanarEdges> aPlanarEdges = std::dynamic_pointer_cast<GeomAPI_PlanarEdges>(aContextShape);
131 if(!aPlanarEdges.get()) {
135 // Check that sketch have faces.
136 if(aConstruction->facesNum() == 0) {
140 anAddedEdges.push_back(anEdgeInList);
141 anAttributesToCheck.push_back(aSelection);
144 // Check if edges have contours.
145 bool isAnyContourAdded = false;
146 for(std::list<AttributeSelectionPtr>::const_iterator aListIt = anAttributesToCheck.cbegin();
147 aListIt != anAttributesToCheck.cend();
149 AttributeSelectionPtr aSelection = *aListIt;
150 std::shared_ptr<GeomAPI_Edge> anEdgeInList(new GeomAPI_Edge(aSelection->value()));
152 ResultConstructionPtr aConstruction = std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aSelection->context());
154 // Iterate on wires and add wire with this edge.
155 std::shared_ptr<GeomAPI_Shape> aFoundWire;
156 for(int anIndex = 0; anIndex < aConstruction->facesNum(); ++anIndex) {
157 std::shared_ptr<GeomAPI_Face> aFace = aConstruction->face(anIndex);
158 for(GeomAPI_ShapeExplorer aWireExp(aFace, GeomAPI_Shape::WIRE); aWireExp.more(); aWireExp.next()) {
159 GeomShapePtr aWireOnFace = aWireExp.current();
160 for(GeomAPI_ShapeExplorer anExp(aWireOnFace, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
161 std::shared_ptr<GeomAPI_Edge> anEdgeOnFace(new GeomAPI_Edge(anExp.current()));
162 if(anEdgeInList->isEqual(anEdgeOnFace)) {
163 aFoundWire = aWireOnFace;
168 if(aFoundWire.get()) {
173 // If wire with the same edge found. Add all other edges to list.
174 if(aFoundWire.get()) {
175 for(GeomAPI_ShapeExplorer anExp(aFoundWire, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
176 std::shared_ptr<GeomAPI_Edge> anEdgeOnFace(new GeomAPI_Edge(anExp.current()));
177 ListOfShape::const_iterator anEdgesIt = anAddedEdges.cbegin();
178 for(; anEdgesIt != anAddedEdges.cend(); ++anEdgesIt) {
179 if(anEdgeOnFace->isEqual(*anEdgesIt)) {
183 if(anEdgesIt == anAddedEdges.cend()) {
184 isAnyContourAdded = true;
185 anAddedEdges.push_back(anEdgeOnFace);
186 aSelectionList->append(aConstruction, anEdgeOnFace);
192 if(!isAnyContourAdded) {
193 Events_InfoMessage("BuildPlugin_Wire",
194 "Error: Contours already closed or no contours found for selected edges.").send();