Octave-fold generated notes instead of dropping out-of-range events

engines.py

@@ -8,7 +8,7 @@ from typing import List, Dict, Any
 
 from midi_model import (
     count_out_of_range_events,
-    filter_events_to_keyboard_range,
+    fold_events_to_keyboard_range,
     get_model,
 )
 
@@ -182,8 +182,8 @@ class GodzillaContinuationEngine:
         )
         out_of_range = count_out_of_range_events(new_events)
         if out_of_range:
-            print(f"Godzilla: dropped {out_of_range} out-of-range events")
-        return filter_events_to_keyboard_range(new_events)
+            print(f"Godzilla: remapped {out_of_range} out-of-range events by octave folding")
+        return fold_events_to_keyboard_range(new_events)
 
 
 # =============================================================================

midi_model.py

@@ -346,6 +346,26 @@ def filter_events_to_keyboard_range(events: list[dict]) -> list[dict]:
     ]
 
 
+def fold_note_to_keyboard_range(note: int) -> int:
+    min_note, max_note = keyboard_note_range()
+    folded = int(note)
+    while folded < min_note:
+        folded += 12
+    while folded > max_note:
+        folded -= 12
+    return max(min_note, min(max_note, folded))
+
+
+def fold_events_to_keyboard_range(events: list[dict]) -> list[dict]:
+    out: list[dict] = []
+    for event in events:
+        copied = dict(event)
+        if copied.get("type") in {"note_on", "note_off"}:
+            copied["note"] = fold_note_to_keyboard_range(int(copied.get("note", 0)))
+        out.append(copied)
+    return out
+
+
 def build_prime_tokens(score_tokens: list[int], seq_len: int) -> list[int]:
     prime = [705, 384, 706]
     if score_tokens: