Salome HOME
673569516d9c61f25c43d4e95503679057166edd
[modules/smesh.git] / doc / examples / modifying_meshes_ex23.py
1 # Extrusion along a Path
2
3 import math
4
5 import salome
6 salome.salome_init_without_session()
7
8 import SMESH
9 from salome.geom import geomBuilder
10 from salome.smesh import smeshBuilder
11
12 geom_builder = geomBuilder.New()
13 smesh_builder = smeshBuilder.New()
14
15 # 1. Create points
16 points = [[0, 0], [50, 30], [50, 110], [0, 150], [-80, 150], [-130, 70], [-130, -20]]
17
18 iv = 1
19 vertices = []
20 for point in points:
21     vert = geom_builder.MakeVertex(point[0], point[1], 0)
22     vertices.append(vert)
23     iv += 1
24
25 # 2. Create edges and wires
26 Edge_straight = geom_builder.MakeEdge(vertices[0], vertices[4])
27 Edge_bezierrr = geom_builder.MakeBezier(vertices)
28 Wire_polyline = geom_builder.MakePolyline(vertices)
29 Edge_Circle   = geom_builder.MakeCircleThreePnt(vertices[0], vertices[1], vertices[2])
30
31 geom_builder.addToStudy(Edge_straight, "Edge_straight")
32 geom_builder.addToStudy(Edge_bezierrr, "Edge_bezierrr")
33 geom_builder.addToStudy(Wire_polyline, "Wire_polyline")
34 geom_builder.addToStudy(Edge_Circle  , "Edge_Circle")
35
36 # 3. Explode wire on edges, as they will be used for mesh extrusion
37 Wire_polyline_edges = geom_builder.SubShapeAll(Wire_polyline, geom_builder.ShapeType["EDGE"])
38 for ii in range(len(Wire_polyline_edges)):
39     geom_builder.addToStudyInFather(Wire_polyline, Wire_polyline_edges[ii], "Edge_" + repr(ii + 1))
40
41 # Mesh
42
43 # Mesh the given shape with the given 1d hypothesis
44 def Mesh1D(shape1d, nbSeg, name):
45   mesh1d_tool = smesh_builder.Mesh(shape1d, name)
46   algo = mesh1d_tool.Segment()
47   hyp  = algo.NumberOfSegments(nbSeg)
48   isDone = mesh1d_tool.Compute()
49   if not isDone: print('Mesh ', name, ': computation failed')
50   return mesh1d_tool
51
52 # Create a mesh with six nodes, seven edges and two quadrangle faces
53 def MakeQuadMesh2(mesh_name):
54   quad_1 = smesh_builder.Mesh(name = mesh_name)
55   
56   # six nodes
57   n1 = quad_1.AddNode(0, 20, 10)
58   n2 = quad_1.AddNode(0, 40, 10)
59   n3 = quad_1.AddNode(0, 40, 30)
60   n4 = quad_1.AddNode(0, 20, 30)
61   n5 = quad_1.AddNode(0,  0, 30)
62   n6 = quad_1.AddNode(0,  0, 10)
63
64   # seven edges
65   quad_1.AddEdge([n1, n2]) # 1
66   quad_1.AddEdge([n2, n3]) # 2
67   quad_1.AddEdge([n3, n4]) # 3
68   quad_1.AddEdge([n4, n1]) # 4
69   quad_1.AddEdge([n4, n5]) # 5
70   quad_1.AddEdge([n5, n6]) # 6
71   quad_1.AddEdge([n6, n1]) # 7
72
73   # two quadrangle faces
74   quad_1.AddFace([n1, n2, n3, n4]) # 8
75   quad_1.AddFace([n1, n4, n5, n6]) # 9
76   return [quad_1, [1,2,3,4,5,6,7], [8,9]]
77
78 # Path meshes
79 Edge_straight_mesh = Mesh1D(Edge_straight, 7, "Edge_straight")
80 Edge_bezierrr_mesh = Mesh1D(Edge_bezierrr, 7, "Edge_bezierrr")
81 Wire_polyline_mesh = Mesh1D(Wire_polyline, 3, "Wire_polyline")
82 Edge_Circle_mesh   = Mesh1D(Edge_Circle  , 8, "Edge_Circle")
83
84 # Initial meshes (to be extruded)
85 [quad_1, ee_1, ff_1] = MakeQuadMesh2("quad_1")
86 [quad_2, ee_2, ff_2] = MakeQuadMesh2("quad_2")
87 [quad_3, ee_3, ff_3] = MakeQuadMesh2("quad_3")
88 [quad_4, ee_4, ff_4] = MakeQuadMesh2("quad_4")
89 [quad_5, ee_5, ff_5] = MakeQuadMesh2("quad_5")
90 [quad_6, ee_6, ff_6] = MakeQuadMesh2("quad_6")
91 [quad_7, ee_7, ff_7] = MakeQuadMesh2("quad_7")
92
93 # ExtrusionAlongPath
94 # IDsOfElements, PathMesh, PathShape, NodeStart,
95 # HasAngles, Angles, HasRefPoint, RefPoint
96 refPoint = SMESH.PointStruct(0, 0, 0)
97 a10 = math.radians( 10.0 )
98 a45 = math.radians( 45.0 )
99
100 # 1. Extrusion of two mesh edges along a straight path
101 error = quad_1.ExtrusionAlongPath([1,2], Edge_straight_mesh, Edge_straight, 1,
102                                   0, [], 0, refPoint)
103
104 # 2. Extrusion of one mesh edge along a curved path
105 error = quad_2.ExtrusionAlongPath([2], Edge_bezierrr_mesh, Edge_bezierrr, 1,
106                                   0, [], 0, refPoint)
107
108 # 3. Extrusion of one mesh edge along a curved path with usage of angles
109 error = quad_3.ExtrusionAlongPath([2], Edge_bezierrr_mesh, Edge_bezierrr, 1,
110                                   1, [a45, a45, a45, 0, -a45, -a45, -a45], 0, refPoint)
111
112 # 4. Extrusion of one mesh edge along the path, which is a part of a meshed wire
113 error = quad_4.ExtrusionAlongPath([4], Wire_polyline_mesh, Wire_polyline_edges[0], 1,
114                                   1, [a10, a10, a10], 0, refPoint)
115
116 # 5. Extrusion of two mesh faces along the path, which is a part of a meshed wire
117 error = quad_5.ExtrusionAlongPath(ff_5 , Wire_polyline_mesh, Wire_polyline_edges[2], 4,
118                                   0, [], 0, refPoint)
119
120 # 6. Extrusion of two mesh faces along a closed path
121 error = quad_6.ExtrusionAlongPath(ff_6 , Edge_Circle_mesh, Edge_Circle, 1,
122                                   0, [], 0, refPoint)
123
124 # 7. Extrusion of two mesh faces along a closed path with usage of angles
125 error = quad_7.ExtrusionAlongPath(ff_7, Edge_Circle_mesh, Edge_Circle, 1,
126                                   1, [a45, -a45, a45, -a45, a45, -a45, a45, -a45], 0, refPoint)
127
128
129
130 # Make the same meshes using a fully functional method ExtrusionAlongPathObjects() having
131 # the following arguments:
132 #   Nodes, Edges, Faces, PathObject, PathShape=None,
133 #   NodeStart=1, HasAngles=False, Angles=[], LinearVariation=False,
134 #   HasRefPoint=False, RefPoint=[0,0,0], MakeGroups=False,
135 #   ScaleFactors=[], ScalesVariation=False
136
137 quad_1 = MakeQuadMesh2("quad_1")[0]
138 quad_2 = MakeQuadMesh2("quad_2")[0]
139 quad_3 = MakeQuadMesh2("quad_3")[0]
140 quad_4 = MakeQuadMesh2("quad_4")[0]
141 quad_5 = MakeQuadMesh2("quad_5")[0]
142 quad_6 = MakeQuadMesh2("quad_6")[0]
143 quad_7 = MakeQuadMesh2("quad_7")[0]
144
145 # 1. Extrusion of two mesh edges along a straight path
146 nn, ee, ff = [], [1,2], []
147 error = quad_1.ExtrusionAlongPathObjects( nn, ee, ff, Edge_straight_mesh )
148
149 # 2. Extrusion of one mesh edge along a curved path
150 nn, ee, ff = [], [2], []
151 error = quad_2.ExtrusionAlongPathObjects( nn, ee, ff, Edge_bezierrr_mesh )
152
153 # 3. Extrusion of one mesh edge along a curved path with usage of angles
154 error = quad_3.ExtrusionAlongPathObjects( nn, ee, ff, Edge_bezierrr_mesh,
155                                           Angles=[a45, a45, a45, 0, -a45, -a45, -a45])
156
157 # 4. Extrusion of one mesh edge along the path, which is a part of a meshed wire
158 nn, ee, ff = [], [4], []
159 error = quad_4.ExtrusionAlongPathObjects( nn, ee, ff, Wire_polyline_mesh, Wire_polyline_edges[0],
160                                           Angles=[a10, a10, a10])
161
162 # 5. Extrusion of two mesh faces along the path, which is a part of a meshed wire
163 nn, ee, ff = [], [], quad_5
164 error = quad_5.ExtrusionAlongPathObjects( nn, ee, ff, Wire_polyline_mesh, Wire_polyline_edges[2],
165                                           NodeStart=4 )
166
167 # 6. Extrusion of two mesh faces along a closed path
168 nn, ee, ff = [], [], quad_6
169 error = quad_6.ExtrusionAlongPathObjects( nn, ee, ff, Edge_Circle_mesh )
170
171 # 7. Extrusion of two mesh faces along a closed path with usage of angles
172 nn, ee, ff = [], [], quad_7
173 error = quad_7.ExtrusionAlongPathObjects( nn, ee, ff, Edge_Circle_mesh, Edge_Circle,
174                                           Angles=[a45, -a45, a45, -a45, a45, -a45, a45, -a45])