src/SMESH_SWIG/SMESH_test5.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2016  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 import salome
  28 salome.salome_init()
  29 import GEOM
  30 from salome.geom import geomBuilder
  31 geompy = geomBuilder.New()
  32
  33 import SMESH, SALOMEDS
  34 from salome.smesh import smeshBuilder
  35 smesh =  smeshBuilder.New()
  36
  37 import CORBA
  38 import os
  39 import os.path
  40
  41 def SetSObjName(theSObj,theName) :
  42     ok, anAttr = theSObj.FindAttribute("AttributeName")
  43     if ok:
  44         aName = anAttr._narrow(SALOMEDS.AttributeName)
  45         #print aName.__dict__
  46         aName.SetValue(theName)
  47
  48 def ConvertMED2UNV(thePath,theFile) :
  49     anInitFileName = thePath + theFile
  50     aMeshes,aResult = smesh.CreateMeshesFromMED(anInitFileName)
  51     print(aResult, aMeshes)
  52
  53     for iMesh in range(len(aMeshes)) :
  54         aMesh = aMeshes[iMesh]
  55         print(aMesh.GetName(), end=' ')
  56         aFileName = anInitFileName
  57         aFileName = os.path.basename(aFileName)
  58         aMesh.SetName(aFileName)
  59         print(aMesh.GetName())
  60
  61         aOutPath = '/tmp/'
  62         aFileName = aOutPath + theFile + "." + str(iMesh) + ".unv"
  63         aMesh.ExportUNV(aFileName)
  64         aMesh = smesh.CreateMeshesFromUNV(aFileName)
  65         print(aMesh.GetName(), end=' ')
  66         os.remove(aFileName)
  67         aFileName = os.path.basename(aFileName)
  68         aMesh.SetName(aFileName)
  69         print(aMesh.GetName())
  70
  71 aPath = os.getenv('DATA_DIR') + '/MedFiles/'
  72 aListDir = os.listdir(aPath)
  73 print(aListDir)
  74
  75 for iFile in range(len(aListDir)) :
  76     aFileName = aListDir[iFile]
  77     aName,anExt = os.path.splitext(aFileName)
  78     if anExt == ".med" :
  79         aFileName = os.path.basename(aFileName)
  80         print(aFileName)
  81         ConvertMED2UNV(aPath,aFileName)
  82         #break
  83
  84 salome.sg.updateObjBrowser()