Salome HOME
Merge from V6_main 01/04/2013
[modules/smesh.git] / src / SMESH_SWIG / SMESH_mechanic_netgen.py
1 #  -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2013  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
24 # Quadrangulation of the geometry generated by the Python script
25 # SMESH_mechanic.py
26 # The new Netgen algorithm is used that discretizes baoundaries itself
27 #
28 import salome
29 import geompy
30
31 geom  = geompy.geom
32
33 import smesh
34
35 # ---------------------------- GEOM --------------------------------------
36
37 # ---- define contigous arcs and segment to define a closed wire
38 p1   = geompy.MakeVertex( 100.0,   0.0,  0.0 )
39 p2   = geompy.MakeVertex(  50.0,  50.0,  0.0 )
40 p3   = geompy.MakeVertex( 100.0, 100.0,  0.0 )
41 arc1 = geompy.MakeArc( p1, p2, p3 )
42
43 p4   = geompy.MakeVertex( 170.0, 100.0, 0.0 )
44 seg1 = geompy.MakeVector( p3, p4 )
45
46 p5   = geompy.MakeVertex( 200.0, 70.0, 0.0 )
47 p6   = geompy.MakeVertex( 170.0, 40.0, 0.0 )
48 arc2 = geompy.MakeArc( p4, p5, p6 )
49
50 p7   = geompy.MakeVertex( 120.0, 30.0, 0.0 )
51 arc3 = geompy.MakeArc( p6, p7, p1 )
52
53 # ---- define a closed wire with arcs and segment
54 List1 = []
55 List1.append( arc1 )
56 List1.append( seg1 )
57 List1.append( arc2 )
58 List1.append( arc3 )
59
60 wire1 = geompy.MakeWire( List1 )
61 Id_wire1 = geompy.addToStudy( wire1, "wire1" )
62
63 # ---- define a planar face with wire
64 WantPlanarFace = 1 #True
65 face1 = geompy.MakeFace( wire1, WantPlanarFace )
66 Id_face1 = geompy.addToStudy( face1, "face1" )
67
68 # ---- create a shape by extrusion
69 pO = geompy.MakeVertex( 0.0, 0.0,   0.0 )
70 pz = geompy.MakeVertex( 0.0, 0.0, 100.0 )
71 vz = geompy.MakeVector( pO, pz )
72
73 prism1 = geompy.MakePrismVecH( face1, vz, 100.0 )
74 Id_prism1 = geompy.addToStudy( prism1, "prism1")
75
76 # ---- create two cylinders
77
78 pc1 = geompy.MakeVertex(  90.0, 50.0, -40.0 )
79 pc2 = geompy.MakeVertex( 170.0, 70.0, -40.0 )
80 radius = 20.0
81 height = 180.0
82 cyl1  = geompy.MakeCylinder( pc1, vz, radius, height )
83 cyl2  = geompy.MakeCylinder( pc2, vz, radius, height )
84
85 Id_Cyl1 = geompy.addToStudy( cyl1, "cyl1" )
86 Id_Cyl2 = geompy.addToStudy( cyl2, "cyl2" )
87
88 # ---- cut with cyl1
89 shape  = geompy.MakeBoolean( prism1, cyl1, 2 )
90
91 # ---- fuse with cyl2 to obtain the final mechanic piece :)
92 mechanic =  geompy.MakeBoolean( shape, cyl2, 3 )
93 Id_mechanic = geompy.addToStudy( mechanic, "mechanic" )
94
95 # ---- Analysis of the geometry
96
97 print "Analysis of the geometry mechanic :"
98
99 subShellList = geompy.SubShapeAll(mechanic,geompy.ShapeType["SHELL"])
100 subFaceList  = geompy.SubShapeAll(mechanic,geompy.ShapeType["FACE"])
101 subEdgeList  = geompy.SubShapeAll(mechanic,geompy.ShapeType["EDGE"])
102
103 print "number of Shells in mechanic : ",len(subShellList)
104 print "number of Faces in mechanic : ",len(subFaceList)
105 print "number of Edges in mechanic : ",len(subEdgeList)
106
107 ### ---------------------------- SMESH --------------------------------------
108 smesh.SetCurrentStudy(salome.myStudy)
109
110 print "-------------------------- create Mesh, algorithm, hypothesis"
111
112 mesh = smesh.Mesh(mechanic, "Mesh_mechanic");
113 netgen = mesh.Triangle(smesh.NETGEN)
114 netgen.SetMaxSize( 50 )
115 #netgen.SetSecondOrder( 0 )
116 netgen.SetFineness( smesh.Fine )
117 netgen.SetQuadAllowed( 1 )
118 #netgen.SetOptimize( 1 )
119
120 salome.sg.updateObjBrowser(1)
121
122 print "-------------------------- compute mesh"
123 ret = mesh.Compute()
124 print ret
125 if ret != 0:
126     print "Information about the MeshcompShel:"
127     print "Number of nodes        : ", mesh.NbNodes()
128     print "Number of edges        : ", mesh.NbEdges()
129     print "Number of faces        : ", mesh.NbFaces()
130     print "Number of triangles    : ", mesh.NbTriangles()
131     print "Number of quadrangles  : ", mesh.NbQuadrangles()
132     print "Number of volumes      : ", mesh.NbVolumes()
133     print "Number of tetrahedrons : ", mesh.NbTetras()
134     
135 else:
136     print "problem when computing the mesh"