Added time_sign and key_sign control

markov_chain.py

@@ -252,7 +252,7 @@ if __name__=="__main__":
 # !lilypond -fpng "new_song-midi.ly"
 
 #Converting abc2ly
-# abc2ly "new_song.ly"
+# abc2ly "new_song.abc"
 #Then same command to convert to png which will generate midi as well 
 
 #midi to abc

n-grams/music.py

@@ -9,9 +9,14 @@ import pickle
 corpus = []
 song = []
 
-"Generate sheet music for a new song from a corpus of existing songs in abc format"
-
-
+#TODO: convert these into inputs
+# lengthofsong = 10 Should we control this? Setting it to random now
+timesignature = ['3/4','4/4','1/8','C'] #Sometimes the letter “C” (meaning common time) will be used in place of 4/4. 
+#Both C and 4/4 indicate that there are four quarter note beats in each measure.
+keysignature = "G"
+key_enforced = False
+key_enforced = True #Set to true if user wants in specific key
+print("hello")
 # get the list of filenames (abc files downloaded from http://www.norbeck.nu/abc/)
 # getdirs = []
 # dirs = ["hn201612/i/*.abc", "hn201612/s/*.abc"]
@@ -21,7 +26,7 @@ song = []
 # 	for filename in glob.iglob(dir1):
 # 		getdirs += [filename]
 
-
+selected_timeSign = '3/4'
 #Finds all absolute paths in directory
 #https://stackoverflow.com/questions/9816816/get-absolute-paths-of-all-files-in-a-directory
 def abs_paths(dir):
@@ -30,16 +35,23 @@ def abs_paths(dir):
             yield os.path.abspath(os.path.join(dir_path, f))
 # ex_filename = "hn201612/i/hnsong1.abc"
 # parsing on file to extract songs and add them to corpus
-for filename in abs_paths("n-grams/data/hard"):
+for filename in abs_paths("n-grams/data/beginner"):
 	with open(filename) as f:
+		print(filename)
 		lines = f.readlines()
 		last = len(lines)
+		accepted = False
 		for index, line in enumerate(lines):
 			if (line.find("|") < 0 and index - 1 == last):
 				# if the next line does not have pipes add song to corpus and then set song variable empty again
-				corpus.append(song)
+				if accepted and key_enforced and key_accepted:
+					corpus.append(song)
+					accepted = False
+					key_accepted = False
+				if accepted:
+					corpus.append(song)
+					accepted = False
 				song = []
-
 			else:
 				if line.find("|") > -1:
 					# a line should be split on "|" and copied to the corpus if it has pipes
@@ -47,7 +59,18 @@ for filename in abs_paths("n-grams/data/hard"):
 					# add the list of measures to the song
 					song += [x.strip("\r\n") for x in sline if len(x.strip("\r\n")) > 0]
 					last = index
-
+				elif "M:" in line:
+					#time signature
+					if selected_timeSign == "4/4":
+						if "4/4" in line or "C|" in line:
+							accepted = True
+					elif selected_timeSign in line:
+						accepted = True
+				elif line.find("K:") and key_enforced:
+					#key signature
+					if keysignature in line:
+						key_accepted = True
+					
 print("Training on {} songs...".format(len(corpus)))
 
 # MARKOV PART
@@ -75,8 +98,9 @@ def writesong(songlength, first):
 # choose a random song length from list of song lengths in corpus
 lengthofsong = random.choice([len(x) for x in corpus if len(x) > 10])
 print("Song length will be {}".format(lengthofsong))
-
-
+song_len = [len(x) for x in corpus if len(x)>10]
+song_len.sort()
+print("Song lengths",song_len)
 firstnote = markov.generate(model, n, max_iterations=3)[0]
 # print "first note: {}".format(firstnote)