test/SMESH_BelongToGeom.py

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