Commit 51d0dbcd authored by Daniel Brown's avatar Daniel Brown
Browse files

adding setRTL method for BS and mirror, fixing db plotting with out.plot

parent 3c94b18d
Pipeline #1608 passed with stage
in 18 seconds
......@@ -334,7 +334,12 @@ class AbstractMirrorComponent(Component):
self.__Fs1 = Param("s1", self, None, canFsig=True, isPutable=False, isPutter=False, isTunable=False, fsig_name="s1")
self._default_fsig_param = self.__phi
def setRTL(self, R=None, T=None, L=None):
if R is not None: self.R.value = R
if T is not None: self.T.value = T
if L is not None: self.L.value = L
@property
def z(self): return self.__z
@property
......
......@@ -331,6 +331,8 @@ class katRun(object):
if kat.noxaxis == True:
raise pkex.BasePyKatException("This kat object has noxaxis=True, so there is nothing to plot.")
original_yaxis = kat.yaxis
if yaxis is not None:
kat.yaxis = yaxis
......@@ -367,7 +369,11 @@ class katRun(object):
dual_plot = True
elif "db:deg" in kat.yaxis:
_func1 = lambda x: 10*np.log10(x)
if "db" not in original_yaxis:
_func1 = lambda x: 10*np.log10(x)
else:
_func1 = lambda x: x
_func2 = lambda x: np.rad2deg(np.angle(x))
plot_cmd1 = plot_cmd
......@@ -379,7 +385,11 @@ class katRun(object):
_func1 = np.real
plot_cmd1 = plot_cmd
elif "db" in kat.yaxis:
_func1 = lambda x: 10*np.log10(x)
if "db" not in original_yaxis:
_func1 = lambda x: 10*np.log10(x)
else:
_func1 = lambda x: x
plot_cmd1 = plot_cmd
elif "deg" in kat.yaxis:
_func1 = lambda x: np.rad2deg(np.angle(x))
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment