comparison Syncopation models/music_objects.py @ 23:df1e7c378ee0

fixed KTH, and WNBD

 
 	def to_string(self):
 		return str(self)[1:-1].replace(" ","")
 
 
-def velocity_sequence_to_note_sequence(velocitySequence):
+def velocity_sequence_to_note_sequence(velocitySequence, nextbarVelocitySequence = None):
 	
 	noteSequence = NoteSequence()
 
 	for index in range(len(velocitySequence)):
 		if (velocitySequence[index]!= 0): # onset detected

 			# add the current note into note sequence
 			noteSequence.append( Note(startTime, 0, velocity) )
 
 	# to set the duration for the last note
 	if( len(noteSequence) > 0):
-		previousNote = noteSequence[-1]
-		previousNote.duration = len(velocitySequence) - previousNote.startTime
+		lastNote = noteSequence[-1]
+		
+		if nextbarVelocitySequence == None:
+			lastNote.duration = len(velocitySequence) - lastNote.startTime
+		else:
+			nextNoteStartTime = next((index for index, v in enumerate(nextbarVelocitySequence) if v), None)
+			lastNote.duration = len(velocitySequence) + nextNoteStartTime-lastNote.startTime
+
 
 	return noteSequence
 
 
 def note_sequence_to_velocity_sequence(noteSequence, timespanTicks = None):

 			self[-1].set_next_bar(bar)
 		super(BarList, self).append(bar)
 
 
 class Bar:
-
 	def __init__(self, rhythmSequence, timeSignature, ticksPerQuarter=None, qpmTempo=None, nextBar=None, prevBar=None):
 		if isinstance(rhythmSequence, NoteSequence):
 			self.noteSequence = rhythmSequence
 			self.velocitySequence = None 
 		elif isinstance(rhythmSequence, VelocitySequence):

 		self.nextBar = nextBar
 		self.prevBar = prevBar
 
 	def get_note_sequence(self):
 		if self.noteSequence == None:
-			self.noteSequence = velocitySequence_to_noteSequence(self.velocitySequence)
+			nextbarVelocitySequence = None
+			if self.nextBar != None:
+				nextbarVelocitySequence = self.nextBar.get_velocity_sequence()
+			self.noteSequence = velocity_sequence_to_note_sequence(self.velocitySequence, nextbarVelocitySequence)
 		return self.noteSequence
 
 	def get_velocity_sequence(self):
 		if self.velocitySequence == None:
-			self.velocitySequence = noteSequence_to_velocitySequence(self.noteSequence)
+			self.velocitySequence = note_sequence_to_velocity_sequence(self.noteSequence)
 		return self.velocitySequence
 
 	def get_binary_sequence(self):
 		return ceiling(self.get_velocity_sequence())
 

 	def get_time_signature(self):
 		return self.timeSignature
 
 	# return the length of a bar in time units (ticks)
 	def get_bar_ticks(self):
-		return calculate_bar_ticks(self.timeSignature.get_numerator(),self.timeSignature.get_denominator(), self.ticksPerQuarter)
+		return calculate_bar_ticks(self.timeSignature.get_numerator(),self.timeSignature.get_denominator(), self.tpq)
 
 
 class TimeSignature():
 	def __init__(self, inputString):
 		if inputString in parameter_setter.read_time_signature():