From bf97af1949e55a431c4606ccf6b9bfada5d5c9a8 Mon Sep 17 00:00:00 2001 From: Ivan Usov Date: Thu, 21 Apr 2022 11:53:19 +0200 Subject: [PATCH] Treat 2theta as gamma in geom files --- pyzebra/sxtal_refgen.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pyzebra/sxtal_refgen.py b/pyzebra/sxtal_refgen.py index 24e343a..a104fa3 100644 --- a/pyzebra/sxtal_refgen.py +++ b/pyzebra/sxtal_refgen.py @@ -108,6 +108,9 @@ def read_geom_file(fileobj): ang, ang_min, ang_max, ang_offset = line.split() ang_lims[ang.lower()] = [ang_min, ang_max, ang_offset] + if "2theta" in ang_lims: # treat 2theta as gamma + ang_lims["gamma"] = ang_lims.pop("2theta") + return ang_lims @@ -130,7 +133,11 @@ def export_geom_file(path, ang_lims, template=None): for _ in range(n_ang): next_line = next(template_file) ang, _, _, _ = next_line.split() + + if ang == "2theta": # treat 2theta as gamma + ang = "Gamma" vals = ang_lims[ang.lower()] + out_file.write(f"{'':<8}{ang:<10}{vals[0]:<10}{vals[1]:<10}{vals[2]:<10}\n")