smesh/doc/examples/transforming_meshes_ex06.py
2023-03-29 14:18:23 +02:00

85 lines
2.7 KiB
Python

# Merging Elements
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()
# create a face to be meshed
px = geom_builder.MakeVertex(100., 0. , 0. )
py = geom_builder.MakeVertex(0. , 100., 0. )
pz = geom_builder.MakeVertex(0. , 0. , 100.)
vxy = geom_builder.MakeVector(px, py)
arc = geom_builder.MakeArc(py, pz, px)
wire = geom_builder.MakeWire([vxy, arc])
isPlanarFace = 1
face1 = geom_builder.MakeFace(wire, isPlanarFace)
id_face1 = geom_builder.addToStudy(face1, "Face1")
# create a circle to be an extrusion path
px1 = geom_builder.MakeVertex( 100., 100., 0.)
py1 = geom_builder.MakeVertex(-100., -100., 0.)
pz1 = geom_builder.MakeVertex( 0., 0., 50.)
circle = geom_builder.MakeCircleThreePnt(py1, pz1, px1)
id_circle = geom_builder.addToStudy(circle, "Path")
# create a 2D mesh on the face
trias = smesh_builder.Mesh(face1, "Face : 2D mesh")
algo1D = trias.Segment()
algo1D.NumberOfSegments(6)
algo2D = trias.Triangle()
algo2D.LengthFromEdges()
if not trias.Compute(): raise Exception("Error when computing Mesh")
# create a group of all triangles currently present in the mesh
faceTriGroup = trias.Group( face1, "face triangles" )
# create a path mesh
circlemesh = smesh_builder.Mesh(circle, "Path mesh")
algo = circlemesh.Segment()
algo.NumberOfSegments(10)
if not circlemesh.Compute(): raise Exception("Error when computing Mesh")
# extrusion of the mesh
trias.ExtrusionAlongPath([], circlemesh, circle, 1, MakeGroups=True )
# get a group "opposite" to faceTriGroup within the generated prismatic mesh
oppositeGroup = trias.GetGroupByName( faceTriGroup.GetName() + "_top" )[0]
# get edges of the groups
edgeGroup = trias.CreateDimGroup([ faceTriGroup, oppositeGroup ], SMESH.EDGE, "face edges")
# merge nodes of the groups only
print("Number of nodes before MergeNodes:", end=' ')
trias.NbNodes()
tolerance = 0.001
array_of_nodes_groups = trias.FindCoincidentNodesOnPart([faceTriGroup, oppositeGroup], tolerance)
trias.MergeNodes(array_of_nodes_groups)
print("Number of nodes after MergeNodes:", trias.NbNodes())
print("")
print("Number of elements before MergeEqualElements:")
print("Edges : ", trias.NbEdges())
print("Faces : ", trias.NbFaces())
print("Volumes : ", trias.NbVolumes())
# merge elements of the groups
equalFaces = trias.FindEqualElements( [faceTriGroup, oppositeGroup, edgeGroup] )
trias.MergeElements( equalFaces )
print("Number of elements after MergeEqualElements:")
print("Edges : ", trias.NbEdges())
print("Faces : ", trias.NbFaces())
print("Volumes : ", trias.NbVolumes())