Salome HOME
Mantis 0054772: crash when select point in viewer inside created face
[modules/geom.git] / src / GEOM_SWIG / GEOM_TestHealing.py
index b5f4d688abd88e6d161859ff80860cbad21a3261..efb2f98d52b3197d51474067c468bab40e28b3e1 100644 (file)
@@ -1,30 +1,31 @@
-#  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
+#  -*- coding: iso-8859-1 -*-
+# Copyright (C) 2007-2022  CEA/DEN, EDF R&D, OPEN CASCADE
 #
-#  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-#  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+# Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+# CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
 #
-#  This library is free software; you can redistribute it and/or
-#  modify it under the terms of the GNU Lesser General Public
-#  License as published by the Free Software Foundation; either
-#  version 2.1 of the License.
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
 #
-#  This library is distributed in the hope that it will be useful,
-#  but WITHOUT ANY WARRANTY; without even the implied warranty of
-#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-#  Lesser General Public License for more details.
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
 #
-#  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
+# 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
 #
-#  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 #
-#  GEOM GEOM_SWIG : binding of C++ implementaion with Python
+
+#  GEOM_SWIG : binding of C++ implementation with Python
 #  File   : GEOM_TestHealing.py
 #  Author : Julia DOROVSKIKH
 #  Module : GEOM
-#  $Header$
-#
+
 def TestProcessShape (geompy):
 
   ##Load shape from BREP file
@@ -50,12 +51,12 @@ def TestProcessShape (geompy):
   theShape = geompy.MakePrismVecH(face, edge, 130)
 
   #Check shape
-  print "Before ProcessShape:"
+  print("Before ProcessShape:")
   isValid = geompy.CheckShape(theShape)
   if isValid == 0:
-    print "The shape is not valid"
+    print("The shape is not valid")
   else:
-    print "The shape seems to be valid"
+    print("The shape seems to be valid")
 
   #Process Shape
   Operators = ["FixShape"]
@@ -65,13 +66,13 @@ def TestProcessShape (geompy):
   PS = geompy.ProcessShape(theShape, Operators, Parameters, Values)
 
   #Check shape
-  print "After ProcessShape:"
+  print("After ProcessShape:")
   isValid = geompy.CheckShape(PS)
   if isValid == 0:
-    print "The shape is not valid"
-    raise RuntimeError, "It seems, that the ProcessShape() has failed"
+    print("The shape is not valid")
+    raise RuntimeError("It seems, that the ProcessShape() has failed")
   else:
-    print "The shape seems to be valid"
+    print("The shape seems to be valid")
 
   #Add In Study
   Id_Shape = geompy.addToStudy(theShape, "Invalid Shape")
@@ -84,7 +85,7 @@ def TestSuppressFaces (geompy):
 
   #IDList for SuppHole
   faces = []
-  faces = geompy.SubShapeAllSorted(Box, geompy.ShapeType["FACE"])
+  faces = geompy.SubShapeAllSortedCentres(Box, geompy.ShapeType["FACE"])
 
   f_glob_id = geompy.GetSubShapeID(Box, faces[5])
 
@@ -137,7 +138,7 @@ def TestSuppressInternalWires (geompy):
     nbw1 = nbw1 + 1
 
   if nbw1 != 2:
-    raise RuntimeError, "GetFreeBoundary(f12) must return 2 closed wires, but returned ", nbw1
+    raise RuntimeError("GetFreeBoundary(f12) must return 2 closed wires, but returned ").with_traceback(nbw1)
 
   #SuppressInternalWires
   face = geompy.SuppressInternalWires(f12, [])
@@ -153,8 +154,8 @@ def TestSuppressInternalWires (geompy):
     nbw2 = nbw2 + 1
 
   if nbw2 != 1:
-    print "GetFreeBoundary(face) must return 1 closed wires, but returned ", nbw2
-    raise RuntimeError, "SuppressInternalWires() works not correctly"
+    print("GetFreeBoundary(face) must return 1 closed wires, but returned ", nbw2)
+    raise RuntimeError("SuppressInternalWires() works not correctly")
 
   #Add In Study
   Id_face = geompy.addToStudy(face, "Face without internal wires")
@@ -180,12 +181,12 @@ def TestCloseContour (geompy):
   Shape = geompy.MakePolyline([p0, pz, py, p200])
 
   #Check shape
-  print "Before closing contour:"
+  print("Before closing contour:")
   isValid = geompy.CheckShape(Shape)
   if isValid == 0:
-    print "The shape is not valid"
+    print("The shape is not valid")
   else:
-    print "The shape seems to be valid"
+    print("The shape seems to be valid")
 
   #Close Contour
   IsCommonVertex = 0 # false
@@ -201,13 +202,13 @@ def TestCloseContour (geompy):
   CC = geompy.CloseContour(Shape, Wires, IsCommonVertex)
 
   #Check shape
-  print "After closing contour:"
+  print("After closing contour:")
   isValid = geompy.CheckShape(CC)
   if isValid == 0:
-    print "The shape is not valid"
-    raise RuntimeError, "It seems, that the contour was not closed"
+    print("The shape is not valid")
+    raise RuntimeError("It seems, that the contour was not closed")
   else:
-    print "The shape seems to be valid"
+    print("The shape seems to be valid")
 
   #Add In Study
   Id_Shape = geompy.addToStudy(Shape, "Shape with open wire")
@@ -235,49 +236,50 @@ def TestSuppressHoles (geompy):
 
   #IDList for SuppressFaces
   faces = []
-  faces = geompy.SubShapeAllSorted(Cut, geompy.ShapeType["FACE"])
+  faces = geompy.SubShapeAllSortedCentres(Cut, geompy.ShapeType["FACE"])
   ind = 0
   for face in faces:
       f_name = "FACE %d"%(ind)
       f_id = geompy.addToStudyInFather(Cut, face, f_name)
 
       f_glob_id = geompy.GetSubShapeID(Cut, face)
-      print "face ", ind, " global index = ", f_glob_id
+      print("face ", ind, " global index = ", f_glob_id)
       ind = ind + 1
 
   f_glob_id_0 = geompy.GetSubShapeID(Cut, faces[0])
   cut_without_f_0 = geompy.SuppressFaces(Cut, [f_glob_id_0])
+  geompy.addToStudy(cut_without_f_0, "Cut without face 0")
 
   faces1 = []
-  faces1 = geompy.SubShapeAllSorted(cut_without_f_0, geompy.ShapeType["FACE"])
+  faces1 = geompy.SubShapeAllSortedCentres(cut_without_f_0, geompy.ShapeType["FACE"])
   ind = 0
   for face in faces1:
       f_name = "FACE %d"%(ind)
       f_id = geompy.addToStudyInFather(cut_without_f_0, face, f_name)
 
       f_glob_id = geompy.GetSubShapeID(cut_without_f_0, face)
-      print "face ", ind, " global index = ", f_glob_id
+      print("face ", ind, " global index = ", f_glob_id)
       ind = ind + 1
 
-  f_glob_id_5 = geompy.GetSubShapeID(cut_without_f_0, faces1[5])
-  cut_without_f_0_5 = geompy.SuppressFaces(cut_without_f_0, [f_glob_id_5])
-  cut_without_f_0_5_id = geompy.addToStudy(cut_without_f_0_5, "Cut without faces 0 and 5")
+  f_glob_id_3 = geompy.GetSubShapeID(cut_without_f_0, faces1[3])
+  cut_without_f_0_3 = geompy.SuppressFaces(cut_without_f_0, [f_glob_id_3])
+  cut_without_f_0_3_id = geompy.addToStudy(cut_without_f_0_3, "Cut without faces 0 and 3")
 
   #IDList for SuppHole
   wires = []
-  wires = geompy.SubShapeAllSorted(cut_without_f_0_5, geompy.ShapeType["WIRE"])
+  wires = geompy.SubShapeAllSortedCentres(cut_without_f_0_3, geompy.ShapeType["WIRE"])
   ind = 0
   for wire in wires:
       w_name = "WIRE %d"%(ind)
-      w_id = geompy.addToStudyInFather(cut_without_f_0_5, wire, w_name)
+      w_id = geompy.addToStudyInFather(cut_without_f_0_3, wire, w_name)
 
-      w_glob_id = geompy.GetSubShapeID(cut_without_f_0_5, wire)
-      print "wire ", ind, " global index = ", w_glob_id
+      w_glob_id = geompy.GetSubShapeID(cut_without_f_0_3, wire)
+      print("wire ", ind, " global index = ", w_glob_id)
       ind = ind + 1
 
-  w_3 = geompy.GetSubShapeID(cut_without_f_0_5, wires[3])
+  w_3 = geompy.GetSubShapeID(cut_without_f_0_3, wires[3])
 
-  SuppHole3 = geompy.SuppressHoles(cut_without_f_0_5, [w_3])
+  SuppHole3 = geompy.SuppressHoles(cut_without_f_0_3, [w_3])
   SuppHole3_id = geompy.addToStudy(SuppHole3, "Supp Hole 3")
 
 def TestMakeSewing (geompy, math):
@@ -312,7 +314,7 @@ def TestDivideEdge (geompy):
   Box = geompy.MakeBoxDXDYDZ(200., 200., 200.)
 
   #Divide Edge
-  box_edges = geompy.SubShapeAllSorted(Box, geompy.ShapeType["EDGE"])
+  box_edges = geompy.SubShapeAllSortedCentres(Box, geompy.ShapeType["EDGE"])
   edge_ind = geompy.GetSubShapeID(Box, box_edges[1])
 
   Divide = geompy.DivideEdge(Box, edge_ind, 0.5, 1) # Obj, ind, param, is_curve_param
@@ -321,8 +323,203 @@ def TestDivideEdge (geompy):
   Id_Box    = geompy.addToStudy(Box, "Box")
   Id_Divide = geompy.addToStudy(Divide, "Box with Divided Edge")
 
+  # using geompy.DivideEdgeByPoint()
+  p    = geompy.MakeVertex( 30, -5, 10, theName="Point to project" )
+  edge = geompy.GetEdgeNearPoint( Box, p, theName="Edge to split")
+  div  = geompy.DivideEdgeByPoint( Box, edge, p, theName="Box (edge divided)")
+  assert geompy.NumberOfEdges( Box ) == geompy.NumberOfEdges( div ) - 1
+
+def TestFuseEdges (geompy):
+
+  # create vertices
+  p1 = geompy.MakeVertex(0, 0, 0)
+  p2 = geompy.MakeVertex(70, 0, 0)
+  p3 = geompy.MakeVertex(70, 50, 0)
+  p4 = geompy.MakeVertex(70, 80, 0)
+  p5 = geompy.MakeVertex(50, 80, 0)
+  p6 = geompy.MakeVertex(20, 80, 0)
+  p7 = geompy.MakeVertex(0, 80, 0)
+  p8 = geompy.MakeVertex(0, 30, 0)
+
+  points = [p1, p2, p3, p4, p5, p6, p7, p8]
+
+  # make a wire
+  wire_1 = geompy.MakePolyline(points, True)
+
+  # suppress some vertices in the wire
+  wire_2 = geompy.FuseCollinearEdgesWithinWire(wire_1, [p3])
+  wire_3 = geompy.FuseCollinearEdgesWithinWire(wire_1, [p5, p6])
+
+  # suppress all suitable vertices in the wire
+  wire_4 = geompy.FuseCollinearEdgesWithinWire(wire_1, [])
+
+  wires = [wire_1, wire_2, wire_3, wire_4]
+
+  # add objects in the study
+  ii = 1
+  for point in points:
+    geompy.addToStudy(point, "p%d"%ii)
+    ii = ii + 1
+    pass
+
+  ii = 1
+  for wire in wires:
+    geompy.addToStudy(wire, "wire_%d"%ii)
+    wire_points = geompy.SubShapeAllSortedCentres(wire, geompy.ShapeType["VERTEX"])
+    jj = 1
+    for point in wire_points:
+      geompy.addToStudyInFather(wire, point, "point_%d"%jj)
+      jj = jj + 1
+      pass
+    ii = ii + 1
+    pass
+
+def TestRemoveWebs (geompy):
+
+  # create solids with some coincident faces
+  Box_1 = geompy.MakeBoxDXDYDZ(200, 200, 200)
+  Translation_1 = geompy.MakeTranslation(Box_1, 200, 0, 0)
+  Translation_2 = geompy.MakeTranslation(Box_1, 200, 200, 0)
+
+  geompy.addToStudy(Box_1, 'Box_1')
+  geompy.addToStudy(Translation_1, 'Translation_1')
+  geompy.addToStudy(Translation_2, 'Translation_2')
+
+  # partition three solids to obtain shared faces
+  Partition_1 = geompy.MakePartition([Box_1, Translation_1, Translation_2])
+  geompy.addToStudy(Partition_1, 'Partition_1')
+
+  # remove shared faces to obtain one solid instead of three
+  Joined_1 = geompy.RemoveInternalFaces(Partition_1)
+  geompy.addToStudy(Joined_1, 'Joined_1')
+
+def TestSewGluing(geompy):
+
+  import GEOM
+  box1 = geompy.MakeBox(0,0,0, 1,1,1)
+  box2 = geompy.MakeBox(1,0,0, 2,1,1)
+  comp = geompy.MakeCompound( [box1, box2] )
+
+  # no sewing with AllowNonManifold=False
+  sew1 = geompy.MakeSewing( [box1,box2], 1e-5, AllowNonManifold=False)
+  assert not sew1
+  sew2 = geompy.MakeSewing( comp, 1e-5, AllowNonManifold=False)
+  assert not sew2
+  sew3 = geompy.MakeSewing( [comp], 1e-5, AllowNonManifold=False)
+  assert not sew3
+  sew1 = geompy.Sew( [box1,box2], 1e-5, AllowNonManifold=False)
+  assert not sew1
+  sew2 = geompy.Sew( comp, 1e-5, AllowNonManifold=False)
+  assert not sew2
+  sew3 = geompy.Sew( [comp], 1e-5, AllowNonManifold=False)
+  assert not sew3
+
+  # check MakeSewing()
+  sew1 = geompy.MakeSewing( [box1,box2], 1e-5, AllowNonManifold=True)
+  assert sew1.GetShapeType() == GEOM.SHELL
+  assert geompy.NumberOfFaces( sew1 ) == geompy.NumberOfFaces( comp )
+  assert geompy.NumberOfEdges( sew1 ) == geompy.NumberOfEdges( comp ) - 4
+  sew2 = geompy.MakeSewing( comp, 1e-5, AllowNonManifold=True)
+  assert sew2.GetShapeType() == GEOM.SHELL
+  assert geompy.NumberOfFaces( sew2 ) == geompy.NumberOfFaces( comp )
+  assert geompy.NumberOfEdges( sew2 ) == geompy.NumberOfEdges( comp ) - 4
+  sew3 = geompy.MakeSewing( [comp], 1e-5, AllowNonManifold=True)
+  assert sew3.GetShapeType() == GEOM.SHELL
+  assert geompy.NumberOfFaces( sew3 ) == geompy.NumberOfFaces( comp )
+  assert geompy.NumberOfEdges( sew3 ) == geompy.NumberOfEdges( comp ) - 4
+  # check Sew()
+  sew1 = geompy.Sew( [box1,box2], 1e-5, AllowNonManifold=True)
+  assert sew1.GetShapeType() == GEOM.SHELL
+  assert geompy.NumberOfFaces( sew1 ) == geompy.NumberOfFaces( comp )
+  assert geompy.NumberOfEdges( sew1 ) == geompy.NumberOfEdges( comp ) - 4
+  sew2 = geompy.Sew( comp, 1e-5, AllowNonManifold=True)
+  assert sew2.GetShapeType() == GEOM.SHELL
+  assert geompy.NumberOfFaces( sew2 ) == geompy.NumberOfFaces( comp )
+  assert geompy.NumberOfEdges( sew2 ) == geompy.NumberOfEdges( comp ) - 4
+  sew3 = geompy.Sew( [comp], 1e-5, AllowNonManifold=True)
+  assert sew3.GetShapeType() == GEOM.SHELL
+  assert geompy.NumberOfFaces( sew3 ) == geompy.NumberOfFaces( comp )
+  assert geompy.NumberOfEdges( sew3 ) == geompy.NumberOfEdges( comp ) - 4
+
+  # check MakeGlueFaces()
+  glueF1 = geompy.MakeGlueFaces( [box1,box2], 1e-5)
+  assert glueF1.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfFaces( glueF1 ) == geompy.NumberOfFaces( comp ) - 1
+  assert geompy.NumberOfEdges( glueF1 ) == geompy.NumberOfEdges( comp ) - 4
+  glueF2 = geompy.MakeGlueFaces( [comp], 1e-5)
+  assert glueF2.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfFaces( glueF2 ) == geompy.NumberOfFaces( comp ) - 1
+  assert geompy.NumberOfEdges( glueF2 ) == geompy.NumberOfEdges( comp ) - 4
+  glueF3 = geompy.MakeGlueFaces( comp, 1e-5)
+  assert glueF3.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfFaces( glueF3 ) == geompy.NumberOfFaces( comp ) - 1
+  assert geompy.NumberOfEdges( glueF3 ) == geompy.NumberOfEdges( comp ) - 4
+
+  # check GetGlueFaces()
+  glueFF1 = geompy.GetGlueFaces( [box1,box2], 1e-5)
+  assert len( glueFF1 ) == 1 and glueFF1[0].GetShapeType() == GEOM.FACE
+  glueFF2 = geompy.GetGlueFaces( [comp], 1e-5)
+  assert len( glueFF2 ) == 1 and glueFF2[0].GetShapeType() == GEOM.FACE
+  glueFF3 = geompy.GetGlueFaces( comp, 1e-5)
+  assert len( glueFF3 ) == 1 and glueFF3[0].GetShapeType() == GEOM.FACE
+
+  #check MakeGlueFacesByList()
+  glueF1 = geompy.MakeGlueFacesByList( [box1,box2], 1e-5, glueFF1)
+  assert glueF1.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfFaces( glueF1 ) == geompy.NumberOfFaces( comp ) - 1
+  assert geompy.NumberOfEdges( glueF1 ) == geompy.NumberOfEdges( comp ) - 4
+  glueF2 = geompy.MakeGlueFacesByList( [comp], 1e-5, glueFF2)
+  assert glueF2.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfFaces( glueF2 ) == geompy.NumberOfFaces( comp ) - 1
+  assert geompy.NumberOfEdges( glueF2 ) == geompy.NumberOfEdges( comp ) - 4
+  glueF3 = geompy.MakeGlueFacesByList( comp, 1e-5, glueFF3 )
+  assert glueF3.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfFaces( glueF3 ) == geompy.NumberOfFaces( comp ) - 1
+  assert geompy.NumberOfEdges( glueF3 ) == geompy.NumberOfEdges( comp ) - 4
+
+  # check MakeGlueEdges()
+  glueE1 = geompy.MakeGlueEdges( [box1,box2], 1e-5)
+  assert glueE1.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfEdges( glueE1 ) == geompy.NumberOfEdges( comp ) - 4
+  glueE2 = geompy.MakeGlueEdges( [comp], 1e-5)
+  assert glueE2.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfEdges( glueE2 ) == geompy.NumberOfEdges( comp ) - 4
+  glueE3 = geompy.MakeGlueEdges( comp, 1e-5)
+  assert glueE3.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfEdges( glueE3 ) == geompy.NumberOfEdges( comp ) - 4
+
+  # check GetGlueEdges()
+  glueEE1 = geompy.GetGlueEdges( [box1,box2], 1e-5)
+  assert len( glueEE1 ) == 4 and glueEE1[0].GetShapeType() == GEOM.EDGE
+  glueEE2 = geompy.GetGlueEdges( [comp], 1e-5)
+  assert len( glueEE2 ) == 4 and glueEE2[0].GetShapeType() == GEOM.EDGE
+  glueEE3 = geompy.GetGlueEdges( comp, 1e-5)
+  assert len( glueEE3 ) == 4 and glueEE3[0].GetShapeType() == GEOM.EDGE
+
+  #check MakeGlueEdgesByList()
+  glueEL1 = geompy.MakeGlueEdgesByList( [box1,box2], 1e-5, glueEE1)
+  assert glueEL1.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfEdges( glueEL1 ) == geompy.NumberOfEdges( comp ) - 4
+  glueEL2 = geompy.MakeGlueEdgesByList( [comp], 1e-5, glueEE2)
+  assert glueEL2.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfEdges( glueEL2 ) == geompy.NumberOfEdges( comp ) - 4
+  glueEL3 = geompy.MakeGlueEdgesByList( comp, 1e-5, glueEE3 )
+  assert glueEL3.GetShapeType() == GEOM.COMPOUND
+  assert geompy.NumberOfEdges( glueEL3 ) == geompy.NumberOfEdges( comp ) - 4
+
+  # check GetSharedShapesMulti()
+  sharedEE = geompy.GetSharedShapesMulti( glueEL3, geompy.ShapeType["EDGE"])
+  assert len( sharedEE ) == 4
+  assert sharedEE[0].GetShapeType() == GEOM.EDGE
+  assert sharedEE[1].GetShapeType() == GEOM.EDGE
+  assert sharedEE[2].GetShapeType() == GEOM.EDGE
+  assert sharedEE[3].GetShapeType() == GEOM.EDGE
+
+  return
+
 def TestHealingOperations (geompy, math):
 
+  TestSewGluing(geompy)
   TestMakeSewing(geompy, math)
   TestDivideEdge(geompy)
   TestSuppressHoles(geompy)
@@ -330,3 +527,5 @@ def TestHealingOperations (geompy, math):
   TestCloseContour(geompy)
   TestSuppressFaces(geompy)
   TestProcessShape(geompy)
+  TestFuseEdges(geompy)
+  TestRemoveWebs(geompy)