m

constellations.py

@@ -1,6 +1,7 @@
 import numpy as np
 from scipy.io.wavfile import read
 from scipy import signal
+from fft_just_do_it import *
 
 def create_constellation(audio, Fs, chunk_size=1000):
     window_length_seconds = 0.05
@@ -17,10 +18,9 @@ def create_constellation(audio, Fs, chunk_size=1000):
         amount_to_pad = window_length_samples - chunk.size % window_length_samples
         padded_chunk = np.pad(chunk, (0, amount_to_pad))
 
-        # Зменшуємо розмір вікна nperseg
         nperseg = min(window_length_samples, len(padded_chunk))
 
-        frequencies, times, stft_result = signal.stft(
+        frequencies, times, stft_result = stft(
             padded_chunk, Fs, nperseg=nperseg, nfft=nperseg, return_onesided=True
         )
 
@@ -48,6 +48,6 @@ def create_constellation(audio, Fs, chunk_size=1000):
     return constellation_map
 
 if __name__ == "__main__":
-    Fs, audio_input = read("Rain Over Me.wav")
+    Fs, audio_input = read("data/Rain Over Me.wav")
     constellation = create_constellation(audio_input, Fs)
     print(constellation)

finding.py

@@ -3,7 +3,7 @@
 from constellations import create_constellation
 from hashes import create_hashes
 
-Fs, audio_input = read("data/Rain Over Me.wav")
+Fs, audio_input = read("video.wav")
 
 constellation = create_constellation(audio_input, Fs)
 hashes = create_hashes(constellation, None)
@@ -44,5 +44,5 @@ def score_songs(hashes):
 
 scores = score_songs(hashes)
 for song_index, score in scores:
-    print(f"{song_index_lookup[song_index]=}: Score of {score[1]} at {score[0]}")
+    print(f"{song_index_lookup[song_index]=}")
 

hashes.py

@@ -7,16 +7,11 @@
 
 def create_hashes(constellation_map, song_id=None):
     hashes = {}
-    for idx, (time, freq) in enumerate(constellation_map):
-        for other_time, other_freq in constellation_map[idx : idx + 100]:
+    for _, (time, freq) in enumerate(constellation_map):
+        for other_time, other_freq in constellation_map:
             diff = other_time - time
-            if diff <= 1 or diff > 10:
-                continue
             freq_binned = freq / upper_frequency * (2 ** frequency_bits)
             other_freq_binned = other_freq / upper_frequency * (2 ** frequency_bits)
             hash = int(freq_binned) | (int(other_freq_binned) << 10) | (int(diff) << 20)
             hashes[hash] = (time, song_id)
     return hashes
-
-if __name__ == "__main__":
-    print(create_hashes([1,2,3], 2))

tempCodeRunnerFile.py

@@ -1,3 +1,8 @@
-# Print the song index
-for index, filename in database.items():
-    print(f"Index: {index}, Filename: {filename}")
+import pickle
+
+# Open the pickle file in binary read mode
+with open("database.pickle", "rb") as db_file:
+    # Load the contents of the pickle file
+    database = pickle.load(db_file)
+
+print(database)