smesh/test/ex21_lamp.py
2024-01-20 14:16:41 +00:00

141 lines
3.1 KiB
Python

# -*- coding: iso-8859-1 -*-
# Copyright (C) 2007-2024 CEA, EDF, OPEN CASCADE
#
# Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
# CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
#
# ==================================
#
import salome
salome.salome_init()
import GEOM
from salome.geom import geomBuilder
geompy = geomBuilder.New()
import SMESH, SALOMEDS
from salome.smesh import smeshBuilder
smesh = smeshBuilder.New()
# Geometry
# ========
# an assembly of a box, a cylinder and a truncated cone meshed with tetrahedral.
# Define values
# -------------
name = "ex21_lamp"
cote = 60
section = 20
size = 200
radius_1 = 80
radius_2 = 40
height = 100
# Build a box
# -----------
box = geompy.MakeBox(-cote, -cote, -cote, +cote, +cote, +cote)
# Build a cylinder
# ----------------
pt1 = geompy.MakeVertex(0, 0, cote/3)
di1 = geompy.MakeVectorDXDYDZ(0, 0, 1)
cyl = geompy.MakeCylinder(pt1, di1, section, size)
# Build a truncated cone
# ----------------------
pt2 = geompy.MakeVertex(0, 0, size)
cone = geompy.MakeCone(pt2, di1, radius_1, radius_2, height)
# Fuse
# ----
box_cyl = geompy.MakeFuse(box, cyl)
piece = geompy.MakeFuse(box_cyl, cone)
# Add in study
# ------------
geompy.addToStudy(piece, name)
# Create a group of faces
# -----------------------
group = geompy.CreateGroup(piece, geompy.ShapeType["FACE"])
group_name = name + "_grp"
geompy.addToStudy(group, group_name)
group.SetName(group_name)
# Add faces in the group
# ----------------------
faces = geompy.SubShapeAllIDs(piece, geompy.ShapeType["FACE"])
geompy.UnionIDs(group, faces)
# Create a mesh
# =============
# Define a mesh on a geometry
# ---------------------------
tetra = smesh.Mesh(piece, name)
# Define 1D hypothesis
# --------------------
algo1d = tetra.Segment()
algo1d.LocalLength(10)
# Define 2D hypothesis
# --------------------
algo2d = tetra.Triangle()
algo2d.LengthFromEdges()
# Define 3D hypothesis
# --------------------
algo3d = tetra.Tetrahedron(smeshBuilder.NETGEN)
algo3d.MaxElementVolume(100)
# Compute the mesh
# ----------------
isDone = tetra.Compute()
if not isDone:
raise Exception("Error when computing Mesh")
# Create a groupe of faces
# ------------------------
tetra.Group(group)
# Update object browser
# ---------------------
salome.sg.updateObjBrowser()