X-Git-Url: http://git.salome-platform.org/gitweb/?p=modules%2Fsmesh.git;a=blobdiff_plain;f=src%2FTools%2FblocFissure%2Fgmu%2FfindWireEndVertices.py;h=e469f43b9d2be65bbe70b516a2a3591c28422da3;hp=be9b34736253b6ec52593cc60ac0a8328fc978ec;hb=140b89fcdf34becd8a3be2f1e276ad0960383eca;hpb=d5029840731bccaa1718e65f0abf3b19198c7293 diff --git a/src/Tools/blocFissure/gmu/findWireEndVertices.py b/src/Tools/blocFissure/gmu/findWireEndVertices.py index be9b34736..e469f43b9 100644 --- a/src/Tools/blocFissure/gmu/findWireEndVertices.py +++ b/src/Tools/blocFissure/gmu/findWireEndVertices.py @@ -1,65 +1,49 @@ # -*- 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 +# +"""Trouver les vertices aux extrémites d'un wire""" import logging + from .geomsmesh import geompy -from .geomsmesh import geomPublish -from .geomsmesh import geomPublishInFather -from . import initLog -# ----------------------------------------------------------------------------- -# --- trouver les vertices extremites d'un wire +from .findWireVertices import findWireVertices def findWireEndVertices(aWire, getNormals=False): - """ - trouver les vertices extremites d'un wire - calcul optionnel des tangentes. Attention à la tolérance qui peut être élevée (> 0.001) + """Trouver les vertices aux extrémites d'un wire + + Calcul optionnel des tangentes. Attention à la tolérance qui peut être élevée (> 0.001) """ logging.info("start") if geompy.NumberOfEdges(aWire) > 1: edges = geompy.ExtractShapes(aWire, geompy.ShapeType["EDGE"], False) else: edges = [aWire] - vertices = [] - idsubs = {} - shortList = [] - if getNormals: - normals = [] - idnorm = {} - shortNorm = [] - for edge in edges: - vert = geompy.ExtractShapes(edge, geompy.ShapeType["VERTEX"], False) - vertices += vert - if getNormals: - v0 = geompy.MakeVertexOnCurve(edge, 0.0) - n0 = geompy.MakeTangentOnCurve(edge, 0.0) - v1 = geompy.MakeVertexOnCurve(edge, 1.0) - n1 = geompy.MakeTangentOnCurve(edge, 1.0) - dist = geompy.MinDistance(v0, vert[0]) - logging.debug("distance %s", dist) - if dist < 1.e-2: - normals += [n0, n1] - else: - normals += [n1, n0] - for i, sub in enumerate(vertices): - subid = geompy.GetSubShapeID(aWire, sub) - if subid in list(idsubs.keys()): - idsubs[subid].append(sub) - else: - idsubs[subid] = [sub] - name='vertex%d'%i - geomPublishInFather(initLog.debug, aWire, sub, name) - if getNormals: - idnorm[subid] = normals[i] - name='norm%d'%i - geomPublishInFather(initLog.debug, aWire, normals[i], name) - logging.debug("idsubs: %s", idsubs) - for k, v in idsubs.items(): - if len(v) == 1: - shortList.append(v[0]) + + idsubs, idnorm = findWireVertices(aWire, edges, getNormals) + + shortList = list() + shortNorm = list() + for k_aux, sub in idsubs.items(): + if ( len(sub) == 1 ): + shortList.append(sub[0]) if getNormals: - shortNorm.append(idnorm[k]) - if getNormals: - return shortList, shortNorm - else: - return shortList + shortNorm.append(idnorm[k_aux]) + return shortList, shortNorm