aboutsummaryrefslogtreecommitdiff
path: root/scripts/detector-shift
diff options
context:
space:
mode:
authorThomas White <taw@physics.org>2016-03-22 14:30:48 +0100
committerThomas White <taw@physics.org>2016-03-22 14:30:48 +0100
commitd21e4d951b05db9e5d7e930c505d81c77772d5cf (patch)
tree63ac45e45eceb618b0f054abc901188b739faf6a /scripts/detector-shift
parentdd243f91a272f8f8240bc7e54d9291c30920e1de (diff)
scripts/detector-shift: Show a 'heat map' histogram of detector offset
Diffstat (limited to 'scripts/detector-shift')
-rwxr-xr-xscripts/detector-shift21
1 files changed, 18 insertions, 3 deletions
diff --git a/scripts/detector-shift b/scripts/detector-shift
index 223ba5b2..a348e3c9 100755
--- a/scripts/detector-shift
+++ b/scripts/detector-shift
@@ -8,11 +8,13 @@
#
# Author:
# 2015-2016 Thomas White <taw@physics.org>
+# 2016 Marmoru Suzuki <mamoru.suzuki@protein.osaka-u.ac.jp>
#
import sys
import os
import re
+import numpy as np
import matplotlib.pyplot as plt
f = open(sys.argv[1], 'r')
@@ -118,12 +120,25 @@ if have_geom:
g.close()
h.close()
-plt.plot(x_shifts, y_shifts, 'rx')
-plt.plot(0, 0, 'bo')
-plt.axis([-2,2,-2,2])
+nbins = 200
+H, xedges, yedges = np.histogram2d(x_shifts,y_shifts,bins=nbins)
+H = np.rot90(H)
+H = np.flipud(H)
+Hmasked = np.ma.masked_where(H==0,H)
+
+# Plot 2D histogram using pcolor
+fig2 = plt.figure()
+plt.pcolormesh(xedges,yedges,Hmasked)
plt.title('Detector shifts according to prediction refinement')
plt.xlabel('x shift / mm')
plt.ylabel('y shift / mm')
+circle1 = plt.Circle((mean_x,mean_y),.1,color='r',fill=False)
+fig = plt.gcf()
+fig.gca().add_artist(circle1)
+cbar = plt.colorbar()
+cbar.ax.set_ylabel('Counts')
+plt.plot(0, 0, 'bH', color='c')
+plt.plot(mean_x, mean_y, 'b8', color='m')
plt.grid(True)
plt.show()