Salome HOME
updated copyright message
[modules/shaper.git] / src / GeomAPI / GeomAPI_Wire.cpp
index 039be40a873ffb450f9b7ad609a0b1bffd9c2453..1ef0fb365a783df05c5cbdc473a171e0577a72d6 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2014-2017  CEA/DEN, EDF R&D
+// Copyright (C) 2014-2023  CEA, EDF
 //
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 //
 // You should have received a copy of the GNU Lesser General Public
 // License along with this library; if not, write to the Free Software
-// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 //
-// See http://www.salome-platform.org/ or
-// email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 //
 
 #include "GeomAPI_Wire.h"
+#include "GeomAPI_Pnt.h"
 
 #include <BRep_Builder.hxx>
+#include <BRepTools_WireExplorer.hxx>
+#include <Geom_Line.hxx>
+#include <Geom_TrimmedCurve.hxx>
+#include <Precision.hxx>
+#include <Standard_Type.hxx>
+#include <TopoDS.hxx>
 #include <TopoDS_Wire.hxx>
 
 //==================================================================================================
@@ -33,3 +39,104 @@ GeomAPI_Wire::GeomAPI_Wire()
 
   this->setImpl(aWire);
 }
+
+//==================================================================================================
+GeomAPI_Wire::GeomAPI_Wire(const std::shared_ptr<GeomAPI_Shape>& theShape)
+{
+  if (!theShape->isNull() && theShape->isWire()) {
+    setImpl(new TopoDS_Shape(theShape->impl<TopoDS_Shape>()));
+  }
+}
+
+//==================================================================================================
+bool GeomAPI_Wire::isClosed() const
+{
+  return BRep_Tool::IsClosed(impl<TopoDS_Shape>());
+}
+
+//==================================================================================================
+bool GeomAPI_Wire::isPolygon(std::list<GeomPointPtr>& thePoints) const
+{
+  const TopoDS_Wire& aWire = TopoDS::Wire(impl<TopoDS_Shape>());
+
+  bool isPolygon = true;
+  const Handle(Standard_Type)& aLineType = STANDARD_TYPE(Geom_Line);
+  for (BRepTools_WireExplorer anExp(aWire); anExp.More() && isPolygon; anExp.Next()) {
+    const TopoDS_Edge& anEdge = anExp.Current();
+    double aT1, aT2;
+    Handle(Geom_Curve) aC3D = BRep_Tool::Curve(anEdge, aT1, aT2);
+    if (!aC3D.IsNull() && aC3D->IsKind(aLineType)) {
+      gp_Pnt aCorner = BRep_Tool::Pnt(anExp.CurrentVertex());
+      thePoints.push_back(GeomPointPtr(new GeomAPI_Pnt(aCorner.X(), aCorner.Y(), aCorner.Z())));
+    }
+    else
+      isPolygon = false;
+  }
+
+  if (!isPolygon)
+    thePoints.clear();
+  return isPolygon;
+}
+
+//==================================================================================================
+bool GeomAPI_Wire::isRectangle(std::list<GeomPointPtr>& thePoints) const
+{
+  thePoints.clear();
+
+  const TopoDS_Wire& aWire = TopoDS::Wire(impl<TopoDS_Shape>());
+  const Handle(Standard_Type)& aLineType = STANDARD_TYPE(Geom_Line);
+  const Handle(Standard_Type)& aTrimmedCurveType = STANDARD_TYPE(Geom_TrimmedCurve);
+
+  gp_XYZ aPrevDir(0, 0, 0);
+
+  for (BRepTools_WireExplorer anExp(aWire); anExp.More(); anExp.Next()) {
+    const TopoDS_Edge& anEdge = anExp.Current();
+    double aT1, aT2;
+    Handle(Geom_Curve) aC3D = BRep_Tool::Curve(anEdge, aT1, aT2);
+    if (aC3D.IsNull())
+      continue;
+    while (aC3D->IsKind(aTrimmedCurveType))
+      aC3D = Handle(Geom_TrimmedCurve)::DownCast(aC3D)->BasisCurve();
+    if (!aC3D.IsNull() && aC3D->IsKind(aLineType)) {
+      gp_Pnt aCorner = BRep_Tool::Pnt(anExp.CurrentVertex());
+      thePoints.push_back(GeomPointPtr(new GeomAPI_Pnt(aCorner.X(), aCorner.Y(), aCorner.Z())));
+    }
+    else
+      return false;
+
+    if (thePoints.size() > 4)
+      return false;
+
+    // collect length of the edge
+    gp_Pnt aStart = aC3D->Value(aT1);
+    gp_Pnt aEnd = aC3D->Value(aT2);
+
+    if (aStart.Distance(aEnd) <= Precision::Confusion())
+      return false;
+
+    // check the edge is orthogonal to the previous
+    gp_XYZ aCurDir = (aEnd.XYZ() - aStart.XYZ()).Normalized();
+    if (aPrevDir.Dot(aCurDir) < Precision::Confusion())
+      aPrevDir = aCurDir;
+    else
+      return false;
+  }
+  return thePoints.size() == 4;
+}
+
+//==================================================================================================
+GeomPointPtr GeomAPI_Wire::middlePoint() const
+{
+  // find middle edge in the wire
+  std::list<GeomShapePtr> aSubs = subShapes(EDGE);
+  size_t aNbSubs = aSubs.size();
+  if (aNbSubs == 0)
+    return GeomPointPtr();
+
+  aNbSubs /= 2;
+  for (; aNbSubs > 0; --aNbSubs)
+    aSubs.pop_front();
+
+  // compute middle point on the middle edge
+  return aSubs.front()->middlePoint();
+}