1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2008 CEA/DEN, EDF R&D, OPEN CASCADE
4 # Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 # GEOM GEOM_SWIG : binding of C++ implementaion with Python
24 # File : GEOM_TestOthers.py
25 # Author : Julia DOROVSKIKH
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
35 def TestExportImport (geompy, shape):
37 print "Test Export/Import ...",
39 tmpDir = os.getenv("TEMP")
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"
49 if os.access(fileExportImport, os.F_OK):
50 if os.access(fileExportImport, os.W_OK):
51 os.remove(fileExportImport)
53 fileExportImport = tmpDir + "/testExportImport1.brep"
55 if os.access(fileExportImportBREP, os.W_OK):
56 os.remove(fileExportImportBREP)
58 fileExportImportBREP = tmpDir + "/testExportImportBREP1.brep"
60 if os.access(fileExportImportIGES, os.W_OK):
61 os.remove(fileExportImportIGES)
63 fileExportImportIGES = tmpDir + "/testExportImportIGES1.iges"
65 if os.access(fileExportImportSTEP, os.W_OK):
66 os.remove(fileExportImportSTEP)
68 fileExportImportSTEP = tmpDir + "/testExportImportSTEP1.step"
71 geompy.Export(shape, fileExportImport, "BREP")
73 # ExportBREP, ExportIGES, ExportSTEP
74 geompy.ExportBREP(shape, fileExportImportBREP)
75 geompy.ExportIGES(shape, fileExportImportIGES)
76 geompy.ExportSTEP(shape, fileExportImportSTEP)
79 Import = geompy.Import(fileExportImport, "BREP")
81 id_Import = geompy.addToStudy(Import, "Import")
83 # ImportBREP, ImportIGES, ImportSTEP
84 ImportBREP = geompy.ImportBREP(fileExportImportBREP)
85 ImportIGES = geompy.ImportIGES(fileExportImportIGES)
86 ImportSTEP = geompy.ImportSTEP(fileExportImportSTEP)
88 id_ImportBREP = geompy.addToStudy(ImportBREP, "ImportBREP")
89 id_ImportIGES = geompy.addToStudy(ImportIGES, "ImportIGES")
90 id_ImportSTEP = geompy.addToStudy(ImportSTEP, "ImportSTEP")
92 # Remove files for Export/Import testing
93 os.remove(fileExportImport)
94 os.remove(fileExportImportBREP)
95 os.remove(fileExportImportIGES)
96 os.remove(fileExportImportSTEP)
101 def TestOtherOperations (geompy, math):
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)
108 v_y = geompy.MakeVectorDXDYDZ( 0, -1, 0)
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)
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)
120 e11 = geompy.MakeEdge(p11, p12)
121 e12 = geompy.MakeEdge(p12, p13)
122 e13 = geompy.MakeEdge(p13, p14)
123 e14 = geompy.MakeEdge(p14, p11)
125 e21 = geompy.MakeEdge(p21, p22)
126 e22 = geompy.MakeEdge(p22, p23)
127 e23 = geompy.MakeEdge(p23, p24)
128 e24 = geompy.MakeEdge(p24, p21)
130 w1 = geompy.MakeWire([e11, e12, e13, e14])
131 w2 = geompy.MakeWire([e21, e22, e23, e24])
132 w3 = geompy.MakeTranslation(w2, 0, 0, 10)
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")
139 f12 = geompy.MakeFaces([w1, w2], 0)
140 id_f12 = geompy.addToStudy(f12, "MakeFaces WO + WI")
143 TestExportImport(geompy, f12)
146 Box = geompy.MakeBoxDXDYDZ(200, 200, 200)
147 Orientation = geompy.OrientationChange(Box)
148 id_Orientation = geompy.addToStudy(Orientation, "OrientationChange")
150 # MakeCommon, MakeCut, MakeFuse, MakeSection
151 Sphere = geompy.MakeSphereR(100)
153 Common = geompy.MakeCommon (Box, Sphere)
154 Cut = geompy.MakeCut (Box, Sphere)
155 Fuse = geompy.MakeFuse (Box, Sphere)
156 Section = geompy.MakeSection(Box, Sphere)
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")
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")
171 # MakeMultiRotation1D, MakeMultiRotation2D
172 pz = geompy.MakeVertex(0, 0, 100)
173 vy = geompy.MakeVectorDXDYDZ(0, 100, 0)
175 MultiRot1D = geompy.MakeMultiRotation1D(f12, vy, pz, 6)
176 MultiRot2D = geompy.MakeMultiRotation2D(f12, vy, pz, 45, 6, 30, 3)
178 id_MultiRot1D = geompy.addToStudy(MultiRot1D, "MakeMultiRotation1D")
179 id_MultiRot2D = geompy.addToStudy(MultiRot2D, "MakeMultiRotation2D")
183 face5 = geompy.SubShapeSorted(Box, geompy.ShapeType["FACE"], [5])
184 f_glob_id = geompy.GetSubShapeID(Box, face5)
185 SuppFace = geompy.SuppressFaces(Box, [f_glob_id])
187 MakeFilletAll = geompy.MakeFilletAll(SuppFace, radius_fillet)
188 id_MakeFilletAll = geompy.addToStudy(MakeFilletAll, "MakeFilletAll")
191 dimension_chamfer = 10.
192 MakeChamferAll = geompy.MakeChamferAll(SuppFace, dimension_chamfer)
193 id_MakeChamferAll = geompy.addToStudy(MakeChamferAll, "MakeChamferAll")
198 box_faces = geompy.SubShapeAllSorted(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])
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")
208 NumberOfFaces = geompy.NumberOfFaces(Box)
209 if NumberOfFaces != 6:
210 print "Bad number of faces in BOX!"
212 NumberOfEdges = geompy.NumberOfEdges(Box)
213 if NumberOfEdges != 12:
214 print "Bad number of edges in BOX!"
216 NumberOfSolids = geompy.NumberOfSolids(Box)
217 if NumberOfSolids != 1:
218 print "Bad number of solids in BOX!"
220 NumberOfShapes = geompy.NumberOfSubShapes(Box, geompy.ShapeType["SHAPE"])
221 if NumberOfShapes != 34:
222 print "Bad number of shapes in BOX!"
225 Compound = geompy.MakeCompound([Box, Sphere])
226 MakeBlockExplode = geompy.MakeBlockExplode(Compound, 6, 6)
228 id_MakeBlockExplode = geompy.addToStudy(MakeBlockExplode[0], "MakeBlockExplode")
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)
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)
241 Compound1 = geompy.MakeCompound([Box, Cyl, Cone, Box3, Box2])
243 IsValid = geompy.CheckCompoundOfBlocks(Compound1)
245 print "The Blocks Compound is NOT VALID"
247 print "The Blocks Compound is VALID"
249 IsValid = geompy.CheckCompoundOfBlocks(Box)
251 print "The Box is NOT VALID"
253 print "The Box is VALID"
256 Cone_ss = geompy.GetSame(Compound1, Cone)
257 id_Cone_ss = geompy.addToStudyInFather(Compound1, Cone_ss, "Cone subshape")
259 # test geometrical groups
262 CreateGroup = geompy.CreateGroup(Box, geompy.ShapeType["FACE"])
264 id_CreateGroup = geompy.addToStudy(CreateGroup, "CreateGroup")
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])
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
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
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
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
289 GetObjectIDs = geompy.GetObjectIDs(CreateGroup)
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:
297 BoxCopy = geompy.GetMainShape(CreateGroup)
300 geompy.DifferenceIDs(CreateGroup, [f_ind_3, f_ind_5])
301 # Now contains f_ind_6
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
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:
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"}
320 GroupType = geompy.GetType(CreateGroup)
321 print "Type of elements of the created group is ", ShapeTypeString[`GroupType`]
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)
328 id_b0 = geompy.addToStudy(b0, "b0")
329 id_s0 = geompy.addToStudy(s0, "s0")
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)
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)
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)
349 #id_part_tool_1 = geompy.addToStudy(part_tool_1, "part_tool_1")
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)
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)
377 #pt_box = geompy.GetBlockNearPoint(part_tool_1, p0)
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")
386 #part = geompy.MakePartition([s0], [part_tool])
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)
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)
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)
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")
417 anIDs = geompy.GetFreeFacesIDs(part)
418 freeFaces = geompy.GetSubShape(part, anIDs)
420 geompy.addToStudy(freeFaces, "freeFaces")
422 # RemoveExtraEdges with union of all faces, sharing common surfaces
423 tools = [pln_pp0, pln_pn0, pln_p0p, pln_p0n]
425 Partition_1 = geompy.MakePartition([Sphere], tools, [], [], geompy.ShapeType["SOLID"], 0, [])
426 geompy.addToStudy(Partition_1, "Partition_1")
428 faces = geompy.SubShapeAllSorted(Partition_1, geompy.ShapeType["FACE"])
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")
438 Vector_5 = geompy.MakeVectorDXDYDZ(0, 20, 0)
439 geompy.addToStudy(Vector_5, "Vector_5")
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)
445 geompy.addToStudy(Rotation_1, "Rotation_1")
446 geompy.addToStudy(Rotation_2, "Rotation_2")
447 geompy.addToStudy(Rotation_3, "Rotation_3")
449 Vector_6 = geompy.MakeVectorDXDYDZ(0, 0, 20)
450 geompy.addToStudy(Vector_6, "Vector_6")
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")
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
461 geompy.addToStudy(Shell_1, "Shell_1")
462 geompy.addToStudy(Solid_1, "Solid_1")
463 geompy.addToStudy(NoExtraEdges_1, "NoExtraEdges_1")
465 # RemoveExtraEdges (by default, doUnionFaces = False)
466 freeFacesWithoutExtra = geompy.RemoveExtraEdges(freeFaces)
468 geompy.addToStudy(freeFacesWithoutExtra, "freeFacesWithoutExtra")
471 sharedFaces = geompy.GetSharedShapes(part, freeFacesWithoutExtra,
472 geompy.ShapeType["FACE"])
474 for shFace in sharedFaces:
475 geompy.addToStudy(shFace, "sharedFace")
478 blocksComp = geompy.CheckAndImprove(part)
480 geompy.addToStudy(blocksComp, "blocksComp")
483 listChains = geompy.Propagate(blocksComp)
485 for chain in listChains:
486 geompy.addToStudyInFather(blocksComp, chain, "propagation chain")
488 # GetPoint(theShape, theX, theY, theZ, theEpsilon)
490 # (-50, 50, 50) .-----. (50, 50, 50)
494 # (-50, -50, 50) '-----' (50, -50, 50)
496 pb0_top_1 = geompy.GetPoint(blocksComp, -50, 50, 50, 0.01)
497 pb0_bot_1 = geompy.GetPoint(blocksComp, -50, -50, -50, 0.01)
499 geompy.addToStudyInFather(blocksComp, pb0_top_1, "point from blocksComp (-50, 50, 50)")
500 geompy.addToStudyInFather(blocksComp, pb0_bot_1, "point from blocksComp (-50, -50, -50)")
502 # GetEdgeNearPoint(theShape, thePoint)
503 pmidle = geompy.MakeVertex(50, 0, 50)
504 edge1 = geompy.GetEdgeNearPoint(blocksComp, pmidle)
506 geompy.addToStudyInFather(blocksComp, edge1, "edge near point (50, 0, 50)")
508 # GetBlockByParts(theCompound, theParts)
509 b0_image = geompy.GetBlockByParts(blocksComp, [pb0_top_1, pb0_bot_1, edge1])
511 geompy.addToStudyInFather(blocksComp, b0_image, "b0 image")
514 faces_on_pln = geompy.GetShapesOnPlane(blocksComp, geompy.ShapeType["FACE"],
515 v_0pp, geompy.GEOM.ST_ONIN)
516 for face_i in faces_on_pln:
517 geompy.addToStudy(face_i, "Face on Plane (N = (0, 1, 1)) or below it")
519 # GetShapesOnPlaneIDs
520 faces_above_pln_ids = geompy.GetShapesOnPlaneIDs(blocksComp, geompy.ShapeType["FACE"],
521 v_0pp, geompy.GEOM.ST_OUT)
522 faces_above = geompy.CreateGroup(blocksComp, geompy.ShapeType["FACE"])
523 geompy.UnionIDs(faces_above, faces_above_pln_ids)
524 geompy.addToStudy(faces_above, "Group of faces above Plane (N = (0, 1, 1))")
526 # GetShapesOnPlaneWithLocation
527 Loc = geompy.MakeVertex(0, -50, 0)
528 edges_on_pln = geompy.GetShapesOnPlaneWithLocation(blocksComp, geompy.ShapeType["EDGE"],
529 v_y, Loc, geompy.GEOM.ST_ON)
530 for edge_i in edges_on_pln:
531 geompy.addToStudy(edge_i, "Edge on Plane (N = (0, -1, 0) & Location = (0, -50, 0)")
533 # GetShapesOnPlaneWithLocationIDs
534 edges_on_pln_ids = geompy.GetShapesOnPlaneWithLocationIDs(
535 blocksComp, geompy.ShapeType["EDGE"], v_y, Loc, geompy.GEOM.ST_ON)
536 group_edges_on_pln = geompy.CreateGroup(blocksComp, geompy.ShapeType["EDGE"])
537 geompy.UnionIDs(group_edges_on_pln, edges_on_pln_ids)
538 grname = "Group of edges on Plane (N = (0, -1, 0) & Location = (0, -50, 0))"
539 geompy.addToStudy(group_edges_on_pln, grname)
541 # GetShapesOnCylinder
542 edges_out_cyl = geompy.GetShapesOnCylinder(blocksComp, geompy.ShapeType["EDGE"],
543 vy, 55, geompy.GEOM.ST_OUT)
544 for edge_i in edges_out_cyl:
545 geompy.addToStudy(edge_i, "Edge out of Cylinder (axis = (0, 1, 0), r = 55)")
547 # GetShapesOnCylinderIDs
548 edges_in_cyl_ids = geompy.GetShapesOnCylinderIDs(blocksComp, geompy.ShapeType["EDGE"],
549 vy, 80, geompy.GEOM.ST_IN)
550 edges_in = geompy.CreateGroup(blocksComp, geompy.ShapeType["EDGE"])
551 geompy.UnionIDs(edges_in, edges_in_cyl_ids)
552 geompy.addToStudy(edges_in, "Group of edges inside Cylinder (axis = (0, 1, 0), r = 55)")
554 # GetShapesOnCylinderWithLocation
555 edges_out_cyl = geompy.GetShapesOnCylinderWithLocation(blocksComp, geompy.ShapeType["EDGE"],
556 vy, p11, 55, geompy.GEOM.ST_OUT)
557 for edge_i in edges_out_cyl:
558 geompy.addToStudy(edge_i, "Edge out of Cylinder (axis = (0, 1, 0), r = 55)")
560 # GetShapesOnCylinderIDs
561 edges_in_cyl_ids = geompy.GetShapesOnCylinderIDs(blocksComp, geompy.ShapeType["EDGE"],
562 vy, 80, geompy.GEOM.ST_IN)
563 edges_in = geompy.CreateGroup(blocksComp, geompy.ShapeType["EDGE"])
564 geompy.UnionIDs(edges_in, edges_in_cyl_ids)
565 geompy.addToStudy(edges_in, "Group of edges inside Cylinder (axis = (0, 1, 0), r = 55)")
568 vertices_on_sph = geompy.GetShapesOnSphere(blocksComp, geompy.ShapeType["VERTEX"],
569 p0, 100, geompy.GEOM.ST_ON)
570 for vertex_i in vertices_on_sph:
571 geompy.addToStudy(vertex_i, "Vertex on Sphere (center = (0, 0, 0), r = 100)")
573 # GetShapesOnSphereIDs
574 vertices_on_sph_ids = geompy.GetShapesOnSphereIDs(blocksComp, geompy.ShapeType["VERTEX"],
575 p0, 100, geompy.GEOM.ST_ON)
576 vertices_on = geompy.CreateGroup(blocksComp, geompy.ShapeType["VERTEX"])
577 geompy.UnionIDs(vertices_on, vertices_on_sph_ids)
578 geompy.addToStudy(vertices_on, "Group of vertices on Sphere (center = (0, 0, 0), r = 100)")
580 # GetShapesOnQuadrangle
582 geompy.addToStudy(f12, "F12" )
584 bl = geompy.MakeVertex(10,-10, 0)
585 br = geompy.MakeVertex(40,-10, 0)
586 tr = geompy.MakeVertex(40, 20, 0)
587 tl = geompy.MakeVertex(10, 20, 0)
588 qe1 = geompy.MakeEdge(bl, br)
589 qe2 = geompy.MakeEdge(br, tr)
590 qe3 = geompy.MakeEdge(tr, tl)
591 qe4 = geompy.MakeEdge(tl, bl)
592 quadrangle = geompy.MakeWire([qe1, qe2, qe3, qe4])
593 geompy.addToStudy(quadrangle, "Quadrangle")
595 edges_onin_quad = geompy.GetShapesOnQuadrangle(f12, geompy.ShapeType["EDGE"],
596 tl, tr, bl, br, geompy.GEOM.ST_ONIN)
597 comp = geompy.MakeCompound(edges_onin_quad)
598 geompy.addToStudy(comp, "Edges of F12 ONIN Quadrangle")
599 if len( edges_onin_quad ) != 4:
600 print "Error in GetShapesOnQuadrangle()"
602 # GetShapesOnQuadrangleIDs
603 vertices_on_quad_ids = geompy.GetShapesOnQuadrangleIDs(f12, geompy.ShapeType["VERTEX"],
604 tl, tr, bl, br, geompy.GEOM.ST_ON)
605 vertices_on_quad = geompy.CreateGroup(f12, geompy.ShapeType["VERTEX"])
606 geompy.UnionIDs(vertices_on_quad, vertices_on_quad_ids)
607 geompy.addToStudy(vertices_on_quad, "Group of vertices on Quadrangle F12")
610 edges_on_box = geompy.GetShapesOnBox(b0, part, geompy.ShapeType["EDGE"],
612 comp = geompy.MakeCompound(edges_on_box)
613 geompy.addToStudy(comp, "Edges of part ON box b0")
614 if len( edges_on_box ) != 12:
615 print "Error in GetShapesOnBox()"
618 faces_on_box_ids = geompy.GetShapesOnBoxIDs(b0, part, geompy.ShapeType["FACE"],
620 faces_on_box = geompy.CreateGroup(part, geompy.ShapeType["FACE"])
621 geompy.UnionIDs(faces_on_box, faces_on_box_ids)
622 geompy.addToStudyInFather(part, faces_on_box, "Group of faces on box b0")
624 # Prepare arguments for GetShapesOnShape
625 sph1 = geompy.MakeSphere(50, 50, 50, 40)
626 sph2 = geompy.MakeSphere(50, 50, -50, 40)
627 pcyl = geompy.MakeVertex(50, 50, -50)
628 cyli = geompy.MakeCylinder(pcyl, vz, 40, 100)
629 fuse = geompy.MakeFuse(sph1, cyli)
630 fuse = geompy.MakeFuse(fuse, sph2)
631 # As after Fuse we have a compound, we need to obtain a solid from it
632 shsh = geompy.SubShapeAll(fuse, geompy.ShapeType["SOLID"])
634 geompy.addToStudy(sh_1, "sh_1")
637 faces_in_sh = geompy.GetShapesOnShape(sh_1, part, geompy.ShapeType["FACE"],
639 comp = geompy.MakeCompound(faces_in_sh)
640 geompy.addToStudy(comp, "Faces of part IN shape sh_1")
641 if len(faces_in_sh) != 11:
642 print "Error in GetShapesOnShape()"
644 # GetShapesOnShapeAsCompound
645 faces_in_sh_c = geompy.GetShapesOnShapeAsCompound(sh_1, part, geompy.ShapeType["FACE"],
647 geompy.addToStudy(faces_in_sh_c, "Faces of part IN shape sh_1 (as compound)")
649 # GetShapesOnShapeIDs
650 edges_in_sh_ids = geompy.GetShapesOnShapeIDs(sh_1, part, geompy.ShapeType["EDGE"],
652 edges_in_sh = geompy.CreateGroup(part, geompy.ShapeType["EDGE"])
653 geompy.UnionIDs(edges_in_sh, edges_in_sh_ids)
654 geompy.addToStudyInFather(part, edges_in_sh, "Group of edges in shape sh_1")
655 if len(edges_in_sh_ids) != 15:
656 print "Error in GetShapesOnShapeIDs()"
658 # Prepare arguments for GetInPlace and GetInPlaceByHistory
659 box5 = geompy.MakeBoxDXDYDZ(100, 100, 100)
660 box6 = geompy.MakeTranslation(box5, 50, 50, 0)
662 geompy.addToStudy(box5, "Box 5")
663 geompy.addToStudy(box6, "Box 6")
665 part = geompy.MakePartition([box5], [box6])
666 geompy.addToStudy(part, "Partitioned")
668 box5_faces = geompy.SubShapeAll(box5, geompy.ShapeType["FACE"])
669 box6_faces = geompy.SubShapeAll(box6, geompy.ShapeType["FACE"])
672 geompy.addToStudyInFather(box5, box5_faces[ifa], "Face" + `ifa + 1`)
673 geompy.addToStudyInFather(box6, box6_faces[ifa], "Face" + `ifa + 1`)
675 # GetInPlace(theShapeWhere, theShapeWhat)
677 faces_list = [box5_faces, box6_faces]
678 for afaces in faces_list:
681 refl_box_face = geompy.GetInPlace(part, aface)
682 if ibb == 6 and (ifa == 2 or ifa == 4):
683 # For two faces of the tool box
684 # there is no reflection in the result.
685 if refl_box_face is not None:
686 error = "Result of GetInPlace must be NULL for face "
687 error += `ifa` + " of box " + `ibb`
688 raise RuntimeError, error
690 ssname = "Reflection of face " + `ifa` + " of box " + `ibb`
691 geompy.addToStudyInFather(part, refl_box_face, ssname)
695 # GetInPlaceByHistory(theShapeWhere, theShapeWhat)
696 part = geompy.MakePartition([box5], [box6])
697 geompy.addToStudy(part, "Partitioned")
700 faces_list = [box5_faces, box6_faces]
701 for afaces in faces_list:
704 ssname = "Reflection of face " + `ifa` + " of box " + `ibb` + " (by history)"
705 if ibb == 6 and (ifa == 2 or ifa == 4):
706 # use IDL interface directly to avoid error message appearence in Python console
707 refl_box_face = geompy.ShapesOp.GetInPlaceByHistory(part, aface)
708 if refl_box_face is not None:
709 geompy.addToStudyInFather(part, refl_box_face, ssname)
710 error = "Result of GetInPlaceByHistory must be NULL for face "
711 error += `ifa` + " of box " + `ibb`
712 raise RuntimeError, error
714 # use geompy interface
715 refl_box_face = geompy.GetInPlaceByHistory(part, aface)
716 geompy.addToStudyInFather(part, refl_box_face, ssname)