test/demo12.py

   1 # Copyright (C) 2010-2016  CEA/DEN, EDF R&D
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License, or (at your option) any later version.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19
  20
  21 if not ('servermanager' in dir()):
  22   from pvsimple import *
  23
  24 import os, inspect
  25
  26 def demo12(fname, impth):
  27     """This method demonstrates the Stream Tracer."""
  28
  29     reader = ExodusIIReader(FileName=fname)
  30     v=GetActiveView()
  31     streamTracer = StreamTracer(reader)
  32     streamTracer.Vectors  = 'V'
  33     streamTracer.SeedType.NumberOfPoints = 100
  34     streamTracer.SeedType.Radius = 1.15
  35     streamTracer.MaximumStreamlineLength = 25.
  36
  37     Show(streamTracer)
  38     camera = GetActiveCamera()
  39     camera.Elevation(-45)
  40     camera.Dolly(0.4)
  41     dp1 = GetDisplayProperties(streamTracer)
  42     streamTracer.PointData[:]
  43     ##[Array: Temp, Array: V, Array: Pres, Array: AsH3, Array: GaMe3, Array: CH4, Array: H2, Array: IntegrationTime, Array: Vorticity, Array: Rotation,     ##Array: AngularVelocity, Array: Normals]
  44
  45     pres = streamTracer.PointData[2]
  46     dp1.LookupTable = MakeBlueToRedLT(pres.GetRange()[0],pres.GetRange()[1])
  47     dp1.ColorAttributeType = 'POINT_DATA'
  48     dp1.ColorArrayName = 'Pres'
  49     Render()
  50     WriteImage(filename = (impth + "streamTracer_1.png"), view=v, Magnification=2)
  51
  52
  53 scriptdir = inspect.getframeinfo(inspect.currentframe())[0]
  54 testdir = os.path.dirname( os.path.abspath(scriptdir) )
  55 pvdata = os.getenv("SAMPLES_SRC_DIR")
  56
  57 if __name__ == "__main__":
  58     demo12(fname=pvdata+"/Data/disk_out_ref.ex2", impth=testdir+"/Pic/")