def update_figure(self):
global ImageDisplayed
self.ax0.cla()
self.axx.cla()
self.axy.cla()
self.ax0.imshow(ImageDisplayed,
aspect='auto',
cmap=beam_map,
origin='lower')
projx, projy = ImageDisplayed.sum(axis=0), ImageDisplayed.sum(axis=1)
x = np.arange(len(projx))
y = np.arange(len(projy))
if FLAG_fit > 0:
p2X = imgtl.FitProfile(projx, x)
p2Y = imgtl.FitProfile(projy, y)
# # top plot
self.axx.plot(projx)
if FLAG_fit > 0:
self.axx.plot(x, imgtl.dg(x, p2X), 'r--', linewidth=3)
self.axx.set_xlim(self.ax0.get_xlim())
# # Right plot
self.axy.plot(projy, range(len(projy)))
if FLAG_fit > 0:
self.axy.plot(imgtl.dg(y, p2Y), y, 'r--', linewidth=3)
self.axy.set_ylim(self.ax0.get_ylim())
# Remove tick labels
nullfmt = ticker.NullFormatter()
self.axx.xaxis.set_major_formatter(nullfmt)
self.axx.yaxis.set_major_formatter(nullfmt)
self.axy.xaxis.set_major_formatter(nullfmt)
self.axy.yaxis.set_major_formatter(nullfmt)
self.draw()
plt.plot (Wx0, stdy0,'go')
plt.subplot (2,2,3)
plt.plot (Wx0, correl0,'ro')
plt.subplot (2,2,4)
plt.plot (Wx0, norm0,'ro')
plt.figure()
plt.imshow(Xc)
print(np.shape(IMGf))
histx, histy, x, y = imgtl.GetImageProjection(IMGf,cal)
x=x-x[np.argmax(histx)]
y=y-y[np.argmax(histy)]
p2X= imgtl.FitProfile(histx, x)
p2Y= imgtl.FitProfile(histy, y)
print('RMS_x=',np.mean(stdx0[len(stdx0)-2:len(stdx0)+2]))
print('RMS_y=',np.mean(stdy0[len(stdy0)-2:len(stdy0)+2]))
print("fitX: ", p2X)
print("fitY: ", p2Y)
plt.figure()
disp=filenameIm.split('/')
print(disp)
print(len(disp))
plt.subplot (2,2,1)
plt.title (disp[len(disp)-1])
histY = np.loadtxt(filenameY)
# create the horizontal coordinate
axisX = np.arange(len(histX))
axisY = np.arange(len(histY))
# center the histogram and select a region of interest (ROI) around the peak
window = 1000.
histXc, axisXc = imgtl.center_and_ROI(histX, axisX, window)
histYc, axisYc = imgtl.center_and_ROI(histY, axisY, window)
# remove background (there are many way of doing this)
window = 10
histXc, axisXc = imgtl.removebackground(histXc, axisXc, window)
histYc, axisYc = imgtl.removebackground(histYc, axisYc, window)
p2X = imgtl.FitProfile(histXc, axisXc)
p2Y = imgtl.FitProfile(histYc, axisYc)
print("fitX: ", p2X)
print("fitY: ", p2Y)
plt.figure()
plt.plot(axisXc, histXc, 'ob', alpha=0.1)
plt.plot(axisXc, imgtl.dg(axisXc, p2X), '--b', linewidth=3)
plt.plot(axisYc, histYc, 'or', alpha=0.1)
plt.plot(axisYc, imgtl.dg(axisYc, p2Y), '--r', linewidth=3)
plt.title("point number: " + str(i))
plt.show()
