Salome HOME
bos #29171 Refactor testing procedure
[modules/smesh.git] / doc / examples / creating_meshes_ex06.py
1 # Creating a hexahedral mesh on a cylinder.
2 #
3 # This example uses Partition to divide the cylinder into blocks, which is
4 # a general approach. But for the case of cylinder there is a dedicated
5 # command creating a blocked cylinder: geom_builder.MakeDividedCylinder()
6
7 import math
8
9 import salome
10 salome.salome_init_without_session()
11
12 import GEOM
13 from salome.geom import geomBuilder
14 from salome.smesh import smeshBuilder
15
16 geom_builder = geomBuilder.New()
17 smesh_builder = smeshBuilder.New()
18
19 # Parameters
20 # ----------
21
22 radius =  50
23 height = 200
24
25 # Build a cylinder
26 # ----------------
27
28 base      = geom_builder.MakeVertex(0, 0, 0)
29 direction = geom_builder.MakeVectorDXDYDZ(0, 0, 1)
30
31 cylinder = geom_builder.MakeCylinder(base, direction, radius, height)
32
33 geom_builder.addToStudy(cylinder, "cylinder")
34
35 # Build blocks
36 # ------------
37
38 size = radius/2.0
39
40 box_rot  = geom_builder.MakeBox(-size, -size, 0,  +size, +size, height)
41 box_axis = geom_builder.MakeLine(base, direction)
42 box      = geom_builder.MakeRotation(box_rot, box_axis, math.pi/4)
43
44 hole = geom_builder.MakeCut(cylinder, box)
45
46 plane_trim = 2000
47
48 plane_a = geom_builder.MakePlane(base, geom_builder.MakeVectorDXDYDZ(1, 0, 0), plane_trim)
49 plane_b = geom_builder.MakePlane(base, geom_builder.MakeVectorDXDYDZ(0, 1, 0), plane_trim)
50
51 blocks_part = geom_builder.MakePartition([hole], [plane_a, plane_b], [], [], geom_builder.ShapeType["SOLID"])
52 blocks_list = [box] + geom_builder.SubShapeAll(blocks_part, geom_builder.ShapeType["SOLID"])
53 blocks_all  = geom_builder.MakeCompound(blocks_list)
54 blocks      = geom_builder.MakeGlueFaces(blocks_all, 0.0001)
55
56 geom_builder.addToStudy(blocks, "cylinder:blocks")
57
58 # Build geometric groups
59 # ----------------------
60
61 group_a = geom_builder.CreateGroup(blocks, geom_builder.ShapeType["FACE"])
62 geom_builder.addToStudyInFather(blocks, group_a, "baseA")
63 items = geom_builder.GetShapesOnPlaneWithLocationIDs(blocks, geom_builder.ShapeType["FACE"], direction, base, GEOM.ST_ON)
64 geom_builder.UnionIDs(group_a, items)
65
66 base_b = geom_builder.MakeVertex(0, 0, height)
67 group_b = geom_builder.CreateGroup(blocks, geom_builder.ShapeType["FACE"])
68 geom_builder.addToStudyInFather(blocks, group_b, "baseB")
69 items = geom_builder.GetShapesOnPlaneWithLocationIDs(blocks, geom_builder.ShapeType["FACE"], direction, base_b, GEOM.ST_ON)
70 geom_builder.UnionIDs(group_b, items)
71
72 group_1 = geom_builder.CreateGroup(blocks, geom_builder.ShapeType["SOLID"])
73 geom_builder.addToStudyInFather(blocks, group_1, "limit")
74 group_1_all = geom_builder.SubShapeAllIDs(blocks, geom_builder.ShapeType["SOLID"])
75 geom_builder.UnionIDs(group_1, group_1_all)
76 group_1_box = geom_builder.GetBlockNearPoint(blocks, base)
77 geom_builder.DifferenceList(group_1, [group_1_box])
78
79 # Mesh the blocks with hexahedral
80 # -------------------------------
81
82 smesh_builder.UpdateStudy()
83
84 hexa = smesh_builder.Mesh(blocks)
85
86 hexa_1d = hexa.Segment()
87 hexa_1d.NumberOfSegments(1)
88
89 vertex = geom_builder.MakeVertex(+radius, +radius, 0)
90 edge = geom_builder.GetEdgeNearPoint(blocks, vertex)
91 algo = hexa.Segment(edge)
92 algo.NumberOfSegments(5)
93 algo.Propagation()
94
95 vertex = geom_builder.MakeVertex(-radius, +radius, 0)
96 edge = geom_builder.GetEdgeNearPoint(blocks, vertex)
97 algo = hexa.Segment(edge)
98 algo.NumberOfSegments(8)
99 algo.Propagation()
100
101 vertex = geom_builder.MakeVertex((radius+size)/2, 0, 0)
102 edge = geom_builder.GetEdgeNearPoint(blocks, vertex)
103 algo = hexa.Segment(edge)
104 algo.NumberOfSegments(10)
105 algo.Propagation()
106
107 vertex = geom_builder.MakeVertex(+radius, 0, height/2)
108 edge = geom_builder.GetEdgeNearPoint(blocks, vertex)
109 algo = hexa.Segment(edge)
110 algo.NumberOfSegments(20)
111 algo.Propagation()
112
113 hexa.Quadrangle()
114 hexa.Hexahedron()
115
116 hexa.Compute()
117
118 hexa.Group(group_a)
119 hexa.Group(group_b)
120 hexa.Group(group_1)