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.
185 lines
5.8 KiB
185 lines
5.8 KiB
#!/usr/bin/env python
|
|
|
|
# This example shows how to align a set of objects together using the
|
|
# Procrustes algorithm. We make three ellipsoids by distorting and
|
|
# translating a sphere and then align them together, using the
|
|
# different modes of Procrustes alignment: rigid-body, similarity and
|
|
# affine.
|
|
|
|
import vtk
|
|
|
|
sphere = vtk.vtkSphereSource()
|
|
|
|
# make two copies of the shape and distort them a little
|
|
transform1 = vtk.vtkTransform()
|
|
transform1.Translate(0.2, 0.1, 0.3)
|
|
transform1.Scale(1.3, 1.1, 0.8)
|
|
|
|
transform2 = vtk.vtkTransform()
|
|
transform2.Translate(0.3, 0.7, 0.1)
|
|
transform2.Scale(1.0, 0.1, 1.8)
|
|
|
|
transformer1 = vtk.vtkTransformPolyDataFilter()
|
|
transformer1.SetInputConnection(sphere.GetOutputPort())
|
|
transformer1.SetTransform(transform1)
|
|
|
|
transformer2 = vtk.vtkTransformPolyDataFilter()
|
|
transformer2.SetInputConnection(sphere.GetOutputPort())
|
|
transformer2.SetTransform(transform2)
|
|
|
|
# map these three shapes into the first renderer
|
|
map1a = vtk.vtkPolyDataMapper()
|
|
map1a.SetInputConnection(sphere.GetOutputPort())
|
|
Actor1a = vtk.vtkActor()
|
|
Actor1a.SetMapper(map1a)
|
|
Actor1a.GetProperty().SetDiffuseColor(1.0000, 0.3882, 0.2784)
|
|
|
|
map1b = vtk.vtkPolyDataMapper()
|
|
map1b.SetInputConnection(transformer1.GetOutputPort())
|
|
Actor1b = vtk.vtkActor()
|
|
Actor1b.SetMapper(map1b)
|
|
Actor1b.GetProperty().SetDiffuseColor(0.3882, 1.0000, 0.2784)
|
|
|
|
map1c = vtk.vtkPolyDataMapper()
|
|
map1c.SetInputConnection(transformer2.GetOutputPort())
|
|
Actor1c = vtk.vtkActor()
|
|
Actor1c.SetMapper(map1c)
|
|
Actor1c.GetProperty().SetDiffuseColor(0.3882, 0.2784, 1.0000)
|
|
|
|
# -- align the shapes using Procrustes (using SetModeToRigidBody) --
|
|
procrustes1 = vtk.vtkProcrustesAlignmentFilter()
|
|
procrustes1.SetNumberOfInputs(3)
|
|
procrustes1.SetInputConnection(0, sphere.GetOutputPort())
|
|
procrustes1.SetInputConnection(1, transformer1.GetOutputPort())
|
|
procrustes1.SetInputConnection(2, transformer2.GetOutputPort())
|
|
procrustes1.GetLandmarkTransform().SetModeToRigidBody()
|
|
|
|
# map the aligned shapes into the second renderer
|
|
map2a = vtk.vtkPolyDataMapper()
|
|
map2a.SetInputConnection(procrustes1.GetOutputPort(0))
|
|
Actor2a = vtk.vtkActor()
|
|
Actor2a.SetMapper(map2a)
|
|
Actor2a.GetProperty().SetDiffuseColor(1.0000, 0.3882, 0.2784)
|
|
|
|
map2b = vtk.vtkPolyDataMapper()
|
|
map2b.SetInputConnection(procrustes1.GetOutputPort(1))
|
|
Actor2b = vtk.vtkActor()
|
|
Actor2b.SetMapper(map2b)
|
|
Actor2b.GetProperty().SetDiffuseColor(0.3882, 1.0000, 0.2784)
|
|
|
|
map2c = vtk.vtkPolyDataMapper()
|
|
map2c.SetInputConnection(procrustes1.GetOutputPort(2))
|
|
Actor2c = vtk.vtkActor()
|
|
Actor2c.SetMapper(map2c)
|
|
Actor2c.GetProperty().SetDiffuseColor(0.3882, 0.2784, 1.0000)
|
|
|
|
# -- align the shapes using Procrustes (using SetModeToSimilarity
|
|
# (default)) --
|
|
procrustes2 = vtk.vtkProcrustesAlignmentFilter()
|
|
procrustes2.SetNumberOfInputs(3)
|
|
procrustes2.SetInputConnection(0, sphere.GetOutputPort())
|
|
procrustes2.SetInputConnection(1, transformer1.GetOutputPort())
|
|
procrustes2.SetInputConnection(2, transformer2.GetOutputPort())
|
|
|
|
# map the aligned shapes into the third renderer
|
|
map3a = vtk.vtkPolyDataMapper()
|
|
map3a.SetInputConnection(procrustes2.GetOutputPort(0))
|
|
Actor3a = vtk.vtkActor()
|
|
Actor3a.SetMapper(map3a)
|
|
Actor3a.GetProperty().SetDiffuseColor(1.0000, 0.3882, 0.2784)
|
|
|
|
map3b = vtk.vtkPolyDataMapper()
|
|
map3b.SetInputConnection(procrustes2.GetOutputPort(1))
|
|
Actor3b = vtk.vtkActor()
|
|
Actor3b.SetMapper(map3b)
|
|
Actor3b.GetProperty().SetDiffuseColor(0.3882, 1.0000, 0.2784)
|
|
|
|
map3c = vtk.vtkPolyDataMapper()
|
|
map3c.SetInputConnection(procrustes2.GetOutputPort(2))
|
|
Actor3c = vtk.vtkActor()
|
|
Actor3c.SetMapper(map3c)
|
|
Actor3c.GetProperty().SetDiffuseColor(0.3882, 0.2784, 1.0000)
|
|
|
|
# -- align the shapes using Procrustes (using SetModeToAffine) --
|
|
procrustes3 = vtk.vtkProcrustesAlignmentFilter()
|
|
procrustes3.SetNumberOfInputs(3)
|
|
procrustes3.SetInputConnection(0, sphere.GetOutputPort())
|
|
procrustes3.SetInputConnection(1, transformer1.GetOutputPort())
|
|
procrustes3.SetInputConnection(2, transformer2.GetOutputPort())
|
|
procrustes3.GetLandmarkTransform().SetModeToAffine()
|
|
|
|
# map the aligned shapes into the fourth renderer
|
|
map4a = vtk.vtkPolyDataMapper()
|
|
map4a.SetInputConnection(procrustes3.GetOutputPort(0))
|
|
Actor4a = vtk.vtkActor()
|
|
Actor4a.SetMapper(map4a)
|
|
Actor4a.GetProperty().SetDiffuseColor(1.0000, 0.3882, 0.2784)
|
|
|
|
map4b = vtk.vtkPolyDataMapper()
|
|
map4b.SetInputConnection(procrustes3.GetOutputPort(1))
|
|
Actor4b = vtk.vtkActor()
|
|
Actor4b.SetMapper(map4b)
|
|
Actor4b.GetProperty().SetDiffuseColor(0.3882, 1.0000, 0.2784)
|
|
|
|
map4c = vtk.vtkPolyDataMapper()
|
|
map4c.SetInputConnection(procrustes3.GetOutputPort(2))
|
|
Actor4c = vtk.vtkActor()
|
|
Actor4c.SetMapper(map4c)
|
|
Actor4c.GetProperty().SetDiffuseColor(0.3882, 0.2784, 1.0000)
|
|
|
|
# Create the RenderWindow and its four Renderers
|
|
ren = vtk.vtkRenderer()
|
|
ren2 = vtk.vtkRenderer()
|
|
ren3 = vtk.vtkRenderer()
|
|
ren4 = vtk.vtkRenderer()
|
|
renWin = vtk.vtkRenderWindow()
|
|
renWin.AddRenderer(ren)
|
|
renWin.AddRenderer(ren2)
|
|
renWin.AddRenderer(ren3)
|
|
renWin.AddRenderer(ren4)
|
|
renWin.SetSize(400, 100)
|
|
iren = vtk.vtkRenderWindowInteractor()
|
|
iren.SetRenderWindow(renWin)
|
|
|
|
# Add the actors to the renderer
|
|
ren.AddActor(Actor1a)
|
|
ren.AddActor(Actor1b)
|
|
ren.AddActor(Actor1c)
|
|
|
|
ren2.AddActor(Actor2a)
|
|
ren2.AddActor(Actor2b)
|
|
ren2.AddActor(Actor2c)
|
|
|
|
ren3.AddActor(Actor3a)
|
|
ren3.AddActor(Actor3b)
|
|
ren3.AddActor(Actor3c)
|
|
|
|
ren4.AddActor(Actor4a)
|
|
ren4.AddActor(Actor4b)
|
|
ren4.AddActor(Actor4c)
|
|
|
|
# set the properties of the renderers
|
|
ren.SetBackground(1, 1, 1)
|
|
ren.SetViewport(0.0, 0.0, 0.25, 1.0)
|
|
ren.GetActiveCamera().SetPosition(1, -1, 0)
|
|
ren.ResetCamera()
|
|
|
|
ren2.SetBackground(1, 1, 1)
|
|
ren2.SetViewport(0.25, 0.0, 0.5, 1.0)
|
|
ren2.GetActiveCamera().SetPosition(1, -1, 0)
|
|
ren2.ResetCamera()
|
|
|
|
ren3.SetBackground(1, 1, 1)
|
|
ren3.SetViewport(0.5, 0.0, 0.75, 1.0)
|
|
ren3.GetActiveCamera().SetPosition(1, -1, 0)
|
|
ren3.ResetCamera()
|
|
|
|
ren4.SetBackground(1, 1, 1)
|
|
ren4.SetViewport(0.75, 0.0, 1.0, 1.0)
|
|
ren4.GetActiveCamera().SetPosition(1, -1, 0)
|
|
ren4.ResetCamera()
|
|
|
|
# Render the image and start interaction.
|
|
iren.Initialize()
|
|
renWin.Render()
|
|
iren.Start()
|
|
|