使用python读取VTK文件

Question

我有以下VTK文件，它由OpenFOAM生成：

# vtk DataFile Version 2.0
sampleSurface
ASCII
DATASET POLYDATA
POINTS 4 float
0.0 0.0 0.0
1.0 0.0 0.0
0.0 1.0 0.0
1.0 1.0 0.0

POLYGONS 2 8
3 0 1 2
3 2 1 3

POINT_DATA 4
FIELD attributes 1
U 3 4 float
1.0 2.0 3.0
1.0 2.0 3.0
1.0 2.0 3.0
1.0 2.0 3.0

它是3D域的平面切割平面。 有4个点，创建两个三角形。 在每个点上，定义向量U. 我可以使用以下代码获得点数和点数：

import vtk
reader = vtk.vtkPolyDataReader()
reader.SetFileName('myVTKfile.vtk')
reader.ReadAllScalarsOn()
reader.ReadAllVectorsOn()
reader.ReadAllTensorsOn()
reader.Update()
vtkdata = reader.GetOutput()

print vtkdata.GetNumberOfPoints()
print vtkdata.GetPoint(0)

不幸的是，我没有找到获得三角形（多边形）列表和数据列表（向量U）的可能性。 有人可以帮我解决这个问题吗？

马塞尔

Answer 1

对于逐点数据（例如标量和向量），您可以通过以下方式访问它：

pointData = vtkdata.GetPointData()
vectorData = pointData.GetVectors()

然后，vectorData将包含一个可以使用的vtkDataArray。

对于几何数据，可以使用GetVerts，GetLines，GetPolys（三角形，四边形和其他多边形）以及可能的GetStrips（用于三角形条带）方法。 对于未拼接成三角形条带的三角形，您可以使用以下方法访问数据：

polyCells = vtkdata.GetPolys()
numPolys = polyCells.GetNumberOfCells() #Number of polygons (triangles in your case)

从python中访问单元格数据（它只是一个点索引列表）有点痛苦，并且自从我上次用VTK编写任何东西以来，显然已经发生了变化。 （回到VTK 5.x）。 至少，您可以获取单元阵列并按照此处所述进行扫描： http ： //vtk.1045678.n5.nabble.com/vtkCellArray-in-python-td3348424.html