mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2024-12-26 09:20:34 +05:00
161 lines
4.5 KiB
Python
161 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import salome
|
|
import math
|
|
|
|
salome.salome_init()
|
|
|
|
import GEOM
|
|
from salome.geom import geomBuilder
|
|
|
|
geompy = geomBuilder.New()
|
|
|
|
# first cylinder
|
|
r1 = 0.5
|
|
h1 = 5
|
|
|
|
# second cylinder
|
|
r2 = 0.3
|
|
h2 = 3
|
|
|
|
length_piquage = 1.5
|
|
|
|
O = geompy.MakeVertex(0, 0, 0)
|
|
OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
|
|
OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
|
|
OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
|
|
geompy.addToStudy( O, 'O' )
|
|
geompy.addToStudy( OX, 'OX' )
|
|
geompy.addToStudy( OY, 'OY' )
|
|
geompy.addToStudy( OZ, 'OZ' )
|
|
|
|
Cylinder_1 = geompy.MakeCylinderRH(r1, h1)
|
|
Cylinder_2 = geompy.MakeCylinderRH(r2, h2)
|
|
Rotation_1 = geompy.MakeRotation(Cylinder_2, OY, -90*math.pi/180.0)
|
|
Translation_1 = geompy.MakeTranslation(Rotation_1, 0, 0, length_piquage)
|
|
|
|
tpipe = geompy.MakeFuseList([Cylinder_1, Translation_1], True, True)
|
|
geompy.addToStudy( tpipe, 'tpipe' )
|
|
|
|
Inlet_z = geompy.GetFaceNearPoint(tpipe, O)
|
|
geompy.addToStudyInFather( tpipe, Inlet_z, 'Inlet_z' )
|
|
|
|
p_inlet_x = geompy.MakeVertex(-h2, 0, length_piquage)
|
|
Inlet_x = geompy.GetFaceNearPoint(tpipe, p_inlet_x)
|
|
geompy.addToStudyInFather( tpipe, Inlet_x, 'Inlet_x' )
|
|
|
|
p_outlet = geompy.MakeVertex(0, 0, h1)
|
|
Outlet = geompy.GetFaceNearPoint(tpipe, p_outlet)
|
|
geompy.addToStudyInFather( tpipe, Outlet, 'Outlet' )
|
|
|
|
Wall = geompy.CreateGroup(tpipe, geompy.ShapeType["FACE"])
|
|
faces = geompy.SubShapeAll(tpipe, geompy.ShapeType["FACE"])
|
|
geompy.UnionList(Wall, faces)
|
|
geompy.DifferenceList(Wall, [Inlet_x, Inlet_z, Outlet])
|
|
geompy.addToStudyInFather( tpipe, Wall, 'Wall' )
|
|
|
|
p_corner = geompy.MakeVertex(-r2, 0, length_piquage+r2)
|
|
corner = geompy.GetVertexNearPoint(tpipe, p_corner)
|
|
geompy.addToStudyInFather( tpipe, corner, 'corner' )
|
|
|
|
geom_groups = [Inlet_x, Inlet_z, Outlet, Wall]
|
|
|
|
volumeGEOM = geompy.BasicProperties(tpipe)[2]
|
|
|
|
###
|
|
### SMESH component
|
|
###
|
|
|
|
import SMESH, SALOMEDS
|
|
from salome.smesh import smeshBuilder
|
|
|
|
from salome.StdMeshers import StdMeshersBuilder
|
|
|
|
smesh = smeshBuilder.New()
|
|
|
|
# Coarse mesh with default hypothesis
|
|
Mesh_1 = smesh.Mesh(tpipe, "Mesh_coarse")
|
|
|
|
Mesh_1.Triangle(algo=smeshBuilder.NETGEN_1D2D)
|
|
|
|
algo3d = Mesh_1.Tetrahedron(algo=smeshBuilder.NETGEN_3D)
|
|
algo3d.MaxElementVolume(0.0002)
|
|
|
|
isDone = Mesh_1.Compute()
|
|
if not isDone:
|
|
raise Exception("Error when computing Mesh")
|
|
|
|
# Create groups
|
|
d_geom_groups = {}
|
|
d_mesh_groups = {}
|
|
for geom_group in geom_groups:
|
|
gr = Mesh_1.Group(geom_group)
|
|
gr_name = gr.GetName()
|
|
d_mesh_groups[gr_name] = gr
|
|
d_geom_groups[gr_name] = geom_group
|
|
|
|
# Check tetra mesh volume
|
|
tetra_volume = smesh.GetVolume(Mesh_1)
|
|
shape_volume = geompy.BasicProperties(tpipe)[2]
|
|
assert abs(tetra_volume-shape_volume)/shape_volume < 0.05
|
|
|
|
dual_Mesh_raw_1 = smesh.CreateDualMesh(Mesh_1, 'dual_Mesh_raw_1', False)
|
|
|
|
# check polyhedrons
|
|
assert dual_Mesh_raw_1.NbPolyhedrons() > 0
|
|
assert dual_Mesh_raw_1.NbPolyhedrons() == dual_Mesh_raw_1.NbVolumes()
|
|
|
|
# Check dual mesh volume
|
|
dual_raw_volume = dual_Mesh_raw_1.GetVolume()
|
|
assert abs(dual_raw_volume-shape_volume)/shape_volume < 0.14
|
|
|
|
# Check groups
|
|
dual_Mesh_raw_groups = dual_Mesh_raw_1.GetGroups()
|
|
dual_Mesh_raw_group_names = dual_Mesh_raw_1.GetGroupNames()
|
|
|
|
assert len(dual_Mesh_raw_groups) == 4
|
|
|
|
## Create dual mesh with projection on faces
|
|
dual_Mesh_1 = smesh.CreateDualMesh(Mesh_1, 'dual_Mesh_1', True)
|
|
|
|
# Check dual mesh volume
|
|
dual_volume = dual_Mesh_1.GetVolume()
|
|
print("shape_volume: ", shape_volume)
|
|
print("tetra_volume: ", tetra_volume)
|
|
print("dual_volume: ", dual_volume)
|
|
print("dual_raw_volume: ", dual_raw_volume)
|
|
assert (dual_volume >= dual_raw_volume)
|
|
|
|
assert abs(dual_volume-shape_volume)/shape_volume < 0.14
|
|
assert abs(dual_raw_volume-tetra_volume)/tetra_volume < 1e-12
|
|
|
|
# Check groups
|
|
dual_Mesh_groups = dual_Mesh_1.GetGroups()
|
|
dual_Mesh_group_names = dual_Mesh_1.GetGroupNames()
|
|
d_mesh_dual_groups = {}
|
|
for gr_name, gr in zip(dual_Mesh_group_names, dual_Mesh_groups):
|
|
gr_name = gr_name.strip()
|
|
d_mesh_dual_groups[gr_name] = gr
|
|
|
|
for gr_dual_raw, gr_name in zip(dual_Mesh_raw_groups, dual_Mesh_raw_group_names):
|
|
gr_name = gr_name.strip()
|
|
gr_tri = d_mesh_groups[gr_name]
|
|
gr_dual = d_mesh_dual_groups[gr_name]
|
|
gr_geom = d_geom_groups[gr_name]
|
|
area_gr_geom = geompy.BasicProperties(gr_geom)[1]
|
|
area_gr_tri = smesh.GetArea(gr_tri)
|
|
area_gr_dual_raw = smesh.GetArea(gr_dual_raw)
|
|
area_gr_dual = smesh.GetArea(gr_dual)
|
|
print(gr_name)
|
|
print("Area geom: ", area_gr_geom)
|
|
print("Area tri: ", area_gr_tri)
|
|
print("Area dual raw:", area_gr_dual_raw)
|
|
print("Area dual:", area_gr_dual)
|
|
assert abs(area_gr_geom-area_gr_dual)/area_gr_geom < 0.015
|
|
assert abs(area_gr_tri-area_gr_dual_raw)/area_gr_tri < 1e-3
|
|
|
|
|
|
if salome.sg.hasDesktop():
|
|
salome.sg.updateObjBrowser()
|