1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
#!/usr/bin/env python
from vtk import *
import os.path
from vtk.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
dataRootDir = VTK_DATA_ROOT + "/Data/Infovis/XML/"
if not os.path.exists(dataRootDir):
dataRootDir = VTK_DATA_ROOT + "/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()
|
