我正在尝试在 Qt/VTK 应用程序中显示 EnSight 文件。更准确地说,我想显示一个特定的 EnSight 部件,并根据其中一个矢量变量的大小对其进行着色。据我了解,VTK 阅读器的输出是 a 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)