Changed loop.py to chache spectrograms

diff --git a/loop.py b/loop.py
index e0c534ad0e37bbef24ba737842d8deadebd33621..34110ed7be30b93862ed32e86080afa84fa8bd82 100755 (executable)
--- a/loop.py
+++ b/loop.py
@@ -90,6 +90,8 @@ def process_loop(
        print("caching data...")
        caching = True
 
+       all_spectrums = np.zeros((segment_count, window_height, window_size, 3), dtype=np.uint8)
+
        for segment_index in tqdm(range(segment_count)):
                # get the specturm of the current sample
                segment_start = segment_index * segment_samples
@@ -108,7 +110,8 @@ def process_loop(
                if segment_index == 10:
                        cv.imwrite("spectrum_sample.jpg", spectrum)
 
-               np.save(f"cache/frame{segment_index}.npy", spectrum)
+               all_spectrums[segment_index] = spectrum
+               #np.save(f"cache/frame{segment_index}.npy", spectrum)
 
        segment_index = 0
        print("cached!")
@@ -116,7 +119,7 @@ def process_loop(
 
        while segment_index < segment_count:
 
-               spectrum = np.load(f"cache/frame{segment_index}.npy")
+               spectrum = all_spectrums[segment_index]#np.load(f"cache/frame{segment_index}.npy")
 
                # display and capture
                camera.display(spectrum)
@@ -171,9 +174,12 @@ notes:
 
 # define parameters
 sample_rate = 22_050
-window_size = 150
+# 64 81 100 121 144 169 192 225
+window_size = 144
 window_height = 80
 
+hop_size = window_size // 2
+
 caching = False
 
 if __name__ == "__main__":