From 2e2e79f5fe4f84cbdd6efd7eaac72d2cfe8ce637 Mon Sep 17 00:00:00 2001
From: Diana <diana.stoll@stundent.uni-tuebingen.de>
Date: Fri, 25 Oct 2024 18:10:57 +0200
Subject: [PATCH] no idea

---
 code/useful_functions.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/code/useful_functions.py b/code/useful_functions.py
index 6d1c854..1b34361 100644
--- a/code/useful_functions.py
+++ b/code/useful_functions.py
@@ -248,23 +248,23 @@ def extract_stim_data(stimulus):
     stim_freq = round(stimulus.metadata[stimulus.name]['Frequency'][0][0])
     stim_dur = stimulus.duration
     # calculates the amplitude modulation
-    amp_mod, ny_freq = AM(eodf, stim_freq)
+    _, ny_freq = AM(eodf, stim_freq)
+    amp_mod = find_AM(eodf, ny_freq, stim_freq)
     return amplitude, df, eodf, stim_freq, stim_dur, amp_mod, ny_freq
 
-def find_AM(frequency, eodf, nyquist, stimulus_frequency):
+def find_AM(eodf, nyquist, stimulus_frequency):
     t = signal.windows.triang(eodf) * nyquist
-    length_t2 = len(frequency) // eodf
-    length_t2 = int(length_t2)
+    length_t2 = int(eodf*10)
     t2 = np.tile(t, length_t2)
     x_values = np.arange(len(t2))
     
-    fig, ax = plt.subplots()
-    ax.plot(t2)
-    ax.scatter(stimulus_frequency, t2[np.argmin(np.abs(x_values - stimulus_frequency))])
-    plt.grid()
+    #fig, ax = plt.subplots()
+    #ax.plot(t2)
+    #ax.scatter(stimulus_frequency, t2[np.argmin(np.abs(x_values - stimulus_frequency))])
+    #plt.grid()
 
     AM = t2[np.argmin(np.abs(x_values - stimulus_frequency))]
-    return fig, AM
+    return AM
 
 def firing_rate(binary_spikes, dt = 0.000025, box_width = 0.01):
     '''