Python VTK-使用矢量数据为vtkStreamTracer线着色

作为作业的一部分，我试图想象粘性液体通过混合器的流动。为此提供的数据是vtk StructuredPoint文件，包含标量和矢量数据。标量数据告诉我们给定的点是否是混合器结构的一部分，矢量数据包含液体流动的信息

我已经使用使用vtkStreamTracer类创建的流线可视化了流程。当输出被传递到vtkPolyDataMapper时，它似乎默认使用数据的标量部分为行着色

我一直在试图找到一种方法来改变这一点，但我没能做到。我试图用流动向量的大小来确定线条的颜色，这样颜色就代表了液体的速度。 有什么提示吗？我已经在下面介绍了可视化代码的相关部分。提前谢谢

seeds = vtk.vtkPointSource()
seeds.SetRadius(15)
seeds.SetCenter(0,30.5,30.5)
seeds.SetNumberOfPoints(100)

integrator=vtk.vtkRungeKutta45()
streamer = vtk.vtkStreamTracer()
streamer.SetInputConnection(reader.GetOutputPort())
streamer.SetSourceConnection(seeds.GetOutputPort())
streamer.SetIntegrationDirectionToBoth()
streamer.SetIntegrator(integrator)
streamer.SetMaximumPropagation(500)
streamer.SetComputeVorticity(True)


mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(streamer.GetOutputPort())
actor = vtk.vtkActor()
actor.SetMapper(mapper)

请访问此网站： 它有一个可能对您有用的流线型示例。它使用了一个颜色传递函数，因此在这种情况下，颜色表示风速，类似于您试图实现的效果

---

希望这有帮助

虽然链接已断开，但可以在存档中访问它：