1 # -*- coding: utf-8 -*-
13 from salome.geom import geomBuilder
18 geompy = geomBuilder.New()
30 O = geompy.MakeVertex(0, 0, 0)
31 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
32 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
33 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
34 geompy.addToStudy( O, 'O' )
35 geompy.addToStudy( OX, 'OX' )
36 geompy.addToStudy( OY, 'OY' )
37 geompy.addToStudy( OZ, 'OZ' )
39 Cylinder_1 = geompy.MakeCylinderRH(r1, h1)
40 Cylinder_2 = geompy.MakeCylinderRH(r2, h2)
41 Rotation_1 = geompy.MakeRotation(Cylinder_2, OY, -90*math.pi/180.0)
42 Translation_1 = geompy.MakeTranslation(Rotation_1, 0, 0, length_piquage)
44 piquage = geompy.MakeFuseList([Cylinder_1, Translation_1], True, True)
45 geompy.addToStudy( piquage, 'piquage' )
47 Inlet_z = geompy.GetFaceNearPoint(piquage, O)
48 geompy.addToStudyInFather( piquage, Inlet_z, 'Inlet_z' )
50 p_inlet_x = geompy.MakeVertex(-h2, 0, length_piquage)
51 Inlet_x = geompy.GetFaceNearPoint(piquage, p_inlet_x)
52 geompy.addToStudyInFather( piquage, Inlet_x, 'Inlet_x' )
54 p_outlet = geompy.MakeVertex(0, 0, h1)
55 Outlet = geompy.GetFaceNearPoint(piquage, p_outlet)
56 geompy.addToStudyInFather( piquage, Outlet, 'Outlet' )
58 Wall = geompy.CreateGroup(piquage, geompy.ShapeType["FACE"])
59 faces = geompy.SubShapeAll(piquage, geompy.ShapeType["FACE"])
60 geompy.UnionList(Wall, faces)
61 geompy.DifferenceList(Wall, [Inlet_x, Inlet_z, Outlet])
62 geompy.addToStudyInFather( piquage, Wall, 'Wall' )
64 p_corner = geompy.MakeVertex(-r2, 0, length_piquage+r2)
65 corner = geompy.GetVertexNearPoint(piquage, p_corner)
66 geompy.addToStudyInFather( piquage, corner, 'corner' )
68 geom_groups = [Inlet_x, Inlet_z, Outlet, Wall]
75 import SMESH, SALOMEDS
76 from salome.smesh import smeshBuilder
78 from salome.StdMeshers import StdMeshersBuilder
79 from salome.HYBRIDPlugin import HYBRIDPluginBuilder
81 smesh = smeshBuilder.New()
84 # Viscous layers without imprinting
85 # =================================
87 Mesh_1 = smesh.Mesh(piquage, "Mesh_without_imprinting")
88 NETGEN_2D = Mesh_1.Triangle(algo=smeshBuilder.NETGEN_1D2D)
89 NETGEN_2D_Parameters = NETGEN_2D.Parameters()
91 NETGEN_2D_Parameters.SetMinSize( 0.01 )
92 NETGEN_2D_Parameters.SetLocalSizeOnShape(corner, 0.01)
93 NETGEN_2D_Parameters.SetFineness( 5 )
94 NETGEN_2D_Parameters.SetGrowthRate( 0.1 )
95 NETGEN_2D_Parameters.SetNbSegPerEdge( 2 )
96 NETGEN_2D_Parameters.SetNbSegPerRadius( 3 )
98 wall_faces = geompy.SubShapeAll(Wall, geompy.ShapeType["FACE"])
99 wall_ids = geompy.GetSubShapesIDs(piquage, wall_faces)
101 MG_Hybrid = Mesh_1.Tetrahedron(algo=smeshBuilder.HYBRID)
102 MG_Hybrid_Parameters_1 = MG_Hybrid.Parameters()
103 MG_Hybrid_Parameters_1.SetLayersOnAllWrap( 0 )
104 MG_Hybrid_Parameters_1.SetElementGeneration( HYBRIDPluginBuilder.Generation_Tetra_Dominant )
105 MG_Hybrid_Parameters_1.SetHeightFirstLayer( 0.01 )
106 MG_Hybrid_Parameters_1.SetBoundaryLayersProgression( 1.1 )
107 MG_Hybrid_Parameters_1.SetNbOfBoundaryLayers( 3 )
108 MG_Hybrid_Parameters_1.SetFacesWithLayers( wall_ids )
111 isDone = Mesh_1.Compute()
114 raise Exception("Error when computing Mesh_without_imprinting")
118 for geom_group in geom_groups:
119 name = geom_group.GetName()
120 gr = Mesh_1.Group(geom_group)
121 d_groups_1[name] = gr
122 d_geom_groups[name] = geom_group
124 assert Mesh_1.NbQuadrangles() == 0
126 shape_volume = geompy.BasicProperties(piquage)[2]
128 # Compare whole mesh volume
129 volume_error_1 = abs(shape_volume-Mesh_1.GetVolume())/shape_volume
131 assert volume_error_1 < 0.02
133 # Viscous layers with imprinting
134 # ==============================
136 Mesh_2 = smesh.Mesh(piquage, "Mesh_with_imprinting")
138 NETGEN_2D_1_1 = Mesh_2.Triangle(algo=smeshBuilder.NETGEN_1D2D)
139 status = Mesh_2.AddHypothesis(NETGEN_2D_Parameters)
141 Inlet_x_id, Inlet_z_id, Outlet_id = geompy.GetSubShapesIDs(piquage, [Inlet_x, Inlet_z, Outlet])
143 MG_Hybrid_2 = Mesh_2.Tetrahedron(algo=smeshBuilder.HYBRID)
144 MG_Hybrid_Parameters_2 = MG_Hybrid_2.Parameters()
145 MG_Hybrid_Parameters_2.SetLayersOnAllWrap( 0 )
146 MG_Hybrid_Parameters_2.SetElementGeneration( HYBRIDPluginBuilder.Generation_Tetra_Dominant )
147 MG_Hybrid_Parameters_2.SetHeightFirstLayer( 0.01 )
148 MG_Hybrid_Parameters_2.SetBoundaryLayersProgression( 1.1 )
149 MG_Hybrid_Parameters_2.SetNbOfBoundaryLayers( 3 )
150 MG_Hybrid_Parameters_2.SetFacesWithLayers( wall_ids )
151 MG_Hybrid_Parameters_2.SetFacesWithImprinting( [ Inlet_x_id, Inlet_z_id, Outlet_id ] )
153 isDone = Mesh_2.Compute()
156 raise Exception("Error when computing Mesh_with_imprinting")
158 assert Mesh_2.NbQuadrangles() > 0
162 for geom_group in geom_groups:
163 name = geom_group.GetName()
164 gr = Mesh_2.Group(geom_group)
165 d_groups_2[name] = gr
167 faces_imprinted = ["Inlet_x", "Inlet_z", "Outlet"]
169 # Check viscous layers with imprinting
170 for name in faces_imprinted:
171 geom_area = geompy.BasicProperties(d_geom_groups[name])[1]
172 gr_1 = d_groups_1[name]
173 gr_2 = d_groups_2[name]
174 #assert gr_1.Size() > 0
175 assert gr_2.Size() > 0
176 # Nb of quadrangles is in 7th index of mesh info
177 assert gr_2.GetMeshInfo()[7] > 0
179 # Compare mesh group and geom group
180 area_error_1 = abs(geom_area-smesh.GetArea(gr_2))/geom_area
181 assert area_error_1 < 0.025
182 # Compare mesh group with imprinting and mesh group without imprinting
183 area_error_2 = abs(smesh.GetArea(gr_1)-smesh.GetArea(gr_2))/smesh.GetArea(gr_1)
184 assert area_error_2 < 1e-10
186 # Compare whole mesh volume
187 mesh_2_volume = Mesh_2.GetVolume()
188 volume_error_2 = abs(shape_volume-mesh_2_volume)/shape_volume
190 assert volume_error_2 < 0.02
192 # Viscous layers with imprinting set by groups
193 # ============================================
195 Mesh_3 = smesh.Mesh(piquage, "Mesh_with_imprinting_set_by_groups")
197 NETGEN_2D_1_1 = Mesh_3.Triangle(algo=smeshBuilder.NETGEN_1D2D)
198 status = Mesh_3.AddHypothesis(NETGEN_2D_Parameters)
200 MG_Hybrid_3 = Mesh_3.Tetrahedron(algo=smeshBuilder.HYBRID)
201 MG_Hybrid_Parameters_3 = MG_Hybrid_3.Parameters()
202 MG_Hybrid_Parameters_3.SetLayersOnAllWrap( 0 )
203 MG_Hybrid_Parameters_3.SetElementGeneration( HYBRIDPluginBuilder.Generation_Tetra_Dominant )
204 MG_Hybrid_Parameters_3.SetHeightFirstLayer( 0.01 )
205 MG_Hybrid_Parameters_3.SetBoundaryLayersProgression( 1.1 )
206 MG_Hybrid_Parameters_3.SetNbOfBoundaryLayers( 3 )
207 MG_Hybrid_3.SetFacesWithLayers( Wall )
208 MG_Hybrid_3.SetFacesWithImprinting( [ Inlet_x, Inlet_z, Outlet ] )
210 isDone = Mesh_3.Compute()
213 raise Exception("Error when computing Mesh_with_imprinting_set_by_groups")
217 for geom_group in geom_groups:
218 name = geom_group.GetName()
219 gr = Mesh_3.Group(geom_group)
220 d_groups_3[name] = gr
222 # Compare whole mesh volume
223 mesh_3_volume = Mesh_3.GetVolume()
224 volume_error_3 = abs(mesh_2_volume-mesh_3_volume)/mesh_2_volume
226 assert abs(mesh_3_volume-mesh_2_volume)/mesh_2_volume < 1e-10
228 # Check viscous layers with imprinting
229 for name in faces_imprinted:
230 gr_2 = d_groups_2[name]
231 gr_3 = d_groups_3[name]
232 assert gr_3.Size() > 0
233 # Nb of quadrangles is in 7th index of mesh info
234 assert gr_2.GetMeshInfo()[7] == gr_3.GetMeshInfo()[7]
236 # Compare mesh group with imprinting set by ids and mesh group with imprinting set by geom group
237 area_error_3 = abs(smesh.GetArea(gr_2)-smesh.GetArea(gr_3))/smesh.GetArea(gr_2)
238 assert area_error_3 < 1e-10