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.
86 lines
2.6 KiB
86 lines
2.6 KiB
2 years ago
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# This example shows how to construct a surface from a point cloud.
|
||
|
|
# First we generate a volume using the
|
||
|
|
# vtkSurfaceReconstructionFilter. The volume values are a distance
|
||
|
|
# field. Once this is generated, the volume is countoured at a
|
||
|
|
# distance value of 0.0.
|
||
|
|
|
||
|
|
import os
|
||
|
|
import string
|
||
|
|
import vtk
|
||
|
|
from vtk.util.misc import vtkGetDataRoot
|
||
|
|
VTK_DATA_ROOT = vtkGetDataRoot()
|
||
|
|
|
||
|
|
# Read some points. Use a programmable filter to read them.
|
||
|
|
pointSource = vtk.vtkProgrammableSource()
|
||
|
|
|
||
|
|
def readPoints():
|
||
|
|
output = pointSource.GetPolyDataOutput()
|
||
|
|
points = vtk.vtkPoints()
|
||
|
|
output.SetPoints(points)
|
||
|
|
|
||
|
|
file = open(os.path.normpath(os.path.join(VTK_DATA_ROOT, "Data/cactus.3337.pts")))
|
||
|
|
|
||
|
|
line = file.readline()
|
||
|
|
while line:
|
||
|
|
data = string.split(line)
|
||
|
|
if data and data[0] == 'p':
|
||
|
|
x, y, z = float(data[1]), float(data[2]), float(data[3])
|
||
|
|
points.InsertNextPoint(x, y, z)
|
||
|
|
line = file.readline()
|
||
|
|
|
||
|
|
pointSource.SetExecuteMethod(readPoints)
|
||
|
|
|
||
|
|
|
||
|
|
# Construct the surface and create isosurface.
|
||
|
|
surf = vtk.vtkSurfaceReconstructionFilter()
|
||
|
|
surf.SetInputConnection(pointSource.GetOutputPort())
|
||
|
|
|
||
|
|
cf = vtk.vtkContourFilter()
|
||
|
|
cf.SetInputConnection(surf.GetOutputPort())
|
||
|
|
cf.SetValue(0, 0.0)
|
||
|
|
|
||
|
|
# Sometimes the contouring algorithm can create a volume whose gradient
|
||
|
|
# vector and ordering of polygon (using the right hand rule) are
|
||
|
|
# inconsistent. vtkReverseSense cures this problem.
|
||
|
|
reverse = vtk.vtkReverseSense()
|
||
|
|
reverse.SetInputConnection(cf.GetOutputPort())
|
||
|
|
reverse.ReverseCellsOn()
|
||
|
|
reverse.ReverseNormalsOn()
|
||
|
|
|
||
|
|
map = vtk.vtkPolyDataMapper()
|
||
|
|
map.SetInputConnection(reverse.GetOutputPort())
|
||
|
|
map.ScalarVisibilityOff()
|
||
|
|
|
||
|
|
surfaceActor = vtk.vtkActor()
|
||
|
|
surfaceActor.SetMapper(map)
|
||
|
|
surfaceActor.GetProperty().SetDiffuseColor(1.0000, 0.3882, 0.2784)
|
||
|
|
surfaceActor.GetProperty().SetSpecularColor(1, 1, 1)
|
||
|
|
surfaceActor.GetProperty().SetSpecular(.4)
|
||
|
|
surfaceActor.GetProperty().SetSpecularPower(50)
|
||
|
|
|
||
|
|
# Create the RenderWindow, Renderer and both Actors
|
||
|
|
ren = vtk.vtkRenderer()
|
||
|
|
renWin = vtk.vtkRenderWindow()
|
||
|
|
renWin.AddRenderer(ren)
|
||
|
|
iren = vtk.vtkRenderWindowInteractor()
|
||
|
|
iren.SetRenderWindow(renWin)
|
||
|
|
|
||
|
|
# Add the actors to the renderer, set the background and size
|
||
|
|
ren.AddActor(surfaceActor)
|
||
|
|
ren.SetBackground(1, 1, 1)
|
||
|
|
renWin.SetSize(400, 400)
|
||
|
|
ren.GetActiveCamera().SetFocalPoint(0, 0, 0)
|
||
|
|
ren.GetActiveCamera().SetPosition(1, 0, 0)
|
||
|
|
ren.GetActiveCamera().SetViewUp(0, 0, 1)
|
||
|
|
ren.ResetCamera()
|
||
|
|
ren.GetActiveCamera().Azimuth(20)
|
||
|
|
ren.GetActiveCamera().Elevation(30)
|
||
|
|
ren.GetActiveCamera().Dolly(1.2)
|
||
|
|
ren.ResetCameraClippingRange()
|
||
|
|
|
||
|
|
iren.Initialize()
|
||
|
|
renWin.Render()
|
||
|
|
iren.Start()
|