annotate Syncopation models/TMC.py @ 0:76ce27beba95

Have uploaded the syncopation dataset and audio wavefies.

'''
Author: Chunyang Song
Institution: Centre for Digital Music, Queen Mary University of London

** Toussaint's Metric Complexity Model **

Algorithm:

Only applicable to monorhythms.

Define metrical hierarchy by given time signature;
Calculate how many onsets and determine Maximum Metricality Max_Metric;
Calculate the Metrical Simplicity - the weights of all onsets;
Syncopation = Max_Metric - Metric_simplicity.
Output the predicted syncopation score; -1 indicates non-applicable

'''

from MeterStructure import MeterStructure

def metricalModel(rhythm, time_sig, category, bar):
	ms = MeterStructure(time_sig)
	meter = ms.getLJWeights(bar)
	if len(meter) !=0:
		
		metricalSimplicity = 0  # sum of weights of onsets per bar
		maxMetrical = 0 	 # maximum metricity per bar
		onsetCount = 0		 # The number of onsets per bar
	
		if 'poly' in category:  # not applicable to polyrhythms
			return -1
	
		# Calculate metricalSimplicity
		else:
			l = len(rhythm)
			for i in range(l):
				if rhythm[i] == 1: # onset detected
					pos = int((float(i)/l) *len(meter)) # looking for the metrical position where this note locates
					metricalSimplicity = metricalSimplicity+meter[pos]
					onsetCount = onsetCount+1
		
			# Calculate max metricity
			meter.sort(reverse=True)
			for i in range(0,onsetCount):
				maxMetrical = maxMetrical+meter[i]
			
			#print 'test', onsetCount, maxMetrical, metricalSimplicity
			syncopation = (maxMetrical - metricalSimplicity)
	
			return syncopation	
	else:
		return -1

# Retrieve the stimuli
f = file('stimuli.txt')
#f = file('stimuli_34only.txt')

#Calculate syncopation for each rhythm pattern
while True:
	line = f.readline().split(';')
	if len(line) == 1:
		 break
	else:
		sti_name = line[0]
		rhythmString = line[1].split()
		time_sig = line[2]
		category = line[3]
		bar = int(line[4])
		
		rhythm = map(int,rhythmString[0].split(','))

		print sti_name, metricalModel(rhythm, time_sig, category, bar)