#!/usr/bin/env python

# This simple example shows how to render a very large image (i.e.
# one that cannot fit on the screen) with a gradient in the background.

import vtk
from vtk.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
VTK_DATA_ROOT = '/Users/nicole/Testing/VTKData'

# We'll import some data to start. Since we are using an importer,
# we've got to give it a render window and such. Note that the render
# window size is set fairly small.
ren = vtk.vtkRenderer()

# set up a gradient blue background
backgroundColor = [0.7568627450980392, 0.7647058823529412, 0.9098039215686275]
backgroundColor2 = [0.4549019607843137, 0.4705882352941176, 0.7450980392156863]
ren.SetBackground(backgroundColor)
ren.SetBackground2(backgroundColor2)
ren.SetGradientBackground(True)


renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
renWin.SetSize(125, 125)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)

importer = vtk.vtk3DSImporter()
importer.SetRenderWindow(renWin)
importer.SetFileName(VTK_DATA_ROOT + "/Data/Viewpoint/iflamigm.3ds")
importer.ComputeNormalsOn()
importer.Read()

# We'll set up the view we want.
ren.GetActiveCamera().SetPosition(0, 1, 0)
ren.GetActiveCamera().SetFocalPoint(0, 0, 0)
ren.GetActiveCamera().SetViewUp(0, 0, 1)

# Let the renderer compute a good position and focal point.
ren.ResetCamera()
ren.GetActiveCamera().Dolly(1.4)
ren.ResetCameraClippingRange()

renderLarge = vtk.vtkRenderLargeImage()
renderLarge.SetInput(ren)
renderLarge.SetMagnification(4)

# We write out the image which causes the rendering to occur. If you
# watch your screen you might see the pieces being rendered right
# after one another.
writer = vtk.vtkPNGWriter()
writer.SetInputConnection(renderLarge.GetOutputPort())
writer.SetFileName("largeImageGradient.png")
writer.Write()