From 754469ec43311947e3e4695cd4b15af8a45fe0d9 Mon Sep 17 00:00:00 2001
From: Thomas Fischer <thomas.fischer@ufz.de>
Date: Wed, 9 Aug 2017 11:58:36 +0200
Subject: [PATCH] script for merging two scalar data arrays into a 2d data
array.
---
post/mergeDataArrays.py | 38 ++++++++++++++++++++++++++++++++++++++
1 file changed, 38 insertions(+)
create mode 100755 post/mergeDataArrays.py
diff --git a/post/mergeDataArrays.py b/post/mergeDataArrays.py
new file mode 100755
index 0000000..0fb66d7
--- /dev/null
+++ b/post/mergeDataArrays.py
@@ -0,0 +1,38 @@
+#!/usr/bin/env python2
+
+from vtk import *
+from sys import argv, exit
+
+if len(argv) < 6:
+ print "Usage:", argv[0], "input.vtu scalar1 scalar2 vector_2d output.vtu"
+ exit(1)
+
+input_file = argv[1]
+output_file = argv[5]
+print("Reading from", input_file)
+print("Writing to", output_file)
+
+r = vtkXMLUnstructuredGridReader()
+r.SetFileName(input_file)
+r.Update()
+
+m = r.GetOutput()
+point_data = m.GetPointData()
+scalar_array_1 = point_data.GetArray(argv[2])
+scalar_array_2 = point_data.GetArray(argv[3])
+new_array = vtkDoubleArray()
+new_array.SetNumberOfComponents(2)
+new_array.SetNumberOfTuples(scalar_array_1.GetNumberOfTuples())
+new_array.SetName(argv[4])
+new_array.CopyComponent(0, scalar_array_1, 0)
+new_array.CopyComponent(1, scalar_array_2, 0)
+point_data.AddArray(new_array)
+point_data.RemoveArray(argv[2])
+point_data.RemoveArray(argv[3])
+
+w = vtkXMLUnstructuredGridWriter()
+w.SetFileName(output_file)
+w.SetInputData(m)
+w.SetDataModeToAscii()
+w.SetCompressorTypeToNone()
+w.Update()
--
GitLab