]> SALOME platform Git repositories - modules/shaper.git/blobdiff - src/GeomAlgoAPI/GeomAlgoAPI_ShapeBuilder.cpp
Salome HOME
Issue #1369: Added "SubShapes" feature.
[modules/shaper.git] / src / GeomAlgoAPI / GeomAlgoAPI_ShapeBuilder.cpp
diff --git a/src/GeomAlgoAPI/GeomAlgoAPI_ShapeBuilder.cpp b/src/GeomAlgoAPI/GeomAlgoAPI_ShapeBuilder.cpp
new file mode 100644 (file)
index 0000000..eea9ae9
--- /dev/null
@@ -0,0 +1,175 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
+
+// File:        GeomAlgoAPI_ShapeBuilder.cpp
+// Created:     27 April 2016
+// Author:      Dmitry Bobylev
+
+#include "GeomAlgoAPI_ShapeBuilder.h"
+
+#include "GeomAlgoAPI_MakeShapeCustom.h"
+
+#include <GeomAPI_ShapeIterator.h>
+
+#include <BRep_Builder.hxx>
+#include <BRepBuilderAPI_Copy.hxx>
+#include <BRepExtrema_DistShapeShape.hxx>
+#include <BRepTools_ReShape.hxx>
+#include <Precision.hxx>
+#include <TopExp_Explorer.hxx>
+#include <TopoDS.hxx>
+#include <TopoDS_Edge.hxx>
+#include <TopoDS_Iterator.hxx>
+#include <TopoDS_Shape.hxx>
+
+//==================================================================================================
+void GeomAlgoAPI_ShapeBuilder::add(std::shared_ptr<GeomAPI_Shape> theShape,
+                                   const std::shared_ptr<GeomAPI_Shape> theShapeToAdd)
+{
+  if(!theShape.get() || !theShapeToAdd.get()) {
+    return;
+  }
+
+  TopoDS_Shape* aShape = theShape->implPtr<TopoDS_Shape>();
+  const TopoDS_Shape& aShapeToAdd = theShapeToAdd->impl<TopoDS_Shape>();
+
+  BRep_Builder aBuilder;
+  aBuilder.Add(*aShape, aShapeToAdd);
+}
+
+
+//==================================================================================================
+void GeomAlgoAPI_ShapeBuilder::remove(std::shared_ptr<GeomAPI_Shape> theShape,
+                                      const std::shared_ptr<GeomAPI_Shape> theShapeToRemove)
+{
+  if(!theShape.get() || !theShapeToRemove.get()) {
+    return;
+  }
+
+  TopoDS_Shape* aShape = theShape->implPtr<TopoDS_Shape>();
+  const TopoDS_Shape& aShapeToRemove = theShapeToRemove->impl<TopoDS_Shape>();
+
+  BRep_Builder aBuilder;
+  aBuilder.Remove(*aShape, aShapeToRemove);
+}
+
+//==================================================================================================
+GeomAlgoAPI_ShapeBuilder::GeomAlgoAPI_ShapeBuilder()
+{
+}
+
+//==================================================================================================
+void GeomAlgoAPI_ShapeBuilder::removeInternal(const std::shared_ptr<GeomAPI_Shape> theShape)
+{
+  GeomShapePtr aResultShape = theShape->emptyCopied();
+  GeomAPI_Shape::ShapeType aBaseShapeType = theShape->shapeType();
+  std::shared_ptr<GeomAlgoAPI_MakeShapeCustom> aMakeShapeCustom(new GeomAlgoAPI_MakeShapeCustom());
+  for(GeomAPI_ShapeIterator anIter(theShape); anIter.more(); anIter.next()) {
+    GeomShapePtr aSubShape = anIter.current();
+    if(aBaseShapeType == GeomAPI_Shape::WIRE) {
+      GeomShapePtr aSubShapeCopy = aSubShape->emptyCopied();
+      aMakeShapeCustom->addModified(aSubShape, aSubShapeCopy);
+      for(GeomAPI_ShapeIterator aSubIter(aSubShape); aSubIter.more(); aSubIter.next()) {
+        GeomShapePtr aSubOfSubShape = aSubIter.current();
+        if(aSubOfSubShape->orientation() != GeomAPI_Shape::INTERNAL) {
+          GeomAlgoAPI_ShapeBuilder::add(aSubShapeCopy, aSubOfSubShape);
+        }
+      }
+      GeomAlgoAPI_ShapeBuilder::add(aResultShape, aSubShapeCopy);
+    } else if(aBaseShapeType == GeomAPI_Shape::FACE) {
+      if(aSubShape->shapeType() == GeomAPI_Shape::WIRE
+          && aSubShape->orientation() != GeomAPI_Shape::INTERNAL) {
+        GeomAlgoAPI_ShapeBuilder::add(aResultShape, aSubShape);
+      }
+    }
+  }
+
+  this->appendAlgo(aMakeShapeCustom);
+
+  setShape(aResultShape);
+  setDone(true);
+}
+
+//==================================================================================================
+void GeomAlgoAPI_ShapeBuilder::add(const std::shared_ptr<GeomAPI_Shape> theShape,
+                                   const ListOfShape& theShapesToAdd)
+{
+  // Get base shape.
+  if(!theShape.get()) {
+    return;
+  }
+  const TopoDS_Shape& aBaseShape = theShape->impl<TopoDS_Shape>();
+  TopAbs_ShapeEnum aBaseShapeType = aBaseShape.ShapeType();
+
+  // Copy base shape.
+  BRepBuilderAPI_Copy* aCopyBuilder = new BRepBuilderAPI_Copy(aBaseShape, Standard_False);
+  this->appendAlgo(std::shared_ptr<GeomAlgoAPI_MakeShape>(new GeomAlgoAPI_MakeShape(aCopyBuilder)));
+  if(!aCopyBuilder->IsDone()) {
+    return;
+  }
+  TopoDS_Shape aResultShape = aCopyBuilder->Shape();
+
+  // Copy sub-shapes from list to new shape.
+  BRep_Builder aBuilder;
+  std::shared_ptr<GeomAlgoAPI_MakeShapeCustom> aMakeShapeCustom(new GeomAlgoAPI_MakeShapeCustom());
+  for(ListOfShape::const_iterator anIt = theShapesToAdd.cbegin(); anIt != theShapesToAdd.cend(); ++anIt) {
+    TopoDS_Shape aShapeToAdd = (*anIt)->impl<TopoDS_Shape>();
+    for(TopExp_Explorer aResExp(aResultShape, TopAbs_VERTEX); aResExp.More(); aResExp.Next()) {
+      const TopoDS_Vertex& aVertexInRes = TopoDS::Vertex(aResExp.Current());
+      const gp_Pnt aPntInRes = BRep_Tool::Pnt(aVertexInRes);
+      for(TopExp_Explorer anAddExp(aShapeToAdd, TopAbs_VERTEX); anAddExp.More(); anAddExp.Next()) {
+        const TopoDS_Vertex& aVertexInAdd = TopoDS::Vertex(anAddExp.Current());
+        const gp_Pnt aPntInAdd = BRep_Tool::Pnt(aVertexInAdd);
+        if(aPntInRes.Distance(aPntInAdd) < Precision::Confusion()) {
+          BRepTools_ReShape aReShape;
+          TopoDS_Shape aVertexInResMod = aVertexInRes;
+          aVertexInResMod.Orientation(aVertexInAdd.Orientation());
+          aReShape.Replace(aVertexInAdd, aVertexInResMod);
+          TopoDS_Shape aModShape = aReShape.Apply(aShapeToAdd);
+
+          GeomShapePtr aGeomBaseShape(new GeomAPI_Shape());
+          GeomShapePtr aGeomModShape(new GeomAPI_Shape());
+          aGeomBaseShape->setImpl(new TopoDS_Shape(aShapeToAdd));
+          aGeomModShape->setImpl(new TopoDS_Shape(aModShape));
+          aMakeShapeCustom->addModified(aGeomBaseShape, aGeomModShape);
+          aShapeToAdd = aModShape;
+        }
+      }
+    }
+    TopAbs_ShapeEnum aShapeToAddType = aShapeToAdd.ShapeType();
+    if(aBaseShapeType == TopAbs_WIRE) {
+      if(aShapeToAddType == TopAbs_VERTEX) {
+        // Find on which edge vertex is lie and add to this edge.
+        for(TopExp_Explorer aResultExp(aResultShape, TopAbs_EDGE); aResultExp.More(); aResultExp.Next()) {
+          TopoDS_Shape anEdge = aResultExp.Current();
+          BRepExtrema_DistShapeShape aDist(anEdge, aShapeToAdd);
+          aDist.Perform();
+          if(aDist.IsDone() && aDist.Value() <= Precision::Confusion()) {
+            aShapeToAdd.Orientation(TopAbs_INTERNAL);
+            Standard_Boolean aFreeFlag = anEdge.Free();
+            anEdge.Free(Standard_True);
+            aBuilder.Add(anEdge, aShapeToAdd);
+            anEdge.Free(aFreeFlag);
+            break;
+          }
+        }
+      }
+    } else if(aBaseShapeType == GeomAPI_Shape::FACE) {
+      if(aShapeToAddType == GeomAPI_Shape::EDGE) {
+        aShapeToAdd.Orientation(TopAbs_INTERNAL);
+        TopoDS_Wire aWire;
+        aBuilder.MakeWire(aWire);
+        aBuilder.Add(aWire, aShapeToAdd);
+        aShapeToAdd = aWire;
+        aShapeToAdd.Orientation(TopAbs_INTERNAL);
+      }
+      aBuilder.Add(aResultShape, aShapeToAdd);
+    }
+  }
+  this->appendAlgo(aMakeShapeCustom);
+
+  // Set result.
+  std::shared_ptr<GeomAPI_Shape> aShape(new GeomAPI_Shape());
+  aShape->setImpl(new TopoDS_Shape(aResultShape));
+  setShape(aShape);
+  setDone(true);
+}