通过矢量幅度着色vtkUnstructuredGrid

时间:2012-07-17 11:52:08

标签: python qt vtk paraview

我正在尝试在Qt / VTK应用程序中显示EnSight文件。更准确地说,我想显示一个特定的EnSight部分,并通过其中一个矢量变量的大小对其进行着色。据我所知,VTK阅读器的输出为vtkMultiBlockDataSet,每个部分有一个块。每个块都是vtkUnstructuredGrid,变量是pointdata中的特定数组。

我到目前为止的代码如下。不幸的是,它显示了统一的颜色(当在ParaView中加载相同的文件时,它会显示一些局部变化)。

显然我错过了一些东西,但我无法想象。我会感激任何提示。

casefile = "data/Results/exported/blahblah.case"
part_id = 0
var_id = 2

reader = vtk.vtkGenericEnSightReader()
reader.SetCaseFileName(casefile)
reader.Update()

# Color map
colormap = vtk.vtkLookupTable()
colormap.SetHueRange(0.667, 0.0)
colormap.SetVectorModeToMagnitude()
colormap.Build()

multiblock = reader.GetOutput()
ugrid = multiblock.GetBlock(part_id)
pointdata = ugrid.GetPointData()
data = pointdata.GetArray(var_id)
data_range = data.GetRange(-1)

mesh_mapper = vtk.vtkDataSetMapper()
mesh_mapper.SetInput(ugrid)
mesh_mapper.SetColorModeToDefault()
mesh_mapper.SetScalarRange(data_range)
mesh_mapper.SetScalarVisibility(True)
mesh_mapper.SetLookupTable(colormap)
#
mesh_actor = vtk.vtkActor()
mesh_actor.SetMapper(mesh_mapper)
mesh_actor.GetProperty().SetDiffuseColor(1., 1., 1.)

renderer = vtk.vtkRenderer()
renderer.AddActor(mesh_actor)
renderer.AddActor2D(colorbar)

1 个答案:

答案 0 :(得分:0)

colormap = vtk.vtkLookupTable()
colormap.SetHueRange(0.667, 0.0)
colormap.SetVectorModeToMagnitude()
colormap.Build()

将生成范围在0和1之间的查找表。这是您的示例数据的数据范围吗?

如果没有,请将数据范围设置为data_range元组中的最小值/最大值,然后调用Build() - 我认为这应该有帮助 - 您是否有示例数据?