1 // Copyright (C) 2014-2017 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
21 #include "BuildPlugin_Polyline.h"
23 #include <ModelAPI_AttributeBoolean.h>
24 #include <ModelAPI_AttributeSelectionList.h>
25 #include <ModelAPI_ResultBody.h>
26 #include <ModelAPI_ResultConstruction.h>
28 #include <Events_InfoMessage.h>
30 #include <GeomAPI_ShapeExplorer.h>
31 #include <GeomAlgoAPI_WireBuilder.h>
33 #include <GeomAlgoAPI_ShapeTools.h>
34 #include <GeomAlgoAPI_EdgeBuilder.h>
35 #include <GeomAlgoAPI_PointBuilder.h>
36 #include <GeomAlgoAPI_WireBuilder.h>
40 //=================================================================================================
41 BuildPlugin_Polyline::BuildPlugin_Polyline()
45 //=================================================================================================
46 void BuildPlugin_Polyline::initAttributes()
48 data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
49 data()->addAttribute(CLOSED_ID(), ModelAPI_AttributeBoolean::typeId());
52 //=================================================================================================
53 void BuildPlugin_Polyline::execute()
55 // Get closed flag value
56 bool isClosed = boolean(CLOSED_ID())->value();
58 // Get base objects list.
59 AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
63 std::set<GeomShapePtr> aContexts;
64 for (int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
65 AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
66 GeomShapePtr aShape = aSelection->value();
68 ResultPtr aContext = aSelection->context();
69 aContexts.insert(aContext->shape());
72 aShape = aContext->shape();
75 aPoints.push_back(aShape);
78 // Prepare a list of edges
81 ListOfShape::const_iterator aPointsIt = aPoints.begin();
82 for (; aPointsIt != std::prev(aPoints.end()); ++aPointsIt) {
83 std::shared_ptr<GeomAPI_Pnt> aPnt = GeomAlgoAPI_PointBuilder::point(*aPointsIt);
84 std::shared_ptr<GeomAPI_Pnt> aPntNext = GeomAlgoAPI_PointBuilder::point(*std::next(aPointsIt));
85 anEdges.push_back(GeomAlgoAPI_EdgeBuilder::line(aPnt, aPntNext));
89 std::shared_ptr<GeomAPI_Pnt> aLastPnt = GeomAlgoAPI_PointBuilder::point(aPoints.back());
90 std::shared_ptr<GeomAPI_Pnt> aFirstPnt = GeomAlgoAPI_PointBuilder::point(aPoints.front());
91 anEdges.push_back(GeomAlgoAPI_EdgeBuilder::line(aLastPnt, aFirstPnt));
94 // Create wire from edges
95 GeomShapePtr aWire = GeomAlgoAPI_WireBuilder::wire(anEdges);
97 setError("Error: Result polyline is empty.");
102 if (GeomAlgoAPI_WireBuilder::isSelfIntersected(aWire)) {
103 setError("Error: Result polyline has self-intersections.");
108 ResultBodyPtr aResultBody = document()->createBody(data());
109 std::set<GeomShapePtr>::const_iterator aContextIt = aContexts.begin();
110 for (; aContextIt != aContexts.end(); aContextIt++) {
111 aResultBody->storeModified(*aContextIt, aWire, aContextIt == aContexts.begin() ? 0 : -2);
114 aPointsIt = aPoints.cbegin();
115 GeomAPI_ShapeExplorer anExp(aWire, GeomAPI_Shape::EDGE);
116 for (; anExp.more() && aPointsIt != aPoints.cend(); anExp.next(), ++aPointsIt) {
117 GeomShapePtr aPoint = *aPointsIt;
118 GeomShapePtr anEdge = anExp.current();
119 aResultBody->generated(aPoint, anEdge, "Edge", 1);
123 ListOfShape aResPoints;
124 anExp.init(aWire, GeomAPI_Shape::VERTEX);
125 for (; anExp.more(); anExp.next()) {
126 aResPoints.push_back(anExp.current());
129 aResultBody->generated(aResPoints.front(), "FirstVertex", 2);
130 aResultBody->generated(aResPoints.back(), "LastVertex", 3);
133 setResult(aResultBody);