src/VISU_SWIG/VISU_Example_01.py

   1 # T 2.23, 28: Displaying fields (scalar and vectorial) on nodes, change presentation options.
   2 # Uses MED files ResOK_0000.med and Fields_group3D.med
   3 #
   4 # This script is equivalent to non-regression test script 003/A1
   5
   6 import salome
   7 import visu_gui
   8 import SALOMEDS
   9 import VISU
  10 import os
  11
  12 datadir = os.getenv("DATA_DIR") + "/MedFiles/"
  13
  14 #%====================Stage1: Creating a new study====================%
  15
  16 print "**** Stage1: Creating a new study"
  17
  18 print "Creating a new study..................",
  19 myVisu = visu_gui.myVisu
  20 myVisu.SetCurrentStudy(salome.myStudy)
  21 myViewManager = myVisu.GetViewManager()
  22 if myViewManager is None : print "Error"
  23 else : print "OK"
  24
  25 #%====================Stage2: Importing MED file====================%
  26
  27 print "**** Stage2: Importing MED file"
  28
  29 print 'Import "ResOK_0000.med"...............',
  30 medFile = datadir + "ResOK_0000.med"
  31 myResult = myVisu.ImportFile(medFile)
  32 if myResult is None : print "Error"
  33 else : print "OK"
  34
  35 print 'Creating new View3D...................',
  36 myView = myViewManager.Create3DView()
  37 if myView is None : print "Error"
  38 else : print "OK"
  39
  40 myMeshName = 'dom'
  41 myCellEntity = VISU.CELL
  42 myNodeEntity = VISU.NODE
  43
  44 #%====================Stage3: Displaying vector field====================%
  45
  46 print "**** Stage3: Displaying vector field"
  47
  48 print "Creating Scalar Map.......",
  49 scalarmap = myVisu.ScalarMapOnField(myResult,myMeshName,myNodeEntity,'vitesse',1);
  50 if scalarmap is None : print "Error"
  51 else : print "OK"
  52 scalarmap.SetSize(0.15, 0.8)
  53 myView.DisplayOnly(scalarmap)
  54 myView.FitAll()
  55
  56 print "Creating Stream Lines.....",
  57 streamlines = myVisu.StreamLinesOnField(myResult,myMeshName,myNodeEntity,'vitesse',1);
  58 if streamlines is None : print "Error"
  59 else : print "OK"
  60 myView.DisplayOnly(streamlines)
  61
  62 print "Creating Vectors..........",
  63 vectors = myVisu.VectorsOnField(myResult,myMeshName,myNodeEntity,'vitesse',1);
  64 if vectors is None : print "Error"
  65 else : print "OK"
  66 myView.DisplayOnly(vectors)
  67
  68
  69 print "Creating Iso Surfaces.....",
  70 isosurfaces = myVisu.IsoSurfacesOnField(myResult,myMeshName,myNodeEntity,'vitesse',1);
  71 if isosurfaces is None : print "Error"
  72 else : print "OK"
  73 myView.DisplayOnly(isosurfaces)
  74
  75 print "Creating Cut Planes.......",
  76 cutplanes = myVisu.CutPlanesOnField(myResult,myMeshName,myNodeEntity,'vitesse',1);
  77 if cutplanes is None : print "Error"
  78 else : print "OK"
  79 cutplanes.SetNbPlanes(30)
  80 cutplanes.SetOrientation(VISU.CutPlanes.YZ, 0, 0)
  81 myView.DisplayOnly(cutplanes)
  82
  83 #%====================Stage4: Opening a new study and Med file import====================%
  84
  85 print "**** Stage4: Opening a new study and Med file import"
  86
  87 print "Creating a new study..................",
  88 newStudy = salome.myStudyManager.NewStudy('newStudy')
  89 myVisu.SetCurrentStudy(newStudy)
  90 myViewManager = myVisu.GetViewManager()
  91 myView = myViewManager.Create3DView()
  92 if myView is None : print "Error"
  93 else : print "OK"
  94
  95 print 'Importing "Fields_group3D.med"........',
  96 medFile = datadir + "Fields_group3D.med"
  97 myResult1 = myVisu.ImportFile(medFile)
  98 if myResult1 is None : print "Error"
  99 myView1 = myViewManager.Create3DView()
 100 if myView1 is None : print "Error"
 101 else : print "OK"
 102
 103 #%====================Stage5: Displaying scalar field====================%
 104
 105 print "**** Stage5: Displaying scalar field"
 106
 107 myMeshName1 = 'mailles_MED'
 108
 109 print "Creating Scalar Map.......",
 110 scalarmap1 = myVisu.ScalarMapOnField(myResult1,myMeshName1,myCellEntity,'scalar field',1);
 111 if scalarmap1 is None : print "Error"
 112 else : print "OK"
 113 myView1.DisplayOnly(scalarmap1)
 114 myView1.FitAll()
 115
 116 print "Creating Iso Surfaces.....",
 117 isosurfaces1 = myVisu.IsoSurfacesOnField(myResult1,myMeshName1,myCellEntity,'scalar field',1);
 118 if isosurfaces1 is None : print "Error"
 119 else : print "OK"
 120 myView1.DisplayOnly(isosurfaces1)
 121
 122 print "Creating Cut Planes.......",
 123 cutplanes1 = myVisu.CutPlanesOnField(myResult1,myMeshName1,myCellEntity,'scalar field',1);
 124 if cutplanes1 is None : print "Error"
 125 else : print "OK"
 126 cutplanes1.SetOrientation(VISU.CutPlanes.YZ, 0, 0)
 127 myView1.DisplayOnly(cutplanes1)
 128
 129 #%====================Stage6: Object browser popup====================%
 130
 131 print "**** Stage6: Object browser popup"
 132
 133 print "Creating mesh.............",
 134 mesh = myVisu.MeshOnEntity(myResult1,myMeshName1,myCellEntity);
 135 if mesh is None : print "Error"
 136 else : print "OK"
 137 myView1.DisplayOnly(mesh)
 138
 139
 140 print "Changing type of presentation of mesh:"
 141 mesh.SetPresentationType(VISU.WIREFRAME)
 142 PrsType = mesh.GetPresentationType()
 143 print "Presentation type..", PrsType
 144 myView1.DisplayOnly(mesh)
 145
 146 mesh.SetPresentationType(VISU.SHADED)
 147 PrsType = mesh.GetPresentationType()
 148 print "Presentation type.....", PrsType
 149 myView1.DisplayOnly(mesh)
 150
 151 mesh.SetPresentationType(VISU.POINT)
 152 PrsType = mesh.GetPresentationType()
 153 print "Presentation type......", PrsType
 154 myView1.DisplayOnly(mesh)
 155 myView1.Update()
 156
 157 mesh.SetPresentationType(VISU.SHRINK)
 158 PrsType = mesh.GetPresentationType()
 159 print "Presentation type.....", PrsType
 160 myView1.DisplayOnly(mesh)
 161
 162 print "Changing color of mesh....",
 163 aColor = SALOMEDS.Color(0,0,1)
 164 mesh.SetCellColor(aColor)
 165 myView1.DisplayOnly(mesh)
 166 print "OK"
 167
 168 print "Renaming ScalarMap........",
 169 SObj = newStudy.FindObjectIOR(scalarmap1.GetID())
 170 newName = 'Renamed Object'
 171 SObj.Name = newName
 172 print "OK"
 173
 174 print "Deleting Cut Planes.......",
 175 SObj = newStudy.FindObjectIOR(cutplanes1.GetID())
 176 myBuilder = newStudy.NewBuilder()
 177 myBuilder.RemoveObject(SObj)
 178 print "OK"
 179
 180 print "Changing first IsoSurfaces",
 181 myVisu.SetCurrentStudy(salome.myStudy)
 182 myView.DisplayOnly(isosurfaces)
 183 isosurfaces.SetNbSurfaces(25)
 184 print "OK"
 185
 186 myView.Maximize()
 187 myView.DisplayOnly(isosurfaces)
 188
 189 print "Hide IsoSurfaces..........",
 190 myView.Erase(isosurfaces)
 191 print "OK"