src/GEOM_SWIG/GEOM_TestHealing.py

   1 #  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
   2 #
   3 #  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 #  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 #
   6 #  This library is free software; you can redistribute it and/or
   7 #  modify it under the terms of the GNU Lesser General Public
   8 #  License as published by the Free Software Foundation; either
   9 #  version 2.1 of the License.
  10 #
  11 #  This library is distributed in the hope that it will be useful,
  12 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 #  Lesser General Public License for more details.
  15 #
  16 #  You should have received a copy of the GNU Lesser General Public
  17 #  License along with this library; if not, write to the Free Software
  18 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 #
  20 #  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 #
  22 #  GEOM GEOM_SWIG : binding of C++ implementaion with Python
  23 #  File   : GEOM_TestHealing.py
  24 #  Author : Julia DOROVSKIKH
  25 #  Module : GEOM
  26 #  $Header$
  27 #
  28 def TestProcessShape (geompy):
  29
  30   ##Load shape from BREP file
  31   #import os
  32   #datadir = os.getenv("GEOM_TEST")
  33   #if datadir != 0:
  34   #  datadir = datadir + "/Resources/"
  35   #else:
  36   #  "Please, define GEOM_TEST variable !"
  37   #
  38   #print "Import ", datadir + "aze2.brep"
  39   #Shape = batchmode_geompy.Import(datadir + "aze2.brep", "BREP")
  40
  41   p1 = geompy.MakeVertex(0,0,0)
  42   p2 = geompy.MakeVertex(200,0,0)
  43   p3 = geompy.MakeVertex(100,150,0)
  44
  45   edge = geompy.MakeEdge(p1,p2)
  46   arc  = geompy.MakeArc(p1,p3,p2)
  47   wire = geompy.MakeWire([edge,arc])
  48   face = geompy.MakeFace(wire, 1)
  49
  50   theShape = geompy.MakePrismVecH(face, edge, 130)
  51
  52   #Check shape
  53   print "Before ProcessShape:"
  54   isValid = geompy.CheckShape(theShape)
  55   if isValid == 0:
  56     print "The shape is not valid"
  57   else:
  58     print "The shape seems to be valid"
  59
  60   #Process Shape
  61   Operators = ["FixShape"]
  62   Parameters = ["FixShape.Tolerance3d"]
  63   Values = ["1e-7"]
  64
  65   PS = geompy.ProcessShape(theShape, Operators, Parameters, Values)
  66
  67   #Check shape
  68   print "After ProcessShape:"
  69   isValid = geompy.CheckShape(PS)
  70   if isValid == 0:
  71     print "The shape is not valid"
  72     raise RuntimeError, "It seems, that the ProcessShape() has failed"
  73   else:
  74     print "The shape seems to be valid"
  75
  76   #Add In Study
  77   Id_Shape = geompy.addToStudy(theShape, "Invalid Shape")
  78   Id_PS    = geompy.addToStudy(PS, "Processed Shape")
  79
  80 def TestSuppressFaces (geompy):
  81
  82   #Create base geometry 3D
  83   Box = geompy.MakeBoxDXDYDZ(200., 200., 200.)
  84
  85   #IDList for SuppHole
  86   faces = []
  87   faces = geompy.SubShapeAllSorted(Box, geompy.ShapeType["FACE"])
  88
  89   f_glob_id = geompy.GetSubShapeID(Box, faces[5])
  90
  91   #Transform objects
  92   SuppFace = geompy.SuppressFaces(Box, [f_glob_id])
  93
  94   #Add In Study
  95   Id_SuppFace = geompy.addToStudy(SuppFace, "SuppFace")
  96
  97 def TestSuppressInternalWires (geompy):
  98
  99   #Create Face with hole
 100   p11 = geompy.MakeVertex( 0,  0, 0)
 101   p12 = geompy.MakeVertex(30,  0, 0)
 102   p13 = geompy.MakeVertex(30, 30, 0)
 103   p14 = geompy.MakeVertex( 0, 30, 0)
 104
 105   p21 = geompy.MakeVertex(10, 10, 0)
 106   p22 = geompy.MakeVertex(20, 10, 0)
 107   p23 = geompy.MakeVertex(20, 20, 0)
 108   p24 = geompy.MakeVertex(10, 20, 0)
 109
 110   e11 = geompy.MakeEdge(p11, p12)
 111   e12 = geompy.MakeEdge(p12, p13)
 112   e13 = geompy.MakeEdge(p13, p14)
 113   e14 = geompy.MakeEdge(p14, p11)
 114
 115   e21 = geompy.MakeEdge(p21, p22)
 116   e22 = geompy.MakeEdge(p22, p23)
 117   e23 = geompy.MakeEdge(p23, p24)
 118   e24 = geompy.MakeEdge(p24, p21)
 119
 120   w1 = geompy.MakeWire([e11, e12, e13, e14])
 121   w2 = geompy.MakeWire([e21, e22, e23, e24])
 122
 123   id_w1 = geompy.addToStudy(w1, "Outside Wire")
 124   id_w2 = geompy.addToStudy(w2, "Inside Wire")
 125
 126   f12 = geompy.MakeFaceWires([w1, w2], 0)
 127   id_f12 = geompy.addToStudy(f12, "Face WO + WI")
 128
 129   #Get Free Boundary
 130   Res1 = geompy.GetFreeBoundary(f12)
 131   isSuccess1   = Res1[0]
 132   ClosedWires1 = Res1[1]
 133   OpenWires1   = Res1[2]
 134   nbw1 = 0
 135
 136   for wire in ClosedWires1:
 137     nbw1 = nbw1 + 1
 138
 139   if nbw1 != 2:
 140     raise RuntimeError, "GetFreeBoundary(f12) must return 2 closed wires, but returned ", nbw1
 141
 142   #SuppressInternalWires
 143   face = geompy.SuppressInternalWires(f12, [])
 144
 145   #Get Free Boundary
 146   Res2 = geompy.GetFreeBoundary(face)
 147   isSuccess2   = Res2[0]
 148   ClosedWires2 = Res2[1]
 149   OpenWires2   = Res2[2]
 150   nbw2 = 0
 151
 152   for wire in ClosedWires2:
 153     nbw2 = nbw2 + 1
 154
 155   if nbw2 != 1:
 156     print "GetFreeBoundary(face) must return 1 closed wires, but returned ", nbw2
 157     raise RuntimeError, "SuppressInternalWires() works not correctly"
 158
 159   #Add In Study
 160   Id_face = geompy.addToStudy(face, "Face without internal wires")
 161
 162 def TestCloseContour (geompy):
 163
 164   ##Load shape from BREP file
 165   #import os
 166   #datadir = os.getenv("GEOM_TEST")
 167   #if datadir != 0:
 168   #  datadir = datadir + "/Resources/"
 169   #else:
 170   #  "Please, define GEOM_TEST variable !"
 171   #
 172   #print "Import ", datadir + "open_cont.brep"
 173   #Shape = geompy.Import(datadir + "open_cont.brep", "BREP")
 174
 175   p0   = geompy.MakeVertex(0.  , 0.  , 0.  )
 176   py   = geompy.MakeVertex(0.  , 100., 0.  )
 177   pz   = geompy.MakeVertex(0.  , 0.  , 100.)
 178   p200 = geompy.MakeVertex(200., 200., 200.)
 179
 180   Shape = geompy.MakePolyline([p0, pz, py, p200])
 181
 182   #Check shape
 183   print "Before closing contour:"
 184   isValid = geompy.CheckShape(Shape)
 185   if isValid == 0:
 186     print "The shape is not valid"
 187   else:
 188     print "The shape seems to be valid"
 189
 190   #Close Contour
 191   IsCommonVertex = 0 # false
 192
 193   shape_wires = geompy.SubShapeAll(Shape, geompy.ShapeType["WIRE"])
 194   Wires = []
 195   wi = 0
 196
 197   for wire in shape_wires:
 198     Wires.append(geompy.GetSubShapeID(Shape, shape_wires[wi]))
 199     wi = wi + 1
 200
 201   CC = geompy.CloseContour(Shape, Wires, IsCommonVertex)
 202
 203   #Check shape
 204   print "After closing contour:"
 205   isValid = geompy.CheckShape(CC)
 206   if isValid == 0:
 207     print "The shape is not valid"
 208     raise RuntimeError, "It seems, that the contour was not closed"
 209   else:
 210     print "The shape seems to be valid"
 211
 212   #Add In Study
 213   Id_Shape = geompy.addToStudy(Shape, "Shape with open wire")
 214   Id_CC    = geompy.addToStudy(CC, "Shape with closed wire")
 215
 216 def TestSuppressHoles (geompy):
 217
 218   #Create base Variables
 219   radius = 50.
 220   height = 300.
 221
 222   #Create base points
 223   p1 = geompy.MakeVertex(100., 100., 50.)
 224
 225   #Create base directions
 226   vz = geompy.MakeVectorDXDYDZ(0., 0., 100.)
 227
 228   #Create base geometry 3D
 229   Box      = geompy.MakeBoxDXDYDZ(200., 200., 200.)
 230   Cylinder = geompy.MakeCylinder(p1, vz, radius, height)
 231
 232   #Boolean (Cut)
 233   Cut = geompy.MakeBoolean(Box, Cylinder, 2)
 234   idCut = geompy.addToStudy(Cut, "CUT")
 235
 236   #IDList for SuppressFaces
 237   faces = []
 238   faces = geompy.SubShapeAllSorted(Cut, geompy.ShapeType["FACE"])
 239   ind = 0
 240   for face in faces:
 241       f_name = "FACE %d"%(ind)
 242       f_id = geompy.addToStudyInFather(Cut, face, f_name)
 243
 244       f_glob_id = geompy.GetSubShapeID(Cut, face)
 245       print "face ", ind, " global index = ", f_glob_id
 246       ind = ind + 1
 247
 248   f_glob_id_0 = geompy.GetSubShapeID(Cut, faces[0])
 249   cut_without_f_0 = geompy.SuppressFaces(Cut, [f_glob_id_0])
 250
 251   faces1 = []
 252   faces1 = geompy.SubShapeAllSorted(cut_without_f_0, geompy.ShapeType["FACE"])
 253   ind = 0
 254   for face in faces1:
 255       f_name = "FACE %d"%(ind)
 256       f_id = geompy.addToStudyInFather(cut_without_f_0, face, f_name)
 257
 258       f_glob_id = geompy.GetSubShapeID(cut_without_f_0, face)
 259       print "face ", ind, " global index = ", f_glob_id
 260       ind = ind + 1
 261
 262   f_glob_id_5 = geompy.GetSubShapeID(cut_without_f_0, faces1[5])
 263   cut_without_f_0_5 = geompy.SuppressFaces(cut_without_f_0, [f_glob_id_5])
 264   cut_without_f_0_5_id = geompy.addToStudy(cut_without_f_0_5, "Cut without faces 0 and 5")
 265
 266   #IDList for SuppHole
 267   wires = []
 268   wires = geompy.SubShapeAllSorted(cut_without_f_0_5, geompy.ShapeType["WIRE"])
 269   ind = 0
 270   for wire in wires:
 271       w_name = "WIRE %d"%(ind)
 272       w_id = geompy.addToStudyInFather(cut_without_f_0_5, wire, w_name)
 273
 274       w_glob_id = geompy.GetSubShapeID(cut_without_f_0_5, wire)
 275       print "wire ", ind, " global index = ", w_glob_id
 276       ind = ind + 1
 277
 278   w_3 = geompy.GetSubShapeID(cut_without_f_0_5, wires[3])
 279
 280   SuppHole3 = geompy.SuppressHoles(cut_without_f_0_5, [w_3])
 281   SuppHole3_id = geompy.addToStudy(SuppHole3, "Supp Hole 3")
 282
 283 def TestMakeSewing (geompy, math):
 284
 285   #Create base points
 286   px = geompy.MakeVertex(100., 0., 0.)
 287   py = geompy.MakeVertex(0., 100., 0.)
 288   pz = geompy.MakeVertex(0., 0., 100.)
 289
 290   #Create base geometry 2D & 3D
 291   Vector = geompy.MakeVector(px, py)
 292   Arc    = geompy.MakeArc(py, pz, px)
 293
 294   #Create base objects
 295   angle     = 45. * math.pi / 180
 296   WantPlanarFace = 1 #True
 297
 298   Wire = geompy.MakeWire([Vector, Arc])
 299   Face = geompy.MakeFace(Wire, WantPlanarFace)
 300   S    = geompy.MakeRotation(Face, Vector, angle)
 301
 302   #Make Sewing
 303   precision = 0.00001
 304   Sewing = geompy.MakeSewing([Face, S], precision)
 305
 306   #Add In Study
 307   id_Sewing = geompy.addToStudy(Sewing, "Sewing")
 308
 309 def TestDivideEdge (geompy):
 310
 311   #Create Box
 312   Box = geompy.MakeBoxDXDYDZ(200., 200., 200.)
 313
 314   #Divide Edge
 315   box_edges = geompy.SubShapeAllSorted(Box, geompy.ShapeType["EDGE"])
 316   edge_ind = geompy.GetSubShapeID(Box, box_edges[1])
 317
 318   Divide = geompy.DivideEdge(Box, edge_ind, 0.5, 1) # Obj, ind, param, is_curve_param
 319
 320   #Add In Study
 321   Id_Box    = geompy.addToStudy(Box, "Box")
 322   Id_Divide = geompy.addToStudy(Divide, "Box with Divided Edge")
 323
 324 def TestHealingOperations (geompy, math):
 325
 326   TestMakeSewing(geompy, math)
 327   TestDivideEdge(geompy)
 328   TestSuppressHoles(geompy)
 329   TestSuppressInternalWires(geompy)
 330   TestCloseContour(geompy)
 331   TestSuppressFaces(geompy)
 332   TestProcessShape(geompy)