Salome HOME
Ménage
[modules/smesh.git] / src / Tools / blocFissure / gmu / creeZoneDefautMaillage.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2014-2020  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
21 import logging
22 from .geomsmesh import geompy
23 import math
24 from .distance2 import distance2
25 import traceback
26 from .fissError import fissError
27
28 # -----------------------------------------------------------------------------
29 # --- zone de defaut extraite du maillage
30
31 def creeZoneDefautMaillage(maillagesSains, shapeDefaut, tailleDefaut,
32                            nomZones, coordsNoeudsFissure):
33   """Identification de la zone à remailler, opérations sur le maillage de l'objet sain.
34
35   La zone à remailler est définie à partir d'un objet géométrique
36   ou à partir d'un jeu de points et d'une distance d'influence.
37   @param maillagesSains : (le maillage de l'objet initial, booleen isHexa)
38   @param shapeDefaut : objet géométrique représentant la fissure
39   (selon les cas, un point central, ou une shape plus complexe, dont on ne garde que les vertices)
40   @param tailleDefaut : distance d'influence définissant la zone à remailler:
41   tous les éléments du maillage initial qui pénètrent dans cette zone
42   sont détectés
43   @param nomZones : préfixe des noms de groupes créés dans le maillage initial. S'il y a un groupe de noeuds
44   @coordsNoeudsFissure : jeu de points donné par une liste (x1,y1,z1, x2,y2,z2, ...)
45   @return (origShapes, verticesShapes, dmoyen) liste id subShapes, listes noeuds de bord, longueur arête moyenne bord
46   """
47   logging.info("start")
48   logging.debug("distance d'influence (tailleDefaut) = %f" % tailleDefaut)
49   #print ("  shapeDefaut : {}".format(shapeDefaut))
50   #print ("  nomZones : {}".format(nomZones))
51
52   maillageSain = maillagesSains[0]
53   isHexa = maillagesSains[1]
54   lists = maillageSain.CreateHoleSkin(tailleDefaut, shapeDefaut, nomZones, coordsNoeudsFissure)
55
56   #print("lists = {}".format(lists))
57
58   trace = True
59   origShapes = list()
60   verticesShapes = list()
61
62   cumul = 0. # somme des distances carrées entre point ordonnés (taille des arêtes)
63   nb = 0     # nombre d'arêtes évaluées
64
65   for aList in lists:
66     aShape = aList[0]
67     origShapes.append(aShape)
68     vertices = list()
69     xyz0 = maillageSain.GetNodeXYZ(aList[1])
70     ###vertices.append(geompy.MakeVertex(xyz0[0], xyz0[1], xyz0[2])) pourquoi pas le premier ?
71     #print ("  node 0 {} : {}".format(aList[1],xyz0))
72     #print (xyz0)
73     for node in aList[2:]:
74       xyz = maillageSain.GetNodeXYZ(node)
75       #print ("  node {} : {}".format(node,xyz))
76       cumul += distance2(xyz, xyz0)
77       xyz0 = xyz
78       #logging.debug("    node %s %s", node, xyz)
79       vertices.append(geompy.MakeVertex(xyz[0], xyz[1], xyz[2]))
80     nb += len(aList) - 2
81     verticesShapes.append(vertices)
82   #print ("nb = {}".format(nb))
83   #print ("cumul = {}".format(cumul))
84
85   if (nb == 0):
86     texte = "La zone à remailler n'est pas détectée correctement.<br>"
87     texte += "Cause possible :<ul>"
88     texte += "<li>La distance d'influence est trop petite. "
89     texte += "L'ordre de grandeur minimal correspond à la taille des mailles du maillage sain dans la zone à remailler.</li></ul>"
90     raise fissError(traceback.extract_stack(),texte)
91
92   dmoyen = math.sqrt(cumul/float(nb)) # ~ taille de l'arête moyenne du maillage global
93   return origShapes, verticesShapes, dmoyen