test/netgen_runner.py

   1 #!/usr/bin/env python
   2 import sys
   3 import salome
   4
   5 salome.salome_init()
   6
   7 from os import path
   8 import tempfile
   9 import subprocess
  10
  11 import GEOM, SMESH, SALOMEDS
  12
  13 from salome.geom import geomBuilder
  14 from salome.smesh import smeshBuilder
  15
  16 import math
  17
  18 smesh = smeshBuilder.New()
  19 geompy = geomBuilder.New()
  20
  21 import medcoupling as mc
  22
  23 def create_param_file(param_file):
  24     """ Create a parameter file for runner """
  25     param = """1
  26 34.64
  27 0.14
  28 16
  29 0.15
  30 1.5
  31 0
  32 0
  33 1
  34 5
  35 1
  36 1
  37 -1
  38 3
  39 3
  40 0.2
  41 2
  42 1
  43 0
  44 0
  45 2
  46 2
  47 0
  48
  49 0
  50 0
  51 0"""
  52     with open(param_file, "w") as ffile:
  53         ffile.write(param)
  54
  55
  56 def test_netgen3d():
  57     """ Test netgen3d mesher """
  58     # Building geometry
  59     box = geompy.MakeBoxDXDYDZ(200, 200, 200)
  60
  61     geompy.ExtractShapes(box, geompy.ShapeType["FACE"], True)
  62     groupe_1 = geompy.CreateGroup(box, geompy.ShapeType["FACE"])
  63     geompy.UnionIDs(groupe_1, [3, 13, 23, 27, 31, 33])
  64
  65     [_, _, _, _, _, _, groupe_1] = geompy.GetExistingSubObjects(box, False)
  66
  67     # Creating 2D mesh
  68     netgen_2d_parameters_1 = smesh.CreateHypothesisByAverageLength(
  69         'NETGEN_Parameters_2D', 'NETGENEngine', 34.641, 0)
  70     mesh_2d = smesh.Mesh(groupe_1, 'Maillage_1')
  71     mesh_2d.AddHypothesis(groupe_1, netgen_2d_parameters_1)
  72     mesh_2d.Triangle(algo=smeshBuilder.NETGEN_1D2D)
  73     isDone = mesh_2d.Compute()
  74     if not isDone:
  75         raise Exception("Error when computing Mesh")
  76
  77     smesh.SetName(mesh_2d, 'Maillage_1')
  78
  79     with tempfile.TemporaryDirectory() as tmp_dir:
  80         mesh_file = path.join(tmp_dir, "mesh.med")
  81         shape_file = path.join(tmp_dir, "shape.step")
  82         param_file = path.join(tmp_dir, "param.txt")
  83         output_mesh = path.join(tmp_dir, "mesh3D.med")
  84
  85         print("Running in folder: ", tmp_dir)
  86         create_param_file(param_file)
  87
  88         mesh_2d.ExportMED(mesh_file, 0, 41, 1, mesh_2d, 1, [], '', -1, 1)
  89         geompy.ExportSTEP(box, shape_file, GEOM.LU_METER)
  90
  91         runner = path.join("${NETGENPLUGIN_ROOT_DIR}",
  92                            "bin",
  93                            "salome",
  94                            "NETGENPlugin_Runner")
  95
  96         if sys.platform == 'win32':
  97             runner += ".exe"
  98
  99         cmd = "{runner} NETGEN3D {mesh_file} {shape_file} "\
 100               "{param_file} NONE NONE {output_mesh}"\
 101               .format(runner=runner,
 102                       mesh_file=mesh_file,
 103                       shape_file=shape_file,
 104                       param_file=param_file,
 105                       output_mesh=output_mesh)
 106         print(cmd)
 107         subprocess.check_call(cmd, shell=True)
 108
 109         mesh_read = mc.ReadUMeshFromFile(output_mesh, "MESH", 0)
 110
 111         nb_tetras = mesh_read.getNumberOfCellsWithType(mc.NORM_TETRA4)
 112         nb_points = mesh_read.getNumberOfNodes()
 113
 114         mesh_read = mc.ReadUMeshFromFile(output_mesh, "MESH", -1)
 115         nb_triangles = mesh_read.getNumberOfCellsWithType(mc.NORM_TRI3)
 116
 117         mesh_read = mc.ReadUMeshFromFile(output_mesh, "MESH", -2)
 118         nb_segments = mesh_read.getNumberOfCellsWithType(mc.NORM_SEG2)
 119
 120         print("Nb Tetras:", nb_tetras)
 121         print("Nb Triangles:", nb_triangles)
 122         print("Nb Segments:", nb_segments)
 123         print("Nb Points:", nb_points)
 124
 125         assert nb_points > 0
 126         assert nb_segments > 0
 127         assert nb_triangles > 0
 128         assert nb_tetras > 0
 129
 130 if __name__ == "__main__":
 131     if sys.platform == "win32":
 132         print("Disabled on windows")
 133         sys.exit(0)
 134     test_netgen3d()