comparison novelty.py @ 1:c11ea9e0357f

adding funcs

 """
 
 import sys, os
 import numpy as np
 from scipy.signal import correlate2d, convolve2d
+import matplotlib.pyplot as plt
 
 # from utils.PeakPickerUtil import PeakPicker
 
 def getNoveltyCurve(ssm, kernel_size, normalise=False):
 	'''Return novelty score from ssm.'''

 			else:
 				kernel[i-1,j-1] = gauss
 				
 	return kernel	
 	
-def getNoveltyPeaks(ssm, kernel_size, peak_picker, normalise=False):
+def process(ssm, kernel_size, peak_picker, normalise=False, plot=False):
 	'''Detect segment boundaries in the ssm.'''
 	novelty = getNoveltyCurve(ssm, kernel_size, normalise=False)
 	smoothed_novelty, novelty_peaks = peak_picker.process(novelty)
 	
+	if plot:
+		plot_detection(smoothed_novelty, novelty_peaks)
 	return novelty, smoothed_novelty, novelty_peaks
+	
+def plot_detection(smoothed_novelty, novelty_peaks):
+	pass