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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#!/usr/bin/env python
from vtk import *
# generate a random graph
source = vtkRandomGraphSource()
source.SetNumberOfVertices(15000)
source.SetAllowSelfLoops(False)
source.SetEdgeProbability(0.003)
source.SetUseEdgeProbability(True)
source.AllowParallelEdgesOff()
# compute the kcore levels for every vertex in the graph
kcore = vtkKCoreDecomposition()
kcore.AddInputConnection(source.GetOutputPort())
kcore.SetOutputArrayName("kcore")
kcore.CheckInputGraphOn()
# generate x/y coordinates for vertices based on coreness
kcoreLayout = vtkKCoreLayout()
kcoreLayout.SetGraphConnection( kcore.GetOutputPort() )
kcoreLayout.SetCartesian(True)
kcoreLayout.SetEpsilon(0.2)
kcoreLayout.SetUnitRadius(1.0)
# assign coordinats for layout purposes based on the x/y coordinates
# that are created in kcoreLayout
kcoreAssignCoords = vtkAssignCoordinates()
kcoreAssignCoords.SetInputConnection(kcoreLayout.GetOutputPort())
kcoreAssignCoords.SetXCoordArrayName("coord_x")
kcoreAssignCoords.SetYCoordArrayName("coord_y")
kcoreAssignCoords.Update()
# draw it
view = vtkGraphLayoutView()
view.AddRepresentationFromInputConnection(kcoreAssignCoords.GetOutputPort())
view.SetVertexLabelArrayName("kcore")
view.SetVertexLabelVisibility(False)
view.SetVertexColorArrayName("kcore")
view.SetColorVertices(True)
# turn off edge visibility since it isn't useful for this view
view.SetEdgeVisibility(False)
# use the coordinates assigned by kcoreAssignCoords
view.SetLayoutStrategyToPassThrough()
theme = vtkViewTheme.CreateNeonTheme()
theme.SetLineWidth(1)
theme.SetPointSize(5)
view.ApplyViewTheme(theme)
theme.FastDelete()
view.GetRenderWindow().SetSize(600, 600)
view.ResetCamera()
view.Render()
view.GetInteractor().Start()
|
