mirror of
https://github.com/OpenFOAM/ThirdParty-6.git
synced 2025-12-08 06:57:43 +00:00
36 lines
942 B
Python
36 lines
942 B
Python
#!/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()
|