Salome HOME
Merge branch 'master' into gni/evolution
[modules/smesh.git] / src / Tools / blocFissure / gmu / construitEdgesRadialesDebouchantes.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2014-2020  EDF R&D
3 #
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.
8 #
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.
13 #
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
17 #
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #
20 """Construit les arêtes débouchantes"""
21
22 import logging
23
24 from .geomsmesh import geompy
25 from .geomsmesh import geomPublish
26 #from .geomsmesh import geomPublishInFather
27 from . import initLog
28 import GEOM
29 from .sortEdges import sortEdges
30
31 def construitEdgesRadialesDebouchantes(idisklim, idiskout, gptsdisks, raydisks,
32                                        facesPipePeau, edgeRadFacePipePeau, nbsegCercle):
33   """construction des listes d'edges radiales sur chaque extrémité débouchante"""
34   logging.info('start')
35
36   # --- listes de nappes radiales en filling à chaque extrémité débouchante
37
38   facesDebouchantes = [False, False]
39   idFacesDebouchantes = [-1, -1] # contiendra les indices des faces disque débouchantes (facesPipePeau)
40   listNappes =[]
41   for i, idisk in enumerate(idisklim):
42     numout = idiskout[i]
43     logging.debug("extremité %s, indices disques interne %s, externe %s",i, idisk, numout)
44     nappes = []
45     if  (idisk != 0) and (idisk != len(gptsdisks)-1): # si extrémité débouchante
46       for k in range(nbsegCercle):
47         if i == 0:
48           iddeb = max(0, numout)
49           idfin = max(iddeb+3,idisk+1) # il faut 3 rayons pour faire un filling qui suive le fond de fissure
50           #logging.debug("extremité %s, indices retenus interne %s, externe %s",i, idfin, iddeb)
51           comp = geompy.MakeCompound(raydisks[k][iddeb:idfin])
52           name='compoundRay%d'%k
53           geomPublish(initLog.debug, comp, name)
54         else:
55           idfin = min(len(gptsdisks), numout+1)
56           iddeb = min(idfin-3, idisk) # il faut 3 rayons pour faire un filling qui suive le fond de fissure
57           #logging.debug("extremité %s, indices retenus interne %s, externe %s",i, idfin, iddeb)
58           comp = geompy.MakeCompound(raydisks[k][iddeb:idfin])
59           name='compoundRay%d'%k
60           geomPublish(initLog.debug, comp, name)
61         nappe = geompy.MakeFilling(comp, 2, 5, 0.0001, 0.0001, 0, GEOM.FOM_Default)
62         nappes.append(nappe)
63         name='nappe%d'%k
64         geomPublish(initLog.debug, nappe, name)
65         facesDebouchantes[i] = True
66     listNappes.append(nappes)
67
68   # --- mise en correspondance avec les indices des faces disque débouchantes (facesPipePeau)
69   for i, nappes in enumerate(listNappes):
70     if facesDebouchantes[i]:
71       for k, face in enumerate(facesPipePeau):
72         #logging.debug('i, k, face, nappes[0] %s %s %s %s', i, k, face, nappes[0])
73         #geomPublish(initLog.debug, nappes[0], 'lanappe')
74         #geomPublish(initLog.debug, face, 'laface')
75         edge = geompy.MakeSection(face, nappes[0])
76         if geompy.NbShapes(edge, geompy.ShapeType["EDGE"]) > 0:
77           idFacesDebouchantes[i] = k
78           break
79   logging.debug("idFacesDebouchantes: %s", idFacesDebouchantes)
80
81   # --- construction des listes d'edges radiales sur chaque extrémité débouchante
82   listEdges = []
83   for i, nappes in enumerate(listNappes):
84     ifd = idFacesDebouchantes[i] # indice de face débouchante (facesPipePeau)
85     if ifd < 0:
86       listEdges.append([])
87     else:
88       face = facesPipePeau[ifd]
89       edges = [edgeRadFacePipePeau[ifd]]
90       for k, nappe in enumerate(nappes):
91         if k > 0:
92           obj = geompy.MakeSection(face, nappes[k]) # normalement une edge, parfois un compound d'edges dont un tout petit
93           edge = obj
94           vs = geompy.ExtractShapes(obj, geompy.ShapeType["VERTEX"], False)
95           if len(vs) > 2:
96             eds = geompy.ExtractShapes(obj, geompy.ShapeType["EDGE"], False)
97             [edsorted, _, maxl] = sortEdges(eds)
98             edge = edsorted[-1]
99           else:
100             maxl = geompy.BasicProperties(edge)[0]
101           if maxl < 0.01: # problème MakeSection
102             logging.info("problème MakeSection recherche edge radiale %s, longueur trop faible: %s, utilisation partition", k, maxl)
103             partNappeFace = geompy.MakePartition([face, nappes[k]], [] , [], [], geompy.ShapeType["FACE"], 0, [], 0)
104             edps= geompy.ExtractShapes(partNappeFace, geompy.ShapeType["EDGE"], False)
105             ednouv = []
106             for ii, ed in enumerate(edps):
107               dmax=100.
108               vxs = geompy.ExtractShapes(ed, geompy.ShapeType["VERTEX"], False)
109               distx = [geompy.MinDistance(vx, face) for vx in vxs]
110               distx += [geompy.MinDistance(vx, nappes[k]) for vx in vxs]
111               dmax = max(distx)
112               lgedge = geompy.BasicProperties(ed)[0]
113               logging.debug("  dmax %s, longueur edge %s",dmax, lgedge)
114               if dmax < 0.01 and  lgedge > 0.01:
115                 ednouv.append(ed)
116             if (len(ednouv) > 0):
117               logging.debug("  edges issues de la partition: %s", ednouv)
118               for ii, ed in enumerate(ednouv):
119                 geomPublish(initLog.debug, ed, "ednouv%d"%ii)
120               [edsorted, _, maxl] = sortEdges(ednouv)
121               logging.debug("  longueur edge trouvée: %s", maxl)
122               edge = edsorted[-1]
123             else:
124               logging.info("problème partition recherche edge radiale %s", k)
125               vxs = geompy.ExtractShapes(partNappeFace, geompy.ShapeType["VERTEX"], False)
126               vxnouv=[]
127               for ii,vx in enumerate(vxs):
128                 distx = geompy.MinDistance(vx, face)
129                 distx += geompy.MinDistance(vx, nappes[k])
130                 logging.debug("vertex distance: %s", distx)
131                 if distx < 0.005:
132                   vxnouv.append(vx)
133               logging.debug("nombre vertex candidats %s", len(vxnouv))
134               if len(vxnouv) >= 2:
135                 eds = [geompy.MakeEdge(vxnouv[j],vxnouv[(j+1)%len(vxnouv)]) for j in range(len(vxnouv))]
136                 [edsorted2, _, maxl] = sortEdges(eds)
137                 edge = edsorted2[-1]
138                 logging.debug("lg edge: %s", maxl)
139               else:
140                 logging.debug("problème recherche edge radiale %s non résolu", k)
141           edges.append(edge)
142           name = 'edgeEndPipe%d'%k
143           geomPublish(initLog.debug, edge, name)
144       listEdges.append(edges)
145
146   return (listEdges, idFacesDebouchantes)