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

Save geometry in ang_lims

Browse Source
This commit is contained in:
usov_i 2022-04-12 16:09:22 +02:00
parent 39b2af60ca
commit e2220760f0
2 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyzebra/app/panel_ccl_prepare.py
View File

@ -34,10 +34,10 @@ def create():
def _update_ang_lims(ang_lims):
sttgamma_ti.value = " ".join(ang_lims["gamma"][:2])
omega_ti.value = " ".join(ang_lims["omega"][:2])
if len(ang_lims) == 3: # NB geom
if ang_lims["geom"] == "nb":
chinu_ti.value = " ".join(ang_lims["nu"][:2])
phi_ti.value = ""
else: # len(ang_lims) == 4, BI geom
else: # ang_lims["geom"] == "bi"
chinu_ti.value = " ".join(ang_lims["chi"][:2])
phi_ti.value = " ".join(ang_lims["phi"][:2])

20
pyzebra/sxtal_refgen.py
View File

@ -90,16 +90,20 @@ def get_zebra_default_cfl_file():
def read_geom_file(fileobj):
ang_lims = dict()
# locate angular limits in .geom text file
for line in fileobj:
if line.startswith("ANG_LIMITS"):
break
# read angular limits
for line in fileobj:
if "!" in line: # remove comments that start with ! sign
line, _ = line.split(sep="!", maxsplit=1)
if line.startswith("GEOM"):
_, val = line.split(maxsplit=1)
if val.startswith("2"):
ang_lims["geom"] = "bi"
else: # val.startswith("3")
ang_lims["geom"] = "nb"
elif line.startswith("ANG_LIMITS"):
# read angular limits
for line in fileobj:
if not line or line.isspace():
break
@ -110,10 +114,10 @@ def read_geom_file(fileobj):
def export_geom_file(path, ang_lims, template=None):
if "chi" in ang_lims: # BI geometry
if ang_lims["geom"] == "bi":
template_file = get_zebraBI_default_geom_file()
n_ang = 4
else: # NB geometry
else: # ang_lims["geom"] == "nb"
template_file = get_zebraNB_default_geom_file()
n_ang = 3