scripts/midi2ly.py

   1 #!@TARGET_PYTHON@
   2 #
   3 # midi2ly.py -- LilyPond midi import script
   4
   5 # This file is part of LilyPond, the GNU music typesetter.
   6 #
   7 # Copyright (C) 1998--2015  Han-Wen Nienhuys <hanwen@xs4all.nl>
   8 #                           Jan Nieuwenhuizen <janneke@gnu.org>
   9 #
  10 # LilyPond is free software: you can redistribute it and/or modify
  11 # it under the terms of the GNU General Public License as published by
  12 # the Free Software Foundation, either version 3 of the License, or
  13 # (at your option) any later version.
  14 #
  15 # LilyPond is distributed in the hope that it will be useful,
  16 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  17 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18 # GNU General Public License for more details.
  19 #
  20 # You should have received a copy of the GNU General Public License
  21 # along with LilyPond.  If not, see <http://www.gnu.org/licenses/>.
  22
  23
  24 '''
  25 TODO:
  26 '''
  27
  28 import os
  29 import sys
  30
  31 """
  32 @relocate-preamble@
  33 """
  34
  35 import lilylib as ly
  36 global _;_=ly._
  37
  38 ################################################################
  39 ## CONSTANTS
  40
  41
  42 LINE_BELL = 60
  43 scale_steps = [0, 2, 4, 5, 7, 9, 11]
  44 global_options = None
  45
  46 clocks_per_1 = 1536
  47 clocks_per_4 = 0
  48
  49 time = None
  50 reference_note = 0
  51 start_quant_clocks = 0
  52
  53 duration_quant_clocks = 0
  54 allowed_tuplet_clocks = []
  55 bar_max = 0
  56
  57 ################################################################
  58
  59
  60 program_name = sys.argv[0]
  61 program_version = '@TOPLEVEL_VERSION@'
  62
  63 authors = ('Jan Nieuwenhuizen <janneke@gnu.org>',
  64            'Han-Wen Nienhuys <hanwen@xs4all.nl>')
  65
  66 errorport = sys.stderr
  67
  68 def identify ():
  69     sys.stdout.write ('%s (GNU LilyPond) %s\n' % (program_name, program_version))
  70
  71 def warranty ():
  72     identify ()
  73     ly.encoded_write (sys.stdout, '''
  74 %s
  75
  76   %s
  77
  78 %s
  79 %s
  80 ''' % ( _ ('Copyright (c) %s by') % '1998--2015',
  81         '\n  '.join (authors),
  82         _ ('Distributed under terms of the GNU General Public License.'),
  83         _ ('It comes with NO WARRANTY.')))
  84
  85 def progress (s):
  86     ly.encoded_write (errorport, s + '\n')
  87
  88 def warning (s):
  89     progress (_ ("warning: ") + s)
  90
  91 def error (s):
  92     progress (_ ("error: ") + s)
  93     raise Exception (_ ("Exiting... "))
  94
  95 def debug (s):
  96     if global_options.debug:
  97         progress ("debug: " + s)
  98
  99 def system (cmd, ignore_error = 0):
 100     return ly.system (cmd, ignore_error=ignore_error)
 101
 102 def strip_extension (f, ext):
 103     (p, e) = os.path.splitext (f)
 104     if e == ext:
 105         e = ''
 106     return p + e
 107
 108
 109 class Duration:
 110     allowed_durs = (1, 2, 4, 8, 16, 32, 64, 128)
 111     def __init__ (self, clocks):
 112         self.clocks = clocks
 113         (self.dur, self.num, self.den) = self.dur_num_den (clocks)
 114
 115     def dur_num_den (self, clocks):
 116         for i in range (len (allowed_tuplet_clocks)):
 117             if clocks == allowed_tuplet_clocks[i]:
 118                 return global_options.allowed_tuplets[i]
 119
 120         dur = 0; num = 1; den = 1;
 121         g = gcd (clocks, clocks_per_1)
 122         if g:
 123             (dur, num) = (clocks_per_1 / g, clocks / g)
 124         if not dur in self.allowed_durs:
 125             dur = 4; num = clocks; den = clocks_per_4
 126         return (dur, num, den)
 127
 128     def dump (self):
 129         if self.den == 1:
 130             if self.num == 1:
 131                 s = '%d' % self.dur
 132             elif self.num == 3 and self.dur != 1:
 133                 s = '%d.' % (self.dur / 2)
 134             else:
 135                 s = '%d*%d' % (self.dur, self.num)
 136         else:
 137             s = '%d*%d/%d' % (self.dur, self.num, self.den)
 138
 139         global reference_note
 140         reference_note.duration = self
 141
 142         return s
 143
 144     def compare (self, other):
 145         return self.clocks - other.clocks
 146
 147 def sign (x):
 148     if x >= 0:
 149         return 1
 150     else:
 151         return -1
 152
 153 class Note:
 154     names = (0, 0, 1, 1, 2, 3, 3, 4, 4, 5, 5, 6)
 155     alterations = (0, 1, 0, 1, 0, 0, 1, 0, 1, 0, 1, 0)
 156     alteration_names = ('eses', 'es', '', 'is' , 'isis')
 157     def __init__ (self, clocks, pitch, velocity):
 158         self.pitch = pitch
 159         self.velocity = velocity
 160         # hmm
 161         self.clocks = clocks
 162         self.duration = Duration (clocks)
 163         (self.octave, self.notename, self.alteration) = self.o_n_a ()
 164
 165     def o_n_a (self):
 166         # major scale: do-do
 167         # minor scale: la-la  (= + 5) '''
 168
 169         n = self.names[(self.pitch) % 12]
 170         a = self.alterations[(self.pitch) % 12]
 171
 172         key = global_options.key
 173         if not key:
 174             key = Key (0, 0, 0)
 175
 176         if a and key.flats:
 177             a = - self.alterations[(self.pitch) % 12]
 178             n = (n - a) % 7
 179
 180         #  By tradition, all scales now consist of a sequence
 181         #  of 7 notes each with a distinct name, from amongst
 182         #  a b c d e f g.  But, minor scales have a wide
 183         #  second interval at the top - the 'leading note' is
 184         #  sharped. (Why? it just works that way! Anything
 185         #  else doesn't sound as good and isn't as flexible at
 186         #  saying things. In medieval times, scales only had 6
 187         #  notes to avoid this problem - the hexachords.)
 188
 189         #  So, the d minor scale is d e f g a b-flat c-sharp d
 190         #  - using d-flat for the leading note would skip the
 191         #  name c and duplicate the name d.  Why isn't c-sharp
 192         #  put in the key signature? Tradition. (It's also
 193         #  supposedly based on the Pythagorean theory of the
 194         #  cycle of fifths, but that really only applies to
 195         #  major scales...)  Anyway, g minor is g a b-flat c d
 196         #  e-flat f-sharp g, and all the other flat minor keys
 197         #  end up with a natural leading note. And there you
 198         #  have it.
 199
 200         #  John Sankey <bf250@freenet.carleton.ca>
 201         #
 202         #  Let's also do a-minor: a b c d e f gis a
 203         #
 204         #  --jcn
 205
 206         o = self.pitch / 12 - 4
 207
 208         if key.minor:
 209             # as -> gis
 210             if (key.sharps == 0 and key.flats == 0
 211                 and n == 5 and a == -1):
 212                 n = 4; a = 1
 213             # des -> cis
 214             elif key.flats == 1 and n == 1 and a == -1:
 215                 n = 0; a = 1
 216             # ges -> fis
 217             elif key.flats == 2 and n == 4 and a == -1:
 218                 n = 3; a = 1
 219             # g -> fisis
 220             elif key.sharps == 5 and n == 4 and a == 0:
 221                 n = 3; a = 2
 222             # d -> cisis
 223             elif key.sharps == 6 and n == 1 and a == 0:
 224                 n = 0; a = 2
 225             # a -> gisis
 226             elif key.sharps == 7 and n == 5 and a == 0:
 227                 n = 4; a = 2
 228
 229         # b -> ces
 230         if key.flats >= 6 and n == 6 and a == 0:
 231             n = 0; a = -1; o = o + 1
 232         # e -> fes
 233         if key.flats >= 7 and n == 2 and a == 0:
 234             n = 3; a = -1
 235
 236         # f -> eis
 237         if key.sharps >= 3 and n == 3 and a == 0:
 238             n = 2; a = 1
 239         # c -> bis
 240         if key.sharps >= 4 and n == 0 and a == 0:
 241             n = 6; a = 1; o = o - 1
 242
 243         return (o, n, a)
 244
 245     def __repr__ (self):
 246         s = chr ((self.notename + 2)  % 7 + ord ('a'))
 247         return 'Note(%s %s)' % (s, self.duration.dump ())
 248
 249     def dump (self, dump_dur=True):
 250         global reference_note
 251         s = chr ((self.notename + 2)  % 7 + ord ('a'))
 252         s = s + self.alteration_names[self.alteration + 2]
 253         if global_options.absolute_pitches:
 254             commas = self.octave
 255         else:
 256             delta = self.pitch - reference_note.pitch
 257             commas = sign (delta) * (abs (delta) / 12)
 258             if (((sign (delta)
 259                   * (self.notename - reference_note.notename) + 7)
 260                  % 7 >= 4)
 261                 or ((self.notename == reference_note.notename)
 262                     and (abs (delta) > 4) and (abs (delta) < 12))):
 263                 commas = commas + sign (delta)
 264
 265         if commas > 0:
 266             s = s + "'" * commas
 267         elif commas < 0:
 268             s = s + "," * -commas
 269
 270         if (dump_dur
 271             and (self.duration.compare (reference_note.duration)
 272                  or global_options.explicit_durations)):
 273             s = s + self.duration.dump ()
 274
 275         # Chords need to handle their reference duration themselves
 276
 277         reference_note = self
 278
 279         # TODO: move space
 280         return s + ' '
 281
 282
 283 class Time:
 284     def __init__ (self, num, den):
 285         self.clocks = 0
 286         self.num = num
 287         self.den = den
 288
 289     def bar_clocks (self):
 290         return clocks_per_1 * self.num / self.den
 291
 292     def __repr__ (self):
 293         return 'Time(%d/%d)' % (self.num, self.den)
 294
 295     def dump (self):
 296         global time
 297         time = self
 298         return '\n  ' + '\\time %d/%d ' % (self.num, self.den) + '\n  '
 299
 300 class Tempo:
 301     def __init__ (self, seconds_per_1):
 302         self.clocks = 0
 303         self.seconds_per_1 = seconds_per_1
 304
 305     def __repr__ (self):
 306         return 'Tempo(%d)' % self.bpm ()
 307
 308     def bpm (self):
 309         return 4 * 60 / self.seconds_per_1
 310
 311     def dump (self):
 312         return '\n  ' + '\\tempo 4 = %d ' % (self.bpm ()) + '\n  '
 313
 314 class Clef:
 315     clefs = ('"bass_8"', 'bass', 'violin', '"violin^8"')
 316     def __init__ (self, type):
 317         self.type = type
 318
 319     def __repr__ (self):
 320         return 'Clef(%s)' % self.clefs[self.type]
 321
 322     def dump (self):
 323         return '\n  \\clef %s\n  ' % self.clefs[self.type]
 324
 325 class Key:
 326     key_sharps = ('c', 'g', 'd', 'a', 'e', 'b', 'fis')
 327     key_flats = ('BUG', 'f', 'bes', 'es', 'as', 'des', 'ges')
 328
 329     def __init__ (self, sharps, flats, minor):
 330         self.clocks = 0
 331         self.flats = flats
 332         self.sharps = sharps
 333         self.minor = minor
 334
 335     def dump (self):
 336         global_options.key = self
 337
 338         s = ''
 339         if self.sharps and self.flats:
 340             pass
 341         else:
 342             if self.flats:
 343                 k = (ord ('cfbeadg'[self.flats % 7]) - ord ('a') - 2 -2 * self.minor + 7) % 7
 344             else:
 345                 k = (ord ('cgdaebf'[self.sharps % 7]) - ord ('a') - 2 -2 * self.minor + 7) % 7
 346
 347             if not self.minor:
 348                 name = chr ((k + 2) % 7 + ord ('a'))
 349             else:
 350                 name = chr ((k + 2) % 7 + ord ('a'))
 351
 352             # fis cis gis dis ais eis bis
 353             sharps = (2, 4, 6, 1, 3, 5, 7)
 354             # bes es as des ges ces fes
 355             flats = (6, 4, 2, 7, 5, 3, 1)
 356             a = 0
 357             if self.flats:
 358                 if flats[k] <= self.flats:
 359                     a = -1
 360             else:
 361                 if sharps[k] <= self.sharps:
 362                     a = 1
 363
 364             if a:
 365                 name = name + Note.alteration_names[a + 2]
 366
 367             s = '\\key ' + name
 368             if self.minor:
 369                 s = s + ' \\minor'
 370             else:
 371                 s = s + ' \\major'
 372
 373         return '\n\n  ' + s + '\n  '
 374
 375
 376 class Text:
 377     text_types = (
 378         'SEQUENCE_NUMBER',
 379         'TEXT_EVENT',
 380         'COPYRIGHT_NOTICE',
 381         'SEQUENCE_TRACK_NAME',
 382         'INSTRUMENT_NAME',
 383         'LYRIC',
 384         'MARKER',
 385         'CUE_POINT',
 386         'PROGRAM_NAME',
 387         'DEVICE_NAME', )
 388
 389     @staticmethod
 390     def _text_only(chr):
 391         if ((' ' <= chr <= '~') or chr in ['\n','\r']):
 392             return chr
 393         else:
 394             return '~'
 395
 396     def __init__ (self, type, text):
 397         self.clocks = 0
 398         self.type = type
 399         self.text =''.join(map(self._text_only, text))
 400
 401     def dump (self):
 402         # urg, we should be sure that we're in a lyrics staff
 403         s = ''
 404         if self.type == midi.LYRIC:
 405             s = '"%s"' % self.text
 406             d = Duration (self.clocks)
 407             if (global_options.explicit_durations
 408                 or d.compare (reference_note.duration)):
 409                 s = s + Duration (self.clocks).dump ()
 410             s = s + ' '
 411         elif (self.text.strip ()
 412               and self.type == midi.SEQUENCE_TRACK_NAME
 413               and not self.text == 'control track'
 414               and not self.track.lyrics_p_):
 415             text = self.text.replace ('(MIDI)', '').strip ()
 416             if text:
 417                 s = '\n  \\set Staff.instrumentName = "%(text)s"\n  ' % locals ()
 418         elif self.text.strip ():
 419             s = '\n  % [' + self.text_types[self.type] + '] ' + self.text + '\n  '
 420         return s
 421
 422     def __repr__ (self):
 423         return 'Text(%d=%s)' % (self.type, self.text)
 424
 425 def get_voice (channel, music):
 426     debug ('channel: ' + str (channel) + '\n')
 427     return unthread_notes (music)
 428
 429 class Channel:
 430     def __init__ (self, number):
 431         self.number = number
 432         self.events = []
 433         self.music = None
 434     def add (self, event):
 435         self.events.append (event)
 436     def get_voice (self):
 437         if not self.music:
 438             self.music = self.parse ()
 439         return get_voice (self.number, self.music)
 440     def parse (self):
 441         pitches = {}
 442         notes = []
 443         music = []
 444         last_lyric = 0
 445         last_time = 0
 446         for e in self.events:
 447             t = e[0]
 448
 449             if start_quant_clocks:
 450                 t = quantise_clocks (t, start_quant_clocks)
 451
 452             if (e[1][0] == midi.NOTE_OFF
 453                 or (e[1][0] == midi.NOTE_ON and e[1][2] == 0)):
 454                 debug ('%d: NOTE OFF: %s' % (t, e[1][1]))
 455                 if not e[1][2]:
 456                     debug ('   ...treated as OFF')
 457                 end_note (pitches, notes, t, e[1][1])
 458
 459             elif e[1][0] == midi.NOTE_ON:
 460                 if not pitches.has_key (e[1][1]):
 461                     debug ('%d: NOTE ON: %s' % (t, e[1][1]))
 462                     pitches[e[1][1]] = (t, e[1][2])
 463                 else:
 464                     debug ('...ignored')
 465
 466             # all include ALL_NOTES_OFF
 467             elif (e[1][0] >= midi.ALL_SOUND_OFF
 468               and e[1][0] <= midi.POLY_MODE_ON):
 469                 for i in pitches:
 470                     end_note (pitches, notes, t, i)
 471
 472             elif e[1][0] == midi.META_EVENT:
 473                 if e[1][1] == midi.END_OF_TRACK:
 474                     for i in pitches:
 475                         end_note (pitches, notes, t, i)
 476                     break
 477
 478                 elif e[1][1] == midi.SET_TEMPO:
 479                     (u0, u1, u2) = map (ord, e[1][2])
 480                     us_per_4 = u2 + 256 * (u1 + 256 * u0)
 481                     seconds_per_1 = us_per_4 * 4 / 1e6
 482                     music.append ((t, Tempo (seconds_per_1)))
 483                 elif e[1][1] == midi.TIME_SIGNATURE:
 484                     (num, dur, clocks4, count32) = map (ord, e[1][2])
 485                     den = 2 ** dur
 486                     music.append ((t, Time (num, den)))
 487                 elif e[1][1] == midi.KEY_SIGNATURE:
 488                     (alterations, minor) = map (ord, e[1][2])
 489                     sharps = 0
 490                     flats = 0
 491                     if alterations < 127:
 492                         sharps = alterations
 493                     else:
 494                         flats = 256 - alterations
 495
 496                     k = Key (sharps, flats, minor)
 497                     if not t and global_options.key:
 498                         # At t == 0, a set --key overrides us
 499                         k = global_options.key
 500                     music.append ((t, k))
 501
 502                     # ugh, must set key while parsing
 503                     # because Note init uses key
 504                     # Better do Note.calc () at dump time?
 505                     global_options.key = k
 506
 507                 elif (e[1][1] == midi.LYRIC
 508                       or (global_options.text_lyrics
 509                           and e[1][1] == midi.TEXT_EVENT)):
 510                     self.lyrics_p_ = True
 511                     if last_lyric:
 512                         last_lyric.clocks = t - last_time
 513                         music.append ((last_time, last_lyric))
 514                     last_time = t
 515                     last_lyric = Text (midi.LYRIC, e[1][2])
 516
 517                 elif (e[1][1] >= midi.SEQUENCE_NUMBER
 518                       and e[1][1] <= midi.CUE_POINT):
 519                     text = Text (e[1][1], e[1][2])
 520                     text.track = self
 521                     music.append ((t, text))
 522                     if (text.type == midi.SEQUENCE_TRACK_NAME):
 523                         self.name = text.text
 524                 else:
 525                     if global_options.verbose:
 526                         sys.stderr.write ("SKIP: %s\n" % `e`)
 527             else:
 528                 if global_options.verbose:
 529                     sys.stderr.write ("SKIP: %s\n" % `e`)
 530
 531         if last_lyric:
 532             # last_lyric.clocks = t - last_time
 533             # hmm
 534             last_lyric.clocks = clocks_per_4
 535             music.append ((last_time, last_lyric))
 536             last_lyric = 0
 537
 538         i = 0
 539         while len (notes):
 540             if i < len (music) and notes[0][0] >= music[i][0]:
 541                 i = i + 1
 542             else:
 543                 music.insert (i, notes[0])
 544                 del notes[0]
 545         return music
 546
 547 class Track (Channel):
 548     def __init__ (self):
 549         Channel.__init__ (self, None)
 550         self.name = None
 551         self.channels = {}
 552         self.lyrics_p_ = False
 553     def _add (self, event):
 554         self.events.append (event)
 555     def add (self, event, channel=None):
 556         if channel == None:
 557             self._add (event)
 558         else:
 559             self.channels[channel] = self.channels.get (channel, Channel (channel))
 560             self.channels[channel].add (event)
 561     def get_voices (self):
 562         return ([self.get_voice ()]
 563                 + [self.channels[k].get_voice ()
 564                    for k in sorted (self.channels.keys ())])
 565
 566 def create_track (events):
 567     track = Track ()
 568     for e in events:
 569         data = list (e[1])
 570         if data[0] > 0x7f and data[0] < 0xf0:
 571             channel = data[0] & 0x0f
 572             e = (e[0], tuple ([data[0] & 0xf0] + data[1:]))
 573             track.add (e, channel)
 574         else:
 575             track.add (e)
 576     return track
 577
 578 def quantise_clocks (clocks, quant):
 579     q = int (clocks / quant) * quant
 580     if q != clocks:
 581         for tquant in allowed_tuplet_clocks:
 582             if int (clocks / tquant) * tquant == clocks:
 583                 return clocks
 584         if 2 * (clocks - q) > quant:
 585             q = q + quant
 586     return q
 587
 588 def end_note (pitches, notes, t, e):
 589     try:
 590         (lt, vel) = pitches[e]
 591         del pitches[e]
 592
 593         i = len (notes) - 1
 594         while i > 0:
 595             if notes[i][0] > lt:
 596                 i = i -1
 597             else:
 598                 break
 599         d = t - lt
 600         if duration_quant_clocks:
 601             d = quantise_clocks (d, duration_quant_clocks)
 602             if not d:
 603                 d = duration_quant_clocks
 604
 605         notes.insert (i + 1,
 606               (lt, Note (d, e, vel)))
 607
 608     except KeyError:
 609         pass
 610
 611 def unthread_notes (channel):
 612     threads = []
 613     while channel:
 614         thread = []
 615         end_busy_t = 0
 616         start_busy_t = 0
 617         todo = []
 618         for e in channel:
 619             t = e[0]
 620             if (e[1].__class__ == Note
 621                 and ((t == start_busy_t
 622                       and e[1].clocks + t == end_busy_t)
 623                      or t >= end_busy_t)):
 624                 thread.append (e)
 625                 start_busy_t = t
 626                 end_busy_t = t + e[1].clocks
 627             elif (e[1].__class__ == Time
 628                   or e[1].__class__ == Key
 629                   or e[1].__class__ == Text
 630                   or e[1].__class__ == Tempo):
 631                 thread.append (e)
 632             else:
 633                 todo.append (e)
 634         threads.append (thread)
 635         channel = todo
 636
 637     return threads
 638
 639 def gcd (a,b):
 640     if b == 0:
 641         return a
 642     c = a
 643     while c:
 644         c = a % b
 645         a = b
 646         b = c
 647     return a
 648
 649 def dump_skip (skip, clocks):
 650     return skip + Duration (clocks).dump () + ' '
 651
 652 def dump (d):
 653     return d.dump ()
 654
 655 def dump_chord (ch):
 656     s = ''
 657     notes = []
 658     for i in ch:
 659         if i.__class__ == Note:
 660             notes.append (i)
 661         else:
 662             s = s + i.dump ()
 663     if len (notes) == 1:
 664         s = s + dump (notes[0])
 665     elif len (notes) > 1:
 666         global reference_note
 667         reference_dur = reference_note.duration
 668         s = s + '<'
 669         s = s + notes[0].dump (dump_dur=False)
 670         r = reference_note
 671         for i in notes[1:]:
 672             s = s + i.dump (dump_dur=False)
 673         s = s + '>'
 674         if (r.duration.compare (reference_dur)
 675             or global_options.explicit_durations):
 676             s = s + r.duration.dump ()
 677         s = s + ' '
 678         reference_note = r
 679     return s
 680
 681 def dump_bar_line (last_bar_t, t, bar_count):
 682     s = ''
 683     bar_t = time.bar_clocks ()
 684     if t - last_bar_t >= bar_t:
 685         bar_count = bar_count + (t - last_bar_t) / bar_t
 686
 687         if t - last_bar_t == bar_t:
 688             s = '\n  | %% %(bar_count)d\n  ' % locals ()
 689             last_bar_t = t
 690         else:
 691             # urg, this will barf at meter changes
 692             last_bar_t = last_bar_t + (t - last_bar_t) / bar_t * bar_t
 693
 694     return (s, last_bar_t, bar_count)
 695
 696
 697 def dump_voice (thread, skip):
 698     global reference_note, time
 699     ref = Note (0, 4*12, 0)
 700     if not reference_note:
 701         reference_note = ref
 702     else:
 703         ref.duration = reference_note.duration
 704         reference_note = ref
 705     last_e = None
 706     chs = []
 707     ch = []
 708
 709     for e in thread:
 710         if last_e and last_e[0] == e[0]:
 711             ch.append (e[1])
 712         else:
 713             if ch:
 714                 chs.append ((last_e[0], ch))
 715
 716             ch = [e[1]]
 717
 718         last_e = e
 719
 720     if ch:
 721         chs.append ((last_e[0], ch))
 722     t = 0
 723     last_t = 0
 724     last_bar_t = 0
 725     bar_count = 1
 726
 727     lines = ['']
 728     for ch in chs:
 729         t = ch[0]
 730
 731         i = lines[-1].rfind ('\n') + 1
 732         if len (lines[-1][i:]) > LINE_BELL:
 733             lines.append ('')
 734
 735         if t - last_t > 0:
 736             d = t - last_t
 737             if bar_max and t > time.bar_clocks () * bar_max:
 738                 d = time.bar_clocks () * bar_max - last_t
 739             lines[-1] = lines[-1] + dump_skip (skip, d)
 740         elif t - last_t < 0:
 741             errorport.write ('BUG: time skew')
 742
 743         (s, last_bar_t, bar_count) = dump_bar_line (last_bar_t,
 744                               t, bar_count)
 745
 746         if bar_max and bar_count > bar_max:
 747             break
 748
 749         lines[-1] = lines[-1] + s
 750         lines[-1] = lines[-1] + dump_chord (ch[1])
 751
 752         clocks = 0
 753         for i in ch[1]:
 754             if i.clocks > clocks:
 755                 clocks = i.clocks
 756
 757         last_t = t + clocks
 758
 759         (s, last_bar_t, bar_count) = dump_bar_line (last_bar_t,
 760                                                     last_t, bar_count)
 761         lines[-1] = lines[-1] + s
 762
 763     return '\n  '.join (lines) + '\n'
 764
 765 def number2ascii (i):
 766     s = ''
 767     i += 1
 768     while i > 0:
 769         m = (i - 1) % 26
 770         s = '%c' % (m + ord ('A')) + s
 771         i = (i - m)/26
 772     return s
 773
 774 def get_track_name (i):
 775     return 'track' + number2ascii (i)
 776
 777 def get_channel_name (i):
 778     return 'channel' + number2ascii (i)
 779
 780 def get_voice_name (i, zero_too_p=False):
 781     if i or zero_too_p:
 782         return 'voice' + number2ascii (i)
 783     return ''
 784
 785 def lst_append (lst, x):
 786     lst.append (x)
 787     return lst
 788
 789 def get_voice_layout (average_pitch):
 790     d = {}
 791     for i in range (len (average_pitch)):
 792         d[average_pitch[i]] = lst_append (d.get (average_pitch[i], []), i)
 793     s = list (reversed (sorted (average_pitch)))
 794     non_empty = len (filter (lambda x: x, s))
 795     names = ['One', 'Two']
 796     if non_empty > 2:
 797         names = ['One', 'Three', 'Four', 'Two']
 798     layout = map (lambda x: '', range (len (average_pitch)))
 799     for i, n in zip (s, names):
 800         if i:
 801             v = d[i]
 802             if type (v) == list:
 803                 d[i] = v[1:]
 804                 v = v[0]
 805             layout[v] = n
 806     return layout
 807
 808 def dump_track (track, n):
 809     s = '\n'
 810     track_name = get_track_name (n)
 811
 812     average_pitch = track_average_pitch (track)
 813     voices = len (filter (lambda x: x, average_pitch[1:]))
 814     clef = get_best_clef (average_pitch[0])
 815
 816     c = 0
 817     vv = 0
 818     for channel in track:
 819         v = 0
 820         channel_name = get_channel_name (c)
 821         c += 1
 822         for voice in channel:
 823             voice_name = get_voice_name (v)
 824             voice_id = track_name + channel_name + voice_name
 825             item = voice_first_item (voice)
 826
 827             if item and item.__class__ == Note:
 828                 skip = 'r'
 829                 if global_options.skip:
 830                     skip = 's'
 831                 s += '%(voice_id)s = ' % locals ()
 832                 if not global_options.absolute_pitches:
 833                     s += '\\relative c '
 834             elif item and item.__class__ == Text:
 835                 skip = '" "'
 836                 s += '%(voice_id)s = \\lyricmode ' % locals ()
 837             else:
 838                 skip = '\\skip '
 839                 s += '%(voice_id)s = ' % locals ()
 840             s += '{\n'
 841             if not n and not vv and global_options.key:
 842                 s += global_options.key.dump ()
 843             if average_pitch[vv+1] and voices > 1:
 844                 vl = get_voice_layout (average_pitch[1:])[vv]
 845                 if vl:
 846                     s += '  \\voice' + vl + '\n'
 847                 else:
 848                     if not global_options.quiet:
 849                         warning (_ ('found more than 5 voices on a staff, expect bad output'))
 850             s += '  ' + dump_voice (voice, skip)
 851             s += '}\n\n'
 852             v += 1
 853             vv += 1
 854
 855     s += '%(track_name)s = <<\n' % locals ()
 856
 857     if clef.type != 2:
 858         s += clef.dump () + '\n'
 859
 860     c = 0
 861     vv = 0
 862     for channel in track:
 863         v = 0
 864         channel_name = get_channel_name (c)
 865         c += 1
 866         for voice in channel:
 867             voice_context_name = get_voice_name (vv, zero_too_p=True)
 868             voice_name = get_voice_name (v)
 869             v += 1
 870             vv += 1
 871             voice_id = track_name + channel_name + voice_name
 872             item = voice_first_item (voice)
 873             context = 'Voice'
 874             if item and item.__class__ == Text:
 875                 context = 'Lyrics'
 876             s += '  \\context %(context)s = %(voice_context_name)s \\%(voice_id)s\n' % locals ()
 877     s += '>>\n\n'
 878     return s
 879
 880 def voice_first_item (voice):
 881     for event in voice:
 882         if (event[1].__class__ == Note
 883             or (event[1].__class__ == Text
 884                 and event[1].type == midi.LYRIC)):
 885             return event[1]
 886     return None
 887
 888 def channel_first_item (channel):
 889     for voice in channel:
 890         first = voice_first_item (voice)
 891         if first:
 892             return first
 893     return None
 894
 895 def track_first_item (track):
 896     for channel in track:
 897         first = channel_first_item (channel)
 898         if first:
 899             return first
 900     return None
 901
 902 def track_average_pitch (track):
 903     i = 0
 904     p = [0]
 905     v = 1
 906     for channel in track:
 907         for voice in channel:
 908             c = 0
 909             p.append (0)
 910             for event in voice:
 911                 if event[1].__class__ == Note:
 912                     i += 1
 913                     c += 1
 914                     p[v] += event[1].pitch
 915             if c:
 916                 p[0] += p[v]
 917                 p[v] = p[v] / c
 918             v += 1
 919     if i:
 920         p[0] = p[0] / i
 921     return p
 922
 923 def get_best_clef (average_pitch):
 924     if average_pitch:
 925         if average_pitch <= 3*12:
 926             return Clef (0)
 927         elif average_pitch <= 5*12:
 928             return Clef (1)
 929         elif average_pitch >= 7*12:
 930             return Clef (3)
 931     return Clef (2)
 932
 933 class Staff:
 934     def __init__ (self, track):
 935         self.voices = track.get_voices ()
 936     def dump (self, i):
 937         return dump_track (self.voices, i)
 938
 939 def convert_midi (in_file, out_file):
 940     global midi
 941     import midi
 942
 943     global clocks_per_1, clocks_per_4, key
 944     global start_quant_clocks
 945     global duration_quant_clocks
 946     global allowed_tuplet_clocks
 947     global time
 948
 949     str = open (in_file, 'rb').read ()
 950     clocks_max = bar_max * clocks_per_1 * 2
 951     midi_dump = midi.parse (str, clocks_max)
 952
 953     clocks_per_1 = midi_dump[0][1]
 954     clocks_per_4 = clocks_per_1 / 4
 955     time = Time (4, 4)
 956
 957     if global_options.start_quant:
 958         start_quant_clocks = clocks_per_1 / global_options.start_quant
 959
 960     if global_options.duration_quant:
 961         duration_quant_clocks = clocks_per_1 / global_options.duration_quant
 962
 963     allowed_tuplet_clocks = []
 964     for (dur, num, den) in global_options.allowed_tuplets:
 965         allowed_tuplet_clocks.append (clocks_per_1 / dur * num / den)
 966
 967     if global_options.verbose:
 968         print 'allowed tuplet clocks:', allowed_tuplet_clocks
 969
 970     tracks = [create_track (t) for t in midi_dump[1]]
 971     # urg, parse all global track events, such as Key first
 972     # this fixes key in different voice/staff problem
 973     for t in tracks:
 974         t.music = t.parse ()
 975     prev = None
 976     staves = []
 977     for t in tracks:
 978         voices = t.get_voices ()
 979         if ((t.name and prev and prev.name)
 980             and t.name.split (':')[0] == prev.name.split (':')[0]):
 981             # staves[-1].voices += voices
 982             # all global track events first
 983             staves[-1].voices = ([staves[-1].voices[0]]
 984                                  + [voices[0]]
 985                                  + staves[-1].voices[1:]
 986                                  + voices[1:])
 987         else:
 988             staves.append (Staff (t))
 989         prev = t
 990
 991     tag = '%% Lily was here -- automatically converted by %s from %s' % ( program_name, in_file)
 992
 993
 994     s = tag
 995     s += r'''
 996 \version "2.14.0"
 997 '''
 998
 999     s += r'''
1000 \layout {
1001   \context {
1002     \Voice
1003     \remove "Note_heads_engraver"
1004     \consists "Completion_heads_engraver"
1005     \remove "Rest_engraver"
1006     \consists "Completion_rest_engraver"
1007   }
1008 }
1009 '''
1010
1011     for i in global_options.include_header:
1012         s += '\n%% included from %(i)s\n' % locals ()
1013         s += open (i).read ()
1014         if s[-1] != '\n':
1015             s += '\n'
1016         s += '% end\n'
1017
1018     for i, t in enumerate (staves):
1019         s += t.dump (i)
1020
1021     s += '\n\\score {\n  <<\n'
1022
1023     control_track = False
1024     i = 0
1025     for i, staff in enumerate (staves):
1026         track_name = get_track_name (i)
1027         item = track_first_item (staff.voices)
1028         staff_name = track_name
1029         context = None
1030         if not i and not item and len (staves) > 1:
1031             control_track = track_name
1032             continue
1033         elif (item and item.__class__ == Note):
1034             context = 'Staff'
1035             if control_track:
1036                 s += '    \\context %(context)s=%(staff_name)s \\%(control_track)s\n' % locals ()
1037         elif item and item.__class__ == Text:
1038             context = 'Lyrics'
1039         if context:
1040             s += '    \\context %(context)s=%(staff_name)s \\%(track_name)s\n' % locals ()
1041
1042     s = s + '''  >>
1043   \layout {}
1044   \midi {}
1045 }
1046 '''
1047
1048     if not global_options.quiet:
1049         progress (_ ("%s output to `%s'...") % ('LY', out_file))
1050
1051     if out_file == '-':
1052         handle = sys.stdout
1053     else:
1054         handle = open (out_file, 'w')
1055
1056     handle.write (s)
1057     handle.close ()
1058
1059
1060 def get_option_parser ():
1061     p = ly.get_option_parser (usage=_ ("%s [OPTION]... FILE") % 'midi2ly',
1062                  description=_ ("Convert %s to LilyPond input.\n") % 'MIDI',
1063                  add_help_option=False)
1064
1065     p.add_option ('-a', '--absolute-pitches',
1066            action='store_true',
1067            help=_ ('print absolute pitches'))
1068     p.add_option ('-d', '--duration-quant',
1069            metavar=_ ('DUR'),
1070            help=_ ('quantise note durations on DUR'))
1071     p.add_option ('-D', '--debug',
1072            action='store_true',
1073            help=_ ('debug printing'))
1074     p.add_option ('-e', '--explicit-durations',
1075            action='store_true',
1076            help=_ ('print explicit durations'))
1077     p.add_option('-h', '--help',
1078            action='help',
1079            help=_ ('show this help and exit'))
1080     p.add_option('-i', '--include-header',
1081            help=_ ('prepend FILE to output'),
1082            action='append',
1083            default=[],
1084            metavar=_ ('FILE'))
1085     p.add_option('-k', '--key', help=_ ('set key: ALT=+sharps|-flats; MINOR=1'),
1086            metavar=_ ('ALT[:MINOR]'),
1087            default=None),
1088     p.add_option ('-o', '--output', help=_ ('write output to FILE'),
1089            metavar=_ ('FILE'),
1090            action='store')
1091     p.add_option ('-p', '--preview', help=_ ('preview of first 4 bars'),
1092            action='store_true')
1093     p.add_option ('-q', '--quiet',
1094            action="store_true",
1095            help=_ ("suppress progress messages and warnings about excess voices"))
1096     p.add_option ('-s', '--start-quant',help= _ ('quantise note starts on DUR'),
1097            metavar=_ ('DUR'))
1098     p.add_option ('-S', '--skip',
1099            action = "store_true",
1100            help =_ ("use s instead of r for rests"))
1101     p.add_option ('-t', '--allow-tuplet',
1102            metavar=_ ('DUR*NUM/DEN'),
1103            action = 'append',
1104            dest='allowed_tuplets',
1105            help=_ ('allow tuplet durations DUR*NUM/DEN'),
1106            default=[])
1107     p.add_option ('-V', '--verbose', help=_ ('be verbose'),
1108            action='store_true')
1109     p.version = 'midi2ly (LilyPond) @TOPLEVEL_VERSION@'
1110     p.add_option ('--version',
1111                  action='version',
1112                  help=_ ('show version number and exit'))
1113     p.add_option ('-w', '--warranty', help=_ ('show warranty and copyright'),
1114            action='store_true',)
1115     p.add_option ('-x', '--text-lyrics', help=_ ('treat every text as a lyric'),
1116            action='store_true')
1117
1118     p.add_option_group (ly.display_encode (_ ('Examples')),
1119               description = r'''
1120   $ midi2ly --key=-2:1 --duration-quant=32 --allow-tuplet=4*2/3 --allow-tuplet=2*4/3 foo.midi
1121 ''')
1122     p.add_option_group ('',
1123                         description=(
1124             _ ('Report bugs via %s')
1125             % 'http://post.gmane.org/post.php'
1126             '?group=gmane.comp.gnu.lilypond.bugs') + '\n')
1127     return p
1128
1129
1130
1131 def do_options ():
1132     opt_parser = get_option_parser ()
1133     (options, args) = opt_parser.parse_args ()
1134
1135     if options.warranty:
1136         warranty ()
1137         sys.exit (0)
1138
1139     if not args or args[0] == '-':
1140         opt_parser.print_help ()
1141         ly.stderr_write ('\n%s: %s %s\n' % (program_name, _ ('error: '),
1142                          _ ('no files specified on command line.')))
1143         sys.exit (2)
1144
1145     if options.duration_quant:
1146         options.duration_quant = int (options.duration_quant)
1147
1148     if options.key:
1149         (alterations, minor) = map (int, (options.key + ':0').split (':'))[0:2]
1150         sharps = 0
1151         flats = 0
1152         if alterations >= 0:
1153             sharps = alterations
1154         else:
1155             flats = - alterations
1156         options.key = Key (sharps, flats, minor)
1157
1158     if options.start_quant:
1159         options.start_quant = int (options.start_quant)
1160
1161     global bar_max
1162     if options.preview:
1163         bar_max = 4
1164
1165     options.allowed_tuplets = [map (int, a.replace ('/','*').split ('*'))
1166                 for a in options.allowed_tuplets]
1167
1168     if options.verbose:
1169         sys.stderr.write ('Allowed tuplets: %s\n' % `options.allowed_tuplets`)
1170
1171     global global_options
1172     global_options = options
1173
1174     return args
1175
1176 def main ():
1177     files = do_options ()
1178
1179     exts = ['.midi', '.mid', '.MID']
1180     for f in files:
1181         g = f
1182         for e in exts:
1183             g = strip_extension (g, e)
1184         if not os.path.exists (f):
1185             for e in exts:
1186                 n = g + e
1187                 if os.path.exists (n):
1188                     f = n
1189                     break
1190
1191         if not global_options.output:
1192             outdir = '.'
1193             outbase = os.path.basename (g)
1194             o = outbase + '-midi.ly'
1195         elif (global_options.output[-1] == os.sep
1196               or os.path.isdir (global_options.output)):
1197             outdir = global_options.output
1198             outbase = os.path.basename (g)
1199             o = os.path.join (outdir, outbase + '-midi.ly')
1200         else:
1201             o = global_options.output
1202             (outdir, outbase) = os.path.split (o)
1203
1204         if outdir and outdir != '.' and not os.path.exists (outdir):
1205             os.mkdir (outdir, 0777)
1206
1207         convert_midi (f, o)
1208
1209 if __name__ == '__main__':
1210     main ()