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