Salome HOME
23586: [EDF] HYDRO: Copy mesh to new geometry
[plugins/hybridplugin.git] / tests / layers_imprinting_hexa_core.py
1 # -*- coding: utf-8 -*-
2
3 # This test is a known failure for MeshGems <= 2.4-5 (but must not crash Salome)
4
5 import sys
6 import salome
7
8 salome.salome_init()
9
10 ###
11 ### GEOM component
12 ###
13
14 import GEOM
15 from salome.geom import geomBuilder
16 import math
17 import SALOMEDS
18
19
20 geompy = geomBuilder.New()
21
22 # first cylinder
23 r1 = 0.5
24 h1 = 5
25
26 # second cylinder
27 r2 = 0.3
28 h2 = 3
29
30 length_piquage = 1.5
31
32 O = geompy.MakeVertex(0, 0, 0)
33 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
34 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
35 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
36 geompy.addToStudy( O, 'O' )
37 geompy.addToStudy( OX, 'OX' )
38 geompy.addToStudy( OY, 'OY' )
39 geompy.addToStudy( OZ, 'OZ' )
40
41 Cylinder_1 = geompy.MakeCylinderRH(r1, h1)
42 Cylinder_2 = geompy.MakeCylinderRH(r2, h2)
43 Rotation_1 = geompy.MakeRotation(Cylinder_2, OY, -90*math.pi/180.0)
44 Translation_1 = geompy.MakeTranslation(Rotation_1, 0, 0, length_piquage)
45
46 piquage = geompy.MakeFuseList([Cylinder_1, Translation_1], True, True)
47 geompy.addToStudy( piquage, 'piquage' )
48
49 Inlet_z = geompy.GetFaceNearPoint(piquage, O)
50 geompy.addToStudyInFather( piquage, Inlet_z, 'Inlet_z' )
51
52 p_inlet_x = geompy.MakeVertex(-h2, 0, length_piquage)
53 Inlet_x = geompy.GetFaceNearPoint(piquage, p_inlet_x)
54 geompy.addToStudyInFather( piquage, Inlet_x, 'Inlet_x' )
55
56 p_outlet = geompy.MakeVertex(0, 0, h1)
57 Outlet = geompy.GetFaceNearPoint(piquage, p_outlet)
58 geompy.addToStudyInFather( piquage, Outlet, 'Outlet' )
59
60 Wall = geompy.CreateGroup(piquage, geompy.ShapeType["FACE"])
61 faces = geompy.SubShapeAll(piquage, geompy.ShapeType["FACE"])
62 geompy.UnionList(Wall, faces)
63 geompy.DifferenceList(Wall, [Inlet_x, Inlet_z, Outlet])
64 geompy.addToStudyInFather( piquage, Wall, 'Wall' )
65
66 p_corner = geompy.MakeVertex(-r2, 0, length_piquage+r2)
67 corner = geompy.GetVertexNearPoint(piquage, p_corner)
68 geompy.addToStudyInFather( piquage, corner, 'corner' )
69
70 geom_groups = [Inlet_x, Inlet_z, Outlet, Wall]
71 d_geom_groups = {}
72
73 ###
74 ### SMESH component
75 ###
76
77 import  SMESH, SALOMEDS
78 from salome.smesh import smeshBuilder
79
80 from salome.StdMeshers import StdMeshersBuilder
81 from salome.HYBRIDPlugin import HYBRIDPluginBuilder
82
83 smesh = smeshBuilder.New()
84
85 # Viscous layers with imprinting (hexa core)
86 # ==========================================
87
88 Mesh_4 = smesh.Mesh(piquage, "Mesh_with_imprinting_hexa_core")
89
90 NETGEN_2D_1_1 = Mesh_4.Triangle(algo=smeshBuilder.NETGEN_1D2D)
91 NETGEN_2D_Parameters = NETGEN_2D_1_1.Parameters()
92
93 NETGEN_2D_Parameters.SetMinSize( 0.01 )
94 NETGEN_2D_Parameters.SetLocalSizeOnShape(corner, 0.01)
95 NETGEN_2D_Parameters.SetFineness( 5 )
96 NETGEN_2D_Parameters.SetGrowthRate( 0.1 )
97 NETGEN_2D_Parameters.SetNbSegPerEdge( 2 )
98 NETGEN_2D_Parameters.SetNbSegPerRadius( 3 )
99
100 wall_faces = geompy.SubShapeAll(Wall, geompy.ShapeType["FACE"])
101 wall_ids = geompy.GetSubShapesIDs(piquage, wall_faces)
102 Inlet_x_id, Inlet_z_id, Outlet_id = geompy.GetSubShapesIDs(piquage, [Inlet_x, Inlet_z, Outlet])
103
104 MG_Hybrid_4 = Mesh_4.Tetrahedron(algo=smeshBuilder.HYBRID)
105 MG_Hybrid_Parameters_4 = MG_Hybrid_4.Parameters()
106 MG_Hybrid_Parameters_4.SetLayersOnAllWrap( 0 )
107 MG_Hybrid_Parameters_4.SetElementGeneration( HYBRIDPluginBuilder.Generation_Hexa_Dominant )
108 MG_Hybrid_Parameters_4.SetHeightFirstLayer( 0.01 )
109 MG_Hybrid_Parameters_4.SetBoundaryLayersProgression( 1.1 )
110 MG_Hybrid_Parameters_4.SetNbOfBoundaryLayers( 3 )
111 MG_Hybrid_Parameters_4.SetFacesWithLayers( wall_ids )
112 MG_Hybrid_Parameters_4.SetFacesWithImprinting( [ Inlet_x_id, Inlet_z_id, Outlet_id ] )
113
114 isDone = Mesh_4.Compute()
115
116 if not isDone:
117   raise Exception("Error when computing Mesh_with_imprinting_hexa_core")
118
119 assert Mesh_4.NbQuadrangles() > 0
120
121 d_groups_4 = {}
122 d_geom_groups = {}
123
124 for geom_group in geom_groups:
125   name = geom_group.GetName()
126   gr = Mesh_4.Group(geom_group)
127   d_groups_4[name] = gr
128   d_geom_groups[name] = geom_group
129
130 # Compare whole mesh volume
131 mesh_4_volume = Mesh_4.GetVolume()
132 shape_volume = geompy.BasicProperties(piquage)[2]
133 volume_error_4 = abs(shape_volume-mesh_4_volume)/shape_volume
134
135 assert volume_error_4 < 0.02
136
137 # Check viscous layers with imprinting
138 faces_imprinted = ["Inlet_x", "Inlet_z", "Outlet"]
139
140 for name in faces_imprinted:
141   gr_4 = d_groups_4[name]
142   #assert gr_1.Size() > 0
143   assert gr_4.Size() > 0
144   # Nb of quadrangles is in 7th index of mesh info
145   assert gr_4.GetMeshInfo()[7] > 0
146
147   # Compare mesh group and geom group
148   geom_area = geompy.BasicProperties(d_geom_groups[name])[1]
149   area_error_1 = abs(geom_area-smesh.GetArea(gr_4))/geom_area
150   assert area_error_1 < 0.025
151