]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOM_SWIG/GEOM_TestAll.py
Salome HOME
NPAL: 16557, 16558, 16578, 16549, 16561
[modules/geom.git] / src / GEOM_SWIG / GEOM_TestAll.py
1 #  GEOM GEOM_SWIG : binding of C++ omplementaion with Python
2 #
3 #  Copyright (C) 2003  CEA
4
5 #
6 #  This library is free software; you can redistribute it and/or
7 #  modify it under the terms of the GNU Lesser General Public
8 #  License as published by the Free Software Foundation; either
9 #  version 2.1 of the License.
10 #
11 #  This library is distributed in the hope that it will be useful,
12 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 #  Lesser General Public License for more details.
15 #
16 #  You should have received a copy of the GNU Lesser General Public
17 #  License along with this library; if not, write to the Free Software
18 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 #
20 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 #
22 #
23 #  File   : GEOM_usinggeom.py
24 #  Author : Damien COQUERET, Open CASCADE
25 #  Module : GEOM
26 #  $Header$
27
28
29 def TestAll (geompy, math):
30
31   #Create base Variables
32   nbtimes1  = 5      #Short
33   nbtimes2  = 5
34   mindeg = 2
35   maxdeg = 5
36   nbiter = 5
37   ShapeTypeFace = geompy.ShapeType["FACE"]
38   ShapeTypeEdge = geompy.ShapeType["EDGE"]
39   WantPlanarFace = 1 #True
40
41   radius  = 10.  #Double
42   radius1 = 100.
43   radius2 = 200.
44   height  = 200.
45   d1      = 10.
46   d2      = 10.
47   step1   = 250.
48   step2   = 250.
49   angle   = 45.
50   angle1  = angle * math.pi / 180
51   angle2  = 2 * angle1
52   factor  = 2.
53   tol3d   = 0.0001
54   tol2d   = 0.0001
55   weight  = 1000000.
56   waterdensity = 1.
57   meshingdeflection = 0.01
58   trimsize  = 1000.
59   precision = 0.00001
60
61   #Create base points
62   p0   = geompy.MakeVertex(0.  , 0.  , 0.  ) #(3 Doubles)->GEOM_Object_ptr
63   px   = geompy.MakeVertex(100., 0.  , 0.  )
64   py   = geompy.MakeVertex(0.  , 100., 0.  )
65   pz   = geompy.MakeVertex(0.  , 0.  , 100.)
66   pxyz = geompy.MakeVertex(100., 100., 100.)
67
68   p200 = geompy.MakeVertexWithRef(pxyz, 100., 100., 100.) #(GEOM_Object_ptr, 3 Doubles)->GEOM_Object_ptr
69
70   #Create base directions
71   vx   = geompy.MakeVector(p0, px) #(GEOM_Object_ptr, GEOM_Object_ptr)->GEOM_Object_ptr
72   vy   = geompy.MakeVector(p0, py)
73   vz   = geompy.MakeVector(p0, pz)
74   vxy  = geompy.MakeVector(px, py)
75
76   vxyz = geompy.MakeVectorDXDYDZ(100., 100., 100.) #(3 Doubles)->GEOM_Object_ptr
77
78   #Create local coordinate systems
79   cs1 = geompy.MakeMarker(50,50,50, 1,0,0, 0,1,0)
80   cs2 = geompy.MakeMarker(70,80,10, 1,0,1, 1,1,0)
81
82   #Create base geometry 2D
83   Line   = geompy.MakeLineTwoPnt(p0, pxyz)                  #(2 GEOM_Object_ptr)->GEOM_Object_ptr
84   Line1  = geompy.MakeLine(pz, vxy)                         #(2 GEOM_Object_ptr)->GEOM_Object_ptr
85   Plane  = geompy.MakePlane(pz, vxyz, trimsize)             #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
86   Plane1 = geompy.MakePlaneThreePnt(px, pz, p200, trimsize) #(4 Doubles)->GEOM_Object_ptr
87
88   Arc      = geompy.MakeArc(py, pz, px)                   #(3 GEOM_Object_ptr)->GEOM_Object_ptr
89   Arc2     = geompy.MakeArcCenter(py, pz, px,0)           #(3 GEOM_Object_ptr,Boolean)->GEOM_Object_ptr
90   Circle   = geompy.MakeCircle(p0, vz, radius1)           #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
91   Circle1  = geompy.MakeCircleThreePnt(p0, pxyz, px)      #(3 GEOM_Object_ptr)->GEOM_Object_ptr
92   Circle2  = geompy.MakeCircleCenter2Pnt(p0, pxyz, py)    #(3 GEOM_Object_ptr)->GEOM_Object_ptr
93   Ellipse  = geompy.MakeEllipse(p0, vy, radius2, radius1) #(2 GEOM_Object_ptr, 2 Doubles)->GEOM_Object_ptr
94   Polyline = geompy.MakePolyline([p0, pz, py, p200])      #(List of GEOM_Object_ptr)->GEOM_Object_ptr
95   Bezier   = geompy.MakeBezier([p0, pz, p200, px])        #(List of GEOM_Object_ptr)->GEOM_Object_ptr
96   Interpol = geompy.MakeInterpol([px, py, p200, pxyz])    #(List of GEOM_Object_ptr)->GEOM_Object_ptr
97   Sketcher = geompy.MakeSketcher("Sketcher:F -100 -100:TT 250 -100:R 0:C 100 150:R 0:L 300:WW",
98                                  [100,0,0, 1,1,1, -1,1,0]) #(String, List of Doubles)->GEOM_Object_ptr
99
100   #Test point on curve creation
101   p_on_arc = geompy.MakeVertexOnCurve(Arc, 0.25) #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
102
103   #Create base geometry 3D
104   Box      = geompy.MakeBoxTwoPnt(p0, p200)                   #(2 GEOM_Object_ptr)->GEOM_Object_ptr
105   Box1     = geompy.MakeBoxDXDYDZ(10, 20, 30)                 #(3 Doubles)->GEOM_Object_ptr
106   Box2     = geompy.MakeBox(10,20,30, 15,25,35)               #(6 Doubles)->GEOM_Object_ptr
107   Cylinder = geompy.MakeCylinder(p0, vz, radius1, height)     #(2 GEOM_Object_ptr, 2 Doubles)->GEOM_Object_ptr
108   Cyl1     = geompy.MakeCylinderRH(radius2, height)           #(2 Doubles)->GEOM_Object_ptr
109   Sphere   = geompy.MakeSpherePntR(p0, radius1)               #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
110   Sphere1  = geompy.MakeSphereR(radius)                       #(Double)->GEOM_Object_ptr
111   Sphere2  = geompy.MakeSphere(50, 70, 30, radius)            #(4 Doubles)->GEOM_Object_ptr
112   Cone     = geompy.MakeCone(p0, vz, radius2, radius, height) #(2 GEOM_Object_ptr, 3 Doubles)->GEOM_Object_ptr
113   Cone1    = geompy.MakeConeR1R2H(radius1, radius, height)    #(3 Doubles)->GEOM_Object_ptr
114   Torus    = geompy.MakeTorus(p0, vz, radius2, radius)        #(2 GEOM_Object_ptr, 2 Doubles)->GEOM_Object_ptr
115   Torus1   = geompy.MakeTorusRR(radius2, radius1)             #(2 Doubles)->GEOM_Object_ptr
116
117   #Boolean (Common, Cut, Fuse, Section)
118   Common  = geompy.MakeBoolean(Box, Sphere, 1) #(2 GEOM_Object_ptr, Short)->GEOM_Object_ptr
119   Cut     = geompy.MakeBoolean(Box, Sphere, 2)
120   Fuse    = geompy.MakeBoolean(Box, Sphere, 3)
121   Section = geompy.MakeBoolean(Box, Sphere, 4)
122
123   #Create base objects
124   Edge     = geompy.MakeEdge(p0, pxyz)               #(2 GEOM_Object_ptr)->GEOM_Object_ptr
125   Wire     = geompy.MakeWire([vxy, Arc])             #(List Of GEOM_Object_ptr)->GEOM_Object_ptr
126   Face     = geompy.MakeFace(Wire, WantPlanarFace)   #(GEOM_Object_ptr, Boolean)->GEOM_Object_ptr
127   Face1    = geompy.MakeFaceWires([Wire, Sketcher],
128                                   WantPlanarFace)    #(List of GEOM_Object_ptr, Boolean)->GEOM_Object_ptr
129   Face2    = geompy.MakeFace(Sketcher, WantPlanarFace)
130   Shell    = geompy.MakeShell([Face, Face1])         #(List of GEOM_Object_ptr)->GEOM_Object_ptr
131
132   Prism1   = geompy.MakePrism(Face2, p0, pxyz)       #(3 GEOM_Object_ptr)->GEOM_Object_ptr
133   prism1_faces = geompy.SubShapeAllSorted(Prism1, ShapeTypeFace)
134   Shell1   = geompy.MakeShell([prism1_faces[0], prism1_faces[1],
135                                prism1_faces[3], prism1_faces[4],
136                                prism1_faces[5], prism1_faces[2]])
137   Solid    = geompy.MakeSolid([Shell1])              #(List of GEOM_Object_ptr)->GEOM_Object_ptr
138
139   ShapeListCompound = []
140   i = 0
141   while i <= 3 :
142         S = geompy.MakeTranslation(Arc, i * 100., i * 100., i * 100.)
143         ShapeListCompound.append(S)
144         i = i + 1
145   Compound = geompy.MakeCompound(ShapeListCompound)  #(List of GEOM_Object_ptr)->GEOM_Object_ptr
146
147   # Test plane from existing face creation
148   Plane2 = geompy.MakePlaneFace(Face, trimsize)      #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
149
150   #ShapeList for Sewing
151   S = geompy.MakeRotation(Face, vxy, angle1)
152
153   #Create advanced objects
154   Copy       = geompy.MakeCopy(Box)                      #(GEOM_Object_ptr)->GEOM_Object_ptr
155   Prism      = geompy.MakePrismVecH(Face, vz, 100.0)     #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
156   Revolution = geompy.MakeRevolution(Face, vz, angle2)   #
157   Filling    = geompy.MakeFilling(Compound, mindeg, maxdeg,
158                                   tol2d, tol3d, nbiter)  #(GEOM_Object_ptr, 4 Doubles, Short)->GEOM_Object_ptr
159   Pipe       = geompy.MakePipe(Wire, Edge)               #(2 GEOM_Object_ptr)->GEOM_Object_ptr
160   Sewing     = geompy.MakeSewing([Face, S], precision)   #(List Of GEOM_Object_ptr, Double)->GEOM_Object_ptr
161
162   #Transform objects
163   Translation = geompy.MakeTranslationTwoPoints(Box, px, pz)    #(3 GEOM_Object_ptr)->GEOM_Object_ptr
164   TranslVect  = geompy.MakeTranslationVector(Box, vxyz)         #(2 GEOM_Object_ptr)->GEOM_Object_ptr
165   Rotation    = geompy.MakeRotation(Box, vz, angle1)            #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
166   RotatPnt    = geompy.MakeRotationThreePoints(Box, px, py, pz) #(4 GEOM_Object_ptr)->GEOM_Object_ptr
167   Scale       = geompy.MakeScaleTransform(Box, p0, factor)      #
168   Mirror      = geompy.MakeMirrorByPlane(Box, Plane)            #(2 GEOM_Object_ptr)->GEOM_Object_ptr
169   MirrorAxis  = geompy.MakeMirrorByAxis(Box, Line1)             #
170   MirrorPnt   = geompy.MakeMirrorByPoint(Box, p200)             #
171   Position    = geompy.MakePosition(Box, cs1, cs2)              #(3 GEOM_Object_ptr)->GEOM_Object_ptr
172   Offset      = geompy.MakeOffset(Box, 10.)                     #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
173   Orientation = geompy.ChangeOrientation(Box)
174
175   #IDList for Fillet/Chamfer
176   prism_edges = geompy.SubShapeAllSorted(Prism, ShapeTypeEdge)
177
178   for anEdge in prism_edges:
179     eid = geompy.GetSubShapeID(Prism, anEdge)
180     sse = geompy.GetSubShape(Prism, [eid])
181
182     sse_id = geompy.GetSubShapeID(Prism, sse)
183     if sse_id != eid:
184       print "Error: GetSubShape() or GetSubShapeID() has failed!"
185
186   IDlist_e = []
187   IDlist_e.append(geompy.GetSubShapeID(Prism, prism_edges[0]))
188   IDlist_e.append(geompy.GetSubShapeID(Prism, prism_edges[1]))
189   IDlist_e.append(geompy.GetSubShapeID(Prism, prism_edges[2]))
190
191   prism_faces = geompy.SubShapeAllSorted(Prism, ShapeTypeFace)
192
193   f_ind_1 = geompy.GetSubShapeID(Prism, prism_faces[0])
194   f_ind_2 = geompy.GetSubShapeID(Prism, prism_faces[1])
195
196   IDlist_f = [f_ind_1, f_ind_2]
197   
198   #Local operations
199   Fillet   = geompy.MakeFillet (Prism, radius, ShapeTypeEdge,
200                                 IDlist_e) #(GEOM_Object_ptr, Double, Short, ListOfLong)->GEOM_Object_ptr
201   Fillet2  = geompy.MakeFilletR1R2 (Prism, 7., 13., ShapeTypeEdge,
202                                     IDlist_e) #(GEOM_Object_ptr, Double, Double, Short, ListOfLong)->GEOM_Object_ptr
203   Chamfer  = geompy.MakeChamferEdge(Prism, d1, d2,
204                                     f_ind_1, f_ind_2) #(GEOM_Object_ptr, 2 Doubles, 2 Long)->GEOM_Object_ptr
205   Chamfer2 = geompy.MakeChamferFaces(Prism, d1, d2,
206                                      IDlist_f) #(GEOM_Object_ptr, 2 Doubles, ListOfLong)->GEOM_Object_ptr
207   Chamfer3 = geompy.MakeChamferEdges(Prism, d1, d2,
208                                      IDlist_e) #(GEOM_Object_ptr, 2 Doubles, ListOfLong)->GEOM_Object_ptr
209   Chamfer4 = geompy.MakeChamferFacesAD(Prism, d1, 0.2,
210                                        IDlist_f) #(GEOM_Object_ptr, 2 Doubles, ListOfLong)->GEOM_Object_ptr
211
212   #Create Patterns
213   MultiTrans1D = geompy.MakeMultiTranslation1D(Fillet, vz, step1, nbtimes1)
214   MultiTrans2D = geompy.MakeMultiTranslation2D(Fillet, vz, step1, nbtimes1, vy, step2, nbtimes2)
215   #!!!!Angle In Degree!!!!
216   MultiRot1D   = geompy.MultiRotate1D(Chamfer, vx, nbtimes1)
217   MultiRot2D   = geompy.MultiRotate2D(Chamfer, vx, angle, nbtimes1, step1, nbtimes2)
218
219   #Create Informations objects
220   CDG        = geompy.MakeCDG(Prism)               #(GEOM_Object_ptr)->GEOM_Object_ptr
221   Archimede  = geompy.Archimede(Box, weight, waterdensity,
222                                 meshingdeflection) #(GEOM_Object_ptr, 3 Doubles)->GEOM_Object_ptr
223   CheckShape = geompy.CheckShape(Prism)            #(GEOM_Object_ptr)->Boolean
224   print "CheckShape(Prism) = ", CheckShape
225
226   #Partition objects
227   Partition  = geompy.MakePartition([Box], [Plane]) #(2 Lists Of GEOM_Object_ptr)->GEOM_Object_ptr
228   Partition1 = geompy.MakeHalfPartition(Box, Plane) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
229
230   #Add In Study
231
232   id_p0   = geompy.addToStudy(p0,   "Vertex 0")
233   id_px   = geompy.addToStudy(px,   "Vertex X")
234   id_py   = geompy.addToStudy(py,   "Vertex Y")
235   id_pz   = geompy.addToStudy(pz,   "Vertex Z")
236   id_pxyz = geompy.addToStudy(pxyz, "Vertex XYZ")
237   id_p200 = geompy.addToStudy(p200, "Vertex 200")
238
239   id_vx   = geompy.addToStudy(vx,   "Vector X")
240   id_vy   = geompy.addToStudy(vy,   "Vector Y")
241   id_vz   = geompy.addToStudy(vz,   "Vector Z")
242   id_vxy  = geompy.addToStudy(vxy,  "Vector XY")
243   id_vxyz = geompy.addToStudy(vxyz, "Vector XYZ")
244
245   id_cs1 = geompy.addToStudy(cs1, "CS 50,50,50, 1,0,0, 0,1,0")
246   id_cs2 = geompy.addToStudy(cs2, "CS 70,80,10, 1,0,1, 1,1,0")
247
248   id_Line   = geompy.addToStudy(Line,   "Line")
249   id_Line1  = geompy.addToStudy(Line1,  "Line by point and vector")
250   id_Plane  = geompy.addToStudy(Plane,  "Plane")
251   id_Plane1 = geompy.addToStudy(Plane1,  "Plane by 3 points")
252
253   id_Arc      = geompy.addToStudy(Arc,      "Arc")
254   id_Circle   = geompy.addToStudy(Circle,   "Circle")
255   id_Circle1  = geompy.addToStudy(Circle1,  "Circle by 3 points")
256   id_Circle2  = geompy.addToStudy(Circle2,  "Circle by center and 2 points")
257   id_Ellipse  = geompy.addToStudy(Ellipse,  "Ellipse")
258   id_Polyline = geompy.addToStudy(Polyline, "Polyline")
259   id_Bezier   = geompy.addToStudy(Bezier,   "Bezier")
260   id_Interpol = geompy.addToStudy(Interpol, "Interpol")
261   id_Sketcher = geompy.addToStudy(Sketcher, "Sketcher")
262
263   id_p_on_arc = geompy.addToStudy(p_on_arc, "Vertex on Arc")
264
265   id_Box      = geompy.addToStudy(Box,      "Box")
266   id_Box1     = geompy.addToStudy(Box1,     "Box 10x20x30")
267   id_Box2     = geompy.addToStudy(Box2,     "Box (10,20,30)-(15,25,35)")
268   id_Cylinder = geompy.addToStudy(Cylinder, "Cylinder")
269   id_Cyl1     = geompy.addToStudy(Cyl1,     "Cylinder RH")
270   id_Sphere   = geompy.addToStudy(Sphere,   "Sphere Pnt R")
271   id_Sphere1  = geompy.addToStudy(Sphere1,  "Sphere R")
272   id_Sphere2  = geompy.addToStudy(Sphere2,  "Sphere")
273   id_Cone     = geompy.addToStudy(Cone,     "Cone")
274   id_Cone1    = geompy.addToStudy(Cone1,    "Cone R1R2H")
275   id_Torus    = geompy.addToStudy(Torus,    "Torus")
276   id_Torus1   = geompy.addToStudy(Torus1,   "Torus RR")
277
278   id_Common  = geompy.addToStudy(Common,  "Common")
279   id_Cut     = geompy.addToStudy(Cut,     "Cut")
280   id_Fuse    = geompy.addToStudy(Fuse,    "Fuse")
281   id_Section = geompy.addToStudy(Section, "Section")
282
283   id_Edge     = geompy.addToStudy(Edge,     "Edge")
284   id_Wire     = geompy.addToStudy(Wire,     "Wire")
285   id_Face     = geompy.addToStudy(Face,     "Face")
286   id_Face1    = geompy.addToStudy(Face1,    "Face from two wires")
287   id_Face2    = geompy.addToStudy(Face2,    "Face from Sketcher")
288   id_Shell    = geompy.addToStudy(Shell,    "Shell")
289
290   id_Prism1   = geompy.addToStudy(Prism1,     "Prism by Two Pnt")
291   id_Shell1   = geompy.addToStudy(Shell1,   "Shell from Prism1 faces")
292   id_Solid    = geompy.addToStudy(Solid,    "Solid")
293   id_Compound = geompy.addToStudy(Compound, "Compound")
294
295   id_Plane2   = geompy.addToStudy(Plane2,   "Plane on Face")
296
297   id_Copy       = geompy.addToStudy(Copy,       "Copy")
298   id_Prism      = geompy.addToStudy(Prism,      "Prism")
299   id_Revolution = geompy.addToStudy(Revolution, "Revolution")
300   id_Filling    = geompy.addToStudy(Filling,    "Filling")
301   id_Pipe       = geompy.addToStudy(Pipe,       "Pipe")
302   id_Sewing     = geompy.addToStudy(Sewing,     "Sewing")
303
304   id_Translation = geompy.addToStudy(Translation, "Translation")
305   id_TranslVect  = geompy.addToStudy(TranslVect , "Translation along vector")
306   id_Rotation    = geompy.addToStudy(Rotation,    "Rotation")
307   id_RotatPnt    = geompy.addToStudy(RotatPnt,    "Rotation by three points")
308   id_Scale       = geompy.addToStudy(Scale,       "Scale")
309   id_Mirror      = geompy.addToStudy(Mirror,      "Mirror by Plane")
310   id_MirrorAxis  = geompy.addToStudy(MirrorAxis,  "Mirror by Axis")
311   id_MirrorPnt   = geompy.addToStudy(MirrorPnt,   "Mirror by Point")
312   id_Position    = geompy.addToStudy(Position,    "Positioned box")
313   id_Offset      = geompy.addToStudy(Offset,      "Offset")
314   id_Orientation = geompy.addToStudy(Orientation, "Orientation")
315
316   id_Fillet   = geompy.addToStudy(Fillet,   "Fillet")
317   id_Fillet2  = geompy.addToStudy(Fillet2,  "Fillet2")
318   id_Chamfer  = geompy.addToStudy(Chamfer,  "Chamfer on Edge")
319   id_Chamfer2 = geompy.addToStudy(Chamfer2, "Chamfer on Faces")
320   id_Chamfer3 = geompy.addToStudy(Chamfer3, "Chamfer on Edges")
321   id_Chamfer4 = geompy.addToStudy(Chamfer4, "Chamfer on Faces with params D Angle")
322
323   id_MultiTrans1D = geompy.addToStudy(MultiTrans1D, "MultiTrans1D")
324   id_MultiTrans2D = geompy.addToStudy(MultiTrans2D, "MultiTrans2D")
325   id_MultiRot1D   = geompy.addToStudy(MultiRot1D,   "MultiRot1D")
326   id_MultiRot2D   = geompy.addToStudy(MultiRot2D,   "MultiRot2D")
327
328   id_CDG       = geompy.addToStudy(CDG,       "CDG")
329   id_Archimede = geompy.addToStudy(Archimede, "Archimede")
330
331   id_Partition  = geompy.addToStudy(Partition, "Partition")
332   id_Partition1 = geompy.addToStudy(Partition1, "Half Partition")
333
334   #Decompose objects
335   SubFace    = geompy.SubShape(Box, geompy.ShapeType["FACE"], [2])
336   name       = geompy.SubShapeName(SubFace, Box)
337   id_SubFace = geompy.addToStudyInFather(Box, SubFace, name)
338
339   SubFaceS   = geompy.SubShapeSorted(Box, geompy.ShapeType["FACE"], [5])
340   nameS      = geompy.SubShapeName(SubFaceS, Box)
341   id_SubFace = geompy.addToStudyInFather(Box, SubFaceS, nameS)
342
343   SubEdgeList = geompy.SubShapeAll(SubFace, geompy.ShapeType["EDGE"])
344   i=0
345   for SubEdge in SubEdgeList :
346     name = geompy.SubShapeName(SubEdge, SubFace)
347     id_SubEdge = geompy.addToStudyInFather(SubFace, SubEdge, name)
348
349   #-------------------
350
351   print "DONE"