You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
2.5 KiB
89 lines
2.5 KiB
2 years ago
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# This example demonstrates how to use the vtkPlaneWidget to probe
|
||
|
|
# a dataset and then generate contours on the probed data.
|
||
|
|
|
||
|
|
import vtk
|
||
|
|
from vtk.util.misc import vtkGetDataRoot
|
||
|
|
VTK_DATA_ROOT = vtkGetDataRoot()
|
||
|
|
|
||
|
|
# Start by loading some data.
|
||
|
|
pl3d = vtk.vtkPLOT3DReader()
|
||
|
|
pl3d.SetXYZFileName(VTK_DATA_ROOT + "/Data/combxyz.bin")
|
||
|
|
pl3d.SetQFileName(VTK_DATA_ROOT + "/Data/combq.bin")
|
||
|
|
pl3d.SetScalarFunctionNumber(100)
|
||
|
|
pl3d.SetVectorFunctionNumber(202)
|
||
|
|
pl3d.Update()
|
||
|
|
|
||
|
|
# The plane widget is used probe the dataset.
|
||
|
|
planeWidget = vtk.vtkPlaneWidget()
|
||
|
|
planeWidget.SetInput(pl3d.GetOutput())
|
||
|
|
planeWidget.NormalToXAxisOn()
|
||
|
|
planeWidget.SetResolution(20)
|
||
|
|
planeWidget.SetRepresentationToOutline()
|
||
|
|
planeWidget.PlaceWidget()
|
||
|
|
plane = vtk.vtkPolyData()
|
||
|
|
planeWidget.GetPolyData(plane)
|
||
|
|
|
||
|
|
probe = vtk.vtkProbeFilter()
|
||
|
|
probe.SetInput(plane)
|
||
|
|
probe.SetSource(pl3d.GetOutput())
|
||
|
|
|
||
|
|
contourMapper = vtk.vtkPolyDataMapper()
|
||
|
|
contourMapper.SetInputConnection(probe.GetOutputPort())
|
||
|
|
contourMapper.SetScalarRange(pl3d.GetOutput().GetScalarRange())
|
||
|
|
contourActor = vtk.vtkActor()
|
||
|
|
contourActor.SetMapper(contourMapper)
|
||
|
|
contourActor.VisibilityOff()
|
||
|
|
|
||
|
|
# An outline is shown for context.
|
||
|
|
outline = vtk.vtkStructuredGridOutlineFilter()
|
||
|
|
outline.SetInputConnection(pl3d.GetOutputPort())
|
||
|
|
outlineMapper = vtk.vtkPolyDataMapper()
|
||
|
|
outlineMapper.SetInputConnection(outline.GetOutputPort())
|
||
|
|
outlineActor = vtk.vtkActor()
|
||
|
|
outlineActor.SetMapper(outlineMapper)
|
||
|
|
|
||
|
|
# Create the RenderWindow, Renderer and both Actors
|
||
|
|
ren = vtk.vtkRenderer()
|
||
|
|
renWin = vtk.vtkRenderWindow()
|
||
|
|
renWin.AddRenderer(ren)
|
||
|
|
iren = vtk.vtkRenderWindowInteractor()
|
||
|
|
iren.SetRenderWindow(renWin)
|
||
|
|
|
||
|
|
# Actually generate contour lines.
|
||
|
|
def BeginInteraction(obj, event):
|
||
|
|
global plane, contourActor
|
||
|
|
obj.GetPolyData(plane)
|
||
|
|
contourActor.VisibilityOn()
|
||
|
|
|
||
|
|
def ProbeData(obj, event):
|
||
|
|
global plane
|
||
|
|
obj.GetPolyData(plane)
|
||
|
|
|
||
|
|
|
||
|
|
# Associate the widget with the interactor
|
||
|
|
planeWidget.SetInteractor(iren)
|
||
|
|
# Handle the events.
|
||
|
|
planeWidget.AddObserver("EnableEvent", BeginInteraction)
|
||
|
|
planeWidget.AddObserver("StartInteractionEvent", BeginInteraction)
|
||
|
|
planeWidget.AddObserver("InteractionEvent", ProbeData)
|
||
|
|
|
||
|
|
# Add the actors to the renderer, set the background and size
|
||
|
|
ren.AddActor(outlineActor)
|
||
|
|
ren.AddActor(contourActor)
|
||
|
|
|
||
|
|
ren.SetBackground(1, 1, 1)
|
||
|
|
renWin.SetSize(300, 300)
|
||
|
|
ren.SetBackground(0.1, 0.2, 0.4)
|
||
|
|
|
||
|
|
cam1 = ren.GetActiveCamera()
|
||
|
|
cam1.SetClippingRange(3.95297, 50)
|
||
|
|
cam1.SetFocalPoint(9.71821, 0.458166, 29.3999)
|
||
|
|
cam1.SetPosition(2.7439, -37.3196, 38.7167)
|
||
|
|
cam1.SetViewUp(-0.16123, 0.264271, 0.950876)
|
||
|
|
|
||
|
|
iren.Initialize()
|
||
|
|
renWin.Render()
|
||
|
|
iren.Start()
|