From 09a86f5d6f171d69214061a9233fe4bc0f22e571 Mon Sep 17 00:00:00 2001 From: Diana Date: Sun, 27 Oct 2024 12:36:32 +0100 Subject: [PATCH] dont remember --- code/plot_functions.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/code/plot_functions.py b/code/plot_functions.py index aa634b8..deccc9f 100644 --- a/code/plot_functions.py +++ b/code/plot_functions.py @@ -79,7 +79,7 @@ def float_formatter(x, _): return f'{x:.5f}' -def plot_highlighted_integrals(ax, frequency, power, points, color_mapping, points_categories, delta=2.5): +def plot_highlighted_integrals(ax, frequency, power, points, nyquist, true_eodf, color_mapping, points_categories, delta=2.5): """ Highlights integrals on the existing axes of the power spectrum for a given dataset. @@ -104,8 +104,6 @@ def plot_highlighted_integrals(ax, frequency, power, points, color_mapping, poin ------- None """ - _, _, _, df, eodf, nyquist, stim_freq = u.sam_data(sam) - AM = u.find_AM(eodf, nyquist, stim_freq) # Plot the power spectrum on the provided axes ax.plot(frequency, power, color="k") @@ -128,6 +126,7 @@ def plot_highlighted_integrals(ax, frequency, power, points, color_mapping, poin ax.set_xlim([0, 1200]) ax.set_ylim([0, 6e-5]) ax.axvline(nyquist, color = "k", linestyle = "--") + ax.axvline(true_eodf, color = "k", linestyle = "--") ax.set_xlabel('Frequency (Hz)') ax.set_ylabel('Power') ax.set_title('Power Spectrum with Highlighted Integrals')