Update app.py

app.py

@@ -18,11 +18,122 @@ import tqdm
 from midi_to_colab_audio import midi_to_colab_audio
 import TMIDIX
 
+import numpy as np
+from scipy.interpolate import make_interp_spline
 import matplotlib.pyplot as plt
     
 # =================================================================================================
 
-def read_MIDI(input_midi)
+def hsv_to_rgb(h, s, v):
+    if s == 0.0:
+        return v, v, v
+    i = int(h*6.0)
+    f = (h*6.0) - i
+    p = v*(1.0 - s)
+    q = v*(1.0 - s*f)
+    t = v*(1.0 - s*(1.0-f))
+    i = i%6
+    return [(v, t, p), (q, v, p), (p, v, t), (p, q, v), (t, p, v), (v, p, q)][i]
+
+def generate_colors(n):
+    return [hsv_to_rgb(i/n, 1, 1) for i in range(n)]
+
+def add_arrays(a, b):
+    return [sum(pair) for pair in zip(a, b)]
+
+def plot_ms_SONG(ms_song,
+                  preview_length_in_notes=0,
+                  block_lines_times_list = None,
+                  plot_title='ms Song',
+                  max_num_colors=129,
+                  drums_color_num=128,
+                  plot_size=(11,4),
+                  note_height = 0.75,
+                  show_grid_lines=False,
+                  return_plt = False,
+                  timings_multiplier=1,
+                  plot_curve_values=None,
+                  save_plot=''
+                  ):
+
+  '''Tegridy ms SONG plotter/vizualizer'''
+
+  notes = [s for s in ms_song if s[0] == 'note']
+
+  if (len(max(notes, key=len)) != 7) and (len(min(notes, key=len)) != 7):
+    print('The song notes do not have patches information')
+    print('Please add patches to the notes in the song')
+
+  else:
+
+    start_times = [(s[1] * timings_multiplier) / 1000 for s in notes]
+    durations = [(s[2]  * timings_multiplier) / 1000 for s in notes]
+    pitches = [s[4] for s in notes]
+    patches = [s[6] for s in notes]
+
+    colors = generate_colors(max_num_colors)
+    colors[drums_color_num] = (1, 1, 1)
+
+    pbl = (notes[preview_length_in_notes][1] * timings_multiplier) / 1000
+
+    fig, ax = plt.subplots(figsize=plot_size)
+
+    # Create a rectangle for each note with color based on patch number
+    for start, duration, pitch, patch in zip(start_times, durations, pitches, patches):
+        rect = plt.Rectangle((start, pitch), duration, note_height, facecolor=colors[patch])
+        ax.add_patch(rect)
+
+    if plot_curve_values is not None:
+
+      min_val = min(plot_curve_values)
+      max_val = max(plot_curve_values)
+      spcva = [((value - min_val) / (max_val - min_val)) * 100 for value in plot_curve_values]
+
+
+      mult = int(math.ceil(max(add_arrays(start_times, durations)) / len(spcva)))
+      pcv = [value for value in spcva for _ in range(mult)][:int(max(add_arrays(start_times, durations)))+mult]
+      
+      x = np.arange(len(pcv))
+      x_smooth = np.linspace(x.min(), x.max(), 300)
+      spl = make_interp_spline(x, pcv, k=3)
+      y_smooth = spl(x_smooth)
+      ax.plot(x_smooth, y_smooth, color='white')
+
+    # Set the limits of the plot
+    ax.set_xlim([min(start_times), max(add_arrays(start_times, durations))])
+    ax.set_ylim([min(y_smooth), max(y_smooth)])
+
+    # Set the background color to black
+    ax.set_facecolor('black')
+    fig.patch.set_facecolor('white')
+
+    if preview_length_in_notes > 0:
+      ax.axvline(x=pbl, c='white')
+
+    if block_lines_times_list:
+      for bl in block_lines_times_list:
+        ax.axvline(x=bl, c='white')
+
+    if show_grid_lines:
+      ax.grid(color='white')
+
+    plt.xlabel('Time (s)', c='black')
+    plt.ylabel('MIDI Pitch', c='black')
+
+    plt.title(plot_title)
+
+    if return_plt:
+      return fig
+
+    if save_plot == '':
+      plt.show()
+
+    else:
+      plt.savefig(save_plot)
+
+# =================================================================================================
+
+def read_MIDI(input_midi):
 
     #===============================================================================
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi)