test/SMESH_create_dual_mesh_tpipe.py

   1 #!/usr/bin/env python3
   2
   3 import sys
   4 import salome
   5 import math
   6
   7 salome.salome_init()
   8
   9 import GEOM
  10 from salome.geom import geomBuilder
  11
  12 geompy = geomBuilder.New()
  13
  14 # first cylinder
  15 r1 = 0.5
  16 h1 = 5
  17
  18 # second cylinder
  19 r2 = 0.3
  20 h2 = 3
  21
  22 length_piquage = 1.5
  23
  24 O = geompy.MakeVertex(0, 0, 0)
  25 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
  26 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
  27 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
  28 geompy.addToStudy( O, 'O' )
  29 geompy.addToStudy( OX, 'OX' )
  30 geompy.addToStudy( OY, 'OY' )
  31 geompy.addToStudy( OZ, 'OZ' )
  32
  33 Cylinder_1 = geompy.MakeCylinderRH(r1, h1)
  34 Cylinder_2 = geompy.MakeCylinderRH(r2, h2)
  35 Rotation_1 = geompy.MakeRotation(Cylinder_2, OY, -90*math.pi/180.0)
  36 Translation_1 = geompy.MakeTranslation(Rotation_1, 0, 0, length_piquage)
  37
  38 tpipe = geompy.MakeFuseList([Cylinder_1, Translation_1], True, True)
  39 geompy.addToStudy( tpipe, 'tpipe' )
  40
  41 Inlet_z = geompy.GetFaceNearPoint(tpipe, O)
  42 geompy.addToStudyInFather( tpipe, Inlet_z, 'Inlet_z' )
  43
  44 p_inlet_x = geompy.MakeVertex(-h2, 0, length_piquage)
  45 Inlet_x = geompy.GetFaceNearPoint(tpipe, p_inlet_x)
  46 geompy.addToStudyInFather( tpipe, Inlet_x, 'Inlet_x' )
  47
  48 p_outlet = geompy.MakeVertex(0, 0, h1)
  49 Outlet = geompy.GetFaceNearPoint(tpipe, p_outlet)
  50 geompy.addToStudyInFather( tpipe, Outlet, 'Outlet' )
  51
  52 Wall = geompy.CreateGroup(tpipe, geompy.ShapeType["FACE"])
  53 faces = geompy.SubShapeAll(tpipe, geompy.ShapeType["FACE"])
  54 geompy.UnionList(Wall, faces)
  55 geompy.DifferenceList(Wall, [Inlet_x, Inlet_z, Outlet])
  56 geompy.addToStudyInFather( tpipe, Wall, 'Wall' )
  57
  58 p_corner = geompy.MakeVertex(-r2, 0, length_piquage+r2)
  59 corner = geompy.GetVertexNearPoint(tpipe, p_corner)
  60 geompy.addToStudyInFather( tpipe, corner, 'corner' )
  61
  62 geom_groups = [Inlet_x, Inlet_z, Outlet, Wall]
  63
  64 volumeGEOM = geompy.BasicProperties(tpipe)[2]
  65
  66 ###
  67 ### SMESH component
  68 ###
  69
  70 import  SMESH, SALOMEDS
  71 from salome.smesh import smeshBuilder
  72
  73 from salome.StdMeshers import StdMeshersBuilder
  74
  75 smesh = smeshBuilder.New()
  76
  77 # Coarse mesh with default hypothesis
  78 Mesh_1 = smesh.Mesh(tpipe, "Mesh_coarse")
  79
  80 Mesh_1.Triangle(algo=smeshBuilder.NETGEN_1D2D)
  81
  82 algo3d = Mesh_1.Tetrahedron(algo=smeshBuilder.NETGEN_3D)
  83 algo3d.MaxElementVolume(0.0002)
  84
  85 isDone = Mesh_1.Compute()
  86 if not isDone:
  87     raise Exception("Error when computing Mesh")
  88
  89 # Create groups
  90 d_geom_groups = {}
  91 d_mesh_groups = {}
  92 for geom_group in geom_groups:
  93   gr = Mesh_1.Group(geom_group)
  94   gr_name = gr.GetName()
  95   d_mesh_groups[gr_name] = gr
  96   d_geom_groups[gr_name] = geom_group
  97
  98 # Check tetra mesh volume
  99 tetra_volume = smesh.GetVolume(Mesh_1)
 100 shape_volume = geompy.BasicProperties(tpipe)[2]
 101 assert abs(tetra_volume-shape_volume)/shape_volume < 0.05
 102
 103 dual_Mesh_raw_1 = smesh.CreateDualMesh(Mesh_1, 'dual_Mesh_raw_1', False)
 104
 105 # check polyhedrons
 106 assert dual_Mesh_raw_1.NbPolyhedrons() > 0
 107 assert dual_Mesh_raw_1.NbPolyhedrons() == dual_Mesh_raw_1.NbVolumes()
 108
 109 # Check dual mesh volume
 110 dual_raw_volume = dual_Mesh_raw_1.GetVolume()
 111 assert abs(dual_raw_volume-shape_volume)/shape_volume < 0.14
 112
 113 # Check groups
 114 dual_Mesh_raw_groups = dual_Mesh_raw_1.GetGroups()
 115 dual_Mesh_raw_group_names = dual_Mesh_raw_1.GetGroupNames()
 116
 117 assert len(dual_Mesh_raw_groups) == 4
 118
 119 ## Create dual mesh with projection on faces
 120 dual_Mesh_1 = smesh.CreateDualMesh(Mesh_1, 'dual_Mesh_1', True)
 121
 122 # Check dual mesh volume
 123 dual_volume = dual_Mesh_1.GetVolume()
 124 print("shape_volume: ", shape_volume)
 125 print("tetra_volume: ", tetra_volume)
 126 print("dual_volume: ", dual_volume)
 127 print("dual_raw_volume: ", dual_raw_volume)
 128 assert (dual_volume >= dual_raw_volume)
 129
 130 assert abs(dual_volume-shape_volume)/shape_volume < 0.14
 131 assert abs(dual_raw_volume-tetra_volume)/tetra_volume < 1e-12
 132
 133 # Check groups
 134 dual_Mesh_groups = dual_Mesh_1.GetGroups()
 135 dual_Mesh_group_names = dual_Mesh_1.GetGroupNames()
 136 d_mesh_dual_groups = {}
 137 for gr_name, gr in zip(dual_Mesh_group_names, dual_Mesh_groups):
 138     gr_name = gr_name.strip()
 139     d_mesh_dual_groups[gr_name] = gr
 140
 141 for gr_dual_raw, gr_name in zip(dual_Mesh_raw_groups, dual_Mesh_raw_group_names):
 142   gr_name = gr_name.strip()
 143   gr_tri = d_mesh_groups[gr_name]
 144   gr_dual = d_mesh_dual_groups[gr_name]
 145   gr_geom = d_geom_groups[gr_name]
 146   area_gr_geom = geompy.BasicProperties(gr_geom)[1]
 147   area_gr_tri = smesh.GetArea(gr_tri)
 148   area_gr_dual_raw = smesh.GetArea(gr_dual_raw)
 149   area_gr_dual = smesh.GetArea(gr_dual)
 150   print(gr_name)
 151   print("Area geom: ", area_gr_geom)
 152   print("Area tri: ", area_gr_tri)
 153   print("Area dual raw:", area_gr_dual_raw)
 154   print("Area dual:", area_gr_dual)
 155   assert abs(area_gr_geom-area_gr_dual)/area_gr_geom < 0.015
 156   assert abs(area_gr_tri-area_gr_dual_raw)/area_gr_tri < 1e-3
 157
 158
 159 if salome.sg.hasDesktop():
 160   salome.sg.updateObjBrowser()