vtk polydata颜色映射

在 VTK 中，可以使用 vtkColorTransferFunction 来创建颜色映射，它将标量值映射到 RGB 颜色。以下是一个简单的示例代码，演示如何使用 vtkColorTransferFunction 来为 PolyData 显示颜色：

import vtk

# 创建一个 PolyData
points = vtk.vtkPoints()
points.InsertNextPoint(0, 0, 0)
points.InsertNextPoint(1, 0, 0)
points.InsertNextPoint(0, 1, 0)
points.InsertNextPoint(1, 1, 0)

polydata = vtk.vtkPolyData()
polydata.SetPoints(points)

# 添加标量数据
scalars = vtk.vtkFloatArray()
scalars.SetNumberOfComponents(1)
scalars.SetName("Scalars")
scalars.InsertNextValue(0)
scalars.InsertNextValue(0.5)
scalars.InsertNextValue(0.8)
scalars.InsertNextValue(1)

polydata.GetPointData().SetScalars(scalars)

# 创建颜色映射
color_transfer_function = vtk.vtkColorTransferFunction()
color_transfer_function.AddRGBPoint(0, 1, 0, 0)
color_transfer_function.AddRGBPoint(0.5, 0, 1, 0)
color_transfer_function.AddRGBPoint(1, 0, 0, 1)

# 创建 mapper 和 actor
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputData(polydata)
mapper.SetLookupTable(color_transfer_function)

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

# 创建 renderer 和 render window
renderer = vtk.vtkRenderer()
renderer.AddActor(actor)

render_window = vtk.vtkRenderWindow()
render_window.AddRenderer(renderer)

# 创建交互器并启动渲染
interactor = vtk.vtkRenderWindowInteractor()
interactor.SetRenderWindow(render_window)

render_window.Render()
interactor.Start()

在这个示例中，我们创建了一个简单的 PolyData，并为它添加了标量数据。然后，我们使用 vtkColorTransferFunction 创建了一个颜色映射，并将其设置为 PolyData 的 mapper 的查找表。最后，我们创建了一个 actor 并将其添加到 renderer 中，启动了渲染过程。