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

Allow export of geom file based on template

Browse Source
This commit is contained in:
usov_i 2022-04-12 15:11:42 +02:00
parent 94df4869d4
commit 39b2af60ca
1 changed files with 8 additions and 5 deletions

13
pyzebra/sxtal_refgen.py

@ -109,21 +109,24 @@ def read_geom_file(fileobj):
return ang_lims return ang_lims
def export_geom_file(path, ang_lims): def export_geom_file(path, ang_lims, template=None):
if "chi" in ang_lims: # BI geometry if "chi" in ang_lims: # BI geometry
default_file = get_zebraBI_default_geom_file() template_file = get_zebraBI_default_geom_file()
n_ang = 4 n_ang = 4
else: # NB geometry else: # NB geometry
default_file = get_zebraNB_default_geom_file() template_file = get_zebraNB_default_geom_file()
n_ang = 3 n_ang = 3
if template is not None:
template_file = template
with open(path, "w") as out_file: with open(path, "w") as out_file:
for line in default_file: for line in template_file:
out_file.write(line) out_file.write(line)
if line.startswith("ANG_LIMITS"): if line.startswith("ANG_LIMITS"):
for _ in range(n_ang): for _ in range(n_ang):
next_line = next(default_file) next_line = next(template_file)
ang, _, _, _ = next_line.split() ang, _, _, _ = next_line.split()
vals = ang_lims[ang.lower()] vals = ang_lims[ang.lower()]
out_file.write(f"{'':<8}{ang:<10}{vals[0]:<10}{vals[1]:<10}{vals[2]:<10}\n") out_file.write(f"{'':<8}{ang:<10}{vals[0]:<10}{vals[1]:<10}{vals[2]:<10}\n")