mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2025-03-11 22:24:34 +05:00
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from geomsmesh import geompy
|
||
|
|
|
||
|
|
# -----------------------------------------------------------------------------
|
||
|
|
# --- trouver les vertices intermediaires d'un wire
|
||
|
|
|
||
|
|
def findWireIntermediateVertices(aWire, getNormals=False):
|
||
|
|
"""
|
||
|
|
trouver les vertices d'un wire qui ne sont pas aux extremités
|
||
|
|
calcul optionnel des tangentes. Attention à la tolérance qui peut être élevée (> 0.001)
|
||
|
|
"""
|
||
|
|
logging.info("start")
|
||
|
|
edges = geompy.ExtractShapes(aWire, geompy.ShapeType["EDGE"], False)
|
||
|
|
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 idsubs.keys():
|
||
|
|
idsubs[subid].append(sub)
|
||
|
|
else:
|
||
|
|
idsubs[subid] = [sub]
|
||
|
|
name='vertex%d'%i
|
||
|
|
geompy.addToStudyInFather(aWire, sub, name)
|
||
|
|
if getNormals:
|
||
|
|
idnorm[subid] = normals[i]
|
||
|
|
name='norm%d'%i
|
||
|
|
geompy.addToStudyInFather(aWire, normals[i], name)
|
||
|
|
for k, v in idsubs.iteritems():
|
||
|
|
if len(v) > 1:
|
||
|
|
shortList.append(v[0])
|
||
|
|
if getNormals:
|
||
|
|
shortNorm.append(idnorm[k])
|
||
|
|
if getNormals:
|
||
|
|
return shortList, shortNorm
|
||
|
|
else:
|
||
|
|
return shortList
|
||
|
|
|