mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2024-11-11 16:19:16 +05:00
Fix polyhedron on penta
Add a test on all kinds of elements
This commit is contained in:
parent
374031c17e
commit
bca8bf9ddb
128
doc/salome/examples/test_polyhedron_per_solid.py
Normal file
128
doc/salome/examples/test_polyhedron_per_solid.py
Normal file
@ -0,0 +1,128 @@
|
||||
#!/usr/bin/env python
|
||||
|
||||
import sys
|
||||
import salome
|
||||
|
||||
salome.salome_init()
|
||||
|
||||
###
|
||||
### GEOM component
|
||||
###
|
||||
|
||||
import GEOM
|
||||
from salome.geom import geomBuilder
|
||||
import math
|
||||
import SALOMEDS
|
||||
|
||||
|
||||
geompy = geomBuilder.New()
|
||||
|
||||
## Creates a polygon given its centre, external radius and number of sides
|
||||
def makePolygon(p_centre, radius, nb_sides, theName=""):
|
||||
points = []
|
||||
x, y, z = geompy.PointCoordinates(p_centre)
|
||||
for i in range(nb_sides):
|
||||
angle = i*2*math.pi/nb_sides
|
||||
p = geompy.MakeVertex(x+radius*math.cos(angle), y+radius*math.sin(angle), 0)
|
||||
points.append(p)
|
||||
wire = geompy.MakePolyline(points, True)
|
||||
face = geompy.MakeFace(wire, 1)
|
||||
if theName:
|
||||
geompy.addToStudy(face, theName)
|
||||
return face
|
||||
|
||||
## Creates a solid by adding a vertex on its top
|
||||
def makeSummitSolid(face, height, theName=""):
|
||||
p_cdg = geompy.MakeCDG(face)
|
||||
p_top = geompy.MakeTranslation(p_cdg, 0, 0, height)
|
||||
edges = geompy.SubShapeAll(face, geompy.ShapeType["EDGE"])
|
||||
faces = [face]
|
||||
for edge in edges:
|
||||
p1, p2 = geompy.SubShapeAll(edge, geompy.ShapeType["VERTEX"])
|
||||
wire = geompy.MakePolyline([p1, p2, p_top], True)
|
||||
face = geompy.MakeFace(wire, 1)
|
||||
faces.append(face)
|
||||
shell = geompy.MakeShell(faces)
|
||||
solid = geompy.MakeSolid(shell)
|
||||
if theName:
|
||||
geompy.addToStudy(solid, theName)
|
||||
return solid
|
||||
|
||||
O = geompy.MakeVertex(0, 0, 0)
|
||||
OX = geompy.MakeVectorDXDYDZ(1, 0, 0)
|
||||
OY = geompy.MakeVectorDXDYDZ(0, 1, 0)
|
||||
OZ = geompy.MakeVectorDXDYDZ(0, 0, 1)
|
||||
geompy.addToStudy( O, 'O' )
|
||||
geompy.addToStudy( OX, 'OX' )
|
||||
geompy.addToStudy( OY, 'OY' )
|
||||
geompy.addToStudy( OZ, 'OZ' )
|
||||
|
||||
height = 0.5
|
||||
|
||||
triangle = makePolygon(O, 1, 3, "triangle")
|
||||
|
||||
P1 = geompy.MakeVertex(2, 0, 0)
|
||||
quadrangle = makePolygon(P1, 1, 4, "quadrangle")
|
||||
|
||||
P2 = geompy.MakeVertex(4, 0, 0)
|
||||
pentagon = makePolygon(P2, 1, 5, "pentagon")
|
||||
|
||||
P3 = geompy.MakeVertex(6, 0, 0)
|
||||
hexagon = makePolygon(P3, 1, 6, "hexagon")
|
||||
|
||||
P4 = geompy.MakeVertex(8, 0, 0)
|
||||
heptagon = makePolygon(P4, 1, 7, "heptagon")
|
||||
|
||||
P5 = geompy.MakeVertex(10, 0, 0)
|
||||
octagon = makePolygon(P5, 1, 7, "octagon")
|
||||
|
||||
polygons = [triangle, quadrangle, pentagon, hexagon, heptagon, octagon]
|
||||
polyhedrons = []
|
||||
for polygon in polygons:
|
||||
name = polygon.GetName()
|
||||
polygon_extruded = geompy.MakePrismVecH(polygon, OZ, height, theName="%s_prism"%name)
|
||||
polygon_summit = makeSummitSolid(polygon, -height, theName="%s_summit"%name)
|
||||
polyhedrons += [polygon_extruded, polygon_summit]
|
||||
|
||||
solids = geompy.MakeCompound(polyhedrons, theName="solids")
|
||||
|
||||
|
||||
###
|
||||
### SMESH component
|
||||
###
|
||||
|
||||
import SMESH, SALOMEDS
|
||||
from salome.smesh import smeshBuilder
|
||||
|
||||
smesh = smeshBuilder.New()
|
||||
|
||||
Mesh_1 = smesh.Mesh(solids)
|
||||
Regular_1D = Mesh_1.Segment()
|
||||
Number_of_Segments_1 = Regular_1D.NumberOfSegments(1)
|
||||
PolygonPerFace_2D = Mesh_1.Polygon()
|
||||
PolyhedronPerSolid_3D = Mesh_1.Polyhedron()
|
||||
|
||||
ok = Mesh_1.Compute()
|
||||
|
||||
if not ok:
|
||||
raise Exception("Error when computing Mesh")
|
||||
|
||||
# check volumes
|
||||
vol_geom = geompy.BasicProperties(solids)[2]
|
||||
vol_mesh = Mesh_1.GetVolume()
|
||||
|
||||
assert abs(vol_geom - vol_mesh) < 1e-12
|
||||
|
||||
assert Mesh_1.NbVolumes() == 12
|
||||
|
||||
# check type of elements
|
||||
assert Mesh_1.NbTetras() == 1
|
||||
assert Mesh_1.NbHexas() == 1
|
||||
assert Mesh_1.NbPyramids() == 1
|
||||
assert Mesh_1.NbPrisms() == 1
|
||||
assert Mesh_1.NbHexagonalPrisms() == 1
|
||||
assert Mesh_1.NbPolyhedrons() == 7
|
||||
|
||||
|
||||
if salome.sg.hasDesktop():
|
||||
salome.sg.updateObjBrowser()
|
@ -182,6 +182,7 @@ SET(GOOD_TESTS
|
||||
split_biquad.py
|
||||
quad_medial_axis_algo.py
|
||||
defining_hypotheses_len_near_vertex.py
|
||||
test_polyhedron_per_solid.py
|
||||
)
|
||||
|
||||
SET(EXAMPLES_TESTS ${BAD_TESTS} ${GOOD_TESTS} testme.py)
|
||||
|
@ -196,6 +196,9 @@ namespace
|
||||
int iSide = trias[0] + 1;
|
||||
if ( iSide == trias[1] )
|
||||
++iSide;
|
||||
if (iSide == 5)
|
||||
// use first side (otherwise, out of bounds)
|
||||
iSide = 0;
|
||||
|
||||
const SMDS_MeshElement* botFace = faces[ trias[0]];
|
||||
const SMDS_MeshElement* topFace = faces[ trias[1]];
|
||||
|
Loading…
Reference in New Issue
Block a user