Salome HOME
29430917716386f1cbd6b808f65cee6d6df2dbd6
[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 sortEdges import sortEdges
7
8 def construitEdgesRadialesDebouchantes(idisklim, idiskout, gptsdisks, raydisks,
9                                        facesPipePeau, edgeRadFacePipePeau, nbsegCercle):
10   """
11   construction des listes d'edges radiales sur chaque extrémité débouchante 
12   """
13   logging.info('start')
14     
15   # --- listes de nappes radiales en filling à chaque extrémité débouchante
16   
17   facesDebouchantes = [False, False]
18   idFacesDebouchantes = [-1, -1] # contiendra les indices des faces disque débouchantes (facesPipePeau)
19   listNappes =[]
20   for i, idisk in enumerate(idisklim):
21     numout = idiskout[i]
22     logging.debug("extremité %s, indices disques interne %s, externe %s",i, idisk, numout)
23     nappes = []
24     if  (idisk != 0) and (idisk != len(gptsdisks)-1): # si extrémité débouchante
25       for k in range(nbsegCercle):
26         if i == 0:
27           iddeb = max(0, numout)
28           idfin = max(iddeb+3,idisk+1) # il faut 3 rayons pour faire un filling qui suive le fond de fissure
29           #logging.debug("extremité %s, indices retenus interne %s, externe %s",i, idfin, iddeb)
30           comp = geompy.MakeCompound(raydisks[k][iddeb:idfin])
31           name='compoundRay%d'%k
32           geompy.addToStudy(comp, name)
33         else:
34           idfin = min(len(gptsdisks), numout+1)
35           iddeb = min(idfin-3, idisk) # il faut 3 rayons pour faire un filling qui suive le fond de fissure
36           #logging.debug("extremité %s, indices retenus interne %s, externe %s",i, idfin, iddeb)
37           comp = geompy.MakeCompound(raydisks[k][iddeb:idfin])
38           name='compoundRay%d'%k
39           geompy.addToStudy(comp, name)
40         nappe = geompy.MakeFilling(comp, 2, 5, 0.0001, 0.0001, 0, GEOM.FOM_Default)
41         nappes.append(nappe)
42         name='nappe%d'%k
43         geompy.addToStudy(nappe, name)
44         facesDebouchantes[i] = True
45     listNappes.append(nappes)
46       
47   # --- mise en correspondance avec les indices des faces disque débouchantes (facesPipePeau)
48   for i, nappes in enumerate(listNappes):
49     if facesDebouchantes[i]:
50       for k, face in enumerate(facesPipePeau):
51         edge = geompy.MakeSection(face, nappes[0])
52         if geompy.NbShapes(edge, geompy.ShapeType["EDGE"]) > 0:
53           idFacesDebouchantes[i] = k
54           break
55   logging.debug("idFacesDebouchantes: %s", idFacesDebouchantes)
56
57   # --- construction des listes d'edges radiales sur chaque extrémité débouchante
58   listEdges = []
59   for i, nappes in enumerate(listNappes):
60     ifd = idFacesDebouchantes[i] # indice de face débouchante (facesPipePeau)
61     if ifd < 0:
62       listEdges.append([])
63     else:
64       face = facesPipePeau[ifd]
65       edges = [edgeRadFacePipePeau[ifd]]
66       for k, nappe in enumerate(nappes):
67         if k > 0:
68           obj = geompy.MakeSection(face, nappes[k]) # normalement une edge, parfois un compound d'edges dont un tout petit
69           edge = obj
70           vs = geompy.ExtractShapes(obj, geompy.ShapeType["VERTEX"], False)
71           if len(vs) > 2:
72             eds = geompy.ExtractShapes(obj, geompy.ShapeType["EDGE"], False)
73             [edsorted, minl,maxl] = sortEdges(eds)
74             edge = edsorted[-1]
75           else:
76             maxl = geompy.BasicProperties(edge)[0]
77           if maxl < 0.01: # problème MakeSection
78             logging.debug("problème MakeSection recherche edge radiale %s, longueur trop faible: %s, utilisation partition", k, maxl)
79             partNappeFace = geompy.MakePartition([face, nappes[k]], [] , [], [], geompy.ShapeType["FACE"], 0, [], 0)
80             edps= geompy.ExtractShapes(partNappeFace, geompy.ShapeType["EDGE"], False)
81             ednouv = []
82             for ii, ed in enumerate(edps):
83               vxs = geompy.ExtractShapes(ed, geompy.ShapeType["VERTEX"], False)
84               distx = [geompy.MinDistance(vx, face) for vx in vxs]
85               distx += [geompy.MinDistance(vx, nappes[k]) for vx in vxs]
86               dmax = max(distx)
87               logging.debug("  dmax %s",dmax)
88               if dmax < 0.01:
89                 ednouv.append(ed)
90             logging.debug("  edges issues de la partition: %s", ednouv)
91             for ii, ed in enumerate(ednouv):
92               geompy.addToStudy(ed, "ednouv%d"%ii)
93             [edsorted, minl,maxl] = sortEdges(ednouv)
94             logging.debug("  longueur edge trouvée: %s", maxl) 
95             edge = edsorted[-1]
96           edges.append(edge)
97           name = 'edgeEndPipe%d'%k
98           geompy.addToStudy(edge, name)
99       listEdges.append(edges)
100       
101   return (listEdges, idFacesDebouchantes)