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