Add a method to get the number of notes depending on the tempo and the desired duration

Author: Kurokabe

Diff for: .gitignore

@@ -113,4 +113,11 @@ venv.bak/
 dmypy.json
 
 # Pyre type checker
-.pyre/
+.pyre/
+
+# Video File
+*.avi
+*.mp4
+
+# Music File
+*.mid

Diff for: BeatGenerator.py

@@ -1,6 +1,10 @@
 from midiutil import MIDIFile
 import numpy as np
 import cv2
+import sys
+import argparse
+import random
+from moviepy.video.io.VideoFileClip import VideoFileClip
 
 def averageRGB(frame, everyNPixels):
     """
@@ -42,37 +46,67 @@ def averageRGBChoice(frame, colorStage, everyNPixels):
 
     return averageColor
 
-cap = cv2.VideoCapture('Class_Room_Tour.avi')
-degrees = []  # MIDI note number
-counter = 0
+def getNumberNote(musicDuration, tempo):
+    """
+    Return the number of notes that the music must have to be of the specified duration
+    musicDuration : the duration of the music (in seconds)
+    tempo : the number of BMP of the music
+    """
+    durationInMinute = musicDuration / 60
+    return int(tempo * durationInMinute)
+
+
+
+if __name__ == "__main__":
+    # TODO use that to get the first argument as the filename to use
+    # # Set an argument parser
+    # parser = argparse.ArgumentParser(description='Convert a video file (.avi) to a music.')
+    # parser.add_argument('filename', metavar='filename',
+    #                     help='a video file')
+
+    # args = vars(parser.parse_args())
+
+    cap = cv2.VideoCapture('Class_Room_Tour.avi')
+    degrees = []  # MIDI note number
+    counter = 0
+
+    # while(cap.isOpened()):
+    #     ret, frame = cap.read()
+    #     everyNImages = 10
+
+    #     if frame is None:
+    #         break
+    #     else:
+    #         if counter % everyNImages == 0:
+    #             degrees.append(averageRGB(frame, 100))
+    #         counter += 1
 
-while(cap.isOpened()):
-    ret, frame = cap.read()
-    everyNImages = 10
+    # cap.release()
+    # cv2.destroyAllWindows()
 
-    if frame is None:
-        break
-    else:
-        if counter % everyNImages == 0:
-            degrees.append(averageRGB(frame, 100))
-        counter += 1
+    track    = 0
+    channel  = 0
+    time     = 0    # In beats
+    duration = 1    # In beats
+    tempo    = 50   # In BPM
+    volume   = 100  # 0-127, as per the MIDI standard
 
-cap.release()
-cv2.destroyAllWindows()
+    # Get the duration of the video
+    clip = VideoFileClip("Class_Room_Tour.avi")
+    print( clip.duration )  
+    # Define the number of note the music must have according to the tempo
+    numberOfNote = getNumberNote(clip.duration, tempo)
 
-track    = 0
-channel  = 0
-time     = 0    # In beats
-duration = 1    # In beats
-tempo    = 200   # In BPM
-volume   = 100  # 0-127, as per the MIDI standard
+    # Generate the correct number of note to have a music of the desired length
+    for i in range(numberOfNote):
+        degrees.append(random.randint(10, 100))
 
-MyMIDI = MIDIFile(1)  # One track, defaults to format 1 (tempo track is created
-                      # automatically)
-MyMIDI.addTempo(track, time, tempo)
+    MyMIDI = MIDIFile(1)  # One track, defaults to format 1 (tempo track is created
+                        # automatically)
+    MyMIDI.addTempo(track, time, tempo)
 
-for i, pitch in enumerate(degrees):
-    MyMIDI.addNote(track, channel, pitch, time + i, duration, volume)
+    for i, pitch in enumerate(degrees):
+        MyMIDI.addNote(track, channel, pitch, time + i, duration, volume)
 
-with open("major-scale.mid", "wb") as output_file:
-    MyMIDI.writeFile(output_file)
+    with open("major-scale.mid", "wb") as output_file:
+        MyMIDI.writeFile(output_file)

Diff for: requirements.txt

@@ -0,0 +1,15 @@
+certifi==2018.11.29
+chardet==3.0.4
+decorator==4.3.2
+idna==2.8
+imageio==2.5.0
+imageio-ffmpeg==0.2.0
+MIDIUtil==1.2.1
+moviepy==1.0.0
+numpy==1.16.2
+opencv-python==4.0.0.21
+Pillow==5.4.1
+proglog==0.1.9
+requests==2.21.0
+tqdm==4.31.1
+urllib3==1.24.1