Salome HOME
Update copyrights
[modules/smesh.git] / src / Tools / blocFissure / gmu / prolongeWire.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2014-2019  CEA/DEN, 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 from .orderEdgesFromWire import orderEdgesFromWire
27
28 # -----------------------------------------------------------------------------
29 # --- prolongation d'un wire par deux segments tangents
30
31 def prolongeWire(aWire, extrem, norms, lg):
32   """
33   """
34   logging.info("start")
35   if geompy.NumberOfEdges(aWire) > 1:
36     edges = geompy.ExtractShapes(aWire, geompy.ShapeType["EDGE"])
37     uneSeuleEdge = False
38   else:
39     edges = [aWire]
40     uneSeuleEdge = True
41   edgesBout = []
42   for i, v1 in enumerate(extrem):
43     exts = [geompy.MakeTranslationVectorDistance(v1, norms[i], l) for l in (-lg, lg)]
44     dists = [(geompy.MinDistance(v, aWire), i , v) for i, v in enumerate(exts)]
45     dists.sort()
46     v2 = dists[-1][-1]
47     edge = geompy.MakeEdge(v1, v2)
48     edges.append(edge)
49     edgesBout.append(edge)
50     name = "extrem%d"%i
51     geomPublish(initLog.debug, edge, name)
52   try:
53     wireProlonge = geompy.MakeWire(edges)
54     geomPublish(initLog.debug, wireProlonge, "wireProlonge")
55   except:
56     logging.warning("probleme MakeWire, approche pas a pas")
57     if uneSeuleEdge:
58       edgelist = [aWire]
59       accessList = [0]
60     else:
61       edgelist, accessList = orderEdgesFromWire(aWire)
62     edge1 = edgelist[accessList[0]]
63     if geompy.MinDistance(edgesBout[0], edge1) < 1.e-4 :
64       i0 = 0
65       i1 = 1
66     else:
67       i0 = 1
68       i1 = 0
69     wireProlonge = edgesBout[i0]
70     for i in range(len(edgelist)):
71       wireProlonge = geompy.MakeWire([wireProlonge, edgelist[accessList[i]]])
72       geomPublish(initLog.debug, wireProlonge, "wireProlonge_%d"%i)
73     wireProlonge = geompy.MakeWire([wireProlonge,edgesBout[i1]])
74     geomPublish(initLog.debug, wireProlonge, "wireProlonge")
75     logging.warning("prolongation wire pas a pas OK")
76   return wireProlonge