1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2014-2020 EDF R&D
4 # This library is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU Lesser General Public
6 # License as published by the Free Software Foundation; either
7 # version 2.1 of the License, or (at your option) any later version.
9 # This library is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 # Lesser General Public License for more details.
14 # You should have received a copy of the GNU Lesser General Public
15 # License along with this library; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 from .geomsmesh import geompy
24 from .geomsmesh import smesh
27 def construitMaillagePipe(gptsdisks, idisklim, nbsegCercle, nbsegRad):
29 maillage effectif du pipe
32 meshPipe = smesh.Mesh(None, "meshPipe")
33 fondFissGroup = meshPipe.CreateEmptyGroup(SMESH.EDGE, "FONDFISS")
34 nodesFondFissGroup = meshPipe.CreateEmptyGroup(SMESH.NODE, "nfondfis")
35 faceFissGroup = meshPipe.CreateEmptyGroup(SMESH.FACE, "fisInPi")
36 edgeFaceFissGroup = meshPipe.CreateEmptyGroup(SMESH.EDGE, "edgeFaceFiss")
37 edgeCircPipe0Group = meshPipe.CreateEmptyGroup(SMESH.EDGE, "edgeCircPipe0")
38 edgeCircPipe1Group = meshPipe.CreateEmptyGroup(SMESH.EDGE, "edgeCircPipe1")
39 faceCircPipe0Group = meshPipe.CreateEmptyGroup(SMESH.FACE, "faceCircPipe0")
40 faceCircPipe1Group = meshPipe.CreateEmptyGroup(SMESH.FACE, "faceCircPipe1")
42 mptsdisks = [] # vertices maillage de tous les disques
43 mEdges = [] # identifiants edges maillage fond de fissure
44 mEdgeFaces = [] # identifiants edges maillage edge face de fissure externe
45 mFaces = [] # identifiants faces maillage fissure
46 mVols = [] # identifiants volumes maillage pipe
49 for idisk in range(idisklim[0], idisklim[1]+1): # boucle sur les disques internes
51 # -----------------------------------------------------------------------
54 gptdsk = gptsdisks[idisk]
55 if idisk > idisklim[0]:
57 mptdsk = [] # vertices maillage d'un disque
58 for k in range(nbsegCercle):
61 for j, pt in enumerate(points):
65 coords = geompy.PointCoordinates(pt)
66 id = meshPipe.AddNode(coords[0], coords[1], coords[2])
69 mptsdisks.append(mptdsk)
71 # -----------------------------------------------------------------------
72 # --- groupes edges cercles debouchants
74 if idisk == idisklim[0]:
76 for k in range(nbsegCercle):
77 pts.append(mptdsk[k][-1])
79 for k in range(len(pts)):
81 idEdge = meshPipe.AddEdge([pts[k], pts[k1]])
83 edgeCircPipe0Group.Add(edges)
85 if idisk == idisklim[1]:
87 for k in range(nbsegCercle):
88 pts.append(mptdsk[k][-1])
90 for k in range(len(pts)):
92 idEdge = meshPipe.AddEdge([pts[k], pts[k1]])
94 edgeCircPipe1Group.Add(edges)
96 # -----------------------------------------------------------------------
97 # --- groupes faces debouchantes
99 if idisk == idisklim[0]:
101 for j in range(nbsegRad):
102 for k in range(nbsegCercle):
104 if k == nbsegCercle-1:
107 idf = meshPipe.AddFace([mptdsk[k][0], mptdsk[k][1], mptdsk[k1][1]]) # triangle
109 idf = meshPipe.AddFace([mptdsk[k][j], mptdsk[k][j+1], mptdsk[k1][j+1], mptdsk[k1][j]]) # quadrangle
111 faceCircPipe0Group.Add(faces)
113 if idisk == idisklim[1]:
115 for j in range(nbsegRad):
116 for k in range(nbsegCercle):
118 if k == nbsegCercle-1:
121 idf = meshPipe.AddFace([mptdsk[k][0], mptdsk[k][1], mptdsk[k1][1]]) # triangle
123 idf = meshPipe.AddFace([mptdsk[k][j], mptdsk[k][j+1], mptdsk[k1][j+1], mptdsk[k1][j]]) # quadrangle
125 faceCircPipe1Group.Add(faces)
127 # -----------------------------------------------------------------------
128 # --- mailles volumiques, groupes noeuds et edges de fond de fissure, groupe de face de fissure
130 if idisk == idisklim[0]:
135 nodesFondFissGroup.Add([mptdsk[0][0]])
137 ide = meshPipe.AddEdge([oldmpts[0][0], mptdsk[0][0]])
139 fondFissGroup.Add([ide])
140 nodesFondFissGroup.Add([mptdsk[0][0]])
141 ide2 = meshPipe.AddEdge([oldmpts[0][-1], mptdsk[0][-1]])
142 mEdgeFaces.append(ide2)
143 edgeFaceFissGroup.Add([ide2])
147 for j in range(nbsegRad):
148 idf = meshPipe.AddFace([oldmpts[0][j], mptdsk[0][j], mptdsk[0][j+1], oldmpts[0][j+1]])
149 faceFissGroup.Add([idf])
153 for k in range(nbsegCercle):
155 if k == nbsegCercle-1:
158 idv = meshPipe.AddVolume([mptdsk[k][j], mptdsk[k][j+1], mptdsk[k1][j+1],
159 oldmpts[k][j], oldmpts[k][j+1], oldmpts[k1][j+1]])
161 idv = meshPipe.AddVolume([mptdsk[k][j], mptdsk[k][j+1], mptdsk[k1][j+1], mptdsk[k1][j],
162 oldmpts[k][j], oldmpts[k][j+1], oldmpts[k1][j+1], oldmpts[k1][j]])
163 idVolCercle.append(idv)
164 idVols.append(idVolCercle)
166 mFaces.append(idFaces)
169 pipeFissGroup = meshPipe.CreateEmptyGroup( SMESH.VOLUME, 'PIPEFISS' )
170 nbAdd = pipeFissGroup.AddFrom( meshPipe.GetMesh() )
172 nb, new_mesh, new_group = meshPipe.MakeBoundaryElements(SMESH.BND_2DFROM3D, "pipeBoundaries")
173 edgesCircPipeGroup = [edgeCircPipe0Group, edgeCircPipe1Group]
175 meshPipeGroups = dict(fondFissGroup = fondFissGroup,
176 nodesFondFissGroup = nodesFondFissGroup,
177 faceFissGroup = faceFissGroup,
178 edgeFaceFissGroup = edgeFaceFissGroup,
179 edgeCircPipe0Group = edgeCircPipe0Group,
180 edgeCircPipe1Group = edgeCircPipe1Group,
181 faceCircPipe0Group = faceCircPipe0Group,
182 faceCircPipe1Group = faceCircPipe1Group,
183 pipeFissGroup = pipeFissGroup,
184 edgesCircPipeGroup = edgesCircPipeGroup
187 return (meshPipe, meshPipeGroups, edgesCircPipeGroup)