From 20effc2fcff2656633e5b899029cb3c65a235358 Mon Sep 17 00:00:00 2001 From: gac-maloja Date: Tue, 25 Oct 2022 18:54:02 +0200 Subject: [PATCH] added selecting the correct key for anti-parallel +/-; added math import --- phases/models/antiparallel_model.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/phases/models/antiparallel_model.py b/phases/models/antiparallel_model.py index c2903cd..7214da5 100644 --- a/phases/models/antiparallel_model.py +++ b/phases/models/antiparallel_model.py @@ -1,3 +1,4 @@ +import math import numpy as np from scipy.optimize import fsolve @@ -29,6 +30,8 @@ def K2gap(Kval, fitparam): def antiparallel_rad2k_fit(radial, phi_in_degree, undudict): + key = "anti-parallel-" if phi_in_degree < 0 else "anti-parallel+" + undudict = undudict[key]["K-value"] phi = math.radians(phi_in_degree) amp1 = getfit(radial, undudict['fitpars_amp1']) amp2 = getfit(radial, undudict['fitpars_amp2']) @@ -52,6 +55,8 @@ def getfit(val, fitparam): def antiparallel_k2rad_fit(Kval, phi_in_degree, undudict): + key = "anti-parallel-" if phi_in_degree < 0 else "anti-parallel+" + undudict = undudict[key]["K-value"] phi = math.radians(phi_in_degree) amp1 = getfit(Kval, undudict['fitpars_amp1']) amp2 = getfit(Kval, undudict['fitpars_amp2'])