def makeNote(track, pitch, velocity):
e = midi.MidiEvent(track,
type=midi.ChannelVoiceMessages.NOTE_ON,
channel=1)
e.pitch = int(pitch)
e.velocity = int(velocity)
return e
def populate_midi_track_from_data(mt, data):
t = 0
tLast = 0
for d, p, v in data:
dt = midi.DeltaTime(mt)
dt.time = t - tLast
# add to track events
mt.events.append(dt)
me = midi.MidiEvent(mt)
me.type = "NOTE_ON"
me.channel = 1
me.time = None # d
me.pitch = p
me.velocity = v
mt.events.append(me)
# add note off / velocity zero message
dt = midi.DeltaTime(mt)
dt.time = d
# add to track events
mt.events.append(dt)
me = midi.MidiEvent(mt)
me.type = "NOTE_OFF"
me.channel = 1
me.time = None # d
me.pitch = p
me.velocity = 0
mt.events.append(me)
tLast = t + d # have delta to note off
t += d # next time
# add end of track
dt = midi.DeltaTime(mt)
dt.time = 0
mt.events.append(dt)
me = midi.MidiEvent(mt)
me.type = "END_OF_TRACK"
me.channel = 1
me.data = '' # must set data to empty string
mt.events.append(me)
return mt
def noteWrite(miditrack, on, pitch, velocity, channel=1):
me = midi.MidiEvent(miditrack)
if bool(on) == True:
me.type = 'NOTE_ON'
elif bool(on) == False:
me.type = 'NOTE_OFF'
me.channel = channel
me.pitch = pitch
me.velocity = velocity
miditrack.events.append(me)
def process_midi(self, message, deltatime):
self._wallclock += deltatime
status, note, velocity = message
dt = midi.DeltaTime(self.track)
dt.time = deltatime * 1000
self.track.events.append(dt)
event = midi.MidiEvent(self.track)
event.channel = 1
event.pitch = note
event.velocity = velocity
if status & 0xF0 == NOTE_ON:
event.type = "NOTE_ON"
self.track.events.append(event)
else:
event.type = "NOTE_OFF"
self.track.events.append(event)
stream = midi.translate.midiTrackToStream(
self.track).flat.notesAndRests
if len(stream) >= seq_len:
print("Go Predict", flush=True)
self.process_stream(stream[-seq_len:])
self.track = midi.MidiTrack(1)
def eventWrite(miditrack, type, data):
me = midi.MidiEvent(miditrack)
me.type = type
me.data = data
miditrack.events.append(me)
def np_seq2mid(np_seq):
"""
Converts a numpy array to a midi file.
:param np_seq: numpy beat sequence
:return: music21.midi.MidiFile
"""
mt = midi.MidiTrack(1)
t = 0
tlast = 0
for step in np_seq:
# onset will be true if at least one trig is > 0.0
# the remaining trigs are added at the same delta time
onset = False # we encountered an onset at this step
for idx, trig in enumerate(step[:15]):
# find the group
group = None
for index, grp in enumerate(PERC_GROUPS):
if idx in grp:
group = index
if trig > 0.0:
vel = int(step[15 + group] * 127)
pitch = PERC_MAP[idx]
dt = midi.DeltaTime(mt)
if onset is False:
dt.time = t - tlast
else:
dt.time = 0
mt.events.append(dt)
me = midi.MidiEvent(mt)
me.type = "NOTE_ON"
me.channel = 10
me.time = None # d
me.pitch = pitch
me.velocity = vel
mt.events.append(me)
if onset is False:
tlast = t + 6
onset = True
if onset is True:
# reset onset for the noteoff
onset = False
# makes the note off now
for idx, trig in enumerate(step[:15]):
if trig > 0.0:
pitch = PERC_MAP[idx]
dt = midi.DeltaTime(mt)
if onset is False:
dt.time = 6
else:
dt.time = 0
mt.events.append(dt)
me = midi.MidiEvent(mt)
me.type = "NOTE_OFF"
me.channel = 10
me.time = None # d
me.pitch = pitch
me.velocity = 0
mt.events.append(me)
if onset is False:
onset = True
t += TICKS_PER_Q / BEAT_DIV
# add end of track
dt = midi.DeltaTime(mt)
dt.time = 0
mt.events.append(dt)
me = midi.MidiEvent(mt)
me.type = "END_OF_TRACK"
me.channel = 1
me.data = '' # must set data to empty string
mt.events.append(me)
# make midi file
mf = midi.MidiFile()
mf.ticksPerQuarterNote = TICKS_PER_Q # cannot use: 10080
mf.tracks.append(mt)
return mf
def idxsToMidi(idxs, verbose=False):
mf = midi.MidiFile()
mf.ticksPerQuarterNote = 1024
# The maestro dataset uses the first track to store tempo data, and the second
# track to store the actual performance. So follow that convention.
tempo_track = midi.MidiTrack(0)
track = midi.MidiTrack(1)
mf.tracks = [tempo_track, track]
tempo = midi.MidiEvent(tempo_track, type=midi.MetaEvents.SET_TEMPO)
# temp.data is the number of microseconds per beat (per quarter note)
# So to set ticks per millis = 1 (easy translation from time-shift values to ticks),
# tempo.data must be 1e3 * 1024, since ticksPerQuarterNote is 1024 (see above)
tempo.data = int(1e3 * 1024).to_bytes(3, 'big')
end_of_track = midi.MidiEvent(tempo_track,
type=midi.MetaEvents.END_OF_TRACK)
end_of_track.data = ''
tempo_track.events = [
# there must always be a delta time before each event
midi.DeltaTime(tempo_track, time=0),
tempo,
midi.DeltaTime(tempo_track, time=0),
end_of_track
]
track.events = [midi.DeltaTime(track, time=0)]
current_velocity = 0
notes_on = set()
errors = {'is_on': 0, 'is_not_on': 0}
for idx in idxs:
if 0 <= idx < 128: # note-on
pitch = idx
if pitch in notes_on:
if verbose:
print(pitch, 'is already on')
errors['is_on'] += 1
continue
if track.events[-1].type != 'DeltaTime':
track.events.append(midi.DeltaTime(track, time=0))
track.events.append(makeNote(track, pitch, current_velocity))
notes_on.add(pitch)
elif 128 <= idx < (128 + 128): # note-off
pitch = idx - 128
if pitch not in notes_on:
if verbose:
print(pitch, 'is not on')
errors['is_not_on'] += 1
continue
if track.events[-1].type != 'DeltaTime':
track.events.append(midi.DeltaTime(track, time=0))
track.events.append(makeNote(track, pitch, 0))
notes_on.remove(pitch)
elif (128 + 128) <= idx < (128 + 128 + 100): # time-shift
t = (1 + idx - (128 + 128)) * 10
if track.events[-1].type == 'DeltaTime':
# combine repeated delta times
track.events[-1].time += t
else:
track.events.append(midi.DeltaTime(track, time=t))
else: # velocity
current_velocity = (idx - (128 + 128 + 100)) * 4
if verbose:
print('remaining notes left on:', notes_on)
if track.events[-1].type != 'DeltaTime':
track.events.append(midi.DeltaTime(track, time=0))
track.events.append(end_of_track)
return mf, errors
def make_event(track, event_type, channel):
event = midi.MidiEvent(track)
event.type = event_type
event.channel = channel
return event
