]> git.donarmstrong.com Git - lilypond.git/blobdiff - scripts/musicxml2ly.py
(grob-cause): replace backslashes by /
[lilypond.git] / scripts / musicxml2ly.py
index a1a1def9b0fc1fc60d78256b80870d81681b97c6..4b5abf9a7061f450fc1fe46b95645adf96cd7294 100644 (file)
@@ -1,6 +1,12 @@
+#!@PYTHON@
+
+import optparse
 import sys
 import re
 import os
+import string
+from gettext import gettext as _
+
 
 datadir = '@local_lilypond_datadir@'
 if not os.path.isdir (datadir):
@@ -17,77 +23,299 @@ elif os.path.exists (os.path.join (datadir, 'share/lilypond/current/')):
 
 sys.path.insert (0, os.path.join (datadir, 'python'))
 
+
 import musicxml
 import musicexp
+import lilylib as ly
+
 from rational import Rational
 
+
+def progress (str):
+       sys.stderr.write (str + '\n')
+       sys.stderr.flush ()
+       
+
 def musicxml_duration_to_lily (mxl_note):
        d = musicexp.Duration ()
        if mxl_note.get_maybe_exist_typed_child (musicxml.Type):
                d.duration_log = mxl_note.get_duration_log ()
        else:
-               d.factor = mxl_note._duration
                d.duration_log = 0
 
        d.dots = len (mxl_note.get_typed_children (musicxml.Dot))
        d.factor = mxl_note._duration / d.get_length ()
 
        return d        
+
+def group_tuplets (music_list, events):
+
+
+       """Collect Musics from
+       MUSIC_LIST demarcated by EVENTS_LIST in TimeScaledMusic objects.
+       """
+
        
-def musicxml_voice_to_lily_voice (voice):
+       indices = []
+
+       j = 0
+       for (ev_chord, tuplet_elt, fraction) in events:
+               while (j < len (music_list)):
+                       if music_list[j]== ev_chord:
+                               break
+                       j += 1
+               if tuplet_elt.type == 'start':
+                       indices.append ((j, None, fraction))
+               elif tuplet_elt.type == 'stop':
+                       indices[-1] = (indices[-1][0], j, indices[-1][2])
+
+       new_list = []
+       last = 0
+       for (i1, i2, frac) in indices:
+               if i1 >= i2:
+                       continue
+
+               new_list.extend (music_list[last:i1])
+               seq = musicexp.SequentialMusic ()
+               last = i2 + 1
+               seq.elements = music_list[i1:last]
+
+               tsm = musicexp.TimeScaledMusic ()
+               tsm.element = seq
+
+               tsm.numerator = frac[0]
+               tsm.denominator  = frac[1]
+
+               new_list.append (tsm)
+
+       new_list.extend (music_list[last:])
+       return new_list
+
+
+def musicxml_clef_to_lily (mxl):
+       sign = mxl.get_maybe_exist_named_child ('sign')
+       change = musicexp.ClefChange ()
+       if sign:
+               change.type = sign.get_text ()
+       return change
+
+       
+def musicxml_time_to_lily (mxl):
+       beats = mxl.get_maybe_exist_named_child ('beats')
+       type = mxl.get_maybe_exist_named_child ('beat-type')
+       change = musicexp.TimeSignatureChange()
+       change.fraction = (string.atoi(beats.get_text ()),
+                          string.atoi(type.get_text ()))
        
+       return change
+
+def musicxml_key_to_lily (mxl):
+       start_pitch  = musicexp.Pitch ()
+       try:
+               mode = mxl.get_maybe_exist_named_child ('mode')
+               if mode:
+                       mode = mode.get_text ()
+               else:
+                       mode = 'major'
+                       
+               (n,a) = { 'major' : (0,0),
+                         'minor' : (6,0),
+                       }[mode]
+               start_pitch.step = n
+               start_pitch.alteration = a
+       except  KeyError:
+               print 'unknown mode', mode
+               
+       fifths = string.atoi (mxl.get_maybe_exist_named_child ('fifths').get_text ())
+
+       fifth = musicexp.Pitch()
+       fifth.step = 4
+       if fifths < 0:
+               fifths *= -1
+               fifth.step *= -1
+               fifth.normalize ()
+       
+       start_pitch = musicexp.Pitch()
+       for x in range (fifths):
+               start_pitch = start_pitch.transposed (fifth)
+
+       start_pitch.octave = 0
+
+       change = musicexp.KeySignatureChange()
+       change.mode = mode
+       change.tonic = start_pitch
+       return change
+       
+def musicxml_attributes_to_lily (attrs):
+       elts = []
+       attr_dispatch =  {
+               'clef': musicxml_clef_to_lily,
+               'time': musicxml_time_to_lily,
+               'key': musicxml_key_to_lily
+       }
+       for (k, func) in attr_dispatch.items ():
+               childs = attrs.get_named_children (k)
+
+               ## ugh: you get clefs spread over staves for piano
+               if childs:
+                       elts.append (func (childs[0]))
+       
+       return elts
+
+def create_skip_music (duration):
+       skip = musicexp.SkipEvent()
+       skip.duration.duration_log = 0
+       skip.duration.factor = duration
+
+       evc = musicexp.EventChord ()
+       evc.append (skip)
+       return evc
+
+spanner_event_dict = {
+       'slur' : musicexp.SlurEvent,
+       'beam' : musicexp.BeamEvent,
+}      
+spanner_type_dict = {
+       'start': -1,
+       'begin': -1,
+       'stop': 1,
+       'end' : 1
+}
+
+def musicxml_spanner_to_lily_event (mxl_event):
+       ev = None
+       
+       name = mxl_event.get_name()
+       try:
+               func = spanner_event_dict[name]
+               ev = func()
+       except KeyError:
+               print 'unknown span event ', mxl_event
+
+       try:
+               key = mxl_event.get_type ()
+               ev.span_direction = spanner_type_dict[key]
+       except KeyError:
+               print 'unknown span type', key, 'for', name
+
+       return ev
+
+def musicxml_note_to_lily_main_event (n):
+       pitch  = None
+       duration = None
+               
+       mxl_pitch = n.get_maybe_exist_typed_child (musicxml.Pitch)
+       event = None
+       if mxl_pitch:
+               pitch = musicxml_pitch_to_lily (mxl_pitch)
+               event = musicexp.NoteEvent()
+               event.pitch = pitch
+
+               acc = n.get_maybe_exist_named_child ('accidental')
+               if acc:
+                       # let's not force accs everywhere. 
+                       event.cautionary = acc.editorial
+               
+       elif n.get_maybe_exist_typed_child (musicxml.Rest):
+               event = musicexp.RestEvent()
+
+       event.duration = musicxml_duration_to_lily (n)
+       return event
+
+def musicxml_voice_to_lily_voice (voice):
        ly_voice = []
        ly_now = Rational (0)
-       for n in voice:
+       pending_skip = Rational (0) 
+
+       tuplet_events = []
+
+       for n in voice._elements:
+               if n.get_name () == 'forward':
+                       continue
+               
+               if isinstance (n, musicxml.Attributes):
+                       ly_now += pending_skip
+                       pending_skip = Rational (0)
+                       
+                       ly_voice.extend (musicxml_attributes_to_lily (n))
+                       continue
+               
                if not n.__class__.__name__ == 'Note':
-                       print 'not a note?'
+                       print 'not a Note or Attributes?', n
                        continue
 
-               pitch  = None
-               duration = None
-               
-               mxl_pitch = n.get_maybe_exist_typed_child (musicxml.Pitch)
-               event = None
+               if n.is_first () and ly_voice:
+                       ly_voice[-1].comment += '\n'
                
-               if mxl_pitch:
-                       pitch = musicxml_pitch_to_lily (mxl_pitch)
-                       event = musicexp.NoteEvent()
-                       event.pitch = pitch
-               elif n.get_maybe_exist_typed_child (musicxml.Rest):
-                       event = musicexp.RestEvent()
-
-               event.duration = musicxml_duration_to_lily (n)
+
+               main_event = musicxml_note_to_lily_main_event (n)
+
                ev_chord = None
                if None ==  n.get_maybe_exist_typed_child (musicxml.Chord):
-                       if ly_voice:
-                               ly_now += ly_voice[-1].get_length ()
+                       ly_now += pending_skip
+                       pending_skip = main_event.get_length ()
 
                        if ly_now <> n._when:
-                               diff = n._when - ly_now 
+                               diff = n._when - ly_now
+
                                if diff < Rational (0):
                                        print 'huh: negative skip', n._when, ly_now, n._duration
                                        diff = Rational (1,314159265)
-                                       
-                               
-                               skip = musicexp.SkipEvent()
-                               skip.duration.duration_log = 0
-                               skip.duration.factor = diff
 
-                               evc = musicexp.EventChord ()
-                               evc.elements.append (skip)
-                               ly_voice.append (evc)
+                               ly_voice.append (create_skip_music (diff))
                                ly_now = n._when
                                
                        ly_voice.append (musicexp.EventChord())
                else:
                        pass
-                       #print 'append'
+               
                ev_chord = ly_voice[-1]
-               ev_chord.elements.append (event)
+               ev_chord.append (main_event)
+               
+               notations = n.get_maybe_exist_typed_child (musicxml.Notations)
+               tuplet_event = None
+               span_events = []
+               if notations:
+                       if notations.get_tuplet():
+                               tuplet_event = notations.get_tuplet()
+                               mod = n.get_maybe_exist_typed_child (musicxml.Time_modification)
+                               frac = (1,1)
+                               if mod:
+                                       frac = mod.get_fraction ()
+                               
+                               tuplet_events.append ((ev_chord, tuplet_event, frac))
 
+                       slurs = [s for s in notations.get_named_children ('slur')
+                                if s.get_type () in ('start','stop')]
+                       if slurs:
+                               if len (slurs) > 1:
+                                       print 'more than 1 slur?'
 
-       seq_music = musicexp.SequentialMusic()
+                               lily_ev = musicxml_spanner_to_lily_event (slurs[0])
+                               ev_chord.append (lily_ev)
+
+                       mxl_tie = notations.get_tie ()
+                       if mxl_tie and mxl_tie.type == 'start':
+                               ev_chord.append (musicexp.TieEvent ())
+
+               mxl_beams = [b for b in n.get_named_children ('beam')
+                            if b.get_type () in ('begin', 'end')] 
+               if mxl_beams:
+                       beam_ev = musicxml_spanner_to_lily_event (mxl_beams[0])
+                       if beam_ev:
+                               ev_chord.append (beam_ev)
+                       
+               if tuplet_event:
+                       mod = n.get_maybe_exist_typed_child (musicxml.Time_modification)
+                       frac = (1,1)
+                       if mod:
+                               frac = mod.get_fraction ()
+                               
+                       tuplet_events.append ((ev_chord, tuplet_event, frac))
 
+       ly_voice = group_tuplets (ly_voice, tuplet_events)
+
+       seq_music = musicexp.SequentialMusic()
        
        seq_music.elements = ly_voice
        return seq_music
@@ -110,34 +338,189 @@ def musicxml_pitch_to_lily (mxl_pitch):
        p = musicexp.Pitch()
        p.alteration = mxl_pitch.get_alteration ()
        p.step = (ord (mxl_pitch.get_step ()) - ord ('A') + 7 - 2) % 7
-       p.octave = mxl_pitch.get_octave () -4
+       p.octave = mxl_pitch.get_octave () - 4
        return p
 
+
+
+def voices_in_part (part):
+       """Return a Name -> Voice dictionary for PART"""
+       part.interpret ()
+       part.extract_voices ()          
+       voice_dict = part.get_voices ()
+
+       return voice_dict
+
+def voices_in_part_in_parts (parts):
+       """return a Part -> Name -> Voice dictionary"""
+       return dict([(p, voices_in_part (p)) for p in parts])
+
+
 def get_all_voices (parts):
-       all_voices = {} 
-       for p in parts:
-               p.interpret ()
-               p.extract_voices ()             
-               voice_dict = p.get_voices ()
+       all_voices = voices_in_part_in_parts (parts)
+
+       all_ly_voices = {}
+       for p, name_voice in all_voices.items ():
+
+               part_ly_voices = {}
+               for n, v in name_voice.items ():
+                       progress ("Converting to LilyPond expressions...")
+                       part_ly_voices[n] = (musicxml_voice_to_lily_voice (v), v)
+
+               all_ly_voices[p] = part_ly_voices
                
-               for (id, voice) in voice_dict.items ():
-                       m = musicxml_voice_to_lily_voice (voice)
-                       m_name = 'Part' + p.name + 'Voice' + id
-                       m_name = musicxml_id_to_lily (m_name)
-                       all_voices[m_name] = m
+       return all_ly_voices
+
+
+def option_parser ():
+       p = ly.get_option_parser(usage='musicxml2ly FILE.xml',
+                                 version = """%prog (LilyPond) @TOPLEVEL_VERSION@
 
-       return all_voices
+This program is free software.  It is covered by the GNU General Public
+License and you are welcome to change it and/or distribute copies of it
+under certain conditions.  Invoke as `lilypond --warranty' for more
+information.
 
-printer = musicexp.Output_printer()
+Copyright (c) 2005--2006 by
+  Han-Wen Nienhuys <hanwen@xs4all.nl> and
+  Jan Nieuwenhuizen <janneke@gnu.org>
+""",
 
+                                 description  =
+                                 """Convert MusicXML file to LilyPond input.
+"""
+                                 )
+       p.add_option ('-v', '--verbose',
+                     action = "store_true",
+                     dest = 'verbose',
+                     help = 'be verbose')
+       p.add_option ('-o', '--output',
+                     metavar = 'FILE',
+                     action = "store",
+                     default = None,
+                     type = 'string',
+                     dest = 'output',
+                     help = 'set output file')
 
-tree = musicxml.read_musicxml (sys.argv[1])
-parts = tree.get_typed_children (musicxml.Part)
+       p.add_option_group  ('', description = '''Report bugs via http://post.gmane.org/post.php?group=gmane.comp.gnu.lilypond.bugs
+''')
+       return p
+
+def music_xml_voice_name_to_lily_name (part, name):
+       str = "Part%sVoice%s" % (part.id, name)
+       return musicxml_id_to_lily (str) 
+
+def print_voice_definitions (printer, voices):
+       for (part, nv_dict) in voices.items():
+               for (name, (voice, mxlvoice)) in nv_dict.items ():
+                       k = music_xml_voice_name_to_lily_name (part, name)
+                       printer.dump ('%s = ' % k)
+                       voice.print_ly (printer)
+                       printer.newline()
+def uniq_list (l):
+       return dict ([(elt,1) for elt in l]).keys ()
+       
+def print_score_setup (printer, part_list, voices):
+       part_dict = dict ([(p.id, p) for p in voices.keys ()]) 
+
+       printer ('<<')
+       printer.newline ()
+       for part_definition in part_list:
+               part_name = part_definition.id
+               try:
+                       part = part_dict[part_name]
+               except KeyError:
+                       print 'unknown part in part-list:', part_name
+                       continue
+
+               nv_dict = voices[part]
+               staves = reduce (lambda x,y: x+ y,
+                                [mxlvoice._staves.keys ()
+                                 for (v, mxlvoice) in nv_dict.values ()],
+                                [])
+
+               if len (staves) > 1:
+                       staves = uniq_list (staves)
+                       staves.sort ()
+                       printer ('\\context PianoStaff << ')
+                       printer.newline ()
+                       
+                       for s in staves:
+                               staff_voices = [music_xml_voice_name_to_lily_name (part, voice_name)
+                                               for (voice_name, (v, mxlvoice)) in nv_dict.items ()
+                                               if mxlvoice._start_staff == s]
+                               
+                               printer ('\\context Staff = "%s" << ' % s)
+                               printer.newline ()
+                               for v in staff_voices:
+                                       printer ('\\context Voice = "%s"  \\%s' % (v,v))
+                                       printer.newline ()
+                               printer ('>>')
+                               printer.newline ()
+                               
+                       printer ('>>')
+                       printer.newline ()
+                       
+               else:
+                       printer ('\\new Staff <<')
+                       printer.newline ()
+                       for (n,v) in nv_dict.items ():
+
+                               n = music_xml_voice_name_to_lily_name (part, n) 
+                               printer ('\\context Voice = "%s"  \\%s' % (n,n))
+                       printer ('>>')
+                       printer.newline ()
+                       
+
+       printer ('>>')
+       printer.newline ()
 
-voices = get_all_voices (parts)
-for  (k,v) in voices.items():
-       print '%s = \n' % k
-       v.print_ly (printer.dump)
-       printer.newline()
+                               
+
+def print_ly_preamble (printer, filename):
+       printer.dump_version ()
+       printer.print_verbatim ('%% converted from %s\n' % filename)
+
+def convert (filename, output_name):
+       printer = musicexp.Output_printer()
+       progress ("Reading MusicXML...")
        
+       tree = musicxml.read_musicxml (filename)
+       parts = tree.get_typed_children (musicxml.Part)
+       voices = get_all_voices (parts)
+
+       part_list = []
+       if tree.get_maybe_exist_typed_child (musicxml.Part_list):
+               pl = tree.get_maybe_exist_typed_child (musicxml.Part_list)
+               part_list = pl.get_named_children ("score-part")
+               
+       if not output_name:
+               output_name = os.path.basename (filename)
+               output_name = os.path.splitext (output_name)[0] + '.ly'
+
+               
+       if output_name:
+               progress ("Output to `%s'" % output_name)
+               printer.set_file (open (output_name, 'w'))
+       
+       progress ("Printing as .ly...")
+
+       print_ly_preamble (printer, filename)
+       print_voice_definitions (printer,  voices)
+       print_score_setup (printer, part_list, voices)
+       printer.newline ()
+       return voices
+
+
+def main ():
+       opt_parser = option_parser()
+
+       (options, args) = opt_parser.parse_args ()
+       if not args:
+               opt_parser.print_usage()
+               sys.exit (2)
+
+       voices = convert (args[0], options.output)
 
+if __name__ == '__main__':
+       main()