Skip to content
Snippets Groups Projects
Commit 8810131d authored by Nikhil Shetty's avatar Nikhil Shetty
Browse files

Adding IOPLY python tests

Change-Id: I40f1c2985891a923e116aea1fe07dfee3695082a
parent ee781ba1
No related branches found
No related tags found
No related merge requests found
add_test_python(TestPLYReadWrite.py IO)
#!/usr/bin/env python
import vtk
from vtk.test import Testing
from vtk.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
ss = vtk.vtkSphereSource()
ss.SetPhiResolution(10)
ss.SetThetaResolution(20)
ele = vtk.vtkSimpleElevationFilter()
ele.SetInputConnection(ss.GetOutputPort())
pd2cd = vtk.vtkPointDataToCellData()
pd2cd.SetInputConnection(ele.GetOutputPort())
# First way or writing
w = vtk.vtkPLYWriter()
w.SetInputConnection(pd2cd.GetOutputPort())
w.SetFileName("plyWriter.ply")
w.SetFileTypeToBinary()
w.SetDataByteOrderToLittleEndian()
w.SetColorModeToUniformCellColor()
w.SetColor(255,0,0)
w.Write()
r = vtk.vtkPLYReader()
r.SetFileName("plyWriter.ply")
r.Update()
file.delete("-force", "plyWriter.ply")
plyMapper = vtk.vtkPolyDataMapper()
plyMapper.SetInputConnection(r.GetOutputPort())
plyActor = vtk.vtkActor()
plyActor.SetMapper(plyMapper)
# Second way or writing - it will map through a lookup table
lut = vtk.vtkLookupTable()
lut.Build()
w2 = vtk.vtkPLYWriter()
w2.SetInputConnection(pd2cd.GetOutputPort())
w2.SetFileName("plyWriter.ply")
w2.SetFileTypeToBinary()
w2.SetDataByteOrderToLittleEndian()
w2.SetColorModeToDefault()
w2.SetLookupTable(lut)
w2.SetArrayName("Elevation")
w2.SetComponent(0)
w2.Write()
r2 = vtk.vtkPLYReader()
r2.SetFileName("plyWriter.ply")
r2.Update()
plyMapper2 = vtk.vtkPolyDataMapper()
plyMapper2.SetInputConnection(r2.GetOutputPort())
plyActor2 = vtk.vtkActor()
plyActor2.SetMapper(plyMapper2)
plyActor2.AddPosition(1,0,0)
# Third way or writing - it will read the previous file with rgb cell color
r3 = vtk.vtkPLYReader()
r3.SetFileName("plyWriter.ply")
r3.Update()
w3 = vtk.vtkPLYWriter()
w3.SetInputConnection(r3.GetOutputPort())
w3.SetFileName("plyWriter.ply")
w3.SetFileTypeToBinary()
w3.SetDataByteOrderToLittleEndian()
w3.SetColorModeToDefault()
w3.SetArrayName("RGB")
w3.SetComponent(0)
w3.Write()
r4 = vtk.vtkPLYReader()
r4.SetFileName("plyWriter.ply")
r4.Update()
plyMapper3 = vtk.vtkPolyDataMapper()
plyMapper3.SetInputConnection(r4.GetOutputPort())
plyActor3 = vtk.vtkActor()
plyActor3.SetMapper(plyMapper3)
plyActor3.AddPosition(2,0,0)
file.delete("-force", "plyWriter.ply")
# Create the RenderWindow, Renderer and both Actors
#
ren1 = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren1)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
# Add the actors to the renderer, set the background and size
#
ren1.AddActor(plyActor)
ren1.AddActor(plyActor2)
ren1.AddActor(plyActor3)
renWin.SetSize(325,125)
iren.Initialize()
renWin.Render()
ren1.GetActiveCamera().Zoom(3.0)
# prevent the tk window from showing up then start the event loop
# --- end of script --
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment