Draw multiple planes in vtk python api

Question

I am trying to render an indoor scene with floor and walls and box like furniture in it using vtk in python. I figured how to draw a single plane in vtk (see code below). But I cannot figure out how to draw multiple planes. Moreover is there a way to give color to this plane?

import vtk

'''VTK'''
planes = vtk.vtkPlaneSource()
planes.SetCenter(1.0, 0.0, 0.0)
planes.SetNormal(1.0, 0.0, 1.0)
# planes.SetExtent(1,100,1,100,7,7);


# mapper
mapper = vtk.vtkPolyDataMapper()
mapper.SetInput(planes.GetOutput())

# actor
actor = vtk.vtkActor()
actor.SetMapper(mapper)

'''RENDER'''
renderer = vtk.vtkRenderer()
# Background
renderer.SetBackground(1., 1., 1.)
# Add actor to the scene
renderer.AddActor(actor)

'''Render window'''
renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)

'''Interactor'''
renderWindowInteractor = vtk.vtkRenderWindowInteractor()
renderWindowInteractor.SetInteractorStyle(vtk.vtkInteractorStyleTrackballCamera())
renderWindowInteractor.SetRenderWindow(renderWindow)

'''Add camera coordinates'''
axes = vtk.vtkAxesActor()
widget = vtk.vtkOrientationMarkerWidget()
widget.SetOutlineColor(0.9300, 0.5700, 0.1300)
widget.SetOrientationMarker(axes)
widget.SetInteractor(renderWindowInteractor)
widget.SetViewport(0.0, 0.0, 0.4, 0.4)
widget.SetEnabled(1)
widget.InteractiveOn()

# Reset camera
renderer.ResetCamera()

'''Begin interaction'''
renderWindow.Render()
renderWindowInteractor.Initialize()
renderWindowInteractor.Start()
renderWindowInteractor.Start()

score 3 · Accepted Answer · answered Jan 12 '16 at 15:11

I just realized that it can be done as easily as shown below.

'''VTK'''
plane1 = vtk.vtkPlaneSource()
plane1.SetCenter(1.0, 0.0, 0.0)
plane1.SetNormal(1.0, 0.0, 0.0)
# planes.SetExtent(1,100,1,100,7,7);

'''VTK'''
plane2 = vtk.vtkPlaneSource()
plane2.SetCenter(1.0, 2.0, 0.0)
plane2.SetNormal(0.0, 1.0, 0.0)
# planes.SetExtent(1,100,1,100,7,7);

# mapper
mapper1 = vtk.vtkPolyDataMapper()
mapper1.SetInput(plane1.GetOutput())

# mapper
mapper2 = vtk.vtkPolyDataMapper()
mapper2.SetInput(plane2.GetOutput())

# actor
actor1 = vtk.vtkActor()
actor1.SetMapper(mapper1)

actor2 = vtk.vtkActor()
actor2.SetMapper(mapper2)


'''RENDER'''
renderer = vtk.vtkRenderer()
# Background
renderer.SetBackground(1., 1., 1.)
# Add actor to the scene
renderer.AddActor(actor1)
renderer.AddActor(actor2)

'''Render window'''
renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)

'''Interactor'''
renderWindowInteractor = vtk.vtkRenderWindowInteractor()
renderWindowInteractor.SetInteractorStyle(vtk.vtkInteractorStyleTrackballCamera())
renderWindowInteractor.SetRenderWindow(renderWindow)

'''Add camera coordinates'''
axes = vtk.vtkAxesActor()
widget = vtk.vtkOrientationMarkerWidget()
widget.SetOutlineColor(0.9300, 0.5700, 0.1300)
widget.SetOrientationMarker(axes)
widget.SetInteractor(renderWindowInteractor)
widget.SetViewport(0.0, 0.0, 0.4, 0.4)
widget.SetEnabled(1)
widget.InteractiveOn()

# Reset camera
renderer.ResetCamera()

'''Begin interaction'''
renderWindow.Render()
renderWindowInteractor.Initialize()
renderWindowInteractor.Start()
renderWindowInteractor.Start()

score 0 · Answer 2 · answered Oct 26 '19 at 23:22

You may want to check out PyVista. It makes all of this pretty easy with minimal code:

import pyvista as pv

plane1 = pv.Plane(center=(1.0, 0.0, 0.0), 
                  direction=(1.0, 0.0, 0.0))
plane2 = pv.Plane(center=(1.0, 2.0, 0.0), 
                  direction=(0.0, 1.0, 0.0))

plotter = pv.Plotter()
plotter.add_mesh(plane1, color='blue')
plotter.add_mesh(plane2, color='red')
plotter.show_grid()
plotter.show()

Draw multiple planes in vtk python api

2 Answers2