comparison Syncopation models/syncopation.py @ 40:6371e8f21f7d

updating the syncopation functions to fix a few problems

 
 
 def sync_perbar_permodel (model, bar, parameters=None):
 	return model.get_syncopation(bar, parameters)
 
-def syncopation_barlist_permodel(model, source, parameters=None):
+def calculate_syncopation(model, source, parameters=None):
  	total = 0.0
  	barResults = []
  	numberOfNotes = 0
 
 	barlist = None
 
  	if isinstance(source, BarList):
  		barlist = source
  		sourceType = "bar list"
+ 	elif isinstance(source, Bar):
+ 		barlist = BarList().append(source)
+ 		sourceType = "single bar"
 	elif isinstance(source, basestring):
 		#treat source as a filename
 		sourceType = source
 		if source[-4:]==".mid":
 			import readmidi

 			print "Error in syncopation_barlist_permodel(): Unrecognised file type."
 	else:
 		print "Error in syncopation_barlist_permodel(): unrecognised source type."
 	
 	barsDiscarded=0
+	discardedlist = []
+	includedlist = []
 
 	if barlist!=None:
 		for bar in barlist:
 			if not bar.is_empty():
 				barSyncopation = sync_perbar_permodel(model, bar, parameters)

 			
 			barResults.append(barSyncopation)
 			if barSyncopation != None:
 				total += barSyncopation
 				numberOfNotes += sum(bar.get_binary_sequence())
+				includedlist.append(barlist.index(bar))
 			else:
 				barsDiscarded += 1
+				discardedlist.append(barlist.index(bar))
 				print 'Model could not measure bar %d, returning None.' % barlist.index(bar)
 
 		import WNBD
 		if model is WNBD:
 			total =  total / numberOfNotes
 
 		average = total / (len(barResults)-barsDiscarded)
 
- 	return {"summed_syncopation":total, "average_syncopation_per_bar":average, "source":sourceType, "number_of_bars":len(barResults), "number_of_bars_not_measured":barsDiscarded, "syncopation_by_bar":barResults}
+ 	return {
+ 			"model_name":model.__name__ , 
+ 			"summed_syncopation":total, 
+ 			"mean_syncopation_per_bar":average, 
+ 			"source":sourceType, 
+ 			"number_of_bars":len(barResults), 
+ 			"number_of_bars_not_measured":barsDiscarded, 
+ 			"bars_with_valid_output":includedlist, 
+ 			"syncopation_by_bar":barResults
+ 			}
 
 
 
 def results_to_xml(results, outputFilename):
 	from xml.etree.ElementTree import Element, ElementTree