Salome HOME
Copyright update 2022
[modules/smesh.git] / src / Tools / blocFissure / gmu / sortGeneratrices.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2014-2022  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 """tri par longueur des 3 generatrices"""
21
22 import logging
23
24 from .geomsmesh import geompy
25 from .geomsmesh import geomPublishInFather
26
27 from . import initLog
28
29 def sortGeneratrices(tore, geners):
30   """
31   tri des 3 edges 'génératrices' selon leur longueur.
32   @param tore
33   @param les edges 'generatrices'
34   @return (genext, genint, gencnt) les 3 edges, de la plus grande à la plus petite
35   """
36   logging.info("start")
37
38   genx = geompy.ExtractShapes(geners[0], geompy.ShapeType["EDGE"], True)
39
40   lenx = list()
41   for gene in genx:
42     props = geompy.BasicProperties(gene)
43     lenx.append(props[0])
44
45   minlen = min(lenx)
46   maxlen = max(lenx)
47   genext=None
48   gencnt=None
49   genint=None
50   for i_aux, gene in enumerate(genx):
51     if lenx[i_aux] == minlen:
52       genint = gene
53     elif lenx[i_aux] == maxlen:
54       genext = gene
55     else:
56       gencnt= gene
57
58   geomPublishInFather(initLog.debug, tore, genext, 'genext' )
59   geomPublishInFather(initLog.debug, tore, genint, 'genint' )
60   geomPublishInFather(initLog.debug, tore, gencnt, 'gencnt' )
61
62   return genext, genint, gencnt