aboutsummaryrefslogtreecommitdiff
path: root/scripts/split-by-mask
diff options
context:
space:
mode:
authorThomas White <taw@physics.org>2016-08-03 17:54:42 +0200
committerThomas White <taw@physics.org>2016-08-03 17:57:28 +0200
commit5b64b3dd7d846fbb26f79ab993994cf1570fe187 (patch)
treefba381ef875f326a2046222ea99c8b72de90e787 /scripts/split-by-mask
parent001705c484e08d1fd8b412f997ba014d4e9adfe3 (diff)
Add scripts/move-entire-detector and scripts/split-by-mask
Diffstat (limited to 'scripts/split-by-mask')
-rwxr-xr-xscripts/split-by-mask87
1 files changed, 87 insertions, 0 deletions
diff --git a/scripts/split-by-mask b/scripts/split-by-mask
new file mode 100755
index 00000000..aede3f43
--- /dev/null
+++ b/scripts/split-by-mask
@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+
+#
+# Split stream according to an external mask
+#
+# Copyright (c) 2014-2016 Deutsches Elektronen-Synchrotron DESY,
+# a research centre of the Helmholtz Association.
+#
+# Author:
+# 2014-2016 Thomas White <taw@physics.org>
+#
+
+import sys
+import optparse
+from collections import deque
+import h5py
+import re
+
+prog = re.compile("^\s+([\d-]+)\s+([\d-]+)\s+([\d-]+)\s+[\d\.\-]+\s+[\d\.\-]+\s+[\d\.\-]+\s+[\d\.\-]+\s+([\d\.]+)\s+([\d\.]+)")
+
+def process_refln(fline, g, h, mask):
+ match = prog.match(fline)
+ if not match:
+ print 'Line not understood, WTF? %s' % fline
+ return
+ fs = int(round(float(match.group(4))))
+ ss = int(round(float(match.group(5))))
+ m = mask[ss][fs]
+ if m:
+ g.write(fline)
+ else:
+ h.write(fline)
+
+op = optparse.OptionParser()
+
+op.add_option('', '--input', action='store', type='string', dest='ifn',
+ help="Input stream")
+
+op.add_option('', '--output1', action='store', type='string', dest='ofn1',
+ help="Output stream", default='mask-split.stream1')
+
+op.add_option('', '--output2', action='store', type='string', dest='ofn2',
+ help="Output stream", default='mask-split.stream2')
+
+op.add_option('', '--mask', action='store', type='string', dest='mask_fn',
+ help="Filename of mask", default='mask.h5')
+
+opt,arg = op.parse_args(sys.argv)
+
+if not opt.ifn:
+ print "You need at least --input"
+ exit(1)
+
+f = open(opt.ifn, 'r')
+g = open(opt.ofn1, 'w')
+h = open(opt.ofn2, 'w')
+
+fh = h5py.File(opt.mask_fn, 'r')
+mask = fh['/data/data'].value
+fh.close()
+
+in_refl = 0
+while True:
+
+ fline = f.readline()
+ if not fline:
+ break
+ if fline.find("End of reflections") != -1:
+ in_refl = 0
+
+ if in_refl == 1:
+ process_refln(fline, g, h, mask)
+ else:
+ g.write(fline)
+ h.write(fline)
+ if in_refl == 2:
+ in_refl = 1
+
+ if fline.find("Reflections measured after indexing") != -1:
+ in_refl = 2
+
+
+
+
+f.close()
+g.close()
+