81 lines
3.2 KiB
Python
81 lines
3.2 KiB
Python
import numpy as np
|
|
|
|
|
|
def correction(value, lorentz=True, zebra_mode="--", ang1=0, ang2=0):
|
|
if lorentz is False:
|
|
return value
|
|
else:
|
|
if zebra_mode == "bi":
|
|
corr_value = np.abs(value * np.sin(ang1))
|
|
return corr_value
|
|
elif zebra_mode == "nb":
|
|
corr_value = np.abs(value * np.sin(ang1) * np.cos(ang2))
|
|
return corr_value
|
|
|
|
|
|
def export_comm(data, path, lorentz=False):
|
|
"""exports data in the *.comm format
|
|
:param lorentz: perform Lorentz correction
|
|
:param path: path to file + name
|
|
:arg data - data to export, is dict after peak fitting
|
|
|
|
"""
|
|
zebra_mode = data["meta"]["zebra_mode"]
|
|
align = ">"
|
|
if data["meta"]["indices"] == "hkl":
|
|
extension = ".comm"
|
|
padding = [6, 4, 10, 8]
|
|
elif data["meta"]["indices"] == "real":
|
|
extension = ".incomm"
|
|
padding = [4, 6, 10, 8]
|
|
|
|
with open(str(path + extension), "w") as out_file:
|
|
for key, scan in data["scan"].items():
|
|
if "fit" not in scan:
|
|
print("Scan skipped - no fit value for:", key)
|
|
continue
|
|
scan_number_str = f"{key:{align}{padding[0]}}"
|
|
h_str = f'{int(scan["h_index"]):{padding[1]}}'
|
|
k_str = f'{int(scan["k_index"]):{padding[1]}}'
|
|
l_str = f'{int(scan["l_index"]):{padding[1]}}'
|
|
if data["meta"]["area_method"] == "fit":
|
|
area = float(scan["fit"]["fit_area"].n)
|
|
sigma_str = (
|
|
f'{"{:8.2f}".format(float(scan["fit"]["fit_area"].s)):{align}{padding[2]}}'
|
|
)
|
|
elif data["meta"]["area_method"] == "integ":
|
|
area = float(scan["fit"]["int_area"].n)
|
|
sigma_str = (
|
|
f'{"{:8.2f}".format(float(scan["fit"]["int_area"].s)):{align}{padding[2]}}'
|
|
)
|
|
|
|
if zebra_mode == "bi":
|
|
area = correction(area, lorentz, zebra_mode, scan["twotheta_angle"])
|
|
int_str = f'{"{:8.2f}".format(area):{align}{padding[2]}}'
|
|
angle_str1 = f'{scan["twotheta_angle"]:{padding[3]}}'
|
|
angle_str2 = f'{scan["omega_angle"]:{padding[3]}}'
|
|
angle_str3 = f'{scan["chi_angle"]:{padding[3]}}'
|
|
angle_str4 = f'{scan["phi_angle"]:{padding[3]}}'
|
|
elif zebra_mode == "nb":
|
|
area = correction(area, lorentz, zebra_mode, scan["gamma_angle"], scan["nu_angle"])
|
|
int_str = f'{"{:8.2f}".format(area):{align}{padding[2]}}'
|
|
angle_str1 = f'{scan["gamma_angle"]:{padding[3]}}'
|
|
angle_str2 = f'{scan["omega_angle"]:{padding[3]}}'
|
|
angle_str3 = f'{scan["nu_angle"]:{padding[3]}}'
|
|
angle_str4 = f'{scan["unkwn_angle"]:{padding[3]}}'
|
|
|
|
line = (
|
|
scan_number_str
|
|
+ h_str
|
|
+ k_str
|
|
+ l_str
|
|
+ int_str
|
|
+ sigma_str
|
|
+ angle_str1
|
|
+ angle_str2
|
|
+ angle_str3
|
|
+ angle_str4
|
|
+ "\n"
|
|
)
|
|
out_file.write(line)
|