src/VISU_SWIG/visu_med.py

   1 import os
   2 import salome
   3 import SALOMEDS
   4 import SALOME_MED
   5 import VISU
   6
   7 from libSALOME_Swig import *
   8 sg = SALOMEGUI_Swig()
   9
  10 def getMedObjectFromStudy():
  11     mySO = salome.myStudy.FindObject("Objet MED")
  12     anAttr = mySO.FindAttribute("AttributeIOR")[1]
  13     obj = salome.orb.string_to_object(anAttr.Value())
  14     myObj = obj._narrow(SALOME_MED.MED)
  15     return myObj
  16
  17 def getFieldObjectFromStudy(number,subnumber):
  18     mySO = salome.myStudy.FindObject("MEDFIELD")
  19     mysub = mySO.FindSubObject(number)[1]
  20     if mysub:
  21         mysubsub = mysub.FindSubObject(subnumber)[1]
  22         if mysubsub:
  23             Builder = salome.myStudy.NewBuilder()
  24             anAttr = Builder.FindOrCreateAttribute(mysubsub, "AttributeIOR")
  25             obj = salome.orb.string_to_object(anAttr.Value())
  26             myObj = obj._narrow(SALOME_MED.FIELDINT)
  27             if (myObj == None):
  28                 myObj = obj._narrow(SALOME_MED.FIELDDOUBLE)
  29             return myObj
  30     else:
  31         print "ERROR: No Field Object stored in this Study"
  32         return None
  33
  34 med_comp = salome.lcc.FindOrLoadComponent("FactoryServer", "Med")
  35
  36 medDir = os.getenv('SALOME_ROOT_DIR') + '/../SALOME_ROOT/data/'
  37
  38 def importMedFrom(medDir,medFile):
  39     medFile = medDir + medFile
  40     med_comp.readStructFileWithFieldType(medFile,salome.myStudyName)
  41     sg.updateObjBrowser(1)
  42
  43 def importMed(medFile):
  44     importMedFrom(medDir,medFile)
  45
  46 #med_obj = getMedObjectFromStudy()
  47 myVisu = salome.lcc.FindOrLoadComponent("FactoryServer", "VISU")
  48 print "Use importMed(medFile) or importMedFrom(medDir,medFile) functions !"
  49
  50 #myField = getFieldObjectFromStudy(2,1)
  51 #myResult = myVisu.ImportMed(myField)
  52 #aMesh = myVisu.CreateMesh(myResult);
  53 #aScalarMap = myVisu.CreateScalarMap(myResult,myField.getName(),0)
  54 #if(myField.getNumberOfComponents() > 1) :
  55 #  aScalarMap = myVisu.CreateVectors(myResult,myField.getName(),0)
  56
  57 #myResult = myVisu.ImportFile(medFile)
  58 #aMesh = myVisu.CreateMesh(myResult);
  59 #aScalarMap = myVisu.CreateScalarMap(myResult,myField.getName(),0)
  60 #if(myField.getNumberOfComponents() > 1) :
  61 #  aScalarMap = myVisu.CreateCutPlanes(myResult,myField.getName(),0)