[extractPointInformationFromPVDUsingPointID.py]

post/extractPointInformation/extractPointInformationFromPVDUsingPointID.py

+#!/usr/bin/env python
+
+import sys
+from vtk import *
+from lxml import etree as ET
+
+def readVTU(filename):
+    reader = vtkXMLUnstructuredGridReader()
+    reader.SetFileName(filename)
+    reader.Update()
+    return reader.GetOutput()
+
+def extractBulkPointData(bulk_vtu, node_id, array_name):
+    point_data = bulk_vtu.GetPointData()
+    data_array = point_data.GetArray(array_name)
+    return data_array.GetTuple1(node_id)
+
+def readPVD(filename):
+    tree = ET.parse(filename)
+    root = tree.getroot()
+    ts_files = {}
+    ts_files['ts'] = []
+    ts_files['filename'] = []
+    for collection in root.getchildren():
+        for dataset in collection.getchildren():
+            ts_files['ts'].append(dataset.attrib['timestep'])
+            ts_files['filename'].append(dataset.attrib['file'])
+    return ts_files
+
+pvd = readPVD(sys.argv[1])
+node_id = int(sys.argv[2])
+array_name=sys.argv[3]
+
+for i in range(1, len(pvd['ts'])):
+    vtu = readVTU(pvd['filename'][i])
+    print(pvd['ts'][i] + ' ' + str(extractBulkPointData(vtu, node_id, array_name)))