Salome HOME
Avoid MED file reading failure caused by low memory
[modules/smesh.git] / src / SMESH_SWIG / smeshBuilder.py
1 # Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
2 #
3 # This library is free software; you can redistribute it and/or
4 # modify it under the terms of the GNU Lesser General Public
5 # License as published by the Free Software Foundation; either
6 # version 2.1 of the License, or (at your option) any later version.
7 #
8 # This library is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 # Lesser General Public License for more details.
12 #
13 # You should have received a copy of the GNU Lesser General Public
14 # License along with this library; if not, write to the Free Software
15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 #
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 #
19 #  File   : smeshBuilder.py
20 #  Author : Francis KLOSS, OCC
21 #  Module : SMESH
22
23 ## @package smeshBuilder
24 #  Python API for SALOME %Mesh module
25
26 ## @defgroup l1_auxiliary Auxiliary methods and structures
27 ## @defgroup l1_creating  Creating meshes
28 ## @{
29 ##   @defgroup l2_impexp     Importing and exporting meshes
30 ##   @defgroup l2_construct  Constructing meshes
31 ##   @defgroup l2_algorithms Defining Algorithms
32 ##   @{
33 ##     @defgroup l3_algos_basic   Basic meshing algorithms
34 ##     @defgroup l3_algos_proj    Projection Algorithms
35 ##     @defgroup l3_algos_radialp Radial Prism
36 ##     @defgroup l3_algos_segmarv Segments around Vertex
37 ##     @defgroup l3_algos_3dextr  3D extrusion meshing algorithm
38
39 ##   @}
40 ##   @defgroup l2_hypotheses Defining hypotheses
41 ##   @{
42 ##     @defgroup l3_hypos_1dhyps 1D Meshing Hypotheses
43 ##     @defgroup l3_hypos_2dhyps 2D Meshing Hypotheses
44 ##     @defgroup l3_hypos_maxvol Max Element Volume hypothesis
45 ##     @defgroup l3_hypos_quad Quadrangle Parameters hypothesis
46 ##     @defgroup l3_hypos_additi Additional Hypotheses
47
48 ##   @}
49 ##   @defgroup l2_submeshes Constructing submeshes
50 ##   @defgroup l2_compounds Building Compounds
51 ##   @defgroup l2_editing   Editing Meshes
52
53 ## @}
54 ## @defgroup l1_meshinfo  Mesh Information
55 ## @defgroup l1_controls  Quality controls and Filtering
56 ## @defgroup l1_grouping  Grouping elements
57 ## @{
58 ##   @defgroup l2_grps_create Creating groups
59 ##   @defgroup l2_grps_edit   Editing groups
60 ##   @defgroup l2_grps_operon Using operations on groups
61 ##   @defgroup l2_grps_delete Deleting Groups
62
63 ## @}
64 ## @defgroup l1_modifying Modifying meshes
65 ## @{
66 ##   @defgroup l2_modif_add      Adding nodes and elements
67 ##   @defgroup l2_modif_del      Removing nodes and elements
68 ##   @defgroup l2_modif_edit     Modifying nodes and elements
69 ##   @defgroup l2_modif_renumber Renumbering nodes and elements
70 ##   @defgroup l2_modif_trsf     Transforming meshes (Translation, Rotation, Symmetry, Sewing, Merging)
71 ##   @defgroup l2_modif_movenode Moving nodes
72 ##   @defgroup l2_modif_throughp Mesh through point
73 ##   @defgroup l2_modif_invdiag  Diagonal inversion of elements
74 ##   @defgroup l2_modif_unitetri Uniting triangles
75 ##   @defgroup l2_modif_changori Changing orientation of elements
76 ##   @defgroup l2_modif_cutquadr Cutting elements
77 ##   @defgroup l2_modif_smooth   Smoothing
78 ##   @defgroup l2_modif_extrurev Extrusion and Revolution
79 ##   @defgroup l2_modif_patterns Pattern mapping
80 ##   @defgroup l2_modif_tofromqu Convert to/from Quadratic Mesh
81
82 ## @}
83 ## @defgroup l1_measurements Measurements
84
85 import salome
86 from salome.geom import geomBuilder
87
88 import SMESH # This is necessary for back compatibility
89 from   SMESH import *
90 from   salome.smesh.smesh_algorithm import Mesh_Algorithm
91
92 import SALOME
93 import SALOMEDS
94 import os
95
96 class MeshMeta(type):
97     def __instancecheck__(cls, inst):
98         """Implement isinstance(inst, cls)."""
99         return any(cls.__subclasscheck__(c)
100                    for c in {type(inst), inst.__class__})
101
102     def __subclasscheck__(cls, sub):
103         """Implement issubclass(sub, cls)."""
104         return type.__subclasscheck__(cls, sub) or (cls.__name__ == sub.__name__ and cls.__module__ == sub.__module__)
105
106 ## @addtogroup l1_auxiliary
107 ## @{
108
109 ## Converts an angle from degrees to radians
110 def DegreesToRadians(AngleInDegrees):
111     from math import pi
112     return AngleInDegrees * pi / 180.0
113
114 import salome_notebook
115 notebook = salome_notebook.notebook
116 # Salome notebook variable separator
117 var_separator = ":"
118
119 ## Return list of variable values from salome notebook.
120 #  The last argument, if is callable, is used to modify values got from notebook
121 def ParseParameters(*args):
122     Result = []
123     Parameters = ""
124     hasVariables = False
125     varModifFun=None
126     if args and callable( args[-1] ):
127         args, varModifFun = args[:-1], args[-1]
128     for parameter in args:
129
130         Parameters += str(parameter) + var_separator
131
132         if isinstance(parameter,str):
133             # check if there is an inexistent variable name
134             if not notebook.isVariable(parameter):
135                 raise ValueError, "Variable with name '" + parameter + "' doesn't exist!!!"
136             parameter = notebook.get(parameter)
137             hasVariables = True
138             if varModifFun:
139                 parameter = varModifFun(parameter)
140                 pass
141             pass
142         Result.append(parameter)
143
144         pass
145     Parameters = Parameters[:-1]
146     Result.append( Parameters )
147     Result.append( hasVariables )
148     return Result
149
150 # Parse parameters converting variables to radians
151 def ParseAngles(*args):
152     return ParseParameters( *( args + (DegreesToRadians, )))
153
154 # Substitute PointStruct.__init__() to create SMESH.PointStruct using notebook variables.
155 # Parameters are stored in PointStruct.parameters attribute
156 def __initPointStruct(point,*args):
157     point.x, point.y, point.z, point.parameters,hasVars = ParseParameters(*args)
158     pass
159 SMESH.PointStruct.__init__ = __initPointStruct
160
161 # Substitute AxisStruct.__init__() to create SMESH.AxisStruct using notebook variables.
162 # Parameters are stored in AxisStruct.parameters attribute
163 def __initAxisStruct(ax,*args):
164     if len( args ) != 6:
165         raise RuntimeError,\
166               "Bad nb args (%s) passed in SMESH.AxisStruct(x,y,z,dx,dy,dz)"%(len( args ))
167     ax.x, ax.y, ax.z, ax.vx, ax.vy, ax.vz, ax.parameters,hasVars = ParseParameters(*args)
168     pass
169 SMESH.AxisStruct.__init__ = __initAxisStruct
170
171 smeshPrecisionConfusion = 1.e-07
172 def IsEqual(val1, val2, tol=smeshPrecisionConfusion):
173     if abs(val1 - val2) < tol:
174         return True
175     return False
176
177 NO_NAME = "NoName"
178
179 ## Gets object name
180 def GetName(obj):
181     if obj:
182         # object not null
183         if isinstance(obj, SALOMEDS._objref_SObject):
184             # study object
185             return obj.GetName()
186         try:
187             ior  = salome.orb.object_to_string(obj)
188         except:
189             ior = None
190         if ior:
191             # CORBA object
192             studies = salome.myStudyManager.GetOpenStudies()
193             for sname in studies:
194                 s = salome.myStudyManager.GetStudyByName(sname)
195                 if not s: continue
196                 sobj = s.FindObjectIOR(ior)
197                 if not sobj: continue
198                 return sobj.GetName()
199             if hasattr(obj, "GetName"):
200                 # unknown CORBA object, having GetName() method
201                 return obj.GetName()
202             else:
203                 # unknown CORBA object, no GetName() method
204                 return NO_NAME
205             pass
206         if hasattr(obj, "GetName"):
207             # unknown non-CORBA object, having GetName() method
208             return obj.GetName()
209         pass
210     raise RuntimeError, "Null or invalid object"
211
212 ## Prints error message if a hypothesis was not assigned.
213 def TreatHypoStatus(status, hypName, geomName, isAlgo, mesh):
214     if isAlgo:
215         hypType = "algorithm"
216     else:
217         hypType = "hypothesis"
218         pass
219     reason = ""
220     if hasattr( status, "__getitem__" ):
221         status,reason = status[0],status[1]
222     if status == HYP_UNKNOWN_FATAL :
223         reason = "for unknown reason"
224     elif status == HYP_INCOMPATIBLE :
225         reason = "this hypothesis mismatches the algorithm"
226     elif status == HYP_NOTCONFORM :
227         reason = "a non-conform mesh would be built"
228     elif status == HYP_ALREADY_EXIST :
229         if isAlgo: return # it does not influence anything
230         reason = hypType + " of the same dimension is already assigned to this shape"
231     elif status == HYP_BAD_DIM :
232         reason = hypType + " mismatches the shape"
233     elif status == HYP_CONCURENT :
234         reason = "there are concurrent hypotheses on sub-shapes"
235     elif status == HYP_BAD_SUBSHAPE :
236         reason = "the shape is neither the main one, nor its sub-shape, nor a valid group"
237     elif status == HYP_BAD_GEOMETRY:
238         reason = "the algorithm is not applicable to this geometry"
239     elif status == HYP_HIDDEN_ALGO:
240         reason = "it is hidden by an algorithm of an upper dimension, which generates elements of all dimensions"
241     elif status == HYP_HIDING_ALGO:
242         reason = "it hides algorithms of lower dimensions by generating elements of all dimensions"
243     elif status == HYP_NEED_SHAPE:
244         reason = "algorithm can't work without shape"
245     elif status == HYP_INCOMPAT_HYPS:
246         pass
247     else:
248         return
249     where = geomName
250     if where:
251         where = '"%s"' % geomName
252         if mesh:
253             meshName = GetName( mesh )
254             if meshName and meshName != NO_NAME:
255                 where = '"%s" in "%s"' % ( geomName, meshName )
256     if status < HYP_UNKNOWN_FATAL and where:
257         print '"%s" was assigned to %s but %s' %( hypName, where, reason )
258     elif where:
259         print '"%s" was not assigned to %s : %s' %( hypName, where, reason )
260     else:
261         print '"%s" was not assigned : %s' %( hypName, reason )
262         pass
263
264 ## Private method. Add geom (sub-shape of the main shape) into the study if not yet there
265 def AssureGeomPublished(mesh, geom, name=''):
266     if not isinstance( geom, geomBuilder.GEOM._objref_GEOM_Object ):
267         return
268     if not geom.GetStudyEntry() and \
269            mesh.smeshpyD.GetCurrentStudy():
270         ## set the study
271         studyID = mesh.smeshpyD.GetCurrentStudy()._get_StudyId()
272         if studyID != mesh.geompyD.myStudyId:
273             mesh.geompyD.init_geom( mesh.smeshpyD.GetCurrentStudy())
274         ## get a name
275         if not name and geom.GetShapeType() != geomBuilder.GEOM.COMPOUND:
276             # for all groups SubShapeName() returns "Compound_-1"
277             name = mesh.geompyD.SubShapeName(geom, mesh.geom)
278         if not name:
279             name = "%s_%s"%(geom.GetShapeType(), id(geom)%10000)
280         ## publish
281         mesh.geompyD.addToStudyInFather( mesh.geom, geom, name )
282     return
283
284 ## Return the first vertex of a geometrical edge by ignoring orientation
285 def FirstVertexOnCurve(mesh, edge):
286     vv = mesh.geompyD.SubShapeAll( edge, geomBuilder.geomBuilder.ShapeType["VERTEX"])
287     if not vv:
288         raise TypeError, "Given object has no vertices"
289     if len( vv ) == 1: return vv[0]
290     v0   = mesh.geompyD.MakeVertexOnCurve(edge,0.)
291     xyz  = mesh.geompyD.PointCoordinates( v0 ) # coords of the first vertex
292     xyz1 = mesh.geompyD.PointCoordinates( vv[0] )
293     xyz2 = mesh.geompyD.PointCoordinates( vv[1] )
294     dist1, dist2 = 0,0
295     for i in range(3):
296         dist1 += abs( xyz[i] - xyz1[i] )
297         dist2 += abs( xyz[i] - xyz2[i] )
298     if dist1 < dist2:
299         return vv[0]
300     else:
301         return vv[1]
302
303 # end of l1_auxiliary
304 ## @}
305
306
307 # Warning: smeshInst is a singleton
308 smeshInst = None
309 engine = None
310 doLcc = False
311 created = False
312
313 ## This class allows to create, load or manipulate meshes
314 #  It has a set of methods to create load or copy meshes, to combine several meshes.
315 #  It also has methods to get infos on meshes.
316 class smeshBuilder(object, SMESH._objref_SMESH_Gen):
317
318     # MirrorType enumeration
319     POINT = SMESH_MeshEditor.POINT
320     AXIS =  SMESH_MeshEditor.AXIS
321     PLANE = SMESH_MeshEditor.PLANE
322
323     # Smooth_Method enumeration
324     LAPLACIAN_SMOOTH = SMESH_MeshEditor.LAPLACIAN_SMOOTH
325     CENTROIDAL_SMOOTH = SMESH_MeshEditor.CENTROIDAL_SMOOTH
326
327     PrecisionConfusion = smeshPrecisionConfusion
328
329     # TopAbs_State enumeration
330     [TopAbs_IN, TopAbs_OUT, TopAbs_ON, TopAbs_UNKNOWN] = range(4)
331
332     # Methods of splitting a hexahedron into tetrahedra
333     Hex_5Tet, Hex_6Tet, Hex_24Tet, Hex_2Prisms, Hex_4Prisms = 1, 2, 3, 1, 2
334
335     def __new__(cls):
336         global engine
337         global smeshInst
338         global doLcc
339         #print "==== __new__", engine, smeshInst, doLcc
340
341         if smeshInst is None:
342             # smesh engine is either retrieved from engine, or created
343             smeshInst = engine
344             # Following test avoids a recursive loop
345             if doLcc:
346                 if smeshInst is not None:
347                     # smesh engine not created: existing engine found
348                     doLcc = False
349                 if doLcc:
350                     doLcc = False
351                     # FindOrLoadComponent called:
352                     # 1. CORBA resolution of server
353                     # 2. the __new__ method is called again
354                     #print "==== smeshInst = lcc.FindOrLoadComponent ", engine, smeshInst, doLcc
355                     smeshInst = salome.lcc.FindOrLoadComponent( "FactoryServer", "SMESH" )
356             else:
357                 # FindOrLoadComponent not called
358                 if smeshInst is None:
359                     # smeshBuilder instance is created from lcc.FindOrLoadComponent
360                     #print "==== smeshInst = super(smeshBuilder,cls).__new__(cls) ", engine, smeshInst, doLcc
361                     smeshInst = super(smeshBuilder,cls).__new__(cls)
362                 else:
363                     # smesh engine not created: existing engine found
364                     #print "==== existing ", engine, smeshInst, doLcc
365                     pass
366             #print "====1 ", smeshInst
367             return smeshInst
368
369         #print "====2 ", smeshInst
370         return smeshInst
371
372     def __init__(self):
373         global created
374         #print "--------------- smeshbuilder __init__ ---", created
375         if not created:
376           created = True
377           SMESH._objref_SMESH_Gen.__init__(self)
378
379     ## Dump component to the Python script
380     #  This method overrides IDL function to allow default values for the parameters.
381     def DumpPython(self, theStudy, theIsPublished=True, theIsMultiFile=True):
382         return SMESH._objref_SMESH_Gen.DumpPython(self, theStudy, theIsPublished, theIsMultiFile)
383
384     ## Set mode of DumpPython(), \a historical or \a snapshot.
385     # In the \a historical mode, the Python Dump script includes all commands
386     # performed by SMESH engine. In the \a snapshot mode, commands
387     # relating to objects removed from the Study are excluded from the script
388     # as well as commands not influencing the current state of meshes
389     def SetDumpPythonHistorical(self, isHistorical):
390         if isHistorical: val = "true"
391         else:            val = "false"
392         SMESH._objref_SMESH_Gen.SetOption(self, "historical_python_dump", val)
393
394     ## Sets the current study and Geometry component
395     #  @ingroup l1_auxiliary
396     def init_smesh(self,theStudy,geompyD = None):
397         #print "init_smesh"
398         self.SetCurrentStudy(theStudy,geompyD)
399         if theStudy:
400             global notebook
401             notebook.myStudy = theStudy
402
403     ## Creates a mesh. This can be either an empty mesh, possibly having an underlying geometry,
404     #  or a mesh wrapping a CORBA mesh given as a parameter.
405     #  @param obj either (1) a CORBA mesh (SMESH._objref_SMESH_Mesh) got e.g. by calling
406     #         salome.myStudy.FindObjectID("0:1:2:3").GetObject() or
407     #         (2) a Geometrical object for meshing or
408     #         (3) none.
409     #  @param name the name for the new mesh.
410     #  @return an instance of Mesh class.
411     #  @ingroup l2_construct
412     def Mesh(self, obj=0, name=0):
413         if isinstance(obj,str):
414             obj,name = name,obj
415         return Mesh(self,self.geompyD,obj,name)
416
417     ## Returns a long value from enumeration
418     #  @ingroup l1_controls
419     def EnumToLong(self,theItem):
420         return theItem._v
421
422     ## Returns a string representation of the color.
423     #  To be used with filters.
424     #  @param c color value (SALOMEDS.Color)
425     #  @ingroup l1_controls
426     def ColorToString(self,c):
427         val = ""
428         if isinstance(c, SALOMEDS.Color):
429             val = "%s;%s;%s" % (c.R, c.G, c.B)
430         elif isinstance(c, str):
431             val = c
432         else:
433             raise ValueError, "Color value should be of string or SALOMEDS.Color type"
434         return val
435
436     ## Gets PointStruct from vertex
437     #  @param theVertex a GEOM object(vertex)
438     #  @return SMESH.PointStruct
439     #  @ingroup l1_auxiliary
440     def GetPointStruct(self,theVertex):
441         [x, y, z] = self.geompyD.PointCoordinates(theVertex)
442         return PointStruct(x,y,z)
443
444     ## Gets DirStruct from vector
445     #  @param theVector a GEOM object(vector)
446     #  @return SMESH.DirStruct
447     #  @ingroup l1_auxiliary
448     def GetDirStruct(self,theVector):
449         vertices = self.geompyD.SubShapeAll( theVector, geomBuilder.geomBuilder.ShapeType["VERTEX"] )
450         if(len(vertices) != 2):
451             print "Error: vector object is incorrect."
452             return None
453         p1 = self.geompyD.PointCoordinates(vertices[0])
454         p2 = self.geompyD.PointCoordinates(vertices[1])
455         pnt = PointStruct(p2[0]-p1[0], p2[1]-p1[1], p2[2]-p1[2])
456         dirst = DirStruct(pnt)
457         return dirst
458
459     ## Makes DirStruct from a triplet
460     #  @param x,y,z vector components
461     #  @return SMESH.DirStruct
462     #  @ingroup l1_auxiliary
463     def MakeDirStruct(self,x,y,z):
464         pnt = PointStruct(x,y,z)
465         return DirStruct(pnt)
466
467     ## Get AxisStruct from object
468     #  @param theObj a GEOM object (line or plane)
469     #  @return SMESH.AxisStruct
470     #  @ingroup l1_auxiliary
471     def GetAxisStruct(self,theObj):
472         import GEOM
473         edges = self.geompyD.SubShapeAll( theObj, geomBuilder.geomBuilder.ShapeType["EDGE"] )
474         axis = None
475         if len(edges) > 1:
476             vertex1, vertex2 = self.geompyD.SubShapeAll( edges[0], geomBuilder.geomBuilder.ShapeType["VERTEX"] )
477             vertex3, vertex4 = self.geompyD.SubShapeAll( edges[1], geomBuilder.geomBuilder.ShapeType["VERTEX"] )
478             vertex1 = self.geompyD.PointCoordinates(vertex1)
479             vertex2 = self.geompyD.PointCoordinates(vertex2)
480             vertex3 = self.geompyD.PointCoordinates(vertex3)
481             vertex4 = self.geompyD.PointCoordinates(vertex4)
482             v1 = [vertex2[0]-vertex1[0], vertex2[1]-vertex1[1], vertex2[2]-vertex1[2]]
483             v2 = [vertex4[0]-vertex3[0], vertex4[1]-vertex3[1], vertex4[2]-vertex3[2]]
484             normal = [ v1[1]*v2[2]-v2[1]*v1[2], v1[2]*v2[0]-v2[2]*v1[0], v1[0]*v2[1]-v2[0]*v1[1] ]
485             axis = AxisStruct(vertex1[0], vertex1[1], vertex1[2], normal[0], normal[1], normal[2])
486             axis._mirrorType = SMESH.SMESH_MeshEditor.PLANE
487         elif len(edges) == 1:
488             vertex1, vertex2 = self.geompyD.SubShapeAll( edges[0], geomBuilder.geomBuilder.ShapeType["VERTEX"] )
489             p1 = self.geompyD.PointCoordinates( vertex1 )
490             p2 = self.geompyD.PointCoordinates( vertex2 )
491             axis = AxisStruct(p1[0], p1[1], p1[2], p2[0]-p1[0], p2[1]-p1[1], p2[2]-p1[2])
492             axis._mirrorType = SMESH.SMESH_MeshEditor.AXIS
493         elif theObj.GetShapeType() == GEOM.VERTEX:
494             x,y,z = self.geompyD.PointCoordinates( theObj )
495             axis = AxisStruct( x,y,z, 1,0,0,)
496             axis._mirrorType = SMESH.SMESH_MeshEditor.POINT
497         return axis
498
499     # From SMESH_Gen interface:
500     # ------------------------
501
502     ## Sets the given name to the object
503     #  @param obj the object to rename
504     #  @param name a new object name
505     #  @ingroup l1_auxiliary
506     def SetName(self, obj, name):
507         if isinstance( obj, Mesh ):
508             obj = obj.GetMesh()
509         elif isinstance( obj, Mesh_Algorithm ):
510             obj = obj.GetAlgorithm()
511         ior  = salome.orb.object_to_string(obj)
512         SMESH._objref_SMESH_Gen.SetName(self, ior, name)
513
514     ## Sets the current mode
515     #  @ingroup l1_auxiliary
516     def SetEmbeddedMode( self,theMode ):
517         #self.SetEmbeddedMode(theMode)
518         SMESH._objref_SMESH_Gen.SetEmbeddedMode(self,theMode)
519
520     ## Gets the current mode
521     #  @ingroup l1_auxiliary
522     def IsEmbeddedMode(self):
523         #return self.IsEmbeddedMode()
524         return SMESH._objref_SMESH_Gen.IsEmbeddedMode(self)
525
526     ## Sets the current study. Calling SetCurrentStudy( None ) allows to
527     #  switch OFF automatic pubilishing in the Study of mesh objects.
528     #  @ingroup l1_auxiliary
529     def SetCurrentStudy( self, theStudy, geompyD = None ):
530         #self.SetCurrentStudy(theStudy)
531         if not geompyD:
532             from salome.geom import geomBuilder
533             geompyD = geomBuilder.geom
534             pass
535         self.geompyD=geompyD
536         self.SetGeomEngine(geompyD)
537         SMESH._objref_SMESH_Gen.SetCurrentStudy(self,theStudy)
538         global notebook
539         if theStudy:
540             notebook = salome_notebook.NoteBook( theStudy )
541         else:
542             notebook = salome_notebook.NoteBook( salome_notebook.PseudoStudyForNoteBook() )
543         if theStudy:
544             sb = theStudy.NewBuilder()
545             sc = theStudy.FindComponent("SMESH")
546             if sc: sb.LoadWith(sc, self)
547             pass
548         pass
549
550     ## Gets the current study
551     #  @ingroup l1_auxiliary
552     def GetCurrentStudy(self):
553         #return self.GetCurrentStudy()
554         return SMESH._objref_SMESH_Gen.GetCurrentStudy(self)
555
556     ## Creates a Mesh object importing data from the given UNV file
557     #  @return an instance of Mesh class
558     #  @ingroup l2_impexp
559     def CreateMeshesFromUNV( self,theFileName ):
560         aSmeshMesh = SMESH._objref_SMESH_Gen.CreateMeshesFromUNV(self,theFileName)
561         aMesh = Mesh(self, self.geompyD, aSmeshMesh)
562         return aMesh
563
564     ## Creates a Mesh object(s) importing data from the given MED file
565     #  @return a tuple ( list of Mesh class instances, SMESH.DriverMED_ReadStatus )
566     #  @ingroup l2_impexp
567     def CreateMeshesFromMED( self,theFileName ):
568         aSmeshMeshes, aStatus = SMESH._objref_SMESH_Gen.CreateMeshesFromMED(self,theFileName)
569         aMeshes = [ Mesh(self, self.geompyD, m) for m in aSmeshMeshes ]
570         return aMeshes, aStatus
571
572     ## Creates a Mesh object(s) importing data from the given SAUV file
573     #  @return a tuple ( list of Mesh class instances, SMESH.DriverMED_ReadStatus )
574     #  @ingroup l2_impexp
575     def CreateMeshesFromSAUV( self,theFileName ):
576         aSmeshMeshes, aStatus = SMESH._objref_SMESH_Gen.CreateMeshesFromSAUV(self,theFileName)
577         aMeshes = [ Mesh(self, self.geompyD, m) for m in aSmeshMeshes ]
578         return aMeshes, aStatus
579
580     ## Creates a Mesh object importing data from the given STL file
581     #  @return an instance of Mesh class
582     #  @ingroup l2_impexp
583     def CreateMeshesFromSTL( self, theFileName ):
584         aSmeshMesh = SMESH._objref_SMESH_Gen.CreateMeshesFromSTL(self,theFileName)
585         aMesh = Mesh(self, self.geompyD, aSmeshMesh)
586         return aMesh
587
588     ## Creates Mesh objects importing data from the given CGNS file
589     #  @return a tuple ( list of Mesh class instances, SMESH.DriverMED_ReadStatus )
590     #  @ingroup l2_impexp
591     def CreateMeshesFromCGNS( self, theFileName ):
592         aSmeshMeshes, aStatus = SMESH._objref_SMESH_Gen.CreateMeshesFromCGNS(self,theFileName)
593         aMeshes = [ Mesh(self, self.geompyD, m) for m in aSmeshMeshes ]
594         return aMeshes, aStatus
595
596     ## Creates a Mesh object importing data from the given GMF file.
597     #  GMF files must have .mesh extension for the ASCII format and .meshb for
598     #  the binary format.
599     #  @return [ an instance of Mesh class, SMESH.ComputeError ]
600     #  @ingroup l2_impexp
601     def CreateMeshesFromGMF( self, theFileName ):
602         aSmeshMesh, error = SMESH._objref_SMESH_Gen.CreateMeshesFromGMF(self,
603                                                                         theFileName,
604                                                                         True)
605         if error.comment: print "*** CreateMeshesFromGMF() errors:\n", error.comment
606         return Mesh(self, self.geompyD, aSmeshMesh), error
607
608     ## Concatenate the given meshes into one mesh. All groups of input meshes will be
609     #  present in the new mesh.
610     #  @param meshes the meshes, sub-meshes and groups to combine into one mesh
611     #  @param uniteIdenticalGroups if true, groups with same names are united, else they are renamed
612     #  @param mergeNodesAndElements if true, equal nodes and elements are merged
613     #  @param mergeTolerance tolerance for merging nodes
614     #  @param allGroups forces creation of groups corresponding to every input mesh
615     #  @param name name of a new mesh
616     #  @return an instance of Mesh class
617     def Concatenate( self, meshes, uniteIdenticalGroups,
618                      mergeNodesAndElements = False, mergeTolerance = 1e-5, allGroups = False,
619                      name = ""):
620         if not meshes: return None
621         for i,m in enumerate(meshes):
622             if isinstance(m, Mesh):
623                 meshes[i] = m.GetMesh()
624         mergeTolerance,Parameters,hasVars = ParseParameters(mergeTolerance)
625         meshes[0].SetParameters(Parameters)
626         if allGroups:
627             aSmeshMesh = SMESH._objref_SMESH_Gen.ConcatenateWithGroups(
628                 self,meshes,uniteIdenticalGroups,mergeNodesAndElements,mergeTolerance)
629         else:
630             aSmeshMesh = SMESH._objref_SMESH_Gen.Concatenate(
631                 self,meshes,uniteIdenticalGroups,mergeNodesAndElements,mergeTolerance)
632         aMesh = Mesh(self, self.geompyD, aSmeshMesh, name=name)
633         return aMesh
634
635     ## Create a mesh by copying a part of another mesh.
636     #  @param meshPart a part of mesh to copy, either a Mesh, a sub-mesh or a group;
637     #                  to copy nodes or elements not contained in any mesh object,
638     #                  pass result of Mesh.GetIDSource( list_of_ids, type ) as meshPart
639     #  @param meshName a name of the new mesh
640     #  @param toCopyGroups to create in the new mesh groups the copied elements belongs to
641     #  @param toKeepIDs to preserve order of the copied elements or not
642     #  @return an instance of Mesh class
643     def CopyMesh( self, meshPart, meshName, toCopyGroups=False, toKeepIDs=False):
644         if (isinstance( meshPart, Mesh )):
645             meshPart = meshPart.GetMesh()
646         mesh = SMESH._objref_SMESH_Gen.CopyMesh( self,meshPart,meshName,toCopyGroups,toKeepIDs )
647         return Mesh(self, self.geompyD, mesh)
648
649     ## From SMESH_Gen interface
650     #  @return the list of integer values
651     #  @ingroup l1_auxiliary
652     def GetSubShapesId( self, theMainObject, theListOfSubObjects ):
653         return SMESH._objref_SMESH_Gen.GetSubShapesId(self,theMainObject, theListOfSubObjects)
654
655     ## From SMESH_Gen interface. Creates a pattern
656     #  @return an instance of SMESH_Pattern
657     #
658     #  <a href="../tui_modifying_meshes_page.html#tui_pattern_mapping">Example of Patterns usage</a>
659     #  @ingroup l2_modif_patterns
660     def GetPattern(self):
661         return SMESH._objref_SMESH_Gen.GetPattern(self)
662
663     ## Sets number of segments per diagonal of boundary box of geometry by which
664     #  default segment length of appropriate 1D hypotheses is defined.
665     #  Default value is 10
666     #  @ingroup l1_auxiliary
667     def SetBoundaryBoxSegmentation(self, nbSegments):
668         SMESH._objref_SMESH_Gen.SetBoundaryBoxSegmentation(self,nbSegments)
669
670     # Filtering. Auxiliary functions:
671     # ------------------------------
672
673     ## Creates an empty criterion
674     #  @return SMESH.Filter.Criterion
675     #  @ingroup l1_controls
676     def GetEmptyCriterion(self):
677         Type = self.EnumToLong(FT_Undefined)
678         Compare = self.EnumToLong(FT_Undefined)
679         Threshold = 0
680         ThresholdStr = ""
681         ThresholdID = ""
682         UnaryOp = self.EnumToLong(FT_Undefined)
683         BinaryOp = self.EnumToLong(FT_Undefined)
684         Tolerance = 1e-07
685         TypeOfElement = ALL
686         Precision = -1 ##@1e-07
687         return Filter.Criterion(Type, Compare, Threshold, ThresholdStr, ThresholdID,
688                                 UnaryOp, BinaryOp, Tolerance, TypeOfElement, Precision)
689
690     ## Creates a criterion by the given parameters
691     #  \n Criterion structures allow to define complex filters by combining them with logical operations (AND / OR) (see example below)
692     #  @param elementType the type of elements(SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME)
693     #  @param CritType the type of criterion (SMESH.FT_Taper, SMESH.FT_Area, etc.)
694     #          Type SMESH.FunctorType._items in the Python Console to see all values.
695     #          Note that the items starting from FT_LessThan are not suitable for CritType.
696     #  @param Compare  belongs to {SMESH.FT_LessThan, SMESH.FT_MoreThan, SMESH.FT_EqualTo}
697     #  @param Threshold the threshold value (range of ids as string, shape, numeric)
698     #  @param UnaryOp  SMESH.FT_LogicalNOT or SMESH.FT_Undefined
699     #  @param BinaryOp a binary logical operation SMESH.FT_LogicalAND, SMESH.FT_LogicalOR or
700     #                  SMESH.FT_Undefined
701     #  @param Tolerance the tolerance used by SMESH.FT_BelongToGeom, SMESH.FT_BelongToSurface,
702     #         SMESH.FT_LyingOnGeom, SMESH.FT_CoplanarFaces criteria
703     #  @return SMESH.Filter.Criterion
704     #
705     #  <a href="../tui_filters_page.html#combining_filters">Example of Criteria usage</a>
706     #  @ingroup l1_controls
707     def GetCriterion(self,elementType,
708                      CritType,
709                      Compare = FT_EqualTo,
710                      Threshold="",
711                      UnaryOp=FT_Undefined,
712                      BinaryOp=FT_Undefined,
713                      Tolerance=1e-07):
714         if not CritType in SMESH.FunctorType._items:
715             raise TypeError, "CritType should be of SMESH.FunctorType"
716         aCriterion               = self.GetEmptyCriterion()
717         aCriterion.TypeOfElement = elementType
718         aCriterion.Type          = self.EnumToLong(CritType)
719         aCriterion.Tolerance     = Tolerance
720
721         aThreshold = Threshold
722
723         if Compare in [FT_LessThan, FT_MoreThan, FT_EqualTo]:
724             aCriterion.Compare = self.EnumToLong(Compare)
725         elif Compare == "=" or Compare == "==":
726             aCriterion.Compare = self.EnumToLong(FT_EqualTo)
727         elif Compare == "<":
728             aCriterion.Compare = self.EnumToLong(FT_LessThan)
729         elif Compare == ">":
730             aCriterion.Compare = self.EnumToLong(FT_MoreThan)
731         elif Compare != FT_Undefined:
732             aCriterion.Compare = self.EnumToLong(FT_EqualTo)
733             aThreshold = Compare
734
735         if CritType in [FT_BelongToGeom,     FT_BelongToPlane, FT_BelongToGenSurface,
736                         FT_BelongToCylinder, FT_LyingOnGeom]:
737             # Check that Threshold is GEOM object
738             if isinstance(aThreshold, geomBuilder.GEOM._objref_GEOM_Object):
739                 aCriterion.ThresholdStr = GetName(aThreshold)
740                 aCriterion.ThresholdID  = aThreshold.GetStudyEntry()
741                 if not aCriterion.ThresholdID:
742                     name = aCriterion.ThresholdStr
743                     if not name:
744                         name = "%s_%s"%(aThreshold.GetShapeType(), id(aThreshold)%10000)
745                     aCriterion.ThresholdID = self.geompyD.addToStudy( aThreshold, name )
746             # or a name of GEOM object
747             elif isinstance( aThreshold, str ):
748                 aCriterion.ThresholdStr = aThreshold
749             else:
750                 raise TypeError, "The Threshold should be a shape."
751             if isinstance(UnaryOp,float):
752                 aCriterion.Tolerance = UnaryOp
753                 UnaryOp = FT_Undefined
754                 pass
755         elif CritType == FT_BelongToMeshGroup:
756             # Check that Threshold is a group
757             if isinstance(aThreshold, SMESH._objref_SMESH_GroupBase):
758                 if aThreshold.GetType() != elementType:
759                     raise ValueError, "Group type mismatches Element type"
760                 aCriterion.ThresholdStr = aThreshold.GetName()
761                 aCriterion.ThresholdID  = salome.orb.object_to_string( aThreshold )
762                 study = self.GetCurrentStudy()
763                 if study:
764                     so = study.FindObjectIOR( aCriterion.ThresholdID )
765                     if so:
766                         entry = so.GetID()
767                         if entry:
768                             aCriterion.ThresholdID = entry
769             else:
770                 raise TypeError, "The Threshold should be a Mesh Group"
771         elif CritType == FT_RangeOfIds:
772             # Check that Threshold is string
773             if isinstance(aThreshold, str):
774                 aCriterion.ThresholdStr = aThreshold
775             else:
776                 raise TypeError, "The Threshold should be a string."
777         elif CritType == FT_CoplanarFaces:
778             # Check the Threshold
779             if isinstance(aThreshold, int):
780                 aCriterion.ThresholdID = str(aThreshold)
781             elif isinstance(aThreshold, str):
782                 ID = int(aThreshold)
783                 if ID < 1:
784                     raise ValueError, "Invalid ID of mesh face: '%s'"%aThreshold
785                 aCriterion.ThresholdID = aThreshold
786             else:
787                 raise TypeError,\
788                       "The Threshold should be an ID of mesh face and not '%s'"%aThreshold
789         elif CritType == FT_ConnectedElements:
790             # Check the Threshold
791             if isinstance(aThreshold, geomBuilder.GEOM._objref_GEOM_Object): # shape
792                 aCriterion.ThresholdID = aThreshold.GetStudyEntry()
793                 if not aCriterion.ThresholdID:
794                     name = aThreshold.GetName()
795                     if not name:
796                         name = "%s_%s"%(aThreshold.GetShapeType(), id(aThreshold)%10000)
797                     aCriterion.ThresholdID = self.geompyD.addToStudy( aThreshold, name )
798             elif isinstance(aThreshold, int): # node id
799                 aCriterion.Threshold = aThreshold
800             elif isinstance(aThreshold, list): # 3 point coordinates
801                 if len( aThreshold ) < 3:
802                     raise ValueError, "too few point coordinates, must be 3"
803                 aCriterion.ThresholdStr = " ".join( [str(c) for c in aThreshold[:3]] )
804             elif isinstance(aThreshold, str):
805                 if aThreshold.isdigit():
806                     aCriterion.Threshold = aThreshold # node id
807                 else:
808                     aCriterion.ThresholdStr = aThreshold # hope that it's point coordinates
809             else:
810                 raise TypeError,\
811                       "The Threshold should either a VERTEX, or a node ID, "\
812                       "or a list of point coordinates and not '%s'"%aThreshold
813         elif CritType == FT_ElemGeomType:
814             # Check the Threshold
815             try:
816                 aCriterion.Threshold = self.EnumToLong(aThreshold)
817                 assert( aThreshold in SMESH.GeometryType._items )
818             except:
819                 if isinstance(aThreshold, int):
820                     aCriterion.Threshold = aThreshold
821                 else:
822                     raise TypeError, "The Threshold should be an integer or SMESH.GeometryType."
823                 pass
824             pass
825         elif CritType == FT_EntityType:
826             # Check the Threshold
827             try:
828                 aCriterion.Threshold = self.EnumToLong(aThreshold)
829                 assert( aThreshold in SMESH.EntityType._items )
830             except:
831                 if isinstance(aThreshold, int):
832                     aCriterion.Threshold = aThreshold
833                 else:
834                     raise TypeError, "The Threshold should be an integer or SMESH.EntityType."
835                 pass
836             pass
837         
838         elif CritType == FT_GroupColor:
839             # Check the Threshold
840             try:
841                 aCriterion.ThresholdStr = self.ColorToString(aThreshold)
842             except:
843                 raise TypeError, "The threshold value should be of SALOMEDS.Color type"
844             pass
845         elif CritType in [FT_FreeBorders, FT_FreeEdges, FT_FreeNodes, FT_FreeFaces,
846                           FT_LinearOrQuadratic, FT_BadOrientedVolume,
847                           FT_BareBorderFace, FT_BareBorderVolume,
848                           FT_OverConstrainedFace, FT_OverConstrainedVolume,
849                           FT_EqualNodes,FT_EqualEdges,FT_EqualFaces,FT_EqualVolumes ]:
850             # At this point the Threshold is unnecessary
851             if aThreshold ==  FT_LogicalNOT:
852                 aCriterion.UnaryOp = self.EnumToLong(FT_LogicalNOT)
853             elif aThreshold in [FT_LogicalAND, FT_LogicalOR]:
854                 aCriterion.BinaryOp = aThreshold
855         else:
856             # Check Threshold
857             try:
858                 aThreshold = float(aThreshold)
859                 aCriterion.Threshold = aThreshold
860             except:
861                 raise TypeError, "The Threshold should be a number."
862                 return None
863
864         if Threshold ==  FT_LogicalNOT or UnaryOp ==  FT_LogicalNOT:
865             aCriterion.UnaryOp = self.EnumToLong(FT_LogicalNOT)
866
867         if Threshold in [FT_LogicalAND, FT_LogicalOR]:
868             aCriterion.BinaryOp = self.EnumToLong(Threshold)
869
870         if UnaryOp in [FT_LogicalAND, FT_LogicalOR]:
871             aCriterion.BinaryOp = self.EnumToLong(UnaryOp)
872
873         if BinaryOp in [FT_LogicalAND, FT_LogicalOR]:
874             aCriterion.BinaryOp = self.EnumToLong(BinaryOp)
875
876         return aCriterion
877
878     ## Creates a filter with the given parameters
879     #  @param elementType the type of elements (SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME)
880     #  @param CritType the type of criterion (SMESH.FT_Taper, SMESH.FT_Area, etc.)
881     #          Type SMESH.FunctorType._items in the Python Console to see all values.
882     #          Note that the items starting from FT_LessThan are not suitable for CritType.
883     #  @param Compare  belongs to {SMESH.FT_LessThan, SMESH.FT_MoreThan, SMESH.FT_EqualTo}
884     #  @param Threshold the threshold value (range of ids as string, shape, numeric)
885     #  @param UnaryOp  SMESH.FT_LogicalNOT or SMESH.FT_Undefined
886     #  @param Tolerance the tolerance used by SMESH.FT_BelongToGeom, SMESH.FT_BelongToSurface,
887     #         SMESH.FT_LyingOnGeom, SMESH.FT_CoplanarFaces and SMESH.FT_EqualNodes criteria
888     #  @param mesh the mesh to initialize the filter with
889     #  @return SMESH_Filter
890     #
891     #  <a href="../tui_filters_page.html#tui_filters">Example of Filters usage</a>
892     #  @ingroup l1_controls
893     def GetFilter(self,elementType,
894                   CritType=FT_Undefined,
895                   Compare=FT_EqualTo,
896                   Threshold="",
897                   UnaryOp=FT_Undefined,
898                   Tolerance=1e-07,
899                   mesh=None):
900         aCriterion = self.GetCriterion(elementType, CritType, Compare, Threshold, UnaryOp, FT_Undefined,Tolerance)
901         aFilterMgr = self.CreateFilterManager()
902         aFilter = aFilterMgr.CreateFilter()
903         aCriteria = []
904         aCriteria.append(aCriterion)
905         aFilter.SetCriteria(aCriteria)
906         if mesh:
907             if isinstance( mesh, Mesh ): aFilter.SetMesh( mesh.GetMesh() )
908             else                       : aFilter.SetMesh( mesh )
909         aFilterMgr.UnRegister()
910         return aFilter
911
912     ## Creates a filter from criteria
913     #  @param criteria a list of criteria
914     #  @param binOp binary operator used when binary operator of criteria is undefined
915     #  @return SMESH_Filter
916     #
917     #  <a href="../tui_filters_page.html#tui_filters">Example of Filters usage</a>
918     #  @ingroup l1_controls
919     def GetFilterFromCriteria(self,criteria, binOp=SMESH.FT_LogicalAND):
920         for i in range( len( criteria ) - 1 ):
921             if criteria[i].BinaryOp == self.EnumToLong( SMESH.FT_Undefined ):
922                 criteria[i].BinaryOp = self.EnumToLong( binOp )
923         aFilterMgr = self.CreateFilterManager()
924         aFilter = aFilterMgr.CreateFilter()
925         aFilter.SetCriteria(criteria)
926         aFilterMgr.UnRegister()
927         return aFilter
928
929     ## Creates a numerical functor by its type
930     #  @param theCriterion functor type - an item of SMESH.FunctorType enumeration.
931     #          Type SMESH.FunctorType._items in the Python Console to see all items.
932     #          Note that not all items correspond to numerical functors.
933     #  @return SMESH_NumericalFunctor
934     #  @ingroup l1_controls
935     def GetFunctor(self,theCriterion):
936         if isinstance( theCriterion, SMESH._objref_NumericalFunctor ):
937             return theCriterion
938         aFilterMgr = self.CreateFilterManager()
939         functor = None
940         if theCriterion == FT_AspectRatio:
941             functor = aFilterMgr.CreateAspectRatio()
942         elif theCriterion == FT_AspectRatio3D:
943             functor = aFilterMgr.CreateAspectRatio3D()
944         elif theCriterion == FT_Warping:
945             functor = aFilterMgr.CreateWarping()
946         elif theCriterion == FT_MinimumAngle:
947             functor = aFilterMgr.CreateMinimumAngle()
948         elif theCriterion == FT_Taper:
949             functor = aFilterMgr.CreateTaper()
950         elif theCriterion == FT_Skew:
951             functor = aFilterMgr.CreateSkew()
952         elif theCriterion == FT_Area:
953             functor = aFilterMgr.CreateArea()
954         elif theCriterion == FT_Volume3D:
955             functor = aFilterMgr.CreateVolume3D()
956         elif theCriterion == FT_MaxElementLength2D:
957             functor = aFilterMgr.CreateMaxElementLength2D()
958         elif theCriterion == FT_MaxElementLength3D:
959             functor = aFilterMgr.CreateMaxElementLength3D()
960         elif theCriterion == FT_MultiConnection:
961             functor = aFilterMgr.CreateMultiConnection()
962         elif theCriterion == FT_MultiConnection2D:
963             functor = aFilterMgr.CreateMultiConnection2D()
964         elif theCriterion == FT_Length:
965             functor = aFilterMgr.CreateLength()
966         elif theCriterion == FT_Length2D:
967             functor = aFilterMgr.CreateLength2D()
968         else:
969             print "Error: given parameter is not numerical functor type."
970         aFilterMgr.UnRegister()
971         return functor
972
973     ## Creates hypothesis
974     #  @param theHType mesh hypothesis type (string)
975     #  @param theLibName mesh plug-in library name
976     #  @return created hypothesis instance
977     def CreateHypothesis(self, theHType, theLibName="libStdMeshersEngine.so"):
978         hyp = SMESH._objref_SMESH_Gen.CreateHypothesis(self, theHType, theLibName )
979
980         if isinstance( hyp, SMESH._objref_SMESH_Algo ):
981             return hyp
982
983         # wrap hypothesis methods
984         #print "HYPOTHESIS", theHType
985         for meth_name in dir( hyp.__class__ ):
986             if not meth_name.startswith("Get") and \
987                not meth_name in dir ( SMESH._objref_SMESH_Hypothesis ):
988                 method = getattr ( hyp.__class__, meth_name )
989                 if callable(method):
990                     setattr( hyp, meth_name, hypMethodWrapper( hyp, method ))
991
992         return hyp
993
994     ## Gets the mesh statistic
995     #  @return dictionary "element type" - "count of elements"
996     #  @ingroup l1_meshinfo
997     def GetMeshInfo(self, obj):
998         if isinstance( obj, Mesh ):
999             obj = obj.GetMesh()
1000         d = {}
1001         if hasattr(obj, "GetMeshInfo"):
1002             values = obj.GetMeshInfo()
1003             for i in range(SMESH.Entity_Last._v):
1004                 if i < len(values): d[SMESH.EntityType._item(i)]=values[i]
1005             pass
1006         return d
1007
1008     ## Get minimum distance between two objects
1009     #
1010     #  If @a src2 is None, and @a id2 = 0, distance from @a src1 / @a id1 to the origin is computed.
1011     #  If @a src2 is None, and @a id2 != 0, it is assumed that both @a id1 and @a id2 belong to @a src1.
1012     #
1013     #  @param src1 first source object
1014     #  @param src2 second source object
1015     #  @param id1 node/element id from the first source
1016     #  @param id2 node/element id from the second (or first) source
1017     #  @param isElem1 @c True if @a id1 is element id, @c False if it is node id
1018     #  @param isElem2 @c True if @a id2 is element id, @c False if it is node id
1019     #  @return minimum distance value
1020     #  @sa GetMinDistance()
1021     #  @ingroup l1_measurements
1022     def MinDistance(self, src1, src2=None, id1=0, id2=0, isElem1=False, isElem2=False):
1023         result = self.GetMinDistance(src1, src2, id1, id2, isElem1, isElem2)
1024         if result is None:
1025             result = 0.0
1026         else:
1027             result = result.value
1028         return result
1029
1030     ## Get measure structure specifying minimum distance data between two objects
1031     #
1032     #  If @a src2 is None, and @a id2 = 0, distance from @a src1 / @a id1 to the origin is computed.
1033     #  If @a src2 is None, and @a id2 != 0, it is assumed that both @a id1 and @a id2 belong to @a src1.
1034     #
1035     #  @param src1 first source object
1036     #  @param src2 second source object
1037     #  @param id1 node/element id from the first source
1038     #  @param id2 node/element id from the second (or first) source
1039     #  @param isElem1 @c True if @a id1 is element id, @c False if it is node id
1040     #  @param isElem2 @c True if @a id2 is element id, @c False if it is node id
1041     #  @return Measure structure or None if input data is invalid
1042     #  @sa MinDistance()
1043     #  @ingroup l1_measurements
1044     def GetMinDistance(self, src1, src2=None, id1=0, id2=0, isElem1=False, isElem2=False):
1045         if isinstance(src1, Mesh): src1 = src1.mesh
1046         if isinstance(src2, Mesh): src2 = src2.mesh
1047         if src2 is None and id2 != 0: src2 = src1
1048         if not hasattr(src1, "_narrow"): return None
1049         src1 = src1._narrow(SMESH.SMESH_IDSource)
1050         if not src1: return None
1051         unRegister = genObjUnRegister()
1052         if id1 != 0:
1053             m = src1.GetMesh()
1054             e = m.GetMeshEditor()
1055             if isElem1:
1056                 src1 = e.MakeIDSource([id1], SMESH.FACE)
1057             else:
1058                 src1 = e.MakeIDSource([id1], SMESH.NODE)
1059             unRegister.set( src1 )
1060             pass
1061         if hasattr(src2, "_narrow"):
1062             src2 = src2._narrow(SMESH.SMESH_IDSource)
1063             if src2 and id2 != 0:
1064                 m = src2.GetMesh()
1065                 e = m.GetMeshEditor()
1066                 if isElem2:
1067                     src2 = e.MakeIDSource([id2], SMESH.FACE)
1068                 else:
1069                     src2 = e.MakeIDSource([id2], SMESH.NODE)
1070                 unRegister.set( src2 )
1071                 pass
1072             pass
1073         aMeasurements = self.CreateMeasurements()
1074         unRegister.set( aMeasurements )
1075         result = aMeasurements.MinDistance(src1, src2)
1076         return result
1077
1078     ## Get bounding box of the specified object(s)
1079     #  @param objects single source object or list of source objects
1080     #  @return tuple of six values (minX, minY, minZ, maxX, maxY, maxZ)
1081     #  @sa GetBoundingBox()
1082     #  @ingroup l1_measurements
1083     def BoundingBox(self, objects):
1084         result = self.GetBoundingBox(objects)
1085         if result is None:
1086             result = (0.0,)*6
1087         else:
1088             result = (result.minX, result.minY, result.minZ, result.maxX, result.maxY, result.maxZ)
1089         return result
1090
1091     ## Get measure structure specifying bounding box data of the specified object(s)
1092     #  @param objects single source object or list of source objects
1093     #  @return Measure structure
1094     #  @sa BoundingBox()
1095     #  @ingroup l1_measurements
1096     def GetBoundingBox(self, objects):
1097         if isinstance(objects, tuple):
1098             objects = list(objects)
1099         if not isinstance(objects, list):
1100             objects = [objects]
1101         srclist = []
1102         for o in objects:
1103             if isinstance(o, Mesh):
1104                 srclist.append(o.mesh)
1105             elif hasattr(o, "_narrow"):
1106                 src = o._narrow(SMESH.SMESH_IDSource)
1107                 if src: srclist.append(src)
1108                 pass
1109             pass
1110         aMeasurements = self.CreateMeasurements()
1111         result = aMeasurements.BoundingBox(srclist)
1112         aMeasurements.UnRegister()
1113         return result
1114
1115     ## Get sum of lengths of all 1D elements in the mesh object.
1116     #  @param obj mesh, submesh or group
1117     #  @return sum of lengths of all 1D elements
1118     #  @ingroup l1_measurements
1119     def GetLength(self, obj):
1120         if isinstance(obj, Mesh): obj = obj.mesh
1121         if isinstance(obj, Mesh_Algorithm): obj = obj.GetSubMesh()
1122         aMeasurements = self.CreateMeasurements()
1123         value = aMeasurements.Length(obj)
1124         aMeasurements.UnRegister()
1125         return value
1126
1127     ## Get sum of areas of all 2D elements in the mesh object.
1128     #  @param obj mesh, submesh or group
1129     #  @return sum of areas of all 2D elements
1130     #  @ingroup l1_measurements
1131     def GetArea(self, obj):
1132         if isinstance(obj, Mesh): obj = obj.mesh
1133         if isinstance(obj, Mesh_Algorithm): obj = obj.GetSubMesh()
1134         aMeasurements = self.CreateMeasurements()
1135         value = aMeasurements.Area(obj)
1136         aMeasurements.UnRegister()
1137         return value
1138
1139     ## Get sum of volumes of all 3D elements in the mesh object.
1140     #  @param obj mesh, submesh or group
1141     #  @return sum of volumes of all 3D elements
1142     #  @ingroup l1_measurements
1143     def GetVolume(self, obj):
1144         if isinstance(obj, Mesh): obj = obj.mesh
1145         if isinstance(obj, Mesh_Algorithm): obj = obj.GetSubMesh()
1146         aMeasurements = self.CreateMeasurements()
1147         value = aMeasurements.Volume(obj)
1148         aMeasurements.UnRegister()
1149         return value
1150
1151     pass # end of class smeshBuilder
1152
1153 import omniORB
1154 #Registering the new proxy for SMESH_Gen
1155 omniORB.registerObjref(SMESH._objref_SMESH_Gen._NP_RepositoryId, smeshBuilder)
1156
1157 ## Create a new smeshBuilder instance.The smeshBuilder class provides the Python
1158 #  interface to create or load meshes.
1159 #
1160 #  Typical use is:
1161 #  \code
1162 #    import salome
1163 #    salome.salome_init()
1164 #    from salome.smesh import smeshBuilder
1165 #    smesh = smeshBuilder.New(theStudy)
1166 #  \endcode
1167 #  @param  study     SALOME study, generally obtained by salome.myStudy.
1168 #  @param  instance  CORBA proxy of SMESH Engine. If None, the default Engine is used.
1169 #  @return smeshBuilder instance
1170
1171 def New( study, instance=None):
1172     """
1173     Create a new smeshBuilder instance.The smeshBuilder class provides the Python
1174     interface to create or load meshes.
1175
1176     Typical use is:
1177         import salome
1178         salome.salome_init()
1179         from salome.smesh import smeshBuilder
1180         smesh = smeshBuilder.New(theStudy)
1181
1182     Parameters:
1183         study     SALOME study, generally obtained by salome.myStudy.
1184         instance  CORBA proxy of SMESH Engine. If None, the default Engine is used.
1185     Returns:
1186         smeshBuilder instance
1187     """
1188     global engine
1189     global smeshInst
1190     global doLcc
1191     engine = instance
1192     if engine is None:
1193       doLcc = True
1194     smeshInst = smeshBuilder()
1195     assert isinstance(smeshInst,smeshBuilder), "Smesh engine class is %s but should be smeshBuilder.smeshBuilder. Import salome.smesh.smeshBuilder before creating the instance."%smeshInst.__class__
1196     smeshInst.init_smesh(study)
1197     return smeshInst
1198
1199
1200 # Public class: Mesh
1201 # ==================
1202
1203 ## This class allows defining and managing a mesh.
1204 #  It has a set of methods to build a mesh on the given geometry, including the definition of sub-meshes.
1205 #  It also has methods to define groups of mesh elements, to modify a mesh (by addition of
1206 #  new nodes and elements and by changing the existing entities), to get information
1207 #  about a mesh and to export a mesh into different formats.
1208 class Mesh:
1209     __metaclass__ = MeshMeta
1210
1211     geom = 0
1212     mesh = 0
1213     editor = 0
1214
1215     ## Constructor
1216     #
1217     #  Creates a mesh on the shape \a obj (or an empty mesh if \a obj is equal to 0) and
1218     #  sets the GUI name of this mesh to \a name.
1219     #  @param smeshpyD an instance of smeshBuilder class
1220     #  @param geompyD an instance of geomBuilder class
1221     #  @param obj Shape to be meshed or SMESH_Mesh object
1222     #  @param name Study name of the mesh
1223     #  @ingroup l2_construct
1224     def __init__(self, smeshpyD, geompyD, obj=0, name=0):
1225         self.smeshpyD=smeshpyD
1226         self.geompyD=geompyD
1227         if obj is None:
1228             obj = 0
1229         objHasName = False
1230         if obj != 0:
1231             if isinstance(obj, geomBuilder.GEOM._objref_GEOM_Object):
1232                 self.geom = obj
1233                 objHasName = True
1234                 # publish geom of mesh (issue 0021122)
1235                 if not self.geom.GetStudyEntry() and smeshpyD.GetCurrentStudy():
1236                     objHasName = False
1237                     studyID = smeshpyD.GetCurrentStudy()._get_StudyId()
1238                     if studyID != geompyD.myStudyId:
1239                         geompyD.init_geom( smeshpyD.GetCurrentStudy())
1240                         pass
1241                     if name:
1242                         geo_name = name + " shape"
1243                     else:
1244                         geo_name = "%s_%s to mesh"%(self.geom.GetShapeType(), id(self.geom)%100)
1245                     geompyD.addToStudy( self.geom, geo_name )
1246                 self.SetMesh( self.smeshpyD.CreateMesh(self.geom) )
1247
1248             elif isinstance(obj, SMESH._objref_SMESH_Mesh):
1249                 self.SetMesh(obj)
1250         else:
1251             self.SetMesh( self.smeshpyD.CreateEmptyMesh() )
1252         if name:
1253             self.smeshpyD.SetName(self.mesh, name)
1254         elif objHasName:
1255             self.smeshpyD.SetName(self.mesh, GetName(obj)) # + " mesh"
1256
1257         if not self.geom:
1258             self.geom = self.mesh.GetShapeToMesh()
1259
1260         self.editor   = self.mesh.GetMeshEditor()
1261         self.functors = [None] * SMESH.FT_Undefined._v
1262
1263         # set self to algoCreator's
1264         for attrName in dir(self):
1265             attr = getattr( self, attrName )
1266             if isinstance( attr, algoCreator ):
1267                 setattr( self, attrName, attr.copy( self ))
1268                 pass
1269             pass
1270         pass
1271
1272     ## Destructor. Clean-up resources
1273     def __del__(self):
1274         if self.mesh:
1275             #self.mesh.UnRegister()
1276             pass
1277         pass
1278         
1279     ## Initializes the Mesh object from an instance of SMESH_Mesh interface
1280     #  @param theMesh a SMESH_Mesh object
1281     #  @ingroup l2_construct
1282     def SetMesh(self, theMesh):
1283         # do not call Register() as this prevents mesh servant deletion at closing study
1284         #if self.mesh: self.mesh.UnRegister()
1285         self.mesh = theMesh
1286         if self.mesh:
1287             #self.mesh.Register()
1288             self.geom = self.mesh.GetShapeToMesh()
1289         pass
1290
1291     ## Returns the mesh, that is an instance of SMESH_Mesh interface
1292     #  @return a SMESH_Mesh object
1293     #  @ingroup l2_construct
1294     def GetMesh(self):
1295         return self.mesh
1296
1297     ## Gets the name of the mesh
1298     #  @return the name of the mesh as a string
1299     #  @ingroup l2_construct
1300     def GetName(self):
1301         name = GetName(self.GetMesh())
1302         return name
1303
1304     ## Sets a name to the mesh
1305     #  @param name a new name of the mesh
1306     #  @ingroup l2_construct
1307     def SetName(self, name):
1308         self.smeshpyD.SetName(self.GetMesh(), name)
1309
1310     ## Gets the subMesh object associated to a \a theSubObject geometrical object.
1311     #  The subMesh object gives access to the IDs of nodes and elements.
1312     #  @param geom a geometrical object (shape)
1313     #  @param name a name for the submesh
1314     #  @return an object of type SMESH_SubMesh, representing a part of mesh, which lies on the given shape
1315     #  @ingroup l2_submeshes
1316     def GetSubMesh(self, geom, name):
1317         AssureGeomPublished( self, geom, name )
1318         submesh = self.mesh.GetSubMesh( geom, name )
1319         return submesh
1320
1321     ## Returns the shape associated to the mesh
1322     #  @return a GEOM_Object
1323     #  @ingroup l2_construct
1324     def GetShape(self):
1325         return self.geom
1326
1327     ## Associates the given shape to the mesh (entails the recreation of the mesh)
1328     #  @param geom the shape to be meshed (GEOM_Object)
1329     #  @ingroup l2_construct
1330     def SetShape(self, geom):
1331         self.mesh = self.smeshpyD.CreateMesh(geom)
1332
1333     ## Loads mesh from the study after opening the study
1334     def Load(self):
1335         self.mesh.Load()
1336
1337     ## Returns true if the hypotheses are defined well
1338     #  @param theSubObject a sub-shape of a mesh shape
1339     #  @return True or False
1340     #  @ingroup l2_construct
1341     def IsReadyToCompute(self, theSubObject):
1342         return self.smeshpyD.IsReadyToCompute(self.mesh, theSubObject)
1343
1344     ## Returns errors of hypotheses definition.
1345     #  The list of errors is empty if everything is OK.
1346     #  @param theSubObject a sub-shape of a mesh shape
1347     #  @return a list of errors
1348     #  @ingroup l2_construct
1349     def GetAlgoState(self, theSubObject):
1350         return self.smeshpyD.GetAlgoState(self.mesh, theSubObject)
1351
1352     ## Returns a geometrical object on which the given element was built.
1353     #  The returned geometrical object, if not nil, is either found in the
1354     #  study or published by this method with the given name
1355     #  @param theElementID the id of the mesh element
1356     #  @param theGeomName the user-defined name of the geometrical object
1357     #  @return GEOM::GEOM_Object instance
1358     #  @ingroup l2_construct
1359     def GetGeometryByMeshElement(self, theElementID, theGeomName):
1360         return self.smeshpyD.GetGeometryByMeshElement( self.mesh, theElementID, theGeomName )
1361
1362     ## Returns the mesh dimension depending on the dimension of the underlying shape
1363     #  or, if the mesh is not based on any shape, basing on deimension of elements
1364     #  @return mesh dimension as an integer value [0,3]
1365     #  @ingroup l1_auxiliary
1366     def MeshDimension(self):
1367         if self.mesh.HasShapeToMesh():
1368             shells = self.geompyD.SubShapeAllIDs( self.geom, self.geompyD.ShapeType["SOLID"] )
1369             if len( shells ) > 0 :
1370                 return 3
1371             elif self.geompyD.NumberOfFaces( self.geom ) > 0 :
1372                 return 2
1373             elif self.geompyD.NumberOfEdges( self.geom ) > 0 :
1374                 return 1
1375             else:
1376                 return 0;
1377         else:
1378             if self.NbVolumes() > 0: return 3
1379             if self.NbFaces()   > 0: return 2
1380             if self.NbEdges()   > 0: return 1
1381         return 0
1382
1383     ## Evaluates size of prospective mesh on a shape
1384     #  @return a list where i-th element is a number of elements of i-th SMESH.EntityType
1385     #  To know predicted number of e.g. edges, inquire it this way
1386     #  Evaluate()[ EnumToLong( Entity_Edge )]
1387     def Evaluate(self, geom=0):
1388         if geom == 0 or not isinstance(geom, geomBuilder.GEOM._objref_GEOM_Object):
1389             if self.geom == 0:
1390                 geom = self.mesh.GetShapeToMesh()
1391             else:
1392                 geom = self.geom
1393         return self.smeshpyD.Evaluate(self.mesh, geom)
1394
1395
1396     ## Computes the mesh and returns the status of the computation
1397     #  @param geom geomtrical shape on which mesh data should be computed
1398     #  @param discardModifs if True and the mesh has been edited since
1399     #         a last total re-compute and that may prevent successful partial re-compute,
1400     #         then the mesh is cleaned before Compute()
1401     #  @param refresh if @c True, Object browser is automatically updated (when running in GUI)
1402     #  @return True or False
1403     #  @ingroup l2_construct
1404     def Compute(self, geom=0, discardModifs=False, refresh=False):
1405         if geom == 0 or not isinstance(geom, geomBuilder.GEOM._objref_GEOM_Object):
1406             if self.geom == 0:
1407                 geom = self.mesh.GetShapeToMesh()
1408             else:
1409                 geom = self.geom
1410         ok = False
1411         try:
1412             if discardModifs and self.mesh.HasModificationsToDiscard(): # issue 0020693
1413                 self.mesh.Clear()
1414             ok = self.smeshpyD.Compute(self.mesh, geom)
1415         except SALOME.SALOME_Exception, ex:
1416             print "Mesh computation failed, exception caught:"
1417             print "    ", ex.details.text
1418         except:
1419             import traceback
1420             print "Mesh computation failed, exception caught:"
1421             traceback.print_exc()
1422         if True:#not ok:
1423             allReasons = ""
1424
1425             # Treat compute errors
1426             computeErrors = self.smeshpyD.GetComputeErrors( self.mesh, geom )
1427             shapeText = ""
1428             for err in computeErrors:
1429                 if self.mesh.HasShapeToMesh():
1430                     shapeText = " on %s" % self.GetSubShapeName( err.subShapeID )
1431                 errText = ""
1432                 stdErrors = ["OK",                   #COMPERR_OK
1433                              "Invalid input mesh",   #COMPERR_BAD_INPUT_MESH
1434                              "std::exception",       #COMPERR_STD_EXCEPTION
1435                              "OCC exception",        #COMPERR_OCC_EXCEPTION
1436                              "..",                   #COMPERR_SLM_EXCEPTION
1437                              "Unknown exception",    #COMPERR_EXCEPTION
1438                              "Memory allocation problem", #COMPERR_MEMORY_PB
1439                              "Algorithm failed",     #COMPERR_ALGO_FAILED
1440                              "Unexpected geometry",  #COMPERR_BAD_SHAPE
1441                              "Warning",              #COMPERR_WARNING
1442                              "Computation cancelled",#COMPERR_CANCELED
1443                              "No mesh on sub-shape"] #COMPERR_NO_MESH_ON_SHAPE
1444                 if err.code > 0:
1445                     if err.code < len(stdErrors): errText = stdErrors[err.code]
1446                 else:
1447                     errText = "code %s" % -err.code
1448                 if errText: errText += ". "
1449                 errText += err.comment
1450                 if allReasons != "":allReasons += "\n"
1451                 if ok:
1452                     allReasons += '-  "%s"%s - %s' %(err.algoName, shapeText, errText)
1453                 else:
1454                     allReasons += '-  "%s" failed%s. Error: %s' %(err.algoName, shapeText, errText)
1455                 pass
1456
1457             # Treat hyp errors
1458             errors = self.smeshpyD.GetAlgoState( self.mesh, geom )
1459             for err in errors:
1460                 if err.isGlobalAlgo:
1461                     glob = "global"
1462                 else:
1463                     glob = "local"
1464                     pass
1465                 dim = err.algoDim
1466                 name = err.algoName
1467                 if len(name) == 0:
1468                     reason = '%s %sD algorithm is missing' % (glob, dim)
1469                 elif err.state == HYP_MISSING:
1470                     reason = ('%s %sD algorithm "%s" misses %sD hypothesis'
1471                               % (glob, dim, name, dim))
1472                 elif err.state == HYP_NOTCONFORM:
1473                     reason = 'Global "Not Conform mesh allowed" hypothesis is missing'
1474                 elif err.state == HYP_BAD_PARAMETER:
1475                     reason = ('Hypothesis of %s %sD algorithm "%s" has a bad parameter value'
1476                               % ( glob, dim, name ))
1477                 elif err.state == HYP_BAD_GEOMETRY:
1478                     reason = ('%s %sD algorithm "%s" is assigned to mismatching'
1479                               'geometry' % ( glob, dim, name ))
1480                 elif err.state == HYP_HIDDEN_ALGO:
1481                     reason = ('%s %sD algorithm "%s" is ignored due to presence of a %s '
1482                               'algorithm of upper dimension generating %sD mesh'
1483                               % ( glob, dim, name, glob, dim ))
1484                 else:
1485                     reason = ("For unknown reason. "
1486                               "Developer, revise Mesh.Compute() implementation in smeshBuilder.py!")
1487                     pass
1488                 if allReasons != "":allReasons += "\n"
1489                 allReasons += "-  " + reason
1490                 pass
1491             if not ok or allReasons != "":
1492                 msg = '"' + GetName(self.mesh) + '"'
1493                 if ok: msg += " has been computed with warnings"
1494                 else:  msg += " has not been computed"
1495                 if allReasons != "": msg += ":"
1496                 else:                msg += "."
1497                 print msg
1498                 print allReasons
1499             pass
1500         if salome.sg.hasDesktop() and self.mesh.GetStudyId() >= 0:
1501             smeshgui = salome.ImportComponentGUI("SMESH")
1502             smeshgui.Init(self.mesh.GetStudyId())
1503             smeshgui.SetMeshIcon( salome.ObjectToID( self.mesh ), ok, (self.NbNodes()==0) )
1504             if refresh: salome.sg.updateObjBrowser(1)
1505             pass
1506         return ok
1507
1508     ## Return a name of a sub-shape by its ID
1509     #  @param subShapeID a unique ID of a sub-shape
1510     #  @return a string describing the sub-shape; possible variants:
1511     #  - "Face_12"    (published sub-shape)
1512     #  - FACE #3      (not published sub-shape)
1513     #  - sub-shape #3 (invalid sub-shape ID)
1514     #  - #3           (error in this function)
1515     def GetSubShapeName(self, subShapeID ):
1516         if not self.mesh.HasShapeToMesh():
1517             return ""
1518         try:
1519             shapeText = ""
1520             mainIOR  = salome.orb.object_to_string( self.GetShape() )
1521             for sname in salome.myStudyManager.GetOpenStudies():
1522                 s = salome.myStudyManager.GetStudyByName(sname)
1523                 if not s: continue
1524                 mainSO = s.FindObjectIOR(mainIOR)
1525                 if not mainSO: continue
1526                 if subShapeID == 1:
1527                     shapeText = '"%s"' % mainSO.GetName()
1528                 subIt = s.NewChildIterator(mainSO)
1529                 while subIt.More():
1530                     subSO = subIt.Value()
1531                     subIt.Next()
1532                     obj = subSO.GetObject()
1533                     if not obj: continue
1534                     go = obj._narrow( geomBuilder.GEOM._objref_GEOM_Object )
1535                     if not go: continue
1536                     try:
1537                         ids = self.geompyD.GetSubShapeID( self.GetShape(), go )
1538                     except:
1539                         continue
1540                     if ids == subShapeID:
1541                         shapeText = '"%s"' % subSO.GetName()
1542                         break
1543             if not shapeText:
1544                 shape = self.geompyD.GetSubShape( self.GetShape(), [subShapeID])
1545                 if shape:
1546                     shapeText = '%s #%s' % (shape.GetShapeType(), subShapeID)
1547                 else:
1548                     shapeText = 'sub-shape #%s' % (subShapeID)
1549         except:
1550             shapeText = "#%s" % (subShapeID)
1551         return shapeText
1552
1553     ## Return a list of sub-shapes meshing of which failed, grouped into GEOM groups by
1554     #  error of an algorithm
1555     #  @param publish if @c True, the returned groups will be published in the study
1556     #  @return a list of GEOM groups each named after a failed algorithm
1557     def GetFailedShapes(self, publish=False):
1558
1559         algo2shapes = {}
1560         computeErrors = self.smeshpyD.GetComputeErrors( self.mesh, self.GetShape() )
1561         for err in computeErrors:
1562             shape = self.geompyD.GetSubShape( self.GetShape(), [err.subShapeID])
1563             if not shape: continue
1564             if err.algoName in algo2shapes:
1565                 algo2shapes[ err.algoName ].append( shape )
1566             else:
1567                 algo2shapes[ err.algoName ] = [ shape ]
1568             pass
1569
1570         groups = []
1571         for algoName, shapes in algo2shapes.items():
1572             while shapes:
1573                 groupType = self.smeshpyD.EnumToLong( shapes[0].GetShapeType() )
1574                 otherTypeShapes = []
1575                 sameTypeShapes  = []
1576                 group = self.geompyD.CreateGroup( self.geom, groupType )
1577                 for shape in shapes:
1578                     if shape.GetShapeType() == shapes[0].GetShapeType():
1579                         sameTypeShapes.append( shape )
1580                     else:
1581                         otherTypeShapes.append( shape )
1582                 self.geompyD.UnionList( group, sameTypeShapes )
1583                 if otherTypeShapes:
1584                     group.SetName( "%s %s" % ( algoName, shapes[0].GetShapeType() ))
1585                 else:
1586                     group.SetName( algoName )
1587                 groups.append( group )
1588                 shapes = otherTypeShapes
1589             pass
1590         if publish:
1591             for group in groups:
1592                 self.geompyD.addToStudyInFather( self.geom, group, group.GetName() )
1593         return groups
1594
1595     ## Return sub-mesh objects list in meshing order
1596     #  @return list of list of sub-meshes
1597     #  @ingroup l2_construct
1598     def GetMeshOrder(self):
1599         return self.mesh.GetMeshOrder()
1600
1601     ## Set order in which concurrent sub-meshes sould be meshed
1602     #  @param submeshes list of sub-meshes
1603     #  @ingroup l2_construct
1604     def SetMeshOrder(self, submeshes):
1605         return self.mesh.SetMeshOrder(submeshes)
1606
1607     ## Removes all nodes and elements
1608     #  @param refresh if @c True, Object browser is automatically updated (when running in GUI)
1609     #  @ingroup l2_construct
1610     def Clear(self, refresh=False):
1611         self.mesh.Clear()
1612         if ( salome.sg.hasDesktop() and 
1613              salome.myStudyManager.GetStudyByID( self.mesh.GetStudyId() ) ):
1614             smeshgui = salome.ImportComponentGUI("SMESH")
1615             smeshgui.Init(self.mesh.GetStudyId())
1616             smeshgui.SetMeshIcon( salome.ObjectToID( self.mesh ), False, True )
1617             if refresh: salome.sg.updateObjBrowser(1)
1618
1619     ## Removes all nodes and elements of indicated shape
1620     #  @param refresh if @c True, Object browser is automatically updated (when running in GUI)
1621     #  @param geomId the ID of a sub-shape to remove elements on
1622     #  @ingroup l2_construct
1623     def ClearSubMesh(self, geomId, refresh=False):
1624         self.mesh.ClearSubMesh(geomId)
1625         if salome.sg.hasDesktop():
1626             smeshgui = salome.ImportComponentGUI("SMESH")
1627             smeshgui.Init(self.mesh.GetStudyId())
1628             smeshgui.SetMeshIcon( salome.ObjectToID( self.mesh ), False, True )
1629             if refresh: salome.sg.updateObjBrowser(1)
1630
1631     ## Computes a tetrahedral mesh using AutomaticLength + MEFISTO + Tetrahedron
1632     #  @param fineness [0.0,1.0] defines mesh fineness
1633     #  @return True or False
1634     #  @ingroup l3_algos_basic
1635     def AutomaticTetrahedralization(self, fineness=0):
1636         dim = self.MeshDimension()
1637         # assign hypotheses
1638         self.RemoveGlobalHypotheses()
1639         self.Segment().AutomaticLength(fineness)
1640         if dim > 1 :
1641             self.Triangle().LengthFromEdges()
1642             pass
1643         if dim > 2 :
1644             self.Tetrahedron()
1645             pass
1646         return self.Compute()
1647
1648     ## Computes an hexahedral mesh using AutomaticLength + Quadrangle + Hexahedron
1649     #  @param fineness [0.0, 1.0] defines mesh fineness
1650     #  @return True or False
1651     #  @ingroup l3_algos_basic
1652     def AutomaticHexahedralization(self, fineness=0):
1653         dim = self.MeshDimension()
1654         # assign the hypotheses
1655         self.RemoveGlobalHypotheses()
1656         self.Segment().AutomaticLength(fineness)
1657         if dim > 1 :
1658             self.Quadrangle()
1659             pass
1660         if dim > 2 :
1661             self.Hexahedron()
1662             pass
1663         return self.Compute()
1664
1665     ## Assigns a hypothesis
1666     #  @param hyp a hypothesis to assign
1667     #  @param geom a subhape of mesh geometry
1668     #  @return SMESH.Hypothesis_Status
1669     #  @ingroup l2_hypotheses
1670     def AddHypothesis(self, hyp, geom=0):
1671         if isinstance( hyp, geomBuilder.GEOM._objref_GEOM_Object ):
1672             hyp, geom = geom, hyp
1673         if isinstance( hyp, Mesh_Algorithm ):
1674             hyp = hyp.GetAlgorithm()
1675             pass
1676         if not geom:
1677             geom = self.geom
1678             if not geom:
1679                 geom = self.mesh.GetShapeToMesh()
1680             pass
1681         isApplicable = True
1682         if self.mesh.HasShapeToMesh():
1683             hyp_type     = hyp.GetName()
1684             lib_name     = hyp.GetLibName()
1685             # checkAll    = ( not geom.IsSame( self.mesh.GetShapeToMesh() ))
1686             # if checkAll and geom:
1687             #     checkAll = geom.GetType() == 37
1688             checkAll     = False
1689             isApplicable = self.smeshpyD.IsApplicable(hyp_type, lib_name, geom, checkAll)
1690         if isApplicable:
1691             AssureGeomPublished( self, geom, "shape for %s" % hyp.GetName())
1692             status = self.mesh.AddHypothesis(geom, hyp)
1693         else:
1694             status = HYP_BAD_GEOMETRY,""
1695         hyp_name = GetName( hyp )
1696         geom_name = ""
1697         if geom:
1698             geom_name = geom.GetName()
1699         isAlgo = hyp._narrow( SMESH_Algo )
1700         TreatHypoStatus( status, hyp_name, geom_name, isAlgo, self )
1701         return status
1702
1703     ## Return True if an algorithm of hypothesis is assigned to a given shape
1704     #  @param hyp a hypothesis to check
1705     #  @param geom a subhape of mesh geometry
1706     #  @return True of False
1707     #  @ingroup l2_hypotheses
1708     def IsUsedHypothesis(self, hyp, geom):
1709         if not hyp: # or not geom
1710             return False
1711         if isinstance( hyp, Mesh_Algorithm ):
1712             hyp = hyp.GetAlgorithm()
1713             pass
1714         hyps = self.GetHypothesisList(geom)
1715         for h in hyps:
1716             if h.GetId() == hyp.GetId():
1717                 return True
1718         return False
1719
1720     ## Unassigns a hypothesis
1721     #  @param hyp a hypothesis to unassign
1722     #  @param geom a sub-shape of mesh geometry
1723     #  @return SMESH.Hypothesis_Status
1724     #  @ingroup l2_hypotheses
1725     def RemoveHypothesis(self, hyp, geom=0):
1726         if not hyp:
1727             return None
1728         if isinstance( hyp, Mesh_Algorithm ):
1729             hyp = hyp.GetAlgorithm()
1730             pass
1731         shape = geom
1732         if not shape:
1733             shape = self.geom
1734             pass
1735         if self.IsUsedHypothesis( hyp, shape ):
1736             return self.mesh.RemoveHypothesis( shape, hyp )
1737         hypName = GetName( hyp )
1738         geoName = GetName( shape )
1739         print "WARNING: RemoveHypothesis() failed as '%s' is not assigned to '%s' shape" % ( hypName, geoName )
1740         return None
1741
1742     ## Gets the list of hypotheses added on a geometry
1743     #  @param geom a sub-shape of mesh geometry
1744     #  @return the sequence of SMESH_Hypothesis
1745     #  @ingroup l2_hypotheses
1746     def GetHypothesisList(self, geom):
1747         return self.mesh.GetHypothesisList( geom )
1748
1749     ## Removes all global hypotheses
1750     #  @ingroup l2_hypotheses
1751     def RemoveGlobalHypotheses(self):
1752         current_hyps = self.mesh.GetHypothesisList( self.geom )
1753         for hyp in current_hyps:
1754             self.mesh.RemoveHypothesis( self.geom, hyp )
1755             pass
1756         pass
1757
1758    ## Exports the mesh in a file in MED format and chooses the \a version of MED format
1759     ## allowing to overwrite the file if it exists or add the exported data to its contents
1760     #  @param f is the file name
1761     #  @param auto_groups boolean parameter for creating/not creating
1762     #  the groups Group_On_All_Nodes, Group_On_All_Faces, ... ;
1763     #  the typical use is auto_groups=false.
1764     #  @param version MED format version(MED_V2_1 or MED_V2_2)
1765     #  @param overwrite boolean parameter for overwriting/not overwriting the file
1766     #  @param meshPart a part of mesh (group, sub-mesh) to export instead of the mesh
1767     #  @param autoDimension: if @c True (default), a space dimension of a MED mesh can be either
1768     #         - 1D if all mesh nodes lie on OX coordinate axis, or
1769     #         - 2D if all mesh nodes lie on XOY coordinate plane, or
1770     #         - 3D in the rest cases.
1771     #         If @a autoDimension is @c False, the space dimension is always 3.
1772     #  @param fields : list of GEOM fields defined on the shape to mesh.
1773     #  @param geomAssocFields : each character of this string means a need to export a 
1774     #         corresponding field; correspondence between fields and characters is following:
1775     #         - 'v' stands for _vertices_ field;
1776     #         - 'e' stands for _edges_ field;
1777     #         - 'f' stands for _faces_ field;
1778     #         - 's' stands for _solids_ field.
1779     #  @ingroup l2_impexp
1780     def ExportMED(self, f, auto_groups=0, version=MED_V2_2,
1781                   overwrite=1, meshPart=None, autoDimension=True, fields=[], geomAssocFields=''):
1782         if meshPart or fields or geomAssocFields:
1783             unRegister = genObjUnRegister()
1784             if isinstance( meshPart, list ):
1785                 meshPart = self.GetIDSource( meshPart, SMESH.ALL )
1786                 unRegister.set( meshPart )
1787             self.mesh.ExportPartToMED( meshPart, f, auto_groups, version, overwrite, autoDimension,
1788                                        fields, geomAssocFields)
1789         else:
1790             self.mesh.ExportToMEDX(f, auto_groups, version, overwrite, autoDimension)
1791
1792     ## Exports the mesh in a file in SAUV format
1793     #  @param f is the file name
1794     #  @param auto_groups boolean parameter for creating/not creating
1795     #  the groups Group_On_All_Nodes, Group_On_All_Faces, ... ;
1796     #  the typical use is auto_groups=false.
1797     #  @ingroup l2_impexp
1798     def ExportSAUV(self, f, auto_groups=0):
1799         self.mesh.ExportSAUV(f, auto_groups)
1800
1801     ## Exports the mesh in a file in DAT format
1802     #  @param f the file name
1803     #  @param meshPart a part of mesh (group, sub-mesh) to export instead of the mesh
1804     #  @ingroup l2_impexp
1805     def ExportDAT(self, f, meshPart=None):
1806         if meshPart:
1807             unRegister = genObjUnRegister()
1808             if isinstance( meshPart, list ):
1809                 meshPart = self.GetIDSource( meshPart, SMESH.ALL )
1810                 unRegister.set( meshPart )
1811             self.mesh.ExportPartToDAT( meshPart, f )
1812         else:
1813             self.mesh.ExportDAT(f)
1814
1815     ## Exports the mesh in a file in UNV format
1816     #  @param f the file name
1817     #  @param meshPart a part of mesh (group, sub-mesh) to export instead of the mesh
1818     #  @ingroup l2_impexp
1819     def ExportUNV(self, f, meshPart=None):
1820         if meshPart:
1821             unRegister = genObjUnRegister()
1822             if isinstance( meshPart, list ):
1823                 meshPart = self.GetIDSource( meshPart, SMESH.ALL )
1824                 unRegister.set( meshPart )
1825             self.mesh.ExportPartToUNV( meshPart, f )
1826         else:
1827             self.mesh.ExportUNV(f)
1828
1829     ## Export the mesh in a file in STL format
1830     #  @param f the file name
1831     #  @param ascii defines the file encoding
1832     #  @param meshPart a part of mesh (group, sub-mesh) to export instead of the mesh
1833     #  @ingroup l2_impexp
1834     def ExportSTL(self, f, ascii=1, meshPart=None):
1835         if meshPart:
1836             unRegister = genObjUnRegister()
1837             if isinstance( meshPart, list ):
1838                 meshPart = self.GetIDSource( meshPart, SMESH.ALL )
1839                 unRegister.set( meshPart )
1840             self.mesh.ExportPartToSTL( meshPart, f, ascii )
1841         else:
1842             self.mesh.ExportSTL(f, ascii)
1843
1844     ## Exports the mesh in a file in CGNS format
1845     #  @param f is the file name
1846     #  @param overwrite boolean parameter for overwriting/not overwriting the file
1847     #  @param meshPart a part of mesh (group, sub-mesh) to export instead of the mesh
1848     #  @ingroup l2_impexp
1849     def ExportCGNS(self, f, overwrite=1, meshPart=None):
1850         unRegister = genObjUnRegister()
1851         if isinstance( meshPart, list ):
1852             meshPart = self.GetIDSource( meshPart, SMESH.ALL )
1853             unRegister.set( meshPart )
1854         if isinstance( meshPart, Mesh ):
1855             meshPart = meshPart.mesh
1856         elif not meshPart:
1857             meshPart = self.mesh
1858         self.mesh.ExportCGNS(meshPart, f, overwrite)
1859
1860     ## Exports the mesh in a file in GMF format.
1861     #  GMF files must have .mesh extension for the ASCII format and .meshb for
1862     #  the bynary format. Other extensions are not allowed.
1863     #  @param f is the file name
1864     #  @param meshPart a part of mesh (group, sub-mesh) to export instead of the mesh
1865     #  @ingroup l2_impexp
1866     def ExportGMF(self, f, meshPart=None):
1867         unRegister = genObjUnRegister()
1868         if isinstance( meshPart, list ):
1869             meshPart = self.GetIDSource( meshPart, SMESH.ALL )
1870             unRegister.set( meshPart )
1871         if isinstance( meshPart, Mesh ):
1872             meshPart = meshPart.mesh
1873         elif not meshPart:
1874             meshPart = self.mesh
1875         self.mesh.ExportGMF(meshPart, f, True)
1876
1877     ## Deprecated, used only for compatibility! Please, use ExportToMEDX() method instead.
1878     #  Exports the mesh in a file in MED format and chooses the \a version of MED format
1879     ## allowing to overwrite the file if it exists or add the exported data to its contents
1880     #  @param f the file name
1881     #  @param version values are SMESH.MED_V2_1, SMESH.MED_V2_2
1882     #  @param opt boolean parameter for creating/not creating
1883     #         the groups Group_On_All_Nodes, Group_On_All_Faces, ...
1884     #  @param overwrite boolean parameter for overwriting/not overwriting the file
1885     #  @param autoDimension: if @c True (default), a space dimension of a MED mesh can be either
1886     #         - 1D if all mesh nodes lie on OX coordinate axis, or
1887     #         - 2D if all mesh nodes lie on XOY coordinate plane, or
1888     #         - 3D in the rest cases.
1889     #
1890     #         If @a autoDimension is @c False, the space dimension is always 3.
1891     #  @ingroup l2_impexp
1892     def ExportToMED(self, f, version, opt=0, overwrite=1, autoDimension=True):
1893         self.mesh.ExportToMEDX(f, opt, version, overwrite, autoDimension)
1894
1895     # Operations with groups:
1896     # ----------------------
1897
1898     ## Creates an empty mesh group
1899     #  @param elementType the type of elements in the group; either of 
1900     #         (SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME)
1901     #  @param name the name of the mesh group
1902     #  @return SMESH_Group
1903     #  @ingroup l2_grps_create
1904     def CreateEmptyGroup(self, elementType, name):
1905         return self.mesh.CreateGroup(elementType, name)
1906
1907     ## Creates a mesh group based on the geometric object \a grp
1908     #  and gives a \a name, \n if this parameter is not defined
1909     #  the name is the same as the geometric group name \n
1910     #  Note: Works like GroupOnGeom().
1911     #  @param grp  a geometric group, a vertex, an edge, a face or a solid
1912     #  @param name the name of the mesh group
1913     #  @return SMESH_GroupOnGeom
1914     #  @ingroup l2_grps_create
1915     def Group(self, grp, name=""):
1916         return self.GroupOnGeom(grp, name)
1917
1918     ## Creates a mesh group based on the geometrical object \a grp
1919     #  and gives a \a name, \n if this parameter is not defined
1920     #  the name is the same as the geometrical group name
1921     #  @param grp  a geometrical group, a vertex, an edge, a face or a solid
1922     #  @param name the name of the mesh group
1923     #  @param typ  the type of elements in the group; either of 
1924     #         (SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME). If not set, it is
1925     #         automatically detected by the type of the geometry
1926     #  @return SMESH_GroupOnGeom
1927     #  @ingroup l2_grps_create
1928     def GroupOnGeom(self, grp, name="", typ=None):
1929         AssureGeomPublished( self, grp, name )
1930         if name == "":
1931             name = grp.GetName()
1932         if not typ:
1933             typ = self._groupTypeFromShape( grp )
1934         return self.mesh.CreateGroupFromGEOM(typ, name, grp)
1935
1936     ## Pivate method to get a type of group on geometry
1937     def _groupTypeFromShape( self, shape ):
1938         tgeo = str(shape.GetShapeType())
1939         if tgeo == "VERTEX":
1940             typ = NODE
1941         elif tgeo == "EDGE":
1942             typ = EDGE
1943         elif tgeo == "FACE" or tgeo == "SHELL":
1944             typ = FACE
1945         elif tgeo == "SOLID" or tgeo == "COMPSOLID":
1946             typ = VOLUME
1947         elif tgeo == "COMPOUND":
1948             sub = self.geompyD.SubShapeAll( shape, self.geompyD.ShapeType["SHAPE"])
1949             if not sub:
1950                 raise ValueError,"_groupTypeFromShape(): empty geometric group or compound '%s'" % GetName(shape)
1951             return self._groupTypeFromShape( sub[0] )
1952         else:
1953             raise ValueError, \
1954                   "_groupTypeFromShape(): invalid geometry '%s'" % GetName(shape)
1955         return typ
1956
1957     ## Creates a mesh group with given \a name based on the \a filter which
1958     ## is a special type of group dynamically updating it's contents during
1959     ## mesh modification
1960     #  @param typ  the type of elements in the group; either of 
1961     #         (SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME).
1962     #  @param name the name of the mesh group
1963     #  @param filter the filter defining group contents
1964     #  @return SMESH_GroupOnFilter
1965     #  @ingroup l2_grps_create
1966     def GroupOnFilter(self, typ, name, filter):
1967         return self.mesh.CreateGroupFromFilter(typ, name, filter)
1968
1969     ## Creates a mesh group by the given ids of elements
1970     #  @param groupName the name of the mesh group
1971     #  @param elementType the type of elements in the group; either of 
1972     #         (SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME).
1973     #  @param elemIDs the list of ids
1974     #  @return SMESH_Group
1975     #  @ingroup l2_grps_create
1976     def MakeGroupByIds(self, groupName, elementType, elemIDs):
1977         group = self.mesh.CreateGroup(elementType, groupName)
1978         if hasattr( elemIDs, "GetIDs" ):
1979             if hasattr( elemIDs, "SetMesh" ):
1980                 elemIDs.SetMesh( self.GetMesh() )
1981             group.AddFrom( elemIDs )
1982         else:
1983             group.Add(elemIDs)
1984         return group
1985
1986     ## Creates a mesh group by the given conditions
1987     #  @param groupName the name of the mesh group
1988     #  @param elementType the type of elements(SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME)
1989     #  @param CritType the type of criterion (SMESH.FT_Taper, SMESH.FT_Area, etc.)
1990     #          Type SMESH.FunctorType._items in the Python Console to see all values.
1991     #          Note that the items starting from FT_LessThan are not suitable for CritType.
1992     #  @param Compare  belongs to {SMESH.FT_LessThan, SMESH.FT_MoreThan, SMESH.FT_EqualTo}
1993     #  @param Threshold the threshold value (range of ids as string, shape, numeric)
1994     #  @param UnaryOp  SMESH.FT_LogicalNOT or SMESH.FT_Undefined
1995     #  @param Tolerance the tolerance used by SMESH.FT_BelongToGeom, SMESH.FT_BelongToSurface,
1996     #         SMESH.FT_LyingOnGeom, SMESH.FT_CoplanarFaces criteria
1997     #  @return SMESH_GroupOnFilter
1998     #  @ingroup l2_grps_create
1999     def MakeGroup(self,
2000                   groupName,
2001                   elementType,
2002                   CritType=FT_Undefined,
2003                   Compare=FT_EqualTo,
2004                   Threshold="",
2005                   UnaryOp=FT_Undefined,
2006                   Tolerance=1e-07):
2007         aCriterion = self.smeshpyD.GetCriterion(elementType, CritType, Compare, Threshold, UnaryOp, FT_Undefined,Tolerance)
2008         group = self.MakeGroupByCriterion(groupName, aCriterion)
2009         return group
2010
2011     ## Creates a mesh group by the given criterion
2012     #  @param groupName the name of the mesh group
2013     #  @param Criterion the instance of Criterion class
2014     #  @return SMESH_GroupOnFilter
2015     #  @ingroup l2_grps_create
2016     def MakeGroupByCriterion(self, groupName, Criterion):
2017         return self.MakeGroupByCriteria( groupName, [Criterion] )
2018
2019     ## Creates a mesh group by the given criteria (list of criteria)
2020     #  @param groupName the name of the mesh group
2021     #  @param theCriteria the list of criteria
2022     #  @param binOp binary operator used when binary operator of criteria is undefined
2023     #  @return SMESH_GroupOnFilter
2024     #  @ingroup l2_grps_create
2025     def MakeGroupByCriteria(self, groupName, theCriteria, binOp=SMESH.FT_LogicalAND):
2026         aFilter = self.smeshpyD.GetFilterFromCriteria( theCriteria, binOp )
2027         group = self.MakeGroupByFilter(groupName, aFilter)
2028         return group
2029
2030     ## Creates a mesh group by the given filter
2031     #  @param groupName the name of the mesh group
2032     #  @param theFilter the instance of Filter class
2033     #  @return SMESH_GroupOnFilter
2034     #  @ingroup l2_grps_create
2035     def MakeGroupByFilter(self, groupName, theFilter):
2036         #group = self.CreateEmptyGroup(theFilter.GetElementType(), groupName)
2037         #theFilter.SetMesh( self.mesh )
2038         #group.AddFrom( theFilter )
2039         group = self.GroupOnFilter( theFilter.GetElementType(), groupName, theFilter )
2040         return group
2041
2042     ## Removes a group
2043     #  @ingroup l2_grps_delete
2044     def RemoveGroup(self, group):
2045         self.mesh.RemoveGroup(group)
2046
2047     ## Removes a group with its contents
2048     #  @ingroup l2_grps_delete
2049     def RemoveGroupWithContents(self, group):
2050         self.mesh.RemoveGroupWithContents(group)
2051
2052     ## Gets the list of groups existing in the mesh in the order
2053     #  of creation (starting from the oldest one)
2054     #  @param elemType type of elements the groups contain; either of 
2055     #         (SMESH.ALL, SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME);
2056     #         by default groups of elements of all types are returned
2057     #  @return a sequence of SMESH_GroupBase
2058     #  @ingroup l2_grps_create
2059     def GetGroups(self, elemType = SMESH.ALL):
2060         groups = self.mesh.GetGroups()
2061         if elemType == SMESH.ALL:
2062             return groups
2063         typedGroups = []
2064         for g in groups:
2065             if g.GetType() == elemType:
2066                 typedGroups.append( g )
2067                 pass
2068             pass
2069         return typedGroups
2070
2071     ## Gets the number of groups existing in the mesh
2072     #  @return the quantity of groups as an integer value
2073     #  @ingroup l2_grps_create
2074     def NbGroups(self):
2075         return self.mesh.NbGroups()
2076
2077     ## Gets the list of names of groups existing in the mesh
2078     #  @return list of strings
2079     #  @ingroup l2_grps_create
2080     def GetGroupNames(self):
2081         groups = self.GetGroups()
2082         names = []
2083         for group in groups:
2084             names.append(group.GetName())
2085         return names
2086
2087     ## Finds groups by name and type
2088     #  @param name name of the group of interest
2089     #  @param elemType type of elements the groups contain; either of 
2090     #         (SMESH.ALL, SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME);
2091     #         by default one group of any type of elements is returned
2092     #         if elemType == SMESH.ALL then all groups of any type are returned
2093     #  @return a list of SMESH_GroupBase's
2094     #  @ingroup l2_grps_create
2095     def GetGroupByName(self, name, elemType = None):
2096         groups = []
2097         for group in self.GetGroups():
2098             if group.GetName() == name:
2099                 if elemType is None:
2100                     return [group]
2101                 if ( elemType == SMESH.ALL or 
2102                      group.GetType() == elemType ):
2103                     groups.append( group )
2104         return groups
2105
2106     ## Produces a union of two groups.
2107     #  A new group is created. All mesh elements that are
2108     #  present in the initial groups are added to the new one
2109     #  @return an instance of SMESH_Group
2110     #  @ingroup l2_grps_operon
2111     def UnionGroups(self, group1, group2, name):
2112         return self.mesh.UnionGroups(group1, group2, name)
2113
2114     ## Produces a union list of groups.
2115     #  New group is created. All mesh elements that are present in
2116     #  initial groups are added to the new one
2117     #  @return an instance of SMESH_Group
2118     #  @ingroup l2_grps_operon
2119     def UnionListOfGroups(self, groups, name):
2120       return self.mesh.UnionListOfGroups(groups, name)
2121
2122     ## Prodices an intersection of two groups.
2123     #  A new group is created. All mesh elements that are common
2124     #  for the two initial groups are added to the new one.
2125     #  @return an instance of SMESH_Group
2126     #  @ingroup l2_grps_operon
2127     def IntersectGroups(self, group1, group2, name):
2128         return self.mesh.IntersectGroups(group1, group2, name)
2129
2130     ## Produces an intersection of groups.
2131     #  New group is created. All mesh elements that are present in all
2132     #  initial groups simultaneously are added to the new one
2133     #  @return an instance of SMESH_Group
2134     #  @ingroup l2_grps_operon
2135     def IntersectListOfGroups(self, groups, name):
2136       return self.mesh.IntersectListOfGroups(groups, name)
2137
2138     ## Produces a cut of two groups.
2139     #  A new group is created. All mesh elements that are present in
2140     #  the main group but are not present in the tool group are added to the new one
2141     #  @return an instance of SMESH_Group
2142     #  @ingroup l2_grps_operon
2143     def CutGroups(self, main_group, tool_group, name):
2144         return self.mesh.CutGroups(main_group, tool_group, name)
2145
2146     ## Produces a cut of groups.
2147     #  A new group is created. All mesh elements that are present in main groups
2148     #  but do not present in tool groups are added to the new one
2149     #  @return an instance of SMESH_Group
2150     #  @ingroup l2_grps_operon
2151     def CutListOfGroups(self, main_groups, tool_groups, name):
2152         return self.mesh.CutListOfGroups(main_groups, tool_groups, name)
2153
2154     ##
2155     #  Create a standalone group of entities basing on nodes of other groups.
2156     #  \param groups - list of groups, sub-meshes or filters, of any type.
2157     #  \param elemType - a type of elements to include to the new group; either of 
2158     #         (SMESH.NODE, SMESH.EDGE, SMESH.FACE, SMESH.VOLUME).
2159     #  \param name - a name of the new group.
2160     #  \param nbCommonNodes - a criterion of inclusion of an element to the new group
2161     #         basing on number of element nodes common with reference \a groups.
2162     #         Meaning of possible values are:
2163     #         - SMESH.ALL_NODES - include if all nodes are common,
2164     #         - SMESH.MAIN - include if all corner nodes are common (meaningful for a quadratic mesh),
2165     #         - SMESH.AT_LEAST_ONE - include if one or more node is common,
2166     #         - SMEHS.MAJORITY - include if half of nodes or more are common.
2167     #  \param underlyingOnly - if \c True (default), an element is included to the
2168     #         new group provided that it is based on nodes of one element of \a groups.
2169     #  @return an instance of SMESH_Group
2170     #  @ingroup l2_grps_operon
2171     def CreateDimGroup(self, groups, elemType, name,
2172                        nbCommonNodes = SMESH.ALL_NODES, underlyingOnly = True):
2173         if isinstance( groups, SMESH._objref_SMESH_IDSource ):
2174             groups = [groups]
2175         return self.mesh.CreateDimGroup(groups, elemType, name, nbCommonNodes, underlyingOnly)
2176
2177
2178     ## Convert group on geom into standalone group
2179     #  @ingroup l2_grps_delete
2180     def ConvertToStandalone(self, group):
2181         return self.mesh.ConvertToStandalone(group)
2182
2183     # Get some info about mesh:
2184     # ------------------------
2185
2186     ## Returns the log of nodes and elements added or removed
2187     #  since the previous clear of the log.
2188     #  @param clearAfterGet log is emptied after Get (safe if concurrents access)
2189     #  @return list of log_block structures:
2190     #                                        commandType
2191     #                                        number
2192     #                                        coords
2193     #                                        indexes
2194     #  @ingroup l1_auxiliary
2195     def GetLog(self, clearAfterGet):
2196         return self.mesh.GetLog(clearAfterGet)
2197
2198     ## Clears the log of nodes and elements added or removed since the previous
2199     #  clear. Must be used immediately after GetLog if clearAfterGet is false.
2200     #  @ingroup l1_auxiliary
2201     def ClearLog(self):
2202         self.mesh.ClearLog()
2203
2204     ## Toggles auto color mode on the object.
2205     #  @param theAutoColor the flag which toggles auto color mode.
2206     #  @ingroup l1_auxiliary
2207     def SetAutoColor(self, theAutoColor):
2208         self.mesh.SetAutoColor(theAutoColor)
2209
2210     ## Gets flag of object auto color mode.
2211     #  @return True or False
2212     #  @ingroup l1_auxiliary
2213     def GetAutoColor(self):
2214         return self.mesh.GetAutoColor()
2215
2216     ## Gets the internal ID
2217     #  @return integer value, which is the internal Id of the mesh
2218     #  @ingroup l1_auxiliary
2219     def GetId(self):
2220         return self.mesh.GetId()
2221
2222     ## Get the study Id
2223     #  @return integer value, which is the study Id of the mesh
2224     #  @ingroup l1_auxiliary
2225     def GetStudyId(self):
2226         return self.mesh.GetStudyId()
2227
2228     ## Checks the group names for duplications.
2229     #  Consider the maximum group name length stored in MED file.
2230     #  @return True or False
2231     #  @ingroup l1_auxiliary
2232     def HasDuplicatedGroupNamesMED(self):
2233         return self.mesh.HasDuplicatedGroupNamesMED()
2234
2235     ## Obtains the mesh editor tool
2236     #  @return an instance of SMESH_MeshEditor
2237     #  @ingroup l1_modifying
2238     def GetMeshEditor(self):
2239         return self.editor
2240
2241     ## Wrap a list of IDs of elements or nodes into SMESH_IDSource which
2242     #  can be passed as argument to a method accepting mesh, group or sub-mesh
2243     #  @param ids list of IDs
2244     #  @param elemType type of elements; this parameter is used to distinguish
2245     #         IDs of nodes from IDs of elements; by default ids are treated as
2246     #         IDs of elements; use SMESH.NODE if ids are IDs of nodes.
2247     #  @return an instance of SMESH_IDSource
2248     #  @warning call UnRegister() for the returned object as soon as it is no more useful:
2249     #          idSrc = mesh.GetIDSource( [1,3,5], SMESH.NODE )
2250     #          mesh.DoSomething( idSrc )
2251     #          idSrc.UnRegister()
2252     #  @ingroup l1_auxiliary
2253     def GetIDSource(self, ids, elemType = SMESH.ALL):
2254         return self.editor.MakeIDSource(ids, elemType)
2255
2256
2257     # Get informations about mesh contents:
2258     # ------------------------------------
2259
2260     ## Gets the mesh stattistic
2261     #  @return dictionary type element - count of elements
2262     #  @ingroup l1_meshinfo
2263     def GetMeshInfo(self, obj = None):
2264         if not obj: obj = self.mesh
2265         return self.smeshpyD.GetMeshInfo(obj)
2266
2267     ## Returns the number of nodes in the mesh
2268     #  @return an integer value
2269     #  @ingroup l1_meshinfo
2270     def NbNodes(self):
2271         return self.mesh.NbNodes()
2272
2273     ## Returns the number of elements in the mesh
2274     #  @return an integer value
2275     #  @ingroup l1_meshinfo
2276     def NbElements(self):
2277         return self.mesh.NbElements()
2278
2279     ## Returns the number of 0d elements in the mesh
2280     #  @return an integer value
2281     #  @ingroup l1_meshinfo
2282     def Nb0DElements(self):
2283         return self.mesh.Nb0DElements()
2284
2285     ## Returns the number of ball discrete elements in the mesh
2286     #  @return an integer value
2287     #  @ingroup l1_meshinfo
2288     def NbBalls(self):
2289         return self.mesh.NbBalls()
2290
2291     ## Returns the number of edges in the mesh
2292     #  @return an integer value
2293     #  @ingroup l1_meshinfo
2294     def NbEdges(self):
2295         return self.mesh.NbEdges()
2296
2297     ## Returns the number of edges with the given order in the mesh
2298     #  @param elementOrder the order of elements:
2299     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2300     #  @return an integer value
2301     #  @ingroup l1_meshinfo
2302     def NbEdgesOfOrder(self, elementOrder):
2303         return self.mesh.NbEdgesOfOrder(elementOrder)
2304
2305     ## Returns the number of faces in the mesh
2306     #  @return an integer value
2307     #  @ingroup l1_meshinfo
2308     def NbFaces(self):
2309         return self.mesh.NbFaces()
2310
2311     ## Returns the number of faces with the given order in the mesh
2312     #  @param elementOrder the order of elements:
2313     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2314     #  @return an integer value
2315     #  @ingroup l1_meshinfo
2316     def NbFacesOfOrder(self, elementOrder):
2317         return self.mesh.NbFacesOfOrder(elementOrder)
2318
2319     ## Returns the number of triangles in the mesh
2320     #  @return an integer value
2321     #  @ingroup l1_meshinfo
2322     def NbTriangles(self):
2323         return self.mesh.NbTriangles()
2324
2325     ## Returns the number of triangles with the given order in the mesh
2326     #  @param elementOrder is the order of elements:
2327     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2328     #  @return an integer value
2329     #  @ingroup l1_meshinfo
2330     def NbTrianglesOfOrder(self, elementOrder):
2331         return self.mesh.NbTrianglesOfOrder(elementOrder)
2332
2333     ## Returns the number of biquadratic triangles in the mesh
2334     #  @return an integer value
2335     #  @ingroup l1_meshinfo
2336     def NbBiQuadTriangles(self):
2337         return self.mesh.NbBiQuadTriangles()
2338
2339     ## Returns the number of quadrangles in the mesh
2340     #  @return an integer value
2341     #  @ingroup l1_meshinfo
2342     def NbQuadrangles(self):
2343         return self.mesh.NbQuadrangles()
2344
2345     ## Returns the number of quadrangles with the given order in the mesh
2346     #  @param elementOrder the order of elements:
2347     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2348     #  @return an integer value
2349     #  @ingroup l1_meshinfo
2350     def NbQuadranglesOfOrder(self, elementOrder):
2351         return self.mesh.NbQuadranglesOfOrder(elementOrder)
2352
2353     ## Returns the number of biquadratic quadrangles in the mesh
2354     #  @return an integer value
2355     #  @ingroup l1_meshinfo
2356     def NbBiQuadQuadrangles(self):
2357         return self.mesh.NbBiQuadQuadrangles()
2358
2359     ## Returns the number of polygons of given order in the mesh
2360     #  @param elementOrder the order of elements:
2361     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2362     #  @return an integer value
2363     #  @ingroup l1_meshinfo
2364     def NbPolygons(self, elementOrder = SMESH.ORDER_ANY):
2365         return self.mesh.NbPolygonsOfOrder(elementOrder)
2366
2367     ## Returns the number of volumes in the mesh
2368     #  @return an integer value
2369     #  @ingroup l1_meshinfo
2370     def NbVolumes(self):
2371         return self.mesh.NbVolumes()
2372
2373     ## Returns the number of volumes with the given order in the mesh
2374     #  @param elementOrder  the order of elements:
2375     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2376     #  @return an integer value
2377     #  @ingroup l1_meshinfo
2378     def NbVolumesOfOrder(self, elementOrder):
2379         return self.mesh.NbVolumesOfOrder(elementOrder)
2380
2381     ## Returns the number of tetrahedrons in the mesh
2382     #  @return an integer value
2383     #  @ingroup l1_meshinfo
2384     def NbTetras(self):
2385         return self.mesh.NbTetras()
2386
2387     ## Returns the number of tetrahedrons with the given order in the mesh
2388     #  @param elementOrder  the order of elements:
2389     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2390     #  @return an integer value
2391     #  @ingroup l1_meshinfo
2392     def NbTetrasOfOrder(self, elementOrder):
2393         return self.mesh.NbTetrasOfOrder(elementOrder)
2394
2395     ## Returns the number of hexahedrons in the mesh
2396     #  @return an integer value
2397     #  @ingroup l1_meshinfo
2398     def NbHexas(self):
2399         return self.mesh.NbHexas()
2400
2401     ## Returns the number of hexahedrons with the given order in the mesh
2402     #  @param elementOrder  the order of elements:
2403     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2404     #  @return an integer value
2405     #  @ingroup l1_meshinfo
2406     def NbHexasOfOrder(self, elementOrder):
2407         return self.mesh.NbHexasOfOrder(elementOrder)
2408
2409     ## Returns the number of triquadratic hexahedrons in the mesh
2410     #  @return an integer value
2411     #  @ingroup l1_meshinfo
2412     def NbTriQuadraticHexas(self):
2413         return self.mesh.NbTriQuadraticHexas()
2414
2415     ## Returns the number of pyramids in the mesh
2416     #  @return an integer value
2417     #  @ingroup l1_meshinfo
2418     def NbPyramids(self):
2419         return self.mesh.NbPyramids()
2420
2421     ## Returns the number of pyramids with the given order in the mesh
2422     #  @param elementOrder  the order of elements:
2423     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2424     #  @return an integer value
2425     #  @ingroup l1_meshinfo
2426     def NbPyramidsOfOrder(self, elementOrder):
2427         return self.mesh.NbPyramidsOfOrder(elementOrder)
2428
2429     ## Returns the number of prisms in the mesh
2430     #  @return an integer value
2431     #  @ingroup l1_meshinfo
2432     def NbPrisms(self):
2433         return self.mesh.NbPrisms()
2434
2435     ## Returns the number of prisms with the given order in the mesh
2436     #  @param elementOrder  the order of elements:
2437     #         SMESH.ORDER_ANY, SMESH.ORDER_LINEAR or SMESH.ORDER_QUADRATIC
2438     #  @return an integer value
2439     #  @ingroup l1_meshinfo
2440     def NbPrismsOfOrder(self, elementOrder):
2441         return self.mesh.NbPrismsOfOrder(elementOrder)
2442
2443     ## Returns the number of hexagonal prisms in the mesh
2444     #  @return an integer value
2445     #  @ingroup l1_meshinfo
2446     def NbHexagonalPrisms(self):
2447         return self.mesh.NbHexagonalPrisms()
2448
2449     ## Returns the number of polyhedrons in the mesh
2450     #  @return an integer value
2451     #  @ingroup l1_meshinfo
2452     def NbPolyhedrons(self):
2453         return self.mesh.NbPolyhedrons()
2454
2455     ## Returns the number of submeshes in the mesh
2456     #  @return an integer value
2457     #  @ingroup l1_meshinfo
2458     def NbSubMesh(self):
2459         return self.mesh.NbSubMesh()
2460
2461     ## Returns the list of mesh elements IDs
2462     #  @return the list of integer values
2463     #  @ingroup l1_meshinfo
2464     def GetElementsId(self):
2465         return self.mesh.GetElementsId()
2466
2467     ## Returns the list of IDs of mesh elements with the given type
2468     #  @param elementType  the required type of elements, either of
2469     #         (SMESH.NODE, SMESH.EDGE, SMESH.FACE or SMESH.VOLUME)
2470     #  @return list of integer values
2471     #  @ingroup l1_meshinfo
2472     def GetElementsByType(self, elementType):
2473         return self.mesh.GetElementsByType(elementType)
2474
2475     ## Returns the list of mesh nodes IDs
2476     #  @return the list of integer values
2477     #  @ingroup l1_meshinfo
2478     def GetNodesId(self):
2479         return self.mesh.GetNodesId()
2480
2481     # Get the information about mesh elements:
2482     # ------------------------------------
2483
2484     ## Returns the type of mesh element
2485     #  @return the value from SMESH::ElementType enumeration
2486     #          Type SMESH.ElementType._items in the Python Console to see all possible values.
2487     #  @ingroup l1_meshinfo
2488     def GetElementType(self, id, iselem=True):
2489         return self.mesh.GetElementType(id, iselem)
2490
2491     ## Returns the geometric type of mesh element
2492     #  @return the value from SMESH::EntityType enumeration
2493     #          Type SMESH.EntityType._items in the Python Console to see all possible values.
2494     #  @ingroup l1_meshinfo
2495     def GetElementGeomType(self, id):
2496         return self.mesh.GetElementGeomType(id)
2497
2498     ## Returns the shape type of mesh element
2499     #  @return the value from SMESH::GeometryType enumeration.
2500     #          Type SMESH.GeometryType._items in the Python Console to see all possible values.
2501     #  @ingroup l1_meshinfo
2502     def GetElementShape(self, id):
2503         return self.mesh.GetElementShape(id)
2504
2505     ## Returns the list of submesh elements IDs
2506     #  @param Shape a geom object(sub-shape) IOR
2507     #         Shape must be the sub-shape of a ShapeToMesh()
2508     #  @return the list of integer values
2509     #  @ingroup l1_meshinfo
2510     def GetSubMeshElementsId(self, Shape):
2511         if isinstance( Shape, geomBuilder.GEOM._objref_GEOM_Object):
2512             ShapeID = self.geompyD.GetSubShapeID( self.geom, Shape )
2513         else:
2514             ShapeID = Shape
2515         return self.mesh.GetSubMeshElementsId(ShapeID)
2516
2517     ## Returns the list of submesh nodes IDs
2518     #  @param Shape a geom object(sub-shape) IOR
2519     #         Shape must be the sub-shape of a ShapeToMesh()
2520     #  @param all If true, gives all nodes of submesh elements, otherwise gives only submesh nodes
2521     #  @return the list of integer values
2522     #  @ingroup l1_meshinfo
2523     def GetSubMeshNodesId(self, Shape, all):
2524         if isinstance( Shape, geomBuilder.GEOM._objref_GEOM_Object):
2525             ShapeID = self.geompyD.GetSubShapeID( self.geom, Shape )
2526         else:
2527             ShapeID = Shape
2528         return self.mesh.GetSubMeshNodesId(ShapeID, all)
2529
2530     ## Returns type of elements on given shape
2531     #  @param Shape a geom object(sub-shape) IOR
2532     #         Shape must be a sub-shape of a ShapeToMesh()
2533     #  @return element type
2534     #  @ingroup l1_meshinfo
2535     def GetSubMeshElementType(self, Shape):
2536         if isinstance( Shape, geomBuilder.GEOM._objref_GEOM_Object):
2537             ShapeID = self.geompyD.GetSubShapeID( self.geom, Shape )
2538         else:
2539             ShapeID = Shape
2540         return self.mesh.GetSubMeshElementType(ShapeID)
2541
2542     ## Gets the mesh description
2543     #  @return string value
2544     #  @ingroup l1_meshinfo
2545     def Dump(self):
2546         return self.mesh.Dump()
2547
2548
2549     # Get the information about nodes and elements of a mesh by its IDs:
2550     # -----------------------------------------------------------
2551
2552     ## Gets XYZ coordinates of a node
2553     #  \n If there is no nodes for the given ID - returns an empty list
2554     #  @return a list of double precision values
2555     #  @ingroup l1_meshinfo
2556     def GetNodeXYZ(self, id):
2557         return self.mesh.GetNodeXYZ(id)
2558
2559     ## Returns list of IDs of inverse elements for the given node
2560     #  \n If there is no node for the given ID - returns an empty list
2561     #  @return a list of integer values
2562     #  @ingroup l1_meshinfo
2563     def GetNodeInverseElements(self, id):
2564         return self.mesh.GetNodeInverseElements(id)
2565
2566     ## @brief Returns the position of a node on the shape
2567     #  @return SMESH::NodePosition
2568     #  @ingroup l1_meshinfo
2569     def GetNodePosition(self,NodeID):
2570         return self.mesh.GetNodePosition(NodeID)
2571
2572     ## @brief Returns the position of an element on the shape
2573     #  @return SMESH::ElementPosition
2574     #  @ingroup l1_meshinfo
2575     def GetElementPosition(self,ElemID):
2576         return self.mesh.GetElementPosition(ElemID)
2577
2578     ## Returns the ID of the shape, on which the given node was generated.
2579     #  @return an integer value > 0 or -1 if there is no node for the given
2580     #          ID or the node is not assigned to any geometry
2581     #  @ingroup l1_meshinfo
2582     def GetShapeID(self, id):
2583         return self.mesh.GetShapeID(id)
2584
2585     ## Returns the ID of the shape, on which the given element was generated.
2586     #  @return an integer value > 0 or -1 if there is no element for the given
2587     #          ID or the element is not assigned to any geometry
2588     #  @ingroup l1_meshinfo
2589     def GetShapeIDForElem(self,id):
2590         return self.mesh.GetShapeIDForElem(id)
2591
2592     ## Returns the number of nodes of the given element
2593     #  @return an integer value > 0 or -1 if there is no element for the given ID
2594     #  @ingroup l1_meshinfo
2595     def GetElemNbNodes(self, id):
2596         return self.mesh.GetElemNbNodes(id)
2597
2598     ## Returns the node ID the given (zero based) index for the given element
2599     #  \n If there is no element for the given ID - returns -1
2600     #  \n If there is no node for the given index - returns -2
2601     #  @return an integer value
2602     #  @ingroup l1_meshinfo
2603     def GetElemNode(self, id, index):
2604         return self.mesh.GetElemNode(id, index)
2605
2606     ## Returns the IDs of nodes of the given element
2607     #  @return a list of integer values
2608     #  @ingroup l1_meshinfo
2609     def GetElemNodes(self, id):
2610         return self.mesh.GetElemNodes(id)
2611
2612     ## Returns true if the given node is the medium node in the given quadratic element
2613     #  @ingroup l1_meshinfo
2614     def IsMediumNode(self, elementID, nodeID):
2615         return self.mesh.IsMediumNode(elementID, nodeID)
2616
2617     ## Returns true if the given node is the medium node in one of quadratic elements
2618     #  @param nodeID ID of the node
2619     #  @param elementType  the type of elements to check a state of the node, either of
2620     #         (SMESH.ALL, SMESH.NODE, SMESH.EDGE, SMESH.FACE or SMESH.VOLUME)
2621     #  @ingroup l1_meshinfo
2622     def IsMediumNodeOfAnyElem(self, nodeID, elementType = SMESH.ALL ):
2623         return self.mesh.IsMediumNodeOfAnyElem(nodeID, elementType)
2624
2625     ## Returns the number of edges for the given element
2626     #  @ingroup l1_meshinfo
2627     def ElemNbEdges(self, id):
2628         return self.mesh.ElemNbEdges(id)
2629
2630     ## Returns the number of faces for the given element
2631     #  @ingroup l1_meshinfo
2632     def ElemNbFaces(self, id):
2633         return self.mesh.ElemNbFaces(id)
2634
2635     ## Returns nodes of given face (counted from zero) for given volumic element.
2636     #  @ingroup l1_meshinfo
2637     def GetElemFaceNodes(self,elemId, faceIndex):
2638         return self.mesh.GetElemFaceNodes(elemId, faceIndex)
2639
2640     ## Returns three components of normal of given mesh face
2641     #  (or an empty array in KO case)
2642     #  @ingroup l1_meshinfo
2643     def GetFaceNormal(self, faceId, normalized=False):
2644         return self.mesh.GetFaceNormal(faceId,normalized)
2645
2646     ## Returns an element based on all given nodes.
2647     #  @ingroup l1_meshinfo
2648     def FindElementByNodes(self,nodes):
2649         return self.mesh.FindElementByNodes(nodes)
2650
2651     ## Returns true if the given element is a polygon
2652     #  @ingroup l1_meshinfo
2653     def IsPoly(self, id):
2654         return self.mesh.IsPoly(id)
2655
2656     ## Returns true if the given element is quadratic
2657     #  @ingroup l1_meshinfo
2658     def IsQuadratic(self, id):
2659         return self.mesh.IsQuadratic(id)
2660
2661     ## Returns diameter of a ball discrete element or zero in case of an invalid \a id
2662     #  @ingroup l1_meshinfo
2663     def GetBallDiameter(self, id):
2664         return self.mesh.GetBallDiameter(id)
2665
2666     ## Returns XYZ coordinates of the barycenter of the given element
2667     #  \n If there is no element for the given ID - returns an empty list
2668     #  @return a list of three double values
2669     #  @ingroup l1_meshinfo
2670     def BaryCenter(self, id):
2671         return self.mesh.BaryCenter(id)
2672
2673     ## Passes mesh elements through the given filter and return IDs of fitting elements
2674     #  @param theFilter SMESH_Filter
2675     #  @return a list of ids
2676     #  @ingroup l1_controls
2677     def GetIdsFromFilter(self, theFilter):
2678         theFilter.SetMesh( self.mesh )
2679         return theFilter.GetIDs()
2680
2681     ## Verifies whether a 2D mesh element has free edges (edges connected to one face only)\n
2682     #  Returns a list of special structures (borders).
2683     #  @return a list of SMESH.FreeEdges.Border structure: edge id and ids of two its nodes.
2684     #  @ingroup l1_controls
2685     def GetFreeBorders(self):
2686         aFilterMgr = self.smeshpyD.CreateFilterManager()
2687         aPredicate = aFilterMgr.CreateFreeEdges()
2688         aPredicate.SetMesh(self.mesh)
2689         aBorders = aPredicate.GetBorders()
2690         aFilterMgr.UnRegister()
2691         return aBorders
2692
2693
2694     # Get mesh measurements information:
2695     # ------------------------------------
2696
2697     ## Get minimum distance between two nodes, elements or distance to the origin
2698     #  @param id1 first node/element id
2699     #  @param id2 second node/element id (if 0, distance from @a id1 to the origin is computed)
2700     #  @param isElem1 @c True if @a id1 is element id, @c False if it is node id
2701     #  @param isElem2 @c True if @a id2 is element id, @c False if it is node id
2702     #  @return minimum distance value
2703     #  @sa GetMinDistance()
2704     def MinDistance(self, id1, id2=0, isElem1=False, isElem2=False):
2705         aMeasure = self.GetMinDistance(id1, id2, isElem1, isElem2)
2706         return aMeasure.value
2707
2708     ## Get measure structure specifying minimum distance data between two objects
2709     #  @param id1 first node/element id
2710     #  @param id2 second node/element id (if 0, distance from @a id1 to the origin is computed)
2711     #  @param isElem1 @c True if @a id1 is element id, @c False if it is node id
2712     #  @param isElem2 @c True if @a id2 is element id, @c False if it is node id
2713     #  @return Measure structure
2714     #  @sa MinDistance()
2715     def GetMinDistance(self, id1, id2=0, isElem1=False, isElem2=False):
2716         if isElem1:
2717             id1 = self.editor.MakeIDSource([id1], SMESH.FACE)
2718         else:
2719             id1 = self.editor.MakeIDSource([id1], SMESH.NODE)
2720         if id2 != 0:
2721             if isElem2:
2722                 id2 = self.editor.MakeIDSource([id2], SMESH.FACE)
2723             else:
2724                 id2 = self.editor.MakeIDSource([id2], SMESH.NODE)
2725             pass
2726         else:
2727             id2 = None
2728
2729         aMeasurements = self.smeshpyD.CreateMeasurements()
2730         aMeasure = aMeasurements.MinDistance(id1, id2)
2731         genObjUnRegister([aMeasurements,id1, id2])
2732         return aMeasure
2733
2734     ## Get bounding box of the specified object(s)
2735     #  @param objects single source object or list of source objects or list of nodes/elements IDs
2736     #  @param isElem if @a objects is a list of IDs, @c True value in this parameters specifies that @a objects are elements,
2737     #  @c False specifies that @a objects are nodes
2738     #  @return tuple of six values (minX, minY, minZ, maxX, maxY, maxZ)
2739     #  @sa GetBoundingBox()
2740     def BoundingBox(self, objects=None, isElem=False):
2741         result = self.GetBoundingBox(objects, isElem)
2742         if result is None:
2743             result = (0.0,)*6
2744         else:
2745             result = (result.minX, result.minY, result.minZ, result.maxX, result.maxY, result.maxZ)
2746         return result
2747
2748     ## Get measure structure specifying bounding box data of the specified object(s)
2749     #  @param IDs single source object or list of source objects or list of nodes/elements IDs
2750     #  @param isElem if @a IDs is a list of IDs, @c True value in this parameters specifies that @a objects are elements,
2751     #  @c False specifies that @a objects are nodes
2752     #  @return Measure structure
2753     #  @sa BoundingBox()
2754     def GetBoundingBox(self, IDs=None, isElem=False):
2755         if IDs is None:
2756             IDs = [self.mesh]
2757         elif isinstance(IDs, tuple):
2758             IDs = list(IDs)
2759         if not isinstance(IDs, list):
2760             IDs = [IDs]
2761         if len(IDs) > 0 and isinstance(IDs[0], int):
2762             IDs = [IDs]
2763         srclist = []
2764         unRegister = genObjUnRegister()
2765         for o in IDs:
2766             if isinstance(o, Mesh):
2767                 srclist.append(o.mesh)
2768             elif hasattr(o, "_narrow"):
2769                 src = o._narrow(SMESH.SMESH_IDSource)
2770                 if src: srclist.append(src)
2771                 pass
2772             elif isinstance(o, list):
2773                 if isElem:
2774                     srclist.append(self.editor.MakeIDSource(o, SMESH.FACE))
2775                 else:
2776                     srclist.append(self.editor.MakeIDSource(o, SMESH.NODE))
2777                 unRegister.set( srclist[-1] )
2778                 pass
2779             pass
2780         aMeasurements = self.smeshpyD.CreateMeasurements()
2781         unRegister.set( aMeasurements )
2782         aMeasure = aMeasurements.BoundingBox(srclist)
2783         return aMeasure
2784
2785     # Mesh edition (SMESH_MeshEditor functionality):
2786     # ---------------------------------------------
2787
2788     ## Removes the elements from the mesh by ids
2789     #  @param IDsOfElements is a list of ids of elements to remove
2790     #  @return True or False
2791     #  @ingroup l2_modif_del
2792     def RemoveElements(self, IDsOfElements):
2793         return self.editor.RemoveElements(IDsOfElements)
2794
2795     ## Removes nodes from mesh by ids
2796     #  @param IDsOfNodes is a list of ids of nodes to remove
2797     #  @return True or False
2798     #  @ingroup l2_modif_del
2799     def RemoveNodes(self, IDsOfNodes):
2800         return self.editor.RemoveNodes(IDsOfNodes)
2801
2802     ## Removes all orphan (free) nodes from mesh
2803     #  @return number of the removed nodes
2804     #  @ingroup l2_modif_del
2805     def RemoveOrphanNodes(self):
2806         return self.editor.RemoveOrphanNodes()
2807
2808     ## Add a node to the mesh by coordinates
2809     #  @return Id of the new node
2810     #  @ingroup l2_modif_add
2811     def AddNode(self, x, y, z):
2812         x,y,z,Parameters,hasVars = ParseParameters(x,y,z)
2813         if hasVars: self.mesh.SetParameters(Parameters)
2814         return self.editor.AddNode( x, y, z)
2815
2816     ## Creates a 0D element on a node with given number.
2817     #  @param IDOfNode the ID of node for creation of the element.
2818     #  @return the Id of the new 0D element
2819     #  @ingroup l2_modif_add
2820     def Add0DElement(self, IDOfNode):
2821         return self.editor.Add0DElement(IDOfNode)
2822
2823     ## Create 0D elements on all nodes of the given elements except those 
2824     #  nodes on which a 0D element already exists.
2825     #  @param theObject an object on whose nodes 0D elements will be created.
2826     #         It can be mesh, sub-mesh, group, list of element IDs or a holder
2827     #         of nodes IDs created by calling mesh.GetIDSource( nodes, SMESH.NODE )
2828     #  @param theGroupName optional name of a group to add 0D elements created
2829     #         and/or found on nodes of \a theObject.
2830     #  @return an object (a new group or a temporary SMESH_IDSource) holding
2831     #          IDs of new and/or found 0D elements. IDs of 0D elements 
2832     #          can be retrieved from the returned object by calling GetIDs()
2833     #  @ingroup l2_modif_add
2834     def Add0DElementsToAllNodes(self, theObject, theGroupName=""):
2835         unRegister = genObjUnRegister()
2836         if isinstance( theObject, Mesh ):
2837             theObject = theObject.GetMesh()
2838         if isinstance( theObject, list ):
2839             theObject = self.GetIDSource( theObject, SMESH.ALL )
2840             unRegister.set( theObject )
2841         return self.editor.Create0DElementsOnAllNodes( theObject, theGroupName )
2842
2843     ## Creates a ball element on a node with given ID.
2844     #  @param IDOfNode the ID of node for creation of the element.
2845     #  @param diameter the bal diameter.
2846     #  @return the Id of the new ball element
2847     #  @ingroup l2_modif_add
2848     def AddBall(self, IDOfNode, diameter):
2849         return self.editor.AddBall( IDOfNode, diameter )
2850
2851     ## Creates a linear or quadratic edge (this is determined
2852     #  by the number of given nodes).
2853     #  @param IDsOfNodes the list of node IDs for creation of the element.
2854     #  The order of nodes in this list should correspond to the description
2855     #  of MED. \n This description is located by the following link:
2856     #  http://www.code-aster.org/outils/med/html/modele_de_donnees.html#3.
2857     #  @return the Id of the new edge
2858     #  @ingroup l2_modif_add
2859     def AddEdge(self, IDsOfNodes):
2860         return self.editor.AddEdge(IDsOfNodes)
2861
2862     ## Creates a linear or quadratic face (this is determined
2863     #  by the number of given nodes).
2864     #  @param IDsOfNodes the list of node IDs for creation of the element.
2865     #  The order of nodes in this list should correspond to the description
2866     #  of MED. \n This description is located by the following link:
2867     #  http://www.code-aster.org/outils/med/html/modele_de_donnees.html#3.
2868     #  @return the Id of the new face
2869     #  @ingroup l2_modif_add
2870     def AddFace(self, IDsOfNodes):
2871         return self.editor.AddFace(IDsOfNodes)
2872
2873     ## Adds a polygonal face to the mesh by the list of node IDs
2874     #  @param IdsOfNodes the list of node IDs for creation of the element.
2875     #  @return the Id of the new face
2876     #  @ingroup l2_modif_add
2877     def AddPolygonalFace(self, IdsOfNodes):
2878         return self.editor.AddPolygonalFace(IdsOfNodes)
2879
2880     ## Adds a quadratic polygonal face to the mesh by the list of node IDs
2881     #  @param IdsOfNodes the list of node IDs for creation of the element;
2882     #         corner nodes follow first.
2883     #  @return the Id of the new face
2884     #  @ingroup l2_modif_add
2885     def AddQuadPolygonalFace(self, IdsOfNodes):
2886         return self.editor.AddQuadPolygonalFace(IdsOfNodes)
2887
2888     ## Creates both simple and quadratic volume (this is determined
2889     #  by the number of given nodes).
2890     #  @param IDsOfNodes the list of node IDs for creation of the element.
2891     #  The order of nodes in this list should correspond to the description
2892     #  of MED. \n This description is located by the following link:
2893     #  http://www.code-aster.org/outils/med/html/modele_de_donnees.html#3.