add time tracker

app.py

@@ -10,6 +10,11 @@ import numpy as np
 
 def select_note(note="C4", context=None):
     context["note"] = note
+    time_index = context.get("time_index", None)
+    if time_index is not None:
+        context["time_index"] += 1
+    else:
+        context["time_index"] = 0
 
 
 def create_note(context={}):
@@ -56,12 +61,6 @@ def get_color(note, size=(256, 256)):
     index = list(NOTE_FREQUENCIES.keys()).index(note)
     color = colors.get(list(colors.keys())[index], [0., 0., 0.])
     img = np.ones((size[1], size[0], 3)) * np.array(color)[None, None, :]
-    border_size = 4
-    border_color = (0.2, 0.2, 0.2)
-    img[:border_size, :] = border_color
-    img[-border_size:, :] = border_color
-    img[:, :border_size] = border_color
-    img[:, -border_size:] = border_color
     text = NOTES_TRANSLATION[index].upper()
     font_scale = size[0] // 64
     thickness = 2
@@ -78,9 +77,18 @@ def get_color(note, size=(256, 256)):
     return img
 
 
+def add_border(img, border_size=10, color=(0.2, 0.2, 0.2)):
+    border_size = 4
+    img[:border_size, :] = color
+    img[-border_size:, :] = color
+    img[:, :border_size] = color
+    img[:, -border_size:] = color
+
+
 def xylo_player():
-    full_song = song_player()
     select_note()
+    select_song()
+    full_song = song_player()
     audio = create_note()
     play_note(audio)
     out_image = display_color()
@@ -91,28 +99,44 @@ SONG_1 = "au clair de la lune"
 SONG_2 = "ainsi font, font, font"
 
 
-def song_player(song: str=SONG_1, context={}):
+def select_song(song: str = SONG_1, context={}):
+    previous_song = context.get("song", None)
+    if previous_song != song:
+        # reset time index
+        context["time_index"] = 0
+    context["song"] = song
+
+
+def song_player(context={}):
+    song = context["song"]
     SONG = {
         SONG_1: "fa fa fa sol la - sol - la sol sol fa - fa fa fa sol la",
         SONG_2: "mi mi do - mi - sol - sol - la sol fa mi re mi do mi mi do - mi - sol - sol - la sol fa mi re do"
     }
     song_str = SONG.get(song, "")
-    return generate_song(song_str)
+    return generate_song(song_str, current_time=context.get("time_index", 0))
 
 
-def generate_song(song_str):
+def generate_song(song_str, current_time=None):
     notes = song_str.split(" ")
     all_notes = []
     size = (64, 128)
-    for note in notes:
+    index_no_silence = -1
+    for idx, note in enumerate(notes):
         if note in ["-", "."]:
             img_note = np.zeros((size[1], size[0], 3))
+            color = (0.2, 0.2, 0.2)
         else:
             note_classic = NOTES_CORRESPONDANCE.get(note, None)
+            index_no_silence += 1
             if note_classic is None:
                 print(f"Note {note} not found")
                 continue
+
             img_note = get_color(note_classic, size=size)
+            color = (0.8, 0., 0.) if current_time == index_no_silence else (
+                0.2, 0.2, 0.2)
+        add_border(img_note, color=color)
         all_notes.append(img_note)
 
     # out_image = np.hstack(all_notes)
@@ -120,8 +144,6 @@ def generate_song(song_str):
     remainder = max_notes_per_line - len(all_notes) % max_notes_per_line
     for _ in range(remainder):
         all_notes.append(np.zeros_like(all_notes[0]))
-    # out_image = np.hstack(all_notes)
-    # print(len(all_notes))
     note_lines = [all_notes[i:i + max_notes_per_line]
                   for i in range(0, len(all_notes), max_notes_per_line)]
     out_image = np.vstack([np.hstack(line) for line in note_lines])
@@ -139,7 +161,7 @@ if __name__ == '__main__':
         img = get_color(note, size=(512, 512))
         Image.save_image(img, icon)
     interactive(note=Control("C4", all_notes, icons=icon_list))(select_note)
-    interactive(song=(SONG_1, [SONG_1, SONG_2]))(song_player)
+    interactive(song=(SONG_1, [SONG_1, SONG_2]))(select_song)
     interactive_pipeline(
         gui=args.backend,
         cache=False,