pyvisi/py_src/scene.py

"""
@author: John Ngui
@author: Lutz Gross
""" 

import vtk
from constants import *

class Scene:
    """
    Class that defines a scene in which components are displayed.
    """

    def __init__(self, renderer = "vtk_online", x_size = 800, y_size = 600):
        """
        Initialize all the instance variables. 


        @type renderer: String
        @param renderer: Type of rendering (I{vtk-online or vtk-offline})       
        @type x_size: Number
        @param x_size: Size of the rendering window on the x-axis
        @type y_size: Number
        @param y_size: Size of the rendering window on the y-axis
        """ 

        self.x_size = x_size
        self.y_size = y_size
        self.vtk_renderer = vtk.vtkRenderer() 
        self.vtk_render_window = vtk.vtkRenderWindow() 

        if(renderer == "vtk_online"):
            self.setRenderWindow()

    def setRenderWindow(self):
        """
        Set up the renderer and rendering window.
        """

        self.vtk_render_window.AddRenderer(self.vtk_renderer)
        self.vtk_render_window.SetSize(self.x_size, self.y_size)
        self.vtk_renderer.SetBackground(
            WHITE[0], WHITE[1], WHITE[2]) 

    def render(self):
        """ 
        Set up the render window interactor and render the image.
        """

        vtk_render_window_interactor = vtk.vtkRenderWindowInteractor()
        vtk_render_window_interactor.SetRenderWindow(self.vtk_render_window)
        vtk_render_window_interactor.Initialize()
        self.vtk_render_window.Render()
        vtk_render_window_interactor.Start()

    def getRenderer(self):
        """
        Return the renderer.

        @rtype: vtkRenderer
        @return: VTK renderer 
        """

        return self.vtk_renderer
1	"""
2	@author: John Ngui
3	@author: Lutz Gross
4	"""
5
6	import vtk
7	from constants import *
8
9	class Scene:
10	"""
11	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):
15	"""
16	Initialize all the instance variables.
17
18
19	@type renderer: String
20	@param renderer: Type of rendering (I{vtk-online or vtk-offline})
21	@type x_size: Number
22	@param x_size: Size of the rendering window on the x-axis
23	@type y_size: Number
24	@param y_size: Size of the rendering window on the y-axis
25	"""
26
27	self.x_size = x_size
28	self.y_size = y_size
29	self.vtk_renderer = vtk.vtkRenderer()
30	self.vtk_render_window = vtk.vtkRenderWindow()
31
32	if(renderer == "vtk_online"):
33	self.setRenderWindow()
34
35	def setRenderWindow(self):
36	"""
37	Set up the renderer and rendering window.
38	"""
39
40	self.vtk_render_window.AddRenderer(self.vtk_renderer)
41	self.vtk_render_window.SetSize(self.x_size, self.y_size)
42	self.vtk_renderer.SetBackground(
43	WHITE[0], WHITE[1], WHITE[2])
44
45	def render(self):
46	"""
47	Set up the render window interactor and render the image.
48	"""
49
50	vtk_render_window_interactor = vtk.vtkRenderWindowInteractor()
51	vtk_render_window_interactor.SetRenderWindow(self.vtk_render_window)
52	vtk_render_window_interactor.Initialize()
53	self.vtk_render_window.Render()
54	vtk_render_window_interactor.Start()
55
56	def getRenderer(self):
57	"""
58	Return the renderer.
59
60	@rtype: vtkRenderer
61	@return: VTK renderer
62	"""
63
64	return self.vtk_renderer