diff options
Diffstat (limited to 'mkiv.py')
| -rw-r--r-- | mkiv.py | 381 |
1 files changed, 251 insertions, 130 deletions
@@ -16,141 +16,262 @@ import xml.etree.ElementTree as ET import midi import sys import os +import optparse -pat = midi.read_midifile(sys.argv[1]) -iv = ET.Element('iv') -iv.set('version', '1') -iv.set('src', os.path.basename(sys.argv[1])) +TRACKS = object() + +parser = optparse.OptionParser() +parser.add_option('-s', '--channel-split', dest='chansplit', action='store_true', help='Split MIDI channels into independent tracks (as far as -T is concerned)') +parser.add_option('-S', '--split-out', dest='chansfname', help='Store the split-format MIDI back into the specified file') +parser.add_option('-c', '--preserve-channels', dest='chanskeep', action='store_true', help='Keep the channel number when splitting channels to tracks (default is to set it to 1)') +parser.add_option('-T', '--track-split', dest='tracks', action='append_const', const=TRACKS, help='Ensure all tracks are on non-mutual streams') +parser.add_option('-t', '--track', dest='tracks', action='append', help='Reserve an exclusive set of streams for certain conditions (try --help-conds)') +parser.add_option('--help-conds', dest='help_conds', action='store_true', help='Print help on filter conditions for streams') +parser.set_defaults(tracks=[]) +options, args = parser.parse_args() + +if options.help_conds: + print '''Filter conditions are used to route events to groups of streams. + +Every filter is an expression; internally, this expression is evaluated as the body of a "lambda ev: ". +The "ev" object will be a MergeEvent with the following properties: +-ev.tidx: the originating track index (starting at 0) +-ev.abstime: the real time in seconds of this event relative to the beginning of playback +-ev.ev: a midi.NoteOnEvent: + -ev.ev.pitch: the MIDI pitch + -ev.ev.velocity: the MIDI velocity + +Specifying a -t <group>=<filter> will group all streams under a filter; if the <group> part is omitted, no group will be added. +For example: + + mkiv -t bass=ev.ev.pitch<35 -t treble=ev.ev.pitch>75 -T -t ev.abstime<10 + +will cause these groups to be made: +-A group "bass" with all notes with pitch less than 35; +-Of those not in "bass", a group in "treble" with pitch>75; +-Of what is not yet consumed, a series of groups "trkN" where N is the track index (starting at 0), which consumes the rest. +-An (unfortunately empty) unnamed group with events prior to ten real seconds. + +As can be seen, order of specification is important. Equally important is the location of -T, which should be at the end. + +NoteOffEvents are always matched to the stream which has their corresponding NoteOnEvent (in track and pitch), and so are +not affected or observed by filters. + +If the filters specified are not a complete cover, an anonymous group will be created with no filter to contain the rest. If +it is desired to force this group to have a name, use -t <group>=True.''' + exit() + +if not args: + parser.print_usage() + exit() + +for fname in args: + pat = midi.read_midifile(fname) + iv = ET.Element('iv') + iv.set('version', '1') + iv.set('src', os.path.basename(fname)) + print fname, ': MIDI format,', len(pat), 'tracks' + + if options.chansplit: + print 'Splitting channels...' + old_pat = pat + pat = midi.Pattern(resolution=old_pat.resolution) + for track in old_pat: + chan_map = {} + for ev in track: + if isinstance(ev, midi.Event): + if options.chanskeep: + newev = ev.copy() + else: + newev = ev.copy(channel=1) + chan_map.setdefault(ev.channel, midi.Track()).append(newev) + else: # MetaEvent + for trk in chan_map.itervalues(): + trk.append(ev) + items = chan_map.items() + items.sort(key=lambda pair: pair[0]) + for chn, trk in items: + pat.append(trk) + print 'Split', len(old_pat), 'tracks into', len(pat), 'tracks by channel' + + if options.chansfname: + midi.write_midifile(options.chansfname, pat) ##### Merge events from all tracks into one master list, annotated with track and absolute times ##### -print 'Merging events...' - -class MergeEvent(object): - __slots__ = ['ev', 'tidx', 'abstime'] - def __init__(self, ev, tidx, abstime): - self.ev = ev - self.tidx = tidx - self.abstime = abstime - def __repr__(self): - return '<ME %r in %d @%f>'%(self.ev, self.tidx, self.abstime) - -events = [] -bpm_at = {0: 120} - -for tidx, track in enumerate(pat): - abstime = 0 - absticks = 0 - for ev in track: - if isinstance(ev, midi.SetTempoEvent): - absticks += ev.tick - bpm_at[absticks] = ev.bpm - else: - if isinstance(ev, midi.NoteOnEvent) and ev.velocity == 0: - ev.__class__ = midi.NoteOffEvent #XXX Oww - bpm = filter(lambda pair: pair[0] <= absticks, bpm_at.items())[-1][1] - abstime += (60.0 * ev.tick) / (bpm * pat.resolution) - absticks += ev.tick - events.append(MergeEvent(ev, tidx, abstime)) - -print 'Sorting events...' - -events.sort(key = lambda ev: ev.abstime) + print 'Merging events...' + + class MergeEvent(object): + __slots__ = ['ev', 'tidx', 'abstime'] + def __init__(self, ev, tidx, abstime): + self.ev = ev + self.tidx = tidx + self.abstime = abstime + def __repr__(self): + return '<ME %r in %d @%f>'%(self.ev, self.tidx, self.abstime) + + events = [] + bpm_at = {0: 120} + + for tidx, track in enumerate(pat): + abstime = 0 + absticks = 0 + for ev in track: + if isinstance(ev, midi.SetTempoEvent): + absticks += ev.tick + bpm_at[absticks] = ev.bpm + else: + if isinstance(ev, midi.NoteOnEvent) and ev.velocity == 0: + ev.__class__ = midi.NoteOffEvent #XXX Oww + bpm = filter(lambda pair: pair[0] <= absticks, bpm_at.items())[-1][1] + abstime += (60.0 * ev.tick) / (bpm * pat.resolution) + absticks += ev.tick + events.append(MergeEvent(ev, tidx, abstime)) + + print 'Sorting events...' + + events.sort(key = lambda ev: ev.abstime) ##### Use merged events to construct a set of streams with non-overlapping durations ##### -print 'Generating streams...' - -class DurationEvent(MergeEvent): - __slots__ = ['duration'] - def __init__(self, me, dur): - MergeEvent.__init__(self, me.ev, me.tidx, me.abstime) - self.duration = dur - -class NoteStream(object): - __slots__ = ['history', 'active'] - def __init__(self): - self.history = [] - self.active = None - def IsActive(self): - return self.active is not None - def Activate(self, mev): - self.active = mev - def Deactivate(self, mev): - self.history.append(DurationEvent(self.active, mev.abstime - self.active.abstime)) - self.active = None - def WouldDeactivate(self, mev): - if not self.IsActive(): - return False - return mev.ev.pitch == self.active.ev.pitch and mev.tidx == self.active.tidx - -notestreams = [] -auxstream = [] - -for mev in events: - if isinstance(mev.ev, midi.NoteOnEvent): - for stream in notestreams: - if not stream.IsActive(): - stream.Activate(mev) - break - else: - stream = NoteStream() - notestreams.append(stream) - stream.Activate(mev) - elif isinstance(mev.ev, midi.NoteOffEvent): - for stream in notestreams: - if stream.WouldDeactivate(mev): - stream.Deactivate(mev) - break - else: - print 'WARNING: Did not match %r with any stream deactivation.'%(mev,) - else: - auxstream.append(mev) - -lastabstime = events[-1].abstime - -for ns in notestreams: - if not ns: - print 'WARNING: Active notes at end of playback.' - ns.Deactivate(MergeEvent(ns.active, ns.active.tidx, lastabstime)) - -print 'Generated %d streams'%(len(notestreams),) + print 'Generating streams...' + + class DurationEvent(MergeEvent): + __slots__ = ['duration'] + def __init__(self, me, dur): + MergeEvent.__init__(self, me.ev, me.tidx, me.abstime) + self.duration = dur + + class NoteStream(object): + __slots__ = ['history', 'active'] + def __init__(self): + self.history = [] + self.active = None + def IsActive(self): + return self.active is not None + def Activate(self, mev): + self.active = mev + def Deactivate(self, mev): + self.history.append(DurationEvent(self.active, mev.abstime - self.active.abstime)) + self.active = None + def WouldDeactivate(self, mev): + if not self.IsActive(): + return False + return mev.ev.pitch == self.active.ev.pitch and mev.tidx == self.active.tidx + + class NSGroup(object): + __slots__ = ['streams', 'filter', 'name'] + def __init__(self, filter=None, name=None): + self.streams = [] + self.filter = (lambda mev: True) if filter is None else filter + self.name = name + def Accept(self, mev): + if not self.filter(mev): + return False + for stream in self.streams: + if not stream.IsActive(): + stream.Activate(mev) + break + else: + stream = NoteStream() + self.streams.append(stream) + stream.Activate(mev) + return True + + notegroups = [] + auxstream = [] + + for spec in options.tracks: + if spec is TRACKS: + for tidx in xrange(len(pat)): + notegroups.append(NSGroup(filter = lambda mev, tidx=tidx: mev.tidx == tidx, name = 'trk%d'%(tidx,))) + else: + if '=' in spec: + name, _, spec = spec.partition('=') + else: + name = None + notegroups.append(NSGroup(filter = eval("lambda ev: "+spec), name = name)) + + print 'Initial group mappings:' + for group in notegroups: + print ('<anonymous>' if group.name is None else group.name), '<=', group.filter + + for mev in events: + if isinstance(mev.ev, midi.NoteOnEvent): + for group in notegroups: + if group.Accept(mev): + break + else: + group = NSGroup() + group.Accept(mev) + notegroups.append(group) + elif isinstance(mev.ev, midi.NoteOffEvent): + for group in notegroups: + found = False + for stream in group.streams: + if stream.WouldDeactivate(mev): + stream.Deactivate(mev) + found = True + break + if found: + break + else: + print 'WARNING: Did not match %r with any stream deactivation.'%(mev,) + else: + auxstream.append(mev) + + lastabstime = events[-1].abstime + + for group in notegroups: + for ns in group.streams: + if ns.IsActive(): + print 'WARNING: Active notes at end of playback.' + ns.Deactivate(MergeEvent(ns.active, ns.active.tidx, lastabstime)) + + print 'Final group mappings:' + for group in notegroups: + print ('<anonymous>' if group.name is None else group.name), '<=', group.filter, '(', len(group.streams), 'streams)' + + print 'Generated %d streams in %d groups'%(sum(map(lambda x: len(x.streams), notegroups)), len(notegroups)) ##### Write to XML and exit ##### -ivmeta = ET.SubElement(iv, 'meta') -ivbpms = ET.SubElement(ivmeta, 'bpms') -abstime = 0 -prevticks = 0 -prev_bpm = 120 -for absticks, bpm in sorted(bpm_at.items(), key = lambda pair: pair[0]): - abstime += ((absticks - prevticks) * 60.0) / (prev_bpm * pat.resolution) - prevticks = absticks - ivbpm = ET.SubElement(ivbpms, 'bpm') - ivbpm.set('bpm', str(bpm)) - ivbpm.set('ticks', str(absticks)) - ivbpm.set('time', str(abstime)) - -ivstreams = ET.SubElement(iv, 'streams') - -for ns in notestreams: - ivns = ET.SubElement(ivstreams, 'stream') - ivns.set('type', 'ns') - for note in ns.history: - ivnote = ET.SubElement(ivns, 'note') - ivnote.set('pitch', str(note.ev.pitch)) - ivnote.set('vel', str(note.ev.velocity)) - ivnote.set('time', str(note.abstime)) - ivnote.set('dur', str(note.duration)) - -ivaux = ET.SubElement(ivstreams, 'stream') -ivaux.set('type', 'aux') - -fw = midi.FileWriter() -fw.RunningStatus = None # XXX Hack - -for mev in auxstream: - ivev = ET.SubElement(ivaux, 'ev') - ivev.set('time', str(mev.abstime)) - ivev.set('data', repr(fw.encode_midi_event(mev.ev))) - -print 'Done.' -open(os.path.splitext(os.path.basename(sys.argv[1]))[0]+'.iv', 'w').write(ET.tostring(iv)) + ivmeta = ET.SubElement(iv, 'meta') + ivbpms = ET.SubElement(ivmeta, 'bpms') + abstime = 0 + prevticks = 0 + prev_bpm = 120 + for absticks, bpm in sorted(bpm_at.items(), key = lambda pair: pair[0]): + abstime += ((absticks - prevticks) * 60.0) / (prev_bpm * pat.resolution) + prevticks = absticks + ivbpm = ET.SubElement(ivbpms, 'bpm') + ivbpm.set('bpm', str(bpm)) + ivbpm.set('ticks', str(absticks)) + ivbpm.set('time', str(abstime)) + + ivstreams = ET.SubElement(iv, 'streams') + + for group in notegroups: + for ns in group.streams: + ivns = ET.SubElement(ivstreams, 'stream') + ivns.set('type', 'ns') + if group.name is not None: + ivns.set('group', group.name) + for note in ns.history: + ivnote = ET.SubElement(ivns, 'note') + ivnote.set('pitch', str(note.ev.pitch)) + ivnote.set('vel', str(note.ev.velocity)) + ivnote.set('time', str(note.abstime)) + ivnote.set('dur', str(note.duration)) + + ivaux = ET.SubElement(ivstreams, 'stream') + ivaux.set('type', 'aux') + + fw = midi.FileWriter() + fw.RunningStatus = None # XXX Hack + + for mev in auxstream: + ivev = ET.SubElement(ivaux, 'ev') + ivev.set('time', str(mev.abstime)) + ivev.set('data', repr(fw.encode_midi_event(mev.ev))) + + print 'Done.' + open(os.path.splitext(os.path.basename(fname))[0]+'.iv', 'w').write(ET.tostring(iv)) |