使用Python编写和修改VTK多数据文件

Question

我有一个vtk多数据文件-我正在编写一个Python代码，该代码读取此多数据文件的方式如下：

fullDirPath  = os.getcwd()
fullFileName = fullDirPath+'/'+fileName
reader = vtkPolyDataReader()
reader.SetFileName(vtkFileName)
reader.ReadAllScalarsOn()
reader.ReadAllVectorsOn()
reader.Update()

data        = reader.GetOutput()
velArray    = vtk_to_numpy(data.GetPointData().GetArray('velocity'))

在这里，我省略了vtk库的所有相关库导入语句。

我现在想对该速度数据velArray进行操作，并创建一个新的标量字段velParam ，然后编写一个新的VTK多数据文件，该文件将velParam作为附加的标量字段。 不幸的是，到目前为止，我还没有找到在Python中操作vtkPolyDataWriter对象及其功能的正确方法-而且我没有找到很多相同的示例。 我想知道从给定的一组点坐标，矢量数据字段和标量数据字段编写vtk Polydata文件的好方法是什么。

非常感谢。

Answer 1

不必操纵编写器（它只需要文件名和一个多数据），而是必须创建所需的多数据，然后将其提供给编写器。 就像是：

velParam_numpy = 2*velArray
velParam_vtk = numpy_to_vtk(velParam_numpy)
velParam_vtk.SetName('VelParam') #rememebr to give an unique name
data.GetPointData().AddArray(velParam_vtk)