from vtk import *
import os.path

dataRootDir = "/usr/share/vtkdata-6.0.0/Data/Infovis/XML/"
if not os.path.exists(dataRootDir):
  dataRootDir = "..//usr/share/vtkdata-6.0.0/Data/Infovis/XML/"


reader1 = vtkXMLTreeReader()
reader1.SetFileName(dataRootDir + "vtklibrary.xml")
reader1.Update()

numeric = vtkStringToNumeric()
numeric.SetInputConnection(reader1.GetOutputPort())

view = vtkTreeMapView()
view.SetAreaSizeArrayName("size");
view.SetAreaColorArrayName("level");
view.SetAreaLabelArrayName("name");
view.SetAreaLabelVisibility(True);
view.SetAreaHoverArrayName("name");
view.SetLayoutStrategyToSquarify();
view.SetRepresentationFromInputConnection(numeric.GetOutputPort());

# Apply a theme to the views
theme = vtkViewTheme.CreateMellowTheme()
view.ApplyViewTheme(theme)
theme.FastDelete()

view.ResetCamera()
view.Render()

view.GetInteractor().Start()