ForkMaster/smesh
1
0
Fork 0
mirror of https://git.salome-platform.org/gitpub/modules/smesh.git synced 2024-12-28 02:10:35 +05:00
Code Issues Packages Projects Releases Wiki Activity
48c30a5c91
smesh/src/SMESH_SWIG/StdMeshersBuilder.py
vsr a274ade365 Copyright update 2021
2021-03-23 17:44:29 +03:00

2087 lines
80 KiB
Python
Raw Blame History

# Copyright (C) 2007-2021 CEA/DEN, EDF R&D, OPEN CASCADE
#
# 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
#
"""
Python API for the standard meshing plug-in module.
"""
LIBRARY = "libStdMeshersEngine.so"
from salome.smesh.smesh_algorithm import Mesh_Algorithm
import StdMeshers
#----------------------------
# Mesh algo type identifiers
#----------------------------
REGULAR = "Regular_1D"
"""
Algorithm type: Regular 1D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_Segment`
"""
PYTHON = "Python_1D"
"""
Algorithm type: Python 1D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_Segment_Python`
"""
COMPOSITE = "CompositeSegment_1D"
"""
Algorithm type: Composite segment 1D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_CompositeSegment`
"""
MEFISTO = "MEFISTO_2D"
"""
Algorithm type: Triangle MEFISTO 2D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_Triangle_MEFISTO`
"""
Hexa = "Hexa_3D"
"""
Algorithm type: Hexahedron 3D (i-j-k) algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_Hexahedron`
"""
QUADRANGLE = "Quadrangle_2D"
"""
Algorithm type: Quadrangle 2D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_Quadrangle`
"""
RADIAL_QUAD = "RadialQuadrangle_1D2D"
"""
Algorithm type: Radial Quadrangle 1D-2D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_RadialQuadrangle1D2D`
"""
QUAD_MA_PROJ = "QuadFromMedialAxis_1D2D"
"""
Algorithm type: Quadrangle (Medial Axis Projection) 1D-2D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_QuadMA_1D2D`
"""
POLYGON = "PolygonPerFace_2D"
"""
Algorithm type: Polygon Per Face 2D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_PolygonPerFace`
"""
POLYHEDRON = "PolyhedronPerSolid_3D"
"""
Algorithm type: Polyhedron Per Solid 3D algorithm, see :class:`~StdMeshersBuilder.StdMeshersBuilder_PolyhedronPerSolid`
"""
# import items of enums
for e in StdMeshers.QuadType._items: exec('%s = StdMeshers.%s'%(e,e))
for e in StdMeshers.VLExtrusionMethod._items: exec('%s = StdMeshers.%s'%(e,e))
#----------------------
# Algorithms
#----------------------
class StdMeshersBuilder_Segment(Mesh_Algorithm):
"""
Defines segment 1D algorithm for edges discretization.
It can be created by calling smeshBuilder.Mesh.Segment(geom=0)
"""
meshMethod = "Segment"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = REGULAR
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create segment 1D algorithm for edges"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def LocalLength(self, l, UseExisting=0, p=1e-07):
"""
Defines "LocalLength" hypothesis to cut an edge in several segments with the same length
Parameters:
l : for the length of segments that cut an edge
UseExisting : if == true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
p : precision, used for calculation of the number of segments.
The precision should be a positive, meaningful value within the range [0,1].
In general, the number of segments is calculated with the formula:
nb = ceil((edge_length / l) - p)
Function ceil rounds its argument to the higher integer.
So, p=0 means rounding of (edge_length / l) to the higher integer,
p=0.5 means rounding of (edge_length / l) to the nearest integer,
p=1 means rounding of (edge_length / l) to the lower integer.
Default value is 1e-07.
Returns:
an instance of StdMeshers_LocalLength hypothesis
"""
from salome.smesh.smeshBuilder import IsEqual
comFun=lambda hyp, args: IsEqual(hyp.GetLength(), args[0]) and IsEqual(hyp.GetPrecision(), args[1])
hyp = self.Hypothesis("LocalLength", [l,p], UseExisting=UseExisting, CompareMethod=comFun)
hyp.SetLength(l)
hyp.SetPrecision(p)
return hyp
def MaxSize(self, length=0.0, UseExisting=0):
"""
Defines "MaxSize" hypothesis to cut an edge into segments not longer than given value
Parameters:
length : is optional maximal allowed length of segment, if it is omitted
the preestimated length is used that depends on geometry size
UseExisting : if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
Returns:
an instance of StdMeshers_MaxLength hypothesis
"""
hyp = self.Hypothesis("MaxLength", [length], UseExisting=UseExisting)
if isinstance(length,str) or length > 0:
# set given length
hyp.SetLength(length)
if not UseExisting:
# set preestimated length
import SMESH
gen = self.mesh.smeshpyD
initHyp = gen.GetHypothesisParameterValues("MaxLength", "libStdMeshersEngine.so",
self.mesh.GetMesh(), self.mesh.GetShape(),
SMESH.HypInitParams( 1, 1.0, False ))
preHyp = initHyp._narrow(StdMeshers.StdMeshers_MaxLength)
if preHyp:
hyp.SetPreestimatedLength( preHyp.GetPreestimatedLength() )
pass
pass
hyp.SetUsePreestimatedLength( length == 0.0 )
return hyp
def NumberOfSegments(self, n, s=[], reversedEdges=[], UseExisting=0):
"""
Defines "NumberOfSegments" hypothesis to cut an edge in a fixed number of segments
Parameters:
n: for the number of segments that cut an edge
s: for the scale factor (optional)
reversedEdges: is a list of edges to mesh using reversed orientation.
A list item can also be a tuple (edge, 1st_vertex_of_edge)
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - create a new one
Returns:
an instance of StdMeshers_NumberOfSegments hypothesis
"""
if not isinstance(reversedEdges,list): #old version script, before adding reversedEdges
reversedEdges, UseExisting = [], reversedEdges
entry = self.MainShapeEntry()
reversedEdgeInd = self.ReversedEdgeIndices(reversedEdges)
if not s:
hyp = self.Hypothesis("NumberOfSegments", [n, reversedEdgeInd, entry],
UseExisting=UseExisting,
CompareMethod=self._compareNumberOfSegments)
else:
hyp = self.Hypothesis("NumberOfSegments", [n,s, reversedEdgeInd, entry],
UseExisting=UseExisting,
CompareMethod=self._compareNumberOfSegments)
hyp.SetScaleFactor(s)
hyp.SetNumberOfSegments(n)
hyp.SetReversedEdges( reversedEdgeInd )
hyp.SetObjectEntry( entry )
return hyp
def _compareNumberOfSegments(self, hyp, args):
"""
Private method
Checks if the given "NumberOfSegments" hypothesis has the same parameters as the given arguments
"""
if hyp.GetNumberOfSegments() == args[0]:
if len(args) == 3:
if hyp.GetReversedEdges() == args[1]:
if not args[1] or hyp.GetObjectEntry() == args[2]:
return True
else:
from salome.smesh.smeshBuilder import IsEqual
if hyp.GetReversedEdges() == args[2]:
if not args[2] or hyp.GetObjectEntry() == args[3]:
if hyp.GetDistrType() == 1:
if IsEqual(hyp.GetScaleFactor(), args[1]):
return True
return False
def Adaptive(self, minSize, maxSize, deflection, UseExisting=False):
"""
Defines "Adaptive" hypothesis to cut an edge into segments keeping segment size
within the given range and considering (1) deflection of segments from the edge
and (2) distance from segments to closest edges and faces to have segment length
not longer than two times shortest distances to edges and faces.
Parameters:
minSize: defines the minimal allowed segment length
maxSize: defines the maximal allowed segment length
deflection: defines the maximal allowed distance from a segment to an edge
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
Returns:
an instance of StdMeshers_Adaptive1D hypothesis
"""
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: ( IsEqual(hyp.GetMinSize(), args[0]) and \
IsEqual(hyp.GetMaxSize(), args[1]) and \
IsEqual(hyp.GetDeflection(), args[2]))
hyp = self.Hypothesis("Adaptive1D", [minSize, maxSize, deflection],
UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetMinSize(minSize)
hyp.SetMaxSize(maxSize)
hyp.SetDeflection(deflection)
return hyp
def Arithmetic1D(self, start, end, reversedEdges=[], UseExisting=0):
"""
Defines "Arithmetic1D" hypothesis to cut an edge in several segments with a length
that changes in arithmetic progression
Parameters:
start: defines the length of the first segment
end: defines the length of the last segment
reversedEdges: is a list of edges to mesh using reversed orientation.
A list item can also be a tuple (edge, 1st_vertex_of_edge)
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
Returns:
an instance of StdMeshers_Arithmetic1D hypothesis
"""
if not isinstance(reversedEdges,list): #old version script, before adding reversedEdges
reversedEdges, UseExisting = [], reversedEdges
reversedEdgeInd = self.ReversedEdgeIndices(reversedEdges)
entry = self.MainShapeEntry()
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: ( IsEqual(hyp.GetLength(1), args[0]) and \
IsEqual(hyp.GetLength(0), args[1]) and \
hyp.GetReversedEdges() == args[2] and \
(not args[2] or hyp.GetObjectEntry() == args[3]))
hyp = self.Hypothesis("Arithmetic1D", [start, end, reversedEdgeInd, entry],
UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetStartLength(start)
hyp.SetEndLength(end)
hyp.SetReversedEdges( reversedEdgeInd )
hyp.SetObjectEntry( entry )
return hyp
def GeometricProgression(self, start, ratio, reversedEdges=[], UseExisting=0):
"""
Defines "GeometricProgression" hypothesis to cut an edge in several
segments with a length that changes in Geometric progression
Parameters:
start: defines the length of the first segment
ratio: defines the common ratio of the geometric progression
reversedEdges: is a list of edges to mesh using reversed orientation.
A list item can also be a tuple (edge, 1st_vertex_of_edge)
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
Returns:
an instance of StdMeshers_Geometric1D hypothesis
"""
reversedEdgeInd = self.ReversedEdgeIndices(reversedEdges)
entry = self.MainShapeEntry()
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: ( IsEqual(hyp.GetLength(1), args[0]) and \
IsEqual(hyp.GetLength(0), args[1]) and \
hyp.GetReversedEdges() == args[2] and \
(not args[2] or hyp.GetObjectEntry() == args[3]))
hyp = self.Hypothesis("GeometricProgression", [start, ratio, reversedEdgeInd, entry],
UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetStartLength( start )
hyp.SetCommonRatio( ratio )
hyp.SetReversedEdges( reversedEdgeInd )
hyp.SetObjectEntry( entry )
return hyp
def FixedPoints1D(self, points, nbSegs=[1], reversedEdges=[], UseExisting=0):
"""
Defines "FixedPoints1D" hypothesis to cut an edge using parameter
on curve from 0 to 1 (additionally it is neecessary to check
orientation of edges and create list of reversed edges if it is
needed) and sets numbers of segments between given points (default
values are 1)
Parameters:
points: defines the list of parameters on curve
nbSegs: defines the list of numbers of segments
reversedEdges: is a list of edges to mesh using reversed orientation.
A list item can also be a tuple (edge, 1st_vertex_of_edge)
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
Returns:
an instance of StdMeshers_FixedPoints1D hypothesis
"""
if not isinstance(reversedEdges,list): #old version script, before adding reversedEdges
reversedEdges, UseExisting = [], reversedEdges
reversedEdgeInd = self.ReversedEdgeIndices(reversedEdges)
entry = self.MainShapeEntry()
compFun = lambda hyp, args: ( hyp.GetPoints() == args[0] and \
hyp.GetNbSegments() == args[1] and \
hyp.GetReversedEdges() == args[2] and \
(not args[2] or hyp.GetObjectEntry() == args[3]))
hyp = self.Hypothesis("FixedPoints1D", [points, nbSegs, reversedEdgeInd, entry],
UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetPoints(points)
hyp.SetNbSegments(nbSegs)
hyp.SetReversedEdges(reversedEdgeInd)
hyp.SetObjectEntry(entry)
return hyp
def StartEndLength(self, start, end, reversedEdges=[], UseExisting=0):
"""
Defines "StartEndLength" hypothesis to cut an edge in several segments with increasing geometric length
Parameters:
start: defines the length of the first segment
end: defines the length of the last segment
reversedEdges: is a list of edges to mesh using reversed orientation.
A list item can also be a tuple (edge, 1st_vertex_of_edge)
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
Returns:
an instance of StdMeshers_StartEndLength hypothesis
"""
if not isinstance(reversedEdges,list): #old version script, before adding reversedEdges
reversedEdges, UseExisting = [], reversedEdges
reversedEdgeInd = self.ReversedEdgeIndices(reversedEdges)
entry = self.MainShapeEntry()
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: ( IsEqual(hyp.GetLength(1), args[0]) and \
IsEqual(hyp.GetLength(0), args[1]) and \
hyp.GetReversedEdges() == args[2] and \
(not args[2] or hyp.GetObjectEntry() == args[3]))
hyp = self.Hypothesis("StartEndLength", [start, end, reversedEdgeInd, entry],
UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetStartLength(start)
hyp.SetEndLength(end)
hyp.SetReversedEdges( reversedEdgeInd )
hyp.SetObjectEntry( entry )
return hyp
def Deflection1D(self, d, UseExisting=0):
"""
Defines "Deflection1D" hypothesis
Parameters:
d: for the deflection
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - create a new one
"""
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: IsEqual(hyp.GetDeflection(), args[0])
hyp = self.Hypothesis("Deflection1D", [d], UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetDeflection(d)
return hyp
def Propagation(self):
"""
Defines "Propagation" hypothesis that propagates 1D hypotheses
from an edge where this hypothesis is assigned to
on all other edges that are at the opposite side in case of quadrangular faces
This hypothesis should be assigned to an edge to propagate a hypothesis from.
"""
return self.Hypothesis("Propagation", UseExisting=1, CompareMethod=self.CompareEqualHyp)
def PropagationOfDistribution(self):
"""
Defines "Propagation of Node Distribution" hypothesis that propagates
distribution of nodes from an edge where this hypothesis is assigned to,
to opposite edges of quadrangular faces, so that number of segments on all these
edges will be the same, as well as relations between segment lengths.
"""
return self.Hypothesis("PropagOfDistribution", UseExisting=1,
CompareMethod=self.CompareEqualHyp)
def AutomaticLength(self, fineness=0, UseExisting=0):
"""
Defines "AutomaticLength" hypothesis
Parameters:
fineness: for the fineness [0-1]
UseExisting: if ==true - searches for an existing hypothesis created with the
same parameters, else (default) - create a new one
"""
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: IsEqual(hyp.GetFineness(), args[0])
hyp = self.Hypothesis("AutomaticLength",[fineness],UseExisting=UseExisting,
CompareMethod=compFun)
hyp.SetFineness( fineness )
return hyp
def LengthNearVertex(self, length, vertex=-1, UseExisting=0):
"""
Defines "SegmentLengthAroundVertex" hypothesis
Parameters:
length: for the segment length
vertex: for the length localization: the vertex index [0,1] | vertex object.
Any other integer value means that the hypothesis will be set on the
whole 1D shape, where Mesh_Segment algorithm is assigned.
UseExisting: if ==true - searches for an existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
import types
store_geom = self.geom
if isinstance(vertex, int):
if vertex == 0 or vertex == 1:
from salome.geom import geomBuilder
vertex = self.mesh.geompyD.ExtractShapes(self.geom, geomBuilder.geomBuilder.ShapeType["VERTEX"],True)[vertex]
self.geom = vertex
pass
pass
else:
self.geom = vertex
pass
# 0D algorithm
if self.geom is None:
self.geom = store_geom
raise RuntimeError("Attempt to create SegmentAroundVertex_0D algorithm on None shape")
from salome.smesh.smeshBuilder import AssureGeomPublished, GetName, TreatHypoStatus
AssureGeomPublished( self.mesh, self.geom )
name = GetName(self.geom)
algo = self.FindAlgorithm("SegmentAroundVertex_0D", self.mesh.smeshpyD)
if algo is None:
algo = self.mesh.smeshpyD.CreateHypothesis("SegmentAroundVertex_0D", "libStdMeshersEngine.so")
pass
status = self.mesh.mesh.AddHypothesis(self.geom, algo)
TreatHypoStatus(status, "SegmentAroundVertex_0D", name, True, self.mesh)
#
from salome.smesh.smeshBuilder import IsEqual
comFun = lambda hyp, args: IsEqual(hyp.GetLength(), args[0])
hyp = self.Hypothesis("SegmentLengthAroundVertex", [length], UseExisting=UseExisting,
CompareMethod=comFun)
self.geom = store_geom
hyp.SetLength( length )
return hyp
def QuadraticMesh(self):
"""
Defines "QuadraticMesh" hypothesis, forcing construction of quadratic edges.
If the 2D mesher sees that all boundary edges are quadratic,
it generates quadratic faces, else it generates linear faces using
medium nodes as if they are vertices.
The 3D mesher generates quadratic volumes only if all boundary faces
are quadratic, else it fails.
"""
hyp = self.Hypothesis("QuadraticMesh", UseExisting=1, CompareMethod=self.CompareEqualHyp)
return hyp
pass # end of StdMeshersBuilder_Segment class
class StdMeshersBuilder_CompositeSegment(StdMeshersBuilder_Segment):
"""
Segment 1D algorithm for discretization of a set of adjacent edges as one edge.
It is created by calling smeshBuilder.Mesh.Segment(smeshBuilder.COMPOSITE,geom=0)
"""
meshMethod = "Segment"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = COMPOSITE
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = False
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create segment 1D algorithm for edges"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
self.Create(mesh, geom, self.algoType)
pass
pass # end of StdMeshersBuilder_CompositeSegment class
class StdMeshersBuilder_Segment_Python(Mesh_Algorithm):
"""
Defines a segment 1D algorithm for discretization of edges with Python function.
It is created by calling smeshBuilder.Mesh.Segment(smeshBuilder.PYTHON,geom=0)
"""
meshMethod = "Segment"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = PYTHON
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create segment 1D algorithm for edges"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
import Python1dPlugin
self.Create(mesh, geom, self.algoType, "libPython1dEngine.so")
pass
def PythonSplit1D(self, n, func, UseExisting=0):
"""
Defines "PythonSplit1D" hypothesis
Parameters:
n: for the number of segments that cut an edge
func: for the python function that calculates the length of all segments
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
compFun = lambda hyp, args: False
hyp = self.Hypothesis("PythonSplit1D", [n], "libPython1dEngine.so",
UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetNumberOfSegments(n)
hyp.SetPythonLog10RatioFunction(func)
return hyp
pass # end of StdMeshersBuilder_Segment_Python class
class StdMeshersBuilder_Triangle_MEFISTO(Mesh_Algorithm):
"""
Triangle MEFISTO 2D algorithm.
It is created by calling smeshBuilder.Mesh.Triangle(smeshBuilder.MEFISTO,geom=0)
"""
meshMethod = "Triangle"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = MEFISTO
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create triangle 2D algorithm for faces"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def MaxElementArea(self, area, UseExisting=0):
"""
Defines "MaxElementArea" hypothesis basing on the definition of the maximum area of each triangle
Parameters:
area: for the maximum area of each triangle
UseExisting: if ==true - searches for an existing hypothesis created with the
same parameters, else (default) - Create a new one
"""
from salome.smesh.smeshBuilder import IsEqual
comparator = lambda hyp, args: IsEqual(hyp.GetMaxElementArea(), args[0])
hyp = self.Hypothesis("MaxElementArea", [area], UseExisting=UseExisting,
CompareMethod=comparator)
hyp.SetMaxElementArea(area)
return hyp
def LengthFromEdges(self):
"""
Defines "LengthFromEdges" hypothesis to build triangles
based on the length of the edges taken from the wire
"""
hyp = self.Hypothesis("LengthFromEdges", UseExisting=1, CompareMethod=self.CompareEqualHyp)
return hyp
pass # end of StdMeshersBuilder_Triangle_MEFISTO class
class StdMeshersBuilder_Quadrangle(Mesh_Algorithm):
"""
Defines a quadrangle 2D algorithm.
It is created by calling smeshBuilder.Mesh.Quadrangle(geom=0)
"""
meshMethod = "Quadrangle"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = QUADRANGLE
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create quadrangle 2D algorithm for faces"
"""
doc string of the method
"""
params = 0
"""
hypothesis associated with algorithm
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def QuadrangleParameters(self, quadType=StdMeshers.QUAD_STANDARD, triangleVertex=0,
enfVertices=[],enfPoints=[],corners=[],UseExisting=0):
"""
Defines "QuadrangleParameters" hypothesis
quadType defines the algorithm of transition between differently descretized
sides of a geometrical face:
- QUAD_STANDARD - both triangles and quadrangles are possible in the transition
area along the finer meshed sides.
- QUAD_TRIANGLE_PREF - only triangles are built in the transition area along the
finer meshed sides.
- QUAD_QUADRANGLE_PREF - only quadrangles are built in the transition area along
the finer meshed sides, iff the total quantity of segments on
all four sides of the face is even (divisible by 2).
- QUAD_QUADRANGLE_PREF_REVERSED - same as QUAD_QUADRANGLE_PREF but the transition
area is located along the coarser meshed sides.
- QUAD_REDUCED - only quadrangles are built and the transition between the sides
is made gradually, layer by layer. This type has a limitation on
the number of segments: one pair of opposite sides must have the
same number of segments, the other pair must have an even difference
between the numbers of segments on the sides.
Parameters:
triangleVertex: vertex of a trilateral geometrical face, around which triangles
will be created while other elements will be quadrangles.
Vertex can be either a GEOM_Object or a vertex ID within the
shape to mesh
enfVertices: list of shapes defining positions where nodes (enforced nodes)
must be created by the mesher. Shapes can be of any type,
vertices of given shapes define positions of enforced nodes.
Only vertices successfully projected to the face are used.
enfPoints: list of points giving positions of enforced nodes.
Point can be defined either as SMESH.PointStruct's
([SMESH.PointStruct(x1,y1,z1), SMESH.PointStruct(x2,y2,z2),...])
or triples of values ([[x1,y1,z1], [x2,y2,z2], ...]).
In the case if the defined QuadrangleParameters() refer to a sole face,
all given points must lie on this face, else the mesher fails.
corners: list of vertices that should be used as quadrangle corners.
The parameter can be useful for faces with more than four vertices,
since in some cases Quadrangle Mapping algorithm chooses corner vertices
differently than it is desired.
A hypothesis can be global and define corners for all CAD faces that
require it, but be sure that each specified vertex is a corner in all
faces the hypothesis will be applied to.
UseExisting: if *True* - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
import GEOM, SMESH
vertexID = triangleVertex
if isinstance( triangleVertex, GEOM._objref_GEOM_Object ):
vertexID = self.mesh.geompyD.GetSubShapeID( self.mesh.geom, triangleVertex )
if isinstance( enfVertices, int ) and not enfPoints and not UseExisting:
# a call of old syntax, before inserting enfVertices and enfPoints before UseExisting
UseExisting, enfVertices = enfVertices, []
pStructs, xyz = [], []
for p in enfPoints:
if isinstance( p, SMESH.PointStruct ):
xyz.append(( p.x, p.y, p.z ))
pStructs.append( p )
else:
xyz.append(( p[0], p[1], p[2] ))
pStructs.append( SMESH.PointStruct( p[0], p[1], p[2] ))
if not self.params:
compFun = lambda hyp,args: \
hyp.GetQuadType() == args[0] and \
(hyp.GetTriaVertex()==args[1] or ( hyp.GetTriaVertex()<1 and args[1]<1)) and \
((hyp.GetEnforcedNodes()) == (args[2],args[3])) # True w/o enfVertices only
entries = [ shape.GetStudyEntry() for shape in enfVertices ]
self.params = self.Hypothesis("QuadrangleParams", [quadType,vertexID,entries,xyz],
UseExisting = UseExisting, CompareMethod=compFun)
pass
if corners and isinstance( corners[0], GEOM._objref_GEOM_Object ):
corners = [ self.mesh.geompyD.GetSubShapeID( self.mesh.geom, v ) for v in corners ]
if self.params.GetQuadType() != quadType:
self.params.SetQuadType(quadType)
if vertexID > 0:
self.params.SetTriaVertex( vertexID )
from salome.smesh.smeshBuilder import AssureGeomPublished
for v in enfVertices:
AssureGeomPublished( self.mesh, v )
self.params.SetEnforcedNodes( enfVertices, pStructs )
self.params.SetCorners( corners )
return self.params
def QuadranglePreference(self, reversed=False, UseExisting=0):
"""
Defines "QuadrangleParams" hypothesis with a type of quadrangulation that only
quadrangles are built in the transition area along the finer meshed sides,
if the total quantity of segments on all four sides of the face is even.
Parameters:
reversed: if True, transition area is located along the coarser meshed sides.
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
if reversed:
return self.QuadrangleParameters(QUAD_QUADRANGLE_PREF_REVERSED,UseExisting=UseExisting)
return self.QuadrangleParameters(QUAD_QUADRANGLE_PREF,UseExisting=UseExisting)
def TrianglePreference(self, UseExisting=0):
"""
Defines "QuadrangleParams" hypothesis with a type of quadrangulation that only
triangles are built in the transition area along the finer meshed sides.
Parameters:
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
return self.QuadrangleParameters(QUAD_TRIANGLE_PREF,UseExisting=UseExisting)
def Reduced(self, UseExisting=0):
"""
Defines "QuadrangleParams" hypothesis with a type of quadrangulation that only
quadrangles are built and the transition between the sides is made gradually,
layer by layer. This type has a limitation on the number of segments: one pair
of opposite sides must have the same number of segments, the other pair must
have an even difference between the numbers of segments on the sides.
Parameters:
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
return self.QuadrangleParameters(QUAD_REDUCED,UseExisting=UseExisting)
def TriangleVertex(self, vertex, UseExisting=0):
"""
Defines "QuadrangleParams" hypothesis with QUAD_STANDARD type of quadrangulation
Parameters:
vertex: vertex of a trilateral geometrical face, around which triangles
will be created while other elements will be quadrangles.
Vertex can be either a GEOM_Object or a vertex ID within the
shape to mesh
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
return self.QuadrangleParameters(QUAD_STANDARD,vertex,UseExisting)
pass # end of StdMeshersBuilder_Quadrangle class
class StdMeshersBuilder_Hexahedron(Mesh_Algorithm):
"""
Defines a hexahedron 3D algorithm.
It is created by calling smeshBuilder.Mesh.Hexahedron(geom=0)
"""
meshMethod = "Hexahedron"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = Hexa
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create hexahedron 3D algorithm for volumes"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, Hexa)
self.renumHypothesis = 0
pass
def Renumber(self, blockCSList=[] ):
if isinstance( blockCSList, StdMeshers.BlockCS ):
blockCSList = [blockCSList]
if not self.renumHypothesis:
self.renumHypothesis = self.Hypothesis("BlockRenumber", blockCSList, UseExisting=0)
self.renumHypothesis.SetBlocksOrientation( blockCSList )
return self.renumHypothesis
pass # end of StdMeshersBuilder_Hexahedron class
class StdMeshersBuilder_Projection1D(Mesh_Algorithm):
"""
Defines a projection 1D algorithm.
It is created by calling smeshBuilder.Mesh.Projection1D(geom=0)
"""
meshMethod = "Projection1D"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Projection_1D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create projection 1D algorithm for edges"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def SourceEdge(self, edge, mesh=None, srcV=None, tgtV=None, UseExisting=0):
"""
Defines "Source Edge" hypothesis, specifying a meshed edge, from where
a mesh pattern is taken, and, optionally, the association of vertices
between the source edge and a target edge (to which a hypothesis is assigned)
Parameters:
edge: from which nodes distribution is taken
mesh: from which nodes distribution is taken (optional)
srcV: a vertex of *edge* to associate with *tgtV* (optional)
tgtV: a vertex of *the edge* to which the algorithm is assigned, to associate with *srcV* (optional)
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
from salome.smesh.smeshBuilder import AssureGeomPublished, Mesh
AssureGeomPublished( self.mesh, edge )
AssureGeomPublished( self.mesh, srcV )
AssureGeomPublished( self.mesh, tgtV )
hyp = self.Hypothesis("ProjectionSource1D", [edge,mesh,srcV,tgtV],
UseExisting=0)
# it does not seem to be useful to reuse the existing "SourceEdge" hypothesis
#UseExisting=UseExisting, CompareMethod=self.CompareSourceEdge)
hyp.SetSourceEdge( edge )
if not mesh is None and isinstance(mesh, Mesh):
mesh = mesh.GetMesh()
hyp.SetSourceMesh( mesh )
hyp.SetVertexAssociation( srcV, tgtV )
return hyp
pass # end of StdMeshersBuilder_Projection1D class
class StdMeshersBuilder_Projection2D(Mesh_Algorithm):
"""
Defines a projection 2D algorithm.
It is created by calling smeshBuilder.Mesh.Projection2D(geom=0)
"""
meshMethod = "Projection2D"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Projection_2D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create projection 2D algorithm for faces"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def SourceFace(self, face, mesh=None, srcV1=None, tgtV1=None,
srcV2=None, tgtV2=None, UseExisting=0):
"""
Defines "Source Face" hypothesis, specifying a meshed face, from where
a mesh pattern is taken, and, optionally, the association of vertices
between the source face and the target face (to which a hypothesis is assigned)
Parameters:
face: from which the mesh pattern is taken
mesh: from which the mesh pattern is taken (optional)
srcV1: a vertex of *face* to associate with *tgtV1* (optional)
tgtV1: a vertex of *the face* to which the algorithm is assigned, to associate with *srcV1* (optional)
srcV2: a vertex of *face* to associate with *tgtV1* (optional)
tgtV2: a vertex of *the face* to which the algorithm is assigned, to associate with *srcV2* (optional)
UseExisting: if ==true - forces the search for the existing hypothesis created with
he same parameters, else (default) - forces the creation a new one
Note:
all association vertices must belong to one edge of a face
"""
from salome.smesh.smeshBuilder import Mesh
if isinstance(mesh, Mesh):
mesh = mesh.GetMesh()
for geom in [ face, srcV1, tgtV1, srcV2, tgtV2 ]:
from salome.smesh.smeshBuilder import AssureGeomPublished
AssureGeomPublished( self.mesh, geom )
hyp = self.Hypothesis("ProjectionSource2D", [face,mesh,srcV1,tgtV1,srcV2,tgtV2],
UseExisting=0, toAdd=False)
# it does not seem to be useful to reuse the existing "SourceFace" hypothesis
#UseExisting=UseExisting, CompareMethod=self.CompareSourceFace)
hyp.SetSourceFace( face )
hyp.SetSourceMesh( mesh )
hyp.SetVertexAssociation( srcV1, srcV2, tgtV1, tgtV2 )
self.mesh.AddHypothesis(hyp, self.geom)
return hyp
pass # end of StdMeshersBuilder_Projection2D class
class StdMeshersBuilder_Projection1D2D(StdMeshersBuilder_Projection2D):
"""
Defines a projection 1D-2D algorithm.
It is created by calling smeshBuilder.Mesh.Projection1D2D(geom=0)
"""
meshMethod = "Projection1D2D"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Projection_1D2D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create projection 1D-2D algorithm for faces"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
StdMeshersBuilder_Projection2D.__init__(self, mesh, geom)
pass
pass # end of StdMeshersBuilder_Projection1D2D class
class StdMeshersBuilder_Projection3D(Mesh_Algorithm):
"""
Defines a projection 3D algorithm.
It is created by calling smeshBuilder.Mesh.Projection3D(geom=0)
"""
meshMethod = "Projection3D"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Projection_3D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create projection 3D algorithm for volumes"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom" geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def SourceShape3D(self, solid, mesh=0, srcV1=0, tgtV1=0,
srcV2=0, tgtV2=0, UseExisting=0):
"""
Defines the "Source Shape 3D" hypothesis, specifying a meshed solid, from where
the mesh pattern is taken, and, optionally, the association of vertices
between the source and the target solid (to which a hipothesis is assigned)
Parameters:
solid: from where the mesh pattern is taken
mesh: from where the mesh pattern is taken (optional)
srcV1: a vertex of *solid* to associate with *tgtV1* (optional)
tgtV1: a vertex of *the solid* where the algorithm is assigned, to associate with *srcV1* (optional)
srcV2: a vertex of *solid* to associate with *tgtV1* (optional)
tgtV2: a vertex of *the solid* to which the algorithm is assigned,to associate with *srcV2* (optional)
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
Note:
association vertices must belong to one edge of a solid
"""
for geom in [ solid, srcV1, tgtV1, srcV2, tgtV2 ]:
from salome.smesh.smeshBuilder import AssureGeomPublished
AssureGeomPublished( self.mesh, geom )
hyp = self.Hypothesis("ProjectionSource3D",
[solid,mesh,srcV1,tgtV1,srcV2,tgtV2],
UseExisting=0)
# seems to be not really useful to reuse existing "SourceShape3D" hypothesis
#UseExisting=UseExisting, CompareMethod=self.CompareSourceShape3D)
hyp.SetSource3DShape( solid )
from salome.smesh.smeshBuilder import Mesh
if isinstance(mesh, Mesh):
mesh = mesh.GetMesh()
if mesh:
hyp.SetSourceMesh( mesh )
if srcV1 and srcV2 and tgtV1 and tgtV2:
hyp.SetVertexAssociation( srcV1, srcV2, tgtV1, tgtV2 )
#elif srcV1 or srcV2 or tgtV1 or tgtV2:
return hyp
pass # end of StdMeshersBuilder_Projection3D class
class StdMeshersBuilder_Prism3D(Mesh_Algorithm):
"""
Defines a Prism 3D algorithm, which is either "Extrusion 3D" or "Radial Prism" depending on geometry.
It is created by calling smeshBuilder.Mesh.Prism(geom=0)
"""
meshMethod = "Prism"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Prism_3D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create prism 3D algorithm for volumes"
"""
doc string of the method
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
shape = geom
if not shape:
shape = mesh.geom
isRadial = mesh.smeshpyD.IsApplicable("RadialPrism_3D", LIBRARY, shape, False )
if not isRadial:
self.Create(mesh, geom, "Prism_3D")
pass
else:
self.algoType = "RadialPrism_3D"
self.Create(mesh, geom, "RadialPrism_3D")
self.distribHyp = None #self.Hypothesis("LayerDistribution", UseExisting=0)
self.nbLayers = None
pass
pass
def Get3DHypothesis(self):
"""
Returns:
3D hypothesis holding the 1D one
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
return self.distribHyp
def OwnHypothesis(self, hypType, args=[], so="libStdMeshersEngine.so"):
"""
Private method creating a 1D hypothesis and storing it in the LayerDistribution
hypothesis.
Returns:
the created hypothesis
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
if not self.nbLayers is None:
self.mesh.GetMesh().RemoveHypothesis( self.geom, self.nbLayers )
self.mesh.GetMesh().AddHypothesis( self.geom, self.distribHyp )
self.mesh.smeshpyD.SetEnablePublish( False ) # prevents publishing own 1D hypothesis
hyp = self.mesh.smeshpyD.CreateHypothesis(hypType, so)
self.mesh.smeshpyD.SetEnablePublish( True ) # enables publishing
if not self.distribHyp:
self.distribHyp = self.Hypothesis("LayerDistribution", UseExisting=0)
self.distribHyp.SetLayerDistribution( hyp )
return hyp
def NumberOfLayers(self, n, UseExisting=0):
"""
Defines "NumberOfLayers" hypothesis, specifying the number of layers of
prisms to build between the inner and outer shells
Parameters:
n: number of layers
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
self.mesh.RemoveHypothesis( self.distribHyp, self.geom )
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: IsEqual(hyp.GetNumberOfLayers(), args[0])
self.nbLayers = self.Hypothesis("NumberOfLayers", [n], UseExisting=UseExisting,
CompareMethod=compFun)
self.nbLayers.SetNumberOfLayers( n )
return self.nbLayers
def LocalLength(self, l, p=1e-07):
"""
Defines "LocalLength" hypothesis, specifying the segment length
to build between the inner and the outer shells
Parameters:
l: the length of segments
p: the precision of rounding
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
hyp = self.OwnHypothesis("LocalLength", [l,p])
hyp.SetLength(l)
hyp.SetPrecision(p)
return hyp
def NumberOfSegments(self, n, s=[]):
"""
Defines "NumberOfSegments" hypothesis, specifying the number of layers of
prisms to build between the inner and the outer shells.
Parameters:
n: the number of layers
s: the scale factor (optional)
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
if not s:
hyp = self.OwnHypothesis("NumberOfSegments", [n])
else:
hyp = self.OwnHypothesis("NumberOfSegments", [n,s])
hyp.SetScaleFactor(s)
hyp.SetNumberOfSegments(n)
return hyp
def Arithmetic1D(self, start, end ):
"""
Defines "Arithmetic1D" hypothesis, specifying the distribution of segments
to build between the inner and the outer shells with a length that changes
in arithmetic progression
Parameters:
start: the length of the first segment
end: the length of the last segment
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
hyp = self.OwnHypothesis("Arithmetic1D", [start, end])
hyp.SetLength(start, 1)
hyp.SetLength(end , 0)
return hyp
def GeometricProgression(self, start, ratio ):
"""
Defines "GeometricProgression" hypothesis, specifying the distribution of segments
to build between the inner and the outer shells with a length that changes
in Geometric progression
Parameters:
start: the length of the first segment
ratio: the common ratio of the geometric progression
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
hyp = self.OwnHypothesis("GeometricProgression", [start, ratio])
hyp.SetStartLength( start )
hyp.SetCommonRatio( ratio )
return hyp
def StartEndLength(self, start, end):
"""
Defines "StartEndLength" hypothesis, specifying distribution of segments
to build between the inner and the outer shells as geometric length increasing
Parameters:
start: for the length of the first segment
end: for the length of the last segment
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
hyp = self.OwnHypothesis("StartEndLength", [start, end])
hyp.SetLength(start, 1)
hyp.SetLength(end , 0)
return hyp
def AutomaticLength(self, fineness=0):
"""
Defines "AutomaticLength" hypothesis, specifying the number of segments
to build between the inner and outer shells
Parameters:
fineness: defines the quality of the mesh within the range [0-1]
"""
if self.algoType != "RadialPrism_3D":
print("Prism_3D algorithm doesn't support any hypothesis")
return None
hyp = self.OwnHypothesis("AutomaticLength")
hyp.SetFineness( fineness )
return hyp
pass # end of StdMeshersBuilder_Prism3D class
class StdMeshersBuilder_RadialPrism3D(StdMeshersBuilder_Prism3D):
"""
Defines Radial Prism 3D algorithm.
It is created by calling smeshBuilder.Mesh.Prism(geom=0).
See :class:`StdMeshersBuilder_Prism3D` for methods defining distribution of mesh layers
build between the inner and outer shells.
"""
meshMethod = "Prism"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "RadialPrism_3D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create Raial Prism 3D algorithm for volumes"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
shape = geom
if not shape:
shape = mesh.geom
self.Create(mesh, geom, "RadialPrism_3D")
self.distribHyp = None
self.nbLayers = None
return
class StdMeshersBuilder_RadialAlgorithm(Mesh_Algorithm):
"""
Base class for algorithms supporting radial distribution hypotheses
"""
def __init__(self):
Mesh_Algorithm.__init__(self)
self.distribHyp = None #self.Hypothesis("LayerDistribution2D", UseExisting=0)
self.nbLayers = None
pass
def Get2DHypothesis(self):
"""
Returns:
2D hypothesis holding the 1D one
"""
if not self.distribHyp:
self.distribHyp = self.Hypothesis("LayerDistribution2D", UseExisting=0)
return self.distribHyp
def OwnHypothesis(self, hypType, args=[], so="libStdMeshersEngine.so"):
"""
Private method creating a 1D hypothesis and storing it in the LayerDistribution
hypothesis.
Returns:
the created hypothesis
"""
if self.nbLayers:
self.mesh.GetMesh().RemoveHypothesis( self.geom, self.nbLayers )
if self.distribHyp is None:
self.distribHyp = self.Hypothesis("LayerDistribution2D", UseExisting=0)
else:
self.mesh.GetMesh().AddHypothesis( self.geom, self.distribHyp )
self.mesh.smeshpyD.SetEnablePublish( False )
hyp = self.mesh.smeshpyD.CreateHypothesis(hypType, so)
self.mesh.smeshpyD.SetEnablePublish( True )
self.distribHyp.SetLayerDistribution( hyp )
return hyp
def NumberOfLayers(self, n, UseExisting=0):
"""
Defines "NumberOfLayers" hypothesis, specifying the number of layers
Parameters:
n: number of layers
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
if self.distribHyp:
self.mesh.GetMesh().RemoveHypothesis( self.geom, self.distribHyp )
from salome.smesh.smeshBuilder import IsEqual
compFun = lambda hyp, args: IsEqual(hyp.GetNumberOfLayers(), args[0])
self.nbLayers = self.Hypothesis("NumberOfLayers2D", [n], UseExisting=UseExisting,
CompareMethod=compFun)
self.nbLayers.SetNumberOfLayers( n )
return self.nbLayers
def LocalLength(self, l, p=1e-07):
"""
Defines "LocalLength" hypothesis, specifying the segment length
Parameters:
l: the length of segments
p: the precision of rounding
"""
hyp = self.OwnHypothesis("LocalLength", [l,p])
hyp.SetLength(l)
hyp.SetPrecision(p)
return hyp
def NumberOfSegments(self, n, s=[]):
"""
Defines "NumberOfSegments" hypothesis, specifying the number of layers
Parameters:
n: the number of layers
s: the scale factor (optional)
"""
if s == []:
hyp = self.OwnHypothesis("NumberOfSegments", [n])
else:
hyp = self.OwnHypothesis("NumberOfSegments", [n,s])
hyp.SetDistrType( 1 )
hyp.SetScaleFactor(s)
hyp.SetNumberOfSegments(n)
return hyp
def Arithmetic1D(self, start, end ):
"""
Defines "Arithmetic1D" hypothesis, specifying the distribution of segments
with a length that changes in arithmetic progression
Parameters:
start: the length of the first segment
end: the length of the last segment
"""
hyp = self.OwnHypothesis("Arithmetic1D", [start, end])
hyp.SetLength(start, 1)
hyp.SetLength(end , 0)
return hyp
def GeometricProgression(self, start, ratio ):
"""
Defines "GeometricProgression" hypothesis, specifying the distribution of segments
with a length that changes in Geometric progression
Parameters:
start: the length of the first segment
ratio: the common ratio of the geometric progression
"""
hyp = self.OwnHypothesis("GeometricProgression", [start, ratio])
hyp.SetStartLength( start )
hyp.SetCommonRatio( ratio )
return hyp
def StartEndLength(self, start, end):
"""
Defines "StartEndLength" hypothesis, specifying distribution of segments
as geometric length increasing
Parameters:
start: for the length of the first segment
end: for the length of the last segment
"""
hyp = self.OwnHypothesis("StartEndLength", [start, end])
hyp.SetLength(start, 1)
hyp.SetLength(end , 0)
return hyp
def AutomaticLength(self, fineness=0):
"""
Defines "AutomaticLength" hypothesis, specifying the number of segments
Parameters:
fineness: defines the quality of the mesh within the range [0-1]
"""
hyp = self.OwnHypothesis("AutomaticLength")
hyp.SetFineness( fineness )
return hyp
pass # end of StdMeshersBuilder_RadialQuadrangle1D2D class
class StdMeshersBuilder_RadialQuadrangle1D2D(StdMeshersBuilder_RadialAlgorithm):
"""
Defines a Radial Quadrangle 1D-2D algorithm.
It is created by calling smeshBuilder.Mesh.Quadrangle(smeshBuilder.RADIAL_QUAD,geom=0)
"""
meshMethod = "Quadrangle"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = RADIAL_QUAD
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create quadrangle 1D-2D algorithm for faces having a shape of disk or a disk segment"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
StdMeshersBuilder_RadialAlgorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
self.distribHyp = None #self.Hypothesis("LayerDistribution2D", UseExisting=0)
self.nbLayers = None
pass
class StdMeshersBuilder_QuadMA_1D2D(StdMeshersBuilder_RadialAlgorithm):
"""
Defines a Quadrangle (Medial Axis Projection) 1D-2D algorithm .
It is created by calling smeshBuilder.Mesh.Quadrangle(smeshBuilder.QUAD_MA_PROJ,geom=0)
"""
meshMethod = "Quadrangle"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = QUAD_MA_PROJ
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create quadrangle 1D-2D algorithm for faces"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
StdMeshersBuilder_RadialAlgorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
pass
class StdMeshersBuilder_PolygonPerFace(Mesh_Algorithm):
""" Defines a Polygon Per Face 2D algorithm.
It is created by calling smeshBuilder.Mesh.Polygon(geom=0)
"""
meshMethod = "Polygon"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = POLYGON
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create polygon 2D algorithm for faces"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
pass
class StdMeshersBuilder_PolyhedronPerSolid(Mesh_Algorithm):
""" Defines a Polyhedron Per Solid 3D algorithm.
It is created by calling smeshBuilder.Mesh.Polyhedron(geom=0)
"""
meshMethod = "Polyhedron"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = POLYHEDRON
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create polyhedron 3D algorithm for solids"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
pass
class StdMeshersBuilder_UseExistingElements_1D(Mesh_Algorithm):
""" Defines a Use Existing Elements 1D algorithm.
It is created by calling smeshBuilder.Mesh.UseExisting1DElements(geom=0)
"""
meshMethod = "UseExisting1DElements"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Import_1D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create 1D algorithm for edges with reusing of existing mesh elements"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def SourceEdges(self, groups, toCopyMesh=False, toCopyGroups=False, UseExisting=False):
"""
Defines "Source edges" hypothesis, specifying groups of edges to import
Parameters:
groups: list of groups of edges
toCopyMesh: if True, the whole mesh *groups* belong to is imported
toCopyGroups: if True, all groups of the mesh *groups* belong to are imported
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
for group in groups:
from salome.smesh.smeshBuilder import AssureGeomPublished
AssureGeomPublished( self.mesh, group )
compFun = lambda hyp, args: ( hyp.GetSourceEdges() == args[0] and \
hyp.GetCopySourceMesh() == args[1], args[2] )
hyp = self.Hypothesis("ImportSource1D", [groups, toCopyMesh, toCopyGroups],
UseExisting=UseExisting, CompareMethod=compFun)
hyp.SetSourceEdges(groups)
hyp.SetCopySourceMesh(toCopyMesh, toCopyGroups)
return hyp
pass # end of StdMeshersBuilder_UseExistingElements_1D class
class StdMeshersBuilder_UseExistingElements_1D2D(Mesh_Algorithm):
""" Defines a Use Existing Elements 1D-2D algorithm.
It is created by calling smeshBuilder.Mesh.UseExisting2DElements(geom=0)
"""
meshMethod = "UseExisting2DElements"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Import_1D2D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create 1D-2D algorithm for faces with reusing of existing mesh elements"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
Mesh_Algorithm.__init__(self)
self.Create(mesh, geom, self.algoType)
pass
def SourceFaces(self, groups, toCopyMesh=False, toCopyGroups=False, UseExisting=False):
"""
Defines "Source faces" hypothesis, specifying groups of faces to import
Parameters:
groups: list of groups of faces
toCopyMesh: if True, the whole mesh *groups* belong to is imported
toCopyGroups: if True, all groups of the mesh *groups* belong to are imported
UseExisting: if ==true - searches for the existing hypothesis created with
the same parameters, else (default) - Create a new one
"""
import SMESH
compFun = lambda hyp, args: ( hyp.GetSourceFaces() == args[0] and \
hyp.GetCopySourceMesh() == args[1], args[2] )
hyp = self.Hypothesis("ImportSource2D", [groups, toCopyMesh, toCopyGroups],
UseExisting=UseExisting, CompareMethod=compFun, toAdd=False)
if groups and isinstance( groups, SMESH._objref_SMESH_GroupBase ):
groups = [groups]
hyp.SetSourceFaces(groups)
hyp.SetCopySourceMesh(toCopyMesh, toCopyGroups)
self.mesh.AddHypothesis(hyp, self.geom)
return hyp
pass # end of StdMeshersBuilder_UseExistingElements_1D2D class
class StdMeshersBuilder_Cartesian_3D(Mesh_Algorithm):
""" Defines a Body Fitting 3D algorithm.
It is created by calling smeshBuilder.Mesh.BodyFitted(geom=0)
"""
meshMethod = "BodyFitted"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "Cartesian_3D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
isDefault = True
"""
flag pointing whether this algorithm should be used by default in dynamic method
of smeshBuilder.Mesh class
"""
docHelper = "Create Body Fitting 3D algorithm for volumes"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
self.Create(mesh, geom, self.algoType)
self.hyp = None
pass
def SetGrid(self, xGridDef, yGridDef, zGridDef, sizeThreshold=4.0, implEdges=False):
"""
Defines "Body Fitting parameters" hypothesis
Parameters:
xGridDef: is definition of the grid along the X asix.
It can be in either of two following forms:
- Explicit coordinates of nodes, e.g. [-1.5, 0.0, 3.1] or range( -100,200,10)
- Functions f(t) defining grid spacing at each point on grid axis. If there are
several functions, they must be accompanied by relative coordinates of
points dividing the whole shape into ranges where the functions apply; points
coordinates should vary within (0.0, 1.0) range. Parameter *t* of the spacing
function f(t) varies from 0.0 to 1.0 within a shape range.
Note:
The actual grid spacing can slightly differ from the defined one. This is done for the
best fitting of polyhedrons and for a better mesh quality on the interval boundaries.
For example, if a constant **Spacing** is defined along an axis, the actual grid will
fill the shape's dimension L along this axis with round number of equal cells:
Spacing_actual = L / round( L / Spacing_defined ).
Examples:
"10.5" - defines a grid with a constant spacing
[["1", "1+10*t", "11"] [0.1, 0.6]] - defines different spacing in 3 ranges.
Parameters:
yGridDef: defines the grid along the Y asix the same way as *xGridDef* does.
zGridDef: defines the grid along the Z asix the same way as *xGridDef* does.
sizeThreshold: (> 1.0) defines a minimal size of a polyhedron so that
a polyhedron of size less than hexSize/sizeThreshold is not created.
implEdges: enables implementation of geometrical edges into the mesh.
"""
if not self.hyp:
compFun = lambda hyp, args: False
self.hyp = self.Hypothesis("CartesianParameters3D",
[xGridDef, yGridDef, zGridDef, sizeThreshold],
UseExisting=False, CompareMethod=compFun)
if not self.mesh.IsUsedHypothesis( self.hyp, self.geom ):
self.mesh.AddHypothesis( self.hyp, self.geom )
for axis, gridDef in enumerate( [xGridDef, yGridDef, zGridDef] ):
if not gridDef: raise ValueError("Empty grid definition")
if isinstance( gridDef, str ):
self.hyp.SetGridSpacing( [gridDef], [], axis )
elif isinstance( gridDef[0], str ):
self.hyp.SetGridSpacing( gridDef, [], axis )
elif isinstance( gridDef[0], int ) or \
isinstance( gridDef[0], float ):
self.hyp.SetGrid(gridDef, axis )
else:
self.hyp.SetGridSpacing( gridDef[0], gridDef[1], axis )
self.hyp.SetSizeThreshold( sizeThreshold )
self.hyp.SetToAddEdges( implEdges )
return self.hyp
def SetAxesDirs( self, xAxis, yAxis, zAxis ):
"""
Defines custom directions of axes of the grid
Parameters:
xAxis: either SMESH.DirStruct or a vector, or 3 vector components
yAxis: either SMESH.DirStruct or a vector, or 3 vector components
zAxis: either SMESH.DirStruct or a vector, or 3 vector components
"""
import GEOM
if hasattr( xAxis, "__getitem__" ):
xAxis = self.mesh.smeshpyD.MakeDirStruct( xAxis[0],xAxis[1],xAxis[2] )
elif isinstance( xAxis, GEOM._objref_GEOM_Object ):
xAxis = self.mesh.smeshpyD.GetDirStruct( xAxis )
if hasattr( yAxis, "__getitem__" ):
yAxis = self.mesh.smeshpyD.MakeDirStruct( yAxis[0],yAxis[1],yAxis[2] )
elif isinstance( yAxis, GEOM._objref_GEOM_Object ):
yAxis = self.mesh.smeshpyD.GetDirStruct( yAxis )
if hasattr( zAxis, "__getitem__" ):
zAxis = self.mesh.smeshpyD.MakeDirStruct( zAxis[0],zAxis[1],zAxis[2] )
elif isinstance( zAxis, GEOM._objref_GEOM_Object ):
zAxis = self.mesh.smeshpyD.GetDirStruct( zAxis )
if not self.hyp:
self.hyp = self.Hypothesis("CartesianParameters3D")
if not self.mesh.IsUsedHypothesis( self.hyp, self.geom ):
self.mesh.AddHypothesis( self.hyp, self.geom )
self.hyp.SetAxesDirs( xAxis, yAxis, zAxis )
return self.hyp
def SetOptimalAxesDirs(self, isOrthogonal=True):
"""
Automatically defines directions of axes of the grid at which
a number of generated hexahedra is maximal
Parameters:
isOrthogonal: defines whether the axes mush be orthogonal
"""
if not self.hyp:
self.hyp = self.Hypothesis("CartesianParameters3D")
if not self.mesh.IsUsedHypothesis( self.hyp, self.geom ):
self.mesh.AddHypothesis( self.hyp, self.geom )
x,y,z = self.hyp.ComputeOptimalAxesDirs( self.geom, isOrthogonal )
self.hyp.SetAxesDirs( x,y,z )
return self.hyp
def SetFixedPoint( self, p, toUnset=False ):
"""
Sets/unsets a fixed point. The algorithm makes a plane of the grid pass
through the fixed point in each direction at which the grid is defined
by spacing
Parameters:
p: coordinates of the fixed point. Either SMESH.PointStruct or
a vertex or 3 components of coordinates.
toUnset: defines whether the fixed point is defined or removed.
"""
import SMESH, GEOM
if toUnset:
if not self.hyp: return
p = SMESH.PointStruct(0,0,0)
elif hasattr( p, "__getitem__" ):
p = SMESH.PointStruct( p[0],p[1],p[2] )
elif isinstance( p, GEOM._objref_GEOM_Object ):
p = self.mesh.smeshpyD.GetPointStruct( p )
if not self.hyp:
self.hyp = self.Hypothesis("CartesianParameters3D")
if not self.mesh.IsUsedHypothesis( self.hyp, self.geom ):
self.mesh.AddHypothesis( self.hyp, self.geom )
self.hyp.SetFixedPoint( p, toUnset )
return self.hyp
pass # end of StdMeshersBuilder_Cartesian_3D class
class StdMeshersBuilder_UseExisting_1D(Mesh_Algorithm):
""" Defines a stub 1D algorithm, which enables "manual" creation of nodes and
segments usable by 2D algorithms.
It is created by calling smeshBuilder.Mesh.UseExistingSegments(geom=0)
"""
meshMethod = "UseExistingSegments"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "UseExisting_1D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create 1D algorithm allowing batch meshing of edges"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
self.Create(mesh, geom, self.algoType)
pass
pass # end of StdMeshersBuilder_UseExisting_1D class
class StdMeshersBuilder_UseExisting_2D(Mesh_Algorithm):
""" Defines a stub 2D algorithm, which enables "manual" creation of nodes and
faces usable by 3D algorithms.
It is created by calling smeshBuilder.Mesh.UseExistingFaces(geom=0)
"""
meshMethod = "UseExistingFaces"
"""
name of the dynamic method in smeshBuilder.Mesh class
"""
algoType = "UseExisting_2D"
"""
type of algorithm used with helper function in smeshBuilder.Mesh class
"""
docHelper = "Create 2D algorithm allowing batch meshing of faces"
"""
doc string of the method
"""
def __init__(self, mesh, geom=0):
"""
Private constructor.
Parameters:
mesh: parent mesh object algorithm is assigned to
geom: geometry (shape/sub-shape) algorithm is assigned to;
if it is :code:`0` (default), the algorithm is assigned to the main shape
"""
self.Create(mesh, geom, self.algoType)
pass
pass # end of StdMeshersBuilder_UseExisting_2D class
Reference in New Issue View Git Blame Copy Permalink