test/SMESH_create_dual_mesh_adapt.py

   1 #!/usr/bin/env python
   2
   3 ###
   4 ### This file is generated automatically by SALOME v9.9.0 with dump python functionality
   5 ###
   6
   7 import sys
   8 import salome
   9
  10 salome.salome_init()
  11 import salome_notebook
  12 notebook = salome_notebook.NoteBook()
  13 sys.path.insert(0, r'/home/B61570/work_in_progress/dual_mesh')
  14
  15 ###
  16 ### GEOM component
  17 ###
  18
  19 import GEOM
  20 from salome.geom import geomBuilder
  21 import math
  22 import SALOMEDS
  23
  24
  25 geompy = geomBuilder.New()
  26
  27 O = geompy.MakeVertex(0, 0, 0)
  28 OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
  29 OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
  30 OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
  31 Cylinder_1 = geompy.MakeCylinderRH(100, 400)
  32 Sphere_1 = geompy.MakeSpherePntR(O, 100)
  33 Fuse_1 = geompy.MakeFuseList([Cylinder_1, Sphere_1], True, True)
  34
  35 [geomObj_1,geomObj_2,geomObj_3] = geompy.ExtractShapes(Fuse_1, geompy.ShapeType["FACE"], True)
  36
  37 top = geompy.CreateGroup(Fuse_1, geompy.ShapeType["FACE"])
  38 geompy.UnionIDs(top, geomObj_1.GetSubShapeIndices())
  39
  40 middle = geompy.CreateGroup(Fuse_1, geompy.ShapeType["FACE"])
  41 geompy.UnionIDs(middle, geomObj_2.GetSubShapeIndices())
  42
  43 bottom = geompy.CreateGroup(Fuse_1, geompy.ShapeType["FACE"])
  44 geompy.UnionIDs(bottom, geomObj_3.GetSubShapeIndices())
  45
  46 #[top, middle, bottom] = geompy.GetExistingSubObjects(Fuse_1, False)
  47
  48 geompy.addToStudy( O, 'O' )
  49 geompy.addToStudy( OX, 'OX' )
  50 geompy.addToStudy( OY, 'OY' )
  51 geompy.addToStudy( OZ, 'OZ' )
  52 geompy.addToStudy( Cylinder_1, 'Cylinder_1' )
  53 geompy.addToStudy( Sphere_1, 'Sphere_1' )
  54 geompy.addToStudy( Fuse_1, 'Fuse_1' )
  55 geompy.addToStudyInFather( Fuse_1, top, 'top' )
  56 geompy.addToStudyInFather( Fuse_1, middle, 'middle' )
  57 geompy.addToStudyInFather( Fuse_1, bottom, 'bottom' )
  58
  59 ###
  60 ### SMESH component
  61 ###
  62
  63 import  SMESH, SALOMEDS
  64 from salome.smesh import smeshBuilder
  65
  66 smesh = smeshBuilder.New()
  67
  68 NETGEN_3D_Parameters_1 = smesh.CreateHypothesisByAverageLength( 'NETGEN_Parameters', 'NETGENEngine', 50, 0 )
  69 Mesh_1 = smesh.Mesh(Fuse_1,'Mesh_1')
  70 status = Mesh_1.AddHypothesis( Fuse_1, NETGEN_3D_Parameters_1 )
  71 NETGEN_1D_2D_3D = Mesh_1.Tetrahedron(algo=smeshBuilder.NETGEN_1D2D3D)
  72 top_1 = Mesh_1.GroupOnGeom(top,'top',SMESH.FACE)
  73 middle_1 = Mesh_1.GroupOnGeom(middle,'middle',SMESH.FACE)
  74 bottom_1 = Mesh_1.GroupOnGeom(bottom,'bottom',SMESH.FACE)
  75 isDone = Mesh_1.Compute()
  76
  77 [ top_1, middle_1, bottom_1 ] = Mesh_1.GetGroups()
  78
  79
  80 dual_Mesh_raw_1 = smesh.CreateDualMesh(Mesh_1, 'dual_Mesh_raw_1', False)
  81 dual_Mesh_1 = smesh.CreateDualMesh(Mesh_1, 'dual_Mesh_1', True)
  82
  83 [ top_2, middle_2, bottom_2 ] = dual_Mesh_1.GetGroups()
  84
  85 #Comparing volumes
  86 dual_volume = dual_Mesh_1.GetVolume()
  87 dual_raw_volume = dual_Mesh_raw_1.GetVolume()
  88 print("dual_volume: ", dual_volume)
  89 print("dual_raw_volume: ", dual_raw_volume)
  90
  91 assert (dual_volume >= dual_raw_volume)
  92
  93 if salome.sg.hasDesktop():
  94   salome.sg.updateObjBrowser()