Salome HOME
Updated modified shapes storing.
[modules/shaper.git] / src / BuildPlugin / BuildPlugin_Polyline.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_Polyline.h"
22
23 #include <ModelAPI_AttributeBoolean.h>
24 #include <ModelAPI_AttributeSelectionList.h>
25 #include <ModelAPI_ResultBody.h>
26 #include <ModelAPI_ResultConstruction.h>
27
28 #include <Events_InfoMessage.h>
29
30 #include <GeomAPI_ShapeExplorer.h>
31
32 #include <GeomAlgoAPI_ShapeTools.h>
33 #include <GeomAlgoAPI_EdgeBuilder.h>
34 #include <GeomAlgoAPI_PointBuilder.h>
35 #include <GeomAlgoAPI_WireBuilder.h>
36
37 #include <algorithm>
38
39 //=================================================================================================
40 BuildPlugin_Polyline::BuildPlugin_Polyline()
41 {
42 }
43
44 //=================================================================================================
45 void BuildPlugin_Polyline::initAttributes()
46 {
47   data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
48   data()->addAttribute(CLOSED_ID(), ModelAPI_AttributeBoolean::typeId());
49 }
50
51 //=================================================================================================
52 void BuildPlugin_Polyline::execute()
53 {
54   // Get closed flag value
55   bool isClosed = boolean(CLOSED_ID())->value();
56
57   // Get base objects list.
58   AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
59
60   // Collect points.
61   ListOfShape aPoints;
62   std::set<GeomShapePtr> aContexts;
63   for (int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
64     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
65                 GeomShapePtr aShape = aSelection->value();
66
67     ResultPtr aContext = aSelection->context();
68     aContexts.insert(aContext->shape());
69
70     if (!aShape.get()) {
71       aShape = aContext->shape();
72                 }
73
74     aPoints.push_back(aShape);
75         }
76
77   // Prepare a list of edges
78   ListOfShape anEdges;
79
80   ListOfShape::const_iterator aPointsIt = aPoints.begin();
81   for (; aPointsIt != std::prev(aPoints.end()); ++aPointsIt) {
82     std::shared_ptr<GeomAPI_Pnt> aPnt = GeomAlgoAPI_PointBuilder::point(*aPointsIt);
83     std::shared_ptr<GeomAPI_Pnt> aPntNext = GeomAlgoAPI_PointBuilder::point(*std::next(aPointsIt));
84     anEdges.push_back(GeomAlgoAPI_EdgeBuilder::line(aPnt, aPntNext));
85   }
86
87   if (isClosed) {
88     std::shared_ptr<GeomAPI_Pnt> aLastPnt = GeomAlgoAPI_PointBuilder::point(aPoints.back());
89     std::shared_ptr<GeomAPI_Pnt> aFirstPnt = GeomAlgoAPI_PointBuilder::point(aPoints.front());
90     anEdges.push_back(GeomAlgoAPI_EdgeBuilder::line(aLastPnt, aFirstPnt));
91   }
92
93   // Create wire from edges
94   GeomShapePtr aWire = GeomAlgoAPI_WireBuilder::wire(anEdges);
95         if (!aWire.get()) {
96                 setError("Error: Result polyline is empty.");
97                 return;
98         }
99
100   // Check the wire.
101   if (GeomAlgoAPI_WireBuilder::isSelfIntersected(aWire)) {
102     setError("Error: Result polyline has self-intersections.");
103     return;
104   }
105
106   // Store result.
107   ResultBodyPtr aResultBody = document()->createBody(data());
108   std::set<GeomShapePtr>::const_iterator aContextIt = aContexts.begin();
109   for (; aContextIt != aContexts.end(); aContextIt++) {
110     aResultBody->storeModified(*aContextIt, aWire, aContextIt == aContexts.begin());
111   }
112
113   aPointsIt = aPoints.cbegin();
114   GeomAPI_ShapeExplorer anExp(aWire, GeomAPI_Shape::EDGE);
115   for (; anExp.more() && aPointsIt != aPoints.cend(); anExp.next(), ++aPointsIt) {
116     GeomShapePtr aPoint = *aPointsIt;
117     GeomShapePtr anEdge = anExp.current();
118     aResultBody->generated(aPoint, anEdge, "Edge");
119   }
120
121   if (!isClosed) {
122     ListOfShape aResPoints;
123     anExp.init(aWire, GeomAPI_Shape::VERTEX);
124     for (; anExp.more(); anExp.next()) {
125       aResPoints.push_back(anExp.current());
126     }
127
128     aResultBody->generated(aResPoints.front(), "FirstVertex");
129     aResultBody->generated(aResPoints.back(), "LastVertex");
130   }
131
132   setResult(aResultBody);
133 }