Salome HOME
Rename tests without test_ prefix
[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 theStudy = salome.myStudy
10
11 ###
12 ### GEOM component
13 ###
14
15 import GEOM
16 from salome.geom import geomBuilder
17 import math
18 import SALOMEDS
19
20
21 geompy = geomBuilder.New(theStudy)
22
23 O = geompy.MakeVertex(0, 0, 0)
24 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
25 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
26 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
27 Cylinder_1 = geompy.MakeCylinderRH(0.5, 5)
28 Cylinder_2 = geompy.MakeCylinderRH(0.3, 3)
29 Rotation_1 = geompy.MakeRotation(Cylinder_2, OY, -90*math.pi/180.0)
30 Translation_1 = geompy.MakeTranslation(Rotation_1, 0, 0, 1.5)
31 piquage = geompy.MakeFuseList([Cylinder_1, Translation_1], True, True)
32 [Outlet] = geompy.SubShapes(piquage, [15])
33 [Inlet_z,Inlet_x] = geompy.SubShapes(piquage, [3, 22])
34 [corner] = geompy.SubShapes(piquage, [14])
35 geomObj_1 = geompy.MakeVertex(0, 0, 0)
36 geomObj_2 = geompy.MakeVectorDXDYDZ(1, 0, 0)
37 geomObj_3 = geompy.MakeVectorDXDYDZ(0, 1, 0)
38 geomObj_4 = geompy.MakeVectorDXDYDZ(0, 0, 1)
39 Wall = geompy.CreateGroup(piquage, geompy.ShapeType["FACE"])
40 geompy.UnionIDs(Wall, [7, 17])
41 geompy.addToStudy( O, 'O' )
42 geompy.addToStudy( OX, 'OX' )
43 geompy.addToStudy( OY, 'OY' )
44 geompy.addToStudy( OZ, 'OZ' )
45 geompy.addToStudy( Cylinder_1, 'Cylinder_1' )
46 geompy.addToStudy( Cylinder_2, 'Cylinder_2' )
47 geompy.addToStudy( Rotation_1, 'Rotation_1' )
48 geompy.addToStudy( Translation_1, 'Translation_1' )
49 geompy.addToStudy( piquage, 'piquage' )
50 geompy.addToStudyInFather( piquage, Inlet_x, 'Inlet_x' )
51 geompy.addToStudyInFather( piquage, Outlet, 'Outlet' )
52 geompy.addToStudyInFather( piquage, Wall, 'Wall' )
53 geompy.addToStudyInFather( piquage, Inlet_z, 'Inlet_z' )
54 geompy.addToStudyInFather( piquage, corner, 'corner' )
55
56 geom_groups = [Inlet_x, Inlet_z, Outlet, Wall]
57 d_geom_groups = {}
58
59 ###
60 ### SMESH component
61 ###
62
63 import  SMESH, SALOMEDS
64 from salome.smesh import smeshBuilder
65
66 from salome.StdMeshers import StdMeshersBuilder
67 from salome.HYBRIDPlugin import HYBRIDPluginBuilder
68
69 smesh = smeshBuilder.New(theStudy)
70
71 # Viscous layers with imprinting (hexa core)
72 # ==========================================
73
74 Mesh_4 = smesh.Mesh(piquage, "Mesh_with_imprinting_hexa_core")
75
76 NETGEN_2D_1_1 = Mesh_4.Triangle(algo=smeshBuilder.NETGEN_1D2D)
77 NETGEN_2D_Parameters = NETGEN_2D_1_1.Parameters()
78
79 NETGEN_2D_Parameters.SetMinSize( 0.01 )
80 NETGEN_2D_Parameters.SetLocalSizeOnShape(corner, 0.01)
81 NETGEN_2D_Parameters.SetFineness( 5 )
82 NETGEN_2D_Parameters.SetGrowthRate( 0.1 )
83 NETGEN_2D_Parameters.SetNbSegPerEdge( 2 )
84 NETGEN_2D_Parameters.SetNbSegPerRadius( 3 )
85
86 MG_Hybrid_4 = Mesh_4.Tetrahedron(algo=smeshBuilder.HYBRID)
87 MG_Hybrid_Parameters_4 = MG_Hybrid_4.Parameters()
88 MG_Hybrid_Parameters_4.SetLayersOnAllWrap( 0 )
89 MG_Hybrid_Parameters_4.SetElementGeneration( HYBRIDPluginBuilder.Generation_Hexa_Dominant )
90 MG_Hybrid_Parameters_4.SetHeightFirstLayer( 0.01 )
91 MG_Hybrid_Parameters_4.SetBoundaryLayersProgression( 1.1 )
92 MG_Hybrid_Parameters_4.SetNbOfBoundaryLayers( 3 )
93 MG_Hybrid_Parameters_4.SetFacesWithLayers( [ 7, 17 ] )
94 MG_Hybrid_Parameters_4.SetFacesWithImprinting( [ 22, 15, 3 ] )
95
96 isDone = Mesh_4.Compute()
97
98 if not isDone:
99   raise Exception("Error when computing Mesh_with_imprinting_hexa_core")
100
101 assert Mesh_4.NbQuadrangles() > 0
102
103 d_groups_4 = {}
104 d_geom_groups = {}
105
106 for geom_group in geom_groups:
107   name = geom_group.GetName()
108   gr = Mesh_4.Group(geom_group)
109   d_groups_4[name] = gr
110   d_geom_groups[name] = geom_group
111
112 # Compare whole mesh volume
113 mesh_4_volume = Mesh_4.GetVolume()
114 shape_volume = geompy.BasicProperties(piquage)[2]
115 volume_error_4 = abs(shape_volume-mesh_4_volume)/shape_volume
116
117 assert volume_error_4 < 0.015
118
119 # Check viscous layers with imprinting
120 faces_imprinted = ["Inlet_x", "Inlet_z", "Outlet"]
121
122 for name in faces_imprinted:
123   gr_4 = d_groups_4[name]
124   #assert gr_1.Size() > 0
125   assert gr_4.Size() > 0
126   # Nb of quadrangles is in 7th index of mesh info
127   assert gr_4.GetMeshInfo()[7] > 0
128
129   # Compare mesh group and geom group
130   geom_area = geompy.BasicProperties(d_geom_groups[name])[1]
131   area_error_1 = abs(geom_area-smesh.GetArea(gr_4))/geom_area
132   assert area_error_1 < 0.025
133
134
135 if salome.sg.hasDesktop():
136   salome.sg.updateObjBrowser(1)