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>
32 #include <GeomAlgoAPI_ShapeTools.h>
33 #include <GeomAlgoAPI_EdgeBuilder.h>
34 #include <GeomAlgoAPI_PointBuilder.h>
35 #include <GeomAlgoAPI_WireBuilder.h>
39 //=================================================================================================
40 BuildPlugin_Polyline::BuildPlugin_Polyline()
44 //=================================================================================================
45 void BuildPlugin_Polyline::initAttributes()
47 data()->addAttribute(BASE_OBJECTS_ID(), ModelAPI_AttributeSelectionList::typeId());
48 data()->addAttribute(CLOSED_ID(), ModelAPI_AttributeBoolean::typeId());
51 //=================================================================================================
52 void BuildPlugin_Polyline::execute()
54 // Get closed flag value
55 bool isClosed = boolean(CLOSED_ID())->value();
57 // Get base objects list.
58 AttributeSelectionListPtr aSelectionList = selectionList(BASE_OBJECTS_ID());
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();
67 ResultPtr aContext = aSelection->context();
68 aContexts.insert(aContext->shape());
71 aShape = aContext->shape();
74 aPoints.push_back(aShape);
77 // Prepare a list of edges
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));
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));
93 // Create wire from edges
94 GeomShapePtr aWire = GeomAlgoAPI_WireBuilder::wire(anEdges);
96 setError("Error: Result polyline is empty.");
101 if (GeomAlgoAPI_WireBuilder::isSelfIntersected(aWire)) {
102 setError("Error: Result polyline has self-intersections.");
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() ? 0 : -2);
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", 1);
122 ListOfShape aResPoints;
123 anExp.init(aWire, GeomAPI_Shape::VERTEX);
124 for (; anExp.more(); anExp.next()) {
125 aResPoints.push_back(anExp.current());
128 aResultBody->generated(aResPoints.front(), "FirstVertex", 2);
129 aResultBody->generated(aResPoints.back(), "LastVertex", 3);
132 setResult(aResultBody);