]> SALOME platform Git repositories - modules/smesh.git/blob - src/Tools/blocFissure/gmu/regroupeSainEtDefaut.py
Salome HOME
Allow saving groups with non-ascii names.
[modules/smesh.git] / src / Tools / blocFissure / gmu / regroupeSainEtDefaut.py
1 # -*- coding: utf-8 -*-
2
3 import logging
4 from .geomsmesh import smesh
5 import SMESH
6 from .geomsmesh import geompy
7
8 # -----------------------------------------------------------------------------
9 # --- maillage complet et fissure
10
11 def RegroupeSainEtDefaut(maillageSain, blocComplet, extrusionFaceFissure, faceGeomFissure, nomVolume, normal = None):
12   """
13   Maillage sain sans la zone de defaut
14   TODO: a completer
15   """
16   logging.info('Concatenation')
17
18   maillageComplet = smesh.Concatenate([maillageSain.GetMesh(), blocComplet.GetMesh()], 1, 1, 1e-05,False)
19
20   groups = maillageComplet.GetGroups()
21   grps = [ grp for grp in groups if grp.GetName() == 'FACE1']
22   faceFissure = grps[0]
23   grps = [ grp for grp in groups if grp.GetName() == 'nfondfis']
24   noeudsFondFissure = grps[0]
25   grps = [ grp for grp in groups if grp.GetName() == 'fisInPi']
26   fisInPi = grps[0]
27   grps = [ grp for grp in groups if grp.GetName() == 'fisOutPi']
28   fisOutPi = grps[0]
29
30   # --- TODO: fiabiliser l'orientation dans le cas general
31   if normal is None:
32     normal  = smesh.MakeDirStruct( 0, 0, 1 )
33   maillageComplet.Reorient2D( fisInPi,  normal, [0,0,0])
34   maillageComplet.Reorient2D( fisOutPi, normal, [0,0,0])
35     
36   shapes = []
37   if extrusionFaceFissure is not None:
38     subIds = geompy.SubShapeAllIDs(extrusionFaceFissure, geompy.ShapeType["SOLID"])
39     if len(subIds) > 1:
40       shapes = geompy.ExtractShapes(extrusionFaceFissure, geompy.ShapeType["SOLID"], False)
41     else:
42       shapes = [extrusionFaceFissure]
43 #  else:
44 #    subIds = geompy.SubShapeAllIDs(faceGeomFissure, geompy.ShapeType["FACE"])
45 #    if len(subIds) > 1:
46 #      shapes = geompy.ExtractShapes(faceGeomFissure, geompy.ShapeType["FACE"], False)
47 #    else:
48 #      shapes = [faceGeomFissure]
49     
50   grpEdges = []
51   grpFaces = []
52   grpVolumes = []
53   if len(shapes) == 0:
54     shapes = [None] # calcul uniquement avec les normales des faces mailles de la fissure
55   for i, aShape in enumerate(shapes):
56     logging.info('Detection elements affectes par le dedoublement de la face %d'%i)
57     affectedGroups = maillageComplet.AffectedElemGroupsInRegion([faceFissure], [noeudsFondFissure], aShape)
58     grps = [ grp for grp in affectedGroups if grp.GetName() == 'affectedEdges']
59     affectedEdges = grps[0]
60     affectedEdges.SetName('affEd%d'%i)
61     grpEdges.append(affectedEdges)
62     grps = [ grp for grp in affectedGroups if grp.GetName() == 'affectedFaces']
63     affectedFaces = grps[0]
64     affectedFaces.SetName('affFa%d'%i)
65     grpFaces.append(affectedFaces)
66     grps = [ grp for grp in affectedGroups if grp.GetName() == 'affectedVolumes']
67     affectedVolumes = grps[0]
68     affectedVolumes.SetName('affVo%d'%i)
69     grpVolumes.append(affectedVolumes)
70   logging.info("union des groupes d'edges") 
71   affectedEdges = maillageComplet.UnionListOfGroups(grpEdges, 'affEdges')
72   logging.info("union des groupes de faces") 
73   affectedFaces = maillageComplet.UnionListOfGroups(grpFaces, 'affFaces')
74   logging.info("union des groupes de volumes") 
75   affectedVolumes = maillageComplet.UnionListOfGroups(grpVolumes, 'affVols')
76   for grp in affectedGroups:
77     logging.debug("nom groupe %s",grp.GetName())
78   [ FACE2, FACE2_nodes ] = maillageComplet.DoubleNodeElemGroups([faceFissure], [noeudsFondFissure], affectedGroups, True, True)
79   FACE2.SetName( 'FACE2' )
80
81   GroupVol = maillageComplet.CreateEmptyGroup( SMESH.VOLUME, nomVolume )
82   nbAdd = GroupVol.AddFrom( maillageComplet.GetMesh() )
83
84   return maillageComplet
85