pyvisi/py_src/scene.py

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

import vtk

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.renderer = renderer
        self.x_size = x_size
        self.y_size = y_size
        self.vtk_renderer = None
        self.vtk_render_window = None

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

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

        self.vtk_renderer = vtk.vtkRenderer()
        self.vtk_render_window = vtk.vtkRenderWindow()
        self.vtk_render_window.AddRenderer(self.vtk_renderer)
        self.vtk_render_window.SetSize(self.x_size, self.y_size)
        self.vtk_renderer.SetBackground(1, 1, 1) # Default color is white

    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
8	class Scene:
9	"""
10	Class that defines a scene in which components are displayed.
11	"""
12
13	def __init__(self, renderer = "vtk_online", x_size = 800, y_size = 600):
14	"""
15	Initialize all the instance variables.
16
17
18	@type renderer: String
19	@param renderer: Type of rendering (I{vtk-online or vtk-offline})
20	@type x_size: Number
21	@param x_size: Size of the rendering window on the x-axis
22	@type y_size: Number
23	@param y_size: Size of the rendering window on the y-axis
24	"""
25
26	self.renderer = renderer
27	self.x_size = x_size
28	self.y_size = y_size
29	self.vtk_renderer = None
30	self.vtk_render_window = None
31
32	if(renderer == "vtk_online"):
33	self.setRenderingWindow()
34
35	def setRenderingWindow(self):
36	"""
37	Set up the renderer and rendering window.
38	"""
39
40	self.vtk_renderer = vtk.vtkRenderer()
41	self.vtk_render_window = vtk.vtkRenderWindow()
42	self.vtk_render_window.AddRenderer(self.vtk_renderer)
43	self.vtk_render_window.SetSize(self.x_size, self.y_size)
44	self.vtk_renderer.SetBackground(1, 1, 1) # Default color is white
45
46	def render(self):
47	"""
48	Set up the render window interactor and render the image.
49	"""
50
51	vtk_render_window_interactor = vtk.vtkRenderWindowInteractor()
52	vtk_render_window_interactor.SetRenderWindow(self.vtk_render_window)
53	vtk_render_window_interactor.Initialize()
54	self.vtk_render_window.Render()
55	vtk_render_window_interactor.Start()
56
57	def getRenderer(self):
58	"""
59	Return the renderer.
60
61	@rtype: vtkRenderer
62	@return: VTK renderer
63	"""
64
65	return self.vtk_renderer
66
67
68
69
70
71
72
73