mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2024-11-11 16:19:16 +05:00
247 lines
6.4 KiB
Python
247 lines
6.4 KiB
Python
# -*- coding: iso-8859-1 -*-
|
|
# Copyright (C) 2007-2024 CEA, EDF, 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
|
|
#
|
|
|
|
# =======================================
|
|
# Procedure that take a triangulation and split all triangles in 4 others triangles
|
|
#
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
|
|
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()
|
|
|
|
# Values
|
|
# ------
|
|
|
|
tmpDir = tempfile.mkdtemp()
|
|
print("Output directory:", tmpDir)
|
|
|
|
# Path for ".med" files
|
|
path = os.path.join(tmpDir, "ex29_")
|
|
|
|
# Name of the shape and the mesh
|
|
name = "Carre"
|
|
|
|
# Add a node and needed edges
|
|
# ---------------------------
|
|
|
|
def node(m, f, n1, n2, lnv):
|
|
x1, y1, z1 = m.GetNodeXYZ(n1)
|
|
x2, y2, z2 = m.GetNodeXYZ(n2)
|
|
|
|
x = (x1 + x2) / 2.0
|
|
y = (y1 + y2) / 2.0
|
|
z = (z1 + z2) / 2.0
|
|
|
|
i = m.AddNode(x, y, z)
|
|
|
|
in1 = m.GetShapeID(n1)
|
|
in2 = m.GetShapeID(n2)
|
|
|
|
if (in1==f) or (in2==f):
|
|
m.SetNodeOnFace(i, f, 0, 0)
|
|
|
|
else:
|
|
e1 = m.AddEdge([ n1, i ])
|
|
e2 = m.AddEdge([ i , n2 ])
|
|
|
|
if n1 in lnv:
|
|
e = in2
|
|
else:
|
|
e = in1
|
|
|
|
m.SetMeshElementOnShape(e1, e)
|
|
m.SetMeshElementOnShape(e2, e)
|
|
m.SetNodeOnEdge(i, e, 0)
|
|
|
|
return i
|
|
|
|
# Add a triangle and associate to the CAD face
|
|
# --------------------------------------------
|
|
|
|
def triangle(m, f, n1, n2, n3):
|
|
i = m.AddFace([ n1, n2, n3 ])
|
|
m.SetMeshElementOnShape(i, f)
|
|
|
|
# Split all triangles in 4 triangles
|
|
# ----------------------------------
|
|
|
|
def SplitTrianglesIn4(m):
|
|
# Get all triangles
|
|
triangles = m.GetElementsByType(SMESH.FACE)
|
|
|
|
# Remove all edges
|
|
m.RemoveElements(m.GetElementsByType(SMESH.EDGE))
|
|
|
|
# Get the list of nodes (ids) associated with the CAD vertices
|
|
shape = m.GetShape()
|
|
lnv = []
|
|
for v in geompy.SubShapeAll(shape, geompy.ShapeType["VERTEX"]):
|
|
lnv = lnv + m.GetSubMeshNodesId(v, True)
|
|
|
|
# Split every triangle
|
|
for t in triangles:
|
|
noeud_1, noeud_2, noeud_3 = m.GetElemNodes(t)
|
|
|
|
face = m.GetShapeIDForElem(t)
|
|
|
|
noeud_12 = node(m, face, noeud_1, noeud_2, lnv)
|
|
noeud_23 = node(m, face, noeud_2, noeud_3, lnv)
|
|
noeud_13 = node(m, face, noeud_1, noeud_3, lnv)
|
|
|
|
triangle(m, face, noeud_1 , noeud_12, noeud_13)
|
|
triangle(m, face, noeud_2 , noeud_23, noeud_12)
|
|
triangle(m, face, noeud_3 , noeud_13, noeud_23)
|
|
triangle(m, face, noeud_12, noeud_23, noeud_13)
|
|
|
|
# Remove all initial triangles
|
|
m.RemoveElements(triangles)
|
|
|
|
# Merge all identical nodes
|
|
m.MergeNodes(m.FindCoincidentNodes(0.0001))
|
|
|
|
# Build a CAD square
|
|
# ------------------
|
|
|
|
x0 = 0.0 ; y0 = 0.0 ; z0 = 0.0
|
|
x1 = 1.0 ; y1 = 0.0 ; z1 = 0.0
|
|
x2 = 1.0 ; y2 = 1.0 ; z2 = 0.0
|
|
x3 = 0.0 ; y3 = 1.0 ; z3 = 0.0
|
|
|
|
P0 = geompy.MakeVertex(x0, y0, z0)
|
|
P1 = geompy.MakeVertex(x1, y1, z1)
|
|
P2 = geompy.MakeVertex(x2, y2, z2)
|
|
P3 = geompy.MakeVertex(x3, y3, z3)
|
|
|
|
square = geompy.MakeQuad4Vertices(P0, P1, P2, P3)
|
|
geompy.addToStudy(square, name)
|
|
|
|
# Refine edges and create group of mesh
|
|
# -------------------------------------
|
|
|
|
def refine(m, p1, p2, n, k, name):
|
|
s = m.GetShape()
|
|
|
|
g = geompy.CreateGroup(s, geompy.ShapeType["EDGE"])
|
|
e = geompy.GetEdge(s, p1, p2)
|
|
i = geompy.GetSubShapeID(s, e)
|
|
geompy.AddObject(g, i)
|
|
m.Group(g, name)
|
|
|
|
a = m.Segment(e)
|
|
a.NumberOfSegments(n, k)
|
|
|
|
# Mesh the square
|
|
# ---------------
|
|
|
|
MyMesh = smesh.Mesh(square)
|
|
|
|
refine(MyMesh, P1, P2, 8, 7, "Droite")
|
|
refine(MyMesh, P3, P0, 9, 10, "Gauche")
|
|
refine(MyMesh, P0, P1, 7, 9, "Bas" )
|
|
refine(MyMesh, P2, P3, 12, 14, "Haut" )
|
|
|
|
algo2D = MyMesh.Triangle()
|
|
algo2D.MaxElementArea(0.07)
|
|
|
|
isDone = MyMesh.Compute()
|
|
if not isDone:
|
|
raise Exception("Error when computing Mesh")
|
|
|
|
MyMesh.ExportMED(path+"110_triangles.med", 0)
|
|
|
|
# Disturb the mesh
|
|
# ----------------
|
|
|
|
MyMesh.MoveNode( 37, 0.05 , 0.368967 , 0 )
|
|
MyMesh.MoveNode( 38, 0.34 , 0.0762294, 0 )
|
|
MyMesh.MoveNode( 40, 0.8 , 0.42 , 0 )
|
|
MyMesh.MoveNode( 42, 0.702662, 0.74 , 0 )
|
|
MyMesh.MoveNode( 46, 0.4 , 0.374656 , 0 )
|
|
MyMesh.MoveNode( 47, 0.13 , 0.63 , 0 )
|
|
MyMesh.MoveNode( 49, 0.222187, 0.3 , 0 )
|
|
MyMesh.MoveNode( 54, 0.557791, 0.05 , 0 )
|
|
MyMesh.MoveNode( 55, 0.7 , 0.2 , 0 )
|
|
MyMesh.MoveNode( 56, 0.73 , 0.52 , 0 )
|
|
MyMesh.MoveNode( 58, 0.313071, 0.31 , 0 )
|
|
MyMesh.MoveNode( 59, 0.8 , 0.56 , 0 )
|
|
MyMesh.MoveNode( 62, 0.592703, 0.95 , 0 )
|
|
MyMesh.MoveNode( 63, 0.28 , 0.5 , 0 )
|
|
MyMesh.MoveNode( 65, 0.49 , 0.93 , 0 )
|
|
MyMesh.MoveNode( 68, 0.501038, 0.65 , 0 )
|
|
MyMesh.MoveNode( 69, 0.37 , 0.63 , 0 )
|
|
MyMesh.MoveNode( 70, 0.597025, 0.52 , 0 )
|
|
MyMesh.MoveNode( 72, 0.899 , 0.878589 , 0 )
|
|
MyMesh.MoveNode( 73, 0.92 , 0.85 , 0 )
|
|
MyMesh.MoveNode( 74, 0.820851, 0.75 , 0 )
|
|
|
|
NbCells1 = 110
|
|
MyMesh.ExportMED(path+"110_triangles_2.med", 0)
|
|
|
|
# First mesh refining
|
|
# -------------------
|
|
|
|
SplitTrianglesIn4(MyMesh)
|
|
|
|
NbCells2 = NbCells1*4
|
|
print(("Mesh with "+str(NbCells2)+" cells computed."))
|
|
|
|
MyMesh.ExportMED(path+str(NbCells2)+"_triangles.med", 0)
|
|
|
|
# Second mesh refining
|
|
# --------------------
|
|
|
|
SplitTrianglesIn4(MyMesh)
|
|
|
|
NbCells3 = NbCells2*4
|
|
print(("Mesh with "+str(NbCells3)+" cells computed."))
|
|
|
|
MyMesh.ExportMED(path+str(NbCells3)+"_triangles.med",0)
|
|
|
|
# Third mesh refining
|
|
# -------------------
|
|
|
|
SplitTrianglesIn4(MyMesh)
|
|
|
|
NbCells4 = NbCells3*4
|
|
print(("Mesh with "+str(NbCells4)+" cells computed."))
|
|
|
|
MyMesh.ExportMED(path+str(NbCells4)+"_triangles.med", 0)
|
|
|
|
# Remove temporary directory
|
|
# --------------------------
|
|
|
|
if os.getenv('SMESH_KEEP_TMP_DIR') != '1':
|
|
shutil.rmtree(tmpDir)
|
|
|
|
# Update the object browser
|
|
# -------------------------
|
|
|
|
salome.sg.updateObjBrowser()
|