src/GEOM_SWIG/GEOM_TestOthers.py

   1 #  GEOM GEOM_SWIG : binding of C++ implementaion with Python
   2 #
   3 #  Copyright (C) 2003  CEA
   4 #
   5 #  This library is free software; you can redistribute it and/or
   6 #  modify it under the terms of the GNU Lesser General Public
   7 #  License as published by the Free Software Foundation; either
   8 #  version 2.1 of the License.
   9 #
  10 #  This library is distributed in the hope that it will be useful,
  11 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 #  Lesser General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU Lesser General Public
  16 #  License along with this library; if not, write to the Free Software
  17 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18 #
  19 #  See http://www.salome-platform.org or email : webmaster.salome@opencascade.org
  20 #
  21 #
  22 #  File   : GEOM_TestOthers.py
  23 #  Author : Julia DOROVSKIKH
  24 #  Module : GEOM
  25 #  $Header$
  26
  27 def TestOtherOperations (geompy, math):
  28
  29   # MakeFaces
  30   p11 = geompy.MakeVertex( 0,  0, 0)
  31   p12 = geompy.MakeVertex(30,  0, 0)
  32   p13 = geompy.MakeVertex(30, 30, 0)
  33   p14 = geompy.MakeVertex( 0, 30, 0)
  34
  35   p21 = geompy.MakeVertex(10, 10, 0)
  36   p22 = geompy.MakeVertex(20, 10, 0)
  37   p23 = geompy.MakeVertex(20, 20, 0)
  38   p24 = geompy.MakeVertex(10, 20, 0)
  39
  40   e11 = geompy.MakeEdge(p11, p12)
  41   e12 = geompy.MakeEdge(p12, p13)
  42   e13 = geompy.MakeEdge(p13, p14)
  43   e14 = geompy.MakeEdge(p14, p11)
  44
  45   e21 = geompy.MakeEdge(p21, p22)
  46   e22 = geompy.MakeEdge(p22, p23)
  47   e23 = geompy.MakeEdge(p23, p24)
  48   e24 = geompy.MakeEdge(p24, p21)
  49
  50   w1 = geompy.MakeWire([e11, e12, e13, e14])
  51   w2 = geompy.MakeWire([e21, e22, e23, e24])
  52   w3 = geompy.MakeTranslation(w2, 0, 0, 10)
  53
  54   id_w1 = geompy.addToStudy(w1, "Outside Wire")
  55   id_w2 = geompy.addToStudy(w2, "Inside Wire")
  56   id_w3 = geompy.addToStudy(w3, "Inside Wire, translated along OZ")
  57
  58   f12 = geompy.MakeFaces([w1, w2], 0)
  59   id_f12 = geompy.addToStudy(f12, "MakeFaces WO + WI")
  60
  61   # OrientationChange
  62   Box = geompy.MakeBoxDXDYDZ(200, 200, 200)
  63   Orientation = geompy.OrientationChange(Box)
  64   id_Orientation = geompy.addToStudy(Orientation, "OrientationChange")
  65
  66   # MakeCommon, MakeCut, MakeFuse, MakeSection
  67   Sphere = geompy.MakeSphereR(100)
  68
  69   Common  = geompy.MakeCommon (Box, Sphere)
  70   Cut     = geompy.MakeCut    (Box, Sphere)
  71   Fuse    = geompy.MakeFuse   (Box, Sphere)
  72   Section = geompy.MakeSection(Box, Sphere)
  73
  74   id_Common  = geompy.addToStudy(Common,  "Common")
  75   id_Cut     = geompy.addToStudy(Cut,     "Cut")
  76   id_Fuse    = geompy.addToStudy(Fuse,    "Fuse")
  77   id_Section = geompy.addToStudy(Section, "Section")
  78
  79   # Partition
  80   p100 = geompy.MakeVertex(100, 100, 100)
  81   p300 = geompy.MakeVertex(300, 300, 300)
  82   Box1 = geompy.MakeBoxTwoPnt(p100, p300)
  83   Partition = geompy.Partition([Box], [Box1], [], [Box])
  84   id_Partition = geompy.addToStudy(Partition, "Partition of Box by Box1")
  85
  86   # MakeMultiRotation1D, MakeMultiRotation2D
  87   pz = geompy.MakeVertex(0, 0, 100)
  88   vy = geompy.MakeVectorDXDYDZ(0, 100, 0)
  89
  90   MultiRot1D = geompy.MakeMultiRotation1D(f12, vy, pz, 6)
  91   MultiRot2D = geompy.MakeMultiRotation2D(f12, vy, pz, 45, 6, 30, 3)
  92
  93   id_MultiRot1D = geompy.addToStudy(MultiRot1D, "MakeMultiRotation1D")
  94   id_MultiRot2D = geompy.addToStudy(MultiRot2D, "MakeMultiRotation2D")
  95
  96   # MakeFilletAll
  97   radius_fillet = 10.
  98   face5 = geompy.SubShapeSorted(Box, geompy.ShapeType["FACE"], [5])
  99   f_glob_id = geompy.GetSubShapeID(Box, face5)
 100   SuppFace = geompy.SuppressFaces(Box, [f_glob_id])
 101
 102   MakeFilletAll = geompy.MakeFilletAll(SuppFace, radius_fillet)
 103   id_MakeFilletAll = geompy.addToStudy(MakeFilletAll, "MakeFilletAll")
 104
 105   # MakeChamferAll
 106   dimension_chamfer = 10.
 107   MakeChamferAll = geompy.MakeChamferAll(SuppFace, dimension_chamfer)
 108   id_MakeChamferAll = geompy.addToStudy(MakeChamferAll, "MakeChamferAll")
 109
 110   # MakeChamfer
 111   d1 = 13.
 112   d2 = 7.
 113   box_faces = geompy.SubShapeAllSorted(Box, geompy.ShapeType["FACE"])
 114   f_ind_1 = geompy.GetSubShapeID(Box, box_faces[0])
 115   f_ind_2 = geompy.GetSubShapeID(Box, box_faces[1])
 116   f_ind_3 = geompy.GetSubShapeID(Box, box_faces[2])
 117
 118   MakeChamfer = geompy.MakeChamfer(Box, d1, d2, geompy.ShapeType["FACE"],
 119                                    [f_ind_1, f_ind_2, f_ind_3])
 120   id_MakeChamfer = geompy.addToStudy(MakeChamfer, "MakeChamfer")
 121
 122   # Export
 123   geompy.Export(f12, "/tmp/testExportImport.brep", "BREP")
 124
 125   # ExportBREP, ExportIGES, ExportSTEP
 126   geompy.ExportBREP(f12, "/tmp/testExportImportBREP.brep")
 127   geompy.ExportIGES(f12, "/tmp/testExportImportIGES.iges")
 128   geompy.ExportSTEP(f12, "/tmp/testExportImportSTEP.step")
 129
 130   # Import
 131   Import = geompy.Import("/tmp/testExportImport.brep", "BREP")
 132
 133   id_Import = geompy.addToStudy(Import, "Import")
 134
 135   # ImportBREP, ImportIGES, ImportSTEP
 136   ImportBREP = geompy.ImportBREP("/tmp/testExportImportBREP.brep")
 137   ImportIGES = geompy.ImportIGES("/tmp/testExportImportIGES.iges")
 138   ImportSTEP = geompy.ImportSTEP("/tmp/testExportImportSTEP.step")
 139
 140   id_ImportBREP = geompy.addToStudy(ImportBREP, "ImportBREP")
 141   id_ImportIGES = geompy.addToStudy(ImportIGES, "ImportIGES")
 142   id_ImportSTEP = geompy.addToStudy(ImportSTEP, "ImportSTEP")
 143
 144   # NumberOfFaces
 145   NumberOfFaces = geompy.NumberOfFaces(Box)
 146   if NumberOfFaces != 6:
 147     print "Bad number of faces in BOX!"
 148
 149   # NumberOfEdges
 150   NumberOfEdges = geompy.NumberOfEdges(Box)
 151   if NumberOfEdges != 12:
 152     print "Bad number of edges in BOX!"
 153
 154   # MakeBlockExplode
 155   Compound = geompy.MakeCompound([Box, Sphere])
 156   MakeBlockExplode = geompy.MakeBlockExplode(Compound, 6, 6)
 157
 158   id_MakeBlockExplode = geompy.addToStudy(MakeBlockExplode[0], "MakeBlockExplode")
 159
 160   # CheckCompoundOfBlocks
 161   p1 = geompy.MakeVertex(200, 0, 0)
 162   p2 = geompy.MakeVertex(400, 200, 200)
 163   p3 = geompy.MakeVertex(400, 50, 50)
 164   p4 = geompy.MakeVertex(600, 250, 250)
 165
 166   Box2 = geompy.MakeBoxTwoPnt(p1, p2)
 167   Box3 = geompy.MakeBoxTwoPnt(p3, p4)
 168   Cyl  = geompy.MakeCylinderRH(50, 300)
 169   Cone = geompy.MakeConeR1R2H(150, 10, 400)
 170
 171   Compound1 = geompy.MakeCompound([Box, Cyl, Cone, Box3, Box2])
 172
 173   IsValid = geompy.CheckCompoundOfBlocks(Compound1)
 174   if IsValid == 0:
 175     print "The Blocks Compound is NOT VALID"
 176   else:
 177     print "The Blocks Compound is VALID"
 178
 179   IsValid = geompy.CheckCompoundOfBlocks(Box)
 180   if IsValid == 0:
 181     print "The Box is NOT VALID"
 182   else:
 183     print "The Box is VALID"
 184
 185   # test geometrical groups
 186
 187   # CreateGroup
 188   CreateGroup = geompy.CreateGroup(Box, geompy.ShapeType["FACE"])
 189
 190   id_CreateGroup = geompy.addToStudy(CreateGroup, "CreateGroup")
 191
 192   # AddObject
 193   f_ind_4 = geompy.GetSubShapeID(Box, box_faces[3])
 194   f_ind_5 = geompy.GetSubShapeID(Box, box_faces[4])
 195   f_ind_6 = geompy.GetSubShapeID(Box, box_faces[5])
 196
 197   geompy.AddObject(CreateGroup, f_ind_6)
 198   geompy.AddObject(CreateGroup, f_ind_1)
 199   geompy.AddObject(CreateGroup, f_ind_4)
 200
 201   # RemoveObject(theGroup, theSubShapeID)
 202   geompy.RemoveObject(CreateGroup, f_ind_1)
 203
 204   # GetObjectIDs
 205   GetObjectIDs = geompy.GetObjectIDs(CreateGroup)
 206
 207   print "Group of Box's faces includes the following IDs:"
 208   for ObjectID in GetObjectIDs:
 209     print " ", ObjectID
 210
 211   BoxCopy = geompy.GetMainShape(CreateGroup)
 212
 213   # -----------------------------------------------------------------------------
 214   # enumeration ShapeTypeString as a dictionary
 215   # -----------------------------------------------------------------------------
 216   ShapeTypeString = {'0':"COMPOUND", '1':"COMPSOLID", '2':"SOLID", '3':"SHELL", '4':"FACE", '5':"WIRE", '6':"EDGE", '7':"VERTEX", '8':"SHAPE"}
 217
 218   GroupType = geompy.GetType(CreateGroup)
 219   print "Type of elements of the created group is ", ShapeTypeString[`GroupType`]
 220
 221   # Prepare data for the following operations
 222   p0 = geompy.MakeVertex(0, 0, 0)
 223   b0 = geompy.MakeBox(-50, -50, -50, 50, 50, 50)
 224   s0 = geompy.MakeSphereR(100)
 225
 226   id_b0 = geompy.addToStudy(b0, "b0")
 227   id_s0 = geompy.addToStudy(s0, "s0")
 228
 229   v_0pp = geompy.MakeVectorDXDYDZ( 0,  1,  1)
 230   v_0np = geompy.MakeVectorDXDYDZ( 0, -1,  1)
 231   v_p0p = geompy.MakeVectorDXDYDZ( 1,  0,  1)
 232   v_n0p = geompy.MakeVectorDXDYDZ(-1,  0,  1)
 233   v_pp0 = geompy.MakeVectorDXDYDZ( 1,  1,  0)
 234   v_np0 = geompy.MakeVectorDXDYDZ(-1,  1,  0)
 235
 236   pln_0pp = geompy.MakePlane(p0, v_0pp, 200)
 237   pln_0np = geompy.MakePlane(p0, v_0np, 200)
 238   pln_p0p = geompy.MakePlane(p0, v_p0p, 200)
 239   pln_n0p = geompy.MakePlane(p0, v_n0p, 200)
 240   pln_pp0 = geompy.MakePlane(p0, v_pp0, 200)
 241   pln_np0 = geompy.MakePlane(p0, v_np0, 200)
 242
 243   part_tool = geompy.MakePartition([b0, pln_0pp, pln_0np, pln_p0p, pln_n0p, pln_pp0, pln_np0],
 244                                    [],
 245                                    [],
 246                                    [b0])
 247   id_part_tool = geompy.addToStudy(part_tool, "part_tool")
 248
 249   part = geompy.MakePartition([s0], [part_tool])
 250   geompy.addToStudy(part, "part")
 251
 252   # GetFreeFacesIDs
 253   anIDs = geompy.GetFreeFacesIDs(part)
 254   freeFaces = geompy.GetSubShape(part, anIDs)
 255
 256   geompy.addToStudy(freeFaces, "freeFaces")
 257
 258   # RemoveExtraEdges
 259   freeFacesWithoutExtra = geompy.RemoveExtraEdges(freeFaces)
 260
 261   geompy.addToStudy(freeFacesWithoutExtra, "freeFacesWithoutExtra")
 262
 263   # GetSharedShapes
 264   sharedFaces = geompy.GetSharedShapes(part, freeFacesWithoutExtra, geompy.ShapeType["FACE"])
 265
 266   for shFace in sharedFaces:
 267     geompy.addToStudy(shFace, "sharedFace")
 268
 269   # CheckAndImprove
 270   blocksComp = geompy.CheckAndImprove(part)
 271
 272   geompy.addToStudy(blocksComp, "blocksComp")
 273
 274   # Propagate
 275   listChains = geompy.Propagate(blocksComp)
 276
 277   for chain in listChains:
 278     geompy.addToStudyInFather(blocksComp, chain, "propagation chain")
 279
 280   # GetPoint(theShape, theX, theY, theZ, theEpsilon)
 281   #
 282   # (-50,  50, 50) .-----. (50,  50, 50)
 283   #      pb0_top_1 |     |
 284   #                |     . pmidle
 285   #                |     |
 286   # (-50, -50, 50) '-----' (50, -50, 50)
 287   #
 288   pb0_top_1 = geompy.GetPoint(blocksComp, -50,  50,  50, 0.01)
 289   pb0_bot_1 = geompy.GetPoint(blocksComp, -50, -50, -50, 0.01)
 290
 291   geompy.addToStudyInFather(blocksComp, pb0_top_1, "point from blocksComp (-50,  50,  50)")
 292   geompy.addToStudyInFather(blocksComp, pb0_bot_1, "point from blocksComp (-50, -50, -50)")
 293
 294   # GetEdgeNearPoint(theShape, thePoint)
 295   pmidle = geompy.MakeVertex(50, 0, 50)
 296   edge1 = geompy.GetEdgeNearPoint(blocksComp, pmidle)
 297
 298   geompy.addToStudyInFather(blocksComp, edge1, "edge near point (50, 0, 50)")
 299
 300   # GetBlockByParts(theCompound, theParts)
 301   b0_image = geompy.GetBlockByParts(blocksComp, [pb0_top_1, pb0_bot_1, edge1])
 302
 303   geompy.addToStudyInFather(blocksComp, b0_image, "b0 image")
 304
 305   # GetShapesOnPlane(theShape, theShapeType, thePlane)
 306   # GetShapesOnCylinder(theShape, theShapeType, theAxis, theRadius)
 307   # GetShapesOnSphere(theShape, theShapeType, theCenter, theRadius)
 308
 309   # GetInPlace(theShapeWhere, theShapeWhat)