mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2024-11-11 16:19:16 +05:00
105 lines
2.5 KiB
Python
105 lines
2.5 KiB
Python
# 3d mesh generation and mesh exploration
|
|
|
|
import salome
|
|
salome.salome_init_without_session()
|
|
|
|
import SMESH
|
|
from salome.geom import geomBuilder
|
|
from salome.smesh import smeshBuilder
|
|
|
|
geom_builder = geomBuilder.New()
|
|
smesh_builder = smeshBuilder.New()
|
|
|
|
###
|
|
# Geometry: an assembly of a box, a cylinder and a truncated cone
|
|
# to be meshed with tetrahedra
|
|
###
|
|
|
|
# Define values
|
|
name = "ex21_lamp"
|
|
cote = 60
|
|
section = 20
|
|
size = 200
|
|
radius_1 = 80
|
|
radius_2 = 40
|
|
height = 100
|
|
|
|
# Build a box
|
|
box = geom_builder.MakeBox(-cote, -cote, -cote, +cote, +cote, +cote)
|
|
|
|
# Build a cylinder
|
|
pt1 = geom_builder.MakeVertex(0, 0, cote/3)
|
|
di1 = geom_builder.MakeVectorDXDYDZ(0, 0, 1)
|
|
cyl = geom_builder.MakeCylinder(pt1, di1, section, size)
|
|
|
|
# Build a truncated cone
|
|
pt2 = geom_builder.MakeVertex(0, 0, size)
|
|
cone = geom_builder.MakeCone(pt2, di1, radius_1, radius_2, height)
|
|
|
|
# Fuse
|
|
box_cyl = geom_builder.MakeFuse(box, cyl)
|
|
piece = geom_builder.MakeFuse(box_cyl, cone)
|
|
|
|
# Add to the study
|
|
geom_builder.addToStudy(piece, name)
|
|
|
|
# Create a group of faces
|
|
faces_group = geom_builder.CreateGroup(piece, geom_builder.ShapeType["FACE"])
|
|
group_name = name + "_grp"
|
|
geom_builder.addToStudy(faces_group, group_name)
|
|
faces_group.SetName(group_name)
|
|
|
|
# Add faces to the group
|
|
faces = geom_builder.SubShapeAllIDs(piece, geom_builder.ShapeType["FACE"])
|
|
geom_builder.UnionIDs(faces_group, faces)
|
|
|
|
###
|
|
# Create a mesh
|
|
###
|
|
|
|
# Define a mesh on a geometry
|
|
tetra = smesh_builder.Mesh(piece, name)
|
|
|
|
# Define 1D algorithm and hypothesis
|
|
algo1d = tetra.Segment()
|
|
algo1d.LocalLength(10)
|
|
|
|
# Define 2D algorithm and hypothesis
|
|
algo2d = tetra.Triangle()
|
|
algo2d.LengthFromEdges()
|
|
|
|
# Define 3D algorithm and hypothesis
|
|
algo3d = tetra.Tetrahedron()
|
|
algo3d.MaxElementVolume(100)
|
|
|
|
# Compute the mesh
|
|
if not tetra.Compute(): raise Exception("Error when computing Mesh")
|
|
|
|
# Create a mesh group of all triangles generated on geom faces present in faces_group
|
|
group = tetra.Group(faces_group)
|
|
|
|
###
|
|
# Explore the mesh
|
|
###
|
|
|
|
# Retrieve coordinates of nodes
|
|
coordStr = ""
|
|
for node in tetra.GetNodesId():
|
|
x,y,z = tetra.GetNodeXYZ( node )
|
|
coordStr += "%s (%s, %s, %s) " % ( node, x,y,z )
|
|
pass
|
|
|
|
# Retrieve nodal connectivity of triangles
|
|
triaStr = ""
|
|
for tria in tetra.GetElementsByType( SMESH.FACE ):
|
|
nodes = tetra.GetElemNodes( tria )
|
|
triaStr += "%s (%s, %s, %s) " % ( tria, nodes[0], nodes[1], nodes[2] )
|
|
|
|
# Retrieve group contents
|
|
groupStr = ""
|
|
for group in tetra.GetGroups():
|
|
ids = group.GetIDs()
|
|
name = group.GetName()
|
|
eType = group.GetType()
|
|
groupStr += "'%s' %s: %s \n" % ( name, eType, ids )
|