smesh/doc/examples/mechanic.py

87 lines
2.4 KiB
Python
Raw Permalink Normal View History

2022-04-11 18:28:01 +05:00
# Inspired by SMESH_mechanic.py.
# Creates geometry and a mesh to be used in other examples
# --------------------------------------------------------
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
# --------
# create planar face on a wire combined from a segment and three arcs
p1 = geom_builder.MakeVertex(100.0, 0.0, 0.0)
p2 = geom_builder.MakeVertex(50.0, 50.0, 0.0)
p3 = geom_builder.MakeVertex(100.0, 100.0, 0.0)
arc1 = geom_builder.MakeArc(p1, p2, p3)
p4 = geom_builder.MakeVertex(170.0, 100.0, 0.0)
seg1 = geom_builder.MakeVector(p3, p4)
p5 = geom_builder.MakeVertex(200.0, 70.0, 0.0)
p6 = geom_builder.MakeVertex(170.0, 40.0, 0.0)
arc2 = geom_builder.MakeArc(p4, p5, p6)
p7 = geom_builder.MakeVertex(120.0, 30.0, 0.0)
arc3 = geom_builder.MakeArc(p6, p7, p1)
wire1 = geom_builder.MakeWire([arc1, seg1, arc2, arc3])
face1 = geom_builder.MakeFace(wire1, True)
# create main shape by extruding just created planar face
pO = geom_builder.MakeVertex(0.0, 0.0, 0.0)
pz = geom_builder.MakeVertex(0.0, 0.0, 100.0)
vz = geom_builder.MakeVector(pO, pz)
prism1 = geom_builder.MakePrismVecH(face1, vz, 100.0)
# create two cylinders
pc1 = geom_builder.MakeVertex(90.0, 50.0, -40.0)
pc2 = geom_builder.MakeVertex(170.0, 70.0, -40.0)
cyl1 = geom_builder.MakeCylinder(pc1, vz, 20.0, 180.0)
cyl2 = geom_builder.MakeCylinder(pc2, vz, 20.0, 180.0)
# create final shape by cutting one cylinder and fusing with another
shape = geom_builder.MakeBoolean(prism1, cyl1, 2)
mechanic = geom_builder.MakeBoolean(shape, cyl2, 3)
# get all faces from shape and keep sub of them into variables for future processing
shape_faces = geom_builder.SubShapeAllSorted(mechanic, geom_builder.ShapeType["FACE"])
sub_face1 = shape_faces[0]
sub_face2 = shape_faces[4]
sub_face3 = shape_faces[5]
sub_face4 = shape_faces[10]
# Mesh
# ----
# create tria+tetra mesh from mechanic shape
mesh = smesh_builder.Mesh(mechanic, "mechanic")
mesh.Segment().NumberOfSegments(10)
mesh.Triangle().MaxElementArea(25)
mesh.Tetrahedron()
# create quad sub-meshes on faces sub_face1 - sub_face4
mesh.Quadrangle(sub_face1)
mesh.Quadrangle(sub_face2)
mesh.Quadrangle(sub_face3)
mesh.Quadrangle(sub_face4)
# finally compute whole mesh
if not mesh.Compute(): raise Exception("Error when computing Mesh")