smesh/src/Tools/blocFissure/gmu/orderEdgesFromWire.py

# -*- 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
#

import logging
from .geomsmesh import geompy

# -----------------------------------------------------------------------------
# --- trouver les vertices intermediaires d'un wire

def orderEdgesFromWire(aWire):
  """
  fournit les edges ordonnées d'un wire selon ExtractShapes(,,False), 
  et l'ordre des edges selon le sens de parcours (ordre des indices de la liste d'edges)
  """
  logging.info("start")
  edges = geompy.ExtractShapes(aWire, geompy.ShapeType["EDGE"], False)
  
  idverts = dict()
  for i, edge in enumerate(edges):
    verts = geompy.ExtractShapes(edge, geompy.ShapeType["VERTEX"], True)
#    idverts[(i,0)] = verts[0]
#    idverts[(i,1)] = verts[1]
    v0 = geompy.MakeVertexOnCurve(edge, 0.0)
    dist = geompy.MinDistance(v0, verts[0])
    if dist < 1.e-4:
      idverts[(i,0)] = verts[0]
      idverts[(i,1)] = verts[1]
    else:
      idverts[(i,0)] = verts[1]
      idverts[(i,1)] = verts[0]
     
  idsubs = dict()
  for kv, sub in idverts.items():
    subid = geompy.GetSubShapeID(aWire, sub)
    if subid in list(idsubs.keys()):
      idsubs[subid].append(kv)
    else:
      idsubs[subid] = [kv]
  
  debut = -1
  fin = -1    
  for k, kvs in idsubs.items():
    if len(kvs) == 1: # une extremité
      kv = kvs[0]
      if kv[1] == 0:
        debut = kv[0]
      else:
        fin = kv[0]
  logging.debug("nombre d'edges: {}, indice edge début: {}, fin: {}".format(len(edges), debut, fin))
  if debut < 0:
    logging.critical("les edges du wire ne sont pas orientées dans le même sens: pas de début trouvé")
    return edges, list(range(len(edges)))
  
  orderedList = [debut]
  while len(orderedList) < len(edges):
    bout = orderedList[-1]
    vertex = idverts[(bout,1)]
    for k, v in idverts.items():
      if k[0] not in orderedList:
        if geompy.MinDistance(vertex, v) < 1.e-4:
          if k[1] == 0:
            orderedList.append(k[0])
            break
          else:
            logging.critical("les edges du wire ne sont pas orientées dans le même sens: une edge à l'envers")
            return edges, list(range(len(edges)))

  logging.debug("liste des edges ordonnées selon le sens de parcours : {}".format(orderedList))
  accessList = list(range(len(orderedList)))
  for i,k in enumerate(orderedList):
    accessList[k] = i
  logging.info("position ordonnée des edges selon le sens de parcours : {}".format(accessList))
  return edges, accessList