From b85f27959d807a43804e9afb80d34019ce83be21 Mon Sep 17 00:00:00 2001
From: Philipp Niedermayer <p.niedermayer@gsi.de>
Date: Tue, 24 May 2022 11:19:10 +0200
Subject: [PATCH] Return spectrum
---
plotting.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/plotting.py b/plotting.py
index 0fa809b..479a694 100644
--- a/plotting.py
+++ b/plotting.py
@@ -158,11 +158,14 @@ def plot_tune_spectrum(ax, libera_data, xy, turn_range=None, time_range=None, tu
q = SI.Measurement(fitr[0][2], (fitr[1][2]**2+fitr[0][3]**2+fitr[1][3]**2)**0.5, '') # conservative estimate of error including width of peak
ax.plot(*fitr[-1], '--', label=f'Fit $q_{xy}={q:~L}$', zorder=50)
return q
+
+ return freq, mag, phase
-
+
+
def plot_tune_spectrogram(ax, libera_data, xy, *, nperseg=2**12, noverlap=None, ninterpol=None, over_time=True, colorbar=False,
turn_range=None, time_range=None, tune_range=None, excitation=None, vmin=None, vmax=None,
cmap='gist_heat_r', bunches=None, show_nperseg=True):
--
GitLab