def push_2():
AllItems = [filebox.itemText(i) for i in range(filebox.count())]
for file in AllItems:
data = detect.load(file)
try:
spectrum = Spectrum()
app.spectrums.append(spectrum)
spectrum.name = file
filename = os.path.basename(file)
if file.endswith('.data'):
file = file.replace('.data', '.input')
elif file.endswith('.txt'):
file = file.replace('.txt', '.input')
elif file.endswith('.undata'):
file = file.replace('.undata', '.input')
elif file.endswith('.npy'):
file = file.replace('.npy', '.input')
else:
file = file + '.input'
spectrum.safio = safari_input.SafariInput(file)
spectrum.run(data)
spectrum.popup.setWindowTitle(file)
except Exception as e:
print(e)
pass
#Wait a second for it to start running
time.sleep(1)
def e_theta_loop(dir, theta1, theta2, theta_step):
if dir != '.':
dir = os.path.join('.',dir)
for filename in os.listdir(dir):
if filename.endswith('.data'):
file = os.path.join(dir, filename)
safio = safari_input.SafariInput(file.replace('.data', '.input'))
print('loading data')
data = detect.load(file)
print('data loaded')
fig, ax = plt.subplots()
num = 0
for theta in frange(theta1, theta2, theta_step):
print('Theta: '+str(theta))
spectrum = detect.Spectrum()
spectrum.plots = False
spectrum.name = filename.replace('.data','')
spectrum.safio = safio
spectrum.safio.DTECTPAR[0] = theta
spectrum.detector = None
spectrum.clean(data)
energy, intensity = spectrum.detector.spectrum(res=spectrum.safio.ESIZE)
intensity = intensity + num
ax.plot(energy, intensity, label=str(theta))
num = num + 1
spectrum = None
ax.legend()
ax.set_title("Intensity vs Energy")
ax.set_xlabel('Energy (eV)')
ax.set_ylabel('Intensity')
fig.show()
input('Press Enter to exit')
def azimuthal_spectrum(dir, theta, size=3, emin=0, emin_rel=0):
if dir != '.':
dir = os.path.join('.',dir)
output = open(os.path.join(dir, "azimuthal_spectrum_{}_{}_{}.txt".format(theta, size, emin)), 'w');
for filename in os.listdir(dir):
if filename.endswith('.data'):
file = os.path.join(dir, filename)
safio = safari_input.SafariInput(file.replace('.data', '.input'))
print('loading: '+filename)
data = detect.load(file)
print('data loaded')
# Setup the spectrum object for this file
spectrum = detect.Spectrum()
spectrum.plots = False
spectrum.name = ""
spectrum.pics = False
spectrum.safio = safio
spectrum.safio.DTECTPAR[0] = theta
spectrum.detector = None
spectrum.detector = detect.SpotDetector(theta, safio.PHI0, size)
if emin_rel!=0:
emin = emin_rel * safio.E0
spectrum.clean(data, emin=emin)
output.write("{}\t{}\n".format(safio.PHI0, len(spectrum.detector.detections)));
output.close();
input('Press Enter to exit')
def push():
data = detect.load(filebox.currentText())
try:
app.spectrums.append(spectrum)
file = filebox.currentText()
spectrum.name = file
filename = os.path.basename(file)
if file.endswith('.data'):
file = file.replace('.data', '.input')
elif file.endswith('.txt'):
file = file.replace('.txt', '.input')
elif file.endswith('.undata'):
file = file.replace('.undata', '.input')
elif file.endswith('.npy'):
file = file.replace('.npy', '.input')
else:
file = file + '.input'
spectrum.safio = safari_input.SafariInput(file)
spectrum.run(data)
spectrum.popup.setWindowTitle(file)
except Exception as e:
print(e)
pass