1 # GEOM GEOM_SWIG : binding of C++ omplementaion with Python
3 # Copyright (C) 2003 CEA
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.
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.
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
20 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 # File : GEOM_usinggeom.py
24 # Author : Damien COQUERET, Open CASCADE
29 def TestAll (geompy, math):
31 #Create base Variables
37 ShapeTypeFace = geompy.ShapeType["FACE"]
38 ShapeTypeEdge = geompy.ShapeType["EDGE"]
39 WantPlanarFace = 1 #True
50 angle1 = angle * math.pi / 180
57 meshingdeflection = 0.01
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.)
68 p200 = geompy.MakeVertexWithRef(pxyz, 100., 100., 100.) #(GEOM_Object_ptr, 3 Doubles)->GEOM_Object_ptr
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)
76 vxyz = geompy.MakeVectorDXDYDZ(100., 100., 100.) #(3 Doubles)->GEOM_Object_ptr
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)
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 Line2 = geompy.MakeLineTwoPnt(pxyz, pz) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
86 Plane = geompy.MakePlane(pz, vxyz, trimsize) #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
87 Plane1 = geompy.MakePlaneThreePnt(px, pz, p200, trimsize) #(4 Doubles)->GEOM_Object_ptr
89 Arc = geompy.MakeArc(py, pz, px) #(3 GEOM_Object_ptr)->GEOM_Object_ptr
90 Arc2 = geompy.MakeArcCenter(py, pz, px,0) #(3 GEOM_Object_ptr,Boolean)->GEOM_Object_ptr
91 Circle = geompy.MakeCircle(p0, vz, radius1) #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
92 Circle1 = geompy.MakeCircleThreePnt(p0, pxyz, px) #(3 GEOM_Object_ptr)->GEOM_Object_ptr
93 Circle2 = geompy.MakeCircleCenter2Pnt(p0, pxyz, py) #(3 GEOM_Object_ptr)->GEOM_Object_ptr
94 Ellipse = geompy.MakeEllipse(p0, vy, radius2, radius1) #(2 GEOM_Object_ptr, 2 Doubles)->GEOM_Object_ptr
95 Polyline = geompy.MakePolyline([p0, pz, py, p200]) #(List of GEOM_Object_ptr)->GEOM_Object_ptr
96 Bezier = geompy.MakeBezier([p0, pz, p200, px]) #(List of GEOM_Object_ptr)->GEOM_Object_ptr
97 Interpol = geompy.MakeInterpol([px, py, p200, pxyz]) #(List of GEOM_Object_ptr)->GEOM_Object_ptr
98 Sketcher = geompy.MakeSketcher("Sketcher:F -100 -100:TT 250 -100:R 0:C 100 150:R 0:L 300:WW",
99 [100,0,0, 1,1,1, -1,1,0]) #(String, List of Doubles)->GEOM_Object_ptr
101 #Test point on curve creation
102 p_on_arc = geompy.MakeVertexOnCurve(Arc, 0.25) #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
104 #Test point on lines intersection
105 pLine = geompy.MakeVertexOnLinesIntersection( Line1, Line2 )
107 #Create base geometry 3D
108 Box = geompy.MakeBoxTwoPnt(p0, p200) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
109 Box1 = geompy.MakeBoxDXDYDZ(10, 20, 30) #(3 Doubles)->GEOM_Object_ptr
110 Box2 = geompy.MakeBox(10,20,30, 15,25,35) #(6 Doubles)->GEOM_Object_ptr
111 Cylinder = geompy.MakeCylinder(p0, vz, radius1, height) #(2 GEOM_Object_ptr, 2 Doubles)->GEOM_Object_ptr
112 Cyl1 = geompy.MakeCylinderRH(radius2, height) #(2 Doubles)->GEOM_Object_ptr
113 Sphere = geompy.MakeSpherePntR(p0, radius1) #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
114 Sphere1 = geompy.MakeSphereR(radius) #(Double)->GEOM_Object_ptr
115 Sphere2 = geompy.MakeSphere(50, 70, 30, radius) #(4 Doubles)->GEOM_Object_ptr
116 Cone = geompy.MakeCone(p0, vz, radius2, radius, height) #(2 GEOM_Object_ptr, 3 Doubles)->GEOM_Object_ptr
117 Cone1 = geompy.MakeConeR1R2H(radius1, radius, height) #(3 Doubles)->GEOM_Object_ptr
118 Torus = geompy.MakeTorus(p0, vz, radius2, radius) #(2 GEOM_Object_ptr, 2 Doubles)->GEOM_Object_ptr
119 Torus1 = geompy.MakeTorusRR(radius2, radius1) #(2 Doubles)->GEOM_Object_ptr
121 #Boolean (Common, Cut, Fuse, Section)
122 Common = geompy.MakeBoolean(Box, Sphere, 1) #(2 GEOM_Object_ptr, Short)->GEOM_Object_ptr
123 Cut = geompy.MakeBoolean(Box, Sphere, 2)
124 Fuse = geompy.MakeBoolean(Box, Sphere, 3)
125 Section = geompy.MakeBoolean(Box, Sphere, 4)
128 Edge = geompy.MakeEdge(p0, pxyz) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
129 Wire = geompy.MakeWire([vxy, Arc]) #(List Of GEOM_Object_ptr)->GEOM_Object_ptr
130 Face = geompy.MakeFace(Wire, WantPlanarFace) #(GEOM_Object_ptr, Boolean)->GEOM_Object_ptr
131 Face1 = geompy.MakeFaceWires([Wire, Sketcher],
132 WantPlanarFace) #(List of GEOM_Object_ptr, Boolean)->GEOM_Object_ptr
133 Face2 = geompy.MakeFace(Sketcher, WantPlanarFace)
134 Shell = geompy.MakeShell([Face, Face1]) #(List of GEOM_Object_ptr)->GEOM_Object_ptr
136 Prism1 = geompy.MakePrism(Face2, p0, pxyz) #(3 GEOM_Object_ptr)->GEOM_Object_ptr
137 prism1_faces = geompy.SubShapeAllSorted(Prism1, ShapeTypeFace)
138 Shell1 = geompy.MakeShell([prism1_faces[0], prism1_faces[1],
139 prism1_faces[3], prism1_faces[4],
140 prism1_faces[5], prism1_faces[2]])
141 Solid = geompy.MakeSolid([Shell1]) #(List of GEOM_Object_ptr)->GEOM_Object_ptr
143 ShapeListCompound = []
146 S = geompy.MakeTranslation(Arc, i * 100., i * 100., i * 100.)
147 ShapeListCompound.append(S)
149 Compound = geompy.MakeCompound(ShapeListCompound) #(List of GEOM_Object_ptr)->GEOM_Object_ptr
151 # Test plane from existing face creation
152 Plane2 = geompy.MakePlaneFace(Face, trimsize) #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
154 #ShapeList for Sewing
155 S = geompy.MakeRotation(Face, vxy, angle1)
157 #Test Line on Faces Intersection
158 Line3 = geompy.MakeLineTwoFaces( prism1_faces[0], prism1_faces[1]) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
160 #Create advanced objects
161 Copy = geompy.MakeCopy(Box) #(GEOM_Object_ptr)->GEOM_Object_ptr
162 Prism = geompy.MakePrismVecH(Face, vz, 100.0) #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
163 Prism2Ways = geompy.MakePrismVecH2Ways(Face, vz, 10.0) #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
164 Revolution = geompy.MakeRevolution(Face, vz, angle2) #
165 Revolution2Ways = geompy.MakeRevolution(Face, vz, angle1) #
166 Filling = geompy.MakeFilling(Compound, mindeg, maxdeg,
167 tol2d, tol3d, nbiter) #(GEOM_Object_ptr, 4 Doubles, Short)->GEOM_Object_ptr
168 Pipe = geompy.MakePipe(Wire, Edge) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
169 Sewing = geompy.MakeSewing([Face, S], precision) #(List Of GEOM_Object_ptr, Double)->GEOM_Object_ptr
172 Translation = geompy.MakeTranslationTwoPoints(Box, px, pz) #(3 GEOM_Object_ptr)->GEOM_Object_ptr
173 TranslVect = geompy.MakeTranslationVector(Box, vxyz) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
174 Rotation = geompy.MakeRotation(Box, vz, angle1) #(2 GEOM_Object_ptr, Double)->GEOM_Object_ptr
175 RotatPnt = geompy.MakeRotationThreePoints(Box, px, py, pz) #(4 GEOM_Object_ptr)->GEOM_Object_ptr
176 Scale = geompy.MakeScaleTransform(Box, p0, factor) #
177 Mirror = geompy.MakeMirrorByPlane(Box, Plane) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
178 MirrorAxis = geompy.MakeMirrorByAxis(Box, Line1) #
179 MirrorPnt = geompy.MakeMirrorByPoint(Box, p200) #
180 Position = geompy.MakePosition(Box, cs1, cs2) #(3 GEOM_Object_ptr)->GEOM_Object_ptr
181 Offset = geompy.MakeOffset(Box, 10.) #(GEOM_Object_ptr, Double)->GEOM_Object_ptr
182 Orientation = geompy.ChangeOrientation(Box)
184 #IDList for Fillet/Chamfer
185 prism_edges = geompy.SubShapeAllSorted(Prism, ShapeTypeEdge)
187 for anEdge in prism_edges:
188 eid = geompy.GetSubShapeID(Prism, anEdge)
189 sse = geompy.GetSubShape(Prism, [eid])
191 sse_id = geompy.GetSubShapeID(Prism, sse)
193 print "Error: GetSubShape() or GetSubShapeID() has failed!"
196 IDlist_e.append(geompy.GetSubShapeID(Prism, prism_edges[0]))
197 IDlist_e.append(geompy.GetSubShapeID(Prism, prism_edges[1]))
198 IDlist_e.append(geompy.GetSubShapeID(Prism, prism_edges[2]))
200 prism_faces = geompy.SubShapeAllSorted(Prism, ShapeTypeFace)
202 f_ind_1 = geompy.GetSubShapeID(Prism, prism_faces[0])
203 f_ind_2 = geompy.GetSubShapeID(Prism, prism_faces[1])
205 IDlist_f = [f_ind_1, f_ind_2]
208 Fillet = geompy.MakeFillet (Prism, radius, ShapeTypeEdge,
209 IDlist_e) #(GEOM_Object_ptr, Double, Short, ListOfLong)->GEOM_Object_ptr
210 Fillet2 = geompy.MakeFilletR1R2 (Prism, 7., 13., ShapeTypeEdge,
211 IDlist_e) #(GEOM_Object_ptr, Double, Double, Short, ListOfLong)->GEOM_Object_ptr
212 Chamfer = geompy.MakeChamferEdge(Prism, d1, d2,
213 f_ind_1, f_ind_2) #(GEOM_Object_ptr, 2 Doubles, 2 Long)->GEOM_Object_ptr
214 Chamfer2 = geompy.MakeChamferFaces(Prism, d1, d2,
215 IDlist_f) #(GEOM_Object_ptr, 2 Doubles, ListOfLong)->GEOM_Object_ptr
216 Chamfer3 = geompy.MakeChamferEdges(Prism, d1, d2,
217 IDlist_e) #(GEOM_Object_ptr, 2 Doubles, ListOfLong)->GEOM_Object_ptr
218 Chamfer4 = geompy.MakeChamferFacesAD(Prism, d1, 20. * math.pi / 180.,
219 IDlist_f) #(GEOM_Object_ptr, 2 Doubles, ListOfLong)->GEOM_Object_ptr
221 MultiTrans1D = geompy.MakeMultiTranslation1D(Fillet, vz, step1, nbtimes1)
222 MultiTrans2D = geompy.MakeMultiTranslation2D(Fillet, vz, step1, nbtimes1, vy, step2, nbtimes2)
223 #!!!!Angle In Degree!!!!
224 MultiRot1D = geompy.MultiRotate1D(Chamfer, vx, nbtimes1)
225 MultiRot2D = geompy.MultiRotate2D(Chamfer, vx, angle, nbtimes1, step1, nbtimes2)
227 #Create Informations objects
228 CDG = geompy.MakeCDG(Prism) #(GEOM_Object_ptr)->GEOM_Object_ptr
229 Archimede = geompy.Archimede(Box, weight, waterdensity,
230 meshingdeflection) #(GEOM_Object_ptr, 3 Doubles)->GEOM_Object_ptr
231 CheckShape = geompy.CheckShape(Prism) #(GEOM_Object_ptr)->Boolean
232 print "CheckShape(Prism) = ", CheckShape
235 Partition = geompy.MakePartition([Box], [Plane]) #(2 Lists Of GEOM_Object_ptr)->GEOM_Object_ptr
236 Partition1 = geompy.MakeHalfPartition(Box, Plane) #(2 GEOM_Object_ptr)->GEOM_Object_ptr
240 id_p0 = geompy.addToStudy(p0, "Vertex 0")
241 id_px = geompy.addToStudy(px, "Vertex X")
242 id_py = geompy.addToStudy(py, "Vertex Y")
243 id_pz = geompy.addToStudy(pz, "Vertex Z")
244 id_pxyz = geompy.addToStudy(pxyz, "Vertex XYZ")
245 id_p200 = geompy.addToStudy(p200, "Vertex 200")
246 id_pLine = geompy.addToStudy(pLine, "Vertex on Lines Intersection")
248 id_vx = geompy.addToStudy(vx, "Vector X")
249 id_vy = geompy.addToStudy(vy, "Vector Y")
250 id_vz = geompy.addToStudy(vz, "Vector Z")
251 id_vxy = geompy.addToStudy(vxy, "Vector XY")
252 id_vxyz = geompy.addToStudy(vxyz, "Vector XYZ")
254 id_cs1 = geompy.addToStudy(cs1, "CS 50,50,50, 1,0,0, 0,1,0")
255 id_cs2 = geompy.addToStudy(cs2, "CS 70,80,10, 1,0,1, 1,1,0")
257 id_Line = geompy.addToStudy(Line, "Line")
258 id_Line1 = geompy.addToStudy(Line1, "Line by point and vector")
259 id_Line3 = geompy.addToStudy(Line3, "Line on Two Faces Intersection")
260 id_Plane = geompy.addToStudy(Plane, "Plane")
261 id_Plane1 = geompy.addToStudy(Plane1, "Plane by 3 points")
263 id_Arc = geompy.addToStudy(Arc, "Arc")
264 id_Circle = geompy.addToStudy(Circle, "Circle")
265 id_Circle1 = geompy.addToStudy(Circle1, "Circle by 3 points")
266 id_Circle2 = geompy.addToStudy(Circle2, "Circle by center and 2 points")
267 id_Ellipse = geompy.addToStudy(Ellipse, "Ellipse")
268 id_Polyline = geompy.addToStudy(Polyline, "Polyline")
269 id_Bezier = geompy.addToStudy(Bezier, "Bezier")
270 id_Interpol = geompy.addToStudy(Interpol, "Interpol")
271 id_Sketcher = geompy.addToStudy(Sketcher, "Sketcher")
273 id_p_on_arc = geompy.addToStudy(p_on_arc, "Vertex on Arc")
275 id_Box = geompy.addToStudy(Box, "Box")
276 id_Box1 = geompy.addToStudy(Box1, "Box 10x20x30")
277 id_Box2 = geompy.addToStudy(Box2, "Box (10,20,30)-(15,25,35)")
278 id_Cylinder = geompy.addToStudy(Cylinder, "Cylinder")
279 id_Cyl1 = geompy.addToStudy(Cyl1, "Cylinder RH")
280 id_Sphere = geompy.addToStudy(Sphere, "Sphere Pnt R")
281 id_Sphere1 = geompy.addToStudy(Sphere1, "Sphere R")
282 id_Sphere2 = geompy.addToStudy(Sphere2, "Sphere")
283 id_Cone = geompy.addToStudy(Cone, "Cone")
284 id_Cone1 = geompy.addToStudy(Cone1, "Cone R1R2H")
285 id_Torus = geompy.addToStudy(Torus, "Torus")
286 id_Torus1 = geompy.addToStudy(Torus1, "Torus RR")
288 id_Common = geompy.addToStudy(Common, "Common")
289 id_Cut = geompy.addToStudy(Cut, "Cut")
290 id_Fuse = geompy.addToStudy(Fuse, "Fuse")
291 id_Section = geompy.addToStudy(Section, "Section")
293 id_Edge = geompy.addToStudy(Edge, "Edge")
294 id_Wire = geompy.addToStudy(Wire, "Wire")
295 id_Face = geompy.addToStudy(Face, "Face")
296 id_Face1 = geompy.addToStudy(Face1, "Face from two wires")
297 id_Face2 = geompy.addToStudy(Face2, "Face from Sketcher")
298 id_Shell = geompy.addToStudy(Shell, "Shell")
300 id_Prism1 = geompy.addToStudy(Prism1, "Prism by Two Pnt")
301 id_Shell1 = geompy.addToStudy(Shell1, "Shell from Prism1 faces")
302 id_Solid = geompy.addToStudy(Solid, "Solid")
303 id_Compound = geompy.addToStudy(Compound, "Compound")
305 id_Plane2 = geompy.addToStudy(Plane2, "Plane on Face")
307 id_Copy = geompy.addToStudy(Copy, "Copy")
308 id_Prism = geompy.addToStudy(Prism, "Prism")
309 id_Prism2Ways = geompy.addToStudy(Prism2Ways, "Prism2Ways")
310 id_Revolution = geompy.addToStudy(Revolution, "Revolution")
311 id_Revolution2Ways = geompy.addToStudy(Revolution2Ways, "Revolution2Ways")
312 id_Filling = geompy.addToStudy(Filling, "Filling")
313 id_Pipe = geompy.addToStudy(Pipe, "Pipe")
314 id_Sewing = geompy.addToStudy(Sewing, "Sewing")
316 id_Translation = geompy.addToStudy(Translation, "Translation")
317 id_TranslVect = geompy.addToStudy(TranslVect , "Translation along vector")
318 id_Rotation = geompy.addToStudy(Rotation, "Rotation")
319 id_RotatPnt = geompy.addToStudy(RotatPnt, "Rotation by three points")
320 id_Scale = geompy.addToStudy(Scale, "Scale")
321 id_Mirror = geompy.addToStudy(Mirror, "Mirror by Plane")
322 id_MirrorAxis = geompy.addToStudy(MirrorAxis, "Mirror by Axis")
323 id_MirrorPnt = geompy.addToStudy(MirrorPnt, "Mirror by Point")
324 id_Position = geompy.addToStudy(Position, "Positioned box")
325 id_Offset = geompy.addToStudy(Offset, "Offset")
326 id_Orientation = geompy.addToStudy(Orientation, "Orientation")
328 id_Fillet = geompy.addToStudy(Fillet, "Fillet")
329 id_Fillet2 = geompy.addToStudy(Fillet2, "Fillet2")
331 id_Chamfer = geompy.addToStudy(Chamfer, "Chamfer on Edge")
332 id_Chamfer2 = geompy.addToStudy(Chamfer2, "Chamfer on Faces")
333 id_Chamfer3 = geompy.addToStudy(Chamfer3, "Chamfer on Edges")
334 id_Chamfer4 = geompy.addToStudy(Chamfer4, "Chamfer on Faces with params D Angle")
336 id_MultiTrans1D = geompy.addToStudy(MultiTrans1D, "MultiTrans1D")
337 id_MultiTrans2D = geompy.addToStudy(MultiTrans2D, "MultiTrans2D")
338 id_MultiRot1D = geompy.addToStudy(MultiRot1D, "MultiRot1D")
339 id_MultiRot2D = geompy.addToStudy(MultiRot2D, "MultiRot2D")
341 id_CDG = geompy.addToStudy(CDG, "CDG")
342 id_Archimede = geompy.addToStudy(Archimede, "Archimede")
344 id_Partition = geompy.addToStudy(Partition, "Partition")
345 id_Partition1 = geompy.addToStudy(Partition1, "Half Partition")
348 SubFace = geompy.SubShape(Box, geompy.ShapeType["FACE"], [2])
349 name = geompy.SubShapeName(SubFace, Box)
350 id_SubFace = geompy.addToStudyInFather(Box, SubFace, name)
352 SubFaceS = geompy.SubShapeSorted(Box, geompy.ShapeType["FACE"], [5])
353 nameS = geompy.SubShapeName(SubFaceS, Box)
354 id_SubFace = geompy.addToStudyInFather(Box, SubFaceS, nameS)
356 SubEdgeList = geompy.SubShapeAll(SubFace, geompy.ShapeType["EDGE"])
358 for SubEdge in SubEdgeList :
359 name = geompy.SubShapeName(SubEdge, SubFace)
360 id_SubEdge = geompy.addToStudyInFather(SubFace, SubEdge, name)