test/SMESH_test5.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 #  File   : SMESH_test5.py
  25 #  Module : SMESH
  26 #
  27
  28 import salome
  29 salome.salome_init()
  30 import GEOM
  31 from salome.geom import geomBuilder
  32 geompy = geomBuilder.New()
  33
  34 import SMESH, SALOMEDS
  35 from salome.smesh import smeshBuilder
  36 smesh =  smeshBuilder.New()
  37
  38 import CORBA
  39 import os
  40 import shutil
  41 import tempfile
  42
  43 aOutPath = tempfile.mkdtemp()
  44
  45 def SetSObjName(theSObj,theName) :
  46     ok, anAttr = theSObj.FindAttribute("AttributeName")
  47     if ok:
  48         aName = anAttr._narrow(SALOMEDS.AttributeName)
  49         #print aName.__dict__
  50         aName.SetValue(theName)
  51
  52 def ConvertMED2UNV(theFile):
  53     print(theFile)
  54     aMeshes, aResult = smesh.CreateMeshesFromMED(theFile)
  55     print(aMeshes, aResult)
  56
  57     for aMesh in aMeshes:
  58         print(aMesh.GetName(), end=' ')
  59         aMesh.SetName(os.path.basename(theFile))
  60         print(aMesh.GetName())
  61
  62         aFileName = os.path.join(aOutPath, theFile + ".unv")
  63         aMesh.ExportUNV(aFileName)
  64
  65         aMesh = smesh.CreateMeshesFromUNV(aFileName)
  66         print(aMesh.GetName(), end=' ')
  67         aMesh.SetName(os.path.basename(aFileName))
  68         print(aMesh.GetName())
  69
  70 aPath = os.path.join(os.getenv('DATA_DIR'), 'MedFiles')
  71
  72 for aFileName in sorted(os.listdir(aPath)):
  73     if os.path.splitext(aFileName)[-1] == ".med":
  74         ConvertMED2UNV(os.path.join(aPath, aFileName))
  75
  76 if os.getenv('SMESH_KEEP_TMP_DIR') != '1':
  77     shutil.rmtree(aOutPath)
  78
  79 salome.sg.updateObjBrowser()