如何在vtkStructuredGrid上设置数据值

时间:2011-10-05 19:58:22

标签: vtk

我正在尝试使用分析字段填充结构化网格,但是尽管阅读了vtk文档,我还没有找到如何在网格点实际设置标量值或设置间距/原点信息网格。从下面的代码开始,我该如何

  1. 将空间信息与网格相关联(即单元格0,0,0位于坐标0,0,0处,间距为每个方向的dx)
  2. 将标量值与每个网格点相关联。首先,我只需要一个,但最终我想在每个点存储3个数据(不是矢量,3个不同的标量)。
  3. grid = vtk.vtkStructuredGrid()
    numPoints = int((maxGrid - minGrid)/dx)
    grid.SetDimensions(numPoints, numPoints, numPoints)
    

1 个答案:

答案 0 :(得分:14)

在VTK中有3种类型的“结构化”网格,vtkImageData(vtkUniformGrid派生自此),vtkRectilinearGrid和vtkStructuredGrid。它们都是在拓扑设置的意义上构建的。 vtkImageData在点之间具有恒定的间距并且是轴对齐的,vtkRectilinearGrid是轴对齐的但是可以改变每个轴方向上的间距,并且vtkStructuredGrid具有任意定位的点(尽管单元格可能无效)。

你想做什么,你应该这样做:

from vtk import *
dx = 2.0
grid = vtkImageData()
grid.SetOrigin(0, 0, 0) # default values
grid.SetSpacing(dx, dx, dx)
grid.SetDimensions(5, 8, 10) # number of points in each direction
# print grid.GetNumberOfPoints()
# print grid.GetNumberOfCells()
array = vtkDoubleArray()
array.SetNumberOfComponents(1) # this is 3 for a vector
array.SetNumberOfTuples(grid.GetNumberOfPoints())
for i in range(grid.GetNumberOfPoints()):
    array.SetValue(i, 1)

grid.GetPointData().AddArray(array)
# print grid.GetPointData().GetNumberOfArrays()
array.SetName("unit array")