X-Git-Url: https://git.ao2.it/vidi-player.git/blobdiff_plain/81266999b597eeac96b846034444f4b240da72bb..6220aaf5bd9b66422ccacf165f54d5b20c79cb22:/vidi-player.py?ds=sidebyside

diff --git a/vidi-player.py b/vidi-player.py
index dbd5a75..4ac51a6 100755
--- a/vidi-player.py
+++ b/vidi-player.py
@@ -22,6 +22,9 @@ import sys
 import mido
 import vidi
 
+# TODO: turn that into a command line option
+ADD_REST_BACKGROUND = True
+
 
 def is_note(msg):
     return msg.type == 'note_on' or msg.type == 'note_off'
@@ -62,12 +65,16 @@ def timeline_from_midi(midi_file, video_font_path):
         elif is_note_off(msg) and msg.channel == 0:
             note = vidi.MidiNote(msg.note)
             duration = elapsed_time - start_time
-            print("Note name: %s start_time: %f duration: %f" %
+            print("Note name: %3s start_time: %f duration: %f" %
                   (note.name, start_time, duration))
 
-            video_sample_uri = "file://%s/sample_%s.mkv" % (video_font_path, note.name)
+            video_sample_path = "%s/sample_%s.webm" % (video_font_path, note.name)
+
+            timeline.add_clip(video_sample_path, start_time, duration)
 
-            timeline.add_clip(video_sample_uri, start_time, duration)
+    if ADD_REST_BACKGROUND:
+        rest_sample_path = "%s/sample_rest.png" % video_font_path
+        timeline.add_layer_clip(rest_sample_path, 0, elapsed_time)
 
     return timeline
 
@@ -86,16 +93,20 @@ def main():
         usage()
         return 1
 
+    if not os.path.isdir(sys.argv[2]):
+        sys.stderr.write("The second argument must be the path of the videofont directory\n")
+        usage()
+        return 1
+
+    if len(sys.argv) > 3 and os.path.exists(sys.argv[3]):
+        sys.stderr.write("File '%s' exists, exiting!\n" % sys.argv[3])
+        return 1
+
     midi_file = mido.MidiFile(sys.argv[1])
 
     overlapping_notes = check_overlapping_notes(midi_file)
     if overlapping_notes:
-        sys.stderr.write("Sorry, supporting only midi file with no overlapping notes")
-        return 1
-
-    if not os.path.isdir(sys.argv[2]):
-        sys.stderr.write("The second argument must be the path of the videofont directory")
-        usage()
+        sys.stderr.write("Sorry, supporting only midi file with no overlapping notes on channel 0\n")
         return 1
 
     video_font_path = os.path.realpath(sys.argv[2])
@@ -103,10 +114,13 @@ def main():
     timeline = timeline_from_midi(midi_file, video_font_path)
 
     if len(sys.argv) > 3:
-        destination_uri = 'file://' + os.path.realpath(sys.argv[3])
-        timeline.save(destination_uri)
+        timeline.save(sys.argv[3])
     else:
-        timeline.play()
+        try:
+            timeline.play()
+        except KeyboardInterrupt:
+            timeline.stop()
+            return 1
 
 
 if __name__ == "__main__":