Salome HOME
Porting to Python 2.6 - add coding page specification for Python scripts
[modules/smesh.git] / src / SMESH_SWIG / SMESH_BelongToGeom.py
1 #! /usr/bin/python
2 #  -*- coding: iso-8859-1 -*-
3 #  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
4 #
5 #  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
6 #  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
7 #
8 #  This library is free software; you can redistribute it and/or
9 #  modify it under the terms of the GNU Lesser General Public
10 #  License as published by the Free Software Foundation; either
11 #  version 2.1 of the License.
12 #
13 #  This library is distributed in the hope that it will be useful,
14 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
15 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16 #  Lesser General Public License for more details.
17 #
18 #  You should have received a copy of the GNU Lesser General Public
19 #  License along with this library; if not, write to the Free Software
20 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
21 #
22 #  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 #
24 from SMESH_test1 import *
25
26 ## Old style
27 def CheckBelongToGeomFilterOld(theMeshGen, theMesh, theShape, theSubShape, theElemType):
28     import geompy
29     if theShape != theSubShape:
30         aName = str(theSubShape)
31         geompy.addToStudyInFather(theShape,theSubShape,aName)
32
33     theMeshGen.Compute(theMesh,theShape)
34
35     aFilterMgr = theMeshGen.CreateFilterManager()
36     aFilter = aFilterMgr.CreateFilter()
37    
38     aBelongToGeom = aFilterMgr.CreateBelongToGeom()
39     aBelongToGeom.SetGeom(theSubShape)
40     aBelongToGeom.SetElementType(theElemType)
41     
42     aFilter.SetPredicate(aBelongToGeom)
43     return aFilter.GetElementsId(theMesh)
44
45 ## Current style
46 def CheckBelongToGeomFilter(theMesh, theShape, theSubShape, theElemType):
47     import geompy
48     import smesh
49     if theShape != theSubShape:
50         aName = str(theSubShape)
51         geompy.addToStudyInFather(theShape,theSubShape,aName)
52
53     theMesh.Compute()
54     aFilter = smesh.GetFilter(theElemType, smesh.FT_BelongToGeom, theSubShape)
55     return aFilter.GetElementsId(theMesh.GetMesh())
56     
57
58 anElemType = smesh.FACE;
59 print "anElemType =", anElemType
60 #anIds = CheckBelongToGeomFilter(mesh,box,subShapeList[1],anElemType)
61 anIds = CheckBelongToGeomFilter(mesh,box,box,anElemType)
62 print "Number of ids = ", len(anIds)
63 print "anIds = ", anIds
64 ## Check old version
65 #anIds = CheckBelongToGeomFilterOld(smesh.smesh,mesh.GetMesh(),box,box,anElemType)
66 #print "anIds = ", anIds
67
68 salome.sg.updateObjBrowser(1);