Spaces:
Sleeping
Sleeping
from threading import Lock | |
import fluidsynth | |
import numpy as np | |
class MidiSynthesizer: | |
def __init__(self, soundfont_path, sample_rate=44100): | |
self.soundfont_path = soundfont_path | |
self.sample_rate = sample_rate | |
fl = fluidsynth.Synth(samplerate=float(sample_rate)) | |
sfid = fl.sfload(soundfont_path) | |
self.devices = [[fl, sfid, False]] | |
self.file_lock = Lock() | |
def get_fluidsynth(self): | |
for device in self.devices: | |
if not device[2]: | |
device[2] = True | |
return device | |
with self.file_lock: | |
fl = fluidsynth.Synth(samplerate=float(self.sample_rate)) | |
sfid = fl.sfload(self.soundfont_path) | |
device = [fl, sfid, True] | |
self.devices.append(device) | |
return device | |
def release_fluidsynth(self, device): | |
device[0].system_reset() | |
device[0].get_samples(self.sample_rate*5) # wait for silence | |
device[2] = False | |
def synthesis(self, midi_opus): | |
ticks_per_beat = midi_opus[0] | |
event_list = [] | |
for track_idx, track in enumerate(midi_opus[1:]): | |
abs_t = 0 | |
for event in track: | |
abs_t += event[1] | |
event_new = [*event] | |
event_new[1] = abs_t | |
event_list.append(event_new) | |
event_list = sorted(event_list, key=lambda e: e[1]) | |
tempo = int((60 / 120) * 10 ** 6) # default 120 bpm | |
ss = np.empty((0, 2), dtype=np.int16) | |
device = self.get_fluidsynth() | |
fl, sfid = device[:-1] | |
last_t = 0 | |
for c in range(16): | |
fl.program_select(c, sfid, 128 if c == 9 else 0, 0) | |
for event in event_list: | |
name = event[0] | |
sample_len = int(((event[1] / ticks_per_beat) * tempo / (10 ** 6)) * self.sample_rate) | |
sample_len -= int(((last_t / ticks_per_beat) * tempo / (10 ** 6)) * self.sample_rate) | |
last_t = event[1] | |
if sample_len > 0: | |
sample = fl.get_samples(sample_len).reshape(sample_len, 2) | |
ss = np.concatenate([ss, sample]) | |
if name == "set_tempo": | |
tempo = event[2] | |
elif name == "patch_change": | |
c, p = event[2:4] | |
fl.program_select(c, sfid, 128 if c == 9 else 0, p) | |
elif name == "control_change": | |
c, cc, v = event[2:5] | |
fl.cc(c, cc, v) | |
elif name == "note_on" and event[3] > 0: | |
c, p, v = event[2:5] | |
fl.noteon(c, p, v) | |
elif name == "note_off" or (name == "note_on" and event[3] == 0): | |
c, p = event[2:4] | |
fl.noteoff(c, p) | |
self.release_fluidsynth(device) | |
if ss.shape[0] > 0: | |
max_val = np.abs(ss).max() | |
if max_val != 0: | |
ss = (ss / max_val) * np.iinfo(np.int16).max | |
ss = ss.astype(np.int16) | |
return ss | |