diff options
| -rw-r--r-- | Makefile.am | 3 | ||||
| -rwxr-xr-x | scripts/detector-shift | 118 | ||||
| -rwxr-xr-x | scripts/plot-detector-shift | 13 |
3 files changed, 133 insertions, 1 deletions
diff --git a/Makefile.am b/Makefile.am index aa479254..2744819e 100644 --- a/Makefile.am +++ b/Makefile.am @@ -187,7 +187,8 @@ script_DATA = scripts/alternate-stream scripts/cell-please \ scripts/gen-sfs-expand scripts/add-beam-params \ scripts/find-pairs scripts/plot-cc-and-scale.R \ scripts/ave-resolution scripts/crystal-frame-number \ - scripts/plot-radius-resolution scripts/plot-predict-refine + scripts/plot-radius-resolution scripts/plot-predict-refine \ + scripts/detector-shift scripts/plot-detector-shift EXTRA_DIST += $(script_DATA) diff --git a/scripts/detector-shift b/scripts/detector-shift new file mode 100755 index 00000000..77634f83 --- /dev/null +++ b/scripts/detector-shift @@ -0,0 +1,118 @@ +#!/usr/bin/env python + +# +# Determine mean detector shift based on prediction refinement results +# +# Copyright (c) 2015 Deutsches Elektronen-Synchrotron DESY, +# a research centre of the Helmholtz Association. +# +# Author: +# 2015 Thomas White <taw@physics.org> +# + +import sys +import os +import re + +f = open(sys.argv[1], 'r') +if len(sys.argv) > 2: + geom = sys.argv[2] + have_geom = 1 +else: + have_geom = 0 + +# Determine the mean shifts +total_x = 0 +total_y = 0 +total_z = 0 +n_xy = 0 +n_z = 0 + +prog1 = re.compile("^predict_refine/det_shift\sx\s=\s([0-9\.\-]+)\sy\s=\s([0-9\.\-]+)\smm$") +prog2 = re.compile("^predict_refine/clen_shift\s=\s([0-9\.\-]+)\smm$") + +while True: + + fline = f.readline() + if not fline: + break + + match = prog1.match(fline) + if match: + xshift = float(match.group(1)) + yshift = float(match.group(2)) + total_x += xshift + total_y += yshift + n_xy += 1 + + match = prog2.match(fline) + if match: + zshift = float(match.group(1)) + total_z += xshift + n_z += 1 + +if n_xy != n_z: + print 'Warning: number of xy shifts not equal to number of z shifts' + +mean_x = total_x / n_xy +mean_y = total_y / n_xy +mean_z = total_z / n_z + +print 'Mean shifts: dx = %.2f mm, dy = %.2f mm; dz = %.2f mm' % (mean_x,mean_y,mean_z) +f.close() + +if not have_geom: + exit(0) + +# Apply shifts to geometry +out = os.path.splitext(geom)[0]+'-predrefine.geom' +print 'Applying corrections to %s, output filename %s' % (geom,out) +g = open(geom, 'r') +h = open(out, 'w') + +prog1 = re.compile("^\s*res\s+=\s+([0-9\.]+)\s") +prog2 = re.compile("^\s*(.*)\/res\s+=\s+([0-9\.]+)\s") +prog3 = re.compile("^\s*(.*)\/corner_x\s+=\s+([0-9\.]+)\s") +prog4 = re.compile("^\s*(.*)\/corner_y\s+=\s+([0-9\.]+)\s") +default_res = 0 +while True: + + fline = g.readline() + if not fline: + break + + match = prog1.match(fline) + if match: + default_res = float(match.group(1)) + print 'default res %f' % (default_res) + h.write(fline) + continue + + match = prog2.match(fline) + if match: + panel = match.group(1) + panel_res = float(match.group(2)) + print 'panel res %s / %f' % (panel, panel_res) + h.write(fline) + continue + + match = prog3.match(fline) + if match: + panel = match.group(1) + panel_cnx = float(match.group(2)) + res = default_res # FIXME! + h.write('%s/corner_x = %f\n' % (panel,panel_cnx+(mean_x*res*1e-3))) + continue + + match = prog4.match(fline) + if match: + panel = match.group(1) + panel_cny = float(match.group(2)) + res = default_res # FIXME! + h.write('%s/corner_y = %f\n' % (panel,panel_cny+(mean_y*res*1e-3))) + continue + + h.write(fline) + +g.close() +h.close() diff --git a/scripts/plot-detector-shift b/scripts/plot-detector-shift new file mode 100755 index 00000000..f1933346 --- /dev/null +++ b/scripts/plot-detector-shift @@ -0,0 +1,13 @@ +#!/bin/sh + +INFILE=$1 + +grep "predict_refine/det_shift" $INFILE > plotme.dat +gnuplot -persist << EOF +set xlabel "x shift / mm" +set ylabel "y shift / mm" +set grid x2tics +set x2tics 10 lw 2 lc 0 +plot "plotme.dat" using 4:7 +replot 0 lw 2 lc 0 +EOF |