/[escript]/trunk/pyvisi/py_src/arrows.py
ViewVC logotype

Contents of /trunk/pyvisi/py_src/arrows.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 827 - (show annotations)
Wed Aug 30 00:08:02 2006 UTC (14 years, 2 months ago) by jongui
File MIME type: text/x-python
File size: 1901 byte(s)
Added:
contour.py and arrows.py


1 """
2 class that shows a vector field by arrows
3
4 @var __author__: name of author
5 @var __license__: licence agreement
6 @var __copyright__: copyrights
7 @var __url__: url entry point on documentation
8 @var __version__: version
9 @var __date__: date of the version
10 """
11 __copyright__=""" Copyright (c) 2006 by ACcESS MNRF
12 http://www.access.edu.au
13 Primary Business: Queensland, Australia"""
14 __license__="""Licensed under the Open Software License version 3.0
15 http://www.opensource.org/licenses/osl-3.0.php"""
16 __author__="Paul Cochrane, L. Gross"
17 __url__="http://www.iservo.edu.au/esys"
18 __version__="$Revision:$"
19 __date__="$Date:$"
20
21 import vtk
22
23 class Arrows:
24
25 def __init__(self, open_scene, data_collector):
26 self.open_scene = open_scene
27 self.data_collector = data_collector
28 self.vtk_glyph = None
29 self.vtk_arrows_mapper = None
30 self.vtk_arrows_actor = None
31
32 self.setArrows()
33 self.setMapper()
34 self.setActor()
35
36 # set up the glyph and use arrows as the source
37 def setArrows(self):
38 vtk_arrows = vtk.vtkArrowSource()
39
40 self.vtk_glyph = vtk.vtkGlyph3D()
41 self.vtk_glyph.SetInput(self.data_collector.getReader().GetOutput())
42 self.vtk_glyph.SetSource(vtk_arrows.GetOutput())
43 self.vtk_glyph.SetVectorModeToUseVector()
44 self.vtk_glyph.SetScaleModeToScaleByVector()
45 self.vtk_glyph.SetColorModeToColorByScalar()
46 self.vtk_glyph.SetScaleFactor(0.2)
47
48 # set up the mapper and data
49 def setMapper(self):
50 self.vtk_arrows_mapper = vtk.vtkPolyDataMapper()
51 self.vtk_arrows_mapper.SetInput(
52 self.vtk_glyph.GetOutput())
53
54 # set up the actor and add the actor to the scene
55 def setActor(self):
56 self.vtk_arrows_actor = vtk.vtkActor()
57 self.vtk_arrows_actor.SetMapper(self.vtk_arrows_mapper)
58
59 self.open_scene.getRenderer().AddActor(self.vtk_arrows_actor)
60
61
62
63
64 #class ArrowsOnPlane:
65 """
66 shows a vector field by arrows on a plane
67 """
68 pass

  ViewVC Help
Powered by ViewVC 1.1.26