Salome HOME
cf70523f06cbb3187661f50fb6746db1738b4f11
[modules/smesh.git] / src / Tools / blocFissure / gmu / findWireEndVertices.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 """Trouver les vertices aux extrémites d'un wire"""
21
22 import logging
23
24 from . import initLog
25
26 from .geomsmesh import geompy
27
28 from .findWireVertices import findWireVertices
29
30 def findWireEndVertices(aWire, getNormals=False):
31   """Trouver les vertices aux extrémites d'un wire
32
33   Calcul optionnel des tangentes. Attention à la tolérance qui peut être élevée (> 0.001)
34   """
35   logging.info("start")
36   if geompy.NumberOfEdges(aWire) > 1:
37     edges = geompy.ExtractShapes(aWire, geompy.ShapeType["EDGE"], False)
38   else:
39     edges = [aWire]
40
41   idsubs, idnorm = findWireVertices(aWire, edges, getNormals)
42
43   shortList = list()
44   shortNorm = list()
45   for k_aux, sub in idsubs.items():
46     if ( len(sub) == 1 ):
47       shortList.append(sub[0])
48       if getNormals:
49         shortNorm.append(idnorm[k_aux])
50
51   return shortList, shortNorm