Salome HOME
Merge branch 'master' into gni/evolution
[modules/smesh.git] / src / Tools / blocFissure / gmu / getCentreFondFiss.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2014-2021  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
21 import logging
22 from .geomsmesh import geompy
23 from .geomsmesh import geomPublish
24 from .geomsmesh import geomPublishInFather
25 from . import initLog
26 import bisect
27
28 publie = False
29
30 def getCentreFondFiss(shapesFissure):
31   """
32   identification du centre de fond de fissure,
33   transformation fond de fissure en edge unique (seulement pour la procédure construitFissureGenerale).
34   On distingue le cas d'utilisation de la procédure insereFissureLongue par le nombre d'éléments de shapesFissure.
35   """
36   global publie
37   logging.debug("start")
38   
39   fondFiss          = shapesFissure[4] # groupe d'edges de fond de fissure
40   if len(shapesFissure) == 6:          # procédure construitFissureGenerale, et edge fond de fissure fournie explicitement
41     edgeFondExt     = shapesFissure[5]
42   else:
43     edgeFondExt     = None
44   
45   if len(shapesFissure) > 6:           # procédure insereFissureLongue (fissure plane, plusieurs edges sur le fond de fissure)
46     centreFondFiss = shapesFissure[1]
47     tgtCentre = None
48   else:                                # procédure construitFissureGenerale, détermination edge unique et milieu de l'edge
49     if geompy.NumberOfEdges(fondFiss) > 1:
50       if geompy.NbShapes(fondFiss, geompy.ShapeType["WIRE"]) > 0: # wire
51         aWire = fondFiss
52       else:                                                       # compound
53         edges = geompy.ExtractShapes(fondFiss, geompy.ShapeType["EDGE"], False)
54         aWire = geompy.MakeWire(edges, 1e-07)
55     else:
56       edgeFondExt = fondFiss
57       aWire = geompy.MakeWire([fondFiss], 1e-07)
58     if not publie:
59       geomPublish(initLog.debug, aWire, "wireFondFissExt")
60         
61     lgWire = geompy.BasicProperties(aWire)[0]
62     edges = geompy.ExtractShapes(aWire, geompy.ShapeType["EDGE"], True)
63     lgEdges = [geompy.BasicProperties(ed)[0] for ed in edges]
64     lgSumEd =[sum(lgEdges[0:i+1]) for i in range(len(lgEdges))]
65     iedr = bisect.bisect_left(lgSumEd, lgWire/2.0)
66     iedg = iedr -1
67     if iedg < 0:
68       lgOnEdge = lgWire/2.0
69     else:
70       lgOnEdge = lgWire/2.0 - lgSumEd[iedg]
71     logging.debug("lgsumEdges %s", lgSumEd)
72     logging.debug("id edge: %s, lgOnEdge: %s, lgEdge: %s",iedr, lgOnEdge, lgEdges[iedr])
73     if iedr > 0: # il y a une edge avant celle du milieu
74       if geompy.MinDistance(edges[iedr-1], geompy.MakeVertexOnCurve(edges[iedr], 0.0 )) < 1.e-3: # edge orientée croissante 
75         centreFondFiss = geompy.MakeVertexOnCurve(edges[iedr], lgOnEdge/lgEdges[iedr])
76       else:
77         centreFondFiss = geompy.MakeVertexOnCurve(edges[iedr], 1.0 - lgOnEdge/lgEdges[iedr])
78     elif iedr < len(edges)-1: # il y a une edge après celle du milieu
79       if geompy.MinDistance(edges[iedr+1], geompy.MakeVertexOnCurve(edges[iedr], 1.0 )) < 1.e-3: # edge orientée croissante
80         centreFondFiss = geompy.MakeVertexOnCurve(edges[iedr], lgOnEdge/lgEdges[iedr])
81       else:
82         centreFondFiss = geompy.MakeVertexOnCurve(edges[iedr], 1.0 - lgOnEdge/lgEdges[iedr])
83     else: # on ne sait pas comment est orientée l'edge unique, mais ça n'a pas d'importance
84       centreFondFiss = geompy.MakeVertexOnCurve(edges[iedr], lgOnEdge/lgEdges[iedr])
85     geomPublishInFather(initLog.debug,aWire, centreFondFiss, "centreFondFiss")
86     tgtCentre = geompy.MakeTangentOnCurve(edges[iedr], lgOnEdge/ lgEdges[iedr])
87     
88     if edgeFondExt is None: # fond de fissure non fourni explicitement sous forme d'edge
89       try:
90         edgeFondExt = geompy.MakeEdgeWire(aWire, 0.0005, 1e-07)
91       except:
92         logging.debug("erreur MakeEdgeWire sur fond de fissure, on fait sans")
93         edgeFondExt = None
94     if not publie and edgeFondExt is not None:
95       geomPublish(initLog.debug, edgeFondExt, "edgeFondExt")
96   
97   publie = True
98   return edgeFondExt, centreFondFiss, tgtCentre