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

Diff of /trunk/pyvisi/py_src/scene.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 851 by jongui, Fri Sep 15 07:11:55 2006 UTC revision 852 by jongui, Wed Sep 20 03:23:16 2006 UTC
# Line 11  class Scene: Line 11  class Scene:
11      Class that defines a scene in which components are displayed.      Class that defines a scene in which components are displayed.
12      """      """
13    
14      def __init__(self, renderer = "vtk_online", x_size = 800, y_size = 600):      def __init__(self, renderer, x_size = 800, y_size = 600):
15          """          """
16          Initialize all the instance variables.          Initialize all the instance variables.
17    
# Line 24  class Scene: Line 24  class Scene:
24          @param y_size: Size of the rendering window on the y-axis          @param y_size: Size of the rendering window on the y-axis
25          """          """
26    
27            self.renderer = renderer
28          self.x_size = x_size          self.x_size = x_size
29          self.y_size = y_size          self.y_size = y_size
30          self.vtk_renderer = vtk.vtkRenderer()          self.vtk_renderer = vtk.vtkRenderer()
31          self.vtk_render_window = vtk.vtkRenderWindow()          self.vtk_render_window = vtk.vtkRenderWindow()
32    
33          if(renderer == "vtk_online"):          self.setRenderWindow()
34              self.setRenderWindow()          
35        def saveImage(self, image_name):
36            self.vtk_render_window.Render()
37    
38            vtk_window_to_image = vtk.vtkWindowToImageFilter()
39            vtk_window_to_image.SetInput(self.vtk_render_window)
40            vtk_window_to_image.Update()
41    
42            vtk_image_writer = self.getImageWriter(self.renderer)
43            vtk_image_writer.SetInput(vtk_window_to_image.GetOutput())
44            vtk_image_writer.SetFileName(image_name)
45            vtk_image_writer.Write()
46    
47    
48      def setRenderWindow(self):      def setRenderWindow(self):
49          """          """
# Line 53  class Scene: Line 66  class Scene:
66          self.vtk_render_window.Render()          self.vtk_render_window.Render()
67          vtk_render_window_interactor.Start()          vtk_render_window_interactor.Start()
68    
69    
70        def getImageWriter(self, renderer):
71            if(renderer == "vtk_jpeg"):
72                return vtk.vtkJPEGWriter()
73            elif(renderer == "vtk_bmp"):
74                return vtk.vtkBMPWriter()
75            elif(renderer == "vtk_pnm"):
76                return vtk.vtkPNMWriter()
77            elif(renderer == "vtk_png"):
78                return vtk.vtkPNGWriter()
79            elif(renderer == "vtk_tiff"):
80                return vtk.vtkTIFFWriter()
81            elif(renderer == "vtk_ps"):
82                return vtk.vtkPostScriptWriter()
83    
84      def getRenderer(self):      def getRenderer(self):
85          """          """
86          Return the renderer.          Return the renderer.
Legend:
Removed from v.851  
changed lines
  Added in v.852
  ViewVC Help
Powered by ViewVC 1.1.26