Spaces:
Build error
Build error
File size: 2,850 Bytes
f35cc94 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
from ..music_transformer.transform import *
class MultitrackItem():
def __init__(self, melody:MusicItem, chords:MusicItem, stream=None):
self.melody,self.chords = melody, chords
self.vocab = melody.vocab
self._stream = stream
@classmethod
def from_file(cls, midi_file, vocab):
return cls.from_stream(file2stream(midi_file), vocab)
@classmethod
def from_stream(cls, stream, vocab):
if not isinstance(stream, music21.stream.Score): stream = stream.voicesToParts()
num_parts = len(stream.parts)
sort_pitch = False
if num_parts > 2:
raise ValueError('Could not extract melody and chords from midi file. Please make sure file contains exactly 2 tracks')
elif num_parts == 1:
print('Warning: only 1 track found. Inferring melody/chords')
stream = separate_melody_chord(stream)
sort_pitch = False
mpart, cpart = stream2npenc_parts(stream, sort_pitch=sort_pitch)
return cls.from_npenc_parts(mpart, cpart, vocab, stream)
@classmethod
def from_npenc_parts(cls, mpart, cpart, vocab, stream=None):
mpart = npenc2idxenc(mpart, seq_type=SEQType.Melody, vocab=vocab, add_eos=False)
cpart = npenc2idxenc(cpart, seq_type=SEQType.Chords, vocab=vocab, add_eos=False)
return MultitrackItem(MusicItem(mpart, vocab), MusicItem(cpart, vocab), stream)
@classmethod
def from_idx(cls, item, vocab):
m, c = item
return MultitrackItem(MusicItem.from_idx(m, vocab), MusicItem.from_idx(c, vocab))
def to_idx(self): return np.array((self.melody.to_idx(), self.chords.to_idx()))
@property
def stream(self):
self._stream = self.to_stream() if self._stream is None else self._stream
return self._stream
def to_stream(self, bpm=120):
ps = self.melody.to_npenc(), self.chords.to_npenc()
ps = [npenc2chordarr(p) for p in ps]
chordarr = chordarr_combine_parts(ps)
return chordarr2stream(chordarr, bpm=bpm)
def show(self, format:str=None):
return self.stream.show(format)
def play(self): self.stream.show('midi')
def transpose(self, val):
return MultitrackItem(self.melody.transpose(val), self.chords.transpose(val))
def pad_to(self, val):
return MultitrackItem(self.melody.pad_to(val), self.chords.pad_to(val))
def trim_to_beat(self, beat):
return MultitrackItem(self.melody.trim_to_beat(beat), self.chords.trim_to_beat(beat))
def combine2chordarr(np1, np2, vocab):
if len(np1.shape) == 1: np1 = idxenc2npenc(np1, vocab)
if len(np2.shape) == 1: np2 = idxenc2npenc(np2, vocab)
p1 = npenc2chordarr(np1)
p2 = npenc2chordarr(np2)
return chordarr_combine_parts((p1, p2))
|