diff --git a/plotting.py b/plotting.py index 0fa809b7f7022f555e4fe0787bdc32a6835a554d..479a694b7d2a022e05129cd964dc34ef1aad94cc 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):