Salome HOME
remplacement de print et corrections mineures
[modules/smesh.git] / src / Tools / blocFissure / gmu / creeZoneDefautMaillage.py
1 # -*- coding: utf-8 -*-
2
3 import logging
4 from geomsmesh import geompy
5 import math
6 from distance2 import distance2
7
8 # -----------------------------------------------------------------------------
9 # --- zone de defaut extraite du maillage
10
11 def creeZoneDefautMaillage(maillagesSains, shapeDefaut, tailleDefaut,
12                            nomZones, coordsNoeudsFissure):
13   """
14   Identification de la zone à remailler, opérations sur le maillage
15   de l'objet sain.
16   La zone à remailler est définie à partir d'un objet géométrique
17   ou à partir d'un jeu de points et d'une distance d'influence.
18   @param maillagesSains : (le maillage de l'objet initial, booleen isHexa)
19   @param shapeDefaut : objet géométrique représentant la fissure
20   (selon les cas, un point central, ou une shape plus complexe,
21   dont on ne garde que les vertices)
22   @param tailleDefaut : distance d'influence définissant la zone à remailler:
23   tous les éléments du maillage initial qui pénètrent dans cette zone
24   sont détectés
25   @param nomZones : préfixe des noms de groupes créés dans le maillage initial. S'il y a un groupe de noeuds
26   @coordsNoeudsFissure : jeu de points donné par une liste (x1,y1,z1, x2,y2,z2, ...)
27   @return (origShapes, verticesShapes, dmoyen) liste id subShapes,
28   listes noeuds de bord, longueur arête moyenne bord
29   """
30   logging.info("start")
31   
32   maillageSain = maillagesSains[0]
33   isHexa = maillagesSains[1]
34   lists = maillageSain.CreateHoleSkin(tailleDefaut, shapeDefaut, nomZones, coordsNoeudsFissure)
35
36   logging.debug("lists=%s", lists)
37
38   trace = True
39   origShapes = []
40   verticesShapes = []
41
42   cumul = 0 # somme des distances carrées entre point ordonnés (taille des arêtes)
43   nb = 0    # nombre d'arêtes évaluées
44
45   for aList in lists:
46     aShape = aList[0]
47     origShapes.append(aShape)
48     logging.debug("  shapeId %s", aShape)
49     vertices = []
50     xyz0 = None
51     for inode in range(1, len(aList)):
52       xyz = maillageSain.GetNodeXYZ(aList[inode])
53       if xyz0 is not None:
54         cumul += distance2(xyz, xyz0)
55         nb += 1
56       xyz0 = xyz
57       #logging.debug("    node %s %s", aList[inode], xyz)
58       vertices.append(geompy.MakeVertex(xyz[0], xyz[1], xyz[2]))
59       pass
60     verticesShapes.append(vertices)
61     pass
62
63   dmoyen = math.sqrt(cumul/nb) # ~ taille de l'arête moyenne du maillage global
64   return origShapes, verticesShapes, dmoyen