From 05a3b569fc30ee53a4a73a25e6883d7dd5d944f7 Mon Sep 17 00:00:00 2001
From: chen <chenzhang8722@gamil.com>
Date: Fri, 14 Oct 2016 12:26:46 -0400
Subject: [PATCH] adding parallel version of addCalc for H5 table

---
 processing/post/h5_addCalculation.py | 56 ++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100755 processing/post/h5_addCalculation.py

diff --git a/processing/post/h5_addCalculation.py b/processing/post/h5_addCalculation.py
new file mode 100755
index 000000000..0ceeeefea
--- /dev/null
+++ b/processing/post/h5_addCalculation.py
@@ -0,0 +1,56 @@
+#!/usr/bin/env python2.7
+# -*- coding: UTF-8 no BOM -*-
+
+import os
+import re
+import sys
+import collections
+import math
+import damask
+import numpy as np
+from optparse import OptionParser
+
+scriptName = os.path.splitext(os.path.basename(__file__))[0]
+scriptID = ' '.join([scriptName, damask.version])
+
+
+# ----- Helper functions ----- #
+def listify(x):
+    return x if isinstance(x, collections.Iterable) else [x]
+
+
+# --------------------------------------------------------------------
+#                                MAIN
+# --------------------------------------------------------------------
+usageEx = """
+usage_in_details:
+    Column labels are tagged by '#label#' in formulas.
+    Use ';' for ',' in functions. Numpy is available as 'np'.
+    Special variables: #_row_# -- row index
+
+    Examples:
+    (1) magnitude of vector -- "np.linalg.norm(#vec#)"
+    (2) rounded root of row number -- "round(math.sqrt(#_row_#);3)"
+"""
+desp = "Add or alter column(s) with derived values according to "
+desp += "user-defined arithmetic operation between column(s)."
+
+parser = OptionParser(option_class=damask.extendableOption,
+                      usage='%prog options [file[s]]' + usageEx,
+                      description=desp,
+                      version=scriptID)
+parser.add_option('-l', '--label',
+                  dest='labels',
+                  action='extend', metavar='<string LIST>',
+                  help='(list of) new column labels')
+parser.add_option('-f', '--formula',
+                  dest='formulas',
+                  action='extend', metavar='<string LIST>',
+                  help='(list of) formulas corresponding to labels')
+parser.add_option('-c', '--condition',
+                  dest='condition', metavar='string',
+                  help='condition to filter rows')
+
+parser.set_defaults(condition=None)
+
+(options, filenames) = parser.parse_args()