zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

subplot in int_box Fix #2

Browse Source
This commit is contained in:
zaharko 2020-04-22 15:36:22 +02:00
parent 17a23ee7df
commit 891a620016
1 changed files with 24 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
pyzebra/xtal.py
View File

@ -1,6 +1,7 @@
import numpy as np import numpy as np
import math import math
from scipy.optimize import curve_fit from scipy.optimize import curve_fit
from matplotlib import pyplot as plt
import pyzebra import pyzebra
def z4frgn(wave,ga,nu): def z4frgn(wave,ga,nu):
@ -400,11 +401,15 @@ def box_int(file,box):
omP = omF + omStep*frStep omP = omF + omStep*frStep
Int = coeff[1]*abs(coeff[2]*omStep)*math.sqrt(2)*math.sqrt(np.pi) Int = coeff[1]*abs(coeff[2]*omStep)*math.sqrt(2)*math.sqrt(np.pi)
# omega plot # omega plot
x0 = range(len(cnts)) x_fit = np.linspace(0, len(cnts), 100)
y0 = cnts y_fit = gauss(x_fit, *coeff)
x0_fit = np.linspace(0, len(cnts), 100) plt.figure()
y0_fit = gauss(x0_fit, *coeff) plt.subplot(131)
plt.plot(range(len(cnts)), cnts)
plt.plot(x_fit, y_fit)
plt.ylabel('Intensity in the box')
plt.xlabel('Frame N of the box')
label='om'
# gamma fit # gamma fit
sliceXZ=np.zeros((nF,iX)) sliceXZ=np.zeros((nF,iX))
sliceYZ=np.zeros((nF,jY)) sliceYZ=np.zeros((nF,jY))
@ -421,11 +426,13 @@ def box_int(file,box):
coeff, var_matrix = curve_fit(gauss, range(len(projX)), projX, p0=p0) coeff, var_matrix = curve_fit(gauss, range(len(projX)), projX, p0=p0)
x= i0+coeff[1] x= i0+coeff[1]
# gamma plot # gamma plot
x1 = range(len(projX)) x_fit = np.linspace(0, len(projX), 100)
y1 = projX y_fit = gauss(x_fit, *coeff)
x1_fit = np.linspace(0, len(projX), 100) plt.subplot(132)
y1_fit = gauss(x1_fit, *coeff) plt.plot(range(len(projX)), projX)
plt.plot(x_fit, y_fit)
plt.ylabel('Intensity in the box')
plt.xlabel('X-pixel of the box')
# nu fit # nu fit
projY = np.sum(sliceYZ, axis=0) projY = np.sum(sliceYZ, axis=0)
@ -433,11 +440,13 @@ def box_int(file,box):
coeff, var_matrix = curve_fit(gauss, range(len(projY)), projY, p0=p0) coeff, var_matrix = curve_fit(gauss, range(len(projY)), projY, p0=p0)
y= j0+coeff[1] y= j0+coeff[1]
# nu plot # nu plot
x2 = range(len(projY)) x_fit = np.linspace(0, len(projY), 100)
y2 = projY y_fit = gauss(x_fit, *coeff)
x2_fit = np.linspace(0, len(projY), 100) plt.subplot(133)
y2_fit = gauss(x2_fit, *coeff) plt.plot(range(len(projY)), projY)
plt.plot(x_fit, y_fit)
plt.ylabel('Intensity in the box')
plt.xlabel('Y-pixel of the box')
ga, nu = pyzebra.det2pol(ddist,sttC,nuC,x,y) ga, nu = pyzebra.det2pol(ddist,sttC,nuC,x,y)