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