# -*- coding: utf-8 -*-
+# Copyright (C) 2014-2021 EDF R&D
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
+"""Prolongation d'un wire par deux segments tangents"""
import logging
-from geomsmesh import geompy
-# -----------------------------------------------------------------------------
-# --- prolongation d'un wire par deux segments tangents
+from .geomsmesh import geompy
+from .geomsmesh import geomPublish
-def prolongeWire(aWire, extrem, norms, long):
- """
- """
+from . import initLog
+
+from .orderEdgesFromWire import orderEdgesFromWire
+
+def prolongeWire(aWire, extrem, norms, longueur):
+ """Prolongation d'un wire par deux segments tangents"""
logging.info("start")
- if geompy.NumberOfEdges(aWire) > 1:
+
+ if ( geompy.NumberOfEdges(aWire) > 1 ):
edges = geompy.ExtractShapes(aWire, geompy.ShapeType["EDGE"])
+ uneSeuleEdge = False
else:
edges = [aWire]
- for i, v1 in enumerate(extrem):
- exts = [geompy.MakeTranslationVectorDistance(v1, norms[i], l) for l in (-long, long)]
- dists = [(geompy.MinDistance(v, aWire), i , v) for i, v in enumerate(exts)]
+ uneSeuleEdge = True
+
+ edgesBout = list()
+ for i_aux, v_1 in enumerate(extrem):
+ exts = [geompy.MakeTranslationVectorDistance(v_1, norms[i_aux], lg_aux) for lg_aux in (-longueur, longueur)]
+ dists = [(geompy.MinDistance(v, aWire), j_aux , v) for j_aux, v in enumerate(exts)]
dists.sort()
- v2 = dists[-1][-1]
- #v2 = geompy.MakeTranslationVectorDistance(v1, norms[i], long)
- edge = geompy.MakeEdge(v1, v2)
+ v_2 = dists[-1][-1]
+ edge = geompy.MakeEdge(v_1, v_2)
edges.append(edge)
- name = "extrem%d"%i
- #geompy.addToStudy(edge,name)
- wireProlonge = geompy.MakeWire(edges)
- geompy.addToStudy(wireProlonge, "wireProlonge")
+ edgesBout.append(edge)
+ name = "extrem{}".format(i_aux)
+ geomPublish(initLog.debug, edge, name)
+
+ try:
+ wireProlonge = geompy.MakeWire(edges)
+ geomPublish(initLog.debug, wireProlonge, "wireProlonge")
+ except:
+ logging.warning("probleme MakeWire, approche pas a pas")
+ if uneSeuleEdge:
+ edgelist = [aWire]
+ accessList = [0]
+ else:
+ edgelist, accessList = orderEdgesFromWire(aWire)
+ edge1 = edgelist[accessList[0]]
+ if geompy.MinDistance(edgesBout[0], edge1) < 1.e-4 :
+ i_0 = 0
+ i_1 = 1
+ else:
+ i_0 = 1
+ i_1 = 0
+ wireProlonge = edgesBout[i_0]
+
+ for i_aux in range(len(edgelist)):
+ wireProlonge = geompy.MakeWire([wireProlonge, edgelist[accessList[i_aux]]])
+ geomPublish(initLog.debug, wireProlonge, "wireProlonge_{}".format(i_aux))
+
+ wireProlonge = geompy.MakeWire([wireProlonge,edgesBout[i_1]])
+ geomPublish(initLog.debug, wireProlonge, "wireProlonge")
+ logging.warning("prolongation wire pas a pas OK")
+
return wireProlonge
