src/GEOM_SWIG/GEOM_TestOthers.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
   3 #
   4 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   6 #
   7 # This library is free software; you can redistribute it and/or
   8 # modify it under the terms of the GNU Lesser General Public
   9 # License as published by the Free Software Foundation; either
  10 # version 2.1 of the License.
  11 #
  12 # This library is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 # Lesser General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU Lesser General Public
  18 # License along with this library; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  20 #
  21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  22 #
  23 #  File   : GEOM_TestOthers.py
  24 #  Author : Julia DOROVSKIKH
  25 #  Module : GEOM
  26 #  $Header$
  27 #
  28 # ! Please, if you edit this example file, update also
  29 # ! GEOM_SRC/doc/salome/gui/GEOM/input/tui_test_others.doc
  30 # ! as some sequences of symbols from this example are used during
  31 # ! documentation generation to identify certain places of this file
  32
  33 import os
  34
  35 def TestExportImport (geompy, shape):
  36
  37   print "Test Export/Import ...",
  38
  39   tmpDir = os.getenv("TEMP")
  40   if tmpDir == None:
  41     tmpDir = "/tmp"
  42
  43   # Files for Export/Import testing
  44   fileExportImport = tmpDir + "/testExportImport.brep"
  45   fileExportImportBREP = tmpDir + "/testExportImportBREP.brep"
  46   fileExportImportIGES = tmpDir + "/testExportImportIGES.iges"
  47   fileExportImportSTEP = tmpDir + "/testExportImportSTEP.step"
  48
  49   if os.access(fileExportImport, os.F_OK):
  50     if os.access(fileExportImport, os.W_OK):
  51       os.remove(fileExportImport)
  52     else:
  53       fileExportImport = tmpDir + "/testExportImport1.brep"
  54
  55     if os.access(fileExportImportBREP, os.W_OK):
  56       os.remove(fileExportImportBREP)
  57     else:
  58       fileExportImportBREP = tmpDir + "/testExportImportBREP1.brep"
  59
  60     if os.access(fileExportImportIGES, os.W_OK):
  61       os.remove(fileExportImportIGES)
  62     else:
  63       fileExportImportIGES = tmpDir + "/testExportImportIGES1.iges"
  64
  65     if os.access(fileExportImportSTEP, os.W_OK):
  66       os.remove(fileExportImportSTEP)
  67     else:
  68       fileExportImportSTEP = tmpDir + "/testExportImportSTEP1.step"
  69
  70   # Export
  71   geompy.Export(shape, fileExportImport, "BREP")
  72
  73   # ExportBREP, ExportIGES, ExportSTEP
  74   geompy.ExportBREP(shape, fileExportImportBREP)
  75   geompy.ExportIGES(shape, fileExportImportIGES)
  76   geompy.ExportSTEP(shape, fileExportImportSTEP)
  77
  78   # Import
  79   Import = geompy.ImportFile(fileExportImport, "BREP")
  80
  81   id_Import = geompy.addToStudy(Import, "Import")
  82
  83   # ImportBREP, ImportIGES, ImportSTEP
  84   ImportBREP = geompy.ImportBREP(fileExportImportBREP)
  85   ImportIGES = geompy.ImportIGES(fileExportImportIGES)
  86   ImportSTEP = geompy.ImportSTEP(fileExportImportSTEP)
  87
  88   id_ImportBREP = geompy.addToStudy(ImportBREP, "ImportBREP")
  89   id_ImportIGES = geompy.addToStudy(ImportIGES, "ImportIGES")
  90   id_ImportSTEP = geompy.addToStudy(ImportSTEP, "ImportSTEP")
  91
  92   # Remove files for Export/Import testing
  93   os.remove(fileExportImport)
  94   os.remove(fileExportImportBREP)
  95   os.remove(fileExportImportIGES)
  96   os.remove(fileExportImportSTEP)
  97
  98   print "OK"
  99
 100
 101 def TestOtherOperations (geompy, math):
 102
 103   # prepare data for further operations
 104   vx = geompy.MakeVectorDXDYDZ( 1,  0,  0)
 105   vy = geompy.MakeVectorDXDYDZ( 0,  1,  0)
 106   vz = geompy.MakeVectorDXDYDZ( 0,  0,  1)
 107
 108   v_y = geompy.MakeVectorDXDYDZ( 0, -1,  0)
 109
 110   p11 = geompy.MakeVertex( 0,  0, 0)
 111   p12 = geompy.MakeVertex(30,  0, 0)
 112   p13 = geompy.MakeVertex(30, 30, 0)
 113   p14 = geompy.MakeVertex( 0, 30, 0)
 114
 115   p21 = geompy.MakeVertex(10, 10, 0)
 116   p22 = geompy.MakeVertex(20, 10, 0)
 117   p23 = geompy.MakeVertex(20, 20, 0)
 118   p24 = geompy.MakeVertex(10, 20, 0)
 119
 120   e11 = geompy.MakeEdge(p11, p12)
 121   e12 = geompy.MakeEdge(p12, p13)
 122   e13 = geompy.MakeEdge(p13, p14)
 123   e14 = geompy.MakeEdge(p14, p11)
 124
 125   e21 = geompy.MakeEdge(p21, p22)
 126   e22 = geompy.MakeEdge(p22, p23)
 127   e23 = geompy.MakeEdge(p23, p24)
 128   e24 = geompy.MakeEdge(p24, p21)
 129
 130   w1 = geompy.MakeWire([e11, e12, e13, e14])
 131   w2 = geompy.MakeWire([e21, e22, e23, e24])
 132   w3 = geompy.MakeTranslation(w2, 0, 0, 10)
 133
 134   id_w1 = geompy.addToStudy(w1, "Outside Wire")
 135   id_w2 = geompy.addToStudy(w2, "Inside Wire")
 136   id_w3 = geompy.addToStudy(w3, "Inside Wire, translated along OZ")
 137
 138   # MakeFaces
 139   f12 = geompy.MakeFaces([w1, w2], 0)
 140   id_f12 = geompy.addToStudy(f12, "MakeFaces WO + WI")
 141
 142   # Export/Import
 143   TestExportImport(geompy, f12)
 144
 145   # OrientationChange
 146   Box = geompy.MakeBoxDXDYDZ(200, 200, 200)
 147   Orientation = geompy.OrientationChange(Box)
 148   id_Orientation = geompy.addToStudy(Orientation, "OrientationChange")
 149
 150   # MakeCommon, MakeCut, MakeFuse, MakeSection
 151   Sphere = geompy.MakeSphereR(100)
 152
 153   Common  = geompy.MakeCommon (Box, Sphere)
 154   Cut     = geompy.MakeCut    (Box, Sphere)
 155   Fuse    = geompy.MakeFuse   (Box, Sphere)
 156   Section = geompy.MakeSection(Box, Sphere)
 157
 158   id_Common  = geompy.addToStudy(Common,  "Common")
 159   id_Cut     = geompy.addToStudy(Cut,     "Cut")
 160   id_Fuse    = geompy.addToStudy(Fuse,    "Fuse")
 161   id_Section = geompy.addToStudy(Section, "Section")
 162
 163   # Partition
 164   p100 = geompy.MakeVertex(100, 100, 100)
 165   p300 = geompy.MakeVertex(300, 300, 300)
 166   Box1 = geompy.MakeBoxTwoPnt(p100, p300)
 167   #Partition = geompy.Partition([Box], [Box1], [], [Box])
 168   Partition = geompy.Partition([Box], [Box1])
 169   id_Partition = geompy.addToStudy(Partition, "Partition of Box by Box1")
 170
 171   # MakeMultiRotation1D, MakeMultiRotation2D
 172   pz = geompy.MakeVertex(0, 0, 100)
 173   vy = geompy.MakeVectorDXDYDZ(0, 100, 0)
 174
 175   MultiRot1D = geompy.MakeMultiRotation1D(f12, vy, pz, 6)
 176   MultiRot2D = geompy.MakeMultiRotation2D(f12, vy, pz, 45, 6, 30, 3)
 177
 178   id_MultiRot1D = geompy.addToStudy(MultiRot1D, "MakeMultiRotation1D")
 179   id_MultiRot2D = geompy.addToStudy(MultiRot2D, "MakeMultiRotation2D")
 180
 181   # MakeFilletAll
 182   radius_fillet = 10.
 183   face5 = geompy.SubShapeSortedCentres(Box, geompy.ShapeType["FACE"], [5])
 184   f_glob_id = geompy.GetSubShapeID(Box, face5)
 185   SuppFace = geompy.SuppressFaces(Box, [f_glob_id])
 186
 187   MakeFilletAll = geompy.MakeFilletAll(SuppFace, radius_fillet)
 188   id_MakeFilletAll = geompy.addToStudy(MakeFilletAll, "MakeFilletAll")
 189
 190   # MakeChamferAll
 191   dimension_chamfer = 10.
 192   MakeChamferAll = geompy.MakeChamferAll(SuppFace, dimension_chamfer)
 193   id_MakeChamferAll = geompy.addToStudy(MakeChamferAll, "MakeChamferAll")
 194
 195   # MakeChamfer
 196   d1 = 13.
 197   d2 = 7.
 198   box_faces = geompy.SubShapeAllSortedCentres(Box, geompy.ShapeType["FACE"])
 199   f_ind_1 = geompy.GetSubShapeID(Box, box_faces[0])
 200   f_ind_2 = geompy.GetSubShapeID(Box, box_faces[1])
 201   f_ind_3 = geompy.GetSubShapeID(Box, box_faces[2])
 202
 203   MakeChamfer = geompy.MakeChamfer(Box, d1, d2, geompy.ShapeType["FACE"],
 204                                    [f_ind_1, f_ind_2, f_ind_3])
 205   id_MakeChamfer = geompy.addToStudy(MakeChamfer, "MakeChamfer")
 206
 207   # NumberOf
 208   NumberOfFaces = geompy.NumberOfFaces(Box)
 209   if NumberOfFaces != 6:
 210     print "Bad number of faces in BOX!"
 211
 212   NumberOfEdges = geompy.NumberOfEdges(Box)
 213   if NumberOfEdges != 12:
 214     print "Bad number of edges in BOX!"
 215
 216   NumberOfSolids = geompy.NumberOfSolids(Box)
 217   if NumberOfSolids != 1:
 218     print "Bad number of solids in BOX!"
 219
 220   NumberOfShapes = geompy.NumberOfSubShapes(Box, geompy.ShapeType["SHAPE"])
 221   if NumberOfShapes != 34:
 222     print "Bad number of shapes in BOX!"
 223
 224   # MakeBlockExplode
 225   Compound = geompy.MakeCompound([Box, Sphere])
 226   MakeBlockExplode = geompy.MakeBlockExplode(Compound, 6, 6)
 227
 228   id_MakeBlockExplode = geompy.addToStudy(MakeBlockExplode[0], "MakeBlockExplode")
 229
 230   # CheckCompoundOfBlocks
 231   p1 = geompy.MakeVertex(200, 0, 0)
 232   p2 = geompy.MakeVertex(400, 200, 200)
 233   p3 = geompy.MakeVertex(400, 50, 50)
 234   p4 = geompy.MakeVertex(600, 250, 250)
 235
 236   Box2 = geompy.MakeBoxTwoPnt(p1, p2)
 237   Box3 = geompy.MakeBoxTwoPnt(p3, p4)
 238   Cyl  = geompy.MakeCylinderRH(50, 300)
 239   Cone = geompy.MakeConeR1R2H(150, 10, 400)
 240
 241   Compound1 = geompy.MakeCompound([Box, Cyl, Cone, Box3, Box2])
 242
 243   IsValid = geompy.CheckCompoundOfBlocks(Compound1)
 244   if IsValid == 0:
 245     print "The Blocks Compound is NOT VALID"
 246   else:
 247     print "The Blocks Compound is VALID"
 248
 249   IsValid = geompy.CheckCompoundOfBlocks(Box)
 250   if IsValid == 0:
 251     print "The Box is NOT VALID"
 252   else:
 253     print "The Box is VALID"
 254
 255   # GetSame
 256   Cone_ss = geompy.GetSame(Compound1, Cone)
 257   id_Cone_ss = geompy.addToStudyInFather(Compound1, Cone_ss, "Cone subshape")
 258
 259   # test geometrical groups
 260
 261   # CreateGroup
 262   CreateGroup = geompy.CreateGroup(Box, geompy.ShapeType["FACE"])
 263
 264   id_CreateGroup = geompy.addToStudy(CreateGroup, "CreateGroup")
 265
 266   # AddObject
 267   f_ind_4 = geompy.GetSubShapeID(Box, box_faces[3])
 268   f_ind_5 = geompy.GetSubShapeID(Box, box_faces[4])
 269   f_ind_6 = geompy.GetSubShapeID(Box, box_faces[5])
 270
 271   geompy.AddObject(CreateGroup, f_ind_6) # box_faces[5]
 272   geompy.AddObject(CreateGroup, f_ind_1) # box_faces[0]
 273   geompy.AddObject(CreateGroup, f_ind_4) # box_faces[3]
 274   # Now contains f_ind_6, f_ind_1, f_ind_4
 275
 276   # UnionList
 277   geompy.UnionList(CreateGroup, [box_faces[2], box_faces[4], box_faces[5]])
 278   # Now contains f_ind_6, f_ind_1, f_ind_4, f_ind_3, f_ind_5
 279
 280   # RemoveObject(theGroup, theSubShapeID)
 281   geompy.RemoveObject(CreateGroup, f_ind_1) # box_faces[0]
 282   # Now contains f_ind_6, f_ind_4, f_ind_3, f_ind_5
 283
 284   # DifferenceList
 285   geompy.DifferenceList(CreateGroup, [box_faces[1], box_faces[0], box_faces[3]])
 286   # Now contains f_ind_6, f_ind_3, f_ind_5
 287
 288   # GetObjectIDs
 289   GetObjectIDs = geompy.GetObjectIDs(CreateGroup)
 290
 291   print "Group of Box's faces includes the following IDs:"
 292   print "(must be ", f_ind_6, ", ", f_ind_3, " and ", f_ind_5, ")"
 293   for ObjectID in GetObjectIDs:
 294     print " ", ObjectID
 295
 296   # GetMainShape
 297   BoxCopy = geompy.GetMainShape(CreateGroup)
 298
 299   # DifferenceIDs
 300   geompy.DifferenceIDs(CreateGroup, [f_ind_3, f_ind_5])
 301   # Now contains f_ind_6
 302
 303   # UnionIDs
 304   geompy.UnionIDs(CreateGroup, [f_ind_1, f_ind_2, f_ind_6])
 305   # Now contains f_ind_6, f_ind_1, f_ind_2
 306
 307   # Check
 308   GetObjectIDs = geompy.GetObjectIDs(CreateGroup)
 309   print "Group of Box's faces includes the following IDs:"
 310   print "(must be ", f_ind_6, ", ", f_ind_1, " and ", f_ind_2, ")"
 311   for ObjectID in GetObjectIDs:
 312     print " ", ObjectID
 313
 314   # -----------------------------------------------------------------------------
 315   # enumeration ShapeTypeString as a dictionary
 316   # -----------------------------------------------------------------------------
 317   ShapeTypeString = {'0':"COMPOUND", '1':"COMPSOLID", '2':"SOLID", '3':"SHELL",
 318                      '4':"FACE", '5':"WIRE", '6':"EDGE", '7':"VERTEX", '8':"SHAPE"}
 319
 320   GroupType = geompy.GetType(CreateGroup)
 321   print "Type of elements of the created group is ", ShapeTypeString[`GroupType`]
 322
 323   # Prepare data for the following operations
 324   p0 = geompy.MakeVertex(0, 0, 0)
 325   b0 = geompy.MakeBox(-50, -50, -50, 50, 50, 50)
 326   s0 = geompy.MakeSphereR(100)
 327
 328   id_b0 = geompy.addToStudy(b0, "b0")
 329   id_s0 = geompy.addToStudy(s0, "s0")
 330
 331   v_0pp = geompy.MakeVectorDXDYDZ( 0,  1,  1)
 332   #v_0np = geompy.MakeVectorDXDYDZ( 0, -1,  1)
 333   v_p0p = geompy.MakeVectorDXDYDZ( 1,  0,  1)
 334   v_p0n = geompy.MakeVectorDXDYDZ(1,  0,  -1)
 335   v_pp0 = geompy.MakeVectorDXDYDZ( 1,  1,  0)
 336   v_pn0 = geompy.MakeVectorDXDYDZ(1,  -1,  0)
 337
 338   #pln_0pp = geompy.MakePlane(p0, v_0pp, 300)
 339   #pln_0np = geompy.MakePlane(p0, v_0np, 300)
 340   pln_p0p = geompy.MakePlane(p0, v_p0p, 300)
 341   pln_p0n = geompy.MakePlane(p0, v_p0n, 300)
 342   pln_pp0 = geompy.MakePlane(p0, v_pp0, 300)
 343   pln_pn0 = geompy.MakePlane(p0, v_pn0, 300)
 344   #
 345   #part_objs = [b0, pln_0pp, pln_0np, pln_p0p, pln_n0p, pln_pp0, pln_np0]
 346   #part_tool_1 = geompy.MakePartition(part_objs, [], [], [b0])
 347   #part_tool_1 = geompy.MakePartition(part_objs)
 348   #
 349   #id_part_tool_1 = geompy.addToStudy(part_tool_1, "part_tool_1")
 350   #
 351   #pt_pnt_1  = geompy.MakeVertex( 55,   0,  55)
 352   #pt_pnt_2  = geompy.MakeVertex(  0,  55,  55)
 353   #pt_pnt_3  = geompy.MakeVertex(-55,   0,  55)
 354   #pt_pnt_4  = geompy.MakeVertex(  0, -55,  55)
 355   #pt_pnt_5  = geompy.MakeVertex( 55,  55,   0)
 356   #pt_pnt_6  = geompy.MakeVertex( 55, -55,   0)
 357   #pt_pnt_7  = geompy.MakeVertex(-55,  55,   0)
 358   #pt_pnt_8  = geompy.MakeVertex(-55, -55,   0)
 359   #pt_pnt_9  = geompy.MakeVertex( 55,   0, -55)
 360   #pt_pnt_10 = geompy.MakeVertex(  0,  55, -55)
 361   #pt_pnt_11 = geompy.MakeVertex(-55,   0, -55)
 362   #pt_pnt_12 = geompy.MakeVertex(  0, -55, -55)
 363   #
 364   #pt_face_1  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_1)
 365   #pt_face_2  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_2)
 366   #pt_face_3  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_3)
 367   #pt_face_4  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_4)
 368   #pt_face_5  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_5)
 369   #pt_face_6  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_6)
 370   #pt_face_7  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_7)
 371   #pt_face_8  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_8)
 372   #pt_face_9  = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_9)
 373   #pt_face_10 = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_10)
 374   #pt_face_11 = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_11)
 375   #pt_face_12 = geompy.GetFaceNearPoint(part_tool_1, pt_pnt_12)
 376   #
 377   #pt_box = geompy.GetBlockNearPoint(part_tool_1, p0)
 378   #
 379   #comp_parts = [pt_face_1, pt_face_4, pt_face_7, pt_face_10,
 380   #              pt_face_2, pt_face_5, pt_face_8, pt_face_11,
 381   #              #pt_face_3, pt_face_6, pt_face_9, pt_face_12, pt_box]
 382   #              pt_face_3, pt_face_6, pt_face_9, pt_face_12]
 383   #part_tool = geompy.MakeCompound(comp_parts)
 384   #id_part_tool = geompy.addToStudy(part_tool, "part_tool")
 385   #
 386   #part = geompy.MakePartition([s0], [part_tool])
 387   #
 388   #part_tools = [pt_face_1, pt_face_4, pt_face_7, pt_face_10,
 389   #              pt_face_2, pt_face_5, pt_face_8, pt_face_11,
 390   #              pt_face_3, pt_face_6, pt_face_9, pt_face_12, b0]
 391   #part = geompy.MakePartition([s0], part_tools)
 392
 393   p1 = geompy.MakeVertex(50, 0, 0)
 394   p2 = geompy.MakeVertex(-50, 0, 0)
 395   p3 = geompy.MakeVertex(0, 50, 0)
 396   p4 = geompy.MakeVertex(0, -50, 0)
 397   p5 = geompy.MakeVertex(0, 0, 50)
 398   p6 = geompy.MakeVertex(0, 0, -50)
 399
 400   plnX1 = geompy.MakePlane(p1, vx, 300)
 401   plnX2 = geompy.MakePlane(p2, vx, 300)
 402   plnY1 = geompy.MakePlane(p3, vy, 300)
 403   plnY2 = geompy.MakePlane(p4, vy, 300)
 404   plnZ1 = geompy.MakePlane(p5, vz, 300)
 405   plnZ2 = geompy.MakePlane(p6, vz, 300)
 406
 407   #part = geompy.MakePartition([s0], [plnX1,plnX2,plnY1,plnY2,plnZ1,plnZ2])
 408   part = geompy.MakePartition([s0], [plnX1])
 409   part = geompy.MakePartition([part], [plnX2])
 410   part = geompy.MakePartition([part], [plnY1])
 411   part = geompy.MakePartition([part], [plnY2])
 412   part = geompy.MakePartition([part], [plnZ1])
 413   part = geompy.MakePartition([part], [plnZ2])
 414   geompy.addToStudy(part, "part")
 415
 416   # GetFreeFacesIDs
 417   anIDs = geompy.GetFreeFacesIDs(part)
 418   freeFaces = geompy.GetSubShape(part, anIDs)
 419
 420   geompy.addToStudy(freeFaces, "freeFaces")
 421
 422   # RemoveExtraEdges with union of all faces, sharing common surfaces
 423   tools = [pln_pp0, pln_pn0, pln_p0p, pln_p0n]
 424
 425   Partition_1 = geompy.MakePartition([Sphere], tools, [], [], geompy.ShapeType["SOLID"], 0, [])
 426   geompy.addToStudy(Partition_1, "Partition_1")
 427
 428   faces = geompy.SubShapeAllSortedCentres(Partition_1, geompy.ShapeType["FACE"])
 429
 430   Face_1 = faces[0]
 431   Face_2 = faces[39]
 432   Face_3 = faces[40]
 433
 434   geompy.addToStudyInFather(Partition_1, Face_1, "Face_1")
 435   geompy.addToStudyInFather(Partition_1, Face_2, "Face_2")
 436   geompy.addToStudyInFather(Partition_1, Face_3, "Face_3")
 437
 438   Vector_5 = geompy.MakeVectorDXDYDZ(0, 20, 0)
 439   geompy.addToStudy(Vector_5, "Vector_5")
 440
 441   Rotation_1 = geompy.MakeRotation(Face_1, Vector_5, 90*math.pi/180.0)
 442   Rotation_2 = geompy.MakeRotation(Face_1, Vector_5, 180*math.pi/180.0)
 443   Rotation_3 = geompy.MakeRotation(Face_1, Vector_5, 270*math.pi/180.0)
 444
 445   geompy.addToStudy(Rotation_1, "Rotation_1")
 446   geompy.addToStudy(Rotation_2, "Rotation_2")
 447   geompy.addToStudy(Rotation_3, "Rotation_3")
 448
 449   Vector_6 = geompy.MakeVectorDXDYDZ(0, 0, 20)
 450   geompy.addToStudy(Vector_6, "Vector_6")
 451
 452   Rotation_4 = geompy.MakeRotation(Face_1, Vector_6, 90*math.pi/180.0)
 453   Rotation_5 = geompy.MakeRotation(Face_1, Vector_6, -90*math.pi/180.0)
 454   geompy.addToStudy(Rotation_4, "Rotation_4")
 455   geompy.addToStudy(Rotation_5, "Rotation_5")
 456
 457   Shell_1 = geompy.MakeShell([Face_1, Rotation_1, Rotation_2, Rotation_3, Rotation_4, Rotation_5])
 458   Solid_1 = geompy.MakeSolid([Shell_1])
 459   NoExtraEdges_1 = geompy.RemoveExtraEdges(Solid_1, True) # doUnionFaces = True
 460
 461   geompy.addToStudy(Shell_1, "Shell_1")
 462   geompy.addToStudy(Solid_1, "Solid_1")
 463   geompy.addToStudy(NoExtraEdges_1, "NoExtraEdges_1")
 464
 465   # RemoveExtraEdges (by default, doUnionFaces = False)
 466   freeFacesWithoutExtra = geompy.RemoveExtraEdges(freeFaces)
 467
 468   geompy.addToStudy(freeFacesWithoutExtra, "freeFacesWithoutExtra")
 469
 470   # GetSharedShapes
 471   sharedFaces = geompy.GetSharedShapes(part, freeFaces,
 472                                        geompy.ShapeType["FACE"])
 473   ind = 1
 474   for shFace in sharedFaces:
 475     geompy.addToStudy(shFace, "sharedFace_" + `ind`)
 476     ind = ind + 1
 477     pass
 478
 479   sharedEdges = geompy.GetSharedShapesMulti([part, freeFaces],
 480                                              geompy.ShapeType["EDGE"])
 481   ind = 1
 482   for shEdge in sharedEdges:
 483     geompy.addToStudy(shEdge, "sharedEdge_" + `ind`)
 484     ind = ind + 1
 485     pass
 486
 487   # CheckAndImprove
 488   blocksComp = geompy.CheckAndImprove(part)
 489
 490   geompy.addToStudy(blocksComp, "blocksComp")
 491
 492   # Propagate
 493   listChains = geompy.Propagate(blocksComp)
 494
 495   for chain in listChains:
 496     geompy.addToStudyInFather(blocksComp, chain, "propagation chain")
 497
 498   # GetPoint(theShape, theX, theY, theZ, theEpsilon)
 499   #
 500   # (-50,  50, 50) .-----. (50,  50, 50)
 501   #      pb0_top_1 |     |
 502   #                |     . pmidle
 503   #                |     |
 504   # (-50, -50, 50) '-----' (50, -50, 50)
 505   #
 506   pb0_top_1 = geompy.GetPoint(blocksComp, -50,  50,  50, 0.01)
 507   pb0_bot_1 = geompy.GetPoint(blocksComp, -50, -50, -50, 0.01)
 508
 509   geompy.addToStudyInFather(blocksComp, pb0_top_1, "point from blocksComp (-50,  50,  50)")
 510   geompy.addToStudyInFather(blocksComp, pb0_bot_1, "point from blocksComp (-50, -50, -50)")
 511
 512   # GetVertexNearPoint(theShape, thePoint)
 513   pb0_top_2_near = geompy.MakeVertex(40, 40, 40)
 514   pb0_top_2      = geompy.GetVertexNearPoint(blocksComp, pb0_top_2_near)
 515
 516   geompy.addToStudyInFather(blocksComp, pb0_top_2, "point from blocksComp near (40,  40,  40)")
 517
 518   # GetEdge(theShape, thePoint1, thePoint2)
 519   edge_top_y50 = geompy.GetEdge(blocksComp, pb0_top_1, pb0_top_2)
 520
 521   geompy.addToStudyInFather(blocksComp, edge_top_y50, "edge from blocksComp by two points")
 522
 523   # GetEdgeNearPoint(theShape, thePoint)
 524   pmidle = geompy.MakeVertex(50, 0, 50)
 525   edge1 = geompy.GetEdgeNearPoint(blocksComp, pmidle)
 526
 527   geompy.addToStudyInFather(blocksComp, edge1, "edge near point (50, 0, 50)")
 528
 529   # GetBlockByParts(theCompound, theParts)
 530   b0_image = geompy.GetBlockByParts(blocksComp, [pb0_top_1, pb0_bot_1, edge1])
 531
 532   geompy.addToStudyInFather(blocksComp, b0_image, "b0 image")
 533
 534   # GetShapesNearPoint(theShape, thePoint, theShapeType, theTolerance)
 535   b0_faces_plus = geompy.GetShapesNearPoint(blocksComp, pb0_top_2_near, geompy.ShapeType["FACE"], 0.01)
 536
 537   geompy.addToStudyInFather(blocksComp, b0_faces_plus, "faces near point (40,  40,  40)")
 538
 539   # GetShapesOnPlane
 540   faces_on_pln = geompy.GetShapesOnPlane(blocksComp, geompy.ShapeType["FACE"],
 541                                          v_0pp, geompy.GEOM.ST_ONIN)
 542   for face_i in faces_on_pln:
 543     geompy.addToStudy(face_i, "Face on Plane (N = (0, 1, 1)) or below it")
 544
 545   # GetShapesOnPlaneIDs
 546   faces_above_pln_ids = geompy.GetShapesOnPlaneIDs(blocksComp, geompy.ShapeType["FACE"],
 547                                                    v_0pp, geompy.GEOM.ST_OUT)
 548   faces_above = geompy.CreateGroup(blocksComp, geompy.ShapeType["FACE"])
 549   geompy.UnionIDs(faces_above, faces_above_pln_ids)
 550   geompy.addToStudy(faces_above, "Group of faces above Plane (N = (0, 1, 1))")
 551
 552   # GetShapesOnPlaneWithLocation
 553   Loc = geompy.MakeVertex(0, -50, 0)
 554   edges_on_pln = geompy.GetShapesOnPlaneWithLocation(blocksComp, geompy.ShapeType["EDGE"],
 555                                                      v_y, Loc, geompy.GEOM.ST_ON)
 556   for edge_i in edges_on_pln:
 557     geompy.addToStudy(edge_i, "Edge on Plane (N = (0, -1, 0) & Location = (0, -50, 0)")
 558
 559   # GetShapesOnPlaneWithLocationIDs
 560   edges_on_pln_ids = geompy.GetShapesOnPlaneWithLocationIDs(
 561            blocksComp, geompy.ShapeType["EDGE"], v_y, Loc, geompy.GEOM.ST_ON)
 562   group_edges_on_pln = geompy.CreateGroup(blocksComp, geompy.ShapeType["EDGE"])
 563   geompy.UnionIDs(group_edges_on_pln, edges_on_pln_ids)
 564   grname = "Group of edges on Plane (N = (0, -1, 0) & Location = (0, -50, 0))"
 565   geompy.addToStudy(group_edges_on_pln, grname)
 566
 567   # GetShapesOnCylinder
 568   edges_out_cyl = geompy.GetShapesOnCylinder(blocksComp, geompy.ShapeType["EDGE"],
 569                                              vy, 55, geompy.GEOM.ST_OUT)
 570   for edge_i in edges_out_cyl:
 571     geompy.addToStudy(edge_i, "Edge out of Cylinder (axis = (0, 1, 0), r = 55)")
 572
 573   # GetShapesOnCylinderIDs
 574   edges_in_cyl_ids = geompy.GetShapesOnCylinderIDs(blocksComp, geompy.ShapeType["EDGE"],
 575                                                    vy, 80, geompy.GEOM.ST_IN)
 576   edges_in = geompy.CreateGroup(blocksComp, geompy.ShapeType["EDGE"])
 577   geompy.UnionIDs(edges_in, edges_in_cyl_ids)
 578   geompy.addToStudy(edges_in, "Group of edges inside Cylinder (axis = (0, 1, 0), r = 55)")
 579
 580   # GetShapesOnCylinderWithLocation
 581   edges_out_cyl = geompy.GetShapesOnCylinderWithLocation(blocksComp, geompy.ShapeType["EDGE"],
 582                                                          vy, p11, 55, geompy.GEOM.ST_OUT)
 583   for edge_i in edges_out_cyl:
 584     geompy.addToStudy(edge_i, "Edge out of Cylinder (axis = (0, 1, 0),  loc = (0, 0, 0), r = 55)")
 585
 586   # GetShapesOnCylinderWithLocationIDs
 587   edges_in_cyl_ids = geompy.GetShapesOnCylinderWithLocationIDs(blocksComp, geompy.ShapeType["EDGE"],
 588                                                                vy, p11, 80, geompy.GEOM.ST_IN)
 589   edges_in = geompy.CreateGroup(blocksComp, geompy.ShapeType["EDGE"])
 590   geompy.UnionIDs(edges_in, edges_in_cyl_ids)
 591   geompy.addToStudy(edges_in, "Group of edges inside Cylinder (axis = (0, 1, 0), loc = (0, 0, 0), r = 80)")
 592
 593   # GetShapesOnSphere
 594   vertices_on_sph = geompy.GetShapesOnSphere(blocksComp, geompy.ShapeType["VERTEX"],
 595                                              p0, 100, geompy.GEOM.ST_ON)
 596   for vertex_i in vertices_on_sph:
 597     geompy.addToStudy(vertex_i, "Vertex on Sphere (center = (0, 0, 0), r = 100)")
 598
 599   # GetShapesOnSphereIDs
 600   vertices_on_sph_ids = geompy.GetShapesOnSphereIDs(blocksComp, geompy.ShapeType["VERTEX"],
 601                                                     p0, 100, geompy.GEOM.ST_ON)
 602   vertices_on = geompy.CreateGroup(blocksComp, geompy.ShapeType["VERTEX"])
 603   geompy.UnionIDs(vertices_on, vertices_on_sph_ids)
 604   geompy.addToStudy(vertices_on, "Group of vertices on Sphere (center = (0, 0, 0), r = 100)")
 605
 606   # GetShapesOnQuadrangle
 607
 608   geompy.addToStudy(f12, "F12" )
 609
 610   bl = geompy.MakeVertex(10,-10, 0)
 611   br = geompy.MakeVertex(40,-10, 0)
 612   tr = geompy.MakeVertex(40, 20, 0)
 613   tl = geompy.MakeVertex(10, 20, 0)
 614   qe1 = geompy.MakeEdge(bl, br)
 615   qe2 = geompy.MakeEdge(br, tr)
 616   qe3 = geompy.MakeEdge(tr, tl)
 617   qe4 = geompy.MakeEdge(tl, bl)
 618   quadrangle = geompy.MakeWire([qe1, qe2, qe3, qe4])
 619   geompy.addToStudy(quadrangle, "Quadrangle")
 620
 621   edges_onin_quad = geompy.GetShapesOnQuadrangle(f12, geompy.ShapeType["EDGE"],
 622                                                  tl, tr, bl, br, geompy.GEOM.ST_ONIN)
 623   comp = geompy.MakeCompound(edges_onin_quad)
 624   geompy.addToStudy(comp, "Edges of F12 ONIN Quadrangle")
 625   if len( edges_onin_quad ) != 4:
 626     print "Error in GetShapesOnQuadrangle()"
 627
 628   # GetShapesOnQuadrangleIDs
 629   vertices_on_quad_ids = geompy.GetShapesOnQuadrangleIDs(f12, geompy.ShapeType["VERTEX"],
 630                                                          tl, tr, bl, br, geompy.GEOM.ST_ON)
 631   vertices_on_quad = geompy.CreateGroup(f12, geompy.ShapeType["VERTEX"])
 632   geompy.UnionIDs(vertices_on_quad, vertices_on_quad_ids)
 633   geompy.addToStudy(vertices_on_quad, "Group of vertices on Quadrangle F12")
 634
 635   # GetShapesOnBox
 636   edges_on_box = geompy.GetShapesOnBox(b0, part, geompy.ShapeType["EDGE"],
 637                                        geompy.GEOM.ST_ON)
 638   comp = geompy.MakeCompound(edges_on_box)
 639   geompy.addToStudy(comp, "Edges of part ON box b0")
 640   if len( edges_on_box ) != 12:
 641     print "Error in GetShapesOnBox()"
 642
 643   # GetShapesOnBoxIDs
 644   faces_on_box_ids = geompy.GetShapesOnBoxIDs(b0, part, geompy.ShapeType["FACE"],
 645                                               geompy.GEOM.ST_ON)
 646   faces_on_box = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
 647   geompy.UnionIDs(faces_on_box, faces_on_box_ids)
 648   geompy.addToStudyInFather(part, faces_on_box, "Group of faces on box b0")
 649
 650   # Prepare arguments for GetShapesOnShape
 651   sph1 = geompy.MakeSphere(50, 50,  50, 40)
 652   sph2 = geompy.MakeSphere(50, 50, -50, 40)
 653   pcyl = geompy.MakeVertex(50, 50, -50)
 654   cyli = geompy.MakeCylinder(pcyl, vz, 40, 100)
 655   fuse = geompy.MakeFuse(sph1, cyli)
 656   sh_1 = geompy.MakeFuse(fuse, sph2)
 657   # As after Fuse we have a compound, we need to obtain a solid from it
 658   #shsh = geompy.SubShapeAll(fuse, geompy.ShapeType["SOLID"])
 659   #sh_1 = shsh[0]
 660   geompy.addToStudy(sh_1, "sh_1")
 661
 662   # GetShapesOnShape
 663   faces_in_sh = geompy.GetShapesOnShape(sh_1, part, geompy.ShapeType["FACE"],
 664                                         geompy.GEOM.ST_IN)
 665   comp = geompy.MakeCompound(faces_in_sh)
 666   geompy.addToStudy(comp, "Faces of part IN shape sh_1")
 667   if len(faces_in_sh) != 11:
 668     print "Error in GetShapesOnShape()"
 669
 670   # GetShapesOnShapeAsCompound
 671   faces_in_sh_c = geompy.GetShapesOnShapeAsCompound(sh_1, part, geompy.ShapeType["FACE"],
 672                                                     geompy.GEOM.ST_IN)
 673   geompy.addToStudy(faces_in_sh_c, "Faces of part IN shape sh_1 (as compound)")
 674
 675   # GetShapesOnShapeIDs
 676   edges_in_sh_ids = geompy.GetShapesOnShapeIDs(sh_1, part, geompy.ShapeType["EDGE"],
 677                                                geompy.GEOM.ST_IN)
 678   edges_in_sh = geompy.CreateGroup(part, geompy.ShapeType["EDGE"])
 679   geompy.UnionIDs(edges_in_sh, edges_in_sh_ids)
 680   geompy.addToStudyInFather(part, edges_in_sh, "Group of edges in shape sh_1")
 681   if len(edges_in_sh_ids) != 15:
 682     print "Error in GetShapesOnShapeIDs()"
 683
 684   # Prepare arguments for GetInPlace and GetInPlaceByHistory
 685   box5 = geompy.MakeBoxDXDYDZ(100, 100, 100)
 686   box6 = geompy.MakeTranslation(box5, 50, 50, 0)
 687
 688   geompy.addToStudy(box5, "Box 5")
 689   geompy.addToStudy(box6, "Box 6")
 690
 691   part = geompy.MakePartition([box5], [box6])
 692   geompy.addToStudy(part, "Partitioned")
 693
 694   box5_faces = geompy.SubShapeAll(box5, geompy.ShapeType["FACE"])
 695   box6_faces = geompy.SubShapeAll(box6, geompy.ShapeType["FACE"])
 696
 697   for ifa in range(6):
 698     geompy.addToStudyInFather(box5, box5_faces[ifa], "Face" + `ifa + 1`)
 699     geompy.addToStudyInFather(box6, box6_faces[ifa], "Face" + `ifa + 1`)
 700
 701   # GetInPlace(theShapeWhere, theShapeWhat)
 702   ibb = 5
 703   faces_list = [box5_faces, box6_faces]
 704   for afaces in faces_list:
 705     ifa = 1
 706     for aface in afaces:
 707       refl_box_face = geompy.GetInPlace(part, aface)
 708       if ibb == 6 and (ifa == 2 or ifa == 4):
 709         # For two faces of the tool box
 710         # there is no reflection in the result.
 711         if refl_box_face is not None:
 712           error = "Result of GetInPlace must be NULL for face "
 713           error += `ifa` + " of box " + `ibb`
 714           raise RuntimeError, error
 715       else:
 716         ssname = "Reflection of face " + `ifa` + " of box " + `ibb`
 717         geompy.addToStudyInFather(part, refl_box_face, ssname)
 718       ifa = ifa + 1
 719     ibb = ibb + 1
 720
 721   # GetInPlaceByHistory(theShapeWhere, theShapeWhat)
 722   part = geompy.MakePartition([box5], [box6])
 723   geompy.addToStudy(part, "Partitioned")
 724
 725   ibb = 5
 726   faces_list = [box5_faces, box6_faces]
 727   for afaces in faces_list:
 728     ifa = 1
 729     for aface in afaces:
 730       ssname = "Reflection of face " + `ifa` + " of box " + `ibb` + " (by history)"
 731       if ibb == 6 and (ifa == 2 or ifa == 4):
 732         # use IDL interface directly to avoid error message appearence in Python console
 733         refl_box_face = geompy.ShapesOp.GetInPlaceByHistory(part, aface)
 734         if refl_box_face is not None:
 735           geompy.addToStudyInFather(part, refl_box_face, ssname)
 736           error = "Result of GetInPlaceByHistory must be NULL for face "
 737           error += `ifa` + " of box " + `ibb`
 738           raise RuntimeError, error
 739       else:
 740         # use geompy interface
 741         refl_box_face = geompy.GetInPlaceByHistory(part, aface)
 742         geompy.addToStudyInFather(part, refl_box_face, ssname)
 743       ifa = ifa + 1
 744     ibb = ibb + 1
 745
 746 #END