Salome HOME
Fix for the issue #2753 : error when dump/load script
[modules/shaper.git] / src / GeomAPI / GeomAPI_Wire.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 "GeomAPI_Wire.h"
22 #include "GeomAPI_Pnt.h"
23
24 #include <BRep_Builder.hxx>
25 #include <BRepTools_WireExplorer.hxx>
26 #include <Geom_Line.hxx>
27 #include <Precision.hxx>
28 #include <Standard_Type.hxx>
29 #include <TopoDS.hxx>
30 #include <TopoDS_Wire.hxx>
31
32 //==================================================================================================
33 GeomAPI_Wire::GeomAPI_Wire()
34 {
35   TopoDS_Wire* aWire = new TopoDS_Wire();
36
37   BRep_Builder aBuilder;
38   aBuilder.MakeWire(*aWire);
39
40   this->setImpl(aWire);
41 }
42
43 //==================================================================================================
44 GeomAPI_Wire::GeomAPI_Wire(const std::shared_ptr<GeomAPI_Shape>& theShape)
45 {
46   if (!theShape->isNull() && theShape->isWire()) {
47     setImpl(new TopoDS_Shape(theShape->impl<TopoDS_Shape>()));
48   }
49 }
50
51 //==================================================================================================
52 bool GeomAPI_Wire::isClosed() const
53 {
54   return BRep_Tool::IsClosed(impl<TopoDS_Shape>());
55 }
56
57 //==================================================================================================
58 bool GeomAPI_Wire::isPolygon(std::list<GeomPointPtr>& thePoints) const
59 {
60   const TopoDS_Wire& aWire = TopoDS::Wire(impl<TopoDS_Shape>());
61
62   bool isPolygon = true;
63   const Handle(Standard_Type)& aLineType = STANDARD_TYPE(Geom_Line);
64   for (BRepTools_WireExplorer anExp(aWire); anExp.More() && isPolygon; anExp.Next()) {
65     const TopoDS_Edge& anEdge = anExp.Current();
66     double aT1, aT2;
67     Handle(Geom_Curve) aC3D = BRep_Tool::Curve(anEdge, aT1, aT2);
68     if (!aC3D.IsNull() && aC3D->IsKind(aLineType)) {
69       gp_Pnt aCorner = BRep_Tool::Pnt(anExp.CurrentVertex());
70       thePoints.push_back(GeomPointPtr(new GeomAPI_Pnt(aCorner.X(), aCorner.Y(), aCorner.Z())));
71     }
72     else
73       isPolygon = false;
74   }
75
76   if (!isPolygon)
77     thePoints.clear();
78   return isPolygon;
79 }
80
81 //==================================================================================================
82 bool GeomAPI_Wire::isRectangle(std::list<GeomPointPtr>& thePoints) const
83 {
84   thePoints.clear();
85
86   const TopoDS_Wire& aWire = TopoDS::Wire(impl<TopoDS_Shape>());
87   const Handle(Standard_Type)& aLineType = STANDARD_TYPE(Geom_Line);
88
89   gp_XYZ aPrevDir(0, 0, 0);
90
91   for (BRepTools_WireExplorer anExp(aWire); anExp.More(); anExp.Next()) {
92     const TopoDS_Edge& anEdge = anExp.Current();
93     double aT1, aT2;
94     Handle(Geom_Curve) aC3D = BRep_Tool::Curve(anEdge, aT1, aT2);
95     if (!aC3D.IsNull() && aC3D->IsKind(aLineType)) {
96       gp_Pnt aCorner = BRep_Tool::Pnt(anExp.CurrentVertex());
97       thePoints.push_back(GeomPointPtr(new GeomAPI_Pnt(aCorner.X(), aCorner.Y(), aCorner.Z())));
98     }
99     else
100       return false;
101
102     if (thePoints.size() > 4)
103       return false;
104
105     // collect length of the edge
106     gp_Pnt aStart = aC3D->Value(aT1);
107     gp_Pnt aEnd = aC3D->Value(aT2);
108
109     if (aStart.Distance(aEnd) <= Precision::Confusion())
110       return false;
111
112     // check the edge is orthogonal to the previous
113     gp_XYZ aCurDir = (aEnd.XYZ() - aStart.XYZ()).Normalized();
114     if (aPrevDir.Dot(aCurDir) < Precision::Confusion())
115       aPrevDir = aCurDir;
116     else
117       return false;
118   }
119   return true;
120 }
121
122 //==================================================================================================
123 GeomPointPtr GeomAPI_Wire::middlePoint() const
124 {
125   // find middle edge in the wire
126   std::list<GeomShapePtr> aSubs = subShapes(EDGE);
127   size_t aNbSubs = aSubs.size();
128   if (aNbSubs == 0)
129     return GeomPointPtr();
130
131   aNbSubs /= 2;
132   for (; aNbSubs > 0; --aNbSubs)
133     aSubs.pop_front();
134
135   // compute middle point on the middle edge
136   return aSubs.front()->middlePoint();
137 }