shearfile.py + test

README.md

@@ -4,6 +4,7 @@
 - [Hawc2io](wetb/hawc2/Hawc2io.py): Read binary, ascii and flex result files
 - [sel_file](wetb/hawc2/sel_file.py): Read/write *.sel (sensor list) files
 - [htcfile](wetb/hawc2/htcfile.py): Read/write/manipulate htc files
+- [shearfile](wetb/hawc2/shearfile.py): Create user defined shear file
 - [at_time_file](wetb/hawc2/at_time_file.py): read at output_at_time files
 
 

wetb/hawc2/shearfile.py

+'''
+Created on 24/04/2014
+
+@author: MMPE
+'''
+
+import numpy as np
+
+def save(filename, y_coordinates, z_coordinates, u=None, v=None, w=None):
+    shape = (len(z_coordinates), len(y_coordinates))
+    vuw = [v, u, w]
+    for i in range(3):
+        if vuw[i] is None:
+            vuw[i] = np.zeros((shape))
+        else:
+            assert vuw[i].shape == shape, (i, vuw[i].shape, shape)
+
+    with open(filename, 'w') as fid:
+        fid.write(" # autogenerated shear file\n")
+        fid.write("  %d %d\n" % (len(y_coordinates), len(z_coordinates)))
+        for i, l in enumerate(['v', 'u', 'w']):
+            fid.write(" # shear %s component\n  " % l)
+            fid.write("\n  ".join([" ".join(["%.10f" % v for v in r ]) for r in vuw[i]]))
+            fid.write("\n")
+        for yz, coor in (['y', y_coordinates], ['z', z_coordinates]):
+            fid.write(" # %s coordinates\n  " % yz)
+            fid.write("\n  ".join("%.10f" % v for v in coor))
+            fid.write("\n")
+
+
+if __name__ == "__main__":
+    save("test.dat", [-55, 55], [30, 100, 160] , u=np.array([[0.7, 1, 1.3], [0.7, 1, 1.3]]).T)

wetb/hawc2/tests/test_shearfile.py

+'''
+Created on 05/11/2015
+
+@author: MMPE
+'''
+import unittest
+from wetb.hawc2 import shearfile
+import numpy as np
+import os
+testfilepath = 'test_files/'
+class Test(unittest.TestCase):
+
+
+    def test_shearfile(self):
+        f = testfilepath + "tmp_shearfile.dat"
+        shearfile.save(f, [-55, 55], [30, 100, 160] , u=np.array([[0.7, 1, 1.3], [0.7, 1, 1.3]]).T)
+        with open(f) as fid:
+            self.assertEqual(fid.read(),
+""" # autogenerated shear file
+  2 3
+ # shear v component
+  0.0000000000 0.0000000000
+  0.0000000000 0.0000000000
+  0.0000000000 0.0000000000
+ # shear u component
+  0.7000000000 0.7000000000
+  1.0000000000 1.0000000000
+  1.3000000000 1.3000000000
+ # shear w component
+  0.0000000000 0.0000000000
+  0.0000000000 0.0000000000
+  0.0000000000 0.0000000000
+ # y coordinates
+  -55.0000000000
+  55.0000000000
+ # z coordinates
+  30.0000000000
+  100.0000000000
+  160.0000000000
+""")
+        os.remove(f)
+
+
+if __name__ == "__main__":
+    #import sys;sys.argv = ['', 'Test.test_shearfile']
+    unittest.main()