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